Procházet zdrojové kódy

Update relay-gemini-native.go

xqx121 před 8 měsíci
rodič
revize
edcdb378fd
1 změnil soubory, kde provedl 3 přidání a 0 odebrání
  1. 3 0
      relay/channel/gemini/relay-gemini-native.go

+ 3 - 0
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)