From 0ff746ebf6f1fc01f20002a863babfac5dd4889d Mon Sep 17 00:00:00 2001
From: jyong <718720800@qq.com>
Date: Fri, 6 Jun 2025 12:08:09 +0800
Subject: [PATCH] r2

---
 api/services/rag_pipeline/rag_pipeline.py | 50 +++++++++++------------
 1 file changed, 25 insertions(+), 25 deletions(-)

diff --git a/api/services/rag_pipeline/rag_pipeline.py b/api/services/rag_pipeline/rag_pipeline.py
index d899e89b02..ac7df87586 100644
--- a/api/services/rag_pipeline/rag_pipeline.py
+++ b/api/services/rag_pipeline/rag_pipeline.py
@@ -450,32 +450,32 @@ class RagPipelineService:
             tenant_id=pipeline.tenant_id,
             datasource_type=DatasourceProviderType(datasource_type),
         )
+        match datasource_type:
+            case DatasourceProviderType.ONLINE_DOCUMENT:
+                datasource_runtime = cast(OnlineDocumentDatasourcePlugin, datasource_runtime)
+                online_document_result: GetOnlineDocumentPagesResponse = datasource_runtime._get_online_document_pages(
+                    user_id=account.id,
+                    datasource_parameters=user_inputs,
+                    provider_type=datasource_runtime.datasource_provider_type(),
+                )
+                return {
+                    "result": [page.model_dump() for page in online_document_result.result],
+                    "provider_type": datasource_node_data.get("provider_type"),
+                }
 
-        if datasource_runtime.datasource_provider_type() == DatasourceProviderType.ONLINE_DOCUMENT:
-            datasource_runtime = cast(OnlineDocumentDatasourcePlugin, datasource_runtime)
-            online_document_result: GetOnlineDocumentPagesResponse = datasource_runtime._get_online_document_pages(
-                user_id=account.id,
-                datasource_parameters=user_inputs,
-                provider_type=datasource_runtime.datasource_provider_type(),
-            )
-            return {
-                "result": [page.model_dump() for page in online_document_result.result],
-                "provider_type": datasource_node_data.get("provider_type"),
-            }
-
-        elif datasource_runtime.datasource_provider_type == DatasourceProviderType.WEBSITE_CRAWL:
-            datasource_runtime = cast(WebsiteCrawlDatasourcePlugin, datasource_runtime)
-            website_crawl_result: GetWebsiteCrawlResponse = datasource_runtime._get_website_crawl(
-                user_id=account.id,
-                datasource_parameters=user_inputs,
-                provider_type=datasource_runtime.datasource_provider_type(),
-            )
-            return {
-                "result": [result.model_dump() for result in website_crawl_result.result],
-                "provider_type": datasource_node_data.get("provider_type"),
-            }
-        else:
-            raise ValueError(f"Unsupported datasource provider: {datasource_runtime.datasource_provider_type}")
+            case DatasourceProviderType.WEBSITE_CRAWL:
+                datasource_runtime = cast(WebsiteCrawlDatasourcePlugin, datasource_runtime)
+                website_crawl_result: GetWebsiteCrawlResponse = datasource_runtime._get_website_crawl(
+                    user_id=account.id,
+                    datasource_parameters=user_inputs,
+                    provider_type=datasource_runtime.datasource_provider_type(),
+                )
+                return {
+                    "result": [result.model_dump() for result in website_crawl_result.result],
+                    "provider_type": datasource_node_data.get("provider_type"),
+                }
+            case _:
+                raise ValueError(f"Unsupported datasource provider: {datasource_runtime.datasource_provider_type}")
 
     def run_free_workflow_node(
         self, node_data: dict, tenant_id: str, user_id: str, node_id: str, user_inputs: dict[str, Any]