oadmin/new-api
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: xAI missing finish_reason #572

Browse Source
This commit is contained in:
CalciumIon
2024-11-30 16:57:31 +08:00
parent 9c4d30602c
commit 1774be8536
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
relay/channel/openai/relay-openai.go
View File

@@ -98,6 +98,11 @@ func OaiStreamHandler(c *gin.Context, resp *http.Response, info *relaycommon.Rel
shouldSendLastResp = false shouldSendLastResp = false
} }
} }
for _, choice := range lastStreamResponse.Choices {
if choice.FinishReason != nil {
shouldSendLastResp = true
}
}
} }
if shouldSendLastResp { if shouldSendLastResp {
service.StringData(c, lastStreamData) service.StringData(c, lastStreamData)