fix: prompt_params now come from params

This commit is contained in:
Gabriel Almeida 2023-03-31 18:19:45 -03:00
commit 9ef1078786

View file

@ -67,8 +67,9 @@ class PromptNode(Node):
else []
)
self.params["tools"] = tools
# Extract the input variables from the prompt
prompt_params = ["prefix", "suffix"]
prompt_params = [
key for key, value in self.params.items() if value["type"] == "str"
]
else:
prompt_params = ["template"]
for param in prompt_params: