From f9cb449c4f5459fac3aab31f54e0bd3f6669464b Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Sat, 30 Mar 2024 17:58:01 -0300
Subject: [PATCH] Set onBuildComplete to run after all layers

---
 src/frontend/src/utils/buildUtils.ts | 11 +++++------
 1 file changed, 5 insertions(+), 6 deletions(-)

diff --git a/src/frontend/src/utils/buildUtils.ts b/src/frontend/src/utils/buildUtils.ts
index 46b81ba9d..bc2cc62e7 100644
--- a/src/frontend/src/utils/buildUtils.ts
+++ b/src/frontend/src/utils/buildUtils.ts
@@ -204,12 +204,11 @@ export async function buildVertices({
     if (stop) {
       break;
     }
-
-    if (onBuildComplete) {
-      const allNodesValid = buildResults.every((result) => result);
-      onBuildComplete(allNodesValid);
-      useFlowStore.getState().setIsBuilding(false);
-    }
+  }
+  if (onBuildComplete) {
+    const allNodesValid = buildResults.every((result) => result);
+    onBuildComplete(allNodesValid);
+    useFlowStore.getState().setIsBuilding(false);
   }
 }
 async function buildVertex({