Niansuh commited on
Commit
ad5b376
·
verified ·
1 Parent(s): 55c6c36

Update api/logger.py

Browse files
Files changed (1) hide show
  1. api/logger.py +20 -54
api/logger.py CHANGED
@@ -1,54 +1,20 @@
1
- import logging
2
-
3
- # Setup logger with a consistent format
4
- def setup_logger(name):
5
- logger = logging.getLogger(name)
6
- if not logger.handlers:
7
- logger.setLevel(logging.INFO)
8
- formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
9
-
10
- # Console handler
11
- console_handler = logging.StreamHandler()
12
- console_handler.setFormatter(formatter)
13
- logger.addHandler(console_handler)
14
-
15
- # File Handler - Error Level
16
- # error_file_handler = logging.FileHandler('error.log')
17
- # error_file_handler.setFormatter(formatter)
18
- # error_file_handler.setLevel(logging.ERROR)
19
- # logger.addHandler(error_file_handler)
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