feat: Enhance dataset info and card components with memoization for improved performance

This commit is contained in:
twwu 2025-07-29 13:46:17 +08:00
parent 657e813c7f
commit c2cbdcd3bf
3 changed files with 17 additions and 4 deletions

View File

@ -1,6 +1,6 @@
'use client' 'use client'
import type { FC } from 'react' import type { FC } from 'react'
import React from 'react' import React, { useMemo } from 'react'
import { useTranslation } from 'react-i18next' import { useTranslation } from 'react-i18next'
import AppIcon from '../../base/app-icon' import AppIcon from '../../base/app-icon'
import Effect from '../../base/effect' import Effect from '../../base/effect'
@ -27,6 +27,9 @@ const DatasetInfo: FC<DatasetInfoProps> = ({
icon_url: '', icon_url: '',
} }
const isExternalProvider = dataset.provider === 'external' 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() const { formatIndexingTechniqueAndMethod } = useKnowledge()
return ( return (
@ -54,7 +57,7 @@ const DatasetInfo: FC<DatasetInfoProps> = ({
</div> </div>
<div className='system-2xs-medium-uppercase text-text-tertiary'> <div className='system-2xs-medium-uppercase text-text-tertiary'>
{isExternalProvider && t('dataset.externalTag')} {isExternalProvider && t('dataset.externalTag')}
{!isExternalProvider && dataset.doc_form && dataset.indexing_technique && ( {!isExternalProvider && isPipelinePublished && dataset.doc_form && dataset.indexing_technique && (
<div className='flex items-center gap-x-2'> <div className='flex items-center gap-x-2'>
<span>{t(`dataset.chunkingMode.${DOC_FORM_TEXT[dataset.doc_form]}`)}</span> <span>{t(`dataset.chunkingMode.${DOC_FORM_TEXT[dataset.doc_form]}`)}</span>
<span>{formatIndexingTechniqueAndMethod(dataset.indexing_technique, dataset.retrieval_model_dict?.search_method)}</span> <span>{formatIndexingTechniqueAndMethod(dataset.indexing_technique, dataset.retrieval_model_dict?.search_method)}</span>

View File

@ -56,6 +56,13 @@ const DatasetCard = ({
const isPipelineUnpublished = useMemo(() => { const isPipelineUnpublished = useMemo(() => {
return dataset.runtime_mode === 'rag_pipeline' && !dataset.is_published return dataset.runtime_mode === 'rag_pipeline' && !dataset.is_published
}, [dataset.runtime_mode, dataset.is_published]) }, [dataset.runtime_mode, dataset.is_published])
const isShowChunkingModeIcon = useMemo(() => {
return dataset.doc_form && (dataset.runtime_mode !== 'rag_pipeline' || dataset.is_published)
}, [dataset.doc_form, dataset.runtime_mode, dataset.is_published])
const isShowDocModeInfo = useMemo(() => {
return dataset.doc_form && dataset.indexing_technique && dataset.retrieval_model_dict?.search_method && (dataset.runtime_mode !== 'rag_pipeline' || dataset.is_published)
}, [dataset.doc_form, dataset.indexing_technique, dataset.retrieval_model_dict?.search_method, dataset.runtime_mode, dataset.is_published])
const chunkingModeIcon = dataset.doc_form ? DOC_FORM_ICON_WITH_BG[dataset.doc_form] : React.Fragment const chunkingModeIcon = dataset.doc_form ? DOC_FORM_ICON_WITH_BG[dataset.doc_form] : React.Fragment
const Icon = isExternalProvider ? DOC_FORM_ICON_WITH_BG.external : chunkingModeIcon const Icon = isExternalProvider ? DOC_FORM_ICON_WITH_BG.external : chunkingModeIcon
const iconInfo = dataset.icon_info || { const iconInfo = dataset.icon_info || {
@ -177,7 +184,7 @@ const DatasetCard = ({
background={iconInfo.icon_type === 'image' ? undefined : iconInfo.icon_background} background={iconInfo.icon_type === 'image' ? undefined : iconInfo.icon_background}
imageUrl={iconInfo.icon_type === 'image' ? iconInfo.icon_url : undefined} imageUrl={iconInfo.icon_type === 'image' ? iconInfo.icon_url : undefined}
/> />
{(dataset.doc_form || isExternalProvider) && ( {(isShowChunkingModeIcon || isExternalProvider) && (
<div className='absolute -bottom-1 -right-1 z-[5]'> <div className='absolute -bottom-1 -right-1 z-[5]'>
<Icon className='size-4' /> <Icon className='size-4' />
</div> </div>
@ -192,7 +199,7 @@ const DatasetCard = ({
</div> </div>
<div className='system-2xs-medium-uppercase flex items-center gap-x-3 text-text-tertiary'> <div className='system-2xs-medium-uppercase flex items-center gap-x-3 text-text-tertiary'>
{isExternalProvider && <span>{t('dataset.externalKnowledgeBase')}</span>} {isExternalProvider && <span>{t('dataset.externalKnowledgeBase')}</span>}
{!isExternalProvider && ( {!isExternalProvider && isShowDocModeInfo && (
<> <>
{dataset.doc_form && <span>{t(`dataset.chunkingMode.${DOC_FORM_TEXT[dataset.doc_form]}`)}</span>} {dataset.doc_form && <span>{t(`dataset.chunkingMode.${DOC_FORM_TEXT[dataset.doc_form]}`)}</span>}
{dataset.indexing_technique && <span>{formatIndexingTechniqueAndMethod(dataset.indexing_technique, dataset.retrieval_model_dict?.search_method)}</span>} {dataset.indexing_technique && <span>{formatIndexingTechniqueAndMethod(dataset.indexing_technique, dataset.retrieval_model_dict?.search_method)}</span>}

View File

@ -38,6 +38,7 @@ import {
import Confirm from '@/app/components/base/confirm' import Confirm from '@/app/components/base/confirm'
import PublishAsKnowledgePipelineModal from '../../publish-as-knowledge-pipeline-modal' import PublishAsKnowledgePipelineModal from '../../publish-as-knowledge-pipeline-modal'
import type { IconInfo } from '@/models/datasets' import type { IconInfo } from '@/models/datasets'
import { useResetDatasetList } from '@/service/knowledge/use-dataset'
const PUBLISH_SHORTCUT = ['ctrl', '⇧', 'P'] const PUBLISH_SHORTCUT = ['ctrl', '⇧', 'P']
@ -76,6 +77,7 @@ const Popup = () => {
}] = useBoolean(false) }] = useBoolean(false)
const invalidPublishedPipelineInfo = useInvalid([...publishedPipelineInfoQueryKeyPrefix, pipelineId]) const invalidPublishedPipelineInfo = useInvalid([...publishedPipelineInfoQueryKeyPrefix, pipelineId])
const resetDatasetList = useResetDatasetList()
const handlePublish = useCallback(async (params?: PublishWorkflowParams) => { const handlePublish = useCallback(async (params?: PublishWorkflowParams) => {
if (publishing) if (publishing)
@ -100,6 +102,7 @@ const Popup = () => {
workflowStore.getState().setPublishedAt(res.created_at) workflowStore.getState().setPublishedAt(res.created_at)
mutateDatasetRes?.() mutateDatasetRes?.()
invalidPublishedPipelineInfo() invalidPublishedPipelineInfo()
resetDatasetList()
} }
} }
} }