123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763 |
- 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,
- }
- }
|