From 01c38d37fabc8e6e859e906e4072fefa7979ec16 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Tue, 28 Nov 2023 15:57:11 -0800 Subject: [PATCH] (test) pass client to Azure completion --- litellm/tests/test_completion.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index d42ffb6d6d..30bedbe7fb 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -591,20 +591,20 @@ async def test_re_use_azure_async_client(): litellm.set_verbose=True import openai client = openai.AsyncAzureOpenAI( - azure_endpoint="https://openai-france-1234.openai.azure.com", - api_key=os.environ["AZURE_FRANCE_API_KEY"], + azure_endpoint=os.environ['AZURE_API_BASE'], + api_key=os.environ["AZURE_API_KEY"], api_version="2023-07-01-preview", ) ## Test azure call for _ in range(3): response = await litellm.acompletion( - model="azure/gpt-turbo", + model="azure/chatgpt-v-2", messages=messages, - azure_client=client + client=client ) print(f"response: {response}") except Exception as e: - print("got Exception", e) + pytest.fail("got Exception", e) import asyncio asyncio.run(