fix(core): add project_id to task

Author: gusye1234

src/server/core/acontext_core/llm/tool/task_lib/append_planning.py

@@ -26,6 +26,7 @@ async def _append_messages_to_planning_section_handler(
         )
     r = await TD.append_messages_to_planning_section(
         ctx.db_session,
+        ctx.project_id,
         ctx.session_id,
         actually_message_ids,
     )

src/server/core/acontext_core/llm/tool/task_lib/insert.py

@@ -10,6 +10,7 @@
 async def insert_task_handler(ctx: TaskCtx, llm_arguments: dict) -> Result[str]:
     r = await TD.insert_task(
         ctx.db_session,
+        ctx.project_id,
         ctx.session_id,
         after_order=llm_arguments["after_task_order"],
         data={

src/server/core/acontext_core/schema/orm/project.py

@@ -9,6 +9,7 @@
 if TYPE_CHECKING:
     from .space import Space
     from .session import Session
+    from .task import Task
     from .tool_reference import ToolReference
 
 
@@ -49,6 +50,15 @@ class Project(CommonMixin):
         },
     )
 
+    tasks: List["Task"] = field(
+        default_factory=list,
+        metadata={
+            "db": relationship(
+                "Task", back_populates="project", cascade="all, delete-orphan"
+            )
+        },
+    )
+
     tool_references: List["ToolReference"] = field(
         default_factory=list,
         metadata={

src/server/core/acontext_core/schema/orm/task.py

@@ -19,6 +19,7 @@
 from ..utils import asUUID
 
 if TYPE_CHECKING:
+    from .project import Project
     from .session import Session
     from .message import Message
 
@@ -55,6 +56,16 @@ class Task(CommonMixin):
         }
     )
 
+    project_id: asUUID = field(
+        metadata={
+            "db": Column(
+                UUID(as_uuid=True),
+                ForeignKey("projects.id", ondelete="CASCADE"),
+                nullable=False,
+            )
+        }
+    )
+
     order: int = field(metadata={"db": Column(Integer, nullable=False)})
 
     data: dict = field(metadata={"db": Column(JSONB, nullable=False)})
@@ -94,3 +105,8 @@ class Task(CommonMixin):
         init=False,
         metadata={"db": relationship("Session", back_populates="tasks")},
     )
+
+    project: "Project" = field(
+        init=False,
+        metadata={"db": relationship("Project", back_populates="tasks")},
+    )

src/server/core/acontext_core/service/data/task.py

