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

Add predecessors and successors properties to Vertex class

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-21 09:57:49 -03:00
commit d35a3d7b87
1 changed files with 8 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/backend/langflow/graph/vertex/base.py
View file

@ -113,6 +113,14 @@ class Vertex:
def edges(self) -> List["ContractEdge"]: def edges(self) -> List["ContractEdge"]:
return self.graph.get_vertex_edges(self.id) 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): def __getstate__(self):
return { return {
"_data": self._data, "_data": self._data,