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 a39baf608d..a83ae4a16c 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
@@ -23,11 +23,13 @@ import CreateModal from './create-modal'
type TemplateCardProps = {
pipeline: PipelineTemplate
showMoreOperations?: boolean
+ type: 'customized' | 'built-in'
}
const TemplateCard = ({
pipeline,
showMoreOperations = true,
+ type,
}: TemplateCardProps) => {
const { t } = useTranslation()
const { push } = useRouter()
@@ -36,7 +38,7 @@ const TemplateCard = ({
const [showDetailModal, setShowDetailModal] = useState(false)
const [showCreateModal, setShowCreateModal] = useState(false)
- const { refetch: getPipelineTemplateInfo } = usePipelineTemplateById(pipeline.id, false)
+ const { refetch: getPipelineTemplateInfo } = usePipelineTemplateById(pipeline.id, type, false)
const { mutateAsync: createEmptyDataset } = useCreatePipelineDataset()
const { handleCheckPluginDependencies } = usePluginDependencies()
@@ -179,6 +181,7 @@ const TemplateCard = ({
>
diff --git a/web/models/pipeline.ts b/web/models/pipeline.ts
index 68ceafbfe8..4552bfe010 100644
--- a/web/models/pipeline.ts
+++ b/web/models/pipeline.ts
@@ -30,17 +30,23 @@ export type PipelineTemplateListResponse = {
}
export type PipelineTemplateByIdResponse = {
+ id: string
name: string
icon: IconInfo
description: string
author: string // todo: TBD
structure: string // todo: TBD
- graph: {
- nodes: Node[]
- edges: Edge[]
- viewport: Viewport
+ export_data: {
+ workflow: {
+ graph: {
+ nodes: Node[]
+ edges: Edge[]
+ viewport: Viewport
+ }
+ environment_variables?: EnvironmentVariable[]
+ rag_pipeline_variables?: RAGPipelineVariables
+ }
}
- export_data: string
}
export type CreateFormData = {
diff --git a/web/service/use-pipeline.ts b/web/service/use-pipeline.ts
index 4993439df6..a6b0794db2 100644
--- a/web/service/use-pipeline.ts
+++ b/web/service/use-pipeline.ts
@@ -42,11 +42,11 @@ export const usePipelineTemplateList = (params: PipelineTemplateListParams) => {
})
}
-export const usePipelineTemplateById = (templateId: string, enabled: boolean) => {
+export const usePipelineTemplateById = (templateId: string, type: string, enabled: boolean) => {
return useQuery
({
queryKey: [NAME_SPACE, 'template', templateId],
queryFn: () => {
- return get(`/rag/pipeline/templates/${templateId}`)
+ return get(`/rag/pipeline/templates/${templateId}?type=${type}`)
},
enabled,
})