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

(feat) pass azure_ad_token to embedding

Browse source
This commit is contained in:
ishaan-jaff 2023-11-14 12:38:01 -08:00
parent 78f42921b9
commit 53ed8f3b7e
1 changed files with 5 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/main.py
View file

@ -1728,6 +1728,11 @@ def embedding(
get_secret("AZURE_API_VERSION")
)
azure_ad_token = (
kwargs.pop("azure_ad_token", None) or
get_secret("AZURE_AD_TOKEN")
)
api_key = (
api_key or
litellm.api_key or