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 //将消息存入数据库