From 270d3c3535ed44e4d6311fc703020eb9c6a3c0ab Mon Sep 17 00:00:00 2001
From: junleea <354425203@qq.com>
Date: Tue, 13 May 2025 13:21:08 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E5=9B=BD=E5=A4=96=E6=9C=8D?=
 =?UTF-8?q?=E5=8A=A1=E5=99=A8=E4=B8=AD=E8=BD=ACai=E8=AF=B7=E6=B1=82?=
 =?UTF-8?q?=E6=95=B0=E6=8D=AE,=E6=B7=BB=E5=8A=A0gemini?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 service/generalModel/openai.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/service/generalModel/openai.go b/service/generalModel/openai.go
index 3f7dab1..6e61ab4 100644
--- a/service/generalModel/openai.go
+++ b/service/generalModel/openai.go
@@ -310,7 +310,7 @@ func OpenAIForeignAIResponseToGeneralMassageAndSendMsgQueue(data *proto.ReceiveM
 		//PromptTokens = chunk.Usage.PromptTokens
 	}
 	//将消息存入数据库
-	if chunk.Created < 0 { //这里由于没有返回结束标志，所以用该字段来判断
+	if chunk.Created < 0 || chunk.Choices[0].FinishReason != "" { //这里由于没有返回结束标志，所以用该字段来判断
 		tokens := chunk.Usage.TotalTokens
 		PromptTokens := chunk.Usage.PromptTokens
 		//将消息存入数据库