o c۷i@s@dZddlmZmZddlmZmZmZmZGdddeZ dS)z8 TaskStore - Abstract interface for task state storage. )ABCabstractmethod)ResultTask TaskMetadata TaskStatusc @seZdZdZe ddededBdefddZedededBfdd Z e  dded e dBd edBdefd d Z edede ddfddZ edede dBfddZe ddedBdeeeedBffddZededefddZededdfddZededdfddZdS) 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_idreturncdS)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 r r Y/home/ubuntu/vllm_env/lib/python3.10/site-packages/mcp/shared/experimental/tasks/store.py create_taskzTaskStore.create_taskcr )z Get a task by ID. Args: task_id: The task identifier Returns: The Task, or None if not found Nr rr r r rget_task(rzTaskStore.get_taskstatusstatus_messagecr )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 rrr r r update_task4rzTaskStore.update_taskresultcr )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 rr r r store_resultMrzTaskStore.store_resultcr )z Get the stored result for a task. Args: task_id: The task identifier Returns: The stored Result, or None if not available Nr rr r r get_resultZrzTaskStore.get_resultcursorcr )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 )rrr r r list_tasksfrzTaskStore.list_taskscr )z Delete a task. Args: task_id: The task identifier Returns: True if deleted, False if not found Nr rr r r delete_taskurzTaskStore.delete_taskcr )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 Nr rr r rwait_for_updaterzTaskStore.wait_for_updatecr )z Signal that a task has been updated. This wakes up any coroutines waiting in wait_for_update(). Args: task_id: The task identifier Nr rr r r notify_updaterzTaskStore.notify_update)N)NN)__name__ __module__ __qualname____doc__rrstrrrrrrrrrtuplelistrboolrrrr r r rr sT        rN) r"abcrr mcp.typesrrrrrr r r rs