o pkj#>@sdZddlZddlZddlZddlmZddlmZddlmZm Z ddl Z zddl Z Wn e y7dZ Ynwe durBe j e jfn e j e je j e jfZGdddZGdd d ZGd d d ZGd d d eZGdddeZde jde efddZde jde efddZ  d0dedededededede edefddZdefd d!Zdefd"d#Z $ % & ' d1d(d)Z $ % &  d2d*d+Z $ % & ' d1d,d-Z! $ % &  d2d.d/Z"dS)3zACode generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.N)datetime)parsedate_to_datetime)ListOptionalc @sdeZdZUdZeed<eed<eed<eed<eeed< d dededededeef dd ZdS) BackoffStrategyz+Exponential backoff strategy configuration.initial_interval max_intervalexponentmax_elapsed_time jitter_msNcCs:|dur |dkr td||_||_||_||_||_dS)aInitialize a backoff strategy. Args: initial_interval: Initial retry interval in milliseconds. max_interval: Maximum retry interval in milliseconds. exponent: Base of the exponential backoff; the interval grows as ``initial_interval * exponent ** retries``. max_elapsed_time: Maximum total elapsed time in milliseconds. Only used by the ``"backoff"`` strategy; ignored by ``"attempt-count-backoff"``. jitter_ms: Additive jitter bound in milliseconds. When set, adds a random value in ``[0, jitter_ms]`` to each computed backoff interval (default -25% for attempt-count, +[0, 1s] for backoff). Note: When a response carries a ``Retry-After`` or ``retry-after-ms`` header, that delay is used as-is and the sleep-shaping parameters (``initial_interval``, ``max_interval``, ``exponent``, ``jitter_ms``) are ignored for that attempt. Nrzjitter_ms must be >= 0) ValueErrorrrr r r )selfrrr r r r^/home/djax/ivt_ai_plugin/venv/lib/python3.10/site-packages/google/genai/_gaos/utils/retries.py__init__1s zBackoffStrategy.__init__N) __name__ __module__ __qualname____doc__int__annotations__floatrrrrrrr(s&  rc @s~eZdZUdZeed<eeed<eed<ee ed<ee eed<  d dedeededee dee ef dd Z dS) RetryConfigzRuntime retry configuration.strategybackoffretry_connection_errors max_retriesstatus_codes_overrideNcCs"||_||_||_||_||_dS)aEInitialize a retry configuration. Args: strategy: Retry strategy: ``"none"``, ``"backoff"`` or ``"attempt-count-backoff"``. backoff: Backoff parameters; falls back to a built-in default when ``None``. retry_connection_errors: Whether to also retry transport-level connection errors. max_retries: Maximum number of retries for the ``"attempt-count-backoff"`` strategy. status_codes_override: Retryable HTTP status codes that take precedence over the per-operation defaults when non-empty. N)rrrrr)r rrrrrrrrr^s  zRetryConfig.__init__)NN) rrrrstrrrrboolrrrrrrrrUs(    rc@s8eZdZUeed<eeed<dedeefddZdS)Retriesconfig status_codescCs||_|jp||_dSr)r"rr#)r r"r#rrrr{szRetries.__init__N)rrrrrrrrrrrrr!ws  r!c@s4eZdZUejed<eeed<dejfddZdS)TemporaryErrorresponse retry_aftercCs||_t||_dSr)r%_parse_retry_after_headerr&)r r%rrrrszTemporaryError.__init__N) rrrhttpxResponserrrrrrrrr$s  r$c@s$eZdZUeed<defddZdS)PermanentErrorinnercCs ||_dSr)r+)r r+rrrrs zPermanentError.__init__N)rrr Exceptionrrrrrrr*s r*r%returnc Cs|jd}|s dSz t|}t|dWStyYnwzt|}|t|j }tt d|dWStt fyCYdSw)zParse Retry-After header from response. Returns: Retry interval in milliseconds, or None if header is missing or invalid. z retry-afterNr) headersgetrroundr rrnowtzinfo total_secondsmax TypeError)r%retry_after_headerseconds retry_datedeltarrrr's"  r'c CsN|jd}|s dSzt|}|dkrt|WSWdSttfy&YdSw)Nzretry-after-msr)r/r0rr1 OverflowErrorr )r%retry_after_ms_header millisecondsrrr_parse_retry_after_ms_headers  r>F exceptionrrr retriesattempt_count_backoffr cCst|tr|jdur|jdkr|jdS|d||}|dur+|td|d7}n|r8|dtd}n|tdd7}t||dS)aEGet sleep interval for retry with exponential backoff. Args: exception: The exception that triggered the retry. initial_interval: Initial retry interval in milliseconds. max_interval: Maximum retry interval in milliseconds. exponent: Base for exponential backoff calculation. retries: Current retry attempt count. attempt_count_backoff: Use multiplicative jitter when ``jitter_ms`` is unset. jitter_ms: Additive jitter bound in ms; see ``BackoffStrategy.jitter_ms``. Returns: Sleep interval in seconds. Nrr.g?) isinstancer$r&randomuniformmin)r?rrr r@rAr sleeprrr_get_sleep_intervals   rHcsjjdvrCdtdtjffdd }jjptdddd }jjd kr5t||j|j |j jj p1d |j St ||j|j |j |j|j Sd S) Nrattempt-count-backoffattemptr-c szB|}d}jD]/}d|vr.t|d}|jd}||kr)|dkr-nq d}q t|}|j|kr9d}q |r@t|W|StyX}z jjrOt||d}~wty_t yo}zt||d}~wwNFXrdrBT r#upperr status_coder$_RETRY_EXCEPTIONSr"rr*r,rKres should_retrycode code_range status_major parsed_coder?funcr@rr do_requests:         zretry..do_request`?6rJr)r"rrr(r)rr retry_with_attempt_count_backoffrrr rr retry_with_backoffr r[r@r\rrrZrretrys* "    rdcsjjdvrJdtdtjffdd }jjptdddd }jjd kr9t||j|j |j jj p2d |j IdHSt ||j|j |j |j|j IdHSd IdHS) NrIrKr-c szE|IdH}d}jD]/}d|vr2t|d}|jd}||kr-|dkr1nqd}qt|}|j|kr=d}q|rDt|W|Sty\}z jjrSt||d}~wtyct ys}zt||d}~wwrLrOrSrZrrr\ s<         zretry_async..do_requestr]r^r_r`rJr)r"rrr(r)rr&retry_with_attempt_count_backoff_asyncrrr rr retry_with_backoff_asyncr rcrrZr retry_asyncs, "      rgr]r^r_r`c Csttd}d} z||WSty}z|jd}~wtyq}zGttd} | ||krCt|trB|jWYd}~St|trTt|j} | durT| |_ t ||||||d} t | |d7}WYd}~nd}~wwq Nr.rT)r rB) r1timer*r+r,rCr$r%r>r&rHrG r[rrr r r startr@r?r2retry_after_msrGrrrrbYs>      rbc Csd} z||WSty}z|jd}~wty`}z>||kr1t|tr0|jWYd}~St|trBt|j}|durB||_t|||||d|d} t | |d7}WYd}~nd}~wwqNrT)rAr rB) r*r+r,rCr$r%r>r&rHrirG r[rrr rr r@r?rlrGrrrras<     rac sttd}d} z||IdHWSty#}z|jd}~wtyx}zJttd} | ||krGt|trF|jWYd}~St|trXt|j} | durX| |_ t ||||||d} t | IdH|d7}WYd}~nd}~wwq rh) r1rir*r+r,rCr$r%r>r&rHasynciorGrjrrrrfs@    rfc sd} z||IdHWSty}z|jd}~wtyg}zA||kr5t|tr4|jWYd}~St|trFt|j}|durF||_t|||||d|d} t | IdH|d7}WYd}~nd}~wwqrm) r*r+r,rCr$r%r>r&rHrorGrnrrrres>    re)FN)r]r^r_r`N)r]r^r_rN)#rrorDrir email.utilsrtypingrrr(httpx2 ImportError NetworkErrorTimeoutExceptionrRrrr!r,r$r*r)rr'r>rr rHrdrgrbrarfrerrrrs     -"   (<> * ) *