Add FunctionComponent alias (run-llama#10673)

add FunctionComponent alias
Izukimat · Mar 29, 2024 · 90e9a83 · 90e9a83
1 parent 54c5b51
commit 90e9a83
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/llama-index-core/llama_index/core/query_pipeline/__init__.py b/llama-index-core/llama_index/core/query_pipeline/__init__.py
@@ -1,12 +1,16 @@
 """Init file."""
+
 from llama_index.core.query_pipeline.components.agent import (
     AgentFnComponent,
     AgentInputComponent,
     CustomAgentComponent,
     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
@@ -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