Merge branch 'refs/heads/feat-knowledgebase'

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)

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