Jelajahi Sumber

文章增加系统DFA标签

kobe6258 4 hari lalu
induk
melakukan
980226353f
2 mengubah file dengan 3 tambahan dan 1 penghapusan
  1. 2 1
      utils/llm/eta_llm/eta_llm_client.go
  2. 1 0
      utils/ws/session_manager.go

+ 2 - 1
utils/llm/eta_llm/eta_llm_client.go

@@ -133,13 +133,14 @@ func (ds *ETALLMClient) CompletionChat(query string, messages []json.RawMessage)
 		TopK:           3,
 		ScoreThreshold: 0.5,
 		Stream:         true,
-		Model:         "qwq:32b",
+		Model:          "qwq:32b",
 		Temperature:    0.7,
 		MaxTokens:      0,
 		PromptName:     DEFALUT_PROMPT_NAME,
 		ReturnDirect:   false,
 	}
 	fmt.Printf("%v", kbReq.Messages)
+	utils.FileLog.Info("kbReq.Messages: %v", kbReq.Messages)
 	body, err := json.Marshal(kbReq)
 	if err != nil {
 		return

+ 1 - 0
utils/ws/session_manager.go

@@ -131,6 +131,7 @@ func (manager *ConnectionManager) HandleMessage(userID int, sessionID string, me
 			utils.FileLog.Warn("没有搜索到相关的研报内容,执行RAG对话 ")
 			resp, err = llmService.KnowledgeBaseChat(userMessage.Query, userMessage.KbName, session.History)
 		} else {
+			utils.FileLog.Info("搜索到相关的研报内容,执行completion对话 ")
 			//直接对话,不需要走RAG
 			var articlesContents string
 			articlesCounts := len(articles)