From edcdb378fd8e2a8449ad294912ed4220c8a877cb Mon Sep 17 00:00:00 2001
From: xqx121 <78908927+xqx121@users.noreply.github.com>
Date: Wed, 18 Jun 2025 14:26:23 +0800
Subject: [PATCH] Update relay-gemini-native.go

---
 relay/channel/gemini/relay-gemini-native.go | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/relay/channel/gemini/relay-gemini-native.go b/relay/channel/gemini/relay-gemini-native.go
index d9d0054d..b57638df 100644
--- 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)