r2

api/services/rag_pipeline/rag_pipeline.py

@@ -450,8 +450,8 @@ class RagPipelineService:
             tenant_id=pipeline.tenant_id,
             datasource_type=DatasourceProviderType(datasource_type),
         )
-
-        if datasource_runtime.datasource_provider_type() == DatasourceProviderType.ONLINE_DOCUMENT:
+        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,
@@ -463,7 +463,7 @@ class RagPipelineService:
                     "provider_type": datasource_node_data.get("provider_type"),
                 }
 
-        elif datasource_runtime.datasource_provider_type == DatasourceProviderType.WEBSITE_CRAWL:
+            case DatasourceProviderType.WEBSITE_CRAWL:
                 datasource_runtime = cast(WebsiteCrawlDatasourcePlugin, datasource_runtime)
                 website_crawl_result: GetWebsiteCrawlResponse = datasource_runtime._get_website_crawl(
                     user_id=account.id,
@@ -474,7 +474,7 @@ class RagPipelineService:
                     "result": [result.model_dump() for result in website_crawl_result.result],
                     "provider_type": datasource_node_data.get("provider_type"),
                 }
-        else:
+            case _:
                 raise ValueError(f"Unsupported datasource provider: {datasource_runtime.datasource_provider_type}")
 
     def run_free_workflow_node(