)`i!dZddlmZmZddlmZmZddlmZddl m Z ddl m Z ddl mZddlmZdd lmZmZdd lmZmZmZmZmZmZmZmZmZmZeGd d Zd S)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 eZdZUdZdZedzed<eddZe dzed<eddZ e dzed<eddZ e dzed<ed efd Zed efd Zd d dedzded edzfdZd d deded edzfdZdedzd efdZddddeegeefdedzdedzd efdZdS) 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 {/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/mcp/server/experimental/request_context.pyis_taskzExperimental.is_task1s!--c0|jdS|jjduS)z*Check if the client declared task support.NF)rtasksr"s r$client_supports_tasksz"Experimental.client_supports_tasks6s#  $ ,5(.d::r&T raise_errortool_task_moder+c|pt}d}|tkr|jsttd}n(|tkr|jrttd}||rt ||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=s4/"& =  %FEE^ # # #%NE   5// ! r&toolcZ|jr |jjnd}|||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_toolks3$.2^Et~))&&t&EEEr&c@|pt}|tkr |jsdSdS)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 5tr&)task_idmodel_immediate_responseworkr:r;c K|jtd|jtd|jtd|j}|j}|j|j|d{V}t||j|j|j|j  d fd }| |d}| t|i}td d|i|rd |iniS) 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.)taskstoresessionqueuehandlerr cZK d{V}tjjs|d{VdSdS#t$rR}tjjs.t |d{VYd}~dSYd}~dSd}~wwxYw)N)r r>statuscomplete Exceptionfailstr)resultetask_ctxr<s r$executez&Experimental.run_task..executes 0#tH~~------"8=#7884"++F3333333333344 0 0 0"8=#7880"--A///////////////000000 0sAA B*AB%%B*r>_meta)r N) r RuntimeErrorrr task_groupr? create_taskrrArB start_soonr r) r#r<r:r;supportrPr>rLmetarKs ` @r$run_taskzExperimental.run_tasks9j   %ijj j = 788 8   %A  $' ]..t/A7KKKKKKKK$-M-O     0 0 0 0 0 0 0 g&&&&* # /02JKDMMTM-Lgt__"MMMr&)__name__ __module__ __qualname____doc__rr__annotations__rrrrr rr propertyboolr%r)rrr2rr7r9rrrrrHrrUrNr&r$rr!s/*.M<$&---6;eDu6U6U6U,t3UUU%*U4e%D%D%DHmd"DDD(-d(G(G(GM;%GGG ....X.;t;;;X;! ,,,)D0, , T  ,,,,d! FFFF F T  FFFF*+\Nt \N #&* \N  \N\N\N\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$rfs}0///////((((((((BBBBBB<<<<<<,,,,,,******[[[[[[[[                         LNLNLNLNLNLNLN LNLNLNr&