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

fix add azure/command-r-plus

Browse source
This commit is contained in:
Ishaan Jaff 2024-04-05 08:53:24 -07:00
parent 65b1a7be73
commit 71352b1b36
1 changed files with 10 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

10
litellm/utils.py
View file

@ -5573,6 +5573,16 @@ def get_llm_provider(
dynamic_api_key = None dynamic_api_key = None
# check if llm provider provided # check if llm provider provided
# AZURE AI-Studio Logic - Azure AI Studio supports AZURE/Cohere
# If User passes azure/command-r-plus -> we should send it to cohere_chat/command-r-plus
if (
model.split("/", 1)[0] == "azure"
and model.split("/", 1)[1] in litellm.cohere_chat_models
):
custom_llm_provider = "openai"
model = model.split("/", 1)[1]
return model, custom_llm_provider, dynamic_api_key, api_base
if custom_llm_provider: if custom_llm_provider:
return model, custom_llm_provider, dynamic_api_key, api_base return model, custom_llm_provider, dynamic_api_key, api_base