From 55a37d6d9e5e681fe20e7a1b3e716c4b67983fd9 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 0e121abb766..772a307868b 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 @@ -756,6 +756,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} @@ -764,7 +767,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":