From 5fe5da7c1d12e13107f971cb4c06036a46a08dd2 Mon Sep 17 00:00:00 2001
From: jyong <718720800@qq.com>
Date: Thu, 5 Jun 2025 11:12:06 +0800
Subject: [PATCH] r2

---
 api/core/plugin/impl/datasource.py          | 8 ++++----
 api/services/datasource_provider_service.py | 8 ++++----
 2 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/api/core/plugin/impl/datasource.py b/api/core/plugin/impl/datasource.py
index f469b51224..775f36b506 100644
--- a/api/core/plugin/impl/datasource.py
+++ b/api/core/plugin/impl/datasource.py
@@ -202,12 +202,12 @@ class PluginDatasourceManager(BasePluginClient):
         raise Exception("No response from plugin daemon")
 
     def validate_provider_credentials(
-        self, tenant_id: str, user_id: str, provider: str, credentials: dict[str, Any]
+        self, tenant_id: str, user_id: str, provider: str, plugin_id: str, credentials: dict[str, Any]
     ) -> bool:
         """
         validate the credentials of the provider
         """
-        datasource_provider_id = GenericProviderID(provider)
+        # datasource_provider_id = GenericProviderID(provider_id)
 
         response = self._request_with_plugin_daemon_response_stream(
             "POST",
@@ -216,12 +216,12 @@ class PluginDatasourceManager(BasePluginClient):
             data={
                 "user_id": user_id,
                 "data": {
-                    "provider": datasource_provider_id.provider_name,
+                    "provider": provider,
                     "credentials": credentials,
                 },
             },
             headers={
-                "X-Plugin-ID": datasource_provider_id.plugin_id,
+                "X-Plugin-ID": plugin_id,
                 "Content-Type": "application/json",
             },
         )
diff --git a/api/services/datasource_provider_service.py b/api/services/datasource_provider_service.py
index 1344dfa9fe..ffc056921b 100644
--- a/api/services/datasource_provider_service.py
+++ b/api/services/datasource_provider_service.py
@@ -32,7 +32,7 @@ class DatasourceProviderService:
         :param credentials:
         """
         credential_valid = self.provider_manager.validate_provider_credentials(
-            tenant_id=tenant_id, user_id=current_user.id, provider=provider, credentials=credentials
+            tenant_id=tenant_id, user_id=current_user.id, provider=provider, plugin_id=plugin_id, credentials=credentials
         )
         if credential_valid:
             # Get all provider configurations of the current workspace
@@ -119,13 +119,13 @@ class DatasourceProviderService:
             )
 
         return copy_credentials_list
-    
+
     def update_datasource_credentials(self, tenant_id: str, auth_id: str, provider: str, plugin_id: str, credentials: dict) -> None:
         """
         update datasource credentials.
         """
         credential_valid = self.provider_manager.validate_provider_credentials(
-            tenant_id=tenant_id, user_id=current_user.id, provider=provider, credentials=credentials
+            tenant_id=tenant_id, user_id=current_user.id, provider=provider,plugin_id=plugin_id, credentials=credentials
         )
         if credential_valid:
             # Get all provider configurations of the current workspace
@@ -156,7 +156,7 @@ class DatasourceProviderService:
                 db.session.commit()
         else:
             raise CredentialsValidateFailedError()
-        
+
     def remove_datasource_credentials(self, tenant_id: str, auth_id: str, provider: str, plugin_id: str) -> None:
         """
         remove datasource credentials.