§ ‘ ¼ió*ãóÈ—dZddlmZddlZddlZddlmZddlmZddl m Z m Z m Z m Z mZddlmZmZmZmZmZmZerddlmZmZGd „d eeeeef¦«ZdS) z"Model retry middleware for agents.é)Ú annotationsN)Ú TYPE_CHECKING)Ú AIMessage)Ú OnFailureÚRetryOnÚcalculate_delayÚshould_retry_exceptionÚvalidate_retry_params)ÚAgentMiddlewareÚ AgentStateÚContextTÚ ModelRequestÚ ModelResponseÚ ResponseT)Ú AwaitableÚCallablecóh‡—eZdZdZdefddddddœd'ˆfd„Zed(d„¦«Zd)d„Zd*d$„Z d+d&„Z ˆxZ S),ÚModelRetryMiddlewarea Middleware that automatically retries failed model calls with configurable backoff. Supports retrying on specific exceptions and exponential backoff. Examples: !!! example "Basic usage with default settings (2 retries, exponential backoff)" ```python from langchain.agents import create_agent from langchain.agents.middleware import ModelRetryMiddleware agent = create_agent(model, tools=[search_tool], middleware=[ModelRetryMiddleware()]) ``` !!! example "Retry specific exceptions only" ```python from anthropic import RateLimitError from openai import APITimeoutError retry = ModelRetryMiddleware( max_retries=4, retry_on=(APITimeoutError, RateLimitError), backoff_factor=1.5, ) ``` !!! example "Custom exception filtering" ```python from anthropic import APIStatusError def should_retry(exc: Exception) -> bool: # Only retry on 5xx errors if isinstance(exc, APIStatusError): return 500 <= exc.status_code < 600 return False retry = ModelRetryMiddleware( max_retries=3, retry_on=should_retry, ) ``` !!! example "Custom error handling" ```python def format_error(exc: Exception) -> str: return "Model temporarily unavailable. Please try again later." retry = ModelRetryMiddleware( max_retries=4, on_failure=format_error, ) ``` !!! example "Constant backoff (no exponential growth)" ```python retry = ModelRetryMiddleware( max_retries=5, backoff_factor=0.0, # No exponential growth initial_delay=2.0, # Always wait 2 seconds ) ``` !!! example "Raise exception on failure" ```python retry = ModelRetryMiddleware( max_retries=2, on_failure="error", # Re-raise exception instead of returning message ) ``` éÚcontinueg@gð?gN@T)Ú max_retriesÚretry_onÚ on_failureÚbackoff_factorÚ initial_delayÚ max_delayÚjitterrÚintrrrrrÚfloatrrrÚboolÚreturnÚNonecóÜ•—t¦« ¦«t||||¦«||_g|_||_||_||_||_||_ ||_ dS)uàInitialize `ModelRetryMiddleware`. Args: max_retries: Maximum number of retry attempts after the initial call. Must be `>= 0`. 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. Default is to retry on all exceptions. on_failure: Behavior when all retries are exhausted. Options: - `'continue'`: Return an `AIMessage` with error details, allowing the agent to continue with an error response. - `'error'`: Re-raise the exception, stopping agent execution. - **Custom callable:** Function that takes the exception and returns a string for the `AIMessage` content, allowing custom error formatting. backoff_factor: Multiplier for exponential backoff. Each retry waits `initial_delay * (backoff_factor ** retry_number)` seconds. 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 (`±25%`) to delay to avoid thundering herd. Raises: ValueError: If `max_retries < 0` or delays are negative. N) ÚsuperÚ__init__r rÚtoolsrrrrrr) ÚselfrrrrrrrÚ __class__s €ú›C:\Users\Dell Inspiron 16\Desktop\tws\AgrotaPowerBi\back-agrota-powerbi\mcp-client-agrota\venv\Lib\site-packages\langchain/agents/middleware/model_retry.pyr%zModelRetryMiddleware.__init__osqø€õ\ ‰Œ×ÒÑÔÐõ ˜k¨=¸)À^ÑTÔTÐTà&ˆÔ؈Œ Ø ˆŒ Ø$ˆŒØ,ˆÔØ*ˆÔØ"ˆŒØˆŒ ˆ ˆ óÚexcÚ ExceptionÚ attempts_madercó˜—t|¦«j}t|¦«}|dkrdnd}d|›d|›d|›d|›}t|¬¦«S) a Format the failure message when retries are exhausted. Args: exc: The exception that caused the failure. attempts_made: Number of attempts actually made. Returns: `AIMessage` with formatted error message. éÚattemptÚattemptszModel call failed after ú z with z: ©Úcontent)ÚtypeÚ__name__Ústrr)r+r-Úexc_typeÚexc_msgÚ attempt_wordr4s r)Ú_format_failure_messagez,ModelRetryMiddleware._format_failure_message«sl€õ˜‘9”9Ô%ˆÝc‘(”(ˆØ$1°QÒ$6Ð$6yy¸Jˆ à ` }Ð `Ð `°|Ð `Ð `È8Ð `Ð `ÐW^Ð `Ð `ð õ Ð)Ñ)Ô)Ð)r*úModelResponse[ResponseT]cóÞ—|jdkr|‚t|j¦«r&| |¦«}t|¬¦«}n| ||¦«}t |g¬¦«S)a\Handle failure when all retries are exhausted. Args: exc: The exception that caused the failure. attempts_made: Number of attempts actually made. Returns: `ModelResponse` with error details. Raises: Exception: If `on_failure` is `'error'`, re-raises the exception. Úerrorr3)Úresult)rÚcallablerr;r)r'r+r-r4Úai_msgs r)Ú_handle_failurez$ModelRetryMiddleware._handle_failure¾st€ð Œ?˜gÒ %Ð %؈Iå D”OÑ $Ô $ð FØ—o’o cÑ*Ô*ˆGÝ wÐ/Ñ/Ô/ˆFˆFà×1Ò1°#°}ÑEÔEˆFå V HÐ-Ñ-Ô-Ð-r*ÚrequestúModelRequest[ContextT]ÚhandlerúrcsZðØ(Ð(à"Ð"Ð"Ð"Ð"Ð"à€€€Ø € € € Ø Ð Ð Ð Ð Ð à-Ð-Ð-Ð-Ð-Ð-ðððððððððððððððððððððððððððððððð4Ø3Ð3Ð3Ð3Ð3Ð3Ð3Ð3ðY ðY ðY ðY ðY ˜?¨:°iÔ+@À(ÈIÐ+UÔVñY ôY ðY ðY ðY r*