100 lines
3.1 KiB
TypeScript
100 lines
3.1 KiB
TypeScript
|
'use client'
|
||
|
|
|
||
|
|
import { useCallback, useMemo } from 'react'
|
||
|
|
import { useTranslation } from 'react-i18next'
|
||
|
|
import { useParams, useRouter } from 'next/navigation'
|
||
|
|
import {
|
||
|
|
RiBook2Fill,
|
||
|
|
RiBook2Line,
|
||
|
|
} from '@remixicon/react'
|
||
|
|
import { flatten } from 'lodash-es'
|
||
|
|
import Nav from '../nav'
|
||
|
|
import type { NavItem } from '../nav/nav-selector'
|
||
|
|
import { basePath } from '@/utils/var'
|
||
|
|
import { useDatasetDetail, useDatasetList } from '@/service/knowledge/use-dataset'
|
||
|
|
import type { DataSet } from '@/models/datasets'
|
||
|
|
|
||
|
|
const DatasetNav = () => {
|
||
|
|
const { t } = useTranslation()
|
||
|
|
const router = useRouter()
|
||
|
|
const { datasetId } = useParams()
|
||
|
|
const { data: currentDataset } = useDatasetDetail(datasetId as string)
|
||
|
|
const {
|
||
|
|
data: datasetList,
|
||
|
|
fetchNextPage,
|
||
|
|
hasNextPage,
|
||
|
|
} = useDatasetList({
|
||
|
|
initialPage: 1,
|
||
|
|
limit: 30,
|
||
|
|
})
|
||
|
|
const datasetItems = flatten(datasetList?.pages.map(datasetData => datasetData.data))
|
||
|
|
|
||
|
|
const curNav = useMemo(() => {
|
||
|
|
if (!currentDataset) return
|
||
|
|
return {
|
||
|
|
id: currentDataset.id,
|
||
|
|
name: currentDataset.name,
|
||
|
|
icon: currentDataset.icon_info.icon,
|
||
|
|
icon_type: currentDataset.icon_info.icon_type,
|
||
|
|
icon_background: currentDataset.icon_info.icon_background,
|
||
|
|
icon_url: currentDataset.icon_info.icon_url,
|
||
|
|
} as Omit<NavItem, 'link'>
|
||
|
|
}, [currentDataset?.id, currentDataset?.name, currentDataset?.icon_info])
|
||
|
|
|
||
|
|
const getDatasetLink = useCallback((dataset: DataSet) => {
|
||
|
|
const isPipelineUnpublished = dataset.runtime_mode === 'rag_pipeline' && !dataset.is_published
|
||
|
|
const link = isPipelineUnpublished
|
||
|
|
? `/datasets/${dataset.id}/pipeline`
|
||
|
|
: `/datasets/${dataset.id}/documents`
|
||
|
|
return dataset.provider === 'external'
|
||
|
|
? `/datasets/${dataset.id}/hitTesting`
|
||
|
|
: link
|
||
|
|
}, [])
|
||
|
|
|
||
|
|
const navigationItems = useMemo(() => {
|
||
|
|
return datasetItems.map((dataset) => {
|
||
|
|
const link = getDatasetLink(dataset)
|
||
|
|
return {
|
||
|
|
id: dataset.id,
|
||
|
|
name: dataset.name,
|
||
|
|
link,
|
||
|
|
icon: dataset.icon_info.icon,
|
||
|
|
icon_type: dataset.icon_info.icon_type,
|
||
|
|
icon_background: dataset.icon_info.icon_background,
|
||
|
|
icon_url: dataset.icon_info.icon_url,
|
||
|
|
}
|
||
|
|
}) as NavItem[]
|
||
|
|
}, [datasetItems, getDatasetLink])
|
||
|
|
|
||
|
|
const createRoute = useMemo(() => {
|
||
|
|
const runtimeMode = currentDataset?.runtime_mode
|
||
|
|
if (runtimeMode === 'rag_pipeline')
|
||
|
|
return `${basePath}/datasets/create-from-pipeline`
|
||
|
|
else
|
||
|
|
return `${basePath}/datasets/create`
|
||
|
|
}, [currentDataset?.runtime_mode])
|
||
|
|
|
||
|
|
const handleLoadMore = useCallback(() => {
|
||
|
|
if (hasNextPage)
|
||
|
|
fetchNextPage()
|
||
|
|
}, [hasNextPage, fetchNextPage])
|
||
|
|
|
||
|
|
return (
|
||
|
|
<Nav
|
||
|
|
isApp={false}
|
||
|
|
icon={<RiBook2Line className='h-4 w-4' />}
|
||
|
|
activeIcon={<RiBook2Fill className='h-4 w-4' />}
|
||
|
|
text={t('common.menus.datasets')}
|
||
|
|
activeSegment='datasets'
|
||
|
|
link='/datasets'
|
||
|
|
curNav={curNav}
|
||
|
|
navigationItems={navigationItems}
|
||
|
|
createText={t('common.menus.newDataset')}
|
||
|
|
onCreate={() => router.push(createRoute)}
|
||
|
|
onLoadMore={handleLoadMore}
|
||
|
|
/>
|
||
|
|
)
|
||
|
|
}
|
||
|
|
|
||
|
|
export default DatasetNav
|