phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 11:14:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

(test) mistral on azure ai studio

Browse source
This commit is contained in:
ishaan-jaff 2024-02-27 06:49:42 -08:00
parent cfebbdfa7b
commit e56cc26e18
1 changed files with 14 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

15
litellm/tests/test_get_llm_provider.py
View file

@ -18,4 +18,17 @@ def test_get_llm_provider():
assert response == "bedrock" 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"
)