✨ MCP Tool test run #541

Author: jiangpeiling

backend/apps/tool_config_app.py

@@ -5,14 +5,15 @@
 from fastapi import APIRouter, Header, HTTPException
 from fastapi.responses import JSONResponse
 
-from consts.exceptions import MCPConnectionError
-from consts.model import ToolInstanceInfoRequest, ToolInstanceSearchRequest
+from consts.exceptions import MCPConnectionError, TimeoutException, NotFoundException
+from consts.model import ToolInstanceInfoRequest, ToolInstanceSearchRequest, ToolValidateRequest
 from services.tool_configuration_service import (
     search_tool_info_impl,
     update_tool_info_impl,
     update_tool_list,
     list_all_tools,
     load_last_tool_config_impl,
+    validate_remote_mcp_tool,
 )
 from utils.auth_utils import get_current_user_id
 
@@ -81,6 +82,7 @@ async def scan_and_update_tool(
         raise HTTPException(
             status_code=HTTPStatus.INTERNAL_SERVER_ERROR, detail="Failed to update tool")
 
+
 @router.get("/load_config/{tool_id}")
 async def load_last_tool_config(tool_id: int, authorization: Optional[str] = Header(None)):
     try:
@@ -97,4 +99,43 @@ async def load_last_tool_config(tool_id: int, authorization: Optional[str] = Hea
     except Exception as e:
         logger.error(f"Failed to load tool config: {e}")
         raise HTTPException(
-            status_code=HTTPStatus.INTERNAL_SERVER_ERROR, detail="Failed to load tool config")
+            status_code=HTTPStatus.INTERNAL_SERVER_ERROR, detail="Failed to load tool config")
+
+
+@router.post("/validate")
+async def validate_tool(
+    request: ToolValidateRequest,
+    authorization: Optional[str] = Header(None)
+):
+    """Validate specific tool based on source type"""
+    try:
+        _, tenant_id = get_current_user_id(authorization)
+        result = await validate_remote_mcp_tool(request, tenant_id)
+
+        return JSONResponse(
+            status_code=HTTPStatus.OK,
+            content=result
+        )
+    except MCPConnectionError as e:
+        logger.error(f"MCP connection failed: {e}")
+        raise HTTPException(
+            status_code=HTTPStatus.SERVICE_UNAVAILABLE,
+            detail=str(e)
+        )
+    except NotFoundException as e:
+        logger.error(f"Tool not found: {e}")
+        raise HTTPException(
+            status_code=HTTPStatus.NOT_FOUND,
+            detail=str(e)
+        )
+    except TimeoutException as e:
+        raise HTTPException(
+            status_code=HTTPStatus.REQUEST_TIMEOUT,
+            detail=str(e)
+        )
+    except Exception as e:
+        logger.error(f"Failed to validate tool: {e}")
+        raise HTTPException(
+            status_code=HTTPStatus.INTERNAL_SERVER_ERROR,
+            detail=f"Failed to validate tool: {str(e)}"
+        )

backend/consts/exceptions.py

@@ -85,4 +85,9 @@ class TTSConnectionException(Exception):
 
 class VoiceConfigException(Exception):
     """Raised when voice configuration is invalid."""
+    pass
+
+
+class ToolExecutionException(Exception):
+    """Raised when mcp tool execution failed."""
     pass

backend/consts/model.py

@@ -340,3 +340,14 @@ class TTSResponse(BaseModel):
     """Response model for TTS conversion"""
     status: str = Field(..., description="Status of the TTS conversion")
     message: Optional[str] = Field(None, description="Additional message")
+
+
+class ToolValidateRequest(BaseModel):
+    """Request model for tool validation"""
+    name: str = Field(..., description="Tool name to validate")
+    source: str = Field(..., description="Tool source (local, mcp, langchain)")
+    usage: Optional[str] = Field(None, description="Tool usage information")
+    inputs: Optional[Dict[str, Any]] = Field(
+        None, description="Tool inputs")
+    params: Optional[Dict[str, Any]] = Field(
+        None, description="Tool configuration parameters")