diff --git a/graphrag/index/run/utils.py b/graphrag/index/run/utils.py index 52b1f0bd31..85e7789aca 100644 --- a/graphrag/index/run/utils.py +++ b/graphrag/index/run/utils.py @@ -43,8 +43,8 @@ def create_callback_chain( ) -> WorkflowCallbacks: """Create a callback manager that encompasses multiple callbacks.""" manager = WorkflowCallbacksManager() - for callback in callbacks or []: - manager.register(callback) + if callbacks: + manager._callbacks.extend(callbacks) return manager