use type annotation

api/controllers/console/workspace/tool_providers.py

@@ -24,7 +24,7 @@ from core.mcp.mcp_client import MCPClient
 from core.model_runtime.utils.encoders import jsonable_encoder
 from core.plugin.entities.plugin_daemon import CredentialType
 from core.plugin.impl.oauth import OAuthHandler
-from core.tools.entities.tool_entities import WorkflowToolParameterConfiguration
+from core.tools.entities.tool_entities import WorkflowToolParameterConfiguration, ApiProviderSchemaType
 from extensions.ext_database import db
 from libs.helper import alphanumeric, uuid_value
 from libs.login import current_account_with_tenant, login_required
@@ -74,7 +74,7 @@ class BuiltinToolUpdatePayload(BaseModel):
 
 class ApiToolProviderBasePayload(BaseModel):
     credentials: dict[str, Any]
-    schema_type: str
+    schema_type: ApiProviderSchemaType
     schema_: str = Field(alias="schema")
     provider: str
     icon: dict[str, Any]

api/services/tools/api_tools_manage_service.py

@@ -113,9 +113,6 @@ class ApiToolManageService:
         """
         create api tool provider
         """
-        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