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

Add checkInputandOutput function to flowStore.ts

Browse source
This commit is contained in:
anovazzi1 2024-01-18 18:18:54 -03:00
commit 9c0d794dd1
1 changed files with 17 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -20,6 +20,8 @@ import {
cleanEdges, cleanEdges,
getHandleId, getHandleId,
getNodeId, getNodeId,
isInputNode,
isOutputNode,
scapeJSONParse, scapeJSONParse,
scapedJSONStringfy, scapedJSONStringfy,
} from "../utils/reactflowUtils"; } from "../utils/reactflowUtils";
@ -153,6 +155,21 @@ const useFlowStore = create<FlowStoreType>((set, get) => ({
}) })
); );
}, },
checkInputandOutput: () => {
let has_input = false;
let has_output = false;
const nodes = get().nodes;
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;
},
getNode: (id: string) => { getNode: (id: string) => {
return get().nodes.find((node) => node.id === id); return get().nodes.find((node) => node.id === id);
}, },