Ensure unique edge IDs for NetworkX

2025-03-18 00:00:56 +08:00
parent 3bf9dbe4fb
commit c641a6290c
1 changed files with 3 additions and 0 deletions
--- a/lightrag/kg/networkx_impl.py
+++ b/lightrag/kg/networkx_impl.py
@@ -373,6 +373,9 @@ class NetworkXStorage(BaseGraphStorage):
        # Add edges to result
        for edge in subgraph.edges():
            source, target = edge
+            # Esure unique edge_id for undirect graph
+            if source > target:
+                source, target = target, source
            edge_id = f"{source}-{target}"
            if edge_id in seen_edges:
                continue