Update api/logger.py
Browse files- 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.")
|