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 })
},