o iG@sddlmZddlmZmZddlmZddlmZddl m Z m Z m Z m Z mZddlmZddlmZdd lmZmZdd lmZmZe rdd d lmZd d lmZd dlmZd dlm Z d dl!m"Z"GdddeZ#eddGddde eZ$Gdddee eZ%dS)) annotations)ABCabstractmethod)Callable) dataclass) TYPE_CHECKINGAnyGenericLiteralProtocol)SchemaValidator)Self) AgentDepsT RunContext)ToolDefinitionToolsPrepareFuncApprovalRequiredToolsetFilteredToolsetPrefixedToolsetPreparedToolsetRenamedToolsetc@s0eZdZdZdddd d Zdddd dZdS)SchemaValidatorProtzsProtocol for a Pydantic Core `SchemaValidator` or `PluggableSchemaValidator` (which is private but API-compatible).F) allow_partialinputstr | bytes | bytearrayr/bool | Literal['off', 'on', 'trailing-strings']kwargsrreturncKdSNselfr rr#r'r'c/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/pydantic_ai/toolsets/abstract.py validate_jsonsz!SchemaValidatorProt.validate_jsoncKr%r&r'r(r'r'r*validate_python!sz#SchemaValidatorProt.validate_pythonN)r r!rr"r#rr$r)r rrr"r#rr$r)__name__ __module__ __qualname____doc__r+r,r'r'r'r*rs  rT)kw_onlyc@s8eZdZUdZded< ded< ded< ded <d S) ToolsetToola:Definition of a tool available on a toolset. This is a wrapper around a plain tool definition that includes information about: - the toolset that provided it, for use in error messages - the maximum number of retries to attempt if the tool call fails - the validator for the tool's arguments AbstractToolset[AgentDepsT]toolsetrtool_defint max_retriesz%SchemaValidator | SchemaValidatorProtargs_validatorN)r-r.r/r0__annotations__r'r'r'r*r2&s  r2c@seZdZdZeedCddZedDddZedDd d ZdEd d Z dFddZ edGddZ edHddZ dId"d#Z dJd&d'ZdKd+d,ZdLd/d0ZdMd4d5ZdNd9d:Zd;d<fdOd@dAZdBS)PAbstractToolsetaA toolset is a collection of tools that can be used by an agent. It is responsible for: - Listing the tools it contains - Validating the arguments of the tools - Calling the tools See [toolset docs](../toolsets.md) for more information. r$ str | NonecCst)aAn ID for the toolset that is unique among all toolsets registered with the same agent. If you're implementing a concrete implementation that users can instantiate more than once, you should let them optionally pass a custom ID to the constructor and return that here. A toolset needs to have an ID in order to be used in a durable execution environment like Temporal, in which case the ID will be used to identify the toolset's activities within the workflow. NotImplementedErrorr)r'r'r*idJs zAbstractToolset.idstrcCs"|jj}|jr|d|j7}|S)z2The name of the toolset for use in error messages. ) __class__r-r?)r)labelr'r'r*rCUszAbstractToolset.labelcCdS)zUA hint for how to avoid name conflicts with other toolsets for use in error messages.zSRename the tool or wrap the toolset in a `PrefixedToolset` to avoid name conflicts.r'r>r'r'r*tool_name_conflict_hint]sz'AbstractToolset.tool_name_conflict_hintr cs|S)z{Enter the toolset context. This is where you can set up network connections in a concrete implementation. r'r>r'r'r* __aenter__bzAbstractToolset.__aenter__argsr bool | NonecsdS)z}Exit the toolset context. This is where you can tear down network connections in a concrete implementation. Nr')r)rHr'r'r* __aexit__irGzAbstractToolset.__aexit__ctxRunContext[AgentDepsT]"dict[str, ToolsetTool[AgentDepsT]]ct)z-The tools that are available in this toolset.r<)r)rKr'r'r* get_toolspszAbstractToolset.get_toolsname tool_argsdict[str, Any]toolToolsetTool[AgentDepsT]crN)aJCall a tool with the given arguments. Args: name: The name of the tool to call. tool_args: The arguments to pass to the tool. ctx: The run context. tool: The tool definition returned by [`get_tools`][pydantic_ai.toolsets.AbstractToolset.get_tools] that was called. r<)r)rPrQrKrSr'r'r* call_toolus zAbstractToolset.call_toolvisitor-Callable[[AbstractToolset[AgentDepsT]], None]NonecCs ||dS)zmRun a visitor function on all "leaf" toolsets (i.e. those that implement their own tool listing and calling).Nr'r)rVr'r'r*applys zAbstractToolset.applyDCallable[[AbstractToolset[AgentDepsT]], AbstractToolset[AgentDepsT]]r3cCs||S)zRun a visitor function on all "leaf" toolsets (i.e. those that implement their own tool listing and calling) and replace them in the hierarchy with the result of the function.r'rYr'r'r*visit_and_replacesz!AbstractToolset.visit_and_replace filter_func8Callable[[RunContext[AgentDepsT], ToolDefinition], bool]FilteredToolset[AgentDepsT]cCddlm}|||S)zReturns a new toolset that filters this toolset's tools using a filter function that takes the agent context and the tool definition. See [toolset docs](../toolsets.md#filtering-tools) for more information. rr)filteredr)r)r]rr'r'r*ras  zAbstractToolset.filteredprefixPrefixedToolset[AgentDepsT]cCr`)zReturns a new toolset that prefixes the names of this toolset's tools. See [toolset docs](../toolsets.md#prefixing-tool-names) for more information. rr)prefixedr)r)rbrr'r'r*rd  zAbstractToolset.prefixed prepare_funcToolsPrepareFunc[AgentDepsT]PreparedToolset[AgentDepsT]cCr`)zReturns a new toolset that prepares this toolset's tools using a prepare function that takes the agent context and the original tool definitions. See [toolset docs](../toolsets.md#preparing-tool-definitions) for more information. rr)preparedr)r)rfrr'r'r*rirezAbstractToolset.preparedname_mapdict[str, str]RenamedToolset[AgentDepsT]cCr`)zReturns a new toolset that renames this toolset's tools using a dictionary mapping new names to original names. See [toolset docs](../toolsets.md#renaming-tools) for more information. rr)renamedr)r)rjrr'r'r*rmrezAbstractToolset.renamedcCrD)NTr')rKr5rQr'r'r*szAbstractToolset.approval_required_funcHCallable[[RunContext[AgentDepsT], ToolDefinition, dict[str, Any]], bool]#ApprovalRequiredToolset[AgentDepsT]cCr`)zReturns a new toolset that requires (some) calls to tools it contains to be approved. See [toolset docs](../toolsets.md#requiring-tool-approval) for more information. rr)approval_requiredr)r)rorr'r'r*rrs z!AbstractToolset.approval_requiredN)r$r;)r$r@)r$r )rHrr$rI)rKrLr$rM) rPr@rQrRrKrLrSrTr$r)rVrWr$rX)rVr[r$r3)r]r^r$r_)rbr@r$rc)rfrgr$rh)rjrkr$rl)rorpr$rq)r-r.r/r0propertyrr?rCrErFrJrOrUrZr\rardrirmrrr'r'r'r*r:>s.           r:N)& __future__rabcrrcollections.abcr dataclassesrtypingrrr r r pydantic_corer typing_extensionsr _run_contextrrtoolsrrrrrrarrdrrirrmrrr2r:r'r'r'r*s&