Ecosystem/ai-gateway
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

修复百度计费问题

Browse Source
This commit is contained in:
glzjin 2023-08-18 23:13:51 +08:00 committed by GitHub
parent 2045c07195
commit 16b387eaaf
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
controller/relay-baidu.go
View File

@ -215,9 +215,11 @@ func baiduStreamHandler(c *gin.Context, resp *http.Response) (*OpenAIErrorWithSt
common.SysError("error unmarshalling stream response: " + err.Error()) common.SysError("error unmarshalling stream response: " + err.Error())
return true return true
} }
usage.PromptTokens += baiduResponse.Usage.PromptTokens if baiduResponse.Usage.TotalTokens != 0 {
usage.CompletionTokens += baiduResponse.Usage.CompletionTokens usage.TotalTokens = baiduResponse.Usage.TotalTokens
usage.TotalTokens += baiduResponse.Usage.TotalTokens usage.PromptTokens = baiduResponse.Usage.PromptTokens
usage.CompletionTokens = baiduResponse.Usage.TotalTokens - baiduResponse.Usage.PromptTokens
}
response := streamResponseBaidu2OpenAI(&baiduResponse) response := streamResponseBaidu2OpenAI(&baiduResponse)
jsonResponse, err := json.Marshal(response) jsonResponse, err := json.Marshal(response)
if err != nil { if err != nil {