Add object and chunk base impl and test cases

src/knowledge/object/object.py

@@ -0,0 +1,79 @@
+
+# define a object type enum
+from abc import ABC, abstractmethod
+from enum import Enum
+from .object_id import ObjectID
+import hashlib
+import json
+import pickle
+
+class KnowledgeObject(ABC):
+    def __init__(self, object_type: int, desc: dict = {}, body: dict = {}):
+        self.desc = desc
+        self.body = body
+        self.object_type = object_type
+
+    def get_object_type(self):
+        return self.object_type
+
+    def object_id(self) -> ObjectID:
+        return self.calculate_id()
+    
+    def set_desc_with_key_value(self, key, value):
+        self.desc[key] = value
+
+    def get_desc_with_key(self, key):
+        return self.desc.get(key)
+
+    def get_desc(self) -> dict:
+        return self.desc
+    
+    def set_body_with_key_value(self, key, value):
+        self.body[key] = value
+
+    def get_body_with_key(self, key):
+        return self.body.get(key)
+
+    def get_body(self) -> dict:
+        return self.body
+    
+    def calculate_id(self):
+        # Convert the object_type and desc to string and compute the SHA256 hash
+        data = json.dumps({"object_type": self.object_type, "desc": self.desc})
+        sha256 = hashlib.sha256()
+        sha256.update(data.encode())
+        return ObjectID(sha256.digest())
+
+    def encode(self) -> bytes:
+        return pickle.dumps(self)
+
+    @staticmethod
+    def decode(data: bytes):
+        return pickle.loads(data)
+    
+
+        
+# define a text chunk class
+class TextChunkObject(KnowledgeObject): # pylint: disable=too-few-public-methods
+    def __init__(self, text: str):
+        super().__init__(ObjectType.TextChunk)
+        self.text = text
+
+
+# define a image class
+class ImageObject(KnowledgeObject): # pylint: disable=too-few-public-methods
+    def __init__(self, meta, path):
+        super().__init__(ObjectType.Image)
+        self.meta = meta
+        self.path = path
+
+
+# define a email class
+class EmailObject(KnowledgeObject): # pylint: disable=too-few-public-methods
+    def __init__(self, meta):
+        super().__init__(ObjectType.Email)
+        self.meta = meta
+        self.text = [ObjectID]
+        self.images = [ObjectID]
+
+