phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 02:34:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(vertex_and_google_ai_studio_gemini.py): fix translation

Browse source
This commit is contained in:
Krrish Dholakia 2025-04-19 09:02:19 -07:00
parent 211fffce89
commit 7bd2c11a7e
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/llms/vertex_ai/gemini/vertex_and_google_ai_studio_gemini.py
View file

@ -385,7 +385,7 @@ class VertexGeminiConfig(VertexAIBaseConfig, BaseConfig):
thinking_param: AnthropicThinkingParam,
) -> GeminiThinkingConfig:
thinking_enabled = thinking_param.get("type") == "enabled"
thinking_budget = thinking_param.get("budgetTokens")
thinking_budget = thinking_param.get("budget_tokens")
params: GeminiThinkingConfig = {}
if thinking_enabled: