ó
    ±"³j  ã                  ó  • S SK Jr  S SKrS SKJr  S SKJr  S SKJr  S SK	J
r
  S SKJr  S SK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  S	SKJr  SSKJrJ r J!r!J"r"J#r#J$r$  S/r%\" SS9 " S S\ 5      5       r&SS jr'g)é    )ÚannotationsN)ÚAsyncGenerator)Úasynccontextmanager)Ú	dataclass)ÚTracebackType)ÚAny)ÚSelfé   )Ú
RunContext)ÚPydanticAIDeprecationWarning)ÚModelMessageÚModelResponse)ÚModelProfile)ÚProvider)ÚModelSettings)ÚRequestUsageé   )ÚKnownModelNameÚModelÚModelRequestContextÚModelRequestParametersÚStreamedResponseÚinfer_modelÚWrapperModelF)Úinitc                  óÖ  ^ • \ rS rSr% SrS\S'    SU 4S j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.     S$S jjr\ S%         S&S jj5       rS'S jr      S(S jr S%     S)S jjr\S*S j5       r\S+S j5       r\S+S j5       r\S+S j5       r\S,S j5       r\S-S j5       r\S.S j5       r\S/S j5       rS0S jrSrU =r$ )1r   é   zQModel which wraps another model.

Does nothing on its own, used as a base class.
r   Úwrappedc                óB   >• [         TU ]  5         [        U5      U l        g ©N)ÚsuperÚ__init__r   r   )Úselfr   Ú	__class__s     €ÚW/home/mande/repo/quber/.venv/lib/python3.13/site-packages/pydantic_ai/models/wrapper.pyr"   ÚWrapperModel.__init__)   s   ø€ Ü‰ÑÔÜ" 7Ó+ˆ�ó    c              ƒ  óV   #   • U R                   R                  5       I S h  v•N   U $  N7fr    )r   Ú
__aenter__©r#   s    r%   r)   ÚWrapperModel.__aenter__-   s%   é € Ø�l‰l×%Ñ%Ó'×'Ð'Øˆñ 	(ùs   ‚) '¡)c              ƒ  óV   #   • U R                   R                  XU5      I S h  v•N $  N7fr    )r   Ú	__aexit__)r#   Úexc_typeÚexc_valÚexc_tbs       r%   r-   ÚWrapperModel.__aexit__1   s$   é € ð —\‘\×+Ñ+¨H¸vÓF×FÐFÑFùó   ‚ )¢'£)c              ƒ  óV   #   • U R                   R                  XU5      I S h  v•N $  N7fr    )r   Úrequest©r#   ÚmessagesÚmodel_settingsÚmodel_request_parameterss       r%   r4   ÚWrapperModel.request9   s%   é € ð —\‘\×)Ñ)¨(ÐD\Ó]×]Ð]Ñ]ùr2   c              ƒ  óT   #   • U R                   R                  U5      I S h  v•N $  N7fr    )r   Úcancel_suspended_response©r#   Úresponses     r%   r;   Ú&WrapperModel.cancel_suspended_responseA   s    é € Ø—\‘\×;Ñ;¸HÓE×EÐEÑEùs   ‚(¡&¢(c                ó8   • U R                   R                  U5      $ r    )r   Úcontinuation_delayr<   s     r%   r@   ÚWrapperModel.continuation_delayD   s   € Ø�|‰|×.Ñ.¨xÓ8Ð8r'   c              ƒ  óV   #   • U R                   R                  XU5      I S h  v•N $  N7fr    )r   Úcount_tokensr5   s       r%   rC   ÚWrapperModel.count_tokensG   s%   é € ð —\‘\×.Ñ.¨xÐIaÓb×bÐbÑbùr2   N©Úinstructionsc             ƒ  óP   #   • U R                   R                  XS9I S h  v•N $  N7f)NrE   )r   Úcompact_messages)r#   Úrequest_contextrF   s      r%   rH   ÚWrapperModel.compact_messagesO   s%   é € ð —\‘\×2Ñ2°?Ð2Ð^×^Ð^Ñ^ùs   ‚&Ÿ$ &c               ó¼   #   • U R                   R                  XX45       IS h  v•N nU7v •  S S S 5      IS h  v•N   g  N N! , IS h  v•N  (       d  f       g = f7fr    )r   Úrequest_stream)r#   r6   r7   r8   Úrun_contextÚresponse_streams         r%   rL   ÚWrapperModel.request_streamW   sK   é € ð —<‘<×.Ñ.ØÐ&>÷
ô 
àØ!Ó!÷
÷ 
÷ 
÷ 
÷ 
ó 
üs@   ‚!A£>¤A§A­A¸A ¹AÁ AÁAÁAÁ	AÁAc                ó8   • U R                   R                  U5      $ r    )r   Úcustomize_request_parameters)r#   r8   s     r%   rQ   Ú)WrapperModel.customize_request_parametersd   s   € Ø�|‰|×8Ñ8Ð9QÓRÐRr'   c                ó8   • U R                   R                  X5      $ r    )r   Úprepare_request)r#   r7   r8   s      r%   rT   ÚWrapperModel.prepare_requestg   s   € ð
 �|‰|×+Ñ+¨NÓUÐUr'   c                ó8   • U R                   R                  X5      $ r    )r   Úprepare_messages)r#   r6   r8   s      r%   rW   ÚWrapperModel.prepare_messagesn   s   € ð
 �|‰|×,Ñ,¨XÓPÐPr'   c                ó.   • U R                   R                  $ r    )r   Úproviderr*   s    r%   rZ   ÚWrapperModel.provideru   s   € à�|‰|×$Ñ$Ð$r'   c                ó.   • U R                   R                  $ r    )r   Ú
