Fix TaskStatusNotificationParams to spread task fields per spec

Author: felixweinberger

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

@@ -672,10 +672,9 @@ export const CreateTaskResultSchema = ResultSchema.extend({
 
 /**
  * Parameters for task status notification.
+ * Task fields are spread directly into params per the spec (NotificationParams & Task).
  */
-export const TaskStatusNotificationParamsSchema = z.object({
-    task: TaskSchema
-});
+export const TaskStatusNotificationParamsSchema = NotificationsParamsSchema.merge(TaskSchema);
 
 /**
  * A notification sent when a task's status changes.