瀏覽代碼

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 <zouzou0208@gmail.com>
tags/v0.17.1
yihong 8 月之前
父節點
當前提交
148a7e7002
沒有連結到貢獻者的電子郵件帳戶。
共有 1 個檔案被更改,包括 4 行新增1 行删除
  1. 4
    1
      graphrag/entity_resolution.py

+ 4
- 1
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]

Loading…
取消
儲存