Update api/logger.py
Browse files- api/logger.py +6 -20
api/logger.py
CHANGED
|
@@ -3,32 +3,18 @@ import logging
|
|
| 3 |
def setup_logger(name):
|
| 4 |
logger = logging.getLogger(name)
|
| 5 |
if not logger.handlers:
|
| 6 |
-
logger.setLevel(logging.INFO)
|
| 7 |
formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
|
| 8 |
-
|
| 9 |
-
# Console
|
| 10 |
console_handler = logging.StreamHandler()
|
| 11 |
console_handler.setFormatter(formatter)
|
| 12 |
logger.addHandler(console_handler)
|
| 13 |
-
|
| 14 |
-
# File
|
| 15 |
# error_file_handler = logging.FileHandler('error.log')
|
| 16 |
# error_file_handler.setFormatter(formatter)
|
| 17 |
# error_file_handler.setLevel(logging.ERROR)
|
| 18 |
# logger.addHandler(error_file_handler)
|
| 19 |
-
|
| 20 |
-
return logger
|
| 21 |
|
| 22 |
-
|
| 23 |
-
|
| 24 |
-
def log_strip_prefix(model_prefix, content):
|
| 25 |
-
"""
|
| 26 |
-
Log when a model prefix is stripped from the content.
|
| 27 |
-
"""
|
| 28 |
-
logger.debug(f"Stripping prefix '{model_prefix}' from content.")
|
| 29 |
-
|
| 30 |
-
def log_model_delay(delay_seconds, model, chat_id):
|
| 31 |
-
"""
|
| 32 |
-
Log the delay introduced for specific models.
|
| 33 |
-
"""
|
| 34 |
-
logger.info(f"Introducing a delay of {delay_seconds} seconds for model '{model}' (Chat ID: {chat_id})")
|
|
|
|
| 3 |
def setup_logger(name):
|
| 4 |
logger = logging.getLogger(name)
|
| 5 |
if not logger.handlers:
|
| 6 |
+
logger.setLevel(logging.INFO) # Set to INFO to avoid DEBUG logs in production
|
| 7 |
formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
|
| 8 |
+
|
| 9 |
+
# Console handler
|
| 10 |
console_handler = logging.StreamHandler()
|
| 11 |
console_handler.setFormatter(formatter)
|
| 12 |
logger.addHandler(console_handler)
|
| 13 |
+
|
| 14 |
+
# File handler for errors (optional)
|
| 15 |
# error_file_handler = logging.FileHandler('error.log')
|
| 16 |
# error_file_handler.setFormatter(formatter)
|
| 17 |
# error_file_handler.setLevel(logging.ERROR)
|
| 18 |
# logger.addHandler(error_file_handler)
|
|
|
|
|
|
|
| 19 |
|
| 20 |
+
return logger
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|