gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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