From f19c18dc1412a9026ed8babacc71016309badf4f Mon Sep 17 00:00:00 2001
From: smyhw <i@smyhw.online>
Date: Mon, 11 Nov 2024 21:50:32 +0800
Subject: [PATCH] Fixes `you have not added provider None` (#10501)

---
 api/core/tools/tool_manager.py                 | 3 ++-
 api/services/tools/api_tools_manage_service.py | 7 ++++++-
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/api/core/tools/tool_manager.py b/api/core/tools/tool_manager.py
index bf2ad13620..d2723df7b2 100644
--- a/api/core/tools/tool_manager.py
+++ b/api/core/tools/tool_manager.py
@@ -555,6 +555,7 @@ class ToolManager:
         """
             get tool provider
         """
+        provider_name = provider
         provider: ApiToolProvider = (
             db.session.query(ApiToolProvider)
             .filter(
@@ -565,7 +566,7 @@ class ToolManager:
         )
 
         if provider is None:
-            raise ValueError(f"you have not added provider {provider}")
+            raise ValueError(f"you have not added provider {provider_name}")
 
         try:
             credentials = json.loads(provider.credentials_str) or {}
diff --git a/api/services/tools/api_tools_manage_service.py b/api/services/tools/api_tools_manage_service.py
index 4a93891855..ed0cebf460 100644
--- a/api/services/tools/api_tools_manage_service.py
+++ b/api/services/tools/api_tools_manage_service.py
@@ -113,6 +113,8 @@ class ApiToolManageService:
         if schema_type not in [member.value for member in ApiProviderSchemaType]:
             raise ValueError(f"invalid schema type {schema}")
 
+        provider_name = provider_name.strip()
+
         # check if the provider exists
         provider: ApiToolProvider = (
             db.session.query(ApiToolProvider)
@@ -203,6 +205,7 @@ class ApiToolManageService:
         """
         list api tool provider tools
         """
+        provider_name = provider
         provider: ApiToolProvider = (
             db.session.query(ApiToolProvider)
             .filter(
@@ -213,7 +216,7 @@ class ApiToolManageService:
         )
 
         if provider is None:
-            raise ValueError(f"you have not added provider {provider}")
+            raise ValueError(f"you have not added provider {provider_name}")
 
         controller = ToolTransformService.api_provider_to_controller(db_provider=provider)
         labels = ToolLabelManager.get_tool_labels(controller)
@@ -246,6 +249,8 @@ class ApiToolManageService:
         if schema_type not in [member.value for member in ApiProviderSchemaType]:
             raise ValueError(f"invalid schema type {schema}")
 
+        provider_name = provider_name.strip()
+
         # check if the provider exists
         provider: ApiToolProvider = (
             db.session.query(ApiToolProvider)