test_azure_instruct

2026-08-05 10:24:03 +00:00 · 2025-03-18 12:56:11 -07:00
parent 2cd49ef096
commit dc3d7b3afc
1 changed files with 2 additions and 0 deletions
@@ -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(