diff --git a/core/providers/google/src/models/chat-models/base-chat-model.google.ts b/core/providers/google/src/models/chat-models/base-chat-model.google.ts index beb5d5a..4223a24 100644 --- a/core/providers/google/src/models/chat-models/base-chat-model.google.ts +++ b/core/providers/google/src/models/chat-models/base-chat-model.google.ts @@ -332,6 +332,10 @@ class BaseChatModel implements ChatModelV1 { acc.thinkingConfig = { includeThoughts: paramValue, }; + } else if (key === "maxReasoningTokens") { + acc.thinkingConfig = acc.thinkingConfig && typeof acc.thinkingConfig === "object" + ? { ...acc.thinkingConfig, thinkingBudget: paramValue } + : { thinkingBudget: paramValue }; } else if (paramKey === "maxOutputTokens" && def.type === "range" && paramValue === 0) { acc[paramKey] = def.max; } else {