scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-15 23:31:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 213

fix: handle 'auto' case-insensitively

Browse source
This commit is contained in:
Mark McDonald 2026-04-01 13:54:06 +08:00 • committed by Sameer Kankute
parent 9bbbf72118
commit c2d2df2277
No known key found for this signature in database
2 changed files with 14 additions and 1 deletions
Show stats 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

@ -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()

13
tests/test_litellm/llms/vertex_ai/gemini/test_vertex_and_google_ai_studio_gemini.py
View file

@ -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.