From 0f0ddcc0fbb8ac9be13c63fc96b049eb7c6e5661 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Tue, 28 Nov 2023 17:17:40 -0800 Subject: [PATCH] (fix) using AzureOpenAI client --- litellm/llms/azure.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/llms/azure.py b/litellm/llms/azure.py index 1066bca3f2..ab79fe0bc2 100644 --- a/litellm/llms/azure.py +++ b/litellm/llms/azure.py @@ -242,7 +242,7 @@ class AzureChatCompletion(BaseLLM): azure_client = AzureOpenAI(**azure_client_params) else: azure_client = client - response = client.chat.completions.create(**data) + response = azure_client.chat.completions.create(**data) streamwrapper = CustomStreamWrapper(completion_stream=response, model=model, custom_llm_provider="azure",logging_obj=logging_obj) for transformed_chunk in streamwrapper: yield transformed_chunk