From 1c0eeaf917356bfa011c99ce14ec020e2e71e0a3 Mon Sep 17 00:00:00 2001 From: Igor Ilic Date: Wed, 30 Jul 2025 12:00:09 +0200 Subject: [PATCH] fix: Return add_data_points --- cognee/tasks/graph/extract_graph_from_data.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/cognee/tasks/graph/extract_graph_from_data.py b/cognee/tasks/graph/extract_graph_from_data.py index 2590d361b..9601568c8 100644 --- a/cognee/tasks/graph/extract_graph_from_data.py +++ b/cognee/tasks/graph/extract_graph_from_data.py @@ -3,6 +3,7 @@ from typing import Type, List from pydantic import BaseModel from cognee.infrastructure.databases.graph import get_graph_engine +from cognee.tasks.storage.add_data_points import add_data_points from cognee.modules.ontology.rdf_xml.OntologyResolver import OntologyResolver from cognee.modules.chunking.models.DocumentChunk import DocumentChunk from cognee.base_config import get_base_config @@ -51,7 +52,7 @@ async def integrate_chunk_graphs( ) if len(graph_nodes) > 0: - await graph_engine.add_nodes(graph_nodes) + await add_data_points(graph_nodes) if len(graph_edges) > 0: await graph_engine.add_edges(graph_edges)