tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-07-15 00:22:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

(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
litellm/tests/test_get_llm_provider.py
+14 -1
View File
@@ -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"
)