From 728a7de6093e8810a4d897821af2e984692521a0 Mon Sep 17 00:00:00 2001 From: Young Date: Wed, 5 Mar 2025 06:01:18 +0000 Subject: [PATCH 1/2] fix: Handle ManagedIdentityCredential in Azure AD token provider --- litellm/secret_managers/get_azure_ad_token_provider.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/litellm/secret_managers/get_azure_ad_token_provider.py b/litellm/secret_managers/get_azure_ad_token_provider.py index 2052f65e151..6bd619def96 100644 --- a/litellm/secret_managers/get_azure_ad_token_provider.py +++ b/litellm/secret_managers/get_azure_ad_token_provider.py @@ -30,6 +30,8 @@ def get_azure_ad_token_provider() -> Callable[[], str]: client_secret=os.environ["AZURE_CLIENT_SECRET"], tenant_id=os.environ["AZURE_TENANT_ID"], ) + if cred == "ManagedIdentityCredential": + credential = cred_cls(client_id=os.environ["AZURE_CLIENT_ID"]) else: credential = cred_cls() From 8d4d90ce1f2b4f1aa1f67fa8232961567e25f770 Mon Sep 17 00:00:00 2001 From: Young Date: Tue, 11 Mar 2025 09:27:31 +0000 Subject: [PATCH 2/2] fix if --- litellm/secret_managers/get_azure_ad_token_provider.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/secret_managers/get_azure_ad_token_provider.py b/litellm/secret_managers/get_azure_ad_token_provider.py index 6bd619def96..5403675b979 100644 --- a/litellm/secret_managers/get_azure_ad_token_provider.py +++ b/litellm/secret_managers/get_azure_ad_token_provider.py @@ -30,7 +30,7 @@ def get_azure_ad_token_provider() -> Callable[[], str]: client_secret=os.environ["AZURE_CLIENT_SECRET"], tenant_id=os.environ["AZURE_TENANT_ID"], ) - if cred == "ManagedIdentityCredential": + elif cred == "ManagedIdentityCredential": credential = cred_cls(client_id=os.environ["AZURE_CLIENT_ID"]) else: credential = cred_cls()