fix(backend): missed test in refactor

cohere-ai · Jan 16, 2025 · fa1eda8 · fa1eda8
1 parent e8779d2
commit fa1eda8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/backend/crud/agent.py b/src/backend/crud/agent.py
@@ -140,7 +140,7 @@ def update_agent(
     if agent.is_private and agent.user_id != user_id:
         return None
 
-    new_agent_cleaned = new_agent.dict(exclude_unset=True, exclude_none=True)
+    new_agent_cleaned = new_agent.model_dump(exclude_unset=True, exclude_none=True)
 
     for attr, value in new_agent_cleaned.items():
         setattr(agent, attr, value)

diff --git a/src/backend/tests/integration/routers/test_deployment.py b/src/backend/tests/integration/routers/test_deployment.py
@@ -93,7 +93,7 @@ def test_list_deployments_no_available_models_404(
     session_client: TestClient, session: Session
 ) -> None:
     session.query(Deployment).delete()
-    with patch("backend.services.deployment.AVAILABLE_MODEL_DEPLOYMENTS", []):
+    with patch("backend.services.deployment.AVAILABLE_MODEL_DEPLOYMENTS", {}):
         response = session_client.get("/v1/deployments")
     assert response.status_code == 404
     assert response.json() == {