From 156c7bec689569d39db96950cb601b17d01ba90d Mon Sep 17 00:00:00 2001 From: Vasilije <8619304+Vasilije1990@users.noreply.github.com> Date: Thu, 8 Aug 2024 13:47:03 +0200 Subject: [PATCH] Refactor of the tasks --- .../source_documents_to_chunks/source_documents_to_chunks.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/cognee/tasks/source_documents_to_chunks/source_documents_to_chunks.py b/cognee/tasks/source_documents_to_chunks/source_documents_to_chunks.py index 116f020a6..419200e07 100644 --- a/cognee/tasks/source_documents_to_chunks/source_documents_to_chunks.py +++ b/cognee/tasks/source_documents_to_chunks/source_documents_to_chunks.py @@ -5,6 +5,10 @@ async def source_documents_to_chunks(documents: list[Document], parent_node_id: str = None, user:str=None, user_permissions:str=None): graph_engine = await get_graph_engine() + if parent_node_id is None: + documents, parent_node_id = documents + + nodes = [] edges = []