iX dUdZddlZddlZddlmZmZmZddlmZddl m Z ddl m Z ddl Z dedejfd Z dd ed ezd ejdzd edzdeddf dZej( dd edzd ejdzd edzdefdZej,ej.ej0ej2ej4ddZeed dzedzfed<Gddej<Z dd ejded dzded dzddfdZ d ejddfdZ!y)zLogging utilities for FastMCP.N)AnyLiteralcast)Console) RichHandler)overridenamereturnc|jdrtj|Stjd|S)zGet a logger nested under FastMCP namespace. Args: name: the name of the logger, which will be prefixed with 'FastMCP.' Returns: a configured logger instance zfastmcp.r ) startswithlogging getLoggerr s i/Users/bowang/.openclaw/workspace/ChatDev/.venv/lib/python3.12/site-packages/fastmcp/utilities/logging.py get_loggerrs9 z"  d++   HTF"3 44levelDEBUGINFOWARNINGERRORCRITICALloggerenable_rich_tracebacks rich_kwargsc tjjsy|tjj}|t j d}t j d}d|_|j||jddD]}|j|tjjsJt j}|jt j d|j|ytddt!di|}|j||j#d d dl}d dl}t!ddd|d t||gd } | j)|tdi| } | j|| j#d |j||j| y)z Configure logging for FastMCP. Args: logger: the logger to configure level: the log level to use rich_kwargs: the parameters to use for creating RichHandler Nfastmcpz %(message)sFz%(levelname)s: %(message)sconsoleT)stderrc|jduSNexc_inforecords rz#configure_logging..Ps V__%.hs vd/Jr)rsettings log_enabledrrr Formatter propagatesetLevelhandlers removeHandlerenable_rich_logging StreamHandler setFormatter addHandlerrr addFiltermcppydanticupdate) rrrr formatterhdlrhandlerr<r=traceback_kwargstraceback_handlers rconfigure_loggingrDs    ' '%!(!1!1!H!H ~""9-!!-0IF OOE"T"#    / /'')W../KLM'"t$ G # <= $'1!" 'h7 K(#7&67""9- JK g '(rc +K|r_tjj}ttd|j }t d|||d| dt d|||d|ydy#t d|||d|wxYww)a Context manager to temporarily set log level and restore it afterwards. Args: level: The temporary log level to set (e.g., "DEBUG", "INFO") logger: Optional logger to configure (defaults to FastMCP logger) enable_rich_tracebacks: Whether to enable rich tracebacks **rich_kwargs: Additional parameters for RichHandler Usage: with temporary_log_level("DEBUG"): # Code that runs with DEBUG logging pass # Original log level is restored here r)rrrNr/)rr0 log_levelrrupperrD)rrrroriginal_levellog_level_literals rtemporary_log_levelrJns*  ))33! C D KKM   ##9      $'=      $'=   sA A=A)A=)A::A=)rrrrrN _level_to_noceZdZUeeefdzed<eeefdzed< d deddzdeddzffd Ze de jde fdZ xZS) _ClampedLogFilterN min_level max_levelrcd|_d|_tj|x}r|t |f|_tj|x}r|t |f|_t |yr")rNrOrKgetstrsuper__init__)selfrNrO min_level_no max_level_no __class__s rrTz_ClampedLogFilter.__init__sj'++I6 6< 6*C N;DN'++I6 6< 6*C N;DN rr&r c|jr-|j\}}|j|kDr||_||_y|jr-|j\}}|j|kr||_||_yy)NT)rOlevelno levelnamerN)rUr&rWmax_level_namerVmin_level_names rfilterz_ClampedLogFilter.filtersk >>+/>> (L.~~ ,!-#1  >>+/>> (L.~~ ,!-#1 rNN)__name__ __module__ __qualname__tupleintrR__annotations__rrTrr LogRecordboolr^ __classcell__)rXs@rrMrMsS#X%%S#X%%  JK JK  "W..4rrMrNrOcVt||jt||y)aClamp the logger to a minimum and maximum level. If min_level is provided, messages logged at a lower level than `min_level` will have their level increased to `min_level`. If max_level is provided, messages logged at a higher level than `max_level` will have their level decreased to `max_level`. Args: min_level: The lower bound of the clamp max_level: The upper bound of the clamp )r)rNrO)r^N)_unclamp_loggerr;rM)rrNrOs r _clamp_loggerrks%6" - YWXrcp|jddD]$}t|ts|j|&y)z/Remove all clamped log filters from the logger.N)filters isinstancerM removeFilter)rr^s rrjrjs...# f/ 0    '$r)rNNr_)"__doc__ contextlibrtypingrrr rich.consoler rich.loggingrtyping_extensionsrrrRLoggerrrdrgrDcontextmanagerrJrrrrrrKdictreFilterrMrkrjr/rrrzs$%% $& 5S 5W^^ 5 MS$(*.N) B Cc IN) NNT !N)!4KN) N)  N)b %)*.0 :0 NNT !0!4K0 00l]] LL ]]    d ;