o i{ @sUddlmZddlZddlmZmZddlmZmZddl m Z m Z ddl m Z ddlmZmZd d lmZmZeeegeedBeeedBBfZd ed < eGd ddeeZdS)) annotationsN) AwaitableCallable) dataclassreplace)Any TypeAlias)Self) AgentDepsT RunContext)AbstractToolset ToolsetToolr ToolsetFunccseZdZUdZded<dZded<dZded <dZd ed <ed.ddZ d/ddZ d0ddZ d1ddZ d2d#d$Z d3fd(d) Zd4fd,d- ZZS)5DynamicToolsetzA toolset that dynamically builds a toolset using a function that takes the run context. It should only be used during a single agent run as it stores the generated toolset. To use it multiple times, copy it using `dataclasses.replace`. zToolsetFunc[AgentDepsT] toolset_funcTbool per_run_stepNz"AbstractToolset[AgentDepsT] | None_toolsetz int | None _run_stepreturn str | NonecCsdSNselfrrc/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/pydantic_ai/toolsets/_dynamic.pyid"szDynamicToolset.idr cs|Srrrrrr __aenter__&szDynamicToolset.__aenter__argsr bool | NonecsNz|jdur|jj|IdHWd|_d|_SWd|_d|_dSd|_d|_wr)r __aexit__r)rr rrrr")s  zDynamicToolset.__aexit__ctxRunContext[AgentDepsT]"dict[str, ToolsetTool[AgentDepsT]]cs|jdus|jr=|j|jkr=|jdur|jIdH||}t|r+|IdH}|dur6|IdH||_|j|_|jdurDiS|j |IdHSr) rrrun_steprr"rinspect isawaitabler get_tools)rr#toolsetrrrr)1s     zDynamicToolset.get_toolsnamestr tool_argsdict[str, Any]toolToolsetTool[AgentDepsT]cs(|jdusJ|j||||IdHSr)r call_tool)rr+r-r#r/rrrr1EszDynamicToolset.call_toolvisitor-Callable[[AbstractToolset[AgentDepsT]], None]Nonecs*|jdur t|dS|j|dSr)rsuperapplyrr2 __class__rrr6Ks zDynamicToolset.applyDCallable[[AbstractToolset[AgentDepsT]], AbstractToolset[AgentDepsT]]AbstractToolset[AgentDepsT]cs*|jdur t|St||j|dS)N)r)rr5visit_and_replacerr7r8rrr<Qs  z DynamicToolset.visit_and_replace)rr)rr )r rrr!)r#r$rr%) r+r,r-r.r#r$r/r0rr)r2r3rr4)r2r:rr;)__name__ __module__ __qualname____doc____annotations__rrrpropertyrrr"r)r1r6r< __classcell__rrr8rrs         r) __future__rr'collections.abcrr dataclassesrrtypingrrtyping_extensionsr _run_contextr r abstractrrrrArrrrrs