From c2d2df22778ba9fccb4e9c851ee8a3b8a2d019d4 Mon Sep 17 00:00:00 2001 From: Mark McDonald Date: Wed, 1 Apr 2026 13:54:06 +0800 Subject: [PATCH] fix: handle 'auto' case-insensitively --- .../gemini/vertex_and_google_ai_studio_gemini.py | 2 +- .../test_vertex_and_google_ai_studio_gemini.py | 13 +++++++++++++ 2 files changed, 14 insertions(+), 1 deletion(-) 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 a98179bff6c..f4932508458 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 @@ -369,7 +369,7 @@ class VertexGeminiConfig(VertexAIBaseConfig, BaseConfig): 'auto' maps to 'priority'. Other values are passed lowercased. """ - if value == "auto": + if value.lower() == "auto": optional_params["service_tier"] = "priority" else: optional_params["service_tier"] = value.lower() diff --git a/tests/test_litellm/llms/vertex_ai/gemini/test_vertex_and_google_ai_studio_gemini.py b/tests/test_litellm/llms/vertex_ai/gemini/test_vertex_and_google_ai_studio_gemini.py index 48f740cda8f..479f94130d4 100644 --- a/tests/test_litellm/llms/vertex_ai/gemini/test_vertex_and_google_ai_studio_gemini.py +++ b/tests/test_litellm/llms/vertex_ai/gemini/test_vertex_and_google_ai_studio_gemini.py @@ -3703,6 +3703,19 @@ def test_vertex_ai_service_tier_in_map_openai_params(): assert result_auto["service_tier"] == "priority" + # Test AUTO (uppercase) -> priority + optional_params_auto_upper = {} + non_default_params_auto_upper = {"service_tier": "AUTO"} + + result_auto_upper = v.map_openai_params( + non_default_params=non_default_params_auto_upper, + optional_params=optional_params_auto_upper, + model="gemini-3-pro-preview", + drop_params=True, + ) + + assert result_auto_upper["service_tier"] == "priority" + def test_vertex_ai_usage_metadata_with_video_tokens_in_prompt(): """Test promptTokensDetails with VIDEO modality for video inputs.