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

fix(main.py): read azure ad token from optional params extra body

Browse source
This commit is contained in:
Krrish Dholakia 2024-01-18 17:13:54 -08:00
parent a8ba5df90e
commit e0aaa94f28
2 changed files with 6 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/main.py
View file

@ -692,9 +692,9 @@ def completion(
or get_secret("AZURE_API_KEY")
)
azure_ad_token = optional_params.pop("azure_ad_token", None) or get_secret(
"AZURE_AD_TOKEN"
)
azure_ad_token = optional_params.get("extra_body", {}).pop(
"azure_ad_token", None
) or get_secret("AZURE_AD_TOKEN")
headers = headers or litellm.headers