chore(a2a): refactored the task manager in a2a

Author: daltonnyx

AgentCrew/__init__.py

@@ -1 +1 @@
-__version__ = "0.10.0"
+__version__ = "0.10.1"

AgentCrew/modules/a2a/exceptions.py

@@ -0,0 +1,5 @@
+from __future__ import annotations
+
+
+class TaskCanceledException(Exception):
+    """Raised when a task is explicitly canceled during processing."""

AgentCrew/modules/a2a/task_cancellation.py

@@ -0,0 +1,32 @@
+from __future__ import annotations
+
+import asyncio
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from typing import Dict
+
+
+class TaskCancellationManager:
+    def __init__(self) -> None:
+        self.cancel_events: Dict[str, asyncio.Event] = {}
+        self.background_tasks: Dict[str, asyncio.Task] = {}
+
+    def register(self, task_id: str, bg_task: asyncio.Task) -> None:
+        cancel_event = asyncio.Event()
+        self.cancel_events[task_id] = cancel_event
+        self.background_tasks[task_id] = bg_task
+
+    def signal_cancel(self, task_id: str) -> None:
+        if task_id in self.cancel_events:
+            self.cancel_events[task_id].set()
+        if task_id in self.background_tasks:
+            self.background_tasks[task_id].cancel()
+
+    def is_canceled(self, task_id: str) -> bool:
+        event = self.cancel_events.get(task_id)
+        return event is not None and event.is_set()
+
+    def cleanup(self, task_id: str) -> None:
+        self.cancel_events.pop(task_id, None)
+        self.background_tasks.pop(task_id, None)