o i; @s:dZddlmZddlmZmZmZmZGdddZdS)z TaskContext - Pure task state management. This module provides TaskContext, which manages task state without any server/session dependencies. It can be used standalone for distributed workers or wrapped by ServerTaskContext for full server integration. ) TaskStore)TASK_STATUS_COMPLETEDTASK_STATUS_FAILEDResultTaskc@seZdZdZdedefddZedefddZ edefd d Z ede fd d Z dddZ dedd fddZdedd fddZdedd fddZd S) TaskContexta Pure task state management - no session dependencies. This class handles: - Task state (status, result) - Cancellation tracking - Store interactions For server-integrated features (elicit, create_message, notifications), use ServerTaskContext from mcp.server.experimental. Example (distributed worker): async def worker_job(task_id: str): store = RedisTaskStore(redis_url) task = await store.get_task(task_id) ctx = TaskContext(task=task, store=store) await ctx.update_status("Working...") result = await do_work() await ctx.complete(result) taskstorecCs||_||_d|_dS)NF)_task_store _cancelled)selfrr rY/home/ubuntu/.local/lib/python3.10/site-packages/mcp/shared/experimental/tasks/context.py__init__$s zTaskContext.__init__returncCs|jjS)zThe task identifier.)r taskIdr rrrtask_id)szTaskContext.task_idcC|jS)zThe current task state.)r rrrrr.zTaskContext.taskcCr)z(Whether cancellation has been requested.r rrrr is_cancelled3rzTaskContext.is_cancelledNcCs d|_dS)z Request cancellation of this task. This sets is_cancelled=True. Task work should check this periodically and exit gracefully if set. TNrrrrrrequest_cancellation8s z TaskContext.request_cancellationmessagecs |jj|j|dIdH|_dS)zn Update the task's status message. Args: message: The new status message )status_messageN)r update_taskrr )r rrrr update_statusAs zTaskContext.update_statusresultcs6|j|j|IdH|jj|jtdIdH|_dS)zv Mark the task as completed with the given result. Args: result: The task result N)status)r store_resultrrrr )r rrrrcompleteMs zTaskContext.completeerrorcs"|jj|jt|dIdH|_dS)zt Mark the task as failed with an error message. Args: error: The error message )rrN)r rrrr )r r"rrrfailZs zTaskContext.fail)rN)__name__ __module__ __qualname____doc__rrrpropertystrrrboolrrrrr!r#rrrrr s    rN) r'#mcp.shared.experimental.tasks.storer mcp.typesrrrrrrrrrs