cr

run-llama · Jun 19, 2024 · 555a8d3 · 555a8d3
1 parent 6758300
commit 555a8d3
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/llama-index-core/llama_index/core/query_pipeline/components/__init__.py b/llama-index-core/llama_index/core/query_pipeline/components/__init__.py
@@ -15,6 +15,7 @@
     SelectorComponent,
 )
 from llama_index.core.query_pipeline.components.tool_runner import ToolRunnerComponent
+from llama_index.core.query_pipeline.components.stateful import StatefulFnComponent
 
 __all__ = [
     "AgentFnComponent",
@@ -28,4 +29,5 @@
     "RouterComponent",
     "SelectorComponent",
     "ToolRunnerComponent",
+    "StatefulFnComponent"
 ]