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 (#302)

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-05-14 23:24:36 +00:00 • committed by GitHub
commit 8c93b9264a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

2
pyproject.toml
View file

@ -1,6 +1,6 @@
[tool.poetry] [tool.poetry]
name = "langflow" name = "langflow"
version = "0.0.70" version = "0.0.71"
description = "A Python package with a built-in web application" description = "A Python package with a built-in web application"
authors = ["Logspace <contact@logspace.ai>"] authors = ["Logspace <contact@logspace.ai>"]
maintainers = [ maintainers = [

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):