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

Revert "feat: update Usage struct to support dynamic token handling with ceil function #1503"

Browse Source 
This reverts commit 71c39c9893.
This commit is contained in:
CaIon
2025-08-07 16:22:40 +08:00
parent d9c1fb5244
commit 865bb7aad8
3 changed files with 11 additions and 124 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -570,11 +570,11 @@ func OpenaiHandlerWithUsage(c *gin.Context, info *relaycommon.RelayInfo, resp *h
// because the upstream has already consumed resources and returned content
// We should still perform billing even if parsing fails
// format
if usageResp.GetInputTokens() > 0 {
usageResp.PromptTokens += usageResp.GetInputTokens()
if usageResp.InputTokens > 0 {
usageResp.PromptTokens += usageResp.InputTokens
}
if usageResp.GetOutputTokens() > 0 {
usageResp.CompletionTokens += usageResp.GetOutputTokens()
if usageResp.OutputTokens > 0 {
usageResp.CompletionTokens += usageResp.OutputTokens
}
if usageResp.InputTokensDetails != nil {
usageResp.PromptTokensDetails.ImageTokens += usageResp.InputTokensDetails.ImageTokens