scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-20 00:11:50 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 75

#14404 BugFix - Add support for Azure AD token-based authorization in image generation request headers definition for Azure

Browse source
This commit is contained in:
Shagun Bansal 2025-09-23 18:49:37 +05:30
parent 7216983f48
commit a2793bdb57
1 changed files with 9 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
litellm/images/main.py
View file

@ -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