From 589c83b88b45b53dd5c6bce05ace5dd1245188e8 Mon Sep 17 00:00:00 2001 From: Shagun Bansal Date: Wed, 24 Sep 2025 16:25:44 +0530 Subject: [PATCH] #14404 BugFix - Add support for Azure AD token-based authorization in image generation request headers definition for Azure --- litellm/llms/azure/azure.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/litellm/llms/azure/azure.py b/litellm/llms/azure/azure.py index 5ee9065f5e1..f41a9bea0c9 100644 --- a/litellm/llms/azure/azure.py +++ b/litellm/llms/azure/azure.py @@ -1117,6 +1117,14 @@ class AzureChatCompletion(BaseAzureLLM, BaseLLM): status_code=422, message="max retries must be an int" ) + if api_key is None and azure_ad_token_provider is not None: + azure_ad_token = azure_ad_token_provider() + if azure_ad_token: + headers.pop( + "api-key", None + ) + headers["Authorization"] = f"Bearer {azure_ad_token}" + # init AzureOpenAI Client azure_client_params: Dict[str, Any] = self.initialize_azure_sdk_client( litellm_params=litellm_params or {},