diff --git a/litellm/utils.py b/litellm/utils.py index 83764961a..16900c617 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -5005,13 +5005,13 @@ def get_llm_provider( # mistral is openai compatible, we just need to set this to custom_openai and have the api_base be https://api.mistral.ai api_base = ( api_base - or get_secret("MISTRAL_API_BASE") + or get_secret("MISTRAL_AZURE_API_BASE") or "https://api.mistral.ai/v1" ) # if api_base does not end with /v1 we add it if api_base is not None and not api_base.endswith("/v1"): api_base = api_base + "/v1" - dynamic_api_key = api_key or get_secret("MISTRAL_API_KEY") + dynamic_api_key = api_key or get_secret("MISTRAL_AZURE_API_KEY") elif custom_llm_provider == "voyage": # voyage is openai compatible, we just need to set this to custom_openai and have the api_base be https://api.voyageai.com/v1 api_base = "https://api.voyageai.com/v1"