gmakstutis/LightRAG
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Apply min_degree filter for full graph query

Browse source
This commit is contained in:
yangdx 2025-03-05 13:21:17 +08:00
parent ca9e958cad
commit 4e2a9b1994
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
lightrag/kg/networkx_impl.py
View file

@ -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)