From 440a8cc70c81a44e640ec663fdef9f8ba7f7d62c Mon Sep 17 00:00:00 2001
From: venti <1308199824@qq.com>
Date: Sat, 30 May 2026 16:39:34 +0800
Subject: [PATCH 1/2] fix: preserve NaN-level entities in
 _filter_under_community_level (fixes #2348)

---
 packages/graphrag/graphrag/query/indexer_adapters.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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()],
     )

From 40986120ab810d4c41372a40215a920c4ef7c7fe Mon Sep 17 00:00:00 2001
From: venti <1308199824@qq.com>
Date: Sat, 30 May 2026 20:55:49 +0800
Subject: [PATCH 2/2] fix: add ModelType enum back to graphrag.config.enums for
 backward compatibility (fixes #2356)

---
 packages/graphrag/graphrag/config/enums.py | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/packages/graphrag/graphrag/config/enums.py b/packages/graphrag/graphrag/config/enums.py
index 5084f2154b..3238d4b8c8 100644
--- a/packages/graphrag/graphrag/config/enums.py
+++ b/packages/graphrag/graphrag/config/enums.py
@@ -8,6 +8,19 @@
 from enum import Enum
 
 
+class ModelType(str, Enum):
+    """The type of language model to use (deprecated: kept for backward compatibility)."""
+
+    Chat = "chat"
+    """Chat-based language model."""
+    Embedding = "embedding"
+    """Embedding-based language model."""
+
+    def __repr__(self):
+        """Get a string representation."""
+        return f'"{self.value}"'
+
+
 class ReportingType(str, Enum):
     """The reporting configuration type for the pipeline."""