15308555518/ MaxKB
1
0
Fork 0
mirror of https://github.com/1Panel-dev/MaxKB.git synced 2025-12-25 17:22:55 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

fix: prevent usage of 'stdio' in MCP server configuration
Some checks failed
sync2gitee / repo-sync (push) Has been cancelled
Details
Typos Check / Spell Check with Typos (push) Has been cancelled
Details

This commit is contained in:
CaptainB 2025-06-24 11:46:32 +08:00
parent 69f024492b
commit b13cd03706
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
apps/application/flow/step_node/ai_chat_step_node/impl/base_chat_node.py
View File

@ -219,7 +219,7 @@ class BaseChatNode(IChatNode):
message_list = self.generate_message_list(system, prompt, history_message) message_list = self.generate_message_list(system, prompt, history_message)
self.context['message_list'] = message_list self.context['message_list'] = message_list
if mcp_enable and mcp_servers is not None: if mcp_enable and mcp_servers is not None and '"stdio"' not in mcp_servers:
r = mcp_response_generator(chat_model, message_list, mcp_servers) r = mcp_response_generator(chat_model, message_list, mcp_servers)
return NodeResult( return NodeResult(
{'result': r, 'chat_model': chat_model, 'message_list': message_list, {'result': r, 'chat_model': chat_model, 'message_list': message_list,

2
apps/application/serializers/application_serializers.py
View File

@ -1325,6 +1325,8 @@ class ApplicationSerializer(serializers.Serializer):
def get_mcp_servers(self, with_valid=True): def get_mcp_servers(self, with_valid=True):
if with_valid: if with_valid:
self.is_valid(raise_exception=True) self.is_valid(raise_exception=True)
if '"stdio"' in self.data.get('mcp_servers'):
raise AppApiException(500, _('stdio is not supported'))
servers = json.loads(self.data.get('mcp_servers')) servers = json.loads(self.data.get('mcp_servers'))
async def get_mcp_tools(servers): async def get_mcp_tools(servers):