i; @dZddlmZddlmZmZmZmZGddZy)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_FAILEDResultTaskceZdZdZdedefdZedefdZ edefdZ ede fdZ dd Z d edd fd Zd edd fdZdedd fdZy ) 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) taskstorec.||_||_d|_y)NF)_task_store _cancelled)selfr r s e/root/aria/mcps/aria-brain/venv/lib/python3.12/site-packages/mcp/shared/experimental/tasks/context.py__init__zTaskContext.__init__$s  returnc.|jjS)zThe task identifier.)r taskIdrs rtask_idzTaskContext.task_id)szz   rc|jS)zThe current task state.)r rs rr zTaskContext.task.szzrc|jS)z(Whether cancellation has been requested.rrs r is_cancelledzTaskContext.is_cancelled3srNcd|_y)z Request cancellation of this task. This sets is_cancelled=True. Task work should check this periodically and exit gracefully if set. TNrrs rrequest_cancellationz TaskContext.request_cancellation8s rmessagecxK|jj|j|d{|_y7 w)zn Update the task's status message. Args: message: The new status message )status_messageN)r update_taskrr )rrs r update_statuszTaskContext.update_statusAs7 ;;22 LL"3    s +:8 :resultcK|jj|j|d{|jj|jtd{|_y7=7 w)zv Mark the task as completed with the given result. Args: result: The task result N)status)r store_resultrr"rr )rr$s rcompletezTaskContext.completeMs[kk&&t||V<<<;;22 LL(3    = s!*A.A*2A.A, A.,A.errorcK|jj|jt|d{|_y7 w)zt Mark the task as failed with an error message. Args: error: The error message )r&r!N)rr"rrr )rr)s rfailzTaskContext.failZs: ;;22 LL% 3    s 0?= ?)rN)__name__ __module__ __qualname____doc__rrrpropertystrrr boolrrr#rr(r+rrr r s, T ) !!!dd  3  4    V          rr N) r/#mcp.shared.experimental.tasks.storer mcp.typesrrrrr r3rrr6s!:MMX X r