From 555a8d351202bd289a56a439ea0b495e0462d2a6 Mon Sep 17 00:00:00 2001 From: Jerry Liu Date: Wed, 19 Jun 2024 01:12:40 -0700 Subject: [PATCH] cr --- .../llama_index/core/query_pipeline/components/__init__.py | 2 ++ 1 file changed, 2 insertions(+) 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" ]