Ki\dZddlZddlZddlZddlmZddlmZddlZddl m Z ddl m Z ddl mZdd lmZmZmZGd d eZGd d eZy)zFError handling middleware for consistent error responses and tracking.N)Callable)Any)McpError) ErrorData) NotFoundError)CallNext MiddlewareMiddlewareContextc eZdZdZ ddej dzdedeee gdfdzdefdZ ded e d dfd Z ded efd Z d e d e d efdZd eeeffdZy)ErrorHandlingMiddlewarea*Middleware that provides consistent error handling and logging. Catches exceptions, logs them appropriately, and converts them to proper MCP error responses. Also tracks error patterns for monitoring. Example: ```python from fastmcp.server.middleware.error_handling import ErrorHandlingMiddleware import logging # Configure logging to see error details logging.basicConfig(level=logging.ERROR) mcp = FastMCP("MyServer") mcp.add_middleware(ErrorHandlingMiddleware()) ``` Nloggerinclude_tracebackerror_callbacktransform_errorscx|xstjd|_||_||_||_i|_y)a}Initialize error handling middleware. Args: logger: Logger instance for error logging. If None, uses 'fastmcp.errors' include_traceback: Whether to include full traceback in error logs error_callback: Optional callback function called for each error transform_errors: Whether to transform non-MCP errors to McpError zfastmcp.errorsN)logging getLoggerrrrr error_counts)selfrrrrs t/home/jay/workspace/scripts/.codegraph-venv/lib/python3.12/site-packages/fastmcp/server/middleware/error_handling.py__init__z ErrorHandlingMiddleware.__init__%s<C 1 12B C !2, 0errorcontextreturnc&t|j}|jxsd}|d|}|jj |ddz|j|<d|d|d|}|j r3|j j|dtjn|j j||jr |j||y y #t$r(}|j jd|Yd }~y d }~wwxYw) z(Log error with appropriate detail level.unknown:rrz Error in :  zError in error callback: N) type__name__methodrgetrrr traceback format_excr Exception)rrr error_typer$ error_key base_messagecallback_errors r _log_errorz"ErrorHandlingMiddleware._log_error:s %[)) ,9"l!F8, '+'8'8'<'*>? ?v1B5)/LM v1A%/KL r call_nextcK ||d{S7#t$r+}|j|||j|}||d}~wwxYww)zHandle errors for all messages.N)r(r-r:)rrr;rtransformed_errors r on_messagez"ErrorHandlingMiddleware.on_messagersT /"7++ ++ / OOE7 +!% 5 5e < # .  /s0A  A  A &AA  A c6|jjS)z$Get error statistics for monitoring.)rcopy)rs rget_error_statsz'ErrorHandlingMiddleware.get_error_stats}s  %%''r)NFNT)r# __module__ __qualname____doc__rLoggerboolrr(r rr-r:r rr>dictstrintrArrr r s()-"'PT!% % !)->!?!EFM   *P P4EP$P.iIB /(9 /h /SV /(c3h(rr ceZdZdZddddeefdfdeded ed ed ee e d fd e jdzf dZ de defdZdedefdZdededefdZy)RetryMiddlewareaiMiddleware that implements automatic retry logic for failed requests. Retries requests that fail with transient errors, using exponential backoff to avoid overwhelming the server or external dependencies. Example: ```python from fastmcp.server.middleware.error_handling import RetryMiddleware # Retry up to 3 times with exponential backoff retry_middleware = RetryMiddleware( max_retries=3, retry_exceptions=(ConnectionError, TimeoutError) ) mcp = FastMCP("MyServer") mcp.add_middleware(retry_middleware) ``` g?gN@g@N max_retries base_delay max_delaybackoff_multiplierretry_exceptions.rc||_||_||_||_||_|xst j d|_y)aInitialize retry middleware. Args: max_retries: Maximum number of retry attempts base_delay: Initial delay between retries in seconds max_delay: Maximum delay between retries in seconds backoff_multiplier: Multiplier for exponential backoff retry_exceptions: Tuple of exception types that should trigger retries logger: Logger for retry attempts z fastmcp.retryN)rNrOrPrQrRrrr)rrNrOrPrQrRrs rrzRetryMiddleware.__init__sA&'$""4 0B 1 1/ B rrrc.t||jS)z-Determine if an error should trigger a retry.)r1rR)rrs r _should_retryzRetryMiddleware._should_retrys%!6!677rattemptcf|j|j|zz}t||jS)z-Calculate delay for the given attempt number.)rOrQminrP)rrVdelays r_calculate_delayz RetryMiddleware._calculate_delays,4#:#:G#CD5$..))rrr;c Kd}t|jdzD]} ||d{cS|r|y7 #t$r}|}||jk(s|j|sYd}~8|j |}|j j d|jd|dzd|jdzdt|jd|d|d d tj|d{7Yd}~d}~wwxYww) z#Implement retry logic for requests.NrzRequest z failed (attempt /z): r z. Retrying in z.1fzs...) rangerNr(rUrZrwarningr$r"r#anyiosleep)rrr; last_errorrVrrYs r on_requestzRetryMiddleware.on_requests T--12 )G )&w/// )&   #0 )" d...d6H6H6O--g6 ##w~~..?! }AdN^N^abNbMccfE{++,Bui~eC[PTV kk%((( )sPD ;9; D; D"C<&D,BC<1C42C<7D<DD)r#rBrCrDConnectionErrorr8rIfloattupler"r(rrErrFrUrZr r rrbrJrrrLrLs,$'9H,8W(,CCC C " C  Y 45 C%C48988*** (9hSVrrL)rDr9rr&collections.abcrtypingrr_mcpr mcp.typesrfastmcp.exceptionsr middlewarer r r r rLrJrrrlsFL$ ,??m(jm(`PjPr