From 3f7c6b5ed17f756400e8fc0239d087d916d89e42 Mon Sep 17 00:00:00 2001
From: junleea <354425203@qq.com>
Date: Wed, 4 Jun 2025 23:36:25 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E9=80=9A=E4=B9=89=E5=8D=83?=
 =?UTF-8?q?=E9=97=AE=E7=9A=84=E5=81=9C=E6=AD=A2=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 service/tongyi.go | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/service/tongyi.go b/service/tongyi.go
index 7f7e50f..b234b49 100644
--- a/service/tongyi.go
+++ b/service/tongyi.go
@@ -61,10 +61,10 @@ func Tongyi(modelParam proto.ModelParam, imCtx *proto.IMParamContext) {
 		log.Println("tongyi stream err:", stream.Err())
 	}
 	//将消息存入数据库
-	DOSaveMessageAndToken(imCtx, answer, "tongyi", tokens, PromptTokens)
+	DoSaveMessageAndToken(imCtx, answer, "tongyi", tokens, PromptTokens)
 }
 
-func DOSaveMessageAndToken(imCtx *proto.IMParamContext, answer, platform string, tokens, PromptTokens int64) {
+func DoSaveMessageAndToken(imCtx *proto.IMParamContext, answer, platform string, tokens, PromptTokens int64) {
 	msg := dao.Message{SessionID: imCtx.SessionID, FromID: imCtx.UserID, ToID: imCtx.ModelID, Msg: answer, Type: proto.ModelToUserMsgType, FunctionID: imCtx.FunctionID, Token: int(tokens), Status: proto.UserToModelMsgType}
 	err2, msgID := dao.CreateMessageV2(msg)
 	if err2 != nil {
@@ -132,7 +132,7 @@ func TongyiToGeneralMassageAndSendMsgQueue(data *openai.ChatCompletionChunk, cha
 	var apiMsg api.GenerateResponse
 	apiMsg.Model = proto.ModelTypeSpark
 	apiMsg.Response = data.Choices[0].Delta.Content
-	if data.Created < 0 { //这里由于没有返回结束标志，所以用该字段来判断
+	if data.Created < 0 || data.Choices[0].FinishReason != "" { //这里由于没有返回结束标志，所以用该字段来判断
 		apiMsg.Done = true
 	} else {
 		apiMsg.Done = false