diff --git a/graphrag/entity_resolution.py b/graphrag/entity_resolution.py index 12b2bc9a6..da1c58c33 100644 --- a/graphrag/entity_resolution.py +++ b/graphrag/entity_resolution.py @@ -104,11 +104,14 @@ class EntityResolution(Extractor): connect_graph = nx.Graph() removed_entities = [] connect_graph.add_edges_from(resolution_result) + # for issue #5600 + all_relationships_data = [] + for sub_connect_graph in nx.connected_components(connect_graph): sub_connect_graph = connect_graph.subgraph(sub_connect_graph) remove_nodes = list(sub_connect_graph.nodes) keep_node = remove_nodes.pop() - self._merge_nodes(keep_node, self._get_entity_(remove_nodes)) + await self._merge_nodes(keep_node, self._get_entity_(remove_nodes), all_relationships_data=all_relationships_data) for remove_node in remove_nodes: removed_entities.append(remove_node) remove_node_neighbors = graph[remove_node]