diff --git a/web/app/components/datasets/create-from-pipeline/list/template-card/edit-pipeline-info.tsx b/web/app/components/datasets/create-from-pipeline/list/template-card/edit-pipeline-info.tsx index 96dec43efa..91ff41cb1a 100644 --- a/web/app/components/datasets/create-from-pipeline/list/template-card/edit-pipeline-info.tsx +++ b/web/app/components/datasets/create-from-pipeline/list/template-card/edit-pipeline-info.tsx @@ -76,7 +76,7 @@ const EditPipelineInfo = ({ const request = { template_id: pipeline.id, name, - icon: { + icon_info: { icon_type: appIcon.type, icon: appIcon.type === 'image' ? appIcon.fileId : appIcon.icon, icon_background: appIcon.type === 'image' ? undefined : appIcon.background, 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 a83ae4a16c..c5d88237c8 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 @@ -5,6 +5,7 @@ import EditPipelineInfo from './edit-pipeline-info' import type { PipelineTemplate } from '@/models/pipeline' import Confirm from '@/app/components/base/confirm' import { + PipelineTemplateListQueryKeyPrefix, useDeleteTemplate, useExportTemplateDSL, usePipelineTemplateById, @@ -19,6 +20,7 @@ import Actions from './actions' import type { CreateDatasetReq } from '@/models/datasets' import { useCreatePipelineDataset } from '@/service/knowledge/use-create-dataset' import CreateModal from './create-modal' +import { useInvalid } from '@/service/use-base' type TemplateCardProps = { pipeline: PipelineTemplate @@ -127,14 +129,16 @@ const TemplateCard = ({ }, []) const { mutateAsync: deletePipeline } = useDeleteTemplate() + const invalidCustomizedTemplateList = useInvalid([...PipelineTemplateListQueryKeyPrefix, 'customized']) const onConfirmDelete = useCallback(async () => { await deletePipeline(pipeline.id, { - onSettled: () => { + onSuccess: () => { + invalidCustomizedTemplateList() setShowConfirmDelete(false) }, }) - }, [pipeline.id, deletePipeline]) + }, [pipeline.id, deletePipeline, invalidCustomizedTemplateList]) return (