o i.@sdZddlZddlZddlZddlmZddlmZddlZddl m Z ddl m Z ddl mZdd lmZmZmZGd d d eZGd d d eZdS)zFError handling middleware for consistent error responses and tracking.N)Callable)Any)McpError) ErrorData) NotFoundError)CallNext MiddlewareMiddlewareContextc @seZdZdZ    ddejdBdedeee gdfdBdefd d Z d ed e d dfddZ d ed efddZ d e de d efddZd eeeffddZdS)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()) ``` NFTloggerinclude_tracebackerror_callbacktransform_errorscCs,|ptd|_||_||_||_i|_dS)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 getLoggerr r rr error_counts)selfr r rrrn/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/fastmcp/server/middleware/error_handling.py__init__%s  z ErrorHandlingMiddleware.__init__errorcontextreturnc Cst|j}|jp d}|d|}|j|dd|j|<d|d|d|}|jr9|j|dt n|j||j rhz | ||Wd St yg}z|jd|WYd }~d Sd }~wwd S) z(Log error with appropriate detail level.unknown:rrz Error in :  zError in error callback: N) type__name__methodrgetr r r traceback format_excr Exception)rrr error_typer error_key base_messagecallback_errorrrr _log_error:s    z"ErrorHandlingMiddleware._log_errorcCst|tr|S|js |St|}|ttfvr!ttdd|dS|ttt fvr3ttdd|dS|t urBttdd|dS|t t j fvrTttdd|dSttd d |dS) z.Transform non-MCP errors to proper MCP errors.izInvalid params: )codemessageizResource not found: izPermission denied: zRequest timeout: izInternal error: ) isinstancerrr ValueError TypeErrorrFileNotFoundErrorKeyErrorrPermissionError TimeoutErrorasyncio)rrr%rrr_transform_errorQs0  z(ErrorHandlingMiddleware._transform_error call_nextc sHz||IdHWSty#}z|||||}||d}~ww)zHandle errors for all messages.N)r$r)r4)rrr5rtransformed_errorrrr on_messagers  z"ErrorHandlingMiddleware.on_messagecCs |jS)z$Get error statistics for monitoring.)rcopy)rrrrget_error_stats}s z'ErrorHandlingMiddleware.get_error_stats)NFNT)r __module__ __qualname____doc__rLoggerboolrr$r rr)r4rrr7dictstrintr9rrrrr s& ! r c@seZdZdZddddeefdfdeded ed ed ee e d fd e j dBf ddZ de defddZdedefddZdededefddZdS)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.r cCs2||_||_||_||_||_|ptd|_dS)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)rDrErFrGrHrrr )rrDrErFrGrHr rrrrs zRetryMiddleware.__init__rrcCs t||jS)z-Determine if an error should trigger a retry.)r,rH)rrrrr _should_retrys zRetryMiddleware._should_retryattemptcCs|j|j|}t||jS)z-Calculate delay for the given attempt number.)rErGminrF)rrJdelayrrr_calculate_delays z RetryMiddleware._calculate_delayrr5csd}t|jdD]a}z ||IdHWStyk}zI|}||jks*||s1WYd}~n;||}|jd|jd|dd|jddt|j d|d|d d t |IdHWYd}~q d}~ww|rp|dS) z#Implement retry logic for requests.NrzRequest z failed (attempt /z): rz. Retrying in z.1fzs...) rangerDr$rIrMr warningr rranyiosleep)rrr5 last_errorrJrrLrrr on_requests2  zRetryMiddleware.on_request)rr:r;r<ConnectionErrorr2rAfloattuplerr$rr=rr>rIrMr rrrTrrrrrBs0 rB)r<r3rr"collections.abcrtypingrrQmcpr mcp.typesrfastmcp.exceptionsr middlewarerr r r rBrrrrs     p