oadmin/new-api
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: update budget calculation logic in relay-gemini to use clamping function

Browse Source
This commit is contained in:
CaIon
2025-08-06 16:25:48 +08:00
parent feef022303
commit f46cefbd39
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
relay/channel/gemini/relay-gemini.go
View File

@@ -107,13 +107,13 @@ func clampThinkingBudgetByEffort(modelName string, effort string) int {
} }
switch effort { switch effort {
case "high": case "high":
return maxBudget * 80 / 100 maxBudget = maxBudget * 80 / 100
case "medium": case "medium":
return maxBudget * 50 / 100 maxBudget = maxBudget * 50 / 100
case "low": 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) { func ThinkingAdaptor(geminiRequest *dto.GeminiChatRequest, info *relaycommon.RelayInfo, oaiRequest ...dto.GeneralOpenAIRequest) {