g-dUddlmZddlZddlZddlmZddlmZmZddl m Z m Z m Z m Z mZmZddlmZddlmZddlmZmZmZmZd d lmZmZmZmZd d lm Z m!Z!e rd d l"m#Z#d Z$eddZ% ejGdde e%Z&eddZ' ee e&e%ge(fe e&e%gee(fe ge(fe gee(ffZ) e ee&e%e'fe fZ* e e'e fZ+ ee*e%e'fe+e'fZ, dZ-de.d< edZ/edGdde e%Z0e1e(e fZ2de.d< eGddZ3y)) annotationsN) Awaitable) dataclassfield) TYPE_CHECKINGAnyCallableGenericUnioncast)ValidationError)SchemaValidator) Concatenate ParamSpec TypeAliasTypeVar) _pydantic_utilsmessagesmodels) ModelRetryUnexpectedModelBehavior)Usage) AgentDeps RunContextSystemPromptFuncToolFuncContext ToolFuncPlainToolFuncEither ToolParamsToolPrepareFuncToolObjectJsonSchemaToolDefinitionr)defaultceZdZUdZded< ded< ded< ded < ee Zd ed < d Zded< dZ ded< dZ ded< d e jf ddZ y )rz#Information about the current call.rdepsz models.Modelmodelrusagestrprompt)default_factoryzlist[_messages.ModelMessage]rN str | None tool_namerintretryrun_stepcni}|||d<|tjur||d<tj|fi|S)Nr1r/)rUNSET dataclassesreplace)selfr1r/kwargss F/var/www/openai/venv/lib/python3.12/site-packages/pydantic_ai/tools.py replace_withzRunContext.replace_with:sD  #F7O FLL ("+F; ""42622)r1 int | Noner/zstr | None | _utils.UnsetreturnRunContext[AgentDeps])__name__ __module__ __qualname____doc____annotations__rlistrr/r1r2rr4r:r;r9rr%s- O% % L, K5-24-HH*H8 Iz (E3N#Hc&#'v|| 3 33L 3  3r;rr!.zSCallable[[RunContext[AgentDeps], ToolDefinition], Awaitable[ToolDefinition | None]]rr"AFinitceZdZUdZded<ded<ded<ded <ded <d ed <ed Zded<ed Zded<ed Zded<ed Z ded<ed d Z ded<ed Z ded<edd Z ded<dddddd d%d Z d&d!Z d'd"Z d(d#Z d)d$Zy)*r#zA tool function for an agent.ToolFuncEither[AgentDeps]functionbool takes_ctxr< max_retriesr+name description!ToolPrepareFunc[AgentDeps] | NoneprepareFrG _is_asyncr._single_arg_namez list[str]_positional_fields_var_positional_field)rHreprr _validatorr$_parameters_json_schemar)r&rHr0 current_retryN)rMrNrOrPrRc|tj|}tj||}||_||_||_|xs |j |_|xs|d|_||_tj|j|_ |d|_ |d|_ |d|_|d|_|d|_y)a0Create a new tool instance. Example usage: ```python {lint="not-imports"} from pydantic_ai import Agent, RunContext, Tool async def my_tool(ctx: RunContext[int], x: int, y: int) -> str: return f'{ctx.deps} {x} {y}' agent = Agent('test', tools=[Tool(my_tool)]) ``` or with a custom prepare method: ```python {lint="not-imports"} from typing import Union from pydantic_ai import Agent, RunContext, Tool from pydantic_ai.tools import ToolDefinition async def my_tool(ctx: RunContext[int], x: int, y: int) -> str: return f'{ctx.deps} {x} {y}' async def prep_my_tool( ctx: RunContext[int], tool_def: ToolDefinition ) -> Union[ToolDefinition, None]: # only register the tool if `deps == 42` if ctx.deps == 42: return tool_def agent = Agent('test', tools=[Tool(my_tool, prepare=prep_my_tool)]) ``` Args: function: The Python function to call as the tool. takes_ctx: Whether the function takes a [`RunContext`][pydantic_ai.tools.RunContext] first argument, this is inferred if unset. max_retries: Maximum number of retries allowed for this tool, set to the agent default if `None`. name: Name of the tool, inferred from the function if `None`. description: Description of the tool, inferred from the function if `None`. 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]. NrPsingle_arg_namepositional_fieldsvar_positional_field validator json_schema)rrMfunction_schemarKrNr?rOrPrRinspectiscoroutinefunctionrSrTrUrVrXrY)r7rKrMrNrOrPrRfs r9__init__z Tool.__init__sp  !++H5I  % %h :  "&-H-- &:!M*:  44T]]C !"3 4"#$7"8%&'=%>"K.'('7$r;cKt|j|j|j}|j|j ||d{S|S7w)aGet the tool definition. By default, this method creates a tool definition, then either returns it, or calls `self.prepare` if it's set. Returns: return a `ToolDefinition` or `None` if the tools should not be registered for this run. )rOrPparameters_json_schemaN)r%rOrPrYrR)r7ctxtool_defs r9prepare_tool_defzTool.prepare_tool_defsU"((#'#?#?  << #c844 4O5sAAAAcxK t|jtjr0|jj |jj }n/|jj|jj}|j|||\}} |jr@tttgt t"f|j$}||i|d{}nItttgt"f|j$}t'j(|g|i|d{}d|_tj.|j0||j2S#t$r}|j||cYd}~Sd}~wwxYw77a#t*$r}|j||cYd}~Sd}~wwxYww)z%Run the tool function asynchronously.Nrr/content tool_call_id) isinstanceargs _messagesArgsJsonrX validate_json args_jsonvalidate_python args_dictr _on_error _call_argsrSr r rrr+rKrrun_in_executorrrZToolReturnPartr/rn) r7message run_contextrverpr8rKresponse_contents r9runzTool.runsn .',, (:(:; OO99',,:P:PQ  OO;;GLL ?O)14)B6)B#B # 4dmmD)/)?)?)Z4)ZSY)Z#Z ''''$ --   .>>!W- - .$C$[ .>>!W- - .sF:BE&F:AF$F%A F.F/F33F:& F /FF F:F  F:FF F7F2,F7-F:2F77F:c|jr|j|i}tj||j|j}|j r|gng}|j D]"}|j|j|$|jr*|j|j|j||fS)N)r1r/) rTr5r6rZr/rMrUappendpoprVextend)r7rvr{r|rhrppositional_fields r9rxzTool._call_argss  .. :I!!+T5G5GSZSdSdeuB $ 7 7  KK &67 8!8  % % KK d&@&@A BYr;c^|xjdz c_|j|j|jkDrtd|j|t|tr|j d}n |j }tj|j||jS)Nrz#Tool exceeded max retries count of F) include_urlrl) rZrNrror errorsr{rqRetryPromptPartr/rn)r7exc call_messagerms r9rwzTool._on_error#s a    #t'9'9Dr=zToolDefinition | None)r{_messages.ToolCallPartr|r>r=z_messages.ModelRequestPart)rvzdict[str, Any]r{rr|r>r=z tuple[list[Any], dict[str, Any]])rzValidationError | ModelRetryrrr=z_messages.RetryPromptPart)r?r@rArBrCrrSrTrUrVrXrYrZrerjrrxrwrEr;r9r#r#s['''O I ..'It'#(e#4j4$)u$5 5(-5(9:9"'U"?J?0550A-Aqu5M35 "&"&"&59G8+G8 G8  G8  G8 G83G8R& - s3 I&,,4'+*+r;r%)4 __future__r _annotationsr5rbcollections.abcrrrtypingrrr r r r pydanticr pydantic_corertyping_extensionsrrrrrrrrqr exceptionsrrresultr__all__rrr!r+rrrr r"rCrFr#dictr$r%rEr;r9rs2%(EE$)HH>>;  K . + 3#33@|S 1 $ j# $c )* j# $in 45 RW R3  "  ;z)'