zhuchaowe/maxkb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore: validate transport types in mcp_servers configuration

Browse Source
This commit is contained in:
CaptainB 2025-07-21 11:23:38 +08:00
parent aee2f43acd
commit b5d2982812
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
apps/application/serializers/application.py
View File

@ -631,9 +631,10 @@ class ApplicationOperateSerializer(serializers.Serializer):
if with_valid: if with_valid:
self.is_valid(raise_exception=True) self.is_valid(raise_exception=True)
McpServersSerializer(data=instance).is_valid(raise_exception=True) McpServersSerializer(data=instance).is_valid(raise_exception=True)
if '"stdio"' in instance.get('mcp_servers'):
raise AppApiException(500, _('stdio is not supported'))
servers = json.loads(instance.get('mcp_servers')) servers = json.loads(instance.get('mcp_servers'))
for server, config in servers.items():
if config.get('transport') not in ['sse', 'streamable_http']:
raise AppApiException(500, _('Only support transport=sse or transport=streamable_http'))
tools = [] tools = []
for server in servers: for server in servers:
tools += [ tools += [