From e794bb88346d4ad6a2c16752026b18d053b2993d Mon Sep 17 00:00:00 2001 From: Leon Luithlen Date: Thu, 14 Nov 2024 09:43:56 +0100 Subject: [PATCH] Return stripped value from get_embeddable_data if its string --- cognee/infrastructure/engine/models/DataPoint.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/cognee/infrastructure/engine/models/DataPoint.py b/cognee/infrastructure/engine/models/DataPoint.py index 222b11ad7..337306cb6 100644 --- a/cognee/infrastructure/engine/models/DataPoint.py +++ b/cognee/infrastructure/engine/models/DataPoint.py @@ -20,5 +20,8 @@ class DataPoint(BaseModel): def get_embeddable_data(self): if self._metadata and len(self._metadata["index_fields"]) > 0 \ and hasattr(self, self._metadata["index_fields"][0]): - - return getattr(self, self._metadata["index_fields"][0]) + attribute = getattr(self, self._metadata["index_fields"][0]) + if isinstance(attribute, str): + return(attribute.strip()) + else: + return (attribute)