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

refactor: add idx property to handle types in GenericNode and reactflowUtils

Browse source
This commit is contained in:
anovazzi1 2024-05-28 17:57:52 -03:00
commit 45011e8fda
3 changed files with 6 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
src/frontend/src/customNodes/genericNode/index.tsx
View file

@ -589,6 +589,7 @@ export default function GenericNode({
baseClasses: data.node!.base_classes, baseClasses: data.node!.base_classes,
id: data.id, id: data.id,
dataType: data.type, dataType: data.type,
idx: 0,
}} }}
type={data.node?.base_classes.join("|")} type={data.node?.base_classes.join("|")}
left={false} left={false}
@ -848,6 +849,7 @@ export default function GenericNode({
baseClasses: [output.selected ?? output.types[0]], baseClasses: [output.selected ?? output.types[0]],
id: data.id, id: data.id,
dataType: data.type, dataType: data.type,
idx: idx,
}} }}
type={output.types.join("|")} type={output.types.join("|")}
left={false} left={false}

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

@ -58,6 +58,7 @@ export type sourceHandleType = {
dataType: string; dataType: string;
id: string; id: string;
baseClasses: string[]; baseClasses: string[];
idx: number;
}; };
//left side //left side
export type targetHandleType = { export type targetHandleType = {

3
src/frontend/src/utils/reactflowUtils.ts
View file

@ -79,6 +79,8 @@ export function cleanEdges(nodes: Node[], edges: Edge[]) {
id: sourceNode.data.id, id: sourceNode.data.id,
baseClasses: sourceNode.data.node!.base_classes, baseClasses: sourceNode.data.node!.base_classes,
dataType: sourceNode.data.type, dataType: sourceNode.data.type,
idx:
sourceNode.data.node!.outputs[scapeJSONParse(sourceHandle).idx] ?? 0,
}; };
if (scapedJSONStringfy(id) !== sourceHandle) { if (scapedJSONStringfy(id) !== sourceHandle) {
newEdges = newEdges.filter((e) => e.id !== edge.id); newEdges = newEdges.filter((e) => e.id !== edge.id);
@ -397,6 +399,7 @@ export function updateEdgesHandleIds({
id: sourceNode.data.id, id: sourceNode.data.id,
baseClasses: sourceNode.data.node!.base_classes, baseClasses: sourceNode.data.node!.base_classes,
dataType: sourceNode.data.type, dataType: sourceNode.data.type,
idx: 0,
}; };
} }
edge.sourceHandle = scapedJSONStringfy(newSource!); edge.sourceHandle = scapedJSONStringfy(newSource!);