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

(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 c903fcc6f3
commit 7a3fa18a64
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