phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 19:24:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

initialize_azure_sdk_client

Browse source
This commit is contained in:
Ishaan Jaff 2025-03-18 12:27:17 -07:00
parent 842625a6f0
commit 6987a73e36
1 changed files with 8 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/litellm/llms/azure/test_azure_common_utils.py
View file

@ -68,6 +68,7 @@ def test_initialize_with_api_key(setup_mocks):
api_base="https://test.openai.azure.com", api_base="https://test.openai.azure.com",
model_name="gpt-4", model_name="gpt-4",
api_version="2023-06-01", api_version="2023-06-01",
is_async=False,
) )
# Verify expected result # Verify expected result
@ -90,6 +91,7 @@ def test_initialize_with_tenant_credentials(setup_mocks):
api_base="https://test.openai.azure.com", api_base="https://test.openai.azure.com",
model_name="gpt-4", model_name="gpt-4",
api_version=None, api_version=None,
is_async=False,
) )
# Verify that get_azure_ad_token_from_entrata_id was called # Verify that get_azure_ad_token_from_entrata_id was called
@ -117,6 +119,7 @@ def test_initialize_with_username_password(setup_mocks):
api_base="https://test.openai.azure.com", api_base="https://test.openai.azure.com",
model_name="gpt-4", model_name="gpt-4",
api_version=None, api_version=None,
is_async=False,
) )
# Verify that get_azure_ad_token_from_username_password was called # Verify that get_azure_ad_token_from_username_password was called
@ -138,6 +141,7 @@ def test_initialize_with_oidc_token(setup_mocks):
api_base="https://test.openai.azure.com", api_base="https://test.openai.azure.com",
model_name="gpt-4", model_name="gpt-4",
api_version=None, api_version=None,
is_async=False,
) )
# Verify that get_azure_ad_token_from_oidc was called # Verify that get_azure_ad_token_from_oidc was called
@ -158,6 +162,7 @@ def test_initialize_with_enable_token_refresh(setup_mocks):
api_base="https://test.openai.azure.com", api_base="https://test.openai.azure.com",
model_name="gpt-4", model_name="gpt-4",
api_version=None, api_version=None,
is_async=False,
) )
# Verify that get_azure_ad_token_provider was called # Verify that get_azure_ad_token_provider was called
@ -179,6 +184,7 @@ def test_initialize_with_token_refresh_error(setup_mocks):
api_base="https://test.openai.azure.com", api_base="https://test.openai.azure.com",
model_name="gpt-4", model_name="gpt-4",
api_version=None, api_version=None,
is_async=False,
) )
# Verify error was logged # Verify error was logged
@ -196,6 +202,7 @@ def test_api_version_from_env_var(setup_mocks):
api_base="https://test.openai.azure.com", api_base="https://test.openai.azure.com",
model_name="gpt-4", model_name="gpt-4",
api_version=None, api_version=None,
is_async=False,
) )
# Verify expected result # Verify expected result
@ -210,6 +217,7 @@ def test_select_azure_base_url_called(setup_mocks):
api_base="https://test.openai.azure.com", api_base="https://test.openai.azure.com",
model_name="gpt-4", model_name="gpt-4",
api_version="2023-06-01", api_version="2023-06-01",
is_async=False,
) )
# Verify that select_azure_base_url_or_endpoint was called # Verify that select_azure_base_url_or_endpoint was called