idZddlmZddlmZmZddlZddlmZerddl m Z ddlm Z m Z m Z mZmZmZmZmZmZmZddlmZeeZGd d Zy) z+Task management methods for FastMCP Client.) annotations) TYPE_CHECKINGAnyN)McpError)Client) CancelTaskRequestCancelTaskRequestParamsGetTaskPayloadRequestGetTaskPayloadRequestParamsGetTaskPayloadResultGetTaskRequestGetTaskRequestParams GetTaskResultListTasksRequestPaginatedRequestParams) get_loggercDeZdZdZddZddZ d d dZd dZy) ClientTaskManagementMixinz3Mixin providing task management methods for Client.cKtt|}|j|jj |t d{S7w)aQuery the status of a background task. Sends a 'tasks/get' MCP protocol request over the existing transport. Args: task_id: The task ID returned from call_tool_as_task Returns: GetTaskResult: Status information including taskId, status, pollInterval, etc. Raises: RuntimeError: If client not connected McpError: If the request results in a TimeoutError | JSONRPCError taskIdparamsrequest result_typeN)r r_await_with_session_monitoringsession send_requestrselftask_idrs u/Users/bowang/.openclaw/workspace/ChatDev/.venv/lib/python3.12/site-packages/fastmcp/client/mixins/task_management.pyget_task_statusz)ClientTaskManagementMixin.get_task_status!sR!( :: LL % %0 &     dT BB sA A& A$ A&NcKt||}t|}|j|jj |t j jd{}|jr|jdSg}t|jd|D]=} |j|d{}|j|jd?|ddS77.#t$rYTwxYww)aList background tasks. Sends a 'tasks/list' MCP protocol request to the server. If the server returns an empty list (indicating client-side tracking), falls back to querying status for locally tracked task IDs. Args: cursor: Optional pagination cursor limit: Maximum number of tasks to return (default 50) Returns: dict: Response with structure: - tasks: List of task status dicts with taskId, status, etc. - nextCursor: Optional cursor for next page Raises: RuntimeError: If client not connected McpError: If the request results in a TimeoutError | JSONRPCError )cursorlimitrrNT)r')tasks nextCursor)rrrrrmcptypesListTasksResultr.r(list_submitted_task_idsr$appendr) r!r,r-rrserver_responser.r"statuss r# list_tasksz$ClientTaskManagementMixin.list_tasksUs2(vUC"&1 $ C C LL % %II55 & !    "--t-< <D445fu=G #33G<< V...=>>d33+ =  sHA!C8#C%$?C8$C)8C'9%C)C8'C)) C52C84C55C8cKtt|}|j|jj |t j jd{S7w)aCancel a task, transitioning it to cancelled state. Sends a 'tasks/cancel' MCP protocol request. Task will halt execution and transition to cancelled state. Args: task_id: The task ID to cancel Returns: CancelTaskResult: The task status showing cancelled state Raises: RuntimeError: If task doesn't exist McpError: If the request results in a TimeoutError | JSONRPCError rrrN)rr rrrr0r1CancelTaskResultr s r# cancel_taskz%ClientTaskManagementMixin.cancel_tasks\ $+B'+RS88 LL % %II66 &     sAA' A%!A')r!rr"strreturnr)r!rr"r<r=r)N2)r!rr,z str | Noner-intr=zdict[str, Any])r!rr"r<r=zmcp.types.CancelTaskResult)__name__ __module__ __qualname____doc__r$r*r8r;r#rrsG= .C>"04040404  04d rEr)rC __future__rtypingrr mcp.typesr0rfastmcp.client.clientrrr r r r r rrrrfastmcp.utilities.loggingrr@loggerrrDrEr#rLsD1"%,   1 H   rE