Fix/gleanings loop (#1564)

AlonsoGuevara · web-flow · commit 5258bc5f4fa1 · 2024-12-30T12:57:33.000-06:00
* Fix gleaning output parsing

* Semver
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

-Original file line number
+Diff line change
@@ @@ -0,0 +1,4 @@ @@
 +{
 +  "type": "patch",
 +  "description": "Fix gleanings loop check"
 +}
Original file line number	Diff line number	Diff line change
`@@ -179,7 +179,8 @@ async def _process_document(`
`179`	`179`	`history=response.history,`
`180`	`180`	`model_parameters=self._loop_args,`
`181`	`181`	`)`
`182`		`- if response.output != "YES":`
	`182`	`+`
	`183`	`+ if response.output.content != "YES":`
`183`	`184`	`break`
`184`	`185`
`185`	`186`	`return results`