Fix/gleanings loop (#1564)

* Fix gleaning output parsing * Semver
microsoft · Dec 30, 2024 · 5258bc5 · 5258bc5
1 parent a2647da
commit 5258bc5
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/.semversioner/next-release/patch-20241227225850465466.json b/.semversioner/next-release/patch-20241227225850465466.json
@@ -0,0 +1,4 @@
+{
+  "type": "patch",
+  "description": "Fix gleanings loop check"
+}
diff --git a/graphrag/index/operations/extract_entities/graph_extractor.py b/graphrag/index/operations/extract_entities/graph_extractor.py
@@ -179,7 +179,8 @@ async def _process_document(
                 history=response.history,
                 model_parameters=self._loop_args,
             )
-            if response.output != "YES":
+
+            if response.output.content != "YES":
                 break
 
         return results