diff --git a/frontend/src/pages/DataManagement/Detail/useFilesOperation.ts b/frontend/src/pages/DataManagement/Detail/useFilesOperation.ts index 9fcd99a..2cb6233 100644 --- a/frontend/src/pages/DataManagement/Detail/useFilesOperation.ts +++ b/frontend/src/pages/DataManagement/Detail/useFilesOperation.ts @@ -25,6 +25,7 @@ import { import { useParams } from "react-router"; const OFFICE_FILE_EXTENSIONS = [".doc", ".docx"]; +const TEXT_DATASET_TYPE_PREFIX = DatasetType.TEXT; const OFFICE_PREVIEW_POLL_INTERVAL = 2000; const OFFICE_PREVIEW_POLL_MAX_TIMES = 60; @@ -86,6 +87,13 @@ export function useFilesOperation(dataset: Dataset) { }; }, [clearOfficePreviewPolling]); + const isTextDataset = (datasetType?: string) => { + if (!datasetType) { + return false; + } + return datasetType.toUpperCase().startsWith(TEXT_DATASET_TYPE_PREFIX); + }; + const fetchFiles = async ( prefix?: string, current?: number, @@ -93,7 +101,7 @@ export function useFilesOperation(dataset: Dataset) { ) => { // 如果明确传了 prefix(包括空字符串),使用传入的值;否则使用当前 pagination.prefix const targetPrefix = prefix !== undefined ? prefix : (pagination.prefix || ''); - const shouldExcludeDerivedFiles = dataset?.datasetType === DatasetType.TEXT; + const shouldExcludeDerivedFiles = isTextDataset(dataset?.datasetType); const params: DatasetFilesQueryParams = { page: current !== undefined ? current : pagination.current,