Niansuh commited on
Commit
71425a9
·
verified ·
1 Parent(s): 01607ce

Update api/logger.py

Browse files
Files changed (1) hide show
  1. api/logger.py +19 -1
api/logger.py CHANGED
@@ -17,4 +17,22 @@ 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
+ logger = setup_logger(__name__)
23
+
24
+ # New functions for logging different scenarios
25
+ def log_generated_chat_id(chat_id, chat_url):
26
+ logger.info(f"Generated Chat ID: {chat_id} - URL: {chat_url}")
27
+
28
+ def log_model_delay(delay_seconds, chat_id, model):
29
+ logger.info(f"Introducing a delay of {delay_seconds} seconds for model '{model}' (Chat ID: {chat_id})")
30
+
31
+ def log_http_error(chat_id, error):
32
+ logger.error(f"HTTP error occurred for Chat ID {chat_id}: {error}")
33
+
34
+ def log_request_error(chat_id, error):
35
+ logger.error(f"Error occurred during request for Chat ID {chat_id}: {error}")
36
+
37
+ def log_strip_prefix(model_prefix):
38
+ logger.debug(f"Stripping prefix '{model_prefix}' from content.")