From 798d0880d6736eb64038f0b3cc1783c4a436df3c Mon Sep 17 00:00:00 2001 From: jyong <718720800@qq.com> Date: Thu, 3 Jul 2025 15:15:01 +0800 Subject: [PATCH] r2 --- api/core/datasource/datasource_manager.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/api/core/datasource/datasource_manager.py b/api/core/datasource/datasource_manager.py index 838fee5b96..47f314e126 100644 --- a/api/core/datasource/datasource_manager.py +++ b/api/core/datasource/datasource_manager.py @@ -10,6 +10,7 @@ from core.datasource.entities.datasource_entities import DatasourceProviderType from core.datasource.errors import DatasourceProviderNotFoundError from core.datasource.local_file.local_file_provider import LocalFileDatasourcePluginProviderController from core.datasource.online_document.online_document_provider import OnlineDocumentDatasourcePluginProviderController +from core.datasource.online_drive.online_drive_provider import OnlineDriveDatasourcePluginProviderController from core.datasource.website_crawl.website_crawl_provider import WebsiteCrawlDatasourcePluginProviderController from core.plugin.impl.datasource import PluginDatasourceManager @@ -54,6 +55,13 @@ class DatasourceManager: plugin_unique_identifier=provider_entity.plugin_unique_identifier, tenant_id=tenant_id, ) + case DatasourceProviderType.ONLINE_DRIVE: + controller = OnlineDriveDatasourcePluginProviderController( + entity=provider_entity.declaration, + plugin_id=provider_entity.plugin_id, + plugin_unique_identifier=provider_entity.plugin_unique_identifier, + tenant_id=tenant_id, + ) case DatasourceProviderType.WEBSITE_CRAWL: controller = WebsiteCrawlDatasourcePluginProviderController( entity=provider_entity.declaration,