From 6987a73e36df84fec6dff8a6a2a80523cf83be4b Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Tue, 18 Mar 2025 12:27:17 -0700 Subject: [PATCH] initialize_azure_sdk_client --- tests/litellm/llms/azure/test_azure_common_utils.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/tests/litellm/llms/azure/test_azure_common_utils.py b/tests/litellm/llms/azure/test_azure_common_utils.py index 21fa3b37ee..f29cc795db 100644 --- a/tests/litellm/llms/azure/test_azure_common_utils.py +++ b/tests/litellm/llms/azure/test_azure_common_utils.py @@ -68,6 +68,7 @@ def test_initialize_with_api_key(setup_mocks): api_base="https://test.openai.azure.com", model_name="gpt-4", api_version="2023-06-01", + is_async=False, ) # Verify expected result @@ -90,6 +91,7 @@ def test_initialize_with_tenant_credentials(setup_mocks): api_base="https://test.openai.azure.com", model_name="gpt-4", api_version=None, + is_async=False, ) # 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", model_name="gpt-4", api_version=None, + is_async=False, ) # 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", model_name="gpt-4", api_version=None, + is_async=False, ) # 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", model_name="gpt-4", api_version=None, + is_async=False, ) # 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", model_name="gpt-4", api_version=None, + is_async=False, ) # Verify error was logged @@ -196,6 +202,7 @@ def test_api_version_from_env_var(setup_mocks): api_base="https://test.openai.azure.com", model_name="gpt-4", api_version=None, + is_async=False, ) # Verify expected result @@ -210,6 +217,7 @@ def test_select_azure_base_url_called(setup_mocks): api_base="https://test.openai.azure.com", model_name="gpt-4", api_version="2023-06-01", + is_async=False, ) # Verify that select_azure_base_url_or_endpoint was called