From cfe68fb2aaff47f2b8641787363c87da0426ad7c Mon Sep 17 00:00:00 2001 From: Gabriel Almeida Date: Wed, 5 Apr 2023 18:10:00 -0300 Subject: [PATCH] fix: remove nodes that don't have edges --- src/backend/langflow/graph/graph.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/src/backend/langflow/graph/graph.py b/src/backend/langflow/graph/graph.py index 4d6bb4172..76cab071d 100644 --- a/src/backend/langflow/graph/graph.py +++ b/src/backend/langflow/graph/graph.py @@ -54,6 +54,12 @@ class Graph: for node in self.nodes: if isinstance(node, ToolkitNode): node.params["llm"] = llm_node + # remove invalid nodes + self.nodes = [node for node in self.nodes if self._validate_node(node)] + + def _validate_node(self, node: Node) -> bool: + # All nodes that do not have edges are invalid + return len(node.edges) > 0 def get_node(self, node_id: str) -> Union[None, Node]: return next((node for node in self.nodes if node.id == node_id), None)