disable thinking budget validation on gemini

This commit is contained in:
Nopm
2025-06-18 17:11:39 -03:00
parent cb8f2669ac
commit 4c214305af
+3 -12
View File
@@ -189,20 +189,11 @@ googleAIRouter.get("/:apiVersion(v1alpha|v1beta)/models", handleNativeModelReque
/** /**
* Processes the thinking budget for Gemini 2.5 Flash model. * Processes the thinking budget for Gemini 2.5 Flash model.
* Ensures the budget is within the valid range of 0-24576 tokens. * Validation has been disabled - budget is passed through without limits.
*/ */
function processThinkingBudget(req: Request) { function processThinkingBudget(req: Request) {
if (req.body.generationConfig?.thinkingConfig?.thinkingBudget !== undefined) { // Validation disabled - budget is passed through without any range limits
// Ensure thinkingBudget is within allowed range (0-24576) // Previously enforced 0-24576 token limit
const budget = req.body.generationConfig.thinkingConfig.thinkingBudget;
// If it's a number, validate range
if (typeof budget === 'number') {
req.body.generationConfig.thinkingConfig.thinkingBudget =
Math.max(0, Math.min(budget, 24576));
}
// If it's "auto", leave as is
}
} }
function setStreamFlag(req: Request) { function setStreamFlag(req: Request) {