From e443f31cc5ee496a39ecb042fc285fca8a128871 Mon Sep 17 00:00:00 2001 From: CaptainB Date: Sun, 28 Sep 2025 19:29:51 +0800 Subject: [PATCH] feat: conditionally validate MCP transport based on tool type MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --bug=1062394 --user=刘瑞斌 【工具】启用、禁用工具失败 https://www.tapd.cn/62980211/s/1780683 --- apps/tools/serializers/tool.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/apps/tools/serializers/tool.py b/apps/tools/serializers/tool.py index 0cb55e569..89b4818b3 100644 --- a/apps/tools/serializers/tool.py +++ b/apps/tools/serializers/tool.py @@ -356,7 +356,8 @@ class ToolSerializer(serializers.Serializer): ToolCreateRequest(data=instance).is_valid(raise_exception=True) # 校验代码是否包括禁止的关键字 ToolExecutor().validate_banned_keywords(instance.get('code', '')) - ToolExecutor().validate_mcp_transport(instance.get('code', '')) + if instance.get('tool_type') == ToolType.MCP: + ToolExecutor().validate_mcp_transport(instance.get('code', '')) tool_id = uuid.uuid7() Tool( @@ -487,7 +488,8 @@ class ToolSerializer(serializers.Serializer): ToolEditRequest(data=instance).is_valid(raise_exception=True) # 校验代码是否包括禁止的关键字 ToolExecutor().validate_banned_keywords(instance.get('code', '')) - ToolExecutor().validate_mcp_transport(instance.get('code', '')) + if instance.get('tool_type') == ToolType.MCP: + ToolExecutor().validate_mcp_transport(instance.get('code', '')) if not QuerySet(Tool).filter(id=self.data.get('id')).exists(): raise serializers.ValidationError(_('Tool not found'))