dify/web/app/components/workflow/nodes/knowledge-retrieval/hooks.ts

15 lines
389 B
TypeScript

import type {
DataSet,
SelectedDatasetsMode,
} from '@/models/datasets'
import { useMemo } from 'react'
import { getSelectedDatasetsMode } from './utils'
export const useSelectedDatasetsMode = (datasets: DataSet[]) => {
const selectedDatasetsMode: SelectedDatasetsMode = useMemo(() => {
return getSelectedDatasetsMode(datasets)
}, [datasets])
return selectedDatasetsMode
}