ljydlv
/
saw-go
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

修复通义千问的停止问题

This commit is contained in:
junleea 2025-06-04 23:36:25 +08:00
parent c809a4e4c7
commit 3f7c6b5ed1
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
service/tongyi.go
View File

@ -61,10 +61,10 @@ func Tongyi(modelParam proto.ModelParam, imCtx *proto.IMParamContext) {
log.Println("tongyi stream err:", stream.Err()) 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} 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) err2, msgID := dao.CreateMessageV2(msg)
if err2 != nil { if err2 != nil {
@ -132,7 +132,7 @@ func TongyiToGeneralMassageAndSendMsgQueue(data *openai.ChatCompletionChunk, cha
var apiMsg api.GenerateResponse var apiMsg api.GenerateResponse
apiMsg.Model = proto.ModelTypeSpark apiMsg.Model = proto.ModelTypeSpark
apiMsg.Response = data.Choices[0].Delta.Content apiMsg.Response = data.Choices[0].Delta.Content
if data.Created < 0 { //这里由于没有返回结束标志，所以用该字段来判断 if data.Created < 0 || data.Choices[0].FinishReason != "" { //这里由于没有返回结束标志，所以用该字段来判断
apiMsg.Done = true apiMsg.Done = true
} else { } else {
apiMsg.Done = false apiMsg.Done = false