fix(tests): Fix e2e tests

Author: LucienShui

src/tasks/task-pipeline.e2e-spec.ts

@@ -519,16 +519,18 @@ describe('Task Pipeline (e2e)', () => {
         );
         if (tasksResponse.status !== 200) return false;
 
-        const tasks = tasksResponse.body;
-        const collectTask = tasks.find((t: any) => t.id === collectTaskId);
-        const extractTagsTask = tasks.find(
-          (t: any) =>
+        const tasks: TaskMetaDto[] = tasksResponse.body;
+        const collectTask: TaskMetaDto | undefined = tasks.find(
+          (t: any) => t.id === collectTaskId,
+        );
+        const extractTagsTask: TaskMetaDto | undefined = tasks.find(
+          (t: TaskMetaDto) =>
             t.function === 'extract_tags' &&
-            t.payload?.parent_task_id === collectTaskId,
+            t.attrs?.parent_task_id === collectTaskId,
         );
 
-        expect(isEmpty(collectTask?.exception)).toBe(true);
-        expect(isEmpty(extractTagsTask?.exception)).toBe(true);
+        expect(collectTask?.status).not.toBe('error');
+        expect(extractTagsTask?.status).not.toBe('error');
 
         return (
           !isEmpty(collectTask?.ended_at) && !isEmpty(extractTagsTask?.ended_at)
@@ -541,27 +543,27 @@ describe('Task Pipeline (e2e)', () => {
       );
       const taskMetaList: TaskMetaDto[] = tasksResponse.body;
       const extractTagsTaskMeta = taskMetaList.find(
-        (t: any) =>
+        (t: TaskMetaDto) =>
           t.function === 'extract_tags' &&
-          t.payload?.parent_task_id === collectTaskId,
+          t.attrs?.parent_task_id === collectTaskId,
       )!;
-      const collectTask = await client
+      const collectTask: TaskDto = await client
         .get(`/api/v1/namespaces/${client.namespace.id}/tasks/${collectTaskId}`)
         .then((res) => res.body);
-      const extractTagsTask = await client
+      const extractTagsTask: TaskDto = await client
         .get(
           `/api/v1/namespaces/${client.namespace.id}/tasks/${extractTagsTaskMeta.id}`,
         )
         .then((res) => res.body);
 
       expect(collectTask).toBeDefined();
       expect(collectTask.ended_at).toBeDefined();
-      expect(collectTask.output.markdown).toBeDefined();
+      expect(collectTask.output?.markdown).toBeDefined();
 
       expect(extractTagsTask).toBeDefined();
       expect(extractTagsTask.ended_at).toBeDefined();
-      expect(extractTagsTask.output.tags).toBeDefined();
-      expect(extractTagsTask.payload.resource_id).toBe(resourceId);
+      expect(extractTagsTask.output?.tags).toBeDefined();
+      expect(extractTagsTask.attrs?.resource_id).toBe(resourceId);
 
       const resource = (
         await client.get(
@@ -596,14 +598,14 @@ describe('Task Pipeline (e2e)', () => {
         const tasksResponse = await client.get(
           `/api/v1/namespaces/${client.namespace.id}/tasks`,
         );
-        const tasks = tasksResponse.body;
+        const tasks: TaskMetaDto[] = tasksResponse.body;
         const generateTitleTask = tasks.find(
-          (t: any) =>
+          (t: TaskMetaDto) =>
             t.function === 'generate_title' &&
-            t.payload.resource_id === resourceId,
+            t.attrs?.resource_id === resourceId,
         );
         if (!generateTitleTask) return false;
-        expect(isEmpty(generateTitleTask.exception)).toBe(true);
+        expect(generateTitleTask.status).not.toBe('error');
         return !isEmpty(generateTitleTask.ended_at);
       });