From a5860700a7255ecbf702411f136d8efe2651e5d2 Mon Sep 17 00:00:00 2001 From: Leon Luithlen Date: Fri, 15 Nov 2024 14:00:59 +0100 Subject: [PATCH] Remove include_root parameter --- cognee/modules/graph/utils/get_graph_from_model.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/cognee/modules/graph/utils/get_graph_from_model.py b/cognee/modules/graph/utils/get_graph_from_model.py index d1e14c878..def6e4840 100644 --- a/cognee/modules/graph/utils/get_graph_from_model.py +++ b/cognee/modules/graph/utils/get_graph_from_model.py @@ -4,7 +4,7 @@ from cognee.modules.storage.utils import copy_model def get_graph_from_model( - data_point: DataPoint, include_root=True, added_nodes=None, added_edges=None + data_point: DataPoint, added_nodes=None, added_edges=None ): if not added_nodes: @@ -53,8 +53,7 @@ def get_graph_from_model( exclude_fields=excluded_properties, ) - if include_root: - nodes.append(SimpleDataPointModel(**data_point_properties)) + nodes.append(SimpleDataPointModel(**data_point_properties)) return nodes, edges @@ -64,7 +63,7 @@ def add_nodes_and_edges( ): property_nodes, property_edges = get_graph_from_model( - field_value, True, added_nodes, added_edges + field_value, added_nodes, added_edges ) for node in property_nodes: