Merge pull request #14374 from dworthen/fix-embedding-with-azure-ad-token-provider

Fix embeddings using azure_ad_token_provider.
This commit is contained in:
Krish Dholakia
2025-09-09 22:21:09 -07:00
committed by GitHub
+1 -1
View File
@@ -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)