xinyin025/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2025-11-21 07:26:47 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge branch 'main' into fork/zijiren233/goroutine

Browse Source
This commit is contained in:
JustSong
2024-06-30 16:36:10 +08:00
parent c7aa38e428 f25aaf7752
commit 1923ded809
7 changed files with 16 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
relay/adaptor/openai/main.go
View File

@@ -130,7 +130,7 @@ func Handler(c *gin.Context, resp *http.Response, promptTokens int, modelName st
return ErrorWrapper(err, "close_response_body_failed", http.StatusInternalServerError), nil
}
if textResponse.Usage.TotalTokens == 0 {
if textResponse.Usage.TotalTokens == 0 || (textResponse.Usage.PromptTokens == 0 && textResponse.Usage.CompletionTokens == 0) {
completionTokens := 0
for _, choice := range textResponse.Choices {
completionTokens += CountTokenText(choice.Message.StringContent(), modelName)