chore: improve ai agent

Signed-off-by: Zzde <zhangxh1997@gmail.com>

Author: zxh326

pkg/ai/anthropic.go

@@ -56,14 +56,13 @@ func (a *Agent) continueChatAnthropic(c *gin.Context, session pendingSession, se
 		toolResult = "Tool error: " + result
 	}
 
-	messages := append([]anthropic.MessageParam(nil), session.AnthropicMessages...)
-	messages = append(
-		messages,
+	session.AnthropicMessages = append(
+		session.AnthropicMessages,
 		anthropic.NewUserMessage(
 			anthropic.NewToolResultBlock(session.ToolCall.ID, toolResult, isError),
 		),
 	)
-	a.runAnthropicConversation(ctx, c, session.SystemPrompt, messages, sendEvent)
+	a.runAnthropicConversation(ctx, c, session.SystemPrompt, session.AnthropicMessages, sendEvent)
 	return nil
 }
 

pkg/ai/openai.go

@@ -58,9 +58,8 @@ func (a *Agent) continueChatOpenAI(c *gin.Context, session pendingSession, sendE
 		result = "Tool error: " + result
 	}
 
-	messages := append([]openai.ChatCompletionMessageParamUnion(nil), session.OpenAIMessages...)
-	messages = append(messages, openai.ToolMessage(result, session.ToolCall.ID))
-	a.runOpenAIConversation(ctx, c, messages, sendEvent)
+	session.OpenAIMessages = append(session.OpenAIMessages, openai.ToolMessage(result, session.ToolCall.ID))
+	a.runOpenAIConversation(ctx, c, session.OpenAIMessages, sendEvent)
 	return nil
 }