Fix on_error Callbacks (#1526)

microsoft · Dec 16, 2024 · 329b83c · 329b83c
1 parent 9836643
commit 329b83c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/.semversioner/next-release/patch-20241216204053395000.json b/.semversioner/next-release/patch-20241216204053395000.json
@@ -0,0 +1,4 @@
+{
+  "type": "patch",
+  "description": "Fix exception on error callbacks"
+}
diff --git a/graphrag/index/run/run.py b/graphrag/index/run/run.py
@@ -276,7 +276,7 @@ async def run_pipeline(
         await _dump_stats(context.stats, context.storage)
     except Exception as e:
         log.exception("error running workflow %s", last_workflow)
-        cast("WorkflowCallbacks", callbacks).on_error(
+        cast("WorkflowCallbacks", callback_chain).on_error(
             "Error running pipeline!", e, traceback.format_exc()
         )
         yield PipelineRunResult(last_workflow, None, [e])