From dc3d7b3afc6ebdfc6a580cc8ee901bf3ecf57a0d Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Tue, 18 Mar 2025 12:56:11 -0700 Subject: [PATCH] test_azure_instruct --- litellm/llms/azure/completion/handler.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/litellm/llms/azure/completion/handler.py b/litellm/llms/azure/completion/handler.py index 9e4fcabb6b..67088a663a 100644 --- a/litellm/llms/azure/completion/handler.py +++ b/litellm/llms/azure/completion/handler.py @@ -95,6 +95,7 @@ class AzureTextCompletion(BaseAzureLLM): azure_ad_token=azure_ad_token, timeout=timeout, client=client, + litellm_params=litellm_params, ) else: return self.acompletion( @@ -109,6 +110,7 @@ class AzureTextCompletion(BaseAzureLLM): client=client, logging_obj=logging_obj, max_retries=max_retries, + litellm_params=litellm_params, ) elif "stream" in optional_params and optional_params["stream"] is True: return self.streaming(