diff --git a/litellm/__init__.py b/litellm/__init__.py index 57604aae76..ec65cf6444 100644 --- a/litellm/__init__.py +++ b/litellm/__init__.py @@ -252,6 +252,7 @@ config_path = None open_ai_chat_completion_models: List = [] open_ai_text_completion_models: List = [] cohere_models: List = [] +cohere_chat_models: List = [] anthropic_models: List = [] openrouter_models: List = [] vertex_language_models: List = [] @@ -274,6 +275,8 @@ for key, value in model_cost.items(): open_ai_text_completion_models.append(key) elif value.get("litellm_provider") == "cohere": cohere_models.append(key) + elif value.get("litellm_provider") == "cohere_chat": + cohere_chat_models.append(key) elif value.get("litellm_provider") == "anthropic": anthropic_models.append(key) elif value.get("litellm_provider") == "openrouter": @@ -421,6 +424,7 @@ model_list = ( open_ai_chat_completion_models + open_ai_text_completion_models + cohere_models + + cohere_chat_models + anthropic_models + replicate_models + openrouter_models @@ -479,6 +483,7 @@ provider_list: List = [ models_by_provider: dict = { "openai": open_ai_chat_completion_models + open_ai_text_completion_models, "cohere": cohere_models, + "cohere_chat": cohere_chat_models, "anthropic": anthropic_models, "replicate": replicate_models, "huggingface": huggingface_models,