From 7a1f06128e5320292e7f443506cfaff94364712b Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Fri, 6 Oct 2023 16:59:29 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20fix(base.py):=20fix=20type=20hin?= =?UTF-8?q?t=20for=20nodes=20parameter=20in=20Graph=20class=20constructor?= =?UTF-8?q?=20to=20improve=20code=20readability=20and=20maintainability=20?= =?UTF-8?q?=F0=9F=94=80=20refactor(base.py):=20refactor=20initialization?= =?UTF-8?q?=20of=20top=5Flevel=5Fnodes=20list=20in=20Graph=20class=20const?= =?UTF-8?q?ructor=20to=20improve=20code=20readability=20and=20maintainabil?= =?UTF-8?q?ity?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/graph/graph/base.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/src/backend/langflow/graph/graph/base.py b/src/backend/langflow/graph/graph/base.py index c60741116..83976ae49 100644 --- a/src/backend/langflow/graph/graph/base.py +++ b/src/backend/langflow/graph/graph/base.py @@ -20,13 +20,18 @@ class Graph: def __init__( self, - nodes: List[Dict[str, Union[str, Dict[str, Union[str, List[str]]]]]], + nodes: List[Dict], edges: List[Dict[str, str]], ) -> None: self._nodes = nodes self._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._nodes = self._graph_data["nodes"] self._edges = self._graph_data["edges"]