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

Update api/logger.py

Browse files
Files changed (1) hide show
  1. api/logger.py +15 -1
api/logger.py CHANGED
@@ -17,4 +17,18 @@ def setup_logger(name):
17
  # error_file_handler.setLevel(logging.ERROR)
18
  # logger.addHandler(error_file_handler)
19
 
20
- return logger
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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})")