|
@@ -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)))
|