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

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

View file

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

View file

@ -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(node_type, params)
return instantiate_prompt(class_object, node_type, params)
elif base_type == "tools":
return instantiate_tool(node_type, class_object, params)
elif base_type == "toolkits":
@ -77,12 +77,12 @@ def instantiate_agent(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 "tools" not in params:
params["tools"] = []
return ZeroShotAgent.create_prompt(**params)
return None # Or some other default action
return class_object(**params)
def instantiate_tool(node_type, class_object, params):