jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

fix: remove nodes that don't have edges

Browse source
This commit is contained in:
Gabriel Almeida 2023-04-05 18:10:00 -03:00
commit cfe68fb2aa
1 changed files with 6 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/backend/langflow/graph/graph.py
View file

@ -54,6 +54,12 @@ class Graph:
for node in self.nodes: for node in self.nodes:
if isinstance(node, ToolkitNode): if isinstance(node, ToolkitNode):
node.params["llm"] = llm_node 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]: def get_node(self, node_id: str) -> Union[None, Node]:
return next((node for node in self.nodes if node.id == node_id), None) return next((node for node in self.nodes if node.id == node_id), None)