From 5885c4a588183f3064986e54b25634365243bb08 Mon Sep 17 00:00:00 2001 From: Sameer Kankute Date: Fri, 20 Feb 2026 09:49:37 +0530 Subject: [PATCH] Fix _map_reasoning_effort_to_thinking_level for all gemini 3 family --- .../vertex_ai/gemini/vertex_and_google_ai_studio_gemini.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/litellm/llms/vertex_ai/gemini/vertex_and_google_ai_studio_gemini.py b/litellm/llms/vertex_ai/gemini/vertex_and_google_ai_studio_gemini.py index cf3461a996..d248d2862e 100644 --- a/litellm/llms/vertex_ai/gemini/vertex_and_google_ai_studio_gemini.py +++ b/litellm/llms/vertex_ai/gemini/vertex_and_google_ai_studio_gemini.py @@ -759,6 +759,9 @@ class VertexGeminiConfig(VertexAIBaseConfig, BaseConfig): "gemini-3-flash-preview" in model.lower() or "gemini-3-flash" in model.lower() ) + is_gemini31pro = model and ( + "gemini-3.1-pro-preview" in model.lower() + ) if reasoning_effort == "minimal": if is_gemini3flash: return {"thinkingLevel": "minimal", "includeThoughts": True} @@ -767,7 +770,10 @@ class VertexGeminiConfig(VertexAIBaseConfig, BaseConfig): elif reasoning_effort == "low": return {"thinkingLevel": "low", "includeThoughts": True} elif reasoning_effort == "medium": + if is_gemini31pro or is_gemini3flash: return {"thinkingLevel": "medium", "includeThoughts": True} + else: + return {"thinkingLevel": "high", "includeThoughts": True} elif reasoning_effort == "high": return {"thinkingLevel": "high", "includeThoughts": True} elif reasoning_effort == "disable":