fix(tool): The MCP tool checks for duplicate additions from the main screen and performs a test before adding.

This commit is contained in:
Timebomb2018
2026-03-11 17:19:07 +08:00
parent 7240dfe793
commit a3e6f67ff7
2 changed files with 81 additions and 5 deletions

View File

@@ -14,6 +14,7 @@ from app.models import User
from app.models.tool_model import ToolType, ToolStatus, AuthType
from app.services.tool_service import ToolService
from app.schemas.response_schema import ApiResponse
from app.core.exceptions import BusinessException
router = APIRouter(prefix="/tools", tags=["Tool System"])
@@ -103,7 +104,7 @@ async def create_tool(
val = getattr(request, key, None)
if val is not None:
request.config[key] = val
tool_id = service.create_tool(
tool_id = await service.create_tool(
name=request.name,
tool_type=request.tool_type,
tenant_id=current_user.tenant_id,
@@ -113,6 +114,8 @@ async def create_tool(
tags=request.tags
)
return success(data={"tool_id": tool_id}, msg="工具创建成功")
except BusinessException as e:
raise HTTPException(status_code=400, detail=e.message)
except ValueError as e:
raise HTTPException(status_code=400, detail=str(e))
except Exception as e:

View File

@@ -93,7 +93,44 @@ class ToolService:
if query.first():
raise BusinessException(f"工具名称 '{name}' 已存在", BizCode.DUPLICATE_NAME)
def create_tool(
def _check_mcp_duplicate(self, name: str, tool_type: ToolType, tenant_id: uuid.UUID, config: Dict[str, Any]):
"""检查MCP工具是否重复市场来源按market_id+market_config_id+mcp_service_id判断名称无关自建按name+tool_type判断"""
from app.models.tool_model import MCPSourceChannel
source_channel = config.get("source_channel")
is_market_source = (
source_channel is not None
and source_channel != MCPSourceChannel.SELF_HOSTED
)
if is_market_source:
exists = (
self.db.query(ToolConfig)
.join(MCPToolConfig, MCPToolConfig.id == ToolConfig.id)
.filter(
ToolConfig.tenant_id == tenant_id,
ToolConfig.tool_type == tool_type,
MCPToolConfig.source_channel == source_channel,
MCPToolConfig.market_id == config.get("market_id"),
MCPToolConfig.market_config_id == config.get("market_config_id"),
MCPToolConfig.mcp_service_id == config.get("mcp_service_id"),
)
.first()
)
if exists:
raise BusinessException(f"该MCP服务已添加", BizCode.DUPLICATE_NAME)
else:
exists = (
self.db.query(ToolConfig)
.filter(
ToolConfig.name == name,
ToolConfig.tool_type == tool_type,
ToolConfig.tenant_id == tenant_id,
)
.first()
)
if exists:
raise BusinessException(f"工具 '{name}' 已存在", BizCode.DUPLICATE_NAME)
async def create_tool(
self,
name: str,
tool_type: ToolType,
@@ -106,7 +143,19 @@ class ToolService:
"""创建工具"""
if tool_type == ToolType.BUILTIN:
raise ValueError("内置工具不允许创建")
self._check_name_duplicate(name, tool_type, tenant_id)
cfg = config or {}
if tool_type == ToolType.MCP:
self._check_mcp_duplicate(name, tool_type, tenant_id, cfg)
# 创建前测试连接
test_result = await self._test_mcp_connection_by_config(cfg)
if not test_result["success"]:
raise BusinessException(f"MCP连接测试失败: {test_result['message']}", BizCode.INVALID_PARAMETER)
# 将发现的工具列表写回 config
if "available_tools" in test_result:
cfg["available_tools"] = test_result["available_tools"]
else:
self._check_name_duplicate(name, tool_type, tenant_id)
try:
# 创建基础配置
@@ -117,19 +166,22 @@ class ToolService:
tool_type=tool_type.value,
tenant_id=tenant_id,
status=ToolStatus.AVAILABLE.value,
config_data=config or {},
config_data=cfg,
tags=tags
)
self.db.add(tool_config)
self.db.flush()
# 创建类型特定配置
self._create_type_config(tool_config, config or {})
self._create_type_config(tool_config, cfg)
self.db.commit()
logger.info(f"工具创建成功: {tool_config.id}")
return str(tool_config.id)
except BusinessException:
self.db.rollback()
raise
except Exception as e:
self.db.rollback()
logger.error(f"创建工具失败: {e}")
@@ -1165,6 +1217,27 @@ class ToolService:
logger.error(f"加载内置工具配置失败: {e}")
return {}
async def _test_mcp_connection_by_config(self, config: Dict[str, Any]) -> Dict[str, Any]:
"""根据配置参数直接测试MCP连接创建前调用无需已存在的工具记录"""
server_url = config.get("server_url")
if not server_url:
return {"success": False, "message": "server_url不能为空"}
connection_config = config.get("connection_config") or {}
try:
test_result = await self.mcp_tool_manager.test_tool_connection(server_url, connection_config)
if not test_result["success"]:
return test_result
success_flag, tools, error = await self.mcp_tool_manager.discover_tools(server_url, connection_config)
if not success_flag:
return {"success": False, "message": f"获取工具列表失败: {error}"}
tool_list = [
{tool["name"]: {"description": tool.get("description", ""), "inputSchema": tool.get("inputSchema", {})}}
for tool in tools if tool.get("name")
]
return {"success": True, "message": "MCP连接测试成功", "available_tools": tool_list}
except Exception as e:
return {"success": False, "message": f"连接测试异常: {str(e)}"}
async def _test_mcp_connection(self, config: ToolConfig) -> Dict[str, Any]:
"""测试MCP连接并自动同步工具列表"""
try: