From d35a3d7b8737bb5908c696d1a49f7ed8d57301b3 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 21 Feb 2024 09:57:49 -0300 Subject: [PATCH] Add predecessors and successors properties to Vertex class --- src/backend/langflow/graph/vertex/base.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/src/backend/langflow/graph/vertex/base.py b/src/backend/langflow/graph/vertex/base.py index c43b8fd0d..47c9bf2ec 100644 --- a/src/backend/langflow/graph/vertex/base.py +++ b/src/backend/langflow/graph/vertex/base.py @@ -113,6 +113,14 @@ class Vertex: def edges(self) -> List["ContractEdge"]: return self.graph.get_vertex_edges(self.id) + @property + def predecessors(self) -> List["Vertex"]: + return self.graph.get_predecessors(self) + + @property + def successors(self) -> List["Vertex"]: + return self.graph.get_successors(self) + def __getstate__(self): return { "_data": self._data,