Преглед изворни кода

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

CaIon пре 3 недеља
родитељ
комит
7cae4a640b
1 измењених фајлова са 1 додато и 1 уклоњено
  1. 1 1
      relay/channel/openai/audio.go

+ 1 - 1
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