diff --git a/src/frontend/src/utils/reactflowUtils.ts b/src/frontend/src/utils/reactflowUtils.ts index 9b62b2624..a2544b247 100644 --- a/src/frontend/src/utils/reactflowUtils.ts +++ b/src/frontend/src/utils/reactflowUtils.ts @@ -1117,6 +1117,18 @@ export function updateProxyIdsOnTemplate( }); } +export function updateProxyIdsOnOutputs( + outputs: OutputFieldType[] | undefined, + idsMap: { [key: string]: string }, +) { + if(!outputs) return; + outputs.forEach((output) => { + if (output.proxy && idsMap[output.proxy.id]) { + output.proxy.id = idsMap[output.proxy.id]; + } + }); +} + export function updateEdgesIds( edges: Edge[], idsMap: { [key: string]: string }, @@ -1485,6 +1497,7 @@ export function updateGroupRecursion(groupNode: NodeType, edges: Edge[]) { let newFlow = groupNode.data.node!.flow; const idsMap = updateIds(newFlow.data!); updateProxyIdsOnTemplate(groupNode.data.node!.template, idsMap); + updateProxyIdsOnOutputs(groupNode.data.node.outputs, idsMap); let flowEdges = edges; updateEdgesIds(flowEdges, idsMap); }