youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-05 01:48:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: replace get_real_credential_by_id with get_datasource_credentials in multiple services for consistency

Browse Source
This commit is contained in:
Harry
2025-08-11 20:03:53 +08:00
parent 7f328328fb
commit 543f80ad5d
6 changed files with 99 additions and 94 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -123,19 +123,12 @@ class DatasourceNode(BaseNode):
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,
)
credentials = datasource_provider_service.get_datasource_credentials(
tenant_id=self.tenant_id,
provider=node_data.provider_name,
plugin_id=node_data.plugin_id,
credential_id=datasource_info.get("credential_id"),
)
match datasource_type:
case DatasourceProviderType.ONLINE_DOCUMENT:
datasource_runtime = cast(OnlineDocumentDatasourcePlugin, datasource_runtime)