i dZddlZddlZddlZddlmZddlmZddlmZm Z m Z ddl m Z GddejZGd d ejZ d dededede ededej$f dZdedej$fdZdej$dedefdZ d!dej$dedeee fde efdZed d d d Zy)"z Logging configuration for the RAG system. This module provides structured logging capabilities with different log levels, JSON formatting for production, and configurable outputs. N)datetime)Path)DictAnyOptional)BASE_DIRc4eZdZdZdej defdZy) JSONFormatterz&JSON formatter for structured logging.recordreturnctj|jj|j|j |j |j|j|jd}|jr|j|j|d<t|dr|j|jtj |dS)N) timestamplevelloggermessagemodulefunctionline exception extra_fieldsF) ensure_ascii)r fromtimestampcreated isoformat levelnamename getMessagerfuncNamelinenoexc_infoformatExceptionhasattrupdaterjsondumps)selfr log_entrys 2/var/www/html/leadgen/airagagent/logging_config.pyformatzJSONFormatter.formats!//?IIK%%kk((*mmMM  ??%)%9%9&//%JIk " 6> *   V00 1zz)%88N__name__ __module__ __qualname____doc__logging LogRecordstrr)r*r(r r s09W..939r*r c4eZdZdZdej defdZy)ConsoleFormatterz!Human-readable console formatter.r r c6tj|jjd}d|jd}d|j d}|j }|d|d|d|}|jr!|d|j|jz }|S)Nz%Y-%m-%d %H:%M:%S[]  ) rrrstrftimerrrr r!)r&r rrrr formatteds r(r)zConsoleFormatter.format-s**6>>:CCDWX F$$%Q'V]]O1%##% k5'6(!G9=  ?? 2d226??CDE EIr*Nr+r3r*r(r5r5*s+ W.. 3 r*r5INFOTFr log_to_filelog_to_consolelog_dir json_formatr c"| tdz }|jddtt|j tj }tj }|j||jddD]}|j|t}t} |rVtjtj} | j|| j||j!| |r |dt#j$j'ddz } tj(| d } | j||r| j| n| j||j!| |d t#j$j'ddz } tj(| d }|jtj*|j||j!||S) a Set up logging configuration for the RAG system. Args: level: Logging level (DEBUG, INFO, WARNING, ERROR, CRITICAL) log_to_file: Whether to log to file log_to_console: Whether to log to console log_dir: Directory for log files (defaults to BASE_DIR/logs) json_format: Whether to use JSON formatting for file logs Returns: Root logger configured with the specified settings NlogsT)parentsexist_ok rag_system_z%Y%m%dz.logzutf-8)encodingrag_system_errors_)rmkdirgetattrr0upperr= getLoggersetLevelhandlers removeHandlerr5r StreamHandlersysstdout setFormatter addHandlerrnowr; FileHandlerWARNING)rr>r?r@rA numeric_levelrhandlerconsole_formatterjson_formatterconsole_handlerlog_file file_handlererror_log_file error_handlers r( setup_loggingra<s(V# MM$M.GU[[]GLLAM   F OOM"??1%&W%&)*"_N!// ;  /$$%67/*{8<<>+B+B8+L*MTRR**8gF m,   % %n 5  % %&7 8,'!%7 8O8OPX8Y7ZZ^#__++NWM w/""#45-( Mr*rc,tj|S)z Get a logger instance for a specific module. Args: name: Logger name (usually __name__) Returns: Configured logger instance )r0rL)rs r( get_loggerrcs   T ""r*r operationdurationc H|jd|d|ddd||d|iy) zLog performance metrics.z Performance: z completed in z.3fsr)rdre)extraN)info)rrdrers r(log_performancerjs: KK  {.#a@YH]P\]^r*errorcontextrc |d|jj}|j||d|jjt|d|iy)z1Log an error with additional context information.NzError occurred: r) error_type error_message)r rh) __class__r,rkr2)rrkrlrs r(log_error_with_contextrqsb$U__%=%=$>? LL #oo66!$U  r*)rr>r?rA)r=TTNF)N)r/r0r$rQrpathlibrtypingrrrconfigr Formatterr r5r2boolLoggerrarcfloatrj Exceptionrqdefault_loggerr3r*r(r{s1 &&9G%%92w((&" E EEEd^ E  E  ^^ EP #S #W^^ #GNNse"  NN #s(^c] 0  r*