scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-08 22:21:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 78

fix get llm provider logic

Browse source
This commit is contained in:
Ishaan Jaff 2024-09-04 09:28:23 -07:00
parent 9ed4dad5ac
commit 4599a0ffd3
1 changed files with 9 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
litellm/litellm_core_utils/get_llm_provider_logic.py
View file

@ -55,13 +55,15 @@ def handle_cohere_chat_model_custom_llm_provider(
if custom_llm_provider == "cohere" and model in litellm.cohere_chat_models:
return model, "cohere_chat"
_custom_llm_provider, _model = model.split("/", 1)
if (
_custom_llm_provider
and _custom_llm_provider == "cohere"
and _model in litellm.cohere_chat_models
):
return _model, "cohere_chat"
if "/" in model:
_custom_llm_provider, _model = model.split("/", 1)
if (
_custom_llm_provider
and _custom_llm_provider == "cohere"
and _model in litellm.cohere_chat_models
):
return _model, "cohere_chat"
return model, custom_llm_provider