From a2793bdb5760cb75f4eb944da9d7f0813c128dbb Mon Sep 17 00:00:00 2001 From: Shagun Bansal Date: Tue, 23 Sep 2025 18:49:37 +0530 Subject: [PATCH] #14404 BugFix - Add support for Azure AD token-based authorization in image generation request headers definition for Azure --- litellm/images/main.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/litellm/images/main.py b/litellm/images/main.py index 2a8b62bce24..0e11d9d3e56 100644 --- a/litellm/images/main.py +++ b/litellm/images/main.py @@ -314,6 +314,15 @@ def image_generation( # noqa: PLR0915 "Content-Type": "application/json", "api-key": api_key, } + + if api_key is None and azure_ad_token_provider is not None: + azure_ad_token = azure_ad_token_provider() + if azure_ad_token: + default_headers.pop( + "api-key", None + ) + default_headers["Authorization"] = f"Bearer {azure_ad_token}" + for k, v in default_headers.items(): if k not in headers: headers[k] = v