gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Remove added_nodes and added_edges default dicts

Browse source
This commit is contained in:
Leon Luithlen 2024-11-15 10:27:06 +01:00
parent b858f0b06b
commit 628f192b8d
1 changed files with 7 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
cognee/modules/graph/utils/get_graph_from_model.py
View file

@ -2,7 +2,13 @@ from datetime import datetime, timezone
from cognee.infrastructure.engine import DataPoint from cognee.infrastructure.engine import DataPoint
from cognee.modules.storage.utils import copy_model from cognee.modules.storage.utils import copy_model
def get_graph_from_model(data_point: DataPoint, include_root = True, added_nodes = {}, added_edges = {}): def get_graph_from_model(data_point: DataPoint, include_root = True, added_nodes = None, added_edges = None):
if not added_nodes:
added_nodes = {}
if not added_edges:
added_edges = {}
nodes = [] nodes = []
edges = [] edges = []