From ec70cdc558504b5f32aef543b3c503d789405c37 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Thu, 29 Feb 2024 08:28:46 -0800 Subject: [PATCH] (feat) use mistral azure with env vars --- litellm/utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/litellm/utils.py b/litellm/utils.py index 83764961af6..16900c617ac 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"