fix: end node limit in next step (#4945)

This commit is contained in:
zxhlyh 2024-06-05 14:00:47 +08:00 committed by GitHub
commit a4041cb40b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 14 additions and 17 deletions

View file

@ -11,7 +11,7 @@ import {
generateNewNode,
} from '../utils'
import {
useNodesExtraData,
useAvailableBlocks,
useNodesReadOnly,
usePanelInteractions,
} from '../hooks'
@ -38,11 +38,10 @@ const AddBlock = ({
const { t } = useTranslation()
const store = useStoreApi()
const workflowStore = useWorkflowStore()
const nodesExtraData = useNodesExtraData()
const { nodesReadOnly } = useNodesReadOnly()
const { handlePaneContextmenuCancel } = usePanelInteractions()
const [open, setOpen] = useState(false)
const availableNextNodes = nodesExtraData[BlockEnum.Start].availableNextNodes
const { availableNextBlocks } = useAvailableBlocks(BlockEnum.Start, false)
const handleOpenChange = useCallback((open: boolean) => {
setOpen(open)
@ -102,7 +101,7 @@ const AddBlock = ({
}}
trigger={renderTrigger || renderTriggerElement}
popupClassName='!min-w-[256px]'
availableBlocksTypes={availableNextNodes}
availableBlocksTypes={availableNextBlocks}
/>
)
}