Refactor of the tasks

topoteretes · Aug 8, 2024 · 156c7be · 156c7be
1 parent 85160da
commit 156c7be
Showing 1 changed file with 4 additions and 0 deletions.
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
@@ -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 = []