g3fidZddlmZddlZddlmZddlmZee e dfee ge fzZ edee ge fzZ d dZ d d Z d d Zy)zShared retry utilities for agent middleware. This module contains common constants, utilities, and logic used by both model and tool retry middleware implementations. ) annotationsN)Callable)Literal.)errorcontinuec|dkr d}t||dkr d}t||dkr d}t||dkr d}t|y)anValidate retry parameters. Args: max_retries: Maximum number of retry attempts. initial_delay: Initial delay in seconds before first retry. max_delay: Maximum delay in seconds between retries. backoff_factor: Multiplier for exponential backoff. Raises: ValueError: If any parameter is invalid (negative values). rzmax_retries must be >= 0zinitial_delay must be >= 0zmax_delay must be >= 0zbackoff_factor must be >= 0N) ValueError) max_retries initial_delay max_delaybackoff_factormsgs `/var/www/auto_recruiter/arenv/lib/python3.12/site-packages/langchain/agents/middleware/_retry.pyvalidate_retry_paramsr#sc"Q(oq*o1}&o+oc@t|r||St||S)aZCheck if an exception should trigger a retry. Args: exc: The exception that occurred. retry_on: Either a tuple of exception types to retry on, or a callable that takes an exception and returns `True` if it should be retried. Returns: `True` if the exception should be retried, `False` otherwise. )callable isinstance)excretry_ons rshould_retry_exceptionrBs#} c8 $$rc|dk(r|}n|||zz}t||}|r0|dkDr+|dz}|tj| |z }td|}|S)a@Calculate delay for a retry attempt with exponential backoff and optional jitter. Args: retry_number: The retry attempt number (0-indexed). backoff_factor: Multiplier for exponential backoff. Set to `0.0` for constant delay. initial_delay: Initial delay in seconds before first retry. max_delay: Maximum delay in seconds between retries. Caps exponential backoff growth. jitter: Whether to add random jitter to delay to avoid thundering herd. Returns: Delay in seconds before next retry. grg?)minrandomuniformmax) retry_numberr r r jitterdelay jitter_amounts rcalculate_delayr!Usg0!=> y !E %!)   >>Au  Lr) r intr floatr r#r r#returnNone)r ExceptionrRetryOnr$bool) rr"r r#r r#r r#rr(r$r#)__doc__ __future__rrcollections.abcrtypingrtupletyper&r(r'str OnFailurerrr!rrr2s # $ Y$ %)d1B(C C ' (8YK4D+E E     >% %% %&&&& &  &  & &r