fix azure_ad_token_provider

This commit is contained in:
Ishaan Jaff 2024-08-22 16:15:53 -07:00
parent 38ce14c66b
commit f13a65be2d

View file

@ -172,7 +172,7 @@ def set_client(litellm_router_instance: LitellmRouter, model: dict):
organization_env_name = organization.replace("os.environ/", "") organization_env_name = organization.replace("os.environ/", "")
organization = litellm.get_secret(organization_env_name) organization = litellm.get_secret(organization_env_name)
litellm_params["organization"] = organization litellm_params["organization"] = organization
azure_ad_token_provider = None
if litellm_params.get("tenant_id"): if litellm_params.get("tenant_id"):
verbose_router_logger.debug("Using Azure AD Token Provider for Azure Auth") verbose_router_logger.debug("Using Azure AD Token Provider for Azure Auth")
azure_ad_token_provider = get_azure_ad_token_from_entrata_id( azure_ad_token_provider = get_azure_ad_token_from_entrata_id(