From 86cc0f087df19da172852251a8ca4aaa6989c104 Mon Sep 17 00:00:00 2001 From: Abhimanyu Saharan Date: Mon, 9 Feb 2026 02:54:11 +0530 Subject: [PATCH] refactor: simplify assigned agent retrieval logic in tasks.py --- backend/app/api/tasks.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/backend/app/api/tasks.py b/backend/app/api/tasks.py index 1e514a9c..88f2aee1 100644 --- a/backend/app/api/tasks.py +++ b/backend/app/api/tasks.py @@ -878,8 +878,9 @@ async def update_task( elif updates["status"] == "in_progress": task.in_progress_at = utcnow() - if "assigned_agent_id" in updates and updates["assigned_agent_id"]: - agent = await Agent.objects.by_id(updates["assigned_agent_id"]).first(session) + assigned_agent_id = updates.get("assigned_agent_id") + if assigned_agent_id: + agent = await Agent.objects.by_id(assigned_agent_id).first(session) if agent is None: raise HTTPException(status_code=status.HTTP_404_NOT_FOUND) if agent.board_id and task.board_id and agent.board_id != task.board_id: