Fix TaskStatusNotificationParamsSchema

Author: LucaButBoring

src/shared/protocol.test.ts

@@ -2657,13 +2657,12 @@ describe('Progress notification support for tasks', () => {
                 jsonrpc: '2.0',
                 method: 'notifications/tasks/status',
                 params: {
-                    task: {
-                        taskId,
-                        status: 'failed',
-                        ttl: 60000,
-                        createdAt: new Date().toISOString(),
-                        statusMessage: 'Task failed'
-                    }
+                    taskId,
+                    status: 'failed',
+                    ttl: 60000,
+                    createdAt: new Date().toISOString(),
+                    lastUpdatedAt: new Date().toISOString(),
+                    statusMessage: 'Task failed'
                 }
             });
         }
@@ -2753,13 +2752,12 @@ describe('Progress notification support for tasks', () => {
                 jsonrpc: '2.0',
                 method: 'notifications/tasks/status',
                 params: {
-                    task: {
-                        taskId,
-                        status: 'cancelled',
-                        ttl: 60000,
-                        createdAt: new Date().toISOString(),
-                        statusMessage: 'User cancelled'
-                    }
+                    taskId,
+                    status: 'cancelled',
+                    ttl: 60000,
+                    createdAt: new Date().toISOString(),
+                    lastUpdatedAt: new Date().toISOString(),
+                    statusMessage: 'User cancelled'
                 }
             });
         }

src/shared/protocol.ts

@@ -1497,9 +1497,7 @@ export abstract class Protocol<SendRequestT extends Request, SendNotificationT e
                 if (task) {
                     const notification: TaskStatusNotification = TaskStatusNotificationSchema.parse({
                         method: 'notifications/tasks/status',
-                        params: {
-                            task
-                        }
+                        params: task
                     });
                     await this.notification(notification as SendNotificationT);
 
@@ -1537,9 +1535,7 @@ export abstract class Protocol<SendRequestT extends Request, SendNotificationT e
                     if (updatedTask) {
                         const notification: TaskStatusNotification = TaskStatusNotificationSchema.parse({
                             method: 'notifications/tasks/status',
-                            params: {
-                                task: updatedTask
-                            }
+                            params: updatedTask
                         });
                         await this.notification(notification as SendNotificationT);
 

src/types.ts

@@ -673,9 +673,7 @@ export const CreateTaskResultSchema = ResultSchema.extend({
 /**
  * Parameters for task status notification.
  */
-export const TaskStatusNotificationParamsSchema = z.object({
-    task: TaskSchema
-});
+export const TaskStatusNotificationParamsSchema = NotificationsParamsSchema.merge(TaskSchema);
 
 /**
  * A notification sent when a task's status changes.