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 6d22f2115a..74e565a494 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 @@ -4,6 +4,7 @@ import CreateCard from './create-card' import { useI18N } from '@/context/i18n' import { useMemo } from 'react' import { LanguagesSupported } from '@/i18n-config/language' +import { useGlobalPublicStore } from '@/context/global-public-context' const BuiltInPipelineList = () => { const { locale } = useI18N() @@ -12,7 +13,8 @@ const BuiltInPipelineList = () => { return locale return LanguagesSupported[0] }, [locale]) - const { data: pipelineList, isLoading } = usePipelineTemplateList({ type: 'built-in', language }) + const enableMarketplace = useGlobalPublicStore(s => s.systemFeatures.enable_marketplace) + const { data: pipelineList, isLoading } = usePipelineTemplateList({ type: 'built-in', language }, enableMarketplace) const list = pipelineList?.pipeline_templates || [] return ( diff --git a/web/service/use-pipeline.ts b/web/service/use-pipeline.ts index a7b9c89410..92a7542c56 100644 --- a/web/service/use-pipeline.ts +++ b/web/service/use-pipeline.ts @@ -39,13 +39,14 @@ import { useInvalid } from './use-base' const NAME_SPACE = 'pipeline' export const PipelineTemplateListQueryKeyPrefix = [NAME_SPACE, 'template-list'] -export const usePipelineTemplateList = (params: PipelineTemplateListParams) => { +export const usePipelineTemplateList = (params: PipelineTemplateListParams, enabled = true) => { const { type, language } = params return useQuery({ queryKey: [...PipelineTemplateListQueryKeyPrefix, type, language], queryFn: () => { return get('/rag/pipeline/templates', { params }) }, + enabled, }) }