From eb86c8be64d7d384bdfd2b1257c525e5cb99865d Mon Sep 17 00:00:00 2001 From: Gabriel Almeida Date: Mon, 15 May 2023 14:12:12 -0300 Subject: [PATCH] refactor(loading.py): swap parameters order in instantiate_prompt function to match other functions --- src/backend/langflow/interface/loading.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/backend/langflow/interface/loading.py b/src/backend/langflow/interface/loading.py index 148efd5b7..6cd9246b8 100644 --- a/src/backend/langflow/interface/loading.py +++ b/src/backend/langflow/interface/loading.py @@ -54,7 +54,7 @@ def instantiate_based_on_type(class_object, base_type, node_type, params): if base_type == "agents": return instantiate_agent(class_object, params) elif base_type == "prompts": - return instantiate_prompt(class_object, node_type, params) + return instantiate_prompt(node_type, class_object, params) elif base_type == "tools": return instantiate_tool(node_type, class_object, params) elif base_type == "toolkits": @@ -77,7 +77,7 @@ def instantiate_agent(class_object, params): return load_agent_executor(class_object, params) -def instantiate_prompt(class_object, node_type, params): +def instantiate_prompt(node_type, class_object, params): if node_type == "ZeroShotPrompt": if "tools" not in params: params["tools"] = [] @@ -96,7 +96,7 @@ def instantiate_tool(node_type, class_object, params): raise ValueError("Function should be a string") elif node_type.lower() == "tool": return class_object(**params) - return None # Or some other default action + return class_object(**params) def instantiate_toolkit(node_type, class_object, params):