From 9ef10787861d92e7033d9ebabb2bdbf0d1ceff5d Mon Sep 17 00:00:00 2001
From: Gabriel Almeida <gabriel@logspace.ai>
Date: Fri, 31 Mar 2023 18:19:45 -0300
Subject: [PATCH] fix: prompt_params now come from params

---
 src/backend/langflow/graph/nodes.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/backend/langflow/graph/nodes.py b/src/backend/langflow/graph/nodes.py
index 4ed0ba8a3..c6834e842 100644
--- a/src/backend/langflow/graph/nodes.py
+++ b/src/backend/langflow/graph/nodes.py
@@ -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: