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

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
}