diff --git a/.semversioner/next-release/patch-20241227225850465466.json b/.semversioner/next-release/patch-20241227225850465466.json new file mode 100644 index 0000000000..75085e8682 --- /dev/null +++ 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 index 890a06d083..1a2ce19695 100644 --- 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