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

Revert built status from building in flowStore.ts

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-26 09:02:52 -03:00
commit 49df8e993d
2 changed files with 10 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -416,6 +416,7 @@ const useFlowStore = create<FlowStoreType>((set, get) => ({
useFlowStore.getState().updateBuildStatus(idList, BuildStatus.BUILDING);
},
});
get().revertBuiltStatusFromBuilding();
},
getFlow: () => {
return {
@ -445,6 +446,14 @@ const useFlowStore = create<FlowStoreType>((set, get) => ({
}
});
},
revertBuiltStatusFromBuilding: () => {
get().nodes.forEach((node) => {
if (node.data.buildStatus === BuildStatus.BUILDING) {
node.data.buildStatus = BuildStatus.TO_BUILD;
}
});
set({ nodes: get().nodes });
},
}));
export default useFlowStore;

1
src/frontend/src/types/zustand/flow/index.ts
View file

@ -90,4 +90,5 @@ export type FlowStoreType = {
removeFromVerticesBuild: (vertices: string[]) => void;
verticesBuild: string[];
updateBuildStatus: (nodeId: string[], status: BuildStatus) => void;
revertBuiltStatusFromBuilding: () => void;
};