Ki rdZddlmZddlZddlmZddlmZddlm Z ddl m Z m Z ddlmZGd d e Zy) z2SamplingTool for use during LLM sampling requests.) annotationsN)Callable)Any)Tool) BaseModel ConfigDict)ParsedFunctionceZdZUdZded<dZded<ded<d ed <ed Zddd ZddZ 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]fnT)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 q/home/jay/workspace/scripts/.codegraph-venv/lib/python3.12/site-packages/fastmcp/server/sampling/sampling_tool.pyrunzSamplingTool.run-sE  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. )r r inputSchema)SDKToolr rr)rs r _to_sdk_toolzSamplingTool._to_sdk_tool>s( ((  )r rctj|d}||jdk(r td||xs |j|xs |j|j |j S)aZCreate 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. 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)r rrr)r from_functionr ValueErrorr input_schemar)clsrr rparseds rr$zSamplingTool.from_functionJsi0 --b4@ r7s+8"$%*-] 9] r!