iv@dZddlmZddlZddlZddlmZmZmZm Z ddl Z ddl m Z er ddlZddlmZmZddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddl m!Z!ddl"m#Z#ee$Z%e e jLjNe jLj zZ(GddZ) d ddZ*y)z(Tool-related methods for FastMCP Client.) annotationsN) TYPE_CHECKINGAnyLiteraloverload) RootModel)CallToolResultClient)ProgressHandler)ToolTask) client_span) ToolError)inject_trace_context)json_schema_to_type) get_logger)normalize_timeout_to_timedelta)get_cached_typeadapterc eZdZdZdd ddZddZ d ddZ d ddZe ddddd ddd  dd Z e ddddd ddd d  ddZ ddddd dddd d ddZ d ddZ y)ClientToolsMixinz0Mixin providing tool-related methods for Client.NcursorcKtjd|jd|j|jj |d{}|S7w)a Send a tools/list request and return the complete MCP protocol result. Args: cursor: Optional pagination cursor from a previous request's nextCursor. Returns: mcp.types.ListToolsResult: The complete response object from the protocol, containing the list of tools and any additional metadata. Raises: RuntimeError: If called while the client is not connected. McpError: If the request results in a TimeoutError | JSONRPCError [z] called list_toolsrN)loggerdebugname_await_with_session_monitoringsession list_tools)selfrresults k/Users/bowang/.openclaw/workspace/ChatDev/.venv/lib/python3.12/site-packages/fastmcp/client/mixins/tools.pylist_tools_mcpzClientToolsMixin.list_tools_mcp%sX  q #678:: LL # #6 # 2    sAAAAcKg}d}t} |j|d{}|j|j|js |S|j|vr3t j d|jd|jd |S|j|j|j}7w)a.Retrieve all tools available on the server. This method automatically fetches all pages if the server paginates results, returning the complete list. For manual pagination control (e.g., to handle large result sets incrementally), use list_tools_mcp() with the cursor parameter. Returns: list[mcp.types.Tool]: A list of all Tool objects. Raises: RuntimeError: If called while the client is not connected. McpError: If the request results in a TimeoutError | JSONRPCError Nrrz.] Server returned duplicate pagination cursor z$ for list_tools; stopping pagination) setr#extendtools nextCursorrwarningradd)r all_toolsr seen_cursorsr!s r"rzClientToolsMixin.list_tools<s+- !!$ ..f.==F   V\\ *$$  L0 {#)),,PR   V.. /&&F=s%CB?BCc Ktd|d||jj5tj d|j d|t |}|j|jj||t||xs |j|r|ndd{}|cdddS7#1swYyxYww)aSend a tools/call request and return the complete MCP protocol result. This method returns the raw CallToolResult object, which includes an isError flag and other metadata. It does not raise an exception if the tool call results in an error. Args: name (str): The name of the tool to call. arguments (dict[str, Any]): Arguments to pass to the tool. timeout (datetime.timedelta | float | int | None, optional): The timeout for the tool call. Defaults to None. progress_handler (ProgressHandler | None, optional): The progress handler to use for the tool call. Defaults to None. meta (dict[str, Any] | None, optional): Additional metadata to include with the request. This is useful for passing contextual information (like user IDs, trace IDs, or preferences) that shouldn't be tool arguments but may influence server-side processing. The server can access this via `context.request_context.meta`. Defaults to None. Returns: mcp.types.CallToolResult: The complete response object from the protocol, containing the tool result and any additional metadata. Raises: RuntimeError: If called while the client is not connected. McpError: If the tool call requests results in a TimeoutError | JSONRPCError z tools/call z tools/call) session_idrz] called call_tool: N)r argumentsread_timeout_secondsprogress_callbackmeta) r transportget_session_idrrrrrr call_toolr_progress_handler)r rr/progress_handlertimeoutr2propagated_metar!s r" call_tool_mcpzClientToolsMixin.call_tool_mcp`s>$  ~~446  LL1TYYK';D6B C348O>> &&')G)P&6&P$:P:P,; 'F)    s/+C A>B=+B;,B=1 C ;B==CC FcKt|||jj|jj|j|d{S7w)aJParse an mcp.types.CallToolResult into our CallToolResult dataclass. Args: name: Tool name (for schema lookup) result: Raw MCP protocol result raise_on_error: Whether to raise ToolError on errors Returns: CallToolResult: Parsed result with structured data )rr!tool_output_schemas list_tools_fn client_nameraise_on_errorN)_parse_call_tool_resultr_tool_output_schemasrr)r rr!r?s r"r@z(ClientToolsMixin._parse_call_tool_resultsI"- $ A A,,11 )     sAA A AT)versionr8r7r?r2taskc KywN) r rr/rBr8r7r?r2rCs r"r5zClientToolsMixin.call_tools `)rBr8r7r?r2task_idttlc KywrErF) r rr/rBr8r7r?r2rCrIrJs r"r5zClientToolsMixin.call_tools rG)rBr8r7r?r2rCrIrJc:K|r t|ni} |i| jdid|i| d<|r"|j||| | | xsdd{S|j||xsi||| xsdd{} |j || |d{S7F7"7w)aCall a tool on the server. Unlike call_tool_mcp, this method raises a ToolError if the tool call results in an error. Args: name (str): The name of the tool to call. arguments (dict[str, Any] | None, optional): Arguments to pass to the tool. Defaults to None. version (str | None, optional): Specific tool version to call. If None, calls highest version. timeout (datetime.timedelta | float | int | None, optional): The timeout for the tool call. Defaults to None. progress_handler (ProgressHandler | None, optional): The progress handler to use for the tool call. Defaults to None. raise_on_error (bool, optional): Whether to raise an exception if the tool call results in an error. Defaults to True. meta (dict[str, Any] | None, optional): Additional metadata to include with the request. This is useful for passing contextual information (like user IDs, trace IDs, or preferences) that shouldn't be tool arguments but may influence server-side processing. The server can access this via `context.request_context.meta`. Defaults to None. task (bool): If True, execute as background task (SEP-1686). Defaults to False. task_id (str | None): Optional client-provided task ID (auto-generated if not provided). ttl (int): Time to keep results available in milliseconds (default 60s). Returns: CallToolResult | ToolTask: The content returned by the tool if task=False, or a ToolTask object if task=True. If the tool returns structured outputs, they are returned as a dataclass (if an output schema is available) or a dictionary; otherwise, a list of content blocks is returned. Note: to receive both structured and unstructured outputs, use call_tool_mcp instead and access the raw result object. Raises: ToolError: If the tool call results in an error. McpError: If the tool call request results in a TimeoutError | JSONRPCError RuntimeError: If called while the client is not connected. NfastmcprB)r2)rr/r8r7r2)r?)dictget_call_tool_as_taskr:r@) r rr/rBr8r7r?r2rCrIrJ request_metar!s r"r5zClientToolsMixin.call_tools`&*tDzr  '""9b1'7'L # 00i#L4HD1 ))o2-% *  11 &2     s6A BB%B4B5BBBBBc Kt|}tjjtjj ||xsitjj ||}|j |jj|td{}|j} t| tjjrd| jj} |jj!| t#|| |d} t%j&| |j(| <| S|j+|| d{} |xst-t/j0} t#|| || S776w)aCall a tool for background execution (SEP-1686). Returns a ToolTask object that handles both background and immediate execution. If the server accepts background execution, ToolTask will poll for results. If the server declines (graceful degradation), ToolTask wraps the immediate result. Args: name: Tool name to call arguments: Tool arguments task_id: Optional client-provided task ID (ignored, for backward compatibility) ttl: Time to keep results available in milliseconds (default 60s) meta: Optional request metadata (e.g., version info) Returns: ToolTask: Future-like object for accessing task status and results )rJ)rr/rC_meta)params)request result_typeN) tool_nameimmediate_result)rmcptypesCallToolRequestCallToolRequestParams TaskMetadatarr send_requestToolTaskResponseUnionroot isinstanceCreateTaskResultrCtaskId_submitted_task_idsr*r weakrefref_task_registryr@struuiduuid4)r rr/rIrJr2r9rUwrapped_result raw_resultserver_task_idtask_obj parsed_resultsynthetic_task_ids r"rPz#ClientToolsMixin._call_tool_as_tasksc4/t4))++9922#/rYY+++4% 3,  $BB LL % %1 &    $(( j#))"<"< ='__33N  $ $ ( ( 8ntH3:++h2GD   /O#'">">tZ"PPM ' <3tzz|+< !!.  - (Qs%B"F $F%B-F F 5F  F )r r r str | Nonereturnzmcp.types.ListToolsResult)r r rrzlist[mcp.types.Tool])NNN)r r rrhr/zdict[str, Any]r7ProgressHandler | Noner8'datetime.timedelta | float | int | Noner2dict[str, Any] | Nonerrmcp.types.CallToolResult)F) r r rrhr!rvr?boolrrr rE)r r rrhr/rurBrqr8rtr7rsr?rwr2rurCzLiteral[False]rrr )r r rrhr/rurBrqr8rtr7rsr?rwr2rurCz Literal[True]rIrqrJintrrr )r r rrhr/rurBrqr8rtr7rsr?rwr2rurCrwrIrqrJrxrrzCallToolResult | ToolTask)NNrHN)r r rrhr/rurIrqrJrxr2rurrr ) __name__ __module__ __qualname____doc__r#rr:r@rr5rPrFr"rr s: /3!+ ". P48;?&* 333"31 3 9 3 $ 3 "3r %    )     4,0  #;?37#&*$   )    9 1  $     ,0  #;?37#&*"   )    9 1  $       $,0E #;?37#&*"E E E )E  E 9 E 1E E $E E E E  #E T,0"&* CCC)C C  C $ C Cr}rc*Kddlm}ddlm}d}|jrE|rC|t j j|jdj} t| |jr ||vr|d{||vr|j|} | r| jdr=| jdijd} |jjd} n |j} t| } t| } | j| }n |j}||j|j|j&||j S7#t $r)}t"j%d|xsd d |Yd}~dd}~wwxYww) aParse an mcp.types.CallToolResult into our CallToolResult dataclass. Args: name: Tool name (for schema lookup) result: Raw MCP protocol result tool_output_schemas: Dictionary mapping tool names to their output schemas list_tools_fn: Async function to refresh tool schemas if needed client_name: Optional client name for logging raise_on_error: Whether to raise ToolError on errors Returns: CallToolResult: Parsed result with structured data r)cast)r Nzx-fastmcp-wrap-result propertiesr!rclientz$] Error parsing structured content: )contentstructured_contentr2datais_error)typingrfastmcp.client.clientr isErrorrYrZ TextContentrtextrstructuredContentrOrrvalidate_python Exceptionrerrorr2)rr!r<r=r>r?rr rmsg output_schemar output_type type_adapteres r"r@r@Zs}*4 D ~~.399((&..*;<AAn  ! ! ..#o%%** 3 7 7 =  $(()@A(5(9(9,(K(O(O$) .4-E-E-I-I(-S*-3-E-E*"5m"DK#9+#FL'778JKD!33D !33 [[   +&   LLK+8,,PQRPST   sCA,F/E=E>B)E'5FE F'F F FF)NF)rrhr!rvr<z dict[str, dict[str, Any] | None]r=rr>rqr?rwrrr )+r| __future__rrirerrrrr mcp.typesrYpydanticrdatetimerr r fastmcp.client.progressr fastmcp.client.tasksr fastmcp.client.telemetryr fastmcp.exceptionsrfastmcp.telemetryr"fastmcp.utilities.json_schema_typerfastmcp.utilities.loggingrfastmcp.utilities.timeoutrfastmcp.utilities.typesrryrrZrbr_rr@rFr}r"rs." 88<3)0(2B0D: H "#))"<"