Fixed azure ad token not being processed properly in embedding models

This commit is contained in:
Dmitry Supranovich 2024-02-22 23:23:06 +00:00 committed by Dmitry Supranovich
parent 0d18f3c0ca
commit 57ebb9582e
4 changed files with 535 additions and 383 deletions

View file

@ -2606,7 +2606,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"
)