Update models/models.py

models/models.py

@@ -13,24 +13,16 @@ class Attachment:
 class Message:
     #structured_data:Optional[List] add this in the below __init__
                  
-    def __init__(self, message_id: str, body_len:int, body: Optional[str], attachments: Optional[List[Attachment]], company: str , high_level_company_type:str,structured_data:Optional[List]):
+    def __init__(self, message_id: str ,company: str ,structured_data:Optional[List]):
         self.id = message_id
-        self.body_len = body_len
-        self.body = body
-        self.attachments = attachments
         self.company = company
-        self.high_level_company_type = high_level_company_type
         self.structured_data = structured_data
         
 
     def to_json(self):
         return {
             "id": self.id,
-            "body_len" : self.body_len,
-            "body": self.body,
-            "attachments": [attachment.__dict__ for attachment in self.attachments] if self.attachments else None,
             "company": self.company,
-            "high_level_company_type":self.high_level_company_type,
             "structured_data": self.structured_data if self.structured_data else None
         }