From 0575f7764006474730f877af8bfe83a0b1b4861a Mon Sep 17 00:00:00 2001 From: venti <1308199824@qq.com> Date: Sat, 30 May 2026 16:42:57 +0800 Subject: [PATCH] fix: preserve entities without community assignment in community level filter (fixes #2348) --- packages/graphrag/graphrag/query/indexer_adapters.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/packages/graphrag/graphrag/query/indexer_adapters.py b/packages/graphrag/graphrag/query/indexer_adapters.py index 7119ad842c..4db0de2e78 100644 --- a/packages/graphrag/graphrag/query/indexer_adapters.py +++ b/packages/graphrag/graphrag/query/indexer_adapters.py @@ -219,7 +219,16 @@ def embed_community_reports( def _filter_under_community_level( df: pd.DataFrame, community_level: int ) -> pd.DataFrame: + nan_count = df.level.isna().sum() + if nan_count > 0: + orphan_pct = nan_count / len(df) * 100 + if orphan_pct > 10: + logger.warning( + "%.0f%% of entities have no community assignment. " + "Consider checking your community detection settings.", + orphan_pct, + ) return cast( "pd.DataFrame", - df[df.level <= community_level], + df[(df.level <= community_level) | df.level.isna()], )