diff --git a/cognee/infrastructure/engine/models/DataPoint.py b/cognee/infrastructure/engine/models/DataPoint.py index 986b13a0e..9658670dd 100644 --- a/cognee/infrastructure/engine/models/DataPoint.py +++ b/cognee/infrastructure/engine/models/DataPoint.py @@ -80,11 +80,7 @@ class DataPoint(BaseModel): """Serialize the instance to pickle-compatible bytes.""" return pickle.dumps(self.dict()) - @classmethod - def from_pickle(self, pickled_data: bytes): - """Deserialize the instance from pickled bytes.""" - data = pickle.loads(pickled_data) - return self(**data) + # The unsafe from_pickle method is removed due to security concerns (CWE-502). def to_dict(self, **kwargs) -> Dict[str, Any]: """Serialize model to a dictionary.""" @@ -93,4 +89,4 @@ class DataPoint(BaseModel): @classmethod def from_dict(cls, data: Dict[str, Any]) -> "DataPoint": """Deserialize model from a dictionary.""" - return cls.model_validate(data) + return cls.model_validate(data) \ No newline at end of file