Update api/logger.py
Browse files- api/logger.py +20 -54
api/logger.py
CHANGED
@@ -1,54 +1,20 @@
|
|
1 |
-
import logging
|
2 |
-
|
3 |
-
|
4 |
-
|
5 |
-
|
6 |
-
|
7 |
-
|
8 |
-
|
9 |
-
|
10 |
-
|
11 |
-
console_handler
|
12 |
-
|
13 |
-
|
14 |
-
|
15 |
-
#
|
16 |
-
# error_file_handler
|
17 |
-
# error_file_handler.
|
18 |
-
#
|
19 |
-
|
20 |
-
|
21 |
-
return logger
|
22 |
-
|
23 |
-
logger = setup_logger(__name__)
|
24 |
-
|
25 |
-
# Log functions to structure specific logs in utils.py
|
26 |
-
def log_generated_chat_id_with_referer(chat_id, model, referer_url):
|
27 |
-
"""
|
28 |
-
Log the generated Chat ID with model and referer URL if it exists.
|
29 |
-
"""
|
30 |
-
logger.info(f"Generated Chat ID: {chat_id} - Model: {model} - URL: {referer_url}")
|
31 |
-
|
32 |
-
def log_model_delay(delay_seconds, model, chat_id):
|
33 |
-
"""
|
34 |
-
Log the delay introduced for specific models.
|
35 |
-
"""
|
36 |
-
logger.info(f"Introducing a delay of {delay_seconds} seconds for model '{model}' (Chat ID: {chat_id})")
|
37 |
-
|
38 |
-
def log_http_error(error, chat_id):
|
39 |
-
"""
|
40 |
-
Log HTTP errors encountered during requests.
|
41 |
-
"""
|
42 |
-
logger.error(f"HTTP error occurred for Chat ID {chat_id}: {error}")
|
43 |
-
|
44 |
-
def log_request_error(error, chat_id):
|
45 |
-
"""
|
46 |
-
Log request errors unrelated to HTTP status.
|
47 |
-
"""
|
48 |
-
logger.error(f"Request error occurred for Chat ID {chat_id}: {error}")
|
49 |
-
|
50 |
-
def log_strip_prefix(model_prefix, content):
|
51 |
-
"""
|
52 |
-
Log when a model prefix is stripped from the content.
|
53 |
-
"""
|
54 |
-
logger.debug(f"Stripping prefix '{model_prefix}' from content.")
|
|
|
1 |
+
import logging
|
2 |
+
|
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
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|