opencode/yao
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

[_]

Browse source
This commit is contained in:
Max 2023-10-20 14:24:28 +08:00
parent bd518610b9
commit ef941450ac
1 changed files with 9 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
neo/neo.go
View file

@ -248,13 +248,13 @@ func (neo *DSL) Answer(ctx command.Context, question string, answer Answer) erro
return true return true
} }
err := neo.write(msg, w, ctx, messages) content = msg.Append(content)
err := neo.write(msg, w, ctx, messages, content)
if err != nil { if err != nil {
log.Warn("Neo write process msg: %v error: %s", msg, err.Error()) log.Warn("Neo write process msg: %v error: %s", msg, err.Error())
msg.Write(w) msg.Write(w)
} }
content = msg.Append(content)
return !msg.IsDone() return !msg.IsDone()
case <-ctx.Done(): case <-ctx.Done():
@ -310,14 +310,19 @@ func (neo *DSL) prompts() []map[string]interface{} {
} }
// after the after hook // after the after hook
func (neo *DSL) write(msg *message.JSON, w io.Writer, ctx command.Context, messages []map[string]interface{}) error { func (neo *DSL) write(msg *message.JSON, w io.Writer, ctx command.Context, messages []map[string]interface{}, content []byte) error {
if neo.Write == "" { if neo.Write == "" {
msg.Write(w) msg.Write(w)
return nil return nil
} }
p, err := process.Of(neo.Write, ctx, messages, msg) args := []interface{}{ctx, messages, msg}
if msg.IsDone() {
args = append(args, string(content))
}
p, err := process.Of(neo.Write, args...)
if err != nil { if err != nil {
return err return err
} }