fix: adds back the ids to the nodes after node_link_graph

topoteretes · Dec 4, 2024 · f444ae2 · f444ae2
1 parent 080143c
commit f444ae2
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/cognee/infrastructure/databases/graph/networkx/adapter.py b/cognee/infrastructure/databases/graph/networkx/adapter.py
@@ -274,6 +274,9 @@ async def load_graph_from_file(self, file_path: str = None):
                             edge["updated_at"] = datetime.strptime(edge["updated_at"], "%Y-%m-%dT%H:%M:%S.%f%z")
 
                     self.graph = nx.readwrite.json_graph.node_link_graph(graph_data)
+
+                    for node_id, node_data in self.graph.nodes(data=True):
+                        node_data['id'] = node_id
             else:
                 # Log that the file does not exist and an empty graph is initialized
                 logger.warning("File %s not found. Initializing an empty graph.", file_path)