api/logger.py

import logging

def setup_logger(name):
    logger = logging.getLogger(name)
    if not logger.handlers:
        logger.setLevel(logging.INFO)
        formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
        
        # Console processor
        console_handler = logging.StreamHandler()
        console_handler.setFormatter(formatter)
        logger.addHandler(console_handler)
        
        # File Handler - Error Level
        # error_file_handler = logging.FileHandler('error.log')
        # error_file_handler.setFormatter(formatter)
        # error_file_handler.setLevel(logging.ERROR)
        # logger.addHandler(error_file_handler)
    
    return logger

# Log functions to structure specific logs in utils.py

def log_strip_prefix(model_prefix, content):
    """
    Log when a model prefix is stripped from the content.
    """
    logger.debug(f"Stripping prefix '{model_prefix}' from content.")

def log_model_delay(delay_seconds, model, chat_id):
    """
    Log the delay introduced for specific models.
    """
    logger.info(f"Introducing a delay of {delay_seconds} seconds for model '{model}' (Chat ID: {chat_id})")