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

🐛 fix(base.py): rename _remove_invalid_nodes() to _validate_nodes() for clarity and accuracy

Browse source 
🔥 chore(base.py): remove unnecessary check for single node and no edges in _validate_nodes()
🔥 chore(base.py): remove unused import statement
🔥 chore(base.py): remove unused _validate_node() method
🔥 chore(base.py): remove unused import statement
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-07-13 15:01:08 -03:00
commit 38da096245
1 changed files with 8 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

17
src/backend/langflow/graph/graph/base.py
View file

@ -60,7 +60,7 @@ class Graph:
# the toolkit node # the toolkit node
self._build_node_params() self._build_node_params()
# remove invalid nodes # remove invalid nodes
self._remove_invalid_nodes() self._validate_nodes()
def _build_node_params(self) -> None: def _build_node_params(self) -> None:
"""Identifies and handles the LLM node within the graph.""" """Identifies and handles the LLM node within the graph."""
@ -75,14 +75,13 @@ class Graph:
if isinstance(node, ToolkitVertex): if isinstance(node, ToolkitVertex):
node.params["llm"] = llm_node node.params["llm"] = llm_node
def _remove_invalid_nodes(self) -> None: def _validate_nodes(self) -> None:
"""Removes invalid nodes from the graph.""" """Check that all nodes have edges"""
self.nodes = [ for node in self.nodes:
node if not self._validate_node(node):
for node in self.nodes raise ValueError(
if self._validate_node(node) f"{node.vertex_type} is not connected to any other components"
or (len(self.nodes) == 1 and len(self.edges) == 0) )
]
def _validate_node(self, node: Vertex) -> bool: def _validate_node(self, node: Vertex) -> bool:
"""Validates a node.""" """Validates a node."""