disable thinking budget validation on gemini
This commit is contained in:
+3
-12
@@ -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) {
|
||||||
|
|||||||
Reference in New Issue
Block a user