Update models/models.py

models/models.py

@@ -8,12 +8,13 @@ class Attachment:
 
 
 class Message:
-    def __init__(self, message_id: str, body: Optional[str], attachments: Optional[List[Attachment]], company: str , document_len:int):
+    def __init__(self, message_id: str, body: Optional[str], attachments: Optional[List[Attachment]], company: str , body_len:int , attachemnt_len:int):
         self.id = message_id
         self.body = body
         self.attachments = attachments
         self.company = company
         self.document_len = document_len
+        self.attachemnt_len = attachemnt_len
 
     def to_json(self):
         return {
@@ -21,5 +22,6 @@ class Message:
             "body": self.body,
             "attachments": [attachment.__dict__ for attachment in self.attachments] if self.attachments else None,
             "company": self.company,
-            "document_len" : self.document_len
+            "document_len" : self.document_len,
+            "attachemnt_len" : attachemnt_len
         }