dify/web/app/components/workflow/nodes/direct-answer/use-config.ts

30 lines
861 B
TypeScript

import { useCallback } from 'react'
import produce from 'immer'
import useVarList from '../_base/hooks/use-var-list'
import type { DirectAnswerNodeType } from './types'
import useNodeCrud from '@/app/components/workflow/nodes/_base/hooks/use-node-crud'
const useConfig = (id: string, payload: DirectAnswerNodeType) => {
const { inputs, setInputs } = useNodeCrud<DirectAnswerNodeType>(id, payload)
// variables
const { handleVarListChange, handleAddVariable } = useVarList<DirectAnswerNodeType>({
inputs,
setInputs,
})
const handleAnswerChange = useCallback((value: string) => {
const newInputs = produce(inputs, (draft) => {
draft.answer = value
})
setInputs(newInputs)
}, [inputs, setInputs])
return {
inputs,
handleVarListChange,
handleAddVariable,
handleAnswerChange,
}
}
export default useConfig