fix: use valid thinking_budget (8192) instead of -1 for Vertex AI compatibility

This commit is contained in:
Greyson LaLonde
2026-03-17 12:41:35 -04:00
parent f4df12a485
commit 7ae3e059f9

View File

@@ -160,7 +160,7 @@ class GeminiCompletion(BaseLLM):
and float(version_match.group(1)) >= 2.5
):
self.thinking_config = types.ThinkingConfig(
include_thoughts=True, thinking_budget=-1
include_thoughts=True, thinking_budget=8192
)
@property