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 [] else []
) )
self.params["tools"] = tools self.params["tools"] = tools
# Extract the input variables from the prompt prompt_params = [
prompt_params = ["prefix", "suffix"] key for key, value in self.params.items() if value["type"] == "str"
]
else: else:
prompt_params = ["template"] prompt_params = ["template"]
for param in prompt_params: for param in prompt_params: