jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

refactor(loading.py): change instantiate_prompt function signature to include class_object parameter and use it to instantiate the prompt object

Browse source
This commit is contained in:
Gabriel Almeida 2023-05-14 20:21:42 -03:00
commit 9627005f8d
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/backend/langflow/interface/loading.py
View file

@ -54,7 +54,7 @@ def instantiate_based_on_type(class_object, base_type, node_type, params):
if base_type == "agents": if base_type == "agents":
return instantiate_agent(class_object, params) return instantiate_agent(class_object, params)
elif base_type == "prompts": elif base_type == "prompts":
return instantiate_prompt(node_type, params) return instantiate_prompt(class_object, node_type, params)
elif base_type == "tools": elif base_type == "tools":
return instantiate_tool(node_type, class_object, params) return instantiate_tool(node_type, class_object, params)
elif base_type == "toolkits": elif base_type == "toolkits":
@ -77,12 +77,12 @@ def instantiate_agent(class_object, params):
return load_agent_executor(class_object, params) return load_agent_executor(class_object, params)
def instantiate_prompt(node_type, params): def instantiate_prompt(class_object, node_type, params):
if node_type == "ZeroShotPrompt": if node_type == "ZeroShotPrompt":
if "tools" not in params: if "tools" not in params:
params["tools"] = [] params["tools"] = []
return ZeroShotAgent.create_prompt(**params) return ZeroShotAgent.create_prompt(**params)
return None # Or some other default action return class_object(**params)
def instantiate_tool(node_type, class_object, params): def instantiate_tool(node_type, class_object, params):