From 59360e64f9cca60312a399cbe1d5e22a31d62292 Mon Sep 17 00:00:00 2001 From: Derek Worthen Date: Tue, 9 Sep 2025 07:01:05 -0700 Subject: [PATCH] Fix embeddings using azure_ad_token_provider. --- litellm/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/main.py b/litellm/main.py index 1bb84cccde..d7395eb145 100644 --- a/litellm/main.py +++ b/litellm/main.py @@ -3854,7 +3854,7 @@ def embedding( # noqa: PLR0915 max_retries = kwargs.get("max_retries", None) litellm_logging_obj: LiteLLMLoggingObj = kwargs.get("litellm_logging_obj") # type: ignore mock_response: Optional[List[float]] = kwargs.get("mock_response", None) # type: ignore - azure_ad_token_provider = kwargs.pop("azure_ad_token_provider", None) + azure_ad_token_provider = kwargs.get("azure_ad_token_provider", None) aembedding = kwargs.get("aembedding", None) extra_headers = kwargs.get("extra_headers", None) headers = kwargs.get("headers", None)