| import { | |
| Position, | |
| getConnectedEdges, | |
| getIncomers, | |
| getOutgoers, | |
| } from 'reactflow' | |
| import dagre from '@dagrejs/dagre' | |
| import { v4 as uuid4 } from 'uuid' | |
| import { | |
| cloneDeep, | |
| groupBy, | |
| isEqual, | |
| uniqBy, | |
| } from 'lodash-es' | |
| import type { | |
| Edge, | |
| InputVar, | |
| Node, | |
| ToolWithProvider, | |
| ValueSelector, | |
| } from './types' | |
| import { BlockEnum, ErrorHandleMode } from './types' | |
| import { | |
| CUSTOM_NODE, | |
| ITERATION_CHILDREN_Z_INDEX, | |
| ITERATION_NODE_Z_INDEX, | |
| NODE_WIDTH_X_OFFSET, | |
| START_INITIAL_POSITION, | |
| } from './constants' | |
| import { CUSTOM_ITERATION_START_NODE } from './nodes/iteration-start/constants' | |
| import type { QuestionClassifierNodeType } from './nodes/question-classifier/types' | |
| import type { IfElseNodeType } from './nodes/if-else/types' | |
| import { branchNameCorrect } from './nodes/if-else/utils' | |
| import type { ToolNodeType } from './nodes/tool/types' | |
| import type { IterationNodeType } from './nodes/iteration/types' | |
| import { CollectionType } from '@/app/components/tools/types' | |
| import { toolParametersToFormSchemas } from '@/app/components/tools/utils/to-form-schema' | |
| const WHITE = 'WHITE' | |
| const GRAY = 'GRAY' | |
| const BLACK = 'BLACK' | |
| const isCyclicUtil = (nodeId: string, color: Record<string, string>, adjList: Record<string, string[]>, stack: string[]) => { | |
| color[nodeId] = GRAY | |
| stack.push(nodeId) | |
| for (let i = 0; i < adjList[nodeId].length; ++i) { | |
| const childId = adjList[nodeId][i] | |
| if (color[childId] === GRAY) { | |
| stack.push(childId) | |
| return true | |
| } | |
| if (color[childId] === WHITE && isCyclicUtil(childId, color, adjList, stack)) | |
| return true | |
| } | |
| color[nodeId] = BLACK | |
| if (stack.length > 0 && stack[stack.length - 1] === nodeId) | |
| stack.pop() | |
| return false | |
| } | |
| const getCycleEdges = (nodes: Node[], edges: Edge[]) => { | |
| const adjList: Record<string, string[]> = {} | |
| const color: Record<string, string> = {} | |
| const stack: string[] = [] | |
| for (const node of nodes) { | |
| color[node.id] = WHITE | |
| adjList[node.id] = [] | |
| } | |
| for (const edge of edges) | |
| adjList[edge.source]?.push(edge.target) | |
| for (let i = 0; i < nodes.length; i++) { | |
| if (color[nodes[i].id] === WHITE) | |
| isCyclicUtil(nodes[i].id, color, adjList, stack) | |
| } | |
| const cycleEdges = [] | |
| if (stack.length > 0) { | |
| const cycleNodes = new Set(stack) | |
| for (const edge of edges) { | |
| if (cycleNodes.has(edge.source) && cycleNodes.has(edge.target)) | |
| cycleEdges.push(edge) | |
| } | |
| } | |
| return cycleEdges | |
| } | |
| export function getIterationStartNode(iterationId: string): Node { | |
| return generateNewNode({ | |
| id: `${iterationId}start`, | |
| type: CUSTOM_ITERATION_START_NODE, | |
| data: { | |
| title: '', | |
| desc: '', | |
| type: BlockEnum.IterationStart, | |
| isInIteration: true, | |
| }, | |
| position: { | |
| x: 24, | |
| y: 68, | |
| }, | |
| zIndex: ITERATION_CHILDREN_Z_INDEX, | |
| parentId: iterationId, | |
| selectable: false, | |
| draggable: false, | |
| }).newNode | |
| } | |
| export function generateNewNode({ data, position, id, zIndex, type, ...rest }: Omit<Node, 'id'> & { id?: string }): { | |
| newNode: Node | |
| newIterationStartNode?: Node | |
| } { | |
| const newNode = { | |
| id: id || `${Date.now()}`, | |
| type: type || CUSTOM_NODE, | |
| data, | |
| position, | |
| targetPosition: Position.Left, | |
| sourcePosition: Position.Right, | |
| zIndex: data.type === BlockEnum.Iteration ? ITERATION_NODE_Z_INDEX : zIndex, | |
| ...rest, | |
| } as Node | |
| if (data.type === BlockEnum.Iteration) { | |
| const newIterationStartNode = getIterationStartNode(newNode.id); | |
| (newNode.data as IterationNodeType).start_node_id = newIterationStartNode.id; | |
| (newNode.data as IterationNodeType)._children = [newIterationStartNode.id] | |
| return { | |
| newNode, | |
| newIterationStartNode, | |
| } | |
| } | |
| return { | |
| newNode, | |
| } | |
| } | |
| export const preprocessNodesAndEdges = (nodes: Node[], edges: Edge[]) => { | |
| const hasIterationNode = nodes.some(node => node.data.type === BlockEnum.Iteration) | |
| if (!hasIterationNode) { | |
| return { | |
| nodes, | |
| edges, | |
| } | |
| } | |
| const nodesMap = nodes.reduce((prev, next) => { | |
| prev[next.id] = next | |
| return prev | |
| }, {} as Record<string, Node>) | |
| const iterationNodesWithStartNode = [] | |
| const iterationNodesWithoutStartNode = [] | |
| for (let i = 0; i < nodes.length; i++) { | |
| const currentNode = nodes[i] as Node<IterationNodeType> | |
| if (currentNode.data.type === BlockEnum.Iteration) { | |
| if (currentNode.data.start_node_id) { | |
| if (nodesMap[currentNode.data.start_node_id]?.type !== CUSTOM_ITERATION_START_NODE) | |
| iterationNodesWithStartNode.push(currentNode) | |
| } | |
| else { | |
| iterationNodesWithoutStartNode.push(currentNode) | |
| } | |
| } | |
| } | |
| const newIterationStartNodesMap = {} as Record<string, Node> | |
| const newIterationStartNodes = [...iterationNodesWithStartNode, ...iterationNodesWithoutStartNode].map((iterationNode, index) => { | |
| const newNode = getIterationStartNode(iterationNode.id) | |
| newNode.id = newNode.id + index | |
| newIterationStartNodesMap[iterationNode.id] = newNode | |
| return newNode | |
| }) | |
| const newEdges = iterationNodesWithStartNode.map((iterationNode) => { | |
| const newNode = newIterationStartNodesMap[iterationNode.id] | |
| const startNode = nodesMap[iterationNode.data.start_node_id] | |
| const source = newNode.id | |
| const sourceHandle = 'source' | |
| const target = startNode.id | |
| const targetHandle = 'target' | |
| return { | |
| id: `${source}-${sourceHandle}-${target}-${targetHandle}`, | |
| type: 'custom', | |
| source, | |
| sourceHandle, | |
| target, | |
| targetHandle, | |
| data: { | |
| sourceType: newNode.data.type, | |
| targetType: startNode.data.type, | |
| isInIteration: true, | |
| iteration_id: startNode.parentId, | |
| _connectedNodeIsSelected: true, | |
| }, | |
| zIndex: ITERATION_CHILDREN_Z_INDEX, | |
| } | |
| }) | |
| nodes.forEach((node) => { | |
| if (node.data.type === BlockEnum.Iteration && newIterationStartNodesMap[node.id]) | |
| (node.data as IterationNodeType).start_node_id = newIterationStartNodesMap[node.id].id | |
| }) | |
| return { | |
| nodes: [...nodes, ...newIterationStartNodes], | |
| edges: [...edges, ...newEdges], | |
| } | |
| } | |
| export const initialNodes = (originNodes: Node[], originEdges: Edge[]) => { | |
| const { nodes, edges } = preprocessNodesAndEdges(cloneDeep(originNodes), cloneDeep(originEdges)) | |
| const firstNode = nodes[0] | |
| if (!firstNode?.position) { | |
| nodes.forEach((node, index) => { | |
| node.position = { | |
| x: START_INITIAL_POSITION.x + index * NODE_WIDTH_X_OFFSET, | |
| y: START_INITIAL_POSITION.y, | |
| } | |
| }) | |
| } | |
| const iterationNodeMap = nodes.reduce((acc, node) => { | |
| if (node.parentId) { | |
| if (acc[node.parentId]) | |
| acc[node.parentId].push(node.id) | |
| else | |
| acc[node.parentId] = [node.id] | |
| } | |
| return acc | |
| }, {} as Record<string, string[]>) | |
| return nodes.map((node) => { | |
| if (!node.type) | |
| node.type = CUSTOM_NODE | |
| const connectedEdges = getConnectedEdges([node], edges) | |
| node.data._connectedSourceHandleIds = connectedEdges.filter(edge => edge.source === node.id).map(edge => edge.sourceHandle || 'source') | |
| node.data._connectedTargetHandleIds = connectedEdges.filter(edge => edge.target === node.id).map(edge => edge.targetHandle || 'target') | |
| if (node.data.type === BlockEnum.IfElse) { | |
| const nodeData = node.data as IfElseNodeType | |
| if (!nodeData.cases && nodeData.logical_operator && nodeData.conditions) { | |
| (node.data as IfElseNodeType).cases = [ | |
| { | |
| case_id: 'true', | |
| logical_operator: nodeData.logical_operator, | |
| conditions: nodeData.conditions, | |
| }, | |
| ] | |
| } | |
| node.data._targetBranches = branchNameCorrect([ | |
| ...(node.data as IfElseNodeType).cases.map(item => ({ id: item.case_id, name: '' })), | |
| { id: 'false', name: '' }, | |
| ]) | |
| } | |
| if (node.data.type === BlockEnum.QuestionClassifier) { | |
| node.data._targetBranches = (node.data as QuestionClassifierNodeType).classes.map((topic) => { | |
| return topic | |
| }) | |
| } | |
| if (node.data.type === BlockEnum.Iteration) { | |
| const iterationNodeData = node.data as IterationNodeType | |
| iterationNodeData._children = iterationNodeMap[node.id] || [] | |
| iterationNodeData.is_parallel = iterationNodeData.is_parallel || false | |
| iterationNodeData.parallel_nums = iterationNodeData.parallel_nums || 10 | |
| iterationNodeData.error_handle_mode = iterationNodeData.error_handle_mode || ErrorHandleMode.Terminated | |
| } | |
| return node | |
| }) | |
| } | |
| export const initialEdges = (originEdges: Edge[], originNodes: Node[]) => { | |
| const { nodes, edges } = preprocessNodesAndEdges(cloneDeep(originNodes), cloneDeep(originEdges)) | |
| let selectedNode: Node | null = null | |
| const nodesMap = nodes.reduce((acc, node) => { | |
| acc[node.id] = node | |
| if (node.data?.selected) | |
| selectedNode = node | |
| return acc | |
| }, {} as Record<string, Node>) | |
| const cycleEdges = getCycleEdges(nodes, edges) | |
| return edges.filter((edge) => { | |
| return !cycleEdges.find(cycEdge => cycEdge.source === edge.source && cycEdge.target === edge.target) | |
| }).map((edge) => { | |
| edge.type = 'custom' | |
| if (!edge.sourceHandle) | |
| edge.sourceHandle = 'source' | |
| if (!edge.targetHandle) | |
| edge.targetHandle = 'target' | |
| if (!edge.data?.sourceType && edge.source && nodesMap[edge.source]) { | |
| edge.data = { | |
| ...edge.data, | |
| sourceType: nodesMap[edge.source].data.type!, | |
| } as any | |
| } | |
| if (!edge.data?.targetType && edge.target && nodesMap[edge.target]) { | |
| edge.data = { | |
| ...edge.data, | |
| targetType: nodesMap[edge.target].data.type!, | |
| } as any | |
| } | |
| if (selectedNode) { | |
| edge.data = { | |
| ...edge.data, | |
| _connectedNodeIsSelected: edge.source === selectedNode.id || edge.target === selectedNode.id, | |
| } as any | |
| } | |
| return edge | |
| }) | |
| } | |
| export const getLayoutByDagre = (originNodes: Node[], originEdges: Edge[]) => { | |
| const dagreGraph = new dagre.graphlib.Graph() | |
| dagreGraph.setDefaultEdgeLabel(() => ({})) | |
| const nodes = cloneDeep(originNodes).filter(node => !node.parentId && node.type === CUSTOM_NODE) | |
| const edges = cloneDeep(originEdges).filter(edge => !edge.data?.isInIteration) | |
| dagreGraph.setGraph({ | |
| rankdir: 'LR', | |
| align: 'UL', | |
| nodesep: 40, | |
| ranksep: 60, | |
| ranker: 'tight-tree', | |
| marginx: 30, | |
| marginy: 200, | |
| }) | |
| nodes.forEach((node) => { | |
| dagreGraph.setNode(node.id, { | |
| width: node.width!, | |
| height: node.height!, | |
| }) | |
| }) | |
| edges.forEach((edge) => { | |
| dagreGraph.setEdge(edge.source, edge.target) | |
| }) | |
| dagre.layout(dagreGraph) | |
| return dagreGraph | |
| } | |
| export const canRunBySingle = (nodeType: BlockEnum) => { | |
| return nodeType === BlockEnum.LLM | |
| || nodeType === BlockEnum.KnowledgeRetrieval | |
| || nodeType === BlockEnum.Code | |
| || nodeType === BlockEnum.TemplateTransform | |
| || nodeType === BlockEnum.QuestionClassifier | |
| || nodeType === BlockEnum.HttpRequest | |
| || nodeType === BlockEnum.Tool | |
| || nodeType === BlockEnum.ParameterExtractor | |
| || nodeType === BlockEnum.Iteration | |
| } | |
| type ConnectedSourceOrTargetNodesChange = { | |
| type: string | |
| edge: Edge | |
| }[] | |
| export const getNodesConnectedSourceOrTargetHandleIdsMap = (changes: ConnectedSourceOrTargetNodesChange, nodes: Node[]) => { | |
| const nodesConnectedSourceOrTargetHandleIdsMap = {} as Record<string, any> | |
| changes.forEach((change) => { | |
| const { | |
| edge, | |
| type, | |
| } = change | |
| const sourceNode = nodes.find(node => node.id === edge.source)! | |
| 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)! | |
| if (targetNode) { | |
| nodesConnectedSourceOrTargetHandleIdsMap[targetNode.id] = nodesConnectedSourceOrTargetHandleIdsMap[targetNode.id] || { | |
| _connectedSourceHandleIds: [...(targetNode?.data._connectedSourceHandleIds || [])], | |
| _connectedTargetHandleIds: [...(targetNode?.data._connectedTargetHandleIds || [])], | |
| } | |
| } | |
| if (sourceNode) { | |
| if (type === 'remove') { | |
| const index = nodesConnectedSourceOrTargetHandleIdsMap[sourceNode.id]._connectedSourceHandleIds.findIndex((handleId: string) => handleId === edge.sourceHandle) | |
| nodesConnectedSourceOrTargetHandleIdsMap[sourceNode.id]._connectedSourceHandleIds.splice(index, 1) | |
| } | |
| if (type === 'add') | |
| nodesConnectedSourceOrTargetHandleIdsMap[sourceNode.id]._connectedSourceHandleIds.push(edge.sourceHandle || 'source') | |
| } | |
| if (targetNode) { | |
| if (type === 'remove') { | |
| const index = nodesConnectedSourceOrTargetHandleIdsMap[targetNode.id]._connectedTargetHandleIds.findIndex((handleId: string) => handleId === edge.targetHandle) | |
| nodesConnectedSourceOrTargetHandleIdsMap[targetNode.id]._connectedTargetHandleIds.splice(index, 1) | |
| } | |
| if (type === 'add') | |
| nodesConnectedSourceOrTargetHandleIdsMap[targetNode.id]._connectedTargetHandleIds.push(edge.targetHandle || 'target') | |
| } | |
| }) | |
| return nodesConnectedSourceOrTargetHandleIdsMap | |
| } | |
| export const genNewNodeTitleFromOld = (oldTitle: string) => { | |
| const regex = /^(.+?)\s*\((\d+)\)\s*$/ | |
| const match = oldTitle.match(regex) | |
| if (match) { | |
| const title = match[1] | |
| const num = parseInt(match[2], 10) | |
| return `${title} (${num + 1})` | |
| } | |
| else { | |
| return `${oldTitle} (1)` | |
| } | |
| } | |
| export const getValidTreeNodes = (nodes: Node[], edges: Edge[]) => { | |
| const startNode = nodes.find(node => node.data.type === BlockEnum.Start) | |
| if (!startNode) { | |
| return { | |
| validNodes: [], | |
| maxDepth: 0, | |
| } | |
| } | |
| const list: Node[] = [startNode] | |
| let maxDepth = 1 | |
| const traverse = (root: Node, depth: number) => { | |
| if (depth > maxDepth) | |
| maxDepth = depth | |
| const outgoers = getOutgoers(root, nodes, edges) | |
| if (outgoers.length) { | |
| outgoers.forEach((outgoer) => { | |
| list.push(outgoer) | |
| if (outgoer.data.type === BlockEnum.Iteration) | |
| list.push(...nodes.filter(node => node.parentId === outgoer.id)) | |
| traverse(outgoer, depth + 1) | |
| }) | |
| } | |
| else { | |
| list.push(root) | |
| if (root.data.type === BlockEnum.Iteration) | |
| list.push(...nodes.filter(node => node.parentId === root.id)) | |
| } | |
| } | |
| traverse(startNode, maxDepth) | |
| return { | |
| validNodes: uniqBy(list, 'id'), | |
| maxDepth, | |
| } | |
| } | |
| export const getToolCheckParams = ( | |
| toolData: ToolNodeType, | |
| buildInTools: ToolWithProvider[], | |
| customTools: ToolWithProvider[], | |
| workflowTools: ToolWithProvider[], | |
| language: string, | |
| ) => { | |
| const { provider_id, provider_type, tool_name } = toolData | |
| const isBuiltIn = provider_type === CollectionType.builtIn | |
| const currentTools = provider_type === CollectionType.builtIn ? buildInTools : provider_type === CollectionType.custom ? customTools : workflowTools | |
| const currCollection = currentTools.find(item => item.id === provider_id) | |
| const currTool = currCollection?.tools.find(tool => tool.name === tool_name) | |
| const formSchemas = currTool ? toolParametersToFormSchemas(currTool.parameters) : [] | |
| const toolInputVarSchema = formSchemas.filter((item: any) => item.form === 'llm') | |
| const toolSettingSchema = formSchemas.filter((item: any) => item.form !== 'llm') | |
| return { | |
| toolInputsSchema: (() => { | |
| const formInputs: InputVar[] = [] | |
| toolInputVarSchema.forEach((item: any) => { | |
| formInputs.push({ | |
| label: item.label[language] || item.label.en_US, | |
| variable: item.variable, | |
| type: item.type, | |
| required: item.required, | |
| }) | |
| }) | |
| return formInputs | |
| })(), | |
| notAuthed: isBuiltIn && !!currCollection?.allow_delete && !currCollection?.is_team_authorization, | |
| toolSettingSchema, | |
| language, | |
| } | |
| } | |
| export const changeNodesAndEdgesId = (nodes: Node[], edges: Edge[]) => { | |
| const idMap = nodes.reduce((acc, node) => { | |
| acc[node.id] = uuid4() | |
| return acc | |
| }, {} as Record<string, string>) | |
| const newNodes = nodes.map((node) => { | |
| return { | |
| ...node, | |
| id: idMap[node.id], | |
| } | |
| }) | |
| const newEdges = edges.map((edge) => { | |
| return { | |
| ...edge, | |
| source: idMap[edge.source], | |
| target: idMap[edge.target], | |
| } | |
| }) | |
| return [newNodes, newEdges] as [Node[], Edge[]] | |
| } | |
| export const isMac = () => { | |
| return navigator.userAgent.toUpperCase().includes('MAC') | |
| } | |
| const specialKeysNameMap: Record<string, string | undefined> = { | |
| ctrl: '⌘', | |
| alt: '⌥', | |
| } | |
| export const getKeyboardKeyNameBySystem = (key: string) => { | |
| if (isMac()) | |
| return specialKeysNameMap[key] || key | |
| return key | |
| } | |
| const specialKeysCodeMap: Record<string, string | undefined> = { | |
| ctrl: 'meta', | |
| } | |
| export const getKeyboardKeyCodeBySystem = (key: string) => { | |
| if (isMac()) | |
| return specialKeysCodeMap[key] || key | |
| return key | |
| } | |
| export const getTopLeftNodePosition = (nodes: Node[]) => { | |
| let minX = Infinity | |
| let minY = Infinity | |
| nodes.forEach((node) => { | |
| if (node.position.x < minX) | |
| minX = node.position.x | |
| if (node.position.y < minY) | |
| minY = node.position.y | |
| }) | |
| return { | |
| x: minX, | |
| y: minY, | |
| } | |
| } | |
| export const isEventTargetInputArea = (target: HTMLElement) => { | |
| if (target.tagName === 'INPUT' || target.tagName === 'TEXTAREA') | |
| return true | |
| if (target.contentEditable === 'true') | |
| return true | |
| } | |
| export const variableTransformer = (v: ValueSelector | string) => { | |
| if (typeof v === 'string') | |
| return v.replace(/^{{#|#}}$/g, '').split('.') | |
| return `{{#${v.join('.')}#}}` | |
| } | |
| type ParallelInfoItem = { | |
| parallelNodeId: string | |
| depth: number | |
| isBranch?: boolean | |
| } | |
| type NodeParallelInfo = { | |
| parallelNodeId: string | |
| edgeHandleId: string | |
| depth: number | |
| } | |
| type NodeHandle = { | |
| node: Node | |
| handle: string | |
| } | |
| type NodeStreamInfo = { | |
| upstreamNodes: Set<string> | |
| downstreamEdges: Set<string> | |
| } | |
| export const getParallelInfo = (nodes: Node[], edges: Edge[], parentNodeId?: string) => { | |
| let startNode | |
| if (parentNodeId) { | |
| const parentNode = nodes.find(node => node.id === parentNodeId) | |
| if (!parentNode) | |
| throw new Error('Parent node not found') | |
| startNode = nodes.find(node => node.id === (parentNode.data as IterationNodeType).start_node_id) | |
| } | |
| else { | |
| startNode = nodes.find(node => node.data.type === BlockEnum.Start) | |
| } | |
| if (!startNode) | |
| throw new Error('Start node not found') | |
| const parallelList = [] as ParallelInfoItem[] | |
| const nextNodeHandles = [{ node: startNode, handle: 'source' }] | |
| let hasAbnormalEdges = false | |
| const traverse = (firstNodeHandle: NodeHandle) => { | |
| const nodeEdgesSet = {} as Record<string, Set<string>> | |
| const totalEdgesSet = new Set<string>() | |
| const nextHandles = [firstNodeHandle] | |
| const streamInfo = {} as Record<string, NodeStreamInfo> | |
| const parallelListItem = { | |
| parallelNodeId: '', | |
| depth: 0, | |
| } as ParallelInfoItem | |
| const nodeParallelInfoMap = {} as Record<string, NodeParallelInfo> | |
| nodeParallelInfoMap[firstNodeHandle.node.id] = { | |
| parallelNodeId: '', | |
| edgeHandleId: '', | |
| depth: 0, | |
| } | |
| while (nextHandles.length) { | |
| const currentNodeHandle = nextHandles.shift()! | |
| const { node: currentNode, handle: currentHandle = 'source' } = currentNodeHandle | |
| const currentNodeHandleKey = currentNode.id | |
| const connectedEdges = edges.filter(edge => edge.source === currentNode.id && edge.sourceHandle === currentHandle) | |
| const connectedEdgesLength = connectedEdges.length | |
| const outgoers = nodes.filter(node => connectedEdges.some(edge => edge.target === node.id)) | |
| const incomers = getIncomers(currentNode, nodes, edges) | |
| if (!streamInfo[currentNodeHandleKey]) { | |
| streamInfo[currentNodeHandleKey] = { | |
| upstreamNodes: new Set<string>(), | |
| downstreamEdges: new Set<string>(), | |
| } | |
| } | |
| if (nodeEdgesSet[currentNodeHandleKey]?.size > 0 && incomers.length > 1) { | |
| const newSet = new Set<string>() | |
| for (const item of totalEdgesSet) { | |
| if (!streamInfo[currentNodeHandleKey].downstreamEdges.has(item)) | |
| newSet.add(item) | |
| } | |
| if (isEqual(nodeEdgesSet[currentNodeHandleKey], newSet)) { | |
| parallelListItem.depth = nodeParallelInfoMap[currentNode.id].depth | |
| nextNodeHandles.push({ node: currentNode, handle: currentHandle }) | |
| break | |
| } | |
| } | |
| if (nodeParallelInfoMap[currentNode.id].depth > parallelListItem.depth) | |
| parallelListItem.depth = nodeParallelInfoMap[currentNode.id].depth | |
| outgoers.forEach((outgoer) => { | |
| const outgoerConnectedEdges = getConnectedEdges([outgoer], edges).filter(edge => edge.source === outgoer.id) | |
| const sourceEdgesGroup = groupBy(outgoerConnectedEdges, 'sourceHandle') | |
| const incomers = getIncomers(outgoer, nodes, edges) | |
| if (outgoers.length > 1 && incomers.length > 1) | |
| hasAbnormalEdges = true | |
| Object.keys(sourceEdgesGroup).forEach((sourceHandle) => { | |
| nextHandles.push({ node: outgoer, handle: sourceHandle }) | |
| }) | |
| if (!outgoerConnectedEdges.length) | |
| nextHandles.push({ node: outgoer, handle: 'source' }) | |
| const outgoerKey = outgoer.id | |
| if (!nodeEdgesSet[outgoerKey]) | |
| nodeEdgesSet[outgoerKey] = new Set<string>() | |
| if (nodeEdgesSet[currentNodeHandleKey]) { | |
| for (const item of nodeEdgesSet[currentNodeHandleKey]) | |
| nodeEdgesSet[outgoerKey].add(item) | |
| } | |
| if (!streamInfo[outgoerKey]) { | |
| streamInfo[outgoerKey] = { | |
| upstreamNodes: new Set<string>(), | |
| downstreamEdges: new Set<string>(), | |
| } | |
| } | |
| if (!nodeParallelInfoMap[outgoer.id]) { | |
| nodeParallelInfoMap[outgoer.id] = { | |
| ...nodeParallelInfoMap[currentNode.id], | |
| } | |
| } | |
| if (connectedEdgesLength > 1) { | |
| const edge = connectedEdges.find(edge => edge.target === outgoer.id)! | |
| nodeEdgesSet[outgoerKey].add(edge.id) | |
| totalEdgesSet.add(edge.id) | |
| streamInfo[currentNodeHandleKey].downstreamEdges.add(edge.id) | |
| streamInfo[outgoerKey].upstreamNodes.add(currentNodeHandleKey) | |
| for (const item of streamInfo[currentNodeHandleKey].upstreamNodes) | |
| streamInfo[item].downstreamEdges.add(edge.id) | |
| if (!parallelListItem.parallelNodeId) | |
| parallelListItem.parallelNodeId = currentNode.id | |
| const prevDepth = nodeParallelInfoMap[currentNode.id].depth + 1 | |
| const currentDepth = nodeParallelInfoMap[outgoer.id].depth | |
| nodeParallelInfoMap[outgoer.id].depth = Math.max(prevDepth, currentDepth) | |
| } | |
| else { | |
| for (const item of streamInfo[currentNodeHandleKey].upstreamNodes) | |
| streamInfo[outgoerKey].upstreamNodes.add(item) | |
| nodeParallelInfoMap[outgoer.id].depth = nodeParallelInfoMap[currentNode.id].depth | |
| } | |
| }) | |
| } | |
| parallelList.push(parallelListItem) | |
| } | |
| while (nextNodeHandles.length) { | |
| const nodeHandle = nextNodeHandles.shift()! | |
| traverse(nodeHandle) | |
| } | |
| return { | |
| parallelList, | |
| hasAbnormalEdges, | |
| } | |
| } | |