dify/web/context/datasets-context.tsx

'use client'

import { createContext, useContext } from 'use-context-selector'
import type { DataSet } from '@/models/datasets'

export type DatasetsContextValue = {
  datasets: DataSet[]
  mutateDatasets: () => void
  currentDataset?: DataSet
}

const DatasetsContext = createContext<DatasetsContextValue>({
  datasets: [],
  mutateDatasets: () => {},
  currentDataset: undefined
})

export const useDatasetsContext = () => useContext(DatasetsContext)

export default DatasetsContext