@@ -154,6 +154,7 @@ async def set_task_space_digested(
 
 async def insert_task(
     db_session: AsyncSession,
+    project_id: asUUID,
     session_id: asUUID,
     after_order: int,
     data: dict,
@@ -192,6 +193,7 @@ async def insert_task(
     # Step 3: Create new task
     task = Task(
         session_id=session_id,
+        project_id=project_id,
         order=after_order + 1,
         data=data,
         status=status,
@@ -223,6 +225,7 @@ async def append_messages_to_task(
 
 async def append_messages_to_planning_section(
     db_session: AsyncSession,
+    project_id: asUUID,
     session_id: asUUID,
     message_ids: list[asUUID],
 ) -> Result[None]:
@@ -237,6 +240,7 @@ async def append_messages_to_planning_section(
     if planning_task is None:
         # add planning section
         planning_task = Task(
+            project_id=project_id,
             session_id=session_id,
             order=0,
             data={},

src/server/core/tests/service/test_task_data.py

@@ -60,7 +60,7 @@ async def test_fetch_all_tasks_success(self):
             ]
 
             for data in tasks_data:
-                task = Task(**data)
+                task = Task(project_id=project.id, **data)
                 session.add(task)
 
             await session.flush()
@@ -106,12 +106,14 @@ async def test_fetch_tasks_with_status_filter(self):
             # Create sample tasks with different statuses
             task1 = Task(
                 session_id=test_session.id,
+                project_id=project.id,
                 order=1,
                 data={"name": "task1"},
                 status="pending",
             )
             task2 = Task(
                 session_id=test_session.id,
+                project_id=project.id,
                 order=2,
                 data={"name": "task2"},
                 status="running",
@@ -174,6 +176,7 @@ async def test_update_status_success(self):
 
             task = Task(
                 session_id=test_session.id,
+                project_id=project.id,
                 order=1,
                 data={"name": "test_task"},
                 status="pending",
@@ -217,6 +220,7 @@ async def test_update_order_success(self):
 
             task = Task(
                 session_id=test_session.id,
+                project_id=project.id,
                 order=1,
                 data={"name": "test_task"},
                 status="pending",
@@ -261,6 +265,7 @@ async def test_update_data_success(self):
 
             task = Task(
                 session_id=test_session.id,
+                project_id=project.id,
                 order=1,
                 data={"name": "original_task"},
                 status="pending",
@@ -303,6 +308,7 @@ async def test_update_multiple_fields(self):
 
             task = Task(
                 session_id=test_session.id,
+                project_id=project.id,
                 order=1,
                 data={"name": "original_task"},
                 status="pending",
@@ -367,6 +373,7 @@ async def test_update_task_with_none_values(self):
 
             task = Task(
                 session_id=test_session.id,
+                project_id=project.id,
                 order=1,
                 data={"name": "test_task"},
                 status="pending",
@@ -423,6 +430,7 @@ async def test_update_task_patch_data_success(self):
             }
             task = Task(
                 session_id=test_session.id,
+                project_id=project.id,
                 order=1,
                 data=initial_data,
                 status="pending",
@@ -524,6 +532,7 @@ async def test_update_task_patch_data_with_status_and_order(self):
 
             task = Task(
                 session_id=test_session.id,
+                project_id=project.id,
                 order=1,
                 data={"task_description": "original", "step": "init"},
                 status="pending",
@@ -579,7 +588,9 @@ async def test_insert_task_success(self):
             data = {"name": "new_task", "description": "A new task"}
             after_order = 0  # Insert after position 0 (will become position 1)
 
-            result = await insert_task(session, test_session.id, after_order, data)
+            result = await insert_task(
+                session, project.id, test_session.id, after_order, data
+            )
 
             t_data, error = result.unpack()
             assert error is None
@@ -617,7 +628,12 @@ async def test_insert_task_with_custom_status(self):
             custom_status = "running"
 
             result = await insert_task(
-                session, test_session.id, after_order, data, status=custom_status
+                session,
+                project.id,
+                test_session.id,
+                after_order,
+                data,
+                status=custom_status,
             )
 
             t_data, error = result.unpack()
@@ -655,7 +671,9 @@ async def test_insert_task_default_status(self):
             data = {"name": "default_status_task"}
             after_order = 2  # Insert after position 2 (will become position 3)
 
-            result = await insert_task(session, test_session.id, after_order, data)
+            result = await insert_task(
+                session, project.id, test_session.id, after_order, data
+            )
 
             data, error = result.unpack()
             assert error is None
@@ -701,7 +719,9 @@ async def test_insert_task_complex_data(self):
                 ],
             }
 
-            result = await insert_task(session, test_session.id, 0, complex_data)
+            result = await insert_task(
+                session, project.id, test_session.id, 0, complex_data
+            )
 
             data, error = result.unpack()
             assert error is None
@@ -737,18 +757,21 @@ async def test_insert_order_increment(self):
             # Create initial tasks with orders 1, 2, 3
             task1 = Task(
                 session_id=test_session.id,
+                project_id=project.id,
                 order=1,
                 data={"name": "task1"},
                 status="pending",
             )
             task2 = Task(
                 session_id=test_session.id,
+                project_id=project.id,
                 order=2,
                 data={"name": "task2"},
                 status="pending",
             )
             task3 = Task(
                 session_id=test_session.id,
+                project_id=project.id,
                 order=3,
                 data={"name": "task3"},
                 status="pending",
@@ -758,7 +781,9 @@ async def test_insert_order_increment(self):
 
             # Insert a new task after position 1 (should become position 2)
             new_data = {"name": "inserted_task"}
-            result = await insert_task(session, test_session.id, 1, new_data)
+            result = await insert_task(
+                session, project.id, test_session.id, 1, new_data
+            )
 
             new_task, error = result.unpack()
             assert error is None
@@ -811,6 +836,7 @@ async def test_delete_task_success(self):
 
             task = Task(
                 session_id=test_session.id,
+                project_id=project.id,
                 order=1,
                 data={"name": "task_to_delete"},
                 status="pending",
@@ -875,18 +901,21 @@ async def test_delete_task_cascade_behavior(self):
             # Create multiple tasks
             task1 = Task(
                 session_id=test_session.id,
+                project_id=project.id,
                 order=1,
                 data={"name": "task1"},
                 status="pending",
             )
             task2 = Task(
                 session_id=test_session.id,
+                project_id=project.id,
                 order=2,
                 data={"name": "task2"},
                 status="running",
             )
             task3 = Task(
                 session_id=test_session.id,
+                project_id=project.id,
                 order=3,
                 data={"name": "task3"},
                 status="success",
@@ -939,7 +968,9 @@ async def test_full_task_lifecycle(self):
 
             # 1. Create a task
             initial_data = {"name": "lifecycle_task", "step": "created"}
-            create_result = await insert_task(session, test_session.id, 0, initial_data)
+            create_result = await insert_task(
+                session, project.id, test_session.id, 0, initial_data
+            )
             created_task, _ = create_result.unpack()
             assert created_task is not None
             task_id = created_task.id
@@ -997,9 +1028,15 @@ async def test_multiple_sessions_isolation(self):
             await session.flush()
 
             # Create tasks in each session
-            await insert_task(session, session1.id, 0, {"session": "1", "task": "A"})
-            await insert_task(session, session1.id, 1, {"session": "1", "task": "B"})
-            await insert_task(session, session2.id, 0, {"session": "2", "task": "A"})
+            await insert_task(
+                session, project.id, session1.id, 0, {"session": "1", "task": "A"}
+            )
+            await insert_task(
+                session, project.id, session1.id, 1, {"session": "1", "task": "B"}
+            )
+            await insert_task(
+                session, project.id, session2.id, 0, {"session": "2", "task": "A"}
+            )
 
             # Fetch tasks for each session
             session1_tasks_result = await fetch_current_tasks(session, session1.id)
@@ -1041,18 +1078,21 @@ async def test_ordering_after_updates(self):
             # Create initial tasks in order
             task1_result = await insert_task(
                 session,
+                project.id,
                 test_session.id,
                 0,
                 {"name": "task1"},  # Insert after 0 -> position 1
             )
             task2_result = await insert_task(
                 session,
+                project.id,
                 test_session.id,
                 1,
                 {"name": "task2"},  # Insert after 1 -> position 2
             )
             task3_result = await insert_task(
                 session,
+                project.id,
                 test_session.id,
                 2,
                 {"name": "task3"},  # Insert after 2 -> position 3
@@ -1065,6 +1105,7 @@ async def test_ordering_after_updates(self):
             # Insert a new task in the middle (after position 1)
             middle_task_result = await insert_task(
                 session,
+                project.id,
                 test_session.id,
                 1,
                 {"name": "middle_task"},  # Insert after 1 -> position 2