scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-07 02:59:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

[Fix] Azure OpenAI - Add handling for v1 under azure api versions (#15984)

Browse source 
* fix _is_azure_v1_api_version

* test_is_azure_v1_api_version
This commit is contained in:
Ishaan Jaff 2025-10-27 13:45:44 -07:00 • committed by GitHub
parent 65afdda3ef
commit 0bb53f5048
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 21 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/llms/azure/common_utils.py
View file

@ -759,4 +759,4 @@ class BaseAzureLLM(BaseOpenAILLM):
def _is_azure_v1_api_version(api_version: Optional[str]) -> bool:
if api_version is None:
return False
return api_version == "preview" or api_version == "latest"
return api_version in {"preview", "latest", "v1"}

20
tests/test_litellm/llms/azure/test_azure_common_utils.py
View file

@ -1504,3 +1504,23 @@ def test_get_azure_ad_token_fallback_to_default_azure_credential(setup_mocks, mo
# Verify the token is what we expect from our DefaultAzureCredential mock
assert token == "mock-default-azure-credential-token"
@pytest.mark.parametrize(
"api_version,expected",
[
("preview", True),
("latest", True),
("v1", True),
(None, False),
("2023-05-15", False),
("2024-01-01", False),
("", False),
],
)
def test_is_azure_v1_api_version(api_version, expected):
"""
Test that _is_azure_v1_api_version correctly identifies v1 API versions.
"""
result = BaseAzureLLM._is_azure_v1_api_version(api_version=api_version)
assert result == expected