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,