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

update proxy field on clean edges

Browse source
This commit is contained in:
anovazzi1 2023-10-04 21:12:19 -03:00
commit 39ddc8eed4
2 changed files with 67 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

62
src/frontend/playwright-report/index.html Normal file
View file

File diff suppressed because one or more lines are too long

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

@ -54,6 +54,9 @@ export function cleanEdges({
id: targetNode.data.id, id: targetNode.data.id,
inputTypes: targetNode.data.node!.template[field]?.input_types, inputTypes: targetNode.data.node!.template[field]?.input_types,
}; };
if(targetNode.data.node!.template[field]?.proxy){
id.proxy = targetNode.data.node!.template[field]?.proxy
}
if (scapedJSONStringfy(id) !== targetHandle) { if (scapedJSONStringfy(id) !== targetHandle) {
newEdges = newEdges.filter((e) => e.id !== edge.id); newEdges = newEdges.filter((e) => e.id !== edge.id);
} }
@ -819,6 +822,7 @@ export function expandGroupNode(
const { template,flow } = _.cloneDeep(groupNode.node!); const { template,flow } = _.cloneDeep(groupNode.node!);
const gNodes: NodeType[] = flow?.data?.nodes!; const gNodes: NodeType[] = flow?.data?.nodes!;
const gEdges = flow!.data!.edges; const gEdges = flow!.data!.edges;
console.log(gEdges)
//redirect edges to correct proxy node //redirect edges to correct proxy node
let updatedEdges: Edge[] = []; let updatedEdges: Edge[] = [];
ReactFlowInstance.getEdges().forEach((edge) => { ReactFlowInstance.getEdges().forEach((edge) => {
@ -907,6 +911,7 @@ export function expandGroupNode(
...gEdges, ...gEdges,
...updatedEdges, ...updatedEdges,
]; ];
console.log(edges)
ReactFlowInstance.setNodes(nodes); ReactFlowInstance.setNodes(nodes);
ReactFlowInstance.setEdges(edges); ReactFlowInstance.setEdges(edges);
} }