jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

♻️ (flowStore.ts): fix updateBuildStatus function to correctly update the flowBuildStatus and improve code structure

Browse source
This commit is contained in:
ogabrielluiz 2024-06-18 16:02:03 -03:00
commit 1aacc8d116
1 changed files with 5 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/frontend/src/stores/flowStore.ts
View file

@ -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 });
},
}));