diff --git a/web/app/components/datasets/create-from-pipeline/list/built-in-pipeline-list.tsx b/web/app/components/datasets/create-from-pipeline/list/built-in-pipeline-list.tsx index af49e3d5c5..ab35066fd3 100644 --- a/web/app/components/datasets/create-from-pipeline/list/built-in-pipeline-list.tsx +++ b/web/app/components/datasets/create-from-pipeline/list/built-in-pipeline-list.tsx @@ -3,7 +3,7 @@ import TemplateCard from './template-card' const BuiltInPipelineList = () => { const { data: pipelineList, isLoading } = usePipelineTemplateList({ type: 'built-in' }) - const list = pipelineList?.pipelines + const list = pipelineList?.pipeline_templates if (isLoading || !list) return null diff --git a/web/app/components/datasets/create-from-pipeline/list/template-card/content.tsx b/web/app/components/datasets/create-from-pipeline/list/template-card/content.tsx index cf6527c092..fc36b4de8c 100644 --- a/web/app/components/datasets/create-from-pipeline/list/template-card/content.tsx +++ b/web/app/components/datasets/create-from-pipeline/list/template-card/content.tsx @@ -9,17 +9,17 @@ type ContentProps = { name: string description: string iconInfo: IconInfo - docForm: ChunkingMode + chunkStructure: ChunkingMode } const Content = ({ name, description, iconInfo, - docForm, + chunkStructure, }: ContentProps) => { const { t } = useTranslation() - const Icon = DOC_FORM_ICON_WITH_BG[docForm] || General + const Icon = DOC_FORM_ICON_WITH_BG[chunkStructure] || General return ( <> @@ -44,7 +44,7 @@ const Content = ({ {name}
- {t(`dataset.chunkingMode.${DOC_FORM_TEXT[docForm]}`)} + {t(`dataset.chunkingMode.${DOC_FORM_TEXT[chunkStructure]}`)}
diff --git a/web/app/components/datasets/create-from-pipeline/list/template-card/details.tsx b/web/app/components/datasets/create-from-pipeline/list/template-card/details.tsx index cbe9e785e4..9cc7093c45 100644 --- a/web/app/components/datasets/create-from-pipeline/list/template-card/details.tsx +++ b/web/app/components/datasets/create-from-pipeline/list/template-card/details.tsx @@ -25,7 +25,7 @@ const Details = ({ const appIcon = React.useMemo(() => { if (!pipelineTemplateInfo) return { type: 'emoji', icon: '📙', background: '#FFF4ED' } - const iconInfo = pipelineTemplateInfo.icon_info + const iconInfo = pipelineTemplateInfo.icon return iconInfo.icon_type === 'image' ? { type: 'image', url: iconInfo.icon_url || '', fileId: iconInfo.icon || '' } : { type: 'icon', icon: iconInfo.icon || '', background: iconInfo.icon_background || '' } 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 c149f37e31..96dec43efa 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 @@ -9,7 +9,8 @@ import Button from '@/app/components/base/button' import { useTranslation } from 'react-i18next' import Toast from '@/app/components/base/toast' import type { PipelineTemplate } from '@/models/pipeline' -import { useUpdateTemplateInfo } from '@/service/use-pipeline' +import { PipelineTemplateListQueryKeyPrefix, useUpdateTemplateInfo } from '@/service/use-pipeline' +import { useInvalid } from '@/service/use-base' type EditPipelineInfoProps = { onClose: () => void @@ -22,7 +23,7 @@ const EditPipelineInfo = ({ }: EditPipelineInfoProps) => { const { t } = useTranslation() const [name, setName] = useState(pipeline.name) - const iconInfo = pipeline.icon_info + const iconInfo = pipeline.icon const [appIcon, setAppIcon] = useState( iconInfo.icon_type === 'image' ? { type: 'image' as const, url: iconInfo.icon_url || '', fileId: iconInfo.icon || '' } @@ -62,6 +63,7 @@ const EditPipelineInfo = ({ }, []) const { mutateAsync: updatePipeline } = useUpdateTemplateInfo() + const invalidCustomizedTemplateList = useInvalid([...PipelineTemplateListQueryKeyPrefix, 'customized']) const handleSave = useCallback(async () => { if (!name) { @@ -74,7 +76,7 @@ const EditPipelineInfo = ({ const request = { template_id: pipeline.id, name, - icon_info: { + icon: { icon_type: appIcon.type, icon: appIcon.type === 'image' ? appIcon.fileId : appIcon.icon, icon_background: appIcon.type === 'image' ? undefined : appIcon.background, @@ -83,11 +85,12 @@ const EditPipelineInfo = ({ description, } await updatePipeline(request, { - onSettled: () => { + onSuccess: () => { + invalidCustomizedTemplateList() onClose() }, }) - }, [name, appIcon, description, pipeline.id, updatePipeline, onClose]) + }, [name, appIcon, description, pipeline.id, updatePipeline, invalidCustomizedTemplateList, onClose]) return (
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 291422a22a..a39baf608d 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 @@ -139,8 +139,8 @@ const TemplateCard = ({ { - console.log('params', params) return useQuery({ - queryKey: [NAME_SPACE, 'template', 'list', params.type], + queryKey: [...PipelineTemplateListQueryKeyPrefix, params.type], queryFn: () => { return get('/rag/pipeline/templates', { params }) },