ChatGPT/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2025-10-19 10:07:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: calculate usage if not given in non-stream mode (#352)

Browse Source
This commit is contained in:
glzjin
2023-08-06 17:40:31 +08:00
committed by GitHub
parent 1dfa190e79
commit 446337c329
3 changed files with 17 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
controller/relay-text.go
View File

@@ -362,7 +362,7 @@ func relayTextHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode {
textResponse.Usage.CompletionTokens = countTokenText(responseText, textRequest.Model)
return nil
} else {
err, usage := openaiHandler(c, resp, consumeQuota)
err, usage := openaiHandler(c, resp, consumeQuota, promptTokens, textRequest.Model)
if err != nil {
return err
}