From c5f24e7dfa2e22119f6d89b090530b9bf664a64b Mon Sep 17 00:00:00 2001 From: Gabriel Almeida Date: Sat, 8 Apr 2023 09:02:42 -0300 Subject: [PATCH] fix: linting --- src/backend/langflow/graph/nodes.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/src/backend/langflow/graph/nodes.py b/src/backend/langflow/graph/nodes.py index a58300854..5320adc61 100644 --- a/src/backend/langflow/graph/nodes.py +++ b/src/backend/langflow/graph/nodes.py @@ -55,7 +55,10 @@ class PromptNode(Node): tools: Optional[Union[List[Node], List[ToolNode]]] = None, ) -> Any: if not self._built or force: - if "input_variables" not in self.params: + if ( + "input_variables" not in self.params + or self.params["input_variables"] is None + ): self.params["input_variables"] = [] # Check if it is a ZeroShotPrompt and needs a tool if "ShotPrompt" in self.node_type: @@ -75,8 +78,6 @@ class PromptNode(Node): for param in prompt_params: prompt_text = self.params[param] variables = extract_input_variables_from_prompt(prompt_text) - if self.params["input_variables"] is None: - self.params["input_variables"] = [] self.params["input_variables"].extend(variables) self.params["input_variables"] = list(set(self.params["input_variables"]))