✨ MCP Tool test run #541

Author: Phinease

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_tool_impl,
 )
 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,38 @@ 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_tool_impl(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 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

@@ -244,6 +244,7 @@ class ToolInfo(BaseModel):
     output_type: str
     class_name: str
     usage: Optional[str]
+    origin_name: Optional[str] = None
 
 
 # used in Knowledge Summary request
@@ -340,3 +341,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")

backend/database/db_models.py

@@ -167,6 +167,7 @@ class ToolInfo(TableBase):
 
     tool_id = Column(Integer, primary_key=True, nullable=False, doc="ID")
     name = Column(String(100), doc="Unique key name")
+    origin_name = Column(String(100), doc="Original name")
     class_name = Column(
         String(100), doc="Tool class name, used when the tool is instantiated")
     description = Column(String(2048), doc="Prompt tool description")

backend/pyproject.toml

@@ -13,7 +13,7 @@ dependencies = [
     "websocket-client>=1.8.0",
     "pyyaml>=6.0.2",
     "redis>=5.0.0",
-    "fastmcp==2.6.0",
+    "fastmcp==2.12.0",
     "langchain>=0.3.26"
 ]