Update api/logger.py

api/logger.py

@@ -1,54 +1,56 @@
-import logging
-
-# Setup logger with a consistent format
-def setup_logger(name):
-    logger = logging.getLogger(name)
-    if not logger.handlers:
-        logger.setLevel(logging.INFO)
-        formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
-        
-        # Console handler
-        console_handler = logging.StreamHandler()
-        console_handler.setFormatter(formatter)
-        logger.addHandler(console_handler)
-        
-        # File Handler - Error Level
-        # error_file_handler = logging.FileHandler('error.log')
-        # error_file_handler.setFormatter(formatter)
-        # error_file_handler.setLevel(logging.ERROR)
-        # logger.addHandler(error_file_handler)
-    
-    return logger
-
-logger = setup_logger(__name__)
-
-# Log functions to structure specific logs in utils.py
-def log_generated_chat_id_with_referer(chat_id, model, referer_url):
-    """
-    Log the generated Chat ID with model and referer URL if it exists.
-    """
-    logger.info(f"Generated Chat ID: {chat_id} - Model: {model} - URL: {referer_url}")
-
-def log_model_delay(delay_seconds, model, chat_id):
-    """
-    Log the delay introduced for specific models.
-    """
-    logger.info(f"Introducing a delay of {delay_seconds} seconds for model '{model}' (Chat ID: {chat_id})")
-
-def log_http_error(error, chat_id):
-    """
-    Log HTTP errors encountered during requests.
-    """
-    logger.error(f"HTTP error occurred for Chat ID {chat_id}: {error}")
-
-def log_request_error(error, chat_id):
-    """
-    Log request errors unrelated to HTTP status.
-    """
-    logger.error(f"Request error occurred for Chat ID {chat_id}: {error}")
-
-def log_strip_prefix(model_prefix, content):
-    """
-    Log when a model prefix is stripped from the content.
-    """
-    logger.debug(f"Stripping prefix '{model_prefix}' from content.")
+# api/logger.py
+
+import logging
+
+# Setup logger with a consistent format
+def setup_logger(name):
+    logger = logging.getLogger(name)
+    if not logger.handlers:
+        logger.setLevel(logging.INFO)
+        formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
+        
+        # Console handler
+        console_handler = logging.StreamHandler()
+        console_handler.setFormatter(formatter)
+        logger.addHandler(console_handler)
+        
+        # File Handler - Uncomment if you want to log errors to a file
+        # error_file_handler = logging.FileHandler('error.log')
+        # error_file_handler.setFormatter(formatter)
+        # error_file_handler.setLevel(logging.ERROR)
+        # logger.addHandler(error_file_handler)
+    
+    return logger
+
+logger = setup_logger(__name__)
+
+# Log functions to structure specific logs in utils.py
+def log_generated_chat_id_with_referer(chat_id, model, referer_url):
+    """
+    Log the generated Chat ID with model and referer URL if it exists.
+    """
+    logger.info(f"Generated Chat ID: {chat_id} - Model: {model} - URL: {referer_url}")
+
+def log_model_delay(delay_seconds, model, chat_id):
+    """
+    Log the delay introduced for specific models.
+    """
+    logger.info(f"Introducing a delay of {delay_seconds} seconds for model '{model}' (Chat ID: {chat_id})")
+
+def log_http_error(error, chat_id):
+    """
+    Log HTTP errors encountered during requests.
+    """
+    logger.error(f"HTTP error occurred for Chat ID {chat_id}: {error}")
+
+def log_request_error(error, chat_id):
+    """
+    Log request errors unrelated to HTTP status.
+    """
+    logger.error(f"Request error occurred for Chat ID {chat_id}: {error}")
+
+def log_strip_prefix(model_prefix, content):
+    """
+    Log when a model prefix is stripped from the content.
+    """
+    logger.debug(f"Stripping prefix '{model_prefix}' from content.")