Update api/logger.py

api/logger.py

@@ -2,11 +2,11 @@ import logging
 
 class ChatIDLoggerAdapter(logging.LoggerAdapter):
     def process(self, msg, kwargs):
-        # Include the chat_id in the log message if provided
-        chat_id = self.extra.get('chat_id', 'N/A')
+        # Check if 'chat_id' is passed in kwargs; otherwise, use 'N/A'
+        chat_id = kwargs.pop('chat_id', 'N/A')
         return f"[Chat ID: {chat_id}] {msg}", kwargs
 
-def setup_logger(name, chat_id=None):
+def setup_logger(name):
     logger = logging.getLogger(name)
     if not logger.handlers:
         logger.setLevel(logging.INFO)
@@ -17,5 +17,5 @@ def setup_logger(name, chat_id=None):
         console_handler.setFormatter(formatter)
         logger.addHandler(console_handler)
     
-    # Ensure we return a LoggerAdapter with or without chat_id
-    return ChatIDLoggerAdapter(logger, {'chat_id': chat_id})
+    # Return the LoggerAdapter without a fixed chat_id
+    return ChatIDLoggerAdapter(logger, {})