From e0e7f10848aa2b524a97728da9a862bf70eb3845 Mon Sep 17 00:00:00 2001
From: junleea <354425203@qq.com>
Date: Fri, 11 Apr 2025 16:35:36 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9im=E6=8E=A5=E6=94=B6=E6=B6=88?=
 =?UTF-8?q?=E6=81=AF=E5=AD=98=E5=85=A5=E6=95=B0=E6=8D=AE=E5=BA=93?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 handler/im.go        | 2 +-
 service/imService.go | 2 ++
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/handler/im.go b/handler/im.go
index 44056da..c103739 100644
--- a/handler/im.go
+++ b/handler/im.go
@@ -348,7 +348,7 @@ func subscribeAndHandleKBaseTextMessages(ws *websocket.Conn, userId int, kBaseSe
 				log.Println("Error parsing JSON:", err1)
 				continue
 			}
-			service.CreateAIStreamMsg(userId, data.IMContext.ModelID, data.IMContext.SessionID, message, proto.KnowledgeBaseServerResponseType) //将消息存入数据库
+			service.CreateAIStreamMsg(userId, data.IMContext.ModelID, data.IMContext.SessionID, data, proto.KnowledgeBaseServerResponseType) //将消息存入数据库
 			//这里主要将接收到的消息转到对应协程
 			//目标chanel
 			targetChanel := fmt.Sprintf("kbase_%s", data.KnowledgeBase.UUID)
diff --git a/service/imService.go b/service/imService.go
index 43794db..9624110 100644
--- a/service/imService.go
+++ b/service/imService.go
@@ -330,6 +330,8 @@ func SelectKBaseContentByQuery(imCtx *proto.IMParamContext, data *proto.WSMessag
 		if err2 != nil {
 			log.Println("Error parsing JSON:", err2)
 		}
+		//讲消息存入数据库
+		CreateAIStreamMsg()
 		break
 	case <-timer.C:
 		log.Println("kbase query timeout")