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 165

chore: tidying up some case insensitivity

Browse source
This commit is contained in:
Mark McDonald 2026-03-30 18:05:09 +08:00 • committed by Sameer Kankute
parent e04af74986
commit af74e80b05
No known key found for this signature in database
2 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
litellm/llms/vertex_ai/gemini/vertex_and_google_ai_studio_gemini.py
View file

@ -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:
"""

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

@ -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 = {}