)`iNdZddlmZmZddlmZmZmZmZGddeZ dS)z8 TaskStore - Abstract interface for task state storage. )ABCabstractmethod)ResultTask TaskMetadata TaskStatusc eZdZdZe ddededzdefdZedededzfdZ e ddede dzd edzdefd Z eded e ddfd Z edede dzfd Ze ddedzdeeeedzffdZededefdZededdfdZededdfdZdS) 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 KdS)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 w/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/mcp/shared/experimental/tasks/store.py create_taskzTaskStore.create_task c KdS)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 KdS)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 KdS)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 KdS)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 KdS)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 KdS)z Delete a task. Args: task_id: The task identifier Returns: True if deleted, False if not found Nrrs r delete_taskzTaskStore.delete_taskurrc KdS)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 KdS)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)rrrr r s9#   t      ^ &  c  dTk    ^  %)%)    T! d      ^ 0  #  v  $    ^          ^  "   d    tDz3:% &    ^          ^   S T   ^ " 3 4   ^   rr N) r-abcrr mcp.typesrrrrr rrrr4s$#######<<<<<<<<<<<<R R R R R R R R R R r