o iA@sddlmZddlmZmZmZddlmZmZddl m Z m Z ddl m Z ddlmZmZddlmZdd lmZmZmZmZmZmZd d lmZmZed d GdddeeZGdddeeZdS)) annotations) AwaitableCallableSequence) dataclassreplace)Anyoverload)GenerateJsonSchema) AgentDepsT RunContext) UserError)DocstringFormatGenerateToolJsonSchemaToolToolFuncEither ToolParamsToolPrepareFunc)AbstractToolset ToolsetToolT)kw_onlyc@s"eZdZUdZded<ded<dS)FunctionToolsetToolzWA tool definition for a function toolset tool that keeps track of the function to call.zBCallable[[dict[str, Any], RunContext[AgentDepsT]], Awaitable[Any]] call_funcboolis_asyncN)__name__ __module__ __qualname____doc____annotations__r"r"c/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/pydantic_ai/toolsets/function.pyrs  rc @seZdZUdZded<ded<ded<ded <d ed <d ed <gfdddedddddd dEddZedFddZe dGd"d#Z e dddddddddddd$ dHd.d#Z dIdddddddddddd$ dJd1d#Z            dKdLd4d5Z dMd8d9Z dNd=d>Z dOdCdDZdS)PFunctionToolsetzA toolset that lets Python functions be used as tools. See [toolset docs](../toolsets.md#function-toolset) for more information. zdict[str, Tool[Any]]toolsint max_retries str | None_idrdocstring_formatrrequire_parameter_descriptionstype[GenerateJsonSchema]schema_generatorrautoFN) r'r*r+r-strict sequentialrequires_approvalmetadataidr")r:r?r"r"r#r;gszFunctionToolset.tool) name descriptionretriespreparer*r+r-r/r0r1r2rBrCrD int | NonerE"ToolPrepareFunc[AgentDepsT] | NoneDocstringFormat | Nonetype[GenerateJsonSchema] | NoneZCallable[[ToolFuncEither[AgentDepsT, ToolParams]], ToolFuncEither[AgentDepsT, ToolParams]]c CrAr>r") r:rBrCrDrEr*r+r-r/r0r1r2r"r"r#r;js-ToolFuncEither[AgentDepsT, ToolParams] | Nonerc  s8d f dd } |dur| S| |S)a Decorator to register a tool function which takes [`RunContext`][pydantic_ai.tools.RunContext] as its first argument. Can decorate a sync or async functions. The docstring is inspected to extract both the tool description and description of each parameter, [learn more](../tools.md#function-tools-and-schema). We can't add overloads for every possible signature of tool, since the return type is a recursive union so the signature of functions decorated with `@toolset.tool` is obscured. Example: ```python from pydantic_ai import Agent, FunctionToolset, RunContext toolset = FunctionToolset() @toolset.tool def foobar(ctx: RunContext[int], x: int) -> int: return ctx.deps + x @toolset.tool(retries=2) async def spam(ctx: RunContext[str], y: float) -> float: return ctx.deps + y agent = Agent('test', toolsets=[toolset], deps_type=int) result = agent.run_sync('foobar', deps=1) print(result.output) #> {"foobar":1,"spam":1.0} ``` Args: func: The tool function to register. name: The name of the tool, defaults to the function name. description: The description of the tool,defaults to the function docstring. retries: The number of retries to allow for this tool, defaults to the agent's default retries, which defaults to 1. prepare: custom method to prepare the tool definition for each step, return `None` to omit this tool from a given step. This is useful if you want to customise a tool at call time, or omit it completely from a step. See [`ToolPrepareFunc`][pydantic_ai.tools.ToolPrepareFunc]. docstring_format: The format of the docstring, see [`DocstringFormat`][pydantic_ai.tools.DocstringFormat]. If `None`, the default value is determined by the toolset. require_parameter_descriptions: If True, raise an error if a parameter description is missing. If `None`, the default value is determined by the toolset. schema_generator: The JSON schema generator class to use for this tool. If `None`, the default value is determined by the toolset. strict: Whether to enforce JSON schema compliance (only affects OpenAI). See [`ToolDefinition`][pydantic_ai.tools.ToolDefinition] for more info. If `None`, the default value is determined by the toolset. sequential: Whether the function requires a sequential/serial execution environment. Defaults to False. If `None`, the default value is determined by the toolset. requires_approval: Whether this tool requires human-in-the-loop approval. Defaults to False. See the [tools documentation](../deferred-tools.md#human-in-the-loop-tool-approval) for more info. If `None`, the default value is determined by the toolset. metadata: Optional metadata for the tool. This is not sent to the model but can be used for filtering and tool behavior customization. If `None`, the default value is determined by the toolset. If provided, it will be merged with the toolset's metadata. func_r@r=cs( j|d d |S)N) r? takes_ctxrBrCrDrEr*r+r-r/r0r1r2)r9)rL rCr*r2rBrEr+r1rDr-r:r0r/r"r#tool_decorators z,FunctionToolset.tool..tool_decoratorN)rLr@r=r@r")r:r?rBrCrDrEr*r+r-r/r0r1r2rOr"rNr#r;|s$JrMNonecCs|dur|j}|dur|j}| dur|j} | dur|j} | dur#|j} | dur*|j} tt||||||||| | | | | d }||dS)ax Add a function as a tool to the toolset. Can take a sync or async function. The docstring is inspected to extract both the tool description and description of each parameter, [learn more](../tools.md#function-tools-and-schema). Args: func: The tool function to register. takes_ctx: Whether the function takes a [`RunContext`][pydantic_ai.tools.RunContext] as its first argument. If `None`, this is inferred from the function signature. name: The name of the tool, defaults to the function name. description: The description of the tool, defaults to the function docstring. retries: The number of retries to allow for this tool, defaults to the agent's default retries, which defaults to 1. prepare: custom method to prepare the tool definition for each step, return `None` to omit this tool from a given step. This is useful if you want to customise a tool at call time, or omit it completely from a step. See [`ToolPrepareFunc`][pydantic_ai.tools.ToolPrepareFunc]. docstring_format: The format of the docstring, see [`DocstringFormat`][pydantic_ai.tools.DocstringFormat]. If `None`, the default value is determined by the toolset. require_parameter_descriptions: If True, raise an error if a parameter description is missing. If `None`, the default value is determined by the toolset. schema_generator: The JSON schema generator class to use for this tool. If `None`, the default value is determined by the toolset. strict: Whether to enforce JSON schema compliance (only affects OpenAI). See [`ToolDefinition`][pydantic_ai.tools.ToolDefinition] for more info. If `None`, the default value is determined by the toolset. sequential: Whether the function requires a sequential/serial execution environment. Defaults to False. If `None`, the default value is determined by the toolset. requires_approval: Whether this tool requires human-in-the-loop approval. Defaults to False. See the [tools documentation](../deferred-tools.md#human-in-the-loop-tool-approval) for more info. If `None`, the default value is determined by the toolset. metadata: Optional metadata for the tool. This is not sent to the model but can be used for filtering and tool behavior customization. If `None`, the default value is determined by the toolset. If provided, it will be merged with the toolset's metadata. N) rMrBrCr'rEr*r+r-r/r0r1r2) r*r+r-r/r0r1rr r8)r:r?rMrBrCrDrEr*r+r-r/r0r1r2r;r"r"r#r9s82zFunctionToolset.add_functionr;Tool[AgentDepsT]cCsZ|j|jvrtd|j|jdur|j|_|jdur%|j|jp"iB|_||j|j<dS)zUAdd a tool to the toolset. Args: tool: The tool to add. z(Tool name conflicts with existing tool: N)rBr%rr'r2)r:r;r"r"r#r8-s   zFunctionToolset.add_toolctxRunContext[AgentDepsT]"dict[str, ToolsetTool[AgentDepsT]]c si}|jD]W\}}|jdur|jn|j}t|||j|d|d}||IdH}|s/q|j}||vrM||krEtd|d|dtd|dt ||||j j |j j |j j d||<q|S) Nr) tool_nameretryr'zRenaming tool z to z conflicts with existing tool.z2Tool name conflicts with previously renamed tool: .)toolsettool_defr'args_validatorrr)r%itemsr'rrDgetprepare_tool_defrBrrfunction_schema validatorcallr) r:rRr% original_namer;r' run_contextrYnew_namer"r"r# get_tools;s6  zFunctionToolset.get_toolsstr tool_argsdict[str, Any]ToolsetTool[AgentDepsT]cs"t|tsJ|||IdHSr>)r7rr)r:rBrfrRr;r"r"r# call_toolZszFunctionToolset.call_tool)r%r4r'r&r*rr+rr-r,r/r5r0rr1rr2r6r3r()r=r()r?r@r=r@)rBr(rCr(rDrFrErGr*rHr+r5r-rIr/r5r0r5r1r5r2r6r=rJr>)r?rKrBr(rCr(rDrFrErGr*rHr+r5r-rIr/r5r0r5r1r5r2r6r=r) NNNNNNNNNNNN)r?r@rMr5rBr(rCr(rDrFrErGr*rHr+r5r-rIr/r5r0r5r1r5r2r6r=rP)r;rQr=rP)rRrSr=rT) rBrerfrgrRrSr;rhr=r)rrrr r!rr<propertyr3r r;r9r8rdrir"r"r"r#r$s 8  d P r$N) __future__rcollections.abcrrr dataclassesrrtypingrr pydantic.json_schemar _run_contextr r exceptionsrr%rrrrrrabstractrrrr$r"r"r"r#s