From f46cefbd392612190679a51b43f68cd4c96989b7 Mon Sep 17 00:00:00 2001 From: CaIon Date: Wed, 6 Aug 2025 16:25:48 +0800 Subject: [PATCH] fix: update budget calculation logic in relay-gemini to use clamping function --- relay/channel/gemini/relay-gemini.go | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/relay/channel/gemini/relay-gemini.go b/relay/channel/gemini/relay-gemini.go index 0f4a54cf..18524afb 100644 --- a/relay/channel/gemini/relay-gemini.go +++ b/relay/channel/gemini/relay-gemini.go @@ -107,13 +107,13 @@ func clampThinkingBudgetByEffort(modelName string, effort string) int { } switch effort { case "high": - return maxBudget * 80 / 100 + maxBudget = maxBudget * 80 / 100 case "medium": - return maxBudget * 50 / 100 + maxBudget = maxBudget * 50 / 100 case "low": - return maxBudget * 20 / 100 + maxBudget = maxBudget * 20 / 100 } - return maxBudget * 50 / 100 // 默认medium + return clampThinkingBudget(modelName, maxBudget) } func ThinkingAdaptor(geminiRequest *dto.GeminiChatRequest, info *relaycommon.RelayInfo, oaiRequest ...dto.GeneralOpenAIRequest) {