From 246548e5537652ae396e738a02d53c4aac8afe48 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Tue, 17 Sep 2024 08:22:16 -0700 Subject: [PATCH] fix(proxy_server.py): use default azure credentials to support azure non-client secret kms --- litellm/proxy/proxy_server.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index de15300712a..043a7628b10 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -652,7 +652,7 @@ def load_from_azure_key_vault(use_azure_key_vault: bool = False): return try: - from azure.identity import ClientSecretCredential + from azure.identity import ClientSecretCredential, DefaultAzureCredential from azure.keyvault.secrets import SecretClient # Set your Azure Key Vault URI @@ -670,9 +670,10 @@ def load_from_azure_key_vault(use_azure_key_vault: bool = False): and tenant_id is not None ): # Initialize the ClientSecretCredential - credential = ClientSecretCredential( - client_id=client_id, client_secret=client_secret, tenant_id=tenant_id - ) + # credential = ClientSecretCredential( + # client_id=client_id, client_secret=client_secret, tenant_id=tenant_id + # ) + credential = DefaultAzureCredential() # Create the SecretClient using the credential client = SecretClient(vault_url=KVUri, credential=credential)