fix: can not create empty knowledge

api/providers/trace/trace-langfuse/src/dify_trace_langfuse/langfuse_trace.py

@@ -13,8 +13,6 @@ from langfuse.api import (
     TraceBody,
 )
 from langfuse.api.commons.types.usage import Usage
-from opentelemetry.sdk.resources import Resource
-from opentelemetry.sdk.trace import TracerProvider
 from sqlalchemy.orm import sessionmaker
 
 from core.ops.base_trace_instance import BaseTraceInstance
@@ -54,40 +52,13 @@ class LangFuseDataTrace(BaseTraceInstance):
         langfuse_config: LangfuseConfig,
     ):
         super().__init__(langfuse_config)
-        # Isolated TracerProvider prevents the langfuse v3 SDK from attaching its
-        # SpanProcessor to the global OpenTelemetry TracerProvider, which would
-        # otherwise siphon every Flask/Celery/SQLAlchemy span in the process into
-        # this tenant's Langfuse project. See langfuse upgrade guide v2 -> v3.
-        self._tracer_provider = TracerProvider(
-            resource=Resource.create({"service.name": "dify-langfuse-app-trace"}),
-        )
         self.langfuse_client = Langfuse(
             public_key=langfuse_config.public_key,
             secret_key=langfuse_config.secret_key,
             host=langfuse_config.host,
-            tracer_provider=self._tracer_provider,
         )
         self.file_base_url = os.getenv("FILES_URL", "http://127.0.0.1:5001")
 
-    def close(self) -> None:
-        """Flush and shut down the isolated TracerProvider.
-
-        Called explicitly when the trace instance is evicted from the cache, or
-        implicitly via ``__del__`` on garbage collection. Idempotent.
-        """
-        provider = getattr(self, "_tracer_provider", None)
-        if provider is None:
-            return
-        try:
-            provider.shutdown()
-        except Exception:
-            logger.debug("Failed to shut down Langfuse TracerProvider", exc_info=True)
-        finally:
-            self._tracer_provider = None
-
-    def __del__(self) -> None:
-        self.close()
-
     @staticmethod
     def _get_completion_start_time(
         start_time: datetime | None, time_to_first_token: float | int | None

api/providers/trace/trace-langfuse/tests/unit_tests/langfuse_trace/test_langfuse_trace.py

@@ -50,92 +50,18 @@ def trace_instance(langfuse_config, monkeypatch: pytest.MonkeyPatch):
 
 
 def test_init(langfuse_config, monkeypatch: pytest.MonkeyPatch):
-    from opentelemetry.sdk.trace import TracerProvider
-
     mock_langfuse = MagicMock()
     monkeypatch.setattr("dify_trace_langfuse.langfuse_trace.Langfuse", mock_langfuse)
     monkeypatch.setenv("FILES_URL", "http://test.url")
 
     instance = LangFuseDataTrace(langfuse_config)
 
-    mock_langfuse.assert_called_once()
-    kwargs = mock_langfuse.call_args.kwargs
-    assert kwargs["public_key"] == langfuse_config.public_key
-    assert kwargs["secret_key"] == langfuse_config.secret_key
-    assert kwargs["host"] == langfuse_config.host
-    assert isinstance(kwargs["tracer_provider"], TracerProvider)
-    assert kwargs["tracer_provider"] is instance._tracer_provider
-    assert instance.file_base_url == "http://test.url"
-
-
-def test_init_passes_isolated_tracer_provider_to_langfuse(
-    langfuse_config, monkeypatch: pytest.MonkeyPatch
-):
-    """Regression test for langfuse v3 SDK side effect.
-
-    Without an explicit ``tracer_provider=`` kwarg, the Langfuse v3 SDK
-    attaches a ``LangfuseSpanProcessor`` to the *global* OpenTelemetry
-    TracerProvider — siphoning every Flask / Celery / SQLAlchemy span in the
-    process into the tenant's Langfuse project. See langfuse upgrade-path
-    docs (v2 -> v3) and GitHub discussion #9136.
-
-    The fix is to construct an isolated ``TracerProvider`` and pass it via
-    ``tracer_provider=`` so the SDK never touches the global one.
-    """
-    from opentelemetry import trace as otel_trace_api
-    from opentelemetry.sdk.trace import TracerProvider
-
-    captured: dict[str, object] = {}
-
-    def fake_langfuse(**kwargs):
-        captured.update(kwargs)
-        return MagicMock()
-
-    monkeypatch.setattr("dify_trace_langfuse.langfuse_trace.Langfuse", fake_langfuse)
-
-    instance = LangFuseDataTrace(langfuse_config)
-
-    # 1. tracer_provider kwarg must be supplied (drives the no-pollution branch
-    #    in langfuse.LangfuseResourceManager._init_tracer_provider).
-    assert "tracer_provider" in captured, (
-        "Langfuse() must receive an explicit tracer_provider=; without it the "
-        "v3 SDK attaches its SpanProcessor to the global OTEL TracerProvider."
+    mock_langfuse.assert_called_once_with(
+        public_key=langfuse_config.public_key,
+        secret_key=langfuse_config.secret_key,
+        host=langfuse_config.host,
     )
-
-    passed_provider = captured["tracer_provider"]
-    assert isinstance(passed_provider, TracerProvider)
-    assert passed_provider is instance._tracer_provider
-
-    # 2. The instance's provider must not be the global one.
-    global_provider = otel_trace_api.get_tracer_provider()
-    assert passed_provider is not global_provider
-
-
-def test_close_shuts_down_tracer_provider(langfuse_config, monkeypatch: pytest.MonkeyPatch):
-    monkeypatch.setattr("dify_trace_langfuse.langfuse_trace.Langfuse", lambda **kwargs: MagicMock())
-
-    instance = LangFuseDataTrace(langfuse_config)
-    provider = instance._tracer_provider
-    provider_shutdown = MagicMock()
-    monkeypatch.setattr(provider, "shutdown", provider_shutdown)
-
-    instance.close()
-
-    provider_shutdown.assert_called_once()
-    assert instance._tracer_provider is None
-
-
-def test_close_is_idempotent(langfuse_config, monkeypatch: pytest.MonkeyPatch):
-    monkeypatch.setattr("dify_trace_langfuse.langfuse_trace.Langfuse", lambda **kwargs: MagicMock())
-
-    instance = LangFuseDataTrace(langfuse_config)
-    provider_shutdown = MagicMock()
-    monkeypatch.setattr(instance._tracer_provider, "shutdown", provider_shutdown)
-
-    instance.close()
-    instance.close()
-
-    provider_shutdown.assert_called_once()
+    assert instance.file_base_url == "http://test.url"
 
 
 def test_trace_dispatch(trace_instance, monkeypatch: pytest.MonkeyPatch):

web/app/components/plugins/install-plugin/install-bundle/index.tsx

@@ -57,7 +57,7 @@ const InstallBundle: FC<Props> = ({
           foldAnimInto()
       }}
     >
-      <DialogContent className={cn('w-full max-w-[480px] overflow-hidden! text-left align-middle', cn(modalClassName, 'shadows-shadow-xl flex min-w-[560px] flex-col items-start rounded-2xl border-[0.5px] border-components-panel-border bg-components-panel-bg p-0'))}>
+      <DialogContent className={cn('relative w-full max-w-[480px] overflow-hidden! text-left align-middle', cn(modalClassName, 'shadows-shadow-xl flex min-w-[560px] flex-col items-start rounded-2xl border-[0.5px] border-components-panel-border bg-components-panel-bg p-0'))}>
         <DialogCloseButton />
 
         <div className="flex items-start gap-2 self-stretch pt-6 pr-14 pb-3 pl-6">

web/app/components/rag-pipeline/components/publish-as-knowledge-pipeline-modal.tsx

@@ -77,7 +77,7 @@ const PublishAsKnowledgePipelineModal = ({
   return (
     <>
       <Dialog open>
-        <DialogContent className="w-full max-w-[480px]! overflow-hidden! border-none p-0! text-left align-middle">
+        <DialogContent className="relative w-full max-w-[480px]! overflow-hidden! border-none p-0! text-left align-middle">
 
           <div className="relative flex items-center p-6 pr-14 pb-3 title-2xl-semi-bold text-text-primary">
             {t('common.publishAs', { ns: 'pipeline' })}

web/app/components/rag-pipeline/components/rag-pipeline-header/publisher/__tests__/index.spec.tsx

@@ -480,9 +480,7 @@ describe('publisher', () => {
       it('should show publish as knowledge pipeline modal when permitted', async () => {
         mockPublishedAt.mockReturnValue(1700000000)
         mockIsAllowPublishAsCustomKnowledgePipelineTemplate.mockReturnValue(true)
-        renderWithQueryClient(<Publisher />)
-
-        fireEvent.click(screen.getByText('workflow.common.publish'))
+        renderWithQueryClient(<Popup />)
 
         const publishAsButton = screen.getAllByRole('button').find(btn =>
           btn.textContent?.includes('pipeline.common.publishAs'),
@@ -497,9 +495,7 @@ describe('publisher', () => {
       it('should close publish as knowledge pipeline modal when cancel is clicked', async () => {
         mockPublishedAt.mockReturnValue(1700000000)
         mockIsAllowPublishAsCustomKnowledgePipelineTemplate.mockReturnValue(true)
-        renderWithQueryClient(<Publisher />)
-
-        fireEvent.click(screen.getByText('workflow.common.publish'))
+        renderWithQueryClient(<Popup />)
 
         const publishAsButton = screen.getAllByRole('button').find(btn =>
           btn.textContent?.includes('pipeline.common.publishAs'),
@@ -520,9 +516,7 @@ describe('publisher', () => {
       it('should call publishAsCustomizedPipeline when confirm is clicked in modal', async () => {
         mockPublishedAt.mockReturnValue(1700000000)
         mockPublishAsCustomizedPipeline.mockResolvedValue({})
-        renderWithQueryClient(<Publisher />)
-
-        fireEvent.click(screen.getByText('workflow.common.publish'))
+        renderWithQueryClient(<Popup />)
 
         const publishAsButton = screen.getAllByRole('button').find(btn =>
           btn.textContent?.includes('pipeline.common.publishAs'),
@@ -544,35 +538,6 @@ describe('publisher', () => {
           })
         })
       })
-
-      it('should publish as template with empty pipeline id fallback', async () => {
-        mockPublishedAt.mockReturnValue(1700000000)
-        mockPipelineId.mockReturnValue(undefined as unknown as string)
-        mockPublishAsCustomizedPipeline.mockResolvedValue({})
-        renderWithQueryClient(<Publisher />)
-
-        fireEvent.click(screen.getByText('workflow.common.publish'))
-
-        const publishAsButton = screen.getAllByRole('button').find(btn =>
-          btn.textContent?.includes('pipeline.common.publishAs'),
-        )
-        fireEvent.click(publishAsButton!)
-
-        await waitFor(() => {
-          expect(screen.getByTestId('publish-as-knowledge-pipeline-modal')).toBeInTheDocument()
-        })
-
-        fireEvent.click(screen.getByTestId('modal-confirm'))
-
-        await waitFor(() => {
-          expect(mockPublishAsCustomizedPipeline).toHaveBeenCalledWith({
-            pipelineId: '',
-            name: 'Test Pipeline',
-            icon_info: { type: 'emoji', emoji: '📚', background: '#fff' },
-            description: 'Test description',
-          })
-        })
-      })
     })
 
     describe('API Calls and Async Operations', () => {
@@ -642,9 +607,7 @@ describe('publisher', () => {
       it('should show success notification for publish as template', async () => {
         mockPublishedAt.mockReturnValue(1700000000)
         mockPublishAsCustomizedPipeline.mockResolvedValue({})
-        renderWithQueryClient(<Publisher />)
-
-        fireEvent.click(screen.getByText('workflow.common.publish'))
+        renderWithQueryClient(<Popup />)
 
         const publishAsButton = screen.getAllByRole('button').find(btn =>
           btn.textContent?.includes('pipeline.common.publishAs'),
@@ -670,9 +633,7 @@ describe('publisher', () => {
       it('should invalidate customized template list after publish as template', async () => {
         mockPublishedAt.mockReturnValue(1700000000)
         mockPublishAsCustomizedPipeline.mockResolvedValue({})
-        renderWithQueryClient(<Publisher />)
-
-        fireEvent.click(screen.getByText('workflow.common.publish'))
+        renderWithQueryClient(<Popup />)
 
         const publishAsButton = screen.getAllByRole('button').find(btn =>
           btn.textContent?.includes('pipeline.common.publishAs'),
@@ -725,9 +686,7 @@ describe('publisher', () => {
       it('should show error notification when publish as template fails', async () => {
         mockPublishedAt.mockReturnValue(1700000000)
         mockPublishAsCustomizedPipeline.mockRejectedValue(new Error('Template publish failed'))
-        renderWithQueryClient(<Publisher />)
-
-        fireEvent.click(screen.getByText('workflow.common.publish'))
+        renderWithQueryClient(<Popup />)
 
         const publishAsButton = screen.getAllByRole('button').find(btn =>
           btn.textContent?.includes('pipeline.common.publishAs'),
@@ -751,9 +710,7 @@ describe('publisher', () => {
       it('should close modal after publish as template error', async () => {
         mockPublishedAt.mockReturnValue(1700000000)
         mockPublishAsCustomizedPipeline.mockRejectedValue(new Error('Template publish failed'))
-        renderWithQueryClient(<Publisher />)
-
-        fireEvent.click(screen.getByText('workflow.common.publish'))
+        renderWithQueryClient(<Popup />)
 
         const publishAsButton = screen.getAllByRole('button').find(btn =>
           btn.textContent?.includes('pipeline.common.publishAs'),
@@ -1094,9 +1051,7 @@ describe('publisher', () => {
     it('should complete full publish as template flow', async () => {
       mockPublishedAt.mockReturnValue(1700000000)
       mockPublishAsCustomizedPipeline.mockResolvedValue({})
-      renderWithQueryClient(<Publisher />)
-
-      fireEvent.click(screen.getByText('workflow.common.publish'))
+      renderWithQueryClient(<Popup />)
 
       const publishAsButton = screen.getAllByRole('button').find(btn =>
         btn.textContent?.includes('pipeline.common.publishAs'),

web/app/components/rag-pipeline/components/rag-pipeline-header/publisher/__tests__/popup.spec.tsx

@@ -327,18 +327,11 @@ describe('Popup', () => {
 
     it('should request closing the outer popover before opening publish-as modal', () => {
       const onRequestClose = vi.fn()
-      const onShowPublishAsKnowledgePipelineModal = vi.fn()
-      render(
-        <Popup
-          onRequestClose={onRequestClose}
-          onShowPublishAsKnowledgePipelineModal={onShowPublishAsKnowledgePipelineModal}
-        />,
-      )
+      render(<Popup onRequestClose={onRequestClose} />)
 
       fireEvent.click(screen.getByText('pipeline.common.publishAs'))
 
       expect(onRequestClose).toHaveBeenCalledTimes(1)
-      expect(onShowPublishAsKnowledgePipelineModal).toHaveBeenCalledTimes(1)
     })
   })
 
@@ -359,6 +352,27 @@ describe('Popup', () => {
     })
   })
 
+  describe('Publish params', () => {
+    it('should publish as template with empty pipeline id fallback', async () => {
+      mockPipelineId = undefined
+      mockUseBoolean
+        .mockImplementationOnce((initial: boolean) => [initial, { setFalse: vi.fn(), setTrue: vi.fn() }])
+        .mockImplementationOnce((initial: boolean) => [initial, { setFalse: vi.fn(), setTrue: vi.fn() }])
+        .mockImplementationOnce(() => [true, { setFalse: vi.fn(), setTrue: vi.fn() }])
+        .mockImplementationOnce((initial: boolean) => [initial, { setFalse: vi.fn(), setTrue: vi.fn() }])
+      render(<Popup />)
+
+      fireEvent.click(screen.getByTestId('publish-as-confirm'))
+
+      expect(mockPublishAsCustomizedPipeline).toHaveBeenCalledWith({
+        pipelineId: '',
+        name: 'My Pipeline',
+        icon_info: { icon_type: 'emoji' },
+        description: 'desc',
+      })
+    })
+  })
+
   describe('Time formatting', () => {
     it('should format published time', () => {
       render(<Popup />)

web/app/components/rag-pipeline/components/rag-pipeline-header/publisher/index.tsx

@@ -1,8 +1,6 @@
-import type { IconInfo } from '@/models/datasets'
 import { Button } from '@langgenius/dify-ui/button'
 import { cn } from '@langgenius/dify-ui/cn'
 import { Popover, PopoverContent, PopoverTrigger } from '@langgenius/dify-ui/popover'
-import { toast } from '@langgenius/dify-ui/toast'
 import { RiArrowDownSLine } from '@remixicon/react'
 import { useBoolean } from 'ahooks'
 import {
@@ -12,11 +10,6 @@ import {
 } from 'react'
 import { useTranslation } from 'react-i18next'
 import { useNodesSyncDraft } from '@/app/components/workflow/hooks'
-import { useStore } from '@/app/components/workflow/store'
-import { useDocLink } from '@/context/i18n'
-import Link from '@/next/link'
-import { useInvalidCustomizedTemplateList, usePublishAsCustomizedPipeline } from '@/service/use-pipeline'
-import PublishAsKnowledgePipelineModal from '../../publish-as-knowledge-pipeline-modal'
 import Popup from './popup'
 
 const Publisher = () => {
@@ -24,12 +17,6 @@ const Publisher = () => {
   const [open, setOpen] = useState(false)
   const [confirmVisible, { setFalse: hideConfirm, setTrue: showConfirm }] = useBoolean(false)
   const { handleSyncWorkflowDraft } = useNodesSyncDraft()
-  const docLink = useDocLink()
-  const pipelineId = useStore(s => s.pipelineId)
-  const { mutateAsync: publishAsCustomizedPipeline } = usePublishAsCustomizedPipeline()
-  const invalidCustomizedTemplateList = useInvalidCustomizedTemplateList()
-  const [showPublishAsKnowledgePipelineModal, setShowPublishAsKnowledgePipelineModal] = useState(false)
-  const [isPublishingAsCustomizedPipeline, setIsPublishingAsCustomizedPipeline] = useState(false)
 
   const handleOpenChange = useCallback((newOpen: boolean) => {
     if (!newOpen && confirmVisible)
@@ -41,86 +28,38 @@ const Publisher = () => {
   const closePopover = useCallback(() => {
     setOpen(false)
   }, [])
-  const openPublishAsKnowledgePipelineModal = useCallback(() => {
-    setShowPublishAsKnowledgePipelineModal(true)
-  }, [])
-  const hidePublishAsKnowledgePipelineModal = useCallback(() => {
-    setShowPublishAsKnowledgePipelineModal(false)
-  }, [])
-  const handlePublishAsKnowledgePipeline = useCallback(async (name: string, icon: IconInfo, description?: string) => {
-    try {
-      setIsPublishingAsCustomizedPipeline(true)
-      await publishAsCustomizedPipeline({
-        pipelineId: pipelineId || '',
-        name,
-        icon_info: icon,
-        description,
-      })
-      toast.success(t('publishTemplate.success.message', { ns: 'datasetPipeline' }), {
-        description: (
-          <div className="flex flex-col gap-y-1">
-            <span className="system-xs-regular text-text-secondary">
-              {t('publishTemplate.success.tip', { ns: 'datasetPipeline' })}
-            </span>
-            <Link href={docLink()} target="_blank" className="inline-block system-xs-medium-uppercase text-text-accent">
-              {t('publishTemplate.success.learnMore', { ns: 'datasetPipeline' })}
-            </Link>
-          </div>
-        ),
-      })
-      invalidCustomizedTemplateList()
-    }
-    catch {
-      toast.error(t('publishTemplate.error.message', { ns: 'datasetPipeline' }))
-    }
-    finally {
-      setIsPublishingAsCustomizedPipeline(false)
-      hidePublishAsKnowledgePipelineModal()
-    }
-  }, [docLink, hidePublishAsKnowledgePipelineModal, invalidCustomizedTemplateList, pipelineId, publishAsCustomizedPipeline, t])
 
   return (
-    <>
-      <Popover
-        open={open}
-        onOpenChange={handleOpenChange}
+    <Popover
+      open={open}
+      onOpenChange={handleOpenChange}
+    >
+      <PopoverTrigger
+        nativeButton
+        render={(
+          <Button
+            className="px-2"
+            variant="primary"
+          >
+            <span className="pl-1">{t('common.publish', { ns: 'workflow' })}</span>
+            <RiArrowDownSLine className="h-4 w-4" />
+          </Button>
+        )}
+      />
+      <PopoverContent
+        placement="bottom-end"
+        sideOffset={4}
+        alignOffset={40}
+        popupClassName={cn('border-none bg-transparent shadow-none', confirmVisible && 'hidden')}
       >
-        <PopoverTrigger
-          nativeButton
-          render={(
-            <Button
-              className="px-2"
-              variant="primary"
-            >
-              <span className="pl-1">{t('common.publish', { ns: 'workflow' })}</span>
-              <RiArrowDownSLine className="h-4 w-4" />
-            </Button>
-          )}
+        <Popup
+          onRequestClose={closePopover}
+          confirmVisible={confirmVisible}
+          onShowConfirm={showConfirm}
+          onHideConfirm={hideConfirm}
         />
-        <PopoverContent
-          placement="bottom-end"
-          sideOffset={4}
-          alignOffset={40}
-          popupClassName={cn('border-none bg-transparent shadow-none', confirmVisible && 'hidden')}
-        >
-          <Popup
-            onRequestClose={closePopover}
-            confirmVisible={confirmVisible}
-            onShowConfirm={showConfirm}
-            onHideConfirm={hideConfirm}
-            isPublishingAsCustomizedPipeline={isPublishingAsCustomizedPipeline}
-            onShowPublishAsKnowledgePipelineModal={openPublishAsKnowledgePipelineModal}
-          />
-        </PopoverContent>
-      </Popover>
-      {showPublishAsKnowledgePipelineModal && (
-        <PublishAsKnowledgePipelineModal
-          confirmDisabled={isPublishingAsCustomizedPipeline}
-          onConfirm={handlePublishAsKnowledgePipeline}
-          onCancel={hidePublishAsKnowledgePipelineModal}
-        />
-      )}
-    </>
+      </PopoverContent>
+    </Popover>
   )
 }
 

web/app/components/rag-pipeline/components/rag-pipeline-header/publisher/popup.tsx

@@ -1,3 +1,4 @@
+import type { IconInfo } from '@/models/datasets'
 import type { PublishWorkflowParams } from '@/types/workflow'
 import {
   AlertDialog,
@@ -24,6 +25,7 @@ import ShortcutsName from '@/app/components/workflow/shortcuts-name'
 import { useStore, useWorkflowStore } from '@/app/components/workflow/store'
 import { getKeyboardKeyCodeBySystem } from '@/app/components/workflow/utils'
 import { useDatasetDetailContextWithSelector } from '@/context/dataset-detail'
+import { useDocLink } from '@/context/i18n'
 import { useModalContextSelector } from '@/context/modal-context'
 import { useProviderContextSelector } from '@/context/provider-context'
 import { useDatasetApiAccessUrl } from '@/hooks/use-api-access-url'
@@ -32,8 +34,9 @@ import Link from '@/next/link'
 import { useParams, useRouter } from '@/next/navigation'
 import { useInvalidDatasetList } from '@/service/knowledge/use-dataset'
 import { useInvalid } from '@/service/use-base'
-import { publishedPipelineInfoQueryKeyPrefix } from '@/service/use-pipeline'
+import { publishedPipelineInfoQueryKeyPrefix, useInvalidCustomizedTemplateList, usePublishAsCustomizedPipeline } from '@/service/use-pipeline'
 import { usePublishWorkflow } from '@/service/use-workflow'
+import PublishAsKnowledgePipelineModal from '../../publish-as-knowledge-pipeline-modal'
 
 const PUBLISH_SHORTCUT = ['ctrl', '⇧', 'P']
 type PopupProps = {
@@ -41,8 +44,6 @@ type PopupProps = {
   confirmVisible?: boolean
   onShowConfirm?: () => void
   onHideConfirm?: () => void
-  isPublishingAsCustomizedPipeline?: boolean
-  onShowPublishAsKnowledgePipelineModal?: () => void
 }
 
 const Popup = ({
@@ -50,12 +51,11 @@ const Popup = ({
   confirmVisible: controlledConfirmVisible,
   onShowConfirm,
   onHideConfirm,
-  isPublishingAsCustomizedPipeline = false,
-  onShowPublishAsKnowledgePipelineModal,
 }: PopupProps) => {
   const { t } = useTranslation()
   const { datasetId } = useParams()
   const { push } = useRouter()
+  const docLink = useDocLink()
   const publishedAt = useStore(s => s.publishedAt)
   const draftUpdatedAt = useStore(s => s.draftUpdatedAt)
   const pipelineId = useStore(s => s.pipelineId)
@@ -73,6 +73,9 @@ const Popup = ({
   const showConfirm = onShowConfirm ?? showLocalConfirm
   const hideConfirm = onHideConfirm ?? hideLocalConfirm
   const [publishing, { setFalse: hidePublishing, setTrue: showPublishing }] = useBoolean(false)
+  const { mutateAsync: publishAsCustomizedPipeline } = usePublishAsCustomizedPipeline()
+  const [showPublishAsKnowledgePipelineModal, { setFalse: hidePublishAsKnowledgePipelineModal, setTrue: setShowPublishAsKnowledgePipelineModal }] = useBoolean(false)
+  const [isPublishingAsCustomizedPipeline, { setFalse: hidePublishingAsCustomizedPipeline, setTrue: showPublishingAsCustomizedPipeline }] = useBoolean(false)
   const invalidPublishedPipelineInfo = useInvalid([...publishedPipelineInfoQueryKeyPrefix, pipelineId])
   const invalidDatasetList = useInvalidDatasetList()
   const handleHideConfirm = useCallback(() => {
@@ -142,15 +145,47 @@ const Popup = ({
   const goToAddDocuments = useCallback(() => {
     push(`/datasets/${datasetId}/documents/create-from-pipeline`)
   }, [datasetId, push])
+  const invalidCustomizedTemplateList = useInvalidCustomizedTemplateList()
+  const handlePublishAsKnowledgePipeline = useCallback(async (name: string, icon: IconInfo, description?: string) => {
+    try {
+      showPublishingAsCustomizedPipeline()
+      await publishAsCustomizedPipeline({
+        pipelineId: pipelineId || '',
+        name,
+        icon_info: icon,
+        description,
+      })
+      toast.success(t('publishTemplate.success.message', { ns: 'datasetPipeline' }), {
+        description: (
+          <div className="flex flex-col gap-y-1">
+            <span className="system-xs-regular text-text-secondary">
+              {t('publishTemplate.success.tip', { ns: 'datasetPipeline' })}
+            </span>
+            <Link href={docLink()} target="_blank" className="inline-block system-xs-medium-uppercase text-text-accent">
+              {t('publishTemplate.success.learnMore', { ns: 'datasetPipeline' })}
+            </Link>
+          </div>
+        ),
+      })
+      invalidCustomizedTemplateList()
+    }
+    catch {
+      toast.error(t('publishTemplate.error.message', { ns: 'datasetPipeline' }))
+    }
+    finally {
+      hidePublishingAsCustomizedPipeline()
+      hidePublishAsKnowledgePipelineModal()
+    }
+  }, [showPublishingAsCustomizedPipeline, publishAsCustomizedPipeline, pipelineId, t, invalidCustomizedTemplateList, hidePublishingAsCustomizedPipeline, hidePublishAsKnowledgePipelineModal, docLink])
   const handleClickPublishAsKnowledgePipeline = useCallback(() => {
     onRequestClose?.()
     if (!isAllowPublishAsCustomKnowledgePipelineTemplate) {
       setShowPricingModal()
     }
     else {
-      onShowPublishAsKnowledgePipelineModal?.()
+      setShowPublishAsKnowledgePipelineModal()
     }
-  }, [isAllowPublishAsCustomKnowledgePipelineTemplate, onRequestClose, onShowPublishAsKnowledgePipelineModal, setShowPricingModal])
+  }, [isAllowPublishAsCustomKnowledgePipelineTemplate, onRequestClose, setShowPublishAsKnowledgePipelineModal, setShowPricingModal])
   return (
     <div className={cn('rounded-2xl border-[0.5px] border-components-panel-border bg-components-panel-bg shadow-xl shadow-shadow-shadow-5', isAllowPublishAsCustomKnowledgePipelineTemplate ? 'w-[360px]' : 'w-[400px]')}>
       <div className="p-4 pt-3">
@@ -244,6 +279,7 @@ const Popup = ({
           </AlertDialogActions>
         </AlertDialogContent>
       </AlertDialog>
+      {showPublishAsKnowledgePipelineModal && (<PublishAsKnowledgePipelineModal confirmDisabled={isPublishingAsCustomizedPipeline} onConfirm={handlePublishAsKnowledgePipeline} onCancel={hidePublishAsKnowledgePipelineModal} />)}
     </div>
   )
 }