fix: force itk agent to create task before updating the status

Author: Krzysztof Dziedzic

.github/workflows/itk.yaml

@@ -28,4 +28,4 @@ jobs:
         run: bash run_itk.sh
         working-directory: itk
         env:
-          A2A_SAMPLES_REVISION: itk-v.0.11-alpha
+          A2A_SAMPLES_REVISION: itk-v.016-alpha

itk/README.md

@@ -36,7 +36,7 @@ You must set the `A2A_SAMPLES_REVISION` environment variable to specify which re
 
 Example:
 ```bash
-export A2A_SAMPLES_REVISION=itk-v.0.11-alpha
+export A2A_SAMPLES_REVISION=itk-v.016-alpha
 ```
 
 ### 2. Execute Tests

itk/main.py

@@ -31,7 +31,9 @@
     Message,
     Part,
     SendMessageRequest,
+    Task,
     TaskState,
+    TaskStatus,
 )
 from a2a.utils import TransportProtocol
 
@@ -196,7 +198,16 @@ async def execute(
             context.context_id,
         )
 
-        await task_updater.update_status(TaskState.TASK_STATE_SUBMITTED)
+        # Explicitly create the task by sending it to the queue
+        task = Task(
+            id=context.task_id,
+            context_id=context.context_id,
+            status=TaskStatus(state=TaskState.TASK_STATE_SUBMITTED),
+            history=[context.message] if context.message else [],
+        )
+        async with task_updater._lock:
+            await event_queue.enqueue_event(task)
+
         await task_updater.update_status(TaskState.TASK_STATE_WORKING)
 
         instruction = extract_instruction(context.message)