diff --git a/web/app/components/datasets/create-from-pipeline/list/template-card/index.tsx b/web/app/components/datasets/create-from-pipeline/list/template-card/index.tsx index 770b342292..bc0fe93108 100644 --- a/web/app/components/datasets/create-from-pipeline/list/template-card/index.tsx +++ b/web/app/components/datasets/create-from-pipeline/list/template-card/index.tsx @@ -18,7 +18,7 @@ import Details from './details' import Content from './content' import Actions from './actions' import type { CreateDatasetReq } from '@/models/datasets' -import { useCreatePipelineDataset } from '@/service/knowledge/use-create-dataset' +import { useCreatePipelineDatasetFromCustomized } from '@/service/knowledge/use-create-dataset' import CreateModal from './create-modal' import { useInvalid } from '@/service/use-base' import { useResetDatasetList } from '@/service/knowledge/use-dataset' @@ -45,7 +45,7 @@ const TemplateCard = ({ template_id: pipeline.id, type, }, false) - const { mutateAsync: createEmptyDataset } = useCreatePipelineDataset() + const { mutateAsync: createDataset } = useCreatePipelineDatasetFromCustomized() const { handleCheckPluginDependencies } = usePluginDependencies() const resetDatasetList = useResetDatasetList() @@ -66,7 +66,7 @@ const TemplateCard = ({ ...payload, yaml_content: pipelineTemplateInfo.export_data, } - await createEmptyDataset(request, { + await createDataset(request, { onSuccess: async (newDataset) => { Toast.notify({ type: 'success', @@ -76,7 +76,7 @@ const TemplateCard = ({ if (newDataset.pipeline_id) await handleCheckPluginDependencies(newDataset.pipeline_id, true) setShowCreateModal(false) - push(`/datasets/${newDataset.id}/pipeline`) + push(`/datasets/${newDataset.dataset_id}/pipeline`) }, onError: () => { Toast.notify({ @@ -85,7 +85,7 @@ const TemplateCard = ({ }) }, }) - }, [getPipelineTemplateInfo, createEmptyDataset, t, handleCheckPluginDependencies, push, resetDatasetList]) + }, [getPipelineTemplateInfo, createDataset, t, handleCheckPluginDependencies, push, resetDatasetList]) const handleShowTemplateDetails = useCallback(() => { setShowDetailModal(true) diff --git a/web/models/datasets.ts b/web/models/datasets.ts index 2d1b8086aa..aafd40945a 100644 --- a/web/models/datasets.ts +++ b/web/models/datasets.ts @@ -774,6 +774,7 @@ export type CreateDatasetResponse = { updated_by: string updated_at: number pipeline_id: string + dataset_id: string } export type NotionPagePreviewRequest = { diff --git a/web/service/knowledge/use-create-dataset.ts b/web/service/knowledge/use-create-dataset.ts index c1d077ebd6..987039a758 100644 --- a/web/service/knowledge/use-create-dataset.ts +++ b/web/service/knowledge/use-create-dataset.ts @@ -251,3 +251,15 @@ export const useCreatePipelineDataset = ( ...mutationOptions, }) } + +export const useCreatePipelineDatasetFromCustomized = ( + mutationOptions: MutationOptions = {}, +) => { + return useMutation({ + mutationKey: [NAME_SPACE, 'create-pipeline-dataset'], + mutationFn: (req: CreateDatasetReq) => { + return post('/rag/pipeline/dataset', { body: req }) + }, + ...mutationOptions, + }) +}