Skip to content

Commit a8043a6

Browse files
authored
fix: Cache Bug (#98)
1 parent 3f8507c commit a8043a6

File tree

1 file changed

+10
-3
lines changed

1 file changed

+10
-3
lines changed

nano_graphrag/_op.py

Lines changed: 10 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -786,10 +786,17 @@ async def _find_most_related_edges_from_entities(
786786
all_related_edges = await asyncio.gather(
787787
*[knowledge_graph_inst.get_node_edges(dp["entity_name"]) for dp in node_datas]
788788
)
789-
all_edges = set()
789+
790+
all_edges = []
791+
seen = set()
792+
790793
for this_edges in all_related_edges:
791-
all_edges.update([tuple(sorted(e)) for e in this_edges])
792-
all_edges = list(all_edges)
794+
for e in this_edges:
795+
sorted_edge = tuple(sorted(e))
796+
if sorted_edge not in seen:
797+
seen.add(sorted_edge)
798+
all_edges.append(sorted_edge)
799+
793800
all_edges_pack = await asyncio.gather(
794801
*[knowledge_graph_inst.get_edge(e[0], e[1]) for e in all_edges]
795802
)

0 commit comments

Comments
 (0)