yangdx commited on
Commit
df1342a
·
1 Parent(s): effc66c

Apply min_degree filter for full graph query

Browse files
Files changed (1) hide show
  1. lightrag/kg/networkx_impl.py +1 -1
lightrag/kg/networkx_impl.py CHANGED
@@ -312,7 +312,7 @@ class NetworkXStorage(BaseGraphStorage):
312
  nodes_to_keep = [
313
  node
314
  for node, degree in subgraph.degree()
315
- if (node_label != "*" and (node in start_nodes or node in direct_connected_nodes))
316
  or degree >= min_degree
317
  ]
318
  subgraph = subgraph.subgraph(nodes_to_keep)
 
312
  nodes_to_keep = [
313
  node
314
  for node, degree in subgraph.degree()
315
+ if node in start_nodes or node in direct_connected_nodes
316
  or degree >= min_degree
317
  ]
318
  subgraph = subgraph.subgraph(nodes_to_keep)