workflow docs nit (#16618)

cr
run-llama · Oct 21, 2024 · d9a219e · d9a219e
1 parent 00fe1d1
commit d9a219e
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/docs/docs/module_guides/workflow/index.md b/docs/docs/module_guides/workflow/index.md
@@ -360,6 +360,9 @@ Since workflows are so flexible, there are many possible ways to implement human
 The easiest way to implement a human-in-the-loop is to use the `InputRequiredEvent` and `HumanResponseEvent` events during event streaming.
 
 ```python
+from llama_index.core.workflow import InputRequiredEvent, HumanResponseEvent
+
+
 class HumanInTheLoopWorkflow(Workflow):
     @step
     async def step1(self, ev: StartEvent) -> InputRequiredEvent:

diff --git a/llama-index-core/llama_index/core/workflow/__init__.py b/llama-index-core/llama_index/core/workflow/__init__.py
@@ -9,7 +9,13 @@
     WorkflowTimeoutError,
     WorkflowValidationError,
 )
-from llama_index.core.workflow.events import Event, StartEvent, StopEvent
+from llama_index.core.workflow.events import (
+    Event,
+    StartEvent,
+    StopEvent,
+    InputRequiredEvent,
+    HumanResponseEvent,
+)
 from llama_index.core.workflow.workflow import Workflow
 from llama_index.core.workflow.context import Context
 
@@ -26,4 +32,6 @@
     "draw_most_recent_execution",
     "step",
     "Context",
+    "InputRequiredEvent",
+    "HumanResponseEvent",
 ]