yangdx commited on
Commit
36947fe
·
1 Parent(s): b101cb0

Ensure unique edge IDs for NetworkX

Browse files
Files changed (1) hide show
  1. lightrag/kg/networkx_impl.py +3 -0
lightrag/kg/networkx_impl.py CHANGED
@@ -373,6 +373,9 @@ class NetworkXStorage(BaseGraphStorage):
373
  # Add edges to result
374
  for edge in subgraph.edges():
375
  source, target = edge
 
 
 
376
  edge_id = f"{source}-{target}"
377
  if edge_id in seen_edges:
378
  continue
 
373
  # Add edges to result
374
  for edge in subgraph.edges():
375
  source, target = edge
376
+ # Esure unique edge_id for undirect graph
377
+ if source > target:
378
+ source, target = target, source
379
  edge_id = f"{source}-{target}"
380
  if edge_id in seen_edges:
381
  continue