g3fi%dZddlmZddlZddlZddlmZmZddlm Z ddl m Z m Z m Z mZddlmZmZddlmZdd lmZmZdd lmZmZdd lmZmZmZmZmZm Z m!Z!dd l"m#Z#e rdd l$m%Z%GddeZ&ddZ'y)zStructured tool.) annotationsN) AwaitableCallable) signature) TYPE_CHECKING AnnotatedAnyLiteral)FieldSkipValidation)override)AsyncCallbackManagerForToolRunCallbackManagerForToolRun)RunnableConfigrun_in_executor) _EMPTY_SET FILTERED_ARGS ArgsSchemaBaseTool_get_runnable_config_param_is_injected_arg_typecreate_schema_from_function)is_basemodel_subclass)ToolCallcTeZdZUdZdZded<eddZded < d Zd ed < d Z d ed< e d dfd Z d d ddZ d d dfdZ e ddddd ddZej"ddZxZS)StructuredToolz.Tool that can operate on any number of inputs.str description.zThe tool schema.)rz'Annotated[ArgsSchema, SkipValidation()] args_schemaNzCallable[..., Any] | Nonefunc$Callable[..., Awaitable[Any]] | None coroutinecK|js!t||j||fi|d{St|||fi|d{S77wN)r#rinvokesuperainvoke)selfinputconfigkwargs __class__s ]/var/www/auto_recruiter/arenv/lib/python3.12/site-packages/langchain_core/tools/structured.pyr(zStructuredTool.ainvoke8sP~~(eVVvVV VW_UF=f===W=s!)AA AA A A) run_managerc|jrq|rAt|jjjdr|j |d<t |jx}r|||<|j|i|Sd}t |)aPUse the tool. Args: *args: Positional arguments to pass to the tool config: Configuration for the run run_manager: Optional callback manager to use for the run **kwargs: Keyword arguments to pass to the tool Returns: The result of the tool execution callbacksz0StructuredTool does not support sync invocation.)r!r parametersget get_childrNotImplementedError)r)r+r/argsr, config_parammsgs r._runzStructuredTool._runGs$ 99y3>>BB;O&1&;&;&={#9$))DD|D'-|$499d-f- -@!#&&cRK|jry|rAt|jjjdr|j |d<t |jx}r|||<|j|i|d{St ||||d|d{S77w)a_Use the tool asynchronously. Args: *args: Positional arguments to pass to the tool config: Configuration for the run run_manager: Optional callback manager to use for the run **kwargs: Keyword arguments to pass to the tool Returns: The result of the tool execution r1N)r+r/)r#rr2r3r4rr'_arun)r)r+r/r6r,r7r-s r.r<zStructuredTool._arunbs$ >>y8CCGG T&1&;&;&={#9$..II|I'-|$'888 8W] &k =C   9 s$BB'B#B'B%B'%B'Fcontent)response_formatparse_docstringerror_on_invalid_docstringc 6||} n||} n d} t| |xs | j}||rt|| | | t| }|}|| s| jxsd}|i|rgt |t r%t|r|j}|rd|vrd}n7|s5d}n2t |tr|jd}nd|} t| | d} t| |#tj|j}|j}|d |||||||d | S) aCreate tool from a given function. A classmethod that helps to create a tool from a function. Args: func: The function from which to create a tool. coroutine: The async function from which to create a tool. name: The name of the tool. Defaults to the function name. description: The description of the tool. Defaults to the function docstring. return_direct: Whether to return the result directly or as a callback. args_schema: The schema of the tool's input arguments. infer_schema: Whether to infer the schema from the function's signature. response_format: The tool response format. If `"content"` then the output of the tool is interpreted as the contents of a `ToolMessage`. If `"content_and_artifact"` then the output is expected to be a two-tuple corresponding to the `(content, artifact)` of a `ToolMessage`. parse_docstring: If `infer_schema` and `parse_docstring`, will attempt to parse parameter descriptions from Google Style function docstrings. error_on_invalid_docstring: if `parse_docstring` is provided, configure whether to raise `ValueError` on invalid Google Style docstrings. **kwargs: Additional arguments to pass to the tool Returns: The tool. Raises: ValueError: If the function is not provided. ValueError: If the function does not have a docstring and description is not provided. TypeError: If the `args_schema` is not a `BaseModel` or dict. Examples: ```python def add(a: int, b: int) -> int: """Add two numbers""" return a + b tool = StructuredTool.from_function(add) tool.run(1, 2) # 3 ``` Nz*Function and/or coroutine must be provided)r?r@ filter_argsz)A base class for creating Pydantic modelsrrz5Invalid args_schema: expected BaseModel or dict, got z;Function must have a docstring if description not provided.)namer!r#r r return_directr>) ValueError__name__r_filter_schema_args__doc__ isinstancetyperdictr3 TypeErrortextwrapdedentstrip)clsr!r#rCrrDr infer_schemar>r?r@r,source_functionr8 description_s r. from_functionzStructuredTool.from_functionsjx  "O  "'O>CS/ !///  <5 /+E/@ K#  *22:dL  K+t,1F{1S*22  C|S#%L%#'LK.*}= &-) n$  OCS/ !  #??<8>>@L',,./   #$'+    r:c|jxs |j}|tStdt |j j DS)Nc3PK|]\}}t|jr| ywr%)r annotation).0kvs r. z5StructuredTool._injected_args_keys..s( 1$Q\\2  s$&)r!r#r frozensetrr2items)r)fns r._injected_args_keysz"StructuredTool._injected_args_keyssL YY ($.. :  !" 00668   r:r%)r*zstr | dict | ToolCallr+zRunnableConfig | Noner,r returnr ) r6r r+rr/z CallbackManagerForToolRun | Noner,r rar ) r6r r+rr/z%AsyncCallbackManagerForToolRun | Noner,r rar )NNNNFNT)r!zCallable | Noner#r"rC str | NonerrbrDboolr zArgsSchema | NonerRrcr>z*Literal['content', 'content_and_artifact']r?rcr@rcr,r rar)razfrozenset[str])rG __module__ __qualname__rIr__annotations__r r r!r#r r(r9r< classmethodrU functoolscached_propertyr` __classcell__)r-s@r.rr(s8K;@ +<K8'&*D #*66:I3:3 )- >$ >& > > > >$9= '''6 '  ' '>>B    ;     >!%:>"&#)-!t GP %+0t t 8t  t  t  t 't t Dt t %)t t  t t l  r:rc`tt}t|x}r|j||Sr%)listrrappend)r!rBr7s r.rHrHs0}%K1$77|7<( r:)r!rraz list[str])(rI __future__rrhrNcollections.abcrrinspectrtypingrrr r pydanticr r typing_extensionsr langchain_core.callbacksrrlangchain_core.runnablesrrlangchain_core.tools.baserrrrrrrlangchain_core.utils.pydanticrlangchain_core.messagesrrrHrEr:r.rysc"/+&E@0Y XY xr: