kobe6258 1 month ago
parent
commit
3fe38bde3e
1 changed files with 5 additions and 3 deletions
  1. 5 3
      utils/ws/session_manager.go

+ 5 - 3
utils/ws/session_manager.go

@@ -66,15 +66,17 @@ func (manager *ConnectionManager) HandleMessage(userID int, sessionID string, me
 	var userMessage Message
 	err := json.Unmarshal(message, &userMessage)
 	if err != nil {
-		return errors.New("消息格式错误")
+		fmt.Printf("消息格式错误:%s", string(message))
+		return errors.New("消息格式错误:" + err.Error())
 	}
 	// 处理业务逻辑
 	session.History = append(session.History, userMessage.LastTopics...)
 
-	//TODO
 	resp, err := llmService.KnowledgeBaseChat(userMessage.Query, userMessage.KbName, session.History)
 	defer func() {
-		_ = resp.Body.Close()
+		if resp != nil && resp.Body != nil {
+			_ = resp.Body.Close()
+		}
 	}()
 	if err != nil {
 		err = errors.New(fmt.Sprintf("知识库问答失败: httpCode:%d,错误信息:%s", resp.StatusCode, http.StatusText(resp.StatusCode)))