mirror of https://github.com/langgenius/dify.git
30 lines
893 B
TypeScript
30 lines
893 B
TypeScript
import { useCallback } from 'react'
|
|
import produce from 'immer'
|
|
import useVarList from './components/var-list/use-var-list'
|
|
import type { VariableAssignerNodeType } from './types'
|
|
import useNodeCrud from '@/app/components/workflow/nodes/_base/hooks/use-node-crud'
|
|
|
|
const useConfig = (id: string, payload: VariableAssignerNodeType) => {
|
|
const { inputs, setInputs } = useNodeCrud<VariableAssignerNodeType>(id, payload)
|
|
|
|
const handleOutputTypeChange = useCallback((outputType: string) => {
|
|
const newInputs = produce(inputs, (draft: VariableAssignerNodeType) => {
|
|
draft.output_type = outputType
|
|
})
|
|
setInputs(newInputs)
|
|
}, [inputs, setInputs])
|
|
|
|
const { handleVarListChange, handleAddVariable } = useVarList({
|
|
inputs,
|
|
setInputs,
|
|
})
|
|
return {
|
|
inputs,
|
|
handleOutputTypeChange,
|
|
handleVarListChange,
|
|
handleAddVariable,
|
|
}
|
|
}
|
|
|
|
export default useConfig
|