feat(datasource): update fetch_datasource_provider

This commit is contained in:
Dongyu Li 2025-07-01 11:57:06 +08:00
parent 618ad4c291
commit cdbba1400c

View File

@ -10,7 +10,7 @@ from core.datasource.entities.datasource_entities import (
OnlineDriveDownloadFileRequest, OnlineDriveDownloadFileRequest,
WebsiteCrawlMessage, WebsiteCrawlMessage,
) )
from core.plugin.entities.plugin import GenericProviderID, ToolProviderID from core.plugin.entities.plugin import DatasourceProviderID, GenericProviderID
from core.plugin.entities.plugin_daemon import ( from core.plugin.entities.plugin_daemon import (
PluginBasicBooleanResponse, PluginBasicBooleanResponse,
PluginDatasourceProviderEntity, PluginDatasourceProviderEntity,
@ -61,7 +61,7 @@ class PluginDatasourceManager(BasePluginClient):
if provider_id == "langgenius/file/file": if provider_id == "langgenius/file/file":
return PluginDatasourceProviderEntity(**self._get_local_file_datasource_provider()) return PluginDatasourceProviderEntity(**self._get_local_file_datasource_provider())
tool_provider_id = ToolProviderID(provider_id) tool_provider_id = DatasourceProviderID(provider_id)
def transformer(json_response: dict[str, Any]) -> dict: def transformer(json_response: dict[str, Any]) -> dict:
data = json_response.get("data") data = json_response.get("data")