Niansuh commited on
Commit
eeba311
·
verified ·
1 Parent(s): 668a15e

Update api/logger.py

Browse files
Files changed (1) hide show
  1. 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 processor
10
  console_handler = logging.StreamHandler()
11
  console_handler.setFormatter(formatter)
12
  logger.addHandler(console_handler)
13
-
14
- # File Handler - Error Level
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
- # Log functions to structure specific logs in utils.py
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