Version: 0.4.5.dev.260307

feat: 📡 更新 SSE 消息流格式

* 将 SSE 消息流格式更新为 Apifox 可识别的 OpenAI 格式
* 便于后续与前端的对接与协作

backend/service/agent.go

@@ -9,6 +9,7 @@ import (
 	"github.com/LoveLosita/smartflow/backend/conv"
 	"github.com/LoveLosita/smartflow/backend/dao"
 	"github.com/LoveLosita/smartflow/backend/inits"
+	"github.com/cloudwego/eino-ext/components/model/ark"
 	"github.com/cloudwego/eino/schema"
 	"github.com/google/uuid"
 )
@@ -35,15 +36,24 @@ func normalizeConversationID(chatID string) string {
 	return trimmed
 }
 
-func (s *AgentService) AgentChat(ctx context.Context, userMessage string, ifThinking bool, userID int, chatID string) (<-chan string, <-chan error) {
-	//1. 创建一个输出通道
+func (s *AgentService) pickChatModel(requestModel string) (*ark.ChatModel, string) {
+	model := strings.TrimSpace(requestModel)
+	if strings.EqualFold(model, "strategist") {
+		return s.AIHub.Strategist, "strategist"
+	}
+	return s.AIHub.Worker, "worker"
+}
+
+func (s *AgentService) AgentChat(ctx context.Context, userMessage string, ifThinking bool, modelName string, userID int, chatID string) (<-chan string, <-chan error) {
+	// 1) 准备输出通道
 	outChan := make(chan string, 5)
 	errChan := make(chan error, 1)
-	//补充：会话 ID 兜底，避免上层漏传
-	chatID = normalizeConversationID(chatID)
 
-	//2. 先确保这个会话存在（如果不存在就创建一个新的）
-	//先看看缓存里面有没有这个会话
+	// 2) 规范化会话并选择模型
+	chatID = normalizeConversationID(chatID)
+	selectedModel, resolvedModelName := s.pickChatModel(modelName)
+
+	// 3) 确保会话存在
 	result, err := s.agentCache.GetConversationStatus(ctx, chatID)
 	if err != nil {
 		errChan <- err
@@ -51,7 +61,6 @@ func (s *AgentService) AgentChat(ctx context.Context, userMessage string, ifThin
 		close(errChan)
 		return outChan, errChan
 	}
-	//如果缓存里面没有，就去查库
 	if !result {
 		innerResult, err := s.repo.IfChatExists(ctx, userID, chatID)
 		if err != nil {
@@ -61,7 +70,6 @@ func (s *AgentService) AgentChat(ctx context.Context, userMessage string, ifThin
 			return outChan, errChan
 		}
 		if !innerResult {
-			//如果会话不存在，先创建一个新的会话
 			if _, err = s.repo.CreateNewChat(userID, chatID); err != nil {
 				errChan <- err
 				close(outChan)
@@ -69,15 +77,12 @@ func (s *AgentService) AgentChat(ctx context.Context, userMessage string, ifThin
 				return outChan, errChan
 			}
 		}
-		//补充：把“会话存在”状态回写缓存，后续请求可直接命中
 		if err = s.agentCache.SetConversationStatus(ctx, chatID); err != nil {
-			//缓存回写失败不影响主流程
 			log.Printf("failed to set conversation status cache for %s: %v", chatID, err)
 		}
 	}
-	//能走到这里，要么缓存里有这个会话，要么数据库里有这个会话了
-	//4. 提取出历史消息，构建上下文
-	//先尝试从缓存里拿历史消息
+
+	// 4) 构建历史上下文
 	chatHistory, err := s.agentCache.GetHistory(ctx, chatID)
 	if err != nil {
 		errChan <- err
@@ -85,9 +90,7 @@ func (s *AgentService) AgentChat(ctx context.Context, userMessage string, ifThin
 		close(errChan)
 		return outChan, errChan
 	}
-	//如果缓存里没有历史消息，就从数据库里拿
 	if chatHistory == nil {
-		//先从数据库拿到历史消息
 		histories, err := s.repo.GetUserChatHistories(ctx, userID, 20, chatID)
 		if err != nil {
 			errChan <- err
@@ -95,9 +98,7 @@ func (s *AgentService) AgentChat(ctx context.Context, userMessage string, ifThin
 			close(errChan)
 			return outChan, errChan
 		}
-		//再转换成 Eino 的消息格式
 		chatHistory = conv.ToEinoMessages(histories)
-		//把历史消息放到缓存里，方便下次直接拿
 		if err = s.agentCache.BackfillHistory(ctx, chatID, chatHistory); err != nil {
 			errChan <- err
 			close(outChan)
@@ -105,9 +106,9 @@ func (s *AgentService) AgentChat(ctx context.Context, userMessage string, ifThin
 			return outChan, errChan
 		}
 	}
-	//3. 将用户消息异步落缓存和库
+
+	// 5) 异步落用户消息
 	go func() {
-		//这里先不管落库成功与否了，毕竟不想因为落库失败而影响用户的聊天体验
 		bg := context.Background()
 		_ = s.agentCache.PushMessage(bg, chatID, &schema.Message{
 			Role:    schema.User,
@@ -116,17 +117,17 @@ func (s *AgentService) AgentChat(ctx context.Context, userMessage string, ifThin
 		_ = s.repo.SaveChatHistory(bg, userID, chatID, "user", userMessage)
 	}()
 
-	//5. 启动一个 goroutine 来处理聊天逻辑
+	// 6) 流式输出模型回复
 	go func() {
-		defer close(outChan) // 确保在函数结束时关闭通道
-		defer close(errChan)
-		//3. 调用 StreamChat 函数进行流式聊天
-		fullText, err := agent.StreamChat(ctx, s.AIHub.Worker, userMessage, ifThinking, chatHistory, outChan)
+		defer close(outChan)
+
+		fullText, err := agent.StreamChat(ctx, selectedModel, resolvedModelName, userMessage, ifThinking, chatHistory, outChan)
 		if err != nil {
 			errChan <- err
 			return
 		}
-		//4. 将 AI 的回复异步落缓存和库
+
+		// 7) 异步落助手消息
 		go func() {
 			bg := context.Background()
 			_ = s.agentCache.PushMessage(bg, chatID, &schema.Message{
@@ -135,9 +136,9 @@ func (s *AgentService) AgentChat(ctx context.Context, userMessage string, ifThin
 			})
 			if saveErr := s.repo.SaveChatHistory(bg, userID, chatID, "assistant", fullText); saveErr != nil {
 				log.Printf("failed to save chat history to database: %v", saveErr)
-				return
 			}
 		}()
 	}()
+
 	return outChan, errChan
 }