From 148a7e7002c0ef9cf227b8f704126afb4cb28b69 Mon Sep 17 00:00:00 2001 From: yihong Date: Wed, 5 Mar 2025 11:10:04 +0800 Subject: [PATCH] fix: issue #5600 (#5620) ### What problem does this PR solve? close issue #5600 ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue) Signed-off-by: yihong0618 --- graphrag/entity_resolution.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) 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]