phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-24 10:14:26 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

(fix) azure: better debugging

Browse source
This commit is contained in:
ishaan-jaff 2023-11-23 15:36:51 -08:00
parent 525e01824a
commit e8b844abae
1 changed files with 11 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

13
litellm/llms/azure.py
View file

@ -122,7 +122,10 @@ class AzureChatCompletion(BaseLLM):
input=messages,
api_key=api_key,
additional_args={
"headers": headers,
"headers": {
"api_key": api_key,
"azure_ad_token": azure_ad_token
},
"api_version": api_version,
"api_base": api_base,
"complete_input_dict": data,
@ -302,7 +305,13 @@ class AzureChatCompletion(BaseLLM):
logging_obj.pre_call(
input=input,
api_key=api_key,
additional_args={"complete_input_dict": data},
additional_args={
"complete_input_dict": data,
"headers": {
"api_key": api_key,
"azure_ad_token": azure_ad_token
}
},
)
## COMPLETION CALL
response = azure_client.embeddings.create(**data) # type: ignore