Update api/logger.py

api/logger.py

@@ -1,5 +1,6 @@
 import logging
 
+# Setup logger with a consistent format
 def setup_logger(name):
     logger = logging.getLogger(name)
     if not logger.handlers:
@@ -21,18 +22,33 @@ def setup_logger(name):
 
 logger = setup_logger(__name__)
 
-# New functions for logging different scenarios
-def log_generated_chat_id(chat_id, chat_url):
-    logger.info(f"Generated Chat ID: {chat_id} - URL: {chat_url}")
+# 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, chat_id, model):
+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(chat_id, error):
+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(chat_id, error):
-    logger.error(f"Error occurred during request 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):
+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.")