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

Refactor flowStore and flowsIOStore

Browse source
This commit is contained in:
anovazzi1 2024-01-18 18:48:18 -03:00
commit 10f67667d5
2 changed files with 51 additions and 143 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -343,6 +343,56 @@ const useFlowStore = create<FlowStoreType>((set, get) => ({
});
});
},
getInputTypes: () => {
let inputType: string[] = [];
const nodes = get().nodes;
nodes.forEach((node) => {
const nodeData: NodeDataType = node.data as NodeDataType;
if (isInputNode(nodeData)) {
// TODO remove count and ramdom key from type before pushing
inputType.push(nodeData.type);
}
});
set({ inputTypes: inputType });
return inputType;
},
getOutputTypes: () => {
let outputType: string[] = [];
const nodes = get().nodes;
nodes.forEach((node) => {
const nodeData: NodeDataType = node.data as NodeDataType;
if (isOutputNode(nodeData)) {
outputType.push(nodeData.type);
}
});
set({ outputTypes: outputType });
return outputType;
},
getInputIds: () => {
let inputIds: string[] = [];
const nodes = get().nodes;
nodes.forEach((node) => {
const nodeData: NodeDataType = node.data as NodeDataType;
if (isInputNode(nodeData)) {
inputIds.push(nodeData.id);
}
});
set({ inputIds });
return inputIds;
},
getOutputIds: () => {
let outputIds: string[] = [];
const nodes = get().nodes;
nodes.forEach((node) => {
const nodeData: NodeDataType = node.data as NodeDataType;
if (isOutputNode(nodeData)) {
outputIds.push(nodeData.id);
}
});
set({ outputIds });
return outputIds;
},
}));
export default useFlowStore;

144
src/frontend/src/stores/flowsIOStore.ts
View file

@ -1,8 +1,4 @@
import { cloneDeep } from "lodash";
import { create } from "zustand";
import { FlowType, NodeDataType, NodeType } from "../types/flow";
import { flowIOStoreType } from "../types/zustand/flowIOStore";
import { isInputNode, isOutputNode } from "../utils/reactflowUtils";
import useAlertStore from "./alertStore";
import useFlowStore from "./flowStore";
import useFlowsManagerStore from "./flowsManagerStore";
@ -13,145 +9,7 @@ const { reactFlowInstance, paste } = useFlowStore();
const { saveFlow } = useFlowsManagerStore();
const { setErrorData, setNoticeData } = useAlertStore();
const useFlowIOStore = create<flowIOStoreType>((set, get) => ({
flowPool: {},
outputTypes: [],
inputTypes: [],
inputIds: [],
outputIds: [],
setFlowPool: (flowPool) => {
set({ flowPool });
},
setOutputTypes: (outputTypes) => {
set({ outputTypes });
},
setInputTypes: (inputTypes) => {
set({ inputTypes });
},
setInputIds: (inputIds) => {
set({ inputIds });
},
setOutputIds: (outputIds) => {
set({ outputIds });
},
updateFlowPoolNodes: (nodes: NodeType[]) => {
//this function will update the removing the old ones
const nodeIdsSet = new Set(nodes.map((node) => node.id));
set((state) => {
let newFlowPool = cloneDeep({ ...state.flowPool });
Object.keys(newFlowPool).forEach((nodeId) => {
if (!nodeIdsSet.has(nodeId)) {
delete newFlowPool[nodeId];
}
});
return { flowPool: newFlowPool };
});
},
addDataToFlowPool: (data: any, nodeId: string) => {
set((state) => {
let newFlowPool = cloneDeep({ ...state.flowPool });
if (!newFlowPool[nodeId]) newFlowPool[nodeId] = [data];
else {
newFlowPool[nodeId].push(data);
}
return { flowPool: newFlowPool };
});
},
CleanFlowPool: () => {
set({ flowPool: {} });
},
checkInputandOutput: (flow?: FlowType) => {
let has_input = false;
let has_output = false;
if (!flow && !reactFlowInstance) {
return false;
}
const nodes = flow?.data?.nodes
? flow.data.nodes
: reactFlowInstance!.getNodes();
nodes.forEach((node) => {
const nodeData: NodeDataType = node.data as NodeDataType;
if (isInputNode(nodeData)) {
has_input = true;
}
if (isOutputNode(nodeData)) {
has_output = true;
}
});
return has_input && has_output;
},
getInputTypes: (flow?: FlowType) => {
let inputType: string[] = [];
if (!flow && !reactFlowInstance) {
return [];
}
const nodes = flow?.data?.nodes
? flow.data.nodes
: reactFlowInstance!.getNodes();
nodes.forEach((node) => {
const nodeData: NodeDataType = node.data as NodeDataType;
if (isInputNode(nodeData)) {
// TODO remove count and ramdom key from type before pushing
inputType.push(nodeData.type);
}
});
set({ inputTypes: inputType });
return inputType;
},
getOutputTypes: (flow?: FlowType) => {
let outputType: string[] = [];
if (!flow && !reactFlowInstance) {
return [];
}
const nodes = flow?.data?.nodes
? flow.data.nodes
: reactFlowInstance!.getNodes();
nodes.forEach((node) => {
const nodeData: NodeDataType = node.data as NodeDataType;
if (isOutputNode(nodeData)) {
// TODO remove count and ramdom key from type before pushing
outputType.push(nodeData.type);
}
});
set({ outputTypes: outputType });
return outputType;
},
getInputIds: (flow?: FlowType) => {
let inputIds: string[] = [];
if (!flow && !reactFlowInstance) {
return [];
}
const nodes = flow?.data?.nodes
? flow.data.nodes
: reactFlowInstance!.getNodes();
nodes.forEach((node) => {
const nodeData: NodeDataType = node.data as NodeDataType;
if (isInputNode(nodeData)) {
inputIds.push(nodeData.id);
}
});
set({ inputIds });
return inputIds;
},
getOutputIds: (flow?: FlowType) => {
let outputIds: string[] = [];
if (!flow && !reactFlowInstance) {
return [];
}
const nodes = flow?.data?.nodes
? flow.data.nodes
: reactFlowInstance!.getNodes();
nodes.forEach((node) => {
const nodeData: NodeDataType = node.data as NodeDataType;
if (isOutputNode(nodeData)) {
outputIds.push(nodeData.id);
}
});
set({ outputIds });
return outputIds;
},
const useFlowIOStore = create<any>((set, get) => ({
/* buildFlow: async (nodeId?: string) => {
function handleBuildUpdate(data: any) {
get().addDataToFlowPool(data.data[data.id], data.id);