youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-04 17:38:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
StyleZhang
2024-03-12 18:47:02 +08:00
parent 30ea3cb702
commit 90e013554c
5 changed files with 116 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files

55
web/app/components/workflow/utils.ts
View File

@ -170,58 +170,3 @@ export const canRunBySingle = (nodeType: BlockEnum) => {
|| nodeType === BlockEnum.HttpRequest
|| nodeType === BlockEnum.Tool
}
export const getTreeLeafNodes = (nodes: Node[], edges: Edge[]) => {
const startNode = nodes.find(node => node.data.type === BlockEnum.Start)
if (!startNode)
return []
const list: Node[] = []
const preOrder = (root: Node, callback: (node: Node) => void) => {
const outgoers = getOutgoers(root, nodes, edges)
if (outgoers.length) {
outgoers.forEach((outgoer) => {
preOrder(outgoer, callback)
})
}
else {
callback(root)
}
}
preOrder(startNode, (node) => {
list.push(node)
})
return list
}
export const getBeforeNodesInSameBranch = (nodeId: string, targetHandle: string, nodes: Node[], edges: Edge[]) => {
const currentNode = nodes.find(node => node.id === nodeId)!
const list: Node[] = []
const traverse = (root: Node, callback: (node: Node) => void) => {
const connectedEdges = getConnectedEdges([root], edges)
const sourceEdge = connectedEdges.filter(edge => edge.targetHandle === targetHandle)
const sourceEdgeLength = sourceEdge.length
if (sourceEdgeLength === 1) {
const before = nodes.find(node => node.id === sourceEdge[0].source)
if (before) {
callback(before)
traverse(before, callback)
}
}
}
traverse(currentNode, (node) => {
list.push(node)
})
const length = list.length
if (length && list[length - 1].data.type === BlockEnum.Start)
return list.reverse()
return []
}