Skip to content

phsl / new-api

Go to a project
Commit 3822f457 by CaIon
Options

fix(audio): correct TotalTokens calculation for accurate usage reporting

parent be2a863b
Showing with 1 additions and 1 deletions
relay/channel/openai/audio.go
......@@ -91,7 +91,6 @@ func OpenaiTTSHandler(c *gin.Context, resp *http.Response, info *relaycommon.Rel
}
usage.PromptTokensDetails.TextTokens = usage.PromptTokens
usage.TotalTokens = usage.PromptTokens + usage.CompletionTokens
if durationErr != nil {
logger.LogWarn(c, fmt.Sprintf("failed to get audio duration: %v", durationErr))
......@@ -106,6 +105,7 @@ func OpenaiTTSHandler(c *gin.Context, resp *http.Response, info *relaycommon.Rel
usage.CompletionTokens = completionTokens
usage.CompletionTokenDetails.AudioTokens = completionTokens
}
usage.TotalTokens = usage.PromptTokens + usage.CompletionTokens
}
return usage
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or sign in to comment