)`idZddlmZddlmZmZmZddlmZddl m Z erddl m Z edej ZGd d ZdS) a Experimental client-side task support. This module provides client methods for interacting with MCP tasks. WARNING: These APIs are experimental and may change without notice. Example: # Call a tool as a task result = await session.experimental.call_tool_as_task("tool_name", {"arg": "value"}) task_id = result.task.taskId # Get task status status = await session.experimental.get_task(task_id) # Get task result when complete if status.status == "completed": result = await session.experimental.get_task_result(task_id, CallToolResult) # List all tasks tasks = await session.experimental.list_tasks() # Cancel a task await session.experimental.cancel_task(task_id) ) AsyncIterator) TYPE_CHECKINGAnyTypeVarN)poll_until_terminal) ClientSessionResultT)boundc eZdZdZddZ ddddd ed eeefdzd ed eeefdzde j f d Z dede j fdZ dedeedefdZ ddedzde jfdZdede jfdZdedee j fdZdS)ExperimentalClientFeaturesz Experimental client features for tasks and other experimental APIs. WARNING: These APIs are experimental and may change without notice. Access via session.experimental: status = await session.experimental.get_task(task_id) sessionrreturnNc||_dSN)_session)selfr s q/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/mcp/client/experimental/tasks.py__init__z#ExperimentalClientFeatures.__init__1s  i`)ttlmetaname argumentsrrc ,Kd}|tjjdi|}|jtjtjtj||tj||tj d{VS)aCall a tool as a task, returning a CreateTaskResult for polling. This is a convenience method for calling tools that support task execution. The server will return a task reference instead of the immediate result, which can then be polled via `get_task()` and retrieved via `get_task_result()`. Args: name: The tool name arguments: Tool arguments ttl: Task time-to-live in milliseconds (default: 60000 = 1 minute) meta: Optional metadata to include in the request Returns: CreateTaskResult containing the task reference Example: # Create task result = await session.experimental.call_tool_as_task( "long_running_tool", {"input": "data"} ) task_id = result.task.taskId # Poll for completion while True: status = await session.experimental.get_task(task_id) if status.status == "completed": break await asyncio.sleep(0.5) # Get result final = await session.experimental.get_task_result(task_id, CallToolResult) N)r)rrtask_metaparams) types RequestParamsMetar send_request ClientRequestCallToolRequestCallToolRequestParams TaskMetadataCreateTaskResult)rrrrrrs rcall_tool_as_taskz,ExperimentalClientFeatures.call_tool_as_task4sP26  ',44t44E]//  % 6!"+"/C888#     "         rtask_idc K|jtjtjtj|tjd{VS)z Get the current status of a task. Args: task_id: The task identifier Returns: GetTaskResult containing the task status and metadata taskIdrN)rr#r r$GetTaskRequestGetTaskRequestParams GetTaskResultrr*s rget_taskz#ExperimentalClientFeatures.get_taskns~]//  $ 5WEEE              r result_typec K|jtjtjtj||d{VS)a Get the result of a completed task. The result type depends on the original request type: - tools/call tasks return CallToolResult - Other request types return their corresponding result type Args: task_id: The task identifier result_type: The expected result type (e.g., CallToolResult) Returns: The task result, validated against result_type r,rN)rr#r r$GetTaskPayloadRequestGetTaskPayloadRequestParams)rr*r3s rget_task_resultz*ExperimentalClientFeatures.get_task_results{&]//  + D(8  8 38nt# 8   8 8 8 8 t c e.A    &  ']       <"  d        *  1G    &s}U=P/Qrr )rIcollections.abcrtypingrrr mcp.typesr %mcp.shared.experimental.tasks.pollingrmcp.client.sessionrResultr r rrrrTs4*)))))..........EEEEEE1000000 ')5< 0 0 0yyyyyyyyyyr