Merge branch 'main' of github.com:xqx121/new-api into xqx121-main

2025-06-19 14:45:41 +08:00
parent 2fdb2be6d0 edcdb378fd
commit 5b54624cd5
1 changed files with 3 additions and 0 deletions
--- a/relay/channel/gemini/relay-gemini-native.go
+++ b/relay/channel/gemini/relay-gemini-native.go
@@ -64,6 +64,9 @@ func GeminiTextGenerationHandler(c *gin.Context, resp *http.Response, info *rela
 			usage.PromptTokensDetails.TextTokens = detail.TokenCount
 		}
 	}
+	
+	// 计算最终使用量
+	usage.CompletionTokens = usage.TotalTokens - usage.PromptTokens

 	// 直接返回 Gemini 原生格式的 JSON 响应
 	jsonResponse, err := json.Marshal(geminiResponse)