From dbd603c706e9658b70ee16893a24b926da144c52 Mon Sep 17 00:00:00 2001 From: EugeneP Date: Wed, 16 Oct 2024 20:04:55 +0200 Subject: [PATCH] TLK-1771 - Improve agent creation flow --- src/backend/tests/unit/routers/test_agent.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/backend/tests/unit/routers/test_agent.py b/src/backend/tests/unit/routers/test_agent.py index bfc7c7c3e..3c1ae0bd0 100644 --- a/src/backend/tests/unit/routers/test_agent.py +++ b/src/backend/tests/unit/routers/test_agent.py @@ -6,7 +6,6 @@ from backend.config.tools import ToolName from backend.crud import agent as agent_crud from backend.crud import deployment as deployment_crud -from backend.crud import model as model_crud from backend.database_models.agent import Agent from backend.database_models.agent_tool_metadata import AgentToolMetadata from backend.database_models.snapshot import Snapshot @@ -118,6 +117,9 @@ def test_create_agent_deployment_not_in_db( deployment_models = cohere_deployment.models deployment_models_list = [model.name for model in deployment_models] assert cohere_deployment + assert cohere_deployment.name == ModelDeploymentName.CoherePlatform + assert cohere_deployment.deployment_class_name == "CoherePlatform" + assert cohere_deployment.id == 1 assert deployment_models_list == ["command-r-plus"] assert response.status_code == 200