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

Add new attributes to Vertex class

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-19 18:53:31 -03:00
commit 9c8d470e5e
1 changed files with 4 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -55,11 +55,14 @@ class Vertex:
self.parent_node_id: Optional[str] = self._data.get("parent_node_id") self.parent_node_id: Optional[str] = self._data.get("parent_node_id")
self.parent_is_top_level = False self.parent_is_top_level = False
self.layer = None self.layer = None
self.should_run = True
try: try:
self.is_interface_component = InterfaceComponentTypes(self.vertex_type) self.is_interface_component = InterfaceComponentTypes(self.vertex_type)
except ValueError: except ValueError:
self.is_interface_component = False self.is_interface_component = False
self.use_result = False
# Build a result dict for each edge # Build a result dict for each edge
# like so: {edge.target.id: {edge.target_param: self._built_object}} # like so: {edge.target.id: {edge.target_param: self._built_object}}
async def get_result_dict(self, force: bool = False) -> Dict[str, Dict[str, Any]]: async def get_result_dict(self, force: bool = False) -> Dict[str, Dict[str, Any]]:
@ -375,7 +378,7 @@ class Vertex:
# PLEASE REVIEW THIS IF STATEMENT # PLEASE REVIEW THIS IF STATEMENT
# Check if the Vertex was built already # Check if the Vertex was built already
if self._built: if self._built:
return self._built_object return self._built_object if not self.use_result else self._built_result
if self.is_task and self.task_id is not None: if self.is_task and self.task_id is not None:
task = self.get_task() task = self.get_task()
@ -561,9 +564,3 @@ class StatefulVertex(Vertex):
class StatelessVertex(Vertex): class StatelessVertex(Vertex):
pass pass
pass
pass
pass
pass
pass
pass