diff --git a/packages/graphrag/graphrag/query/indexer_adapters.py b/packages/graphrag/graphrag/query/indexer_adapters.py index 7119ad842c..a6ded1f9f7 100644 --- a/packages/graphrag/graphrag/query/indexer_adapters.py +++ b/packages/graphrag/graphrag/query/indexer_adapters.py @@ -221,5 +221,5 @@ def _filter_under_community_level( ) -> pd.DataFrame: return cast( "pd.DataFrame", - df[df.level <= community_level], + df[(df.level <= community_level) | df.level.isna()], ) diff --git a/packages/graphrag/pyproject.toml b/packages/graphrag/pyproject.toml index f9fa8b5373..727feb7f75 100644 --- a/packages/graphrag/pyproject.toml +++ b/packages/graphrag/pyproject.toml @@ -46,7 +46,7 @@ dependencies = [ "graspologic-native~=1.2", "json-repair~=0.30", "networkx~=3.4", - "nltk~=3.9", + "nltk~=3.9.3", "numpy~=2.1", "pandas~=2.3", "pyarrow~=22.0",