From 9c8d470e5e44102209c515718154ae1dc2fda18b Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Mon, 19 Feb 2024 18:53:31 -0300
Subject: [PATCH] Add new attributes to Vertex class

---
 src/backend/langflow/graph/vertex/base.py | 11 ++++-------
 1 file changed, 4 insertions(+), 7 deletions(-)

diff --git a/src/backend/langflow/graph/vertex/base.py b/src/backend/langflow/graph/vertex/base.py
index 503855e6e..4029f422e 100644
--- a/src/backend/langflow/graph/vertex/base.py
+++ b/src/backend/langflow/graph/vertex/base.py
@@ -55,11 +55,14 @@ class Vertex:
         self.parent_node_id: Optional[str] = self._data.get("parent_node_id")
         self.parent_is_top_level = False
         self.layer = None
+        self.should_run = True
         try:
             self.is_interface_component = InterfaceComponentTypes(self.vertex_type)
         except ValueError:
             self.is_interface_component = False
 
+        self.use_result = False
+
     # Build a result dict for each edge
     # 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]]:
@@ -375,7 +378,7 @@ class Vertex:
         # PLEASE REVIEW THIS IF STATEMENT
         # Check if the Vertex was built already
         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:
             task = self.get_task()
@@ -561,9 +564,3 @@ class StatefulVertex(Vertex):
 
 class StatelessVertex(Vertex):
     pass
-    pass
-    pass
-    pass
-    pass
-    pass
-    pass