Fix on_error Callbacks (#1526)

AlonsoGuevara · web-flow · commit 329b83cf7fe7 · 2024-12-16T14:56:41.000-06:00
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])

-Original file line number
+Diff line change
@@ @@ -0,0 +1,4 @@ @@
 +{
 +  "type": "patch",
 +  "description": "Fix exception on error callbacks"
 +}
Original file line number	Diff line number	Diff line change
`@@ -276,7 +276,7 @@ async def run_pipeline(`
`276`	`276`	`await _dump_stats(context.stats, context.storage)`
`277`	`277`	`except Exception as e:`
`278`	`278`	`log.exception("error running workflow %s", last_workflow)`
`279`		`- cast("WorkflowCallbacks", callbacks).on_error(`
	`279`	`+ cast("WorkflowCallbacks", callback_chain).on_error(`
`280`	`280`	`"Error running pipeline!", e, traceback.format_exc()`
`281`	`281`	`)`
`282`	`282`	`yield PipelineRunResult(last_workflow, None, [e])`