o i$@sdZddlZddlZddlZddlmZddlmZddlmZddl Z ddl m Z m Z m Z ded efd d ZGd d d e ZGdddeZGdddeZded efddZdS)z5Comprehensive logging middleware for FastMCP servers.N)Callable)Logger)Any)CallNext MiddlewareMiddlewareContextdatareturncCstj|tdS)z>The default serializer for Payloads in the logging middleware.)fallback) pydantic_coreto_jsonstrdecode)r rg/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/fastmcp/server/middleware/logging.pydefault_serializersrc @sveZdZUdZeed<eed<eed<eed<eed<edBed<ee dBed <eed <e e ge fdBed <d e e d e fddZ dee e eBeBfd e fddZd e e d ee e eBeBffddZd e e deded ee e eBeBffddZd e e ded ee e eBeBffddZ d dee e eBeBfdedBfddZd e e dee e fd e fddZdS)!BaseLoggingMiddlewarez"Base class for logging middleware.logger log_levelinclude_payloadsinclude_payload_lengthestimate_payload_tokensNmax_payload_lengthmethodsstructured_loggingpayload_serializercontextr cCs|js t|j}|Sz ||j}W|StyA}z"|jd|d|jd|jd|jd t|j}WYd}~|Sd}~ww)Nz#Failed to serialize payload due to z:  .) rrmessage Exceptionrwarningtypemethodsource)selfrpayloaderrr_serialize_payload!s  "z(BaseLoggingMiddleware._serialize_payloadr cCs(|jrt|Sddd|DS)zFormat a message for logging.rcSsg|] \}}|d|qS)=r).0kvrrr 6sz9BaseLoggingMiddleware._format_message..)rjsondumpsjoinitems)r&r rrr_format_message1s z%BaseLoggingMiddleware._format_messagecCs|jd|jpd|jd}|js|js|jrY||}|js!|jr7t|}|d}|jr0||d<|jr7||d<|jrJt||jkrJ|d|jd}|jrY||d<t|j j |d <|S) N_startunknown)eventr$r%payload_tokenspayload_lengthz...r' payload_type) r#r$r%rrrr)lenrr __name__)r&rr r'r9r8rrr_create_before_message8s0  z,BaseLoggingMiddleware._create_before_message start_timeerrorcCs0t|}|jd|jp d|j|t|dd}|S)N_errorr5)object)r6r$r% duration_msr?)_get_duration_msr#r$r%r)r&rr>r?rBr rrr_create_error_messageYsz+BaseLoggingMiddleware._create_error_messagecCs(t|}|jd|jp d|j|d}|S)N_successr5)r6r$r%rB)rCr#r$r%)r&rr>rBr rrr_create_after_messageisz+BaseLoggingMiddleware._create_after_messagecCs|j|p|j||dSN)rlogrr3)r&r rrrr _log_messagewsz"BaseLoggingMiddleware._log_message call_nextc s|jr|j|jvr||IdHS|||t}z||IdH}|||||WStyI}z |||||t j d}~ww)z$Log messages for configured methods.N) rr$rIr=time perf_counterrFr!rDloggingERROR)r&rrJr>resultr(rrr on_message|s z BaseLoggingMiddleware.on_messagerG)r< __module__ __qualname____doc__r__annotations__intboollistrrrrr)dictfloatr3r=r!rDrFrIrrPrrrrrs^  " !    rc@sleZdZdZdejddddddddejdBdeded ed ed ed e e dBd e e ge fdBfddZ dS)LoggingMiddlewareaMiddleware that provides comprehensive request and response logging. Logs all MCP messages with configurable detail levels. Useful for debugging, monitoring, and understanding server usage patterns. Example: ```python from fastmcp.server.middleware.logging import LoggingMiddleware import logging # Configure logging logging.basicConfig(level=logging.INFO) mcp = FastMCP("MyServer") mcp.add_middleware(LoggingMiddleware()) ``` NF)rrrrrrrrrrrrrrrrc CsD|ptd|_||_||_||_||_||_||_||_ d|_ dS)aInitialize logging middleware. Args: logger: Logger instance to use. If None, creates a logger named 'fastmcp.requests' log_level: Log level for messages (default: INFO) include_payloads: Whether to include message payloads in logs include_payload_length: Whether to include response size in logs estimate_payload_tokens: Whether to estimate response tokens max_payload_length: Maximum length of payload to log (prevents huge logs) methods: List of methods to log. If None, logs all methods. payload_serializer: Callable that converts objects to a JSON string for the payload. If not provided, uses FastMCP's default tool serializer. zfastmcp.middleware.loggingFN) rM getLoggerrrrrrrrrr) r&rrrrrrrrrrr__init__s zLoggingMiddleware.__init__r<rQrRrSrMINFOrrUrVrWrrrr]rrrrrZs6  rZc@sfeZdZdZdejdddddddejdBdededed ed e e dBd e e ge fdBfd d Z dS)StructuredLoggingMiddlewareaMiddleware that provides structured JSON logging for better log analysis. Outputs structured logs that are easier to parse and analyze with log aggregation tools like ELK stack, Splunk, or cloud logging services. Example: ```python from fastmcp.server.middleware.logging import StructuredLoggingMiddleware import logging mcp = FastMCP("MyServer") mcp.add_middleware(StructuredLoggingMiddleware()) ``` NF)rrrrrrrrrrrrrrcCsD|ptd|_||_||_||_||_||_||_d|_ d|_ dS)aInitialize structured logging middleware. Args: logger: Logger instance to use. If None, creates a logger named 'fastmcp.structured' log_level: Log level for messages (default: INFO) include_payloads: Whether to include message payloads in logs include_payload_length: Whether to include payload size in logs estimate_payload_tokens: Whether to estimate token count using length // 4 methods: List of methods to log. If None, logs all methods. payload_serializer: Callable that converts objects to a JSON string for the payload. If not provided, uses FastMCP's default tool serializer. z%fastmcp.middleware.structured_loggingNT) rMr\rrrrrrrrr)r&rrrrrrrrrrr]s z$StructuredLoggingMiddleware.__init__r^rrrrr`s0  r`r>cCstt|dddS)Nr[)numberndigits)roundrKrL)r>rrrrCsrC)rSr/rMrKcollections.abcrrtypingrr middlewarerrrrrrrZr`rYrCrrrrs   74