admin777/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
Some checks failed
Publish Docker image (Multi Registries) / Push Docker image to multiple registries (push) Has been cancelled
Details
Linux Release / release (push) Has been cancelled
Details
macOS Release / release (push) Has been cancelled
Details
Windows Release / release (push) Has been cancelled
Details

Browse Source
This commit is contained in:
CaIon 2025-08-06 16:25:48 +08:00
parent 6960a06322
commit 2e41362f2e
1 changed files with 4 additions and 4 deletions
Show Stats 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) {