Undo make notifier private, not needed

Author: stellasia

src/neo4j_graphrag/experimental/pipeline/orchestrator.py

@@ -85,9 +85,7 @@ async def run_task(self, task: TaskPipelineNode, data: dict[str, Any]) -> None:
             run_id=self.run_id,
             task_name=task.name,
         )
-        context = RunContext(
-            run_id=self.run_id, task_name=task.name, _notifier=notifier
-        )
+        context = RunContext(run_id=self.run_id, task_name=task.name, notifier=notifier)
         res = await task.run(context, inputs)
         await self.set_task_status(task.name, RunStatus.DONE)
         await self.event_notifier.notify_task_finished(self.run_id, task.name, res)

src/neo4j_graphrag/experimental/pipeline/types/context.py

@@ -28,10 +28,10 @@ class RunContext(BaseModel):
 
     run_id: str
     task_name: str
-    _notifier: Optional[TaskProgressCallbackProtocol] = None
+    notifier: Optional[TaskProgressCallbackProtocol] = None
 
     model_config = ConfigDict(arbitrary_types_allowed=True)
 
     async def notify(self, message: str, data: dict[str, Any]) -> None:
-        if self._notifier:
-            await self._notifier(message=message, data=data)
+        if self.notifier:
+            await self.notifier(message=message, data=data)

tests/unit/experimental/pipeline/test_component.py

@@ -68,7 +68,7 @@ async def test_component_run_with_context() -> None:
     c = ComponentMultiplyWithContext()
     notifier_mock = AsyncMock()
     result = await c.run_with_context(
-        RunContext(run_id="run_id", task_name="task_name", _notifier=notifier_mock),
+        RunContext(run_id="run_id", task_name="task_name", notifier=notifier_mock),
         number1=1,
         number2=2,
     )