Update models/models.py

models/models.py

@@ -1,3 +1,48 @@
+Hugging Face's logo
+Hugging Face
+Search models, datasets, users...
+Models
+Datasets
+Spaces
+Posts
+Docs
+Pricing
+
+
+
+Spaces:
+
+Hushh
+/
+hushh-luxury-app
+
+
+like
+0
+
+App
+Files
+Community
+Settings
+hushh-luxury-app
+/
+models
+/
+models.py
+
+Omkar008's picture
+Omkar008
+Update models/models.py
+760e491
+VERIFIED
+19 days ago
+raw
+history
+blame
+edit
+delete
+No virus
+1.28 kB
 import json
 from typing import Optional, List, Dict
 
@@ -11,12 +56,16 @@ class Attachment:
 
 
 class Message:
-    def __init__(self, message_id: str, body_len:int, body: Optional[str], attachments: Optional[List[Attachment]], company: str ):
+    #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]):
         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):
@@ -25,5 +74,8 @@ class Message:
             "body_len" : self.body_len,
             "body": self.body,
             "attachments": [attachment.__dict__ for attachment in self.attachments] if self.attachments else None,
-            "company": self.company
+            "company": self.company,
+            "high_level_company_type":self.high_level_company_type,
+            "structured_data": self.structured_data if self.structured_data else None
         }
+