scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-14 23:21:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 102

fix(azure): add service_tier to AzureOpenAIConfig supported params (#39719)

Browse source
This commit is contained in:
amasen02 2026-09-04 16:46:49 +05:30
parent c8635ecc67
commit a2b260a9b1
2 changed files with 23 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
litellm/llms/azure/chat/gpt_transformation.py
View file

@ -130,6 +130,7 @@ class AzureOpenAIConfig(BaseConfig):
"web_search_options",
"prompt_cache_key",
"store",
"service_tier",
]
@classmethod

22
tests/test_litellm/llms/azure/chat/test_azure_chat_gpt_transformation.py
View file

@ -52,6 +52,28 @@ class TestAzureOpenAIConfig:
supported_params = config.get_supported_openai_params("gpt-4.1")
assert "prompt_cache_key" in supported_params
def test_service_tier_supported(self):
"""Test that 'service_tier' is in supported params for Azure OpenAI models and preserved with drop_params=True."""
config: Final = AzureOpenAIConfig()
supported_params: Final = config.get_supported_openai_params("gpt-4.1")
assert "service_tier" in supported_params
mapped: Final = config.map_openai_params(
non_default_params={"service_tier": "priority"},
optional_params={},
model="azure/gpt-4.1",
drop_params=True,
)
assert mapped.get("service_tier") == "priority"
opt: Final = get_optional_params(
model="azure/gpt-4o",
custom_llm_provider="azure",
service_tier="priority",
drop_params=True,
)
assert opt.get("service_tier") == "priority"
def test_map_openai_params_with_preview_api_version():
config = AzureOpenAIConfig()