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

This commit is contained in:
zxhlyh
2024-05-17 11:42:03 +08:00
committed by GitHub
parent 0ac5d621b6
commit de3a7603ac
3 changed files with 81 additions and 11 deletions

View File

@@ -101,9 +101,40 @@ export const useSelectionInteractions = () => {
setNodes(newNodes)
}, [store, workflowStore])
const handleSelectionCancel = useCallback(() => {
const {
getNodes,
setNodes,
edges,
setEdges,
} = store.getState()
store.setState({
userSelectionRect: null,
userSelectionActive: true,
})
const nodes = getNodes()
const newNodes = produce(nodes, (draft) => {
draft.forEach((node) => {
if (node.data._isBundled)
node.data._isBundled = false
})
})
setNodes(newNodes)
const newEdges = produce(edges, (draft) => {
draft.forEach((edge) => {
if (edge.data._isBundled)
edge.data._isBundled = false
})
})
setEdges(newEdges)
}, [store])
return {
handleSelectionStart,
handleSelectionChange,
handleSelectionDrag,
handleSelectionCancel,
}
}