From f4fbe75af31c2f74d45adce0fc58041d2bb8c8f8 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Tue, 27 Aug 2024 19:32:37 -0700 Subject: [PATCH] test: fix test --- litellm/tests/test_exceptions.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/litellm/tests/test_exceptions.py b/litellm/tests/test_exceptions.py index 9978f6df3e8..0388e026b9f 100644 --- a/litellm/tests/test_exceptions.py +++ b/litellm/tests/test_exceptions.py @@ -914,12 +914,16 @@ async def test_exception_with_headers(sync_mode, provider, model, call_type, str if provider == "openai": openai_client = openai.OpenAI(api_key="") elif provider == "azure": - openai_client = openai.AzureOpenAI(api_key="", base_url="") + openai_client = openai.AzureOpenAI( + api_key="", base_url="", api_version=litellm.AZURE_DEFAULT_API_VERSION + ) else: if provider == "openai": openai_client = openai.AsyncOpenAI(api_key="") elif provider == "azure": - openai_client = openai.AsyncAzureOpenAI(api_key="", base_url="") + openai_client = openai.AsyncAzureOpenAI( + api_key="", base_url="", api_version=litellm.AZURE_DEFAULT_API_VERSION + ) data = {"model": model} data, original_function, mapped_target = _pre_call_utils(