TLK-1771 - Improve agent creation flow

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