available nodes

This commit is contained in:
StyleZhang
2024-03-15 13:24:09 +08:00
parent 8d9c86ac4c
commit 5fbf8ee6c6
17 changed files with 165 additions and 38 deletions

View File

@ -1,4 +1,4 @@
import type { NodeDefault } from '../../types'
import { BlockEnum, type NodeDefault } from '../../types'
import { type IfElseNodeType, LogicalOperator } from './types'
import { ALL_CHAT_AVAILABLE_BLOCKS, ALL_COMPLETION_AVAILABLE_BLOCKS } from '@/app/components/workflow/constants'
@ -23,7 +23,7 @@ const nodeDefault: NodeDefault<IfElseNodeType> = {
},
getAvailableNextNodes(isChatMode: boolean) {
const nodes = isChatMode ? ALL_CHAT_AVAILABLE_BLOCKS : ALL_COMPLETION_AVAILABLE_BLOCKS
return nodes
return nodes.filter(type => type !== BlockEnum.VariableAssigner)
},
checkValid(payload: IfElseNodeType) {
let isValid = true