KMS/dify
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: prevent Loop and Iteration nodes from nesting within each other.

Browse Source
This commit is contained in:
woo0ood 2025-01-26 06:08:01 +00:00 committed by Wood
parent 3e18b5d1ae
commit 8d910c7f34
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
web/app/components/workflow/hooks/use-nodes-data.ts
View File

@ -48,17 +48,19 @@ export const useAvailableBlocks = (nodeType?: BlockEnum, isInIteration?: boolean
return useMemo(() => { return useMemo(() => {
return { return {
availablePrevBlocks: availablePrevBlocks.filter((nType) => { availablePrevBlocks: availablePrevBlocks.filter((nType) => {
if (isInIteration && (nType === BlockEnum.Iteration || nType === BlockEnum.End)) if (isInIteration && (nType === BlockEnum.Iteration || nType === BlockEnum.Loop || nType === BlockEnum.End))
return false return false
if (isInLoop && (nType === BlockEnum.Loop || nType === BlockEnum.End))
if (isInLoop && (nType === BlockEnum.Iteration || nType === BlockEnum.Loop || nType === BlockEnum.End))
return false return false
return true return true
}), }),
availableNextBlocks: availableNextBlocks.filter((nType) => { availableNextBlocks: availableNextBlocks.filter((nType) => {
if (isInIteration && (nType === BlockEnum.Iteration || nType === BlockEnum.End)) if (isInIteration && (nType === BlockEnum.Iteration || nType === BlockEnum.Loop || nType === BlockEnum.End))
return false return false
if (isInLoop && (nType === BlockEnum.Loop || nType === BlockEnum.End))
if (isInLoop && (nType === BlockEnum.Iteration || nType === BlockEnum.Loop || nType === BlockEnum.End))
return false return false
return true return true