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

🐛 fix(base.py): fix type hint for nodes parameter in Graph class constructor to improve code readability and maintainability

Browse source 
🔀 refactor(base.py): refactor initialization of top_level_nodes list in Graph class constructor to improve code readability and maintainability
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-10-06 16:59:29 -03:00
commit 7a1f06128e
1 changed files with 7 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -20,13 +20,18 @@ class Graph:
def __init__( def __init__(
self, self,
nodes: List[Dict[str, Union[str, Dict[str, Union[str, List[str]]]]]], nodes: List[Dict],
edges: List[Dict[str, str]], edges: List[Dict[str, str]],
) -> None: ) -> None:
self._nodes = nodes self._nodes = nodes
self._edges = edges self._edges = edges
self.raw_graph_data = {"nodes": nodes, "edges": edges} self.raw_graph_data = {"nodes": nodes, "edges": edges}
self.top_level_nodes = [node.get("id") for node in nodes]
self.top_level_nodes = []
for node in self._nodes:
if node_id := node.get("id"):
self.top_level_nodes.append(node_id)
self._graph_data = process_flow(self.raw_graph_data) self._graph_data = process_flow(self.raw_graph_data)
self._nodes = self._graph_data["nodes"] self._nodes = self._graph_data["nodes"]
self._edges = self._graph_data["edges"] self._edges = self._graph_data["edges"]