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 {
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) {