From 1aacc8d116530aa6d5da27ddb8561ee4ea185f88 Mon Sep 17 00:00:00 2001
From: ogabrielluiz <gabriel@langflow.org>
Date: Tue, 18 Jun 2024 16:02:03 -0300
Subject: [PATCH] =?UTF-8?q?=E2=99=BB=EF=B8=8F=20(flowStore.ts):=20fix=20up?=
 =?UTF-8?q?dateBuildStatus=20function=20to=20correctly=20update=20the=20fl?=
 =?UTF-8?q?owBuildStatus=20and=20improve=20code=20structure?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

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

diff --git a/src/frontend/src/stores/flowStore.ts b/src/frontend/src/stores/flowStore.ts
index 278a673c9..c2b84ae5a 100644
--- a/src/frontend/src/stores/flowStore.ts
+++ b/src/frontend/src/stores/flowStore.ts
@@ -509,6 +509,7 @@ const useFlowStore = create<FlowStoreType>((set, get) => ({
           runId: runId,
           verticesToRun: get().verticesBuild!.verticesToRun,
         });
+
         get().updateBuildStatus(top_level_vertices, BuildStatus.TO_BUILD);
       }
 
@@ -522,10 +523,12 @@ const useFlowStore = create<FlowStoreType>((set, get) => ({
       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);
       }
@@ -625,6 +628,7 @@ const useFlowStore = create<FlowStoreType>((set, get) => ({
     });
   },
   updateBuildStatus: (nodeIdList: string[], status: BuildStatus) => {
+
     const newFlowBuildStatus = { ...get().flowBuildStatus };
     nodeIdList.forEach((id) => {
       newFlowBuildStatus[id] = {
@@ -644,6 +648,7 @@ const useFlowStore = create<FlowStoreType>((set, get) => ({
         newFlowBuildStatus[id].status = BuildStatus.BUILT;
       }
     });
+    set({ flowBuildStatus: newFlowBuildStatus });
   },
 }));