jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: workflow add next node from knowledge retrieval node (#4467)

Browse source
This commit is contained in:
zxhlyh 2024-05-17 11:42:03 +08:00 • committed by GitHub
commit de3a7603ac
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 81 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

14
web/app/components/workflow/hooks/use-nodes-interactions.ts
View file

@ -466,8 +466,7 @@ export const useNodesInteractions = () => {
const prevNode = nodes[prevNodeIndex]
const outgoers = getOutgoers(prevNode, nodes, edges).sort((a, b) => a.position.y - b.position.y)
const lastOutgoer = outgoers[outgoers.length - 1]
if (prevNode.data.type === BlockEnum.KnowledgeRetrieval)
targetHandle = prevNodeId
newNode.data._connectedTargetHandleIds = [targetHandle]
newNode.data._connectedSourceHandleIds = []
newNode.position = {
@ -593,8 +592,7 @@ export const useNodesInteractions = () => {
if (prevNodeId && nextNodeId) {
const prevNode = nodes.find(node => node.id === prevNodeId)!
const nextNode = nodes.find(node => node.id === nextNodeId)!
if (prevNode.data.type === BlockEnum.KnowledgeRetrieval)
targetHandle = prevNodeId
newNode.data._connectedTargetHandleIds = [targetHandle]
newNode.data._connectedSourceHandleIds = [sourceHandle]
newNode.position = {
@ -925,10 +923,6 @@ export const useNodesInteractions = () => {
edges,
} = store.getState()
const edgeSelected = edges.some(edge => edge.selected)
if (edgeSelected)
return
const nodes = getNodes()
const bundledNodes = nodes.filter(node => node.data._isBundled && node.data.type !== BlockEnum.Start)
@ -937,6 +931,10 @@ export const useNodesInteractions = () => {
return
}
const edgeSelected = edges.some(edge => edge.selected)
if (edgeSelected)
return
const selectedNode = nodes.find(node => node.data.selected && node.data.type !== BlockEnum.Start)
if (selectedNode)