From af74e80b0598d0258ba0d30d216d3385cb5dffa2 Mon Sep 17 00:00:00 2001 From: Mark McDonald Date: Mon, 30 Mar 2026 18:05:09 +0800 Subject: [PATCH] chore: tidying up some case insensitivity --- .../vertex_ai/gemini/vertex_and_google_ai_studio_gemini.py | 4 ++-- .../gemini/test_vertex_and_google_ai_studio_gemini.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) 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 ae2faeccdd8..bf2a8240652 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 @@ -367,12 +367,12 @@ class VertexGeminiConfig(VertexAIBaseConfig, BaseConfig): """ Map OpenAI service_tier (string) to Gemini serviceTier. 'auto' maps to 'priority'. - Other values are passed blindly. + Other values are passed lowercased. """ if value == "auto": optional_params["service_tier"] = "priority" else: - optional_params["service_tier"] = value + optional_params["service_tier"] = value.lower() def _transform_computer_use_config(self, computer_use_config: dict) -> dict: """ 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 f0c899d35ea..48f740cda8f 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 @@ -3688,7 +3688,7 @@ def test_vertex_ai_service_tier_in_map_openai_params(): drop_params=True, ) - assert result["service_tier"] == "FLEX" + assert result["service_tier"] == "flex" # Test auto -> priority optional_params_auto = {}