92 lines
3.2 KiB
TypeScript
92 lines
3.2 KiB
TypeScript
'use client'
|
|
import type { FC } from 'react'
|
|
import React, { useMemo } from 'react'
|
|
import { useTranslation } from 'react-i18next'
|
|
import AppIcon from '../../base/app-icon'
|
|
import Effect from '../../base/effect'
|
|
import { useDatasetDetailContextWithSelector } from '@/context/dataset-detail'
|
|
import type { DataSet } from '@/models/datasets'
|
|
import { DOC_FORM_TEXT } from '@/models/datasets'
|
|
import { useKnowledge } from '@/hooks/use-knowledge'
|
|
import cn from '@/utils/classnames'
|
|
import Dropdown from './dropdown'
|
|
|
|
type DatasetInfoProps = {
|
|
expand: boolean
|
|
}
|
|
|
|
const DatasetInfo: FC<DatasetInfoProps> = ({
|
|
expand,
|
|
}) => {
|
|
const { t } = useTranslation()
|
|
const dataset = useDatasetDetailContextWithSelector(state => state.dataset) as DataSet
|
|
const iconInfo = dataset.icon_info || {
|
|
icon: '📙',
|
|
icon_type: 'emoji',
|
|
icon_background: '#FFF4ED',
|
|
icon_url: '',
|
|
}
|
|
const isExternalProvider = dataset.provider === 'external'
|
|
const isPipelinePublished = useMemo(() => {
|
|
return dataset.runtime_mode === 'rag_pipeline' && dataset.is_published
|
|
}, [dataset.runtime_mode, dataset.is_published])
|
|
const { formatIndexingTechniqueAndMethod } = useKnowledge()
|
|
|
|
return (
|
|
<div className={cn('relative flex flex-col', expand ? '' : 'p-1')}>
|
|
{expand && (
|
|
<Effect className='-left-5 top-[-22px] opacity-15' />
|
|
)}
|
|
|
|
<div className='flex flex-col gap-2 p-2'>
|
|
<div className='flex items-center gap-1'>
|
|
<div className={cn(!expand && '-ml-1')}>
|
|
<AppIcon
|
|
size={expand ? 'large' : 'small'}
|
|
iconType={iconInfo.icon_type}
|
|
icon={iconInfo.icon}
|
|
background={iconInfo.icon_background}
|
|
imageUrl={iconInfo.icon_url}
|
|
/>
|
|
</div>
|
|
{expand && (
|
|
<div className='ml-auto'>
|
|
<Dropdown expand />
|
|
</div>
|
|
)}
|
|
</div>
|
|
{!expand && (
|
|
<div className='-mb-2 -mt-1 flex items-center justify-center'>
|
|
<Dropdown expand={false} />
|
|
</div>
|
|
)}
|
|
{expand && (
|
|
<div className='flex flex-col gap-y-1 pb-0.5'>
|
|
<div
|
|
className='system-md-semibold truncate text-text-secondary'
|
|
title={dataset.name}
|
|
>
|
|
{dataset.name}
|
|
</div>
|
|
<div className='system-2xs-medium-uppercase text-text-tertiary'>
|
|
{isExternalProvider && t('dataset.externalTag')}
|
|
{!isExternalProvider && isPipelinePublished && dataset.doc_form && dataset.indexing_technique && (
|
|
<div className='flex items-center gap-x-2'>
|
|
<span>{t(`dataset.chunkingMode.${DOC_FORM_TEXT[dataset.doc_form]}`)}</span>
|
|
<span>{formatIndexingTechniqueAndMethod(dataset.indexing_technique, dataset.retrieval_model_dict?.search_method)}</span>
|
|
</div>
|
|
)}
|
|
</div>
|
|
{!!dataset.description && (
|
|
<p className='system-xs-regular line-clamp-3 text-text-tertiary first-letter:capitalize'>
|
|
{dataset.description}
|
|
</p>
|
|
)}
|
|
</div>
|
|
)}
|
|
</div>
|
|
</div>
|
|
)
|
|
}
|
|
export default React.memo(DatasetInfo)
|