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 = ({
-
-
{dataset.name}
+
+
+ {dataset.name} +
{!isExternalProvider ? ( <> diff --git a/web/models/datasets.ts b/web/models/datasets.ts index e0a09d7af6..210b8349f6 100644 --- a/web/models/datasets.ts +++ b/web/models/datasets.ts @@ -729,10 +729,6 @@ export type CreateDatasetReq = { user_id: string role?: 'owner' | 'admin' | 'editor' | 'normal' | 'dataset_operator' }[] - indexing_technique?: IndexingType - retrieval_mode?: RetrievalConfig - embedding_model?: string - embedding_model_provider?: string } export type CreateDatasetResponse = { @@ -740,7 +736,6 @@ export type CreateDatasetResponse = { name: string description: string permission: DatasetPermission - data_source_type: DataSourceType indexing_technique: IndexingType created_by: string created_at: number diff --git a/web/service/knowledge/use-create-dataset.ts b/web/service/knowledge/use-create-dataset.ts index 8ca171312c..c1d077ebd6 100644 --- a/web/service/knowledge/use-create-dataset.ts +++ b/web/service/knowledge/use-create-dataset.ts @@ -240,13 +240,13 @@ export const useFetchDefaultProcessRule = ( }) } -export const useCreateDataset = ( +export const useCreatePipelineDataset = ( mutationOptions: MutationOptions = {}, ) => { return useMutation({ - mutationKey: [NAME_SPACE, 'create-dataset'], + mutationKey: [NAME_SPACE, 'create-pipeline-empty-dataset'], mutationFn: (req: CreateDatasetReq) => { - return post('/datasets', { body: req }) + return post('/rag/pipeline/empty-dataset', { body: req }) }, ...mutationOptions, })