From 89a853212b87f98b944eb93f4b6452293f968110 Mon Sep 17 00:00:00 2001 From: Joel Date: Wed, 10 Apr 2024 11:16:54 +0800 Subject: [PATCH] fix: var assigner input node can not find caused error (#3274) --- web/app/components/workflow/utils.ts | 17 +++++++++++------ 1 file changed, 11 insertions(+), 6 deletions(-) diff --git a/web/app/components/workflow/utils.ts b/web/app/components/workflow/utils.ts index f586992dc7..d9239a8b6d 100644 --- a/web/app/components/workflow/utils.ts +++ b/web/app/components/workflow/utils.ts @@ -214,14 +214,19 @@ export const getNodesConnectedSourceOrTargetHandleIdsMap = (changes: ConnectedSo type, } = change const sourceNode = nodes.find(node => node.id === edge.source)! - nodesConnectedSourceOrTargetHandleIdsMap[sourceNode.id] = nodesConnectedSourceOrTargetHandleIdsMap[sourceNode.id] || { - _connectedSourceHandleIds: [...(sourceNode?.data._connectedSourceHandleIds || [])], - _connectedTargetHandleIds: [...(sourceNode?.data._connectedTargetHandleIds || [])], + if (sourceNode) { + nodesConnectedSourceOrTargetHandleIdsMap[sourceNode.id] = nodesConnectedSourceOrTargetHandleIdsMap[sourceNode.id] || { + _connectedSourceHandleIds: [...(sourceNode?.data._connectedSourceHandleIds || [])], + _connectedTargetHandleIds: [...(sourceNode?.data._connectedTargetHandleIds || [])], + } } + const targetNode = nodes.find(node => node.id === edge.target)! - nodesConnectedSourceOrTargetHandleIdsMap[targetNode.id] = nodesConnectedSourceOrTargetHandleIdsMap[targetNode.id] || { - _connectedSourceHandleIds: [...(targetNode?.data._connectedSourceHandleIds || [])], - _connectedTargetHandleIds: [...(targetNode?.data._connectedTargetHandleIds || [])], + if (targetNode) { + nodesConnectedSourceOrTargetHandleIdsMap[targetNode.id] = nodesConnectedSourceOrTargetHandleIdsMap[targetNode.id] || { + _connectedSourceHandleIds: [...(targetNode?.data._connectedSourceHandleIds || [])], + _connectedTargetHandleIds: [...(targetNode?.data._connectedTargetHandleIds || [])], + } } if (sourceNode) {