From e56cc26e18b83aa39617b7d2cf8ce3c1e3dfee87 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Tue, 27 Feb 2024 06:49:42 -0800 Subject: [PATCH] (test) mistral on azure ai studio --- litellm/tests/test_get_llm_provider.py | 15 ++++++++++++++- 1 file changed, 14 insertions(+), 1 deletion(-) diff --git a/litellm/tests/test_get_llm_provider.py b/litellm/tests/test_get_llm_provider.py index cf28f2d0d7..e443830b2f 100644 --- a/litellm/tests/test_get_llm_provider.py +++ b/litellm/tests/test_get_llm_provider.py @@ -18,4 +18,17 @@ def test_get_llm_provider(): assert response == "bedrock" -test_get_llm_provider() +# test_get_llm_provider() + + +def test_get_llm_provider_mistral_custom_api_base(): + model, custom_llm_provider, dynamic_api_key, api_base = litellm.get_llm_provider( + model="mistral/mistral-large-fr", + api_base="https://mistral-large-fr-ishaan.francecentral.inference.ai.azure.com/v1", + ) + assert custom_llm_provider == "mistral" + assert model == "mistral-large-fr" + assert ( + api_base + == "https://mistral-large-fr-ishaan.francecentral.inference.ai.azure.com/v1" + )