From 785714fc464253a4cc3d01ad9b71d837986e0b83 Mon Sep 17 00:00:00 2001
From: junleea <354425203@qq.com>
Date: Mon, 12 May 2025 12:16:36 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E8=85=BE=E8=AE=AF=E6=B7=B7?=
 =?UTF-8?q?=E5=85=83=E5=A4=A7=E6=A8=A1=E5=9E=8B=E6=94=AF=E6=8C=81,?=
 =?UTF-8?q?=E4=BF=AE=E6=94=B9openai=E5=BE=AA=E7=8E=AF=E5=BC=95=E7=94=A8?=
 =?UTF-8?q?=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 service/generalModel/openai.go | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/service/generalModel/openai.go b/service/generalModel/openai.go
index 07f4c04..0c2ab57 100644
--- a/service/generalModel/openai.go
+++ b/service/generalModel/openai.go
@@ -3,7 +3,6 @@ package generalModel
 import (
 	"StuAcaWorksAI/dao"
 	"StuAcaWorksAI/proto"
-	"StuAcaWorksAI/service"
 	"StuAcaWorksAI/worker"
 	"context"
 	"encoding/json"
@@ -47,7 +46,12 @@ func OpenAI(modelParam proto.ModelParam, imCtx *proto.IMParamContext, platform s
 
 	for stream.Next() {
 		chunk := stream.Current()
-		go service.CreateAIStreamMsg(imCtx.UserID, imCtx.ModelID, imCtx.SessionID, chunk, modelParam.Model)
+		msgStr, err3 := json.Marshal(chunk)
+		if err3 != nil {
+			log.Println("openai stream msg Error parsing JSON:", err)
+		}
+		//go service.CreateAIStreamMsg(imCtx.UserID, imCtx.ModelID, imCtx.SessionID, chunk, modelParam.Model)
+		go dao.InsertAIStreamMsgToDB(imCtx.UserID, imCtx.SessionID, uint(imCtx.ModelID), string(msgStr), modelParam.Model) //存入数据库
 		acc.AddChunk(chunk)
 		if _, ok := acc.JustFinishedToolCall(); ok {
 			chunk.Created = -1 // 这里由于没有返回结束标志，所以用该字段来判断