Spaces:
Sleeping
Sleeping
File size: 1,644 Bytes
f3dfbd4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 |
"""
Logging configuration for the CRE Chatbot application.
"""
import logging
import os
from logging.handlers import RotatingFileHandler
from .config import LOG_LEVEL, LOG_FORMAT, LOG_FILE
def setup_logging():
"""Set up logging configuration for the application."""
# Create logs directory if it doesn't exist
os.makedirs('logs', exist_ok=True)
# Set up root logger
logger = logging.getLogger()
logger.setLevel(LOG_LEVEL)
# Create formatters and handlers
formatter = logging.Formatter(LOG_FORMAT)
# Console Handler
console_handler = logging.StreamHandler()
console_handler.setFormatter(formatter)
logger.addHandler(console_handler)
# File Handler
file_handler = RotatingFileHandler(
LOG_FILE,
maxBytes=10485760, # 10MB
backupCount=5
)
file_handler.setFormatter(formatter)
logger.addHandler(file_handler)
# Create separate loggers for different components
loggers = {
'api': setup_component_logger('api'),
'pdf': setup_component_logger('pdf'),
'rag': setup_component_logger('rag'),
'app': setup_component_logger('app')
}
return loggers
def setup_component_logger(name):
"""Set up a logger for a specific component."""
logger = logging.getLogger(name)
logger.setLevel(LOG_LEVEL)
# Create component-specific log file
handler = RotatingFileHandler(
f'logs/{name}.log',
maxBytes=10485760, # 10MB
backupCount=3
)
handler.setFormatter(logging.Formatter(LOG_FORMAT))
logger.addHandler(handler)
return logger
|