mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-25 02:34:29 +00:00
b81072d90c
All checks were successful
Read Version from pyproject.toml / read-version (push) Successful in 11s
* fix: add default credential for azure (#7095) * fix: fix linting error * fix: remove redundant test * test: skip redundant test --------- Co-authored-by: you-n-g <you-n-g@users.noreply.github.com>
46 lines
1.5 KiB
Python
46 lines
1.5 KiB
Python
import os
|
|
from typing import Callable, Union
|
|
|
|
from litellm._logging import verbose_logger
|
|
|
|
|
|
def get_azure_ad_token_provider() -> Callable[[], str]:
|
|
"""
|
|
Get Azure AD token provider based on Service Principal with Secret workflow.
|
|
|
|
Based on: https://github.com/openai/openai-python/blob/main/examples/azure_ad.py
|
|
See Also:
|
|
https://learn.microsoft.com/en-us/python/api/overview/azure/identity-readme?view=azure-python#service-principal-with-secret;
|
|
https://learn.microsoft.com/en-us/python/api/azure-identity/azure.identity.clientsecretcredential?view=azure-python.
|
|
|
|
Returns:
|
|
Callable that returns a temporary authentication token.
|
|
"""
|
|
from azure.identity import (
|
|
ClientSecretCredential,
|
|
DefaultAzureCredential,
|
|
get_bearer_token_provider,
|
|
)
|
|
|
|
try:
|
|
credential: Union[ClientSecretCredential, DefaultAzureCredential] = (
|
|
ClientSecretCredential(
|
|
client_id=os.environ["AZURE_CLIENT_ID"],
|
|
client_secret=os.environ["AZURE_CLIENT_SECRET"],
|
|
tenant_id=os.environ["AZURE_TENANT_ID"],
|
|
)
|
|
)
|
|
except KeyError as e:
|
|
verbose_logger.exception(
|
|
"Missing environment variable required by Azure AD workflow. "
|
|
"DefaultAzureCredential will be used"
|
|
" {}".format(str(e))
|
|
)
|
|
credential = DefaultAzureCredential()
|
|
except Exception:
|
|
raise
|
|
|
|
return get_bearer_token_provider(
|
|
credential,
|
|
"https://cognitiveservices.azure.com/.default",
|
|
)
|