i!dZddlmZmZddlmZmZddlmZddl m Z ddl m Z ddl mZddlmZdd lmZmZdd lmZmZmZmZmZmZmZmZmZmZeGd d Zy )a  Experimental request context features. This module provides the Experimental class which gives access to experimental features within a request context, such as task-augmented request handling. WARNING: These APIs are experimental and may change without notice. ) AwaitableCallable) dataclassfield)Any)ServerTaskContext) TaskSupport) ServerSession)McpError)MODEL_IMMEDIATE_RESPONSE_KEY is_terminal) METHOD_NOT_FOUNDTASK_FORBIDDEN TASK_REQUIREDClientCapabilitiesCreateTaskResult ErrorDataResultTaskExecutionMode TaskMetadataToolc XeZdZUdZdZedzed<eddZe dzed<eddZ e dzed<eddZ e dzed<ed efd Zed efd Zd d dedzded edzfdZd d deded edzfdZdedzd efdZddddeegeefdedzdedzd efdZy) Experimentala Experimental features context for task-augmented requests. Provides helpers for validating task execution compatibility and running tasks with automatic lifecycle management. WARNING: This API is experimental and may change without notice. N task_metadataF)defaultrepr_client_capabilities_session _task_supportreturnc|jduS)z(Check if this request is task-augmented.N)rselfs g/root/aria/mcps/aria-brain/venv/lib/python3.12/site-packages/mcp/server/experimental/request_context.pyis_taskzExperimental.is_task1s!!--cL|jy|jjduS)z*Check if the client declared task support.NF)rtasksr"s r$client_supports_tasksz"Experimental.client_supports_tasks6s*  $ $ ,((..d::r&T raise_errortool_task_moder+c|xst}d}|tk(r|jsttd}n&|tk(r|jrttd}| |r t ||S)a Validate that the request is compatible with the tool's task execution mode. Per MCP spec: - "required": Clients MUST invoke as task. Server returns -32601 if not. - "forbidden" (or None): Clients MUST NOT invoke as task. Server returns -32601 if they do. - "optional": Either is acceptable. Args: tool_task_mode: The tool's execution.taskSupport value ("forbidden", "optional", "required", or None) raise_error: If True, raises McpError on validation failure. If False, returns ErrorData. Returns: None if valid, ErrorData if invalid and raise_error=False Raises: McpError: If invalid and raise_error=True Nz,This tool requires task-augmented invocation)codemessagez4This tool does not support task-augmented invocation)rrr%rrr )r#r,r+modeerrors r$validate_task_modezExperimental.validate_task_mode=sj4/"& = %FE^ # %NE  5/ ! r&toolcp|jr|jjnd}|j||S)a Validate that the request is compatible with the given tool. Convenience wrapper around validate_task_mode that extracts the mode from a Tool. Args: tool: The Tool definition raise_error: If True, raises McpError on validation failure. Returns: None if valid, ErrorData if invalid and raise_error=False Nr*) execution taskSupportr2)r#r3r+r0s r$validate_for_toolzExperimental.validate_for_toolks1$.2^^t~~))&&t&EEr&cD|xst}|tk(r |jsyy)a Check if this client can use a tool with the given task mode. Useful for filtering tool lists or providing warnings. Returns False if tool requires "required" but client doesn't support tasks. Args: tool_task_mode: The tool's execution.taskSupport value Returns: True if the client can use this tool, False otherwise FT)rrr))r#r,r0s r$ can_use_toolzExperimental.can_use_tools$/ = )C)Cr&)task_idmodel_immediate_responseworkr:r;c  K|j td|j td|j td|j}|j}|j j |j|d{}t||j |j|j|j d fd }|j|d}|t|i}td d|i|rd|iSiS7uw) as Create a task, spawn background work, and return CreateTaskResult immediately. This is the recommended way to handle task-augmented tool calls. It: 1. Creates a task in the store 2. Spawns the work function in a background task 3. Returns CreateTaskResult immediately The work function receives a ServerTaskContext with: - elicit() for sending elicitation requests - create_message() for sampling requests - update_status() for progress updates - complete()/fail() for finishing the task When work() returns a Result, the task is auto-completed with that result. If work() raises an exception, the task is auto-failed. Args: work: Async function that does the actual work task_id: Optional task ID (generated if not provided) model_immediate_response: Optional string to include in _meta as io.modelcontextprotocol/model-immediate-response Returns: CreateTaskResult to return to the client Raises: RuntimeError: If task support is not enabled or task_metadata is missing Example: @server.call_tool() async def handle_tool(name: str, args: dict): ctx = server.request_context async def work(task: ServerTaskContext) -> CallToolResult: result = await task.elicit( message="Are you sure?", requestedSchema={"type": "object", ...} ) confirmed = result.content.get("confirm", False) return CallToolResult(content=[TextContent(text="Done" if confirmed else "Cancelled")]) return await ctx.experimental.run_task(work) WARNING: This API is experimental and may change without notice. NzHTask support not enabled. Call server.experimental.enable_tasks() first.zSession not available.zgRequest is not task-augmented (no task field in params). The client must send a task-augmented request.)taskstoresessionqueuehandlercdK d{}tjjsj|d{yy7>7#t$rQ}tjjs(j t |d{7Yd}~yYd}~yd}~wwxYww)N)r r>statuscomplete Exceptionfailstr)resultetask_ctxr<s r$executez&Experimental.run_task..executes 0#H~-"8==#7#78"++F3339.3 0"8==#7#78"--A///9 0s\B0 AA7AA A B0AA B-_meta)r N) r RuntimeErrorrr task_groupr? create_taskrrArB start_soonr r) r#r<r:r;supportrPr>rLmetarKs ` @r$run_taskzExperimental.run_tasks j    %ij j == 78 8    %A  $$'' ]]..t/A/A7KK$--MM--OO   0 g&&* # /02JKDMTMgt_MM"MM3LsBD D A6D)__name__ __module__ __qualname____doc__rr__annotations__rrrrr rr propertyboolr%r)rrr2rr7r9rrrrrHrrUrNr&r$rr!sa*.M<$&-6;Du6U,t3U%*4e%DHmd"D(-d(GM;%G ...;t;;! ,)D0, , T  ,d! FF F T  F*+\Nt \N #&* \N  \Nr&rN) rYcollections.abcrr dataclassesrrtypingr$mcp.server.experimental.task_contextr$mcp.server.experimental.task_supportr mcp.server.sessionr mcp.shared.exceptionsr %mcp.shared.experimental.tasks.helpersr r mcp.typesrrrrrrrrrrrrNr&r$rfsR0(B<,*[    LNLN LNr&