�
��g� � �$ � d dl Z d dlZd dlmZ d dlmZ ej rd dlmZ G d� de� � Zej eej dgej
e f f Z G d� d e� � Z
G d
� d e� � Z G d � d
e� � Z G d� de� � ZdS )� N)�_utils)�
retry_base)�RetryCallStatec �| � e Zd ZdZej dddefd�� � Z dd �Z dd
�Z dd �Z
dd
�Z dS )�async_retry_basez/Abstract base class for async retry strategies.� retry_stater �returnc � �
K � d S �N� ��selfr s �f/home/asafur/pinokio/api/open-webui.git/app/env/lib/python3.11/site-packages/tenacity/asyncio/retry.py�__call__zasync_retry_base.__call__ s � � � � �� �other�*typing.Union[retry_base, async_retry_base]� retry_allc �" � t | |� � S r �r �r r s r �__and__zasync_retry_base.__and__! � � � ��u�%�%�%r c �" � t || � � S r r r s r �__rand__zasync_retry_base.__rand__& � � � ���%�%�%r � retry_anyc �" � t | |� � S r �r r s r �__or__zasync_retry_base.__or__+ r r c �" � t || � � S r r r s r �__ror__zasync_retry_base.__ror__0 r r N)r r r r )r r r r ) �__name__�
__module__� __qualname__�__doc__�abc�abstractmethod�boolr r r r r" r r r r r s� � � � � � �9�9���
�*:�
�t�
�
�
� ��
�&�A�&� �&� &� &� &�
&�A�&� �&� &� &� &�
&�A�&� �&� &� &� &�
&�A�&� �&� &� &� &� &� &r r r c �d � e Zd ZdZdej egej e f ddfd�Z dddefd�Z
dS ) �retry_if_exceptionzARetry strategy that retries if an exception verifies a predicate.� predicater Nc � � || _ d S r �r, �r r, s r �__init__zretry_if_exception.__init__> � � � #����r r r c � �� K � |j �t d� � �|j j rE|j � � � }|�t d� � �| � |� � � d {V ��S dS )N�(__call__() called before outcome was setz(outcome failed but the exception is NoneF)�outcome� RuntimeError�failed� exceptionr, )r r r7 s r r zretry_if_exception.__call__C s~ � � � � � � &� � I�J�J� J� � � %� �#�+�5�5�7�7�I�� �"�#M�N�N�N� � � � �2�2�2�2�2�2�2�2� 2��5r ) r# r$ r% r& �typing�Callable�
BaseException� Awaitabler) r0 r r r r r+ r+ ; sw � � � � � �K�K�#� � �-� �&�:J�4�:P�)P�Q�#�
�#� #� #� #�
�*:�
�t�
�
�
�
�
�
r r+ c �n � e Zd ZdZdej ej gej e f ddfd�Z dddefd�Z
dS ) �retry_if_resultz+Retries if the result verifies a predicate.r, r Nc � � || _ d S r r. r/ s r r0 zretry_if_result.__init__S r1 r r r c � � K � |j �t d� � �|j j s2| � |j � � � � � � d {V ��S dS )Nr3 F)r4 r5 r6 r, �resultr
s r r zretry_if_result.__call__X sd � � � � � � &� � I�J�J� J��"�)� � � � � �(;�(B�(B�(D�(D�E�E�E�E�E�E�E�E� E��5r ) r# r$ r% r& r8 r9 �Anyr; r) r0 r r r r r= r= P sy � � � � � �5�5�#� � �&�*� �v�7G��7M�)M�N�#�
�#� #� #� #�
�*:� �t� � � � � � r r= c �L � e Zd ZdZdej eef ddfd�Zddde fd�Z
dS ) r z1Retries if any of the retries condition is valid.�retriesr Nc � � || _ d S r �rC �r rC s r r0 zretry_any.__init__e �
� � �� � � r r r c � �t K � d}| j D ]+}|p" t j |� � |� � � d {V ��}|r n�,|S )NF�rC r �wrap_to_async_func�r r r@ �rs r r zretry_any.__call__h se � � � ���� � � �A��N�%A�V�%>�q�%A�%A�+�%N�%N� N� N� N� N� N� N�F��
���
��
r � r# r$ r% r& r8 �Unionr r r0 r) r r r r r r b sm � � � � � �;�;� �� �j�:J�.J�!K� �PT� � � � ��*:� �t� � � � � � r r c �L � e Zd ZdZdej eef ddfd�Zddde fd�Z
dS ) r z/Retries if all the retries condition are valid.rC r Nc � � || _ d S r rE rF s r r0 zretry_all.__init__t rG r r r c � �t K � d}| j D ]+}|o" t j |� � |� � � d {V ��}|s n�,|S )NTrI rK s r r zretry_all.__call__w se � � � ���� � � �A��O�&B�f�&?��&B�&B�;�&O�&O� O� O� O� O� O� O�F��
���
��
r rM r r r r r q sm � � � � � �9�9� �� �j�:J�.J�!K� �PT� � � � ��*:� �t� � � � � � r r )r' r8 �tenacityr r �
TYPE_CHECKINGr r rN r9 r; r) �
RetryBaseTr+ r= r r r r r �<module>rU sh �� �
�
�
�
�
�
�
� � � � � � � � � � � � � �� (�'�'�'�'�'�'�&� &� &� &� &�z� &� &� &�8 �\��f�o�'7�&8�&�:J�4�:P�&P�Q�Q��
�
� � � � �)� � � �*� � � � �&� � � �$ � � � � � � � � � � � � � � � � � � � r