Change the TaskUpdater constructor to take the RequestContext

Author: kabir

src/main/java/io/a2a/server/tasks/TaskUpdater.java

@@ -4,6 +4,7 @@
 import java.util.Map;
 import java.util.UUID;
 
+import io.a2a.server.agentexecution.RequestContext;
 import io.a2a.server.events.EventQueue;
 import io.a2a.spec.Artifact;
 import io.a2a.spec.Message;
@@ -18,10 +19,10 @@ public class TaskUpdater {
     private final String taskId;
     private final String contextId;
 
-    public TaskUpdater(EventQueue eventQueue, String taskId, String contextId) {
+    public TaskUpdater(EventQueue eventQueue, RequestContext context) {
         this.eventQueue = eventQueue;
-        this.taskId = taskId;
-        this.contextId = contextId;
+        this.taskId = context.getTaskId();
+        this.contextId = context.getContextId();
     }
 
     public void updateStatus(TaskState taskState) {

src/test/java/io/a2a/server/tasks/TaskUpdaterTest.java

@@ -11,6 +11,7 @@
 import java.util.List;
 import java.util.Map;
 
+import io.a2a.server.agentexecution.RequestContext;
 import io.a2a.server.events.Event;
 import io.a2a.server.events.EventQueue;
 import io.a2a.spec.Message;
@@ -43,7 +44,11 @@ public class TaskUpdaterTest {
     @BeforeEach
     public void init() {
         eventQueue = EventQueue.create();
-        taskUpdater = new TaskUpdater(eventQueue, TEST_TASK_ID, TEST_TASK_CONTEXT_ID);
+        RequestContext context = new RequestContext.Builder()
+                .setTaskId(TEST_TASK_ID)
+                .setContextId(TEST_TASK_CONTEXT_ID)
+                .build();
+        taskUpdater = new TaskUpdater(eventQueue, context);
     }
 
     //@Test

tck/src/main/java/io/a2a/tck/server/AgentExecutorProducer.java

@@ -43,7 +43,7 @@ public void execute(RequestContext context, EventQueue eventQueue) throws JSONRP
                 eventQueue.enqueueEvent(task);
             }
 
-            TaskUpdater updater = new TaskUpdater(eventQueue, context.getTaskId(), context.getTaskId());
+            TaskUpdater updater = new TaskUpdater(eventQueue, context);
 
             // Immediately set to WORKING state
             updater.startWork();
@@ -68,7 +68,7 @@ public void cancel(RequestContext context, EventQueue eventQueue) throws JSONRPC
                 throw new TaskNotCancelableError();
             }
 
-            TaskUpdater updater = new TaskUpdater(eventQueue, context.getTaskId(), context.getTaskId());
+            TaskUpdater updater = new TaskUpdater(eventQueue, context);
             updater.cancel();
             eventQueue.enqueueEvent(new TaskStatusUpdateEvent.Builder()
                     .taskId(task.getId())