diff --git a/llama-index-core/llama_index/core/query_pipeline/__init__.py b/llama-index-core/llama_index/core/query_pipeline/__init__.py index a82c409cfa499b..109025f54058d1 100644 --- a/llama-index-core/llama_index/core/query_pipeline/__init__.py +++ b/llama-index-core/llama_index/core/query_pipeline/__init__.py @@ -1,4 +1,5 @@ """Init file.""" + from llama_index.core.query_pipeline.components.agent import ( AgentFnComponent, AgentInputComponent, @@ -6,7 +7,10 @@ QueryComponent, ) from llama_index.core.query_pipeline.components.argpacks import ArgPackComponent -from llama_index.core.query_pipeline.components.function import FnComponent +from llama_index.core.query_pipeline.components.function import ( + FnComponent, + FunctionComponent, +) from llama_index.core.query_pipeline.components.input import InputComponent from llama_index.core.query_pipeline.components.router import RouterComponent from llama_index.core.query_pipeline.components.tool_runner import ToolRunnerComponent @@ -25,6 +29,7 @@ "AgentInputComponent", "ArgPackComponent", "FnComponent", + "FunctionComponent", "InputComponent", "RouterComponent", "ToolRunnerComponent", diff --git a/llama-index-core/llama_index/core/query_pipeline/components/function.py b/llama-index-core/llama_index/core/query_pipeline/components/function.py index bef5c377ab0848..f6a180da0666c1 100644 --- a/llama-index-core/llama_index/core/query_pipeline/components/function.py +++ b/llama-index-core/llama_index/core/query_pipeline/components/function.py @@ -114,3 +114,7 @@ def input_keys(self) -> InputKeys: def output_keys(self) -> OutputKeys: """Output keys.""" return OutputKeys.from_keys({self.output_key}) + + +# alias +FunctionComponent = FnComponent