phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

(fix) failing cohere test

Browse source
This commit is contained in:
ishaan-jaff 2024-03-12 12:42:12 -07:00
parent b2d5f775ca
commit a18c941621
3 changed files with 15 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/utils.py
View file

@ -5371,6 +5371,9 @@ def get_llm_provider(
## cohere
elif model in litellm.cohere_models or model in litellm.cohere_embedding_models:
custom_llm_provider = "cohere"
## cohere chat models
elif model in litellm.cohere_chat_models:
custom_llm_provider = "cohere_chat"
## replicate
elif model in litellm.replicate_models or (":" in model and len(model) > 64):
model_parts = model.split(":")