fix use azure_ai/mistral

This commit is contained in:
Ishaan Jaff 2024-04-05 10:07:43 -07:00
parent 5ce80d82d3
commit 6b9c04618e
2 changed files with 12 additions and 43 deletions

View file

@ -5579,7 +5579,7 @@ def get_llm_provider(
model_name = model.split("/", 1)[1]
if (
model_name in litellm.cohere_chat_models
or model_name in litellm.mistral_chat_models
or f"mistral/{model_name}" in litellm.mistral_chat_models
):
custom_llm_provider = "openai"
model = model_name