idZddlmZddlZddlmZddlmZee e dfee ge fzZ edee ge fzZ ddZddZddZdS)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.)errorcontinue max_retriesint initial_delayfloat max_delaybackoff_factorreturnNonec|dkrd}t||dkrd}t||dkrd}t||dkrd}t|dS)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)rr r r msgs C:\Users\Dell Inspiron 16\Desktop\tws\AgrotaPowerBi\back-agrota-powerbi\mcp-client-agrota\venv\Lib\site-packages\langchain/agents/middleware/_retry.pyvalidate_retry_paramsr#sx"Q(ooq*oo1}}&oo+ooexc Exceptionretry_onRetryOnboolcVt|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)rrs rshould_retry_exceptionrBs2x}} c8 $ $$r retry_numberjitterc|dkr|}n|||zz}t||}|r4|dkr.|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)rr r r r delay jitter_amounts rcalculate_delayr(Usy0!=> y ! !E %!))   >>>Au  Lr) rr r r r r r r rr)rrrrrr) rr r r r r r r r rrr )__doc__ __future__rr#collections.abcrtypingrtupletyperrrstr OnFailurerrr(rrr2s #""""" $$$$$$ Y$ %)d1B(C C ' (8YK4D+E E  >%%%%&&&&&&&r