From 8d55e058cefcd10d67af80e40e3526dee40ea8a4 Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Wed, 27 Mar 2024 16:52:10 -0300
Subject: [PATCH] Fix build status update in flowStore.ts

---
 src/frontend/src/stores/flowStore.ts | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/src/frontend/src/stores/flowStore.ts b/src/frontend/src/stores/flowStore.ts
index 09b8d5240..54b71f1ea 100644
--- a/src/frontend/src/stores/flowStore.ts
+++ b/src/frontend/src/stores/flowStore.ts
@@ -505,6 +505,17 @@ const useFlowStore = create<FlowStoreType>((set, get) => ({
       );
 
       useFlowStore.getState().updateBuildStatus([vertexBuildData.id], status);
+
+      const verticesIds = get().verticesBuild?.verticesIds;
+      const newFlowBuildStatus = { ...get().flowBuildStatus };
+      // filter out the vertices that are not status
+      const verticesToUpdate = verticesIds?.filter(
+        (id) => newFlowBuildStatus[id]?.status !== BuildStatus.BUILT
+      );
+
+      if (verticesToUpdate) {
+        useFlowStore.getState().updateBuildStatus(verticesToUpdate, status);
+      }
     }
     await buildVertices({
       input_value,
@@ -591,7 +602,6 @@ const useFlowStore = create<FlowStoreType>((set, get) => ({
   },
   updateBuildStatus: (nodeIdList: string[], status: BuildStatus) => {
     const newFlowBuildStatus = { ...get().flowBuildStatus };
-
     nodeIdList.forEach((id) => {
       newFlowBuildStatus[id] = {
         status,