ó
    ±"³jf  ã                  ó¦   • S SK Jr  S SKJrJr  S SKJrJr  S SKJ	r	  S SK
Jr  SSKJrJr  SSKJr  S	S
KJr  S	SKJrJr  \ " S S\\   5      5       rg)é    )Úannotations)ÚCallableÚSequence)Ú	dataclassÚreplace)ÚAny)ÚSelfé   )Ú
AgentDepsTÚ
RunContext)ÚInstructionParté   )Ú!flatten_instruction_contributions)ÚAbstractToolsetÚToolsetToolc                  óÞ   • \ rS rSr% SrS\S'   \SS j5       r\SS j5       rSS jr	SS jr
SS	 jrSS
 jr    SS jrSS jrSS jrSS jr          SS jrSS jr    SS jrSrg) ÚWrapperToolseté   zŒA toolset that wraps another toolset and delegates to it.

See [toolset docs](../toolsets.md#changing-tool-execution) for more information.
úAbstractToolset[AgentDepsT]Úwrappedc                ó   • g ©N© ©Úselfs    ÚY/home/mande/repo/quber/.venv/lib/python3.13/site-packages/pydantic_ai/toolsets/wrapper.pyÚidÚWrapperToolset.id   s   € àó    c                ób   • U R                   R                   SU R                  R                   S3$ )NÚ(Ú))Ú	__class__Ú__name__r   Úlabelr   s    r   r%   ÚWrapperToolset.label   s+   € à—.‘.×)Ñ)Ð*¨!¨D¯L©L×,>Ñ,>Ð+?¸qÐAÐAr   c              ƒ  ó†   #   • U R                   R                  U5      I S h  v•N nX R                   L a  U $ [        XS9$  N7f©N©r   )r   Úfor_runr   ©r   ÚctxÚnew_wrappeds      r   r*   ÚWrapperToolset.for_run    s;   é € Ø ŸL™L×0Ñ0°Ó5×5ˆØŸ,™,Ò&ØˆKÜ�tÑ1Ð1ñ 6ùó   ‚A¡?¢Ac              ƒ  ó†   #   • U R                   R                  U5      I S h  v•N nX R                   L a  U $ [        XS9$  N7fr(   )r   Úfor_run_stepr   r+   s      r   r1   ÚWrapperToolset.for_run_step&   s;   é € Ø ŸL™L×5Ñ5°cÓ:×:ˆØŸ,™,Ò&ØˆKÜ�tÑ1Ð1ñ ;ùr/   c              ƒ  óV   #   • U R                   R                  5       I S h  v•N   U $  N7fr   )r   Ú
__aenter__r   s    r   r4   ÚWrapperToolset.__aenter__,   s%   é € Ø�l‰l×%Ñ%Ó'×'Ð'Øˆñ 	(ùs   ‚) '¡)c              ‡  óP   #   • U R                   R                  " U6 I S h  v•N $  N7fr   )r   Ú	__aexit__)r   Úargss     r   r7   ÚWrapperToolset.__aexit__0   s    é € Ø—\‘\×+Ò+¨TÐ2×2Ð2Ñ2ùs   ‚&Ÿ$ &c              ƒ  óh   #   • U R                  U5      I Sh  v•N n[        U5      =(       d    S$  N7f)z8Collect instructions from the wrapped authoring toolset.N)Ú(_collect_child_instruction_contributionsr   )r   r,   Úcontributionss      r   Úget_instructionsÚWrapperToolset.get_instructions3   s0   é € ð #×KÑKÈCÓP×PˆÜ0°Ó?×GÀ4ÐGñ Qùs   ‚2—0˜2c                ó   • U R                   4$ r   r)   r   s    r   Ú_instruction_childrenÚ$WrapperToolset._instruction_children:   s   € Ø—‘ˆÐr   c                óL   • [        U 5      R                  [        R                  L$ r   )Útyper=   r   r   s    r   Ú_authors_own_instructionsÚ(WrapperToolset._authors_own_instructions=   s   € Ü�D‹z×*Ñ*´.×2QÑ2QÐQÐQr   c              ƒ  óT   #   • U R                   R                  U5      I S h  v•N $  N7fr   )r   Ú	get_tools)r   r,   s     r   rG   ÚWrapperToolset.get_tools@   s    é € Ø—\‘\×+Ñ+¨CÓ0×0Ð0Ñ0ùs   ‚(¡&¢(c              ƒ  óV   #   • U R                   R                  XX45      I S h  v•N $  N7fr   )r   Ú	call_tool)r   ÚnameÚ	tool_argsr,   Útools        r   rJ   ÚWrapperToolset.call_toolC   s$   é € ð —\‘\×+Ñ+¨D¸SÓG×GÐGÑGùs   ‚ )¢'£)c                ó:   • U R                   R                  U5        g r   )r   Úapply©r   Úvisitors     r   rP   ÚWrapperToolset.applyH   s   € Ø�‰×Ñ˜7Õ#r   c                óF   • [        X R                  R                  U5      S9$ r(   )r   r   Úvisit_and_replacerQ   s     r   rU   Ú WrapperToolset.visit_and_replaceK   s   € ô �t§\¡\×%CÑ%CÀGÓ%LÑMÐMr   r   N)Úreturnz
str | None)rW   Ústr)r,   úRunContext[AgentDepsT]rW   r   )rW   r	   )r8   r   rW   zbool | None)r,   rY   rW   z>str | InstructionPart | Sequence[str | InstructionPart] | None)rW   z%Sequence[AbstractToolset[AgentDepsT]])rW   Úbool)r,   rY   rW   z"dict[str, ToolsetTool[AgentDepsT]])
rK   rX   rL   zdict[str, Any]r,   rY   rM   zToolsetTool[AgentDepsT]rW   r   )rR   z-Callable[[AbstractToolset[AgentDepsT]], None]rW   ÚNone)rR   zDCallable[[AbstractToolset[AgentDepsT]], AbstractToolset[AgentDepsT]]rW   r   )r$   Ú
__module__Ú__qualname__Ú__firstlineno__Ú__doc__Ú__annotations__Úpropertyr   r%   r*   r1   r4   r7   r=   r@   rD   rG   rJ   rP   rU   Ú__static_attributes__r   r   r   r   r      sÃ   ‡ ñð
 )Ó(àóó ðð óBó ðBô2ô2ôô3ðHØ)ðHà	GôHôôRô1ðHØðHØ$2ðHØ9OðHØWnðHà	ôHô
$ðNØ[ðNà	$÷Nr   r   N)Ú
__future__r   Úcollections.abcr   r   Údataclassesr   r   Útypingr   Útyping_extensionsr	   Ú_run_contextr   r   Úmessagesr   Ú_instruction_collectionr   Úabstractr   r   r   r   r   r   Ú<module>rl      sC   ðÝ "ç .ß *Ý å "ç 1Ý &Ý Fß 2ð ô>N�_ ZÑ0ó >Nó ñ>Nr   