feat: add prev and next nodes

This commit is contained in:
Joel
2024-03-14 20:29:47 +08:00
parent 64e44d1709
commit ae6a558662
14 changed files with 85 additions and 47 deletions

View File

@ -1,16 +1,19 @@
import type { NodeDefault } from '../../types'
import type { ToolNodeType } from './types'
import { ALL_CHAT_AVAILABLE_BLOCKS, ALL_COMPLETION_AVAILABLE_BLOCKS } from '@/app/components/workflow/constants'
const nodeDefault: NodeDefault<ToolNodeType> = {
defaultValue: {
tool_parameters: [],
tool_configurations: {},
},
getAvailablePrevNodes() {
return []
getAvailablePrevNodes(isChatMode: boolean) {
const nodes = isChatMode ? ALL_CHAT_AVAILABLE_BLOCKS : ALL_COMPLETION_AVAILABLE_BLOCKS
return nodes
},
getAvailableNextNodes() {
return []
getAvailableNextNodes(isChatMode: boolean) {
const nodes = isChatMode ? ALL_CHAT_AVAILABLE_BLOCKS : ALL_COMPLETION_AVAILABLE_BLOCKS
return nodes
},
}