From 4e2a9b19940037b9a360cf752eeb7c599be95558 Mon Sep 17 00:00:00 2001 From: yangdx Date: Wed, 5 Mar 2025 13:21:17 +0800 Subject: [PATCH] Apply min_degree filter for full graph query --- lightrag/kg/networkx_impl.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lightrag/kg/networkx_impl.py b/lightrag/kg/networkx_impl.py index 60122166..4084993d 100644 --- a/lightrag/kg/networkx_impl.py +++ b/lightrag/kg/networkx_impl.py @@ -312,7 +312,7 @@ class NetworkXStorage(BaseGraphStorage): nodes_to_keep = [ node for node, degree in subgraph.degree() - if (node_label != "*" and (node in start_nodes or node in direct_connected_nodes)) + if node in start_nodes or node in direct_connected_nodes or degree >= min_degree ] subgraph = subgraph.subgraph(nodes_to_keep)