model_namer*   s    r%   r]   ÚWrapperModel.model_namey   s   € à�|‰|×&Ñ&Ð&r'   c                ó.   • U R                   R                  $ r    )r   Úsystemr*   s    r%   r`   ÚWrapperModel.system}   s   € à�|‰|×"Ñ"Ð"r'   c                ó.   • U R                   R                  $ r    )r   Úmodel_idr*   s    r%   rc   ÚWrapperModel.model_id�   ó   € ð �|‰|×$Ñ$Ð$r'   c                ó.   • U R                   R                  $ r    )r   Úprofiler*   s    r%   rg   ÚWrapperModel.profileŠ   s   € à�|‰|×#Ñ#Ð#r'   c                ó.   • U R                   R                  $ r    )r   Úcontext_windowr*   s    r%   rj   ÚWrapperModel.context_windowŽ   s   € ð �|‰|×*Ñ*Ð*r'   c                ó.   • U R                   R                  $ )z(Get the settings from the wrapped model.)r   Úsettingsr*   s    r%   rm   ÚWrapperModel.settings”   s   € ð �|‰|×$Ñ$Ð$r'   c                ó.   • U R                   R                  $ r    )r   Úbase_urlr*   s    r%   rp   ÚWrapperModel.base_url™   re   r'   c                ó.   • [        U R                  U5      $ r    )Úgetattrr   )r#   Úitems     r%   Ú__getattr__ÚWrapperModel.__getattr__¢   s   € Ü�t—|‘| TÓ*Ð*r'   )r   )r   zModel | KnownModelName)Úreturnr	   )r.   ztype[BaseException] | Noner/   zBaseException | Noner0   zTracebackType | Nonerw   zbool | None)r6   úlist[ModelMessage]r7   úModelSettings | Noner8   r   rw   r   )r=   r   rw   ÚNone)r=   r   rw   zfloat | None)r6   rx   r7   ry   r8   r   rw   r   )rI   r   rF   ú
str | Nonerw   r   r    )
r6   rx   r7   ry   r8   r   rM   zRunContext[Any] | Nonerw   z AsyncGenerator[StreamedResponse])r8   r   rw   r   )r7   ry   r8   r   rw   z3tuple[ModelSettings | None, ModelRequestParameters])r6   rx   r8   zModelRequestParameters | Nonerw   rx   )rw   zProvider[Any] | None)rw   Ústr)rw   r   )rw   z
int | None)rw   ry   )rw   r{   )rt   r|   )Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__Ú__doc__Ú__annotations__r"   r)   r-   r4   r;   r@   rC   rH   r   rL   rQ   rT   rW   ÚpropertyrZ   r]   r`   rc   rg   rj   rm   rp   ru   Ú__static_attributes__Ú__classcell__)r$   s   @r%   r   r      s,  ø‡ ñð
 ƒNØ-÷,ôðGà,ðGð &ðGð %ð	Gð
 
ôGð^à$ð^ð -ð^ð #9ð	^ð
 
ô^ôFô9ðcà$ðcð -ðcð #9ð	cð
 
ôcð $(ñ	_à,ð_ð !ð	_ð
 
õ_ð ð /3ð
"à$ð
"ð -ð
"ð #9ð	
"ð
 ,ð
"ð 
*ô
"ó ð
"ôSðVà,ðVð #9ðVð 
=ô	Vð CGðQà$ðQð #@ðQð 
õ	Qð ó%ó ð%ð ó'ó ð'ð ó#ó ð#ð ó%ó ð%ð ó$ó ð$ð ó+ó ð+ð
 ó%ó ð%ð ó%ó ð%÷+ò +r'   c                ó€   • U S:X  a"  [         R                  " S[        SS9  SSKJn  U$ [        S[        < SU < 35      e)	NÚCompletedStreamedResponsez~`CompletedStreamedResponse` has moved from `pydantic_ai.models.wrapper` to `pydantic_ai.models`; import it from there instead.r
   )Ú
stacklevelr   )r‡   zmodule z has no attribute )ÚwarningsÚwarnr   Ú r‡   ÚAttributeErrorr}   )Únamer‡   s     r%   ru   ru   §   sF   € ØÐ*Ó*Ü�Šð,ä(Øò		
õ 	0à(Ð(Ü
˜7¤8¡,Ð.@ÀÁÐIÓ
JÐJr'   )r�   r|   rw   r   )(Ú
__future__r   r‰   Úcollections.abcr   Ú
contextlibr   Údataclassesr   Útypesr   Útypingr   Útyping_extensionsr	   Ú_run_contextr   Ú	_warningsr   r6   r   r   Úprofilesr   Ú	providersr   rm   r   Úusager   r‹   r   r   r   r   r   r   Ú__all__r   ru   © r'   r%   Ú<module>rœ      sp   ðÝ "ã Ý *Ý *Ý !Ý Ý å "å %Ý 4ß 2Ý #Ý  Ý $Ý  ÷÷ ð Ð
€ñ �ÑôC+�5ó C+ó ðC+õNKr'   