From 0d15491d36910c46c2e88996476eb5b64f50b4e7 Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Wed, 20 Mar 2024 16:07:19 -0300
Subject: [PATCH] Update useFlowStore to use top_level_vertices instead of
 next_vertices_ids

---
 src/frontend/src/stores/flowStore.ts | 2 +-
 src/frontend/src/types/api/index.ts  | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/frontend/src/stores/flowStore.ts b/src/frontend/src/stores/flowStore.ts
index 16f6b75f4..5ca1de1e0 100644
--- a/src/frontend/src/stores/flowStore.ts
+++ b/src/frontend/src/stores/flowStore.ts
@@ -473,7 +473,7 @@ const useFlowStore = create<FlowStoreType>((set, get) => ({
           runId: runId,
         });
         get().updateBuildStatus(
-          vertexBuildData.next_vertices_ids,
+          vertexBuildData.top_level_vertices,
           BuildStatus.TO_BUILD
         );
       }
diff --git a/src/frontend/src/types/api/index.ts b/src/frontend/src/types/api/index.ts
index 172f01dad..297984d28 100644
--- a/src/frontend/src/types/api/index.ts
+++ b/src/frontend/src/types/api/index.ts
@@ -146,6 +146,7 @@ export type VertexBuildTypeAPI = {
   id: string;
   inactivated_vertices: Array<string> | null;
   next_vertices_ids: Array<string>;
+  top_level_vertices: Array<string>;
   run_id: string;
   valid: boolean;
   params: string;