فهرست منبع

Fixes `you have not added provider None` (#10501)

smyhw 5 ماه پیش
والد
کامیت
f19c18dc14
2فایلهای تغییر یافته به همراه8 افزوده شده و 2 حذف شده
  1. 2 1
      api/core/tools/tool_manager.py
  2. 6 1
      api/services/tools/api_tools_manage_service.py

+ 2 - 1
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 {}

+ 6 - 1
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)