chatgraph 1.2.5__tar.gz → 1.2.6__tar.gz

{chatgraph-1.2.5 → chatgraph-1.2.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: chatgraph
-Version: 1.2.5
+Version: 1.2.6
 Summary: A user-friendly chatbot library
 Project-URL: Homepage, https://github.com/irissonnlima/chatgraph
 Project-URL: Repository, https://github.com/irissonnlima/chatgraph

{chatgraph-1.2.5 → chatgraph-1.2.6}/chatgraph/logger/user_logger.py

@@ -9,6 +9,14 @@ _ENV_LEVEL = os.getenv("CHATGRAPH_LOG_LEVEL", "").upper()
 _DEFAULT_LEVEL: int = getattr(logging, _ENV_LEVEL, logging.INFO)
 
 
+class ChatIDFilter(logging.Filter):
+    def filter(self, record: logging.LogRecord) -> bool:
+        chat_id = getattr(record, 'chat_id', None)
+        if chat_id:
+            record.msg = f'[ChatID: {chat_id}] - {record.msg}'
+        return True
+
+
 class UserLoggerManager:
     _loggers: dict[str, logging.Logger] = {}
     _level: int = _DEFAULT_LEVEL
@@ -48,6 +56,7 @@ class UserLoggerManager:
             stream_handler.setLevel(cls._level)
             stream_handler.setFormatter(logging.Formatter(LOG_FORMAT))
             logger.addHandler(stream_handler)
+            logger.addFilter(ChatIDFilter())
 
         cls._loggers[key] = logger
         return logger

{chatgraph-1.2.5 → chatgraph-1.2.6}/chatgraph/types/usercall.py

@@ -43,8 +43,13 @@ class UserCall:  # noqa: PLR0904
         self.__content_message = self.__message.text_message.detail
 
     @property
-    def logger(self) -> logging.Logger:
-        return UserLoggerManager.get_user_logger(self.user_id, self.company_id)
+    def logger(self) -> logging.LoggerAdapter:
+        base_logger = UserLoggerManager.get_user_logger(
+            self.user_id, self.company_id
+        )
+        return logging.LoggerAdapter(
+            base_logger, {'chat_id': str(self.__user_state.chat_id)}
+        )
 
     def __str__(self):
         return (

{chatgraph-1.2.5 → chatgraph-1.2.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "chatgraph"
-version = "1.2.5"
+version = "1.2.6"
 description = "A user-friendly chatbot library"
 authors = [
     {name = "Irisson N. Lima", email = "irisson.lima@verdecard.com.br"}