phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 11:43:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test_azure_instruct

Browse source
This commit is contained in:
Ishaan Jaff 2025-03-18 12:56:11 -07:00
parent 2cd49ef096
commit dc3d7b3afc
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/llms/azure/completion/handler.py
View file

@ -95,6 +95,7 @@ class AzureTextCompletion(BaseAzureLLM):
azure_ad_token=azure_ad_token, azure_ad_token=azure_ad_token,
timeout=timeout, timeout=timeout,
client=client, client=client,
litellm_params=litellm_params,
) )
else: else:
return self.acompletion( return self.acompletion(
@ -109,6 +110,7 @@ class AzureTextCompletion(BaseAzureLLM):
client=client, client=client,
logging_obj=logging_obj, logging_obj=logging_obj,
max_retries=max_retries, max_retries=max_retries,
litellm_params=litellm_params,
) )
elif "stream" in optional_params and optional_params["stream"] is True: elif "stream" in optional_params and optional_params["stream"] is True:
return self.streaming( return self.streaming(