KiFdZddlmZmZddlmZmZmZmZGddeZ y)z8 TaskStore - Abstract interface for task state storage. )ABCabstractmethod)ResultTask TaskMetadata TaskStatusc ReZdZdZe ddededzdefdZedededzfdZ e ddede dzd edzdefd Z eded e ddfd Z edede dzfd Ze ddedzdeeeedzffdZededefdZededdfdZededdfdZy) TaskStorez Abstract interface for task state storage. This is a pure storage interface - it doesn't manage execution. Implementations can use in-memory storage, databases, Redis, etc. All methods are async to support various backends. Nmetadatatask_idreturnc Kyw)a? Create a new task. Args: metadata: Task metadata (ttl, etc.) task_id: Optional task ID. If None, implementation should generate one. Returns: The created Task with status="working" Raises: ValueError: If task_id already exists N)selfr r s o/home/jay/workspace/scripts/.codegraph-venv/lib/python3.12/site-packages/mcp/shared/experimental/tasks/store.py create_taskzTaskStore.create_taskc Kyw)z Get a task by ID. Args: task_id: The task identifier Returns: The Task, or None if not found Nrrr s rget_taskzTaskStore.get_task(rrstatusstatus_messagec Kyw)a Update a task's status and/or message. Args: task_id: The task identifier status: New status (if changing) status_message: New status message (if changing) Returns: The updated Task Raises: ValueError: If task not found ValueError: If attempting to transition from a terminal status (completed, failed, cancelled). Per spec, terminal states MUST NOT transition to any other status. Nr)rr rrs r update_taskzTaskStore.update_task4rrresultc Kyw)z Store the result for a task. Args: task_id: The task identifier result: The result to store Raises: ValueError: If task not found Nr)rr rs r store_resultzTaskStore.store_resultMrrc Kyw)z Get the stored result for a task. Args: task_id: The task identifier Returns: The stored Result, or None if not available Nrrs r get_resultzTaskStore.get_resultZrrcursorc Kyw)z List tasks with pagination. Args: cursor: Optional cursor for pagination Returns: Tuple of (tasks, next_cursor). next_cursor is None if no more pages. Nr)rr!s r list_taskszTaskStore.list_tasksfrrc Kyw)z Delete a task. Args: task_id: The task identifier Returns: True if deleted, False if not found Nrrs r delete_taskzTaskStore.delete_taskurrc Kyw)a\ Wait until the task status changes. This blocks until either: 1. The task status changes 2. The wait is cancelled Used by tasks/result to wait for task completion or status changes. Args: task_id: The task identifier Raises: ValueError: If task not found Nrrs rwait_for_updatezTaskStore.wait_for_updaterrc Kyw)z Signal that a task has been updated. This wakes up any coroutines waiting in wait_for_update(). Args: task_id: The task identifier Nrrs r notify_updatezTaskStore.notify_updaterr)N)NN)__name__ __module__ __qualname____doc__rrstrrrrrrrrr tuplelistr#boolr%r'r)rrr r s#  t     &  c  dTk    %)%)   T! d     0  #  v  $            "  d    tDz3:% &             S T  " 3 4  r2r N) r-abcrr mcp.typesrrrrr rr2rr5s#$<<R R r2