yangdx
commited on
Commit
·
df1342a
1
Parent(s):
effc66c
Apply min_degree filter for full graph query
Browse files
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
|
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)
|