From dc6967805ebf184cd64fdd3396fb4f19e8998ceb Mon Sep 17 00:00:00 2001
From: Max <shadow.iqka@qq.com>
Date: Sun, 9 Feb 2025 18:25:09 +0800
Subject: [PATCH] Fix assistant context switching and ID handling

- Prevent switching to an empty assistant ID
- Update context assistant ID during execution
- Ensure proper assistant context management in chat flow
---
 neo/assistant/api.go | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/neo/assistant/api.go b/neo/assistant/api.go
index a2883121..ed0e9266 100644
--- a/neo/assistant/api.go
+++ b/neo/assistant/api.go
@@ -95,7 +95,7 @@ func (ast *Assistant) execute(c *gin.Context, ctx chatctx.Context, input []chatM
 	// }
 
 	// 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)
 		if err != nil {
 			chatMessage.New().
@@ -244,6 +244,8 @@ func (next *NextAction) Execute(c *gin.Context, ctx chatctx.Context, contents *c
 		msg.Write(c.Writer)
 		newContents := chatMessage.NewContents()
 
+		// Update the context id
+		ctx.AssistantID = assistant.ID
 		return assistant.execute(c, ctx, messages, options, newContents)
 
 	case "exit":