phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-28 04:04:31 +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 4b7402a1de
commit 150d030d9f
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, input=messages,
api_key=api_key, api_key=api_key,
additional_args={ additional_args={
"headers": headers, "headers": {
"api_key": api_key,
"azure_ad_token": azure_ad_token
},
"api_version": api_version, "api_version": api_version,
"api_base": api_base, "api_base": api_base,
"complete_input_dict": data, "complete_input_dict": data,
@ -302,7 +305,13 @@ class AzureChatCompletion(BaseLLM):
logging_obj.pre_call( logging_obj.pre_call(
input=input, input=input,
api_key=api_key, 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 ## COMPLETION CALL
response = azure_client.embeddings.create(**data) # type: ignore response = azure_client.embeddings.create(**data) # type: ignore