From e66c2baddad16e9af105b8ce29bb9a485f8bd95f Mon Sep 17 00:00:00 2001
From: twwu <twwu@dify.ai>
Date: Thu, 10 Jul 2025 13:55:24 +0800
Subject: [PATCH] refactor: update error handling to use
 DataSourceNodeErrorResponse in OnlineDocuments and WebsiteCrawl components

---
 .../data-source/online-documents/index.tsx                  | 6 +++---
 .../data-source/website-crawl/index.tsx                     | 5 +++--
 2 files changed, 6 insertions(+), 5 deletions(-)

diff --git a/web/app/components/datasets/documents/create-from-pipeline/data-source/online-documents/index.tsx b/web/app/components/datasets/documents/create-from-pipeline/data-source/online-documents/index.tsx
index a03d681d53..4308fa3068 100644
--- a/web/app/components/datasets/documents/create-from-pipeline/data-source/online-documents/index.tsx
+++ b/web/app/components/datasets/documents/create-from-pipeline/data-source/online-documents/index.tsx
@@ -8,7 +8,7 @@ import { useDatasetDetailContextWithSelector } from '@/context/dataset-detail'
 import { DatasourceType } from '@/models/pipeline'
 import { ssePost } from '@/service/base'
 import Toast from '@/app/components/base/toast'
-import type { DataSourceNodeCompletedResponse } from '@/types/pipeline'
+import type { DataSourceNodeCompletedResponse, DataSourceNodeErrorResponse } from '@/types/pipeline'
 import type { DataSourceNodeType } from '@/app/components/workflow/nodes/data-source/types'
 import { useDataSourceStore, useDataSourceStoreWithSelector } from '../store'
 
@@ -64,10 +64,10 @@ const OnlineDocuments = ({
           setDocumentsData(documentsData.data as DataSourceNotionWorkspace[])
           setCurrentWorkspaceId(documentsData.data[0].workspace_id)
         },
-        onError: (message: string) => {
+        onDataSourceNodeError: (error: DataSourceNodeErrorResponse) => {
           Toast.notify({
             type: 'error',
-            message,
+            message: error.error,
           })
         },
       },
diff --git a/web/app/components/datasets/documents/create-from-pipeline/data-source/website-crawl/index.tsx b/web/app/components/datasets/documents/create-from-pipeline/data-source/website-crawl/index.tsx
index 2464d5402f..d28e1cbf55 100644
--- a/web/app/components/datasets/documents/create-from-pipeline/data-source/website-crawl/index.tsx
+++ b/web/app/components/datasets/documents/create-from-pipeline/data-source/website-crawl/index.tsx
@@ -17,6 +17,7 @@ import { DatasourceType } from '@/models/pipeline'
 import { ssePost } from '@/service/base'
 import type {
   DataSourceNodeCompletedResponse,
+  DataSourceNodeErrorResponse,
   DataSourceNodeProcessingResponse,
 } from '@/types/pipeline'
 import type { DataSourceNodeType } from '@/app/components/workflow/nodes/data-source/types'
@@ -138,8 +139,8 @@ const WebsiteCrawl = ({
           setCrawlErrorMessage('')
           setStep(CrawlStep.finished)
         },
-        onError: (message: string) => {
-          setCrawlErrorMessage(message || t(`${I18N_PREFIX}.unknownError`))
+        onDataSourceNodeError: (error: DataSourceNodeErrorResponse) => {
+          setCrawlErrorMessage(error.error || t(`${I18N_PREFIX}.unknownError`))
           setStep(CrawlStep.finished)
         },
       },