diff --git a/src/backend/database_models/seeders/deplyments_models_seed.py b/src/backend/database_models/seeders/deplyments_models_seed.py index 057bda1cc1..0b8cef3685 100644 --- a/src/backend/database_models/seeders/deplyments_models_seed.py +++ b/src/backend/database_models/seeders/deplyments_models_seed.py @@ -47,6 +47,18 @@ "cohere_name": "c4ai-aya-23", "is_default": False, }, + "c4ai-aya-23-35b": { + "cohere_name": "c4ai-aya-23-35b", + "is_default": False, + }, + "command-r-08-2024": { + "cohere_name": "command-r-08-2024", + "is_default": False, + }, + "command-r-plus-08-2024": { + "cohere_name": "command-r-plus-08-2024", + "is_default": False, + }, }, ModelDeploymentName.SingleContainer: { "command": { @@ -77,6 +89,18 @@ "cohere_name": "c4ai-aya-23", "is_default": False, }, + "c4ai-aya-23-35b": { + "cohere_name": "c4ai-aya-23-35b", + "is_default": False, + }, + "command-r-08-2024": { + "cohere_name": "command-r-08-2024", + "is_default": False, + }, + "command-r-plus-08-2024": { + "cohere_name": "command-r-plus-08-2024", + "is_default": False, + }, }, ModelDeploymentName.SageMaker: { "sagemaker-command": { diff --git a/src/backend/model_deployments/cohere_platform.py b/src/backend/model_deployments/cohere_platform.py index e6d08a1075..9f6b042e08 100644 --- a/src/backend/model_deployments/cohere_platform.py +++ b/src/backend/model_deployments/cohere_platform.py @@ -54,11 +54,7 @@ def list_models(cls) -> List[str]: return [] models = response.json()["models"] - return [ - model["name"] - for model in models - if model.get("endpoints") and "chat" in model["endpoints"] - ] + return [model["name"] for model in models if model.get("endpoints") and "chat" in model["endpoints"]] @classmethod def is_available(cls) -> bool: