yujunjun/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2025-11-06 05:44:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

notion fix

Browse Source
This commit is contained in:
jyong 2025-08-06 15:39:22 +08:00
parent 13f38045d4
commit 1567b615dc
1 changed files with 14 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
api/core/workflow/nodes/datasource/datasource_node.py
View File

@ -122,15 +122,23 @@ class DatasourceNode(BaseNode):
) )
try: try:
datasource_provider_service = DatasourceProviderService()
if datasource_info.get("credential_id"):
credentials = datasource_provider_service.get_real_credential_by_id(
tenant_id=self.tenant_id,
credential_id=datasource_info.get("credential_id"),
provider=node_data.provider_name,
plugin_id=node_data.plugin_id,
)
else:
credentials = datasource_provider_service.get_default_credentials(
tenant_id=self.tenant_id,
provider=node_data.provider_name,
plugin_id=node_data.plugin_id,
)
match datasource_type: match datasource_type:
case DatasourceProviderType.ONLINE_DOCUMENT: case DatasourceProviderType.ONLINE_DOCUMENT:
datasource_runtime = cast(OnlineDocumentDatasourcePlugin, datasource_runtime) datasource_runtime = cast(OnlineDocumentDatasourcePlugin, datasource_runtime)
datasource_provider_service = DatasourceProviderService()
credentials = datasource_provider_service.get_default_credentials(
tenant_id=self.tenant_id,
provider=node_data.provider_name,
plugin_id=node_data.plugin_id,
)
if credentials: if credentials:
datasource_runtime.runtime.credentials = credentials datasource_runtime.runtime.credentials = credentials
online_document_result: Generator[DatasourceMessage, None, None] = ( online_document_result: Generator[DatasourceMessage, None, None] = (
@ -151,12 +159,6 @@ class DatasourceNode(BaseNode):
) )
case DatasourceProviderType.ONLINE_DRIVE: case DatasourceProviderType.ONLINE_DRIVE:
datasource_runtime = cast(OnlineDriveDatasourcePlugin, datasource_runtime) datasource_runtime = cast(OnlineDriveDatasourcePlugin, datasource_runtime)
datasource_provider_service = DatasourceProviderService()
credentials = datasource_provider_service.get_real_datasource_credentials(
tenant_id=self.tenant_id,
provider=node_data.provider_name,
plugin_id=node_data.plugin_id,
)
if credentials: if credentials:
datasource_runtime.runtime.credentials = credentials datasource_runtime.runtime.credentials = credentials
online_drive_result: Generator[DatasourceMessage, None, None] = ( online_drive_result: Generator[DatasourceMessage, None, None] = (