diff --git a/web/app/components/datasets/create-from-pipeline/create-options/create-from-scratch.tsx b/web/app/components/datasets/create-from-pipeline/create-options/create-from-scratch.tsx index c8828edf0c..17bcd1253c 100644 --- a/web/app/components/datasets/create-from-pipeline/create-options/create-from-scratch.tsx +++ b/web/app/components/datasets/create-from-pipeline/create-options/create-from-scratch.tsx @@ -8,12 +8,12 @@ import type { AppIconType } from '@/types/app' import { RiCloseLine } from '@remixicon/react' import PermissionSelector from '../../settings/permission-selector' import type { CreateDatasetReq } from '@/models/datasets' -import { DatasetPermission } from '@/models/datasets' +import { ChunkingMode, DatasetPermission } from '@/models/datasets' import { useMembers } from '@/service/use-common' import Button from '@/app/components/base/button' import { useTranslation } from 'react-i18next' import Toast from '@/app/components/base/toast' -import { useCreateDataset } from '@/service/knowledge/use-create-dataset' +import { useCreatePipelineDataset } from '@/service/knowledge/use-create-dataset' import type { Member } from '@/models/common' type CreateFromScratchProps = { @@ -75,7 +75,7 @@ const CreateFromScratch = ({ setPermission(value!) }, []) - const { mutateAsync: createEmptyDataset } = useCreateDataset() + const { mutateAsync: createEmptyDataset } = useCreatePipelineDataset() const handleCreate = useCallback(async () => { if (!name) { @@ -94,6 +94,7 @@ const CreateFromScratch = ({ icon_background: appIcon.type === 'image' ? undefined : appIcon.background, icon_url: appIcon.type === 'image' ? appIcon.url : undefined, }, + doc_form: ChunkingMode.text, permission, } // Handle permission diff --git a/web/app/components/datasets/list/dataset-card/index.tsx b/web/app/components/datasets/list/dataset-card/index.tsx index 60b00740b9..e8345fd844 100644 --- a/web/app/components/datasets/list/dataset-card/index.tsx +++ b/web/app/components/datasets/list/dataset-card/index.tsx @@ -51,6 +51,12 @@ const DatasetCard = ({ return dataset.provider === EXTERNAL_PROVIDER }, [dataset.provider]) const Icon = isExternalProvider ? DOC_FORM_ICON.external : (DOC_FORM_ICON[dataset.doc_form] || General) + const iconInfo = dataset.icon_info || { + icon: '📙', + icon_type: 'emoji', + icon_background: '#FFF4ED', + icon_url: '', + } const { formatIndexingTechniqueAndMethod } = useKnowledge() const documentCount = useMemo(() => { const availableDocCount = dataset.available_document_count || dataset.document_count @@ -122,16 +128,23 @@ const DatasetCard = ({