i&dZddlmZddlZddlmZddlmZddlm Z ddlm Z ddl m Z dd lmZdd lmZdd lmZdd lmZdd lmZGddeZy)z2SamplingTool for use during LLM sampling requests.) annotationsN)Callable)Any) TextContent)Tool) ConfigDict)ParsedFunction) FunctionTool) ToolResult)TransformedTool)FastMCPBaseModelceZdZUdZded<dZded<ded<d ed <d Zd ed <edZdddZ ddZ e ddd d ddZ e ddd ddZ y) SamplingToolaA tool that can be used during LLM sampling. SamplingTools bundle a tool's schema (name, description, parameters) with an executor function, enabling servers to execute agentic workflows where the LLM can request tool calls during sampling. In most cases, pass functions directly to ctx.sample(): def search(query: str) -> str: '''Search the web.''' return web_search(query) result = await context.sample( messages="Find info about Python", tools=[search], # Plain functions work directly ) Create a SamplingTool explicitly when you need custom name/description: tool = SamplingTool.from_function(search, name="web_search") strnameN str | None descriptionzdict[str, Any] parametersCallable[..., Any]fnFbool sequentialT)arbitrary_types_allowedc|K|i}|jdi|}tj|r |d{}|S7w)zExecute the tool with the given arguments. Args: arguments: Dictionary of arguments to pass to the tool function. Returns: The result of executing the tool function. N)rinspect isawaitable)self argumentsresults u/Users/bowang/.openclaw/workspace/ChatDev/.venv/lib/python3.12/site-packages/fastmcp/server/sampling/sampling_tool.pyrunzSamplingTool.run3sE  I%9%   v &!\F "s 1<:<cZt|j|j|jS)zConvert to an mcp.types.Tool for SDK compatibility. This is used internally when passing tools to the MCP SDK's create_message() method. )rr inputSchema)SDKToolrrr)rs r! _to_sdk_toolzSamplingTool._to_sdk_toolDs( ((  )rrrctj|d}||jdk(r td||xs |j|xs |j|j |j |S)aCreate a SamplingTool from a function. The function's signature is analyzed to generate a JSON schema for the tool's parameters. Type hints are used to determine parameter types. Args: fn: The function to create a tool from. name: Optional name override. Defaults to the function's name. description: Optional description override. Defaults to the function's docstring. sequential: If True, this tool requires sequential execution and prevents parallel execution of all tools in the batch. Set to True for tools with shared state, file writes, or other operations that cannot run concurrently. Defaults to False. Returns: A SamplingTool wrapping the function. Raises: ValueError: If the function is a lambda without a name override. T)validatezz,You must provide a name for lambda functions)rrrrr)r from_functionr ValueErrorr input_schemar)clsrrrrparseds r!r*zSamplingTool.from_functionPsl: --b4@ .wrappers88F++F&*-,,8))d.@.@.D.D// &88<>c&..&9A&=$*NN1$5M!-=,111M',sCCCC)rrrr)r8rreturnr) r1r r TypeErrortype__name__rrr)r-r:rrr;rs ` r!from_callable_toolzSamplingTool.from_callable_toolzs.$ ?@@dATAT@UVII  ,"#7t'7'7   r')N)rzdict[str, Any] | Noner<r)r<r%) rrrrrrrrr<r)r:zFunctionTool | TransformedToolrrrrr<r)r? __module__ __qualname____doc____annotations__rrr model_configr"r& classmethodr*r@rr'r!rrs, I"K"Jd;L"    "& ' '  '  '  '  ' ' R  "& < ,<  <  <  < < r'r)rC __future__rrcollections.abcrtypingr mcp.typesrrr%pydanticrfastmcp.tools.function_parsingr fastmcp.tools.function_toolr fastmcp.tools.toolr fastmcp.tools.tool_transformr fastmcp.utilities.typesr rrr'r!rQs;8"$!%94)84c #c r'