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

Fix assistant context switching and ID handling

Browse source 
- Prevent switching to an empty assistant ID
- Update context assistant ID during execution
- Ensure proper assistant context management in chat flow
This commit is contained in:
Max 2025-02-09 18:25:09 +08:00
parent ccb1c54927
commit dc6967805e
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
neo/assistant/api.go
View file

@ -95,7 +95,7 @@ func (ast *Assistant) execute(c *gin.Context, ctx chatctx.Context, input []chatM
// } // }
// Switch to the new assistant if necessary // Switch to the new assistant if necessary
if res != nil && res.AssistantID != ctx.AssistantID { if res != nil && res.AssistantID != "" && res.AssistantID != ctx.AssistantID {
newAst, err := Get(res.AssistantID) newAst, err := Get(res.AssistantID)
if err != nil { if err != nil {
chatMessage.New(). chatMessage.New().
@ -244,6 +244,8 @@ func (next *NextAction) Execute(c *gin.Context, ctx chatctx.Context, contents *c
msg.Write(c.Writer) msg.Write(c.Writer)
newContents := chatMessage.NewContents() newContents := chatMessage.NewContents()
// Update the context id
ctx.AssistantID = assistant.ID
return assistant.execute(c, ctx, messages, options, newContents) return assistant.execute(c, ctx, messages, options, newContents)
case "exit": case "exit":