diff --git a/api/controllers/console/workspace/tool_providers.py b/api/controllers/console/workspace/tool_providers.py
index cb02277c4e..48eb866def 100644
--- a/api/controllers/console/workspace/tool_providers.py
+++ b/api/controllers/console/workspace/tool_providers.py
@@ -119,7 +119,7 @@ class ApiToolTestPayload(BaseModel):
     provider_name: str | None = None
     credentials: dict[str, Any]
     parameters: dict[str, Any]
-    schema_type: str
+    schema_type: ApiProviderSchemaType
     schema_: str = Field(alias="schema")
 
 
diff --git a/api/services/tools/api_tools_manage_service.py b/api/services/tools/api_tools_manage_service.py
index ce97a1a9d0..00b64d1790 100644
--- a/api/services/tools/api_tools_manage_service.py
+++ b/api/services/tools/api_tools_manage_service.py
@@ -244,7 +244,7 @@ class ApiToolManageService:
         original_provider: str,
         icon: dict,
         credentials: dict,
-        schema_type: str,
+        _schema_type: ApiProviderSchemaType,
         schema: str,
         privacy_policy: str | None,
         custom_disclaimer: str,
@@ -362,7 +362,7 @@ class ApiToolManageService:
         tool_name: str,
         credentials: dict,
         parameters: dict,
-        schema_type: str,
+        schema_type: ApiProviderSchemaType,
         schema: str,
     ):
         """