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
index 7a1d1d2493d38..a5523b251d955 100644
--- 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"
 ]