)`idZddlmZddlmZddlmZmZddlmZddl m Z ddl m Z ddl mZdd lmZmZmZmZmZmZmZmZmZmZd Zd Zd ed efdZdeded efdZd efdZ ddededzd efdZ!ededed ee fdZ"dS)z Helper functions for pure task management. These helpers work with pure TaskContext and don't require server dependencies. For server-integrated task helpers, use mcp.server.experimental. ) AsyncIterator)asynccontextmanager)datetimetimezone)uuid4)McpError) TaskContext) TaskStore) INVALID_PARAMSTASK_STATUS_CANCELLEDTASK_STATUS_COMPLETEDTASK_STATUS_FAILEDTASK_STATUS_WORKINGCancelTaskResult ErrorDataTask TaskMetadata TaskStatusz0io.modelcontextprotocol/model-immediate-responsez$io.modelcontextprotocol/related-taskstatusreturnc.|tttfvS)a Check if a task status represents a terminal state. Terminal states are those where the task has finished and will not change. Args: status: The task status to check Returns: True if the status is terminal (completed, failed, or cancelled) )r rr rs y/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/mcp/shared/experimental/tasks/helpers.py is_terminalr&s +-?AVW WWstoretask_idcK||d{V}|&tttd|t |jr,tttd|jd||td{V}tdi| S)a7 Cancel a task with spec-compliant validation. Per spec: "Receivers MUST reject cancellation of terminal status tasks with -32602 (Invalid params)" This helper validates that the task exists and is not in a terminal state before setting it to "cancelled". Args: store: The task store task_id: The task identifier to cancel Returns: CancelTaskResult with the cancelled task state Raises: McpError: With INVALID_PARAMS (-32602) if: - Task does not exist - Task is already in a terminal state (completed, failed, cancelled) Example: @server.experimental.cancel_task() async def handle_cancel(request: CancelTaskRequest) -> CancelTaskResult: return await cancel_task(store, request.params.taskId) NzTask not found: )codemessagez&Cannot cancel task in terminal state ''r) get_taskrrr rr update_taskr r model_dump)rrtaskcancelled_tasks r cancel_taskr(5s<(( ( ( ( ( ( (D | #4744      4;  #OOOO      !,,W=R,SSSSSSSSN  : :n7799 : ::rc8ttS)zGenerate a unique task ID.)strrr"rrgenerate_task_idr+is uww<<rNmetadatactjtj}t |p t t |||jdS)a Create a Task object with initial state. This is a helper for TaskStore implementations. Args: metadata: Task metadata task_id: Optional task ID (generated if not provided) Returns: A new Task in "working" status i)taskIdr createdAt lastUpdatedAtttl pollInterval)rnowrutcrr+rr1)r,rr3s rcreate_task_stater5nsL ,x| $ $C ,*,," L    rcTK||d{V}|td|dt||} |WVdS#t$rR}t |jjs.|t|d{VYd}~dSYd}~dSd}~wwxYw)a Context manager for safe task execution (pure, no server dependencies). Loads a task from the store and provides a TaskContext for the work. If an unhandled exception occurs, the task is automatically marked as failed and the exception is suppressed (since the failure is captured in task state). This is useful for distributed workers that don't have a server session. Args: task_id: The task identifier to execute store: The task store (must be accessible by the worker) Yields: TaskContext for updating status and completing/failing the task Raises: ValueError: If the task is not found in the store Example (distributed worker): async def worker_process(task_id: str): store = RedisTaskStore(redis_url) async with task_execution(task_id, store) as ctx: await ctx.update_status("Working...") result = await do_work() await ctx.complete(result) NzTask z not found) r# ValueErrorr Exceptionrr&rfailr*)rrr&ctxes rtask_executionr<s@(( ( ( ( ( ( (D |4444555 dE " "C# ###38?++ #((3q66"" " " " " " " " " " " " " " # # # # # ##sA B'AB""B')N)#__doc__collections.abcr contextlibrrruuidrmcp.shared.exceptionsr%mcp.shared.experimental.tasks.contextr #mcp.shared.experimental.tasks.storer mcp.typesr r r rrrrrrrMODEL_IMMEDIATE_RESPONSE_KEYRELATED_TASK_METADATA_KEYboolrr*r(r+r5r<r"rrrHs*)))))******''''''''******======999999                         RC X Xt X X X X1; 1; 1;1;1;1;1;h# 4Z 6*# *# *#;*#*#*#*#*#*#r