phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

Merge pull request #2142 from vilmar-hillow/azure_embedding_ad_token

Browse source 
Fixed azure ad token not being processed properly in embedding models
This commit is contained in:
Krish Dholakia 2024-03-19 11:51:28 -07:00 • committed by GitHub
commit c840fecdeb
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 535 additions and 383 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/main.py
View file

@ -2614,7 +2614,7 @@ def embedding(
api_version or litellm.api_version or get_secret("AZURE_API_VERSION")
)
azure_ad_token = kwargs.pop("azure_ad_token", None) or get_secret(
azure_ad_token = optional_params.pop("azure_ad_token", None) or get_secret(
"AZURE_AD_TOKEN"
)