panyy
/
UnisKB
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: ensure only active tools are processed in base_chat_node.py 

--bug=1061252 --user=刘瑞斌 【工具】MCP被禁用后，依然可以在应用中调用 https://www.tapd.cn/62980211/s/1765960
v3.2
CaptainB 2025-09-02 15:42:42 +08:00 committed by 刘瑞斌
parent a8e8a0580f
commit 0b6b45928d
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -279,7 +279,7 @@ class BaseChatNode(IChatNode):
mcp_servers_config = json.loads(mcp_servers) mcp_servers_config = json.loads(mcp_servers)
elif mcp_tool_id: elif mcp_tool_id:
mcp_tool = QuerySet(Tool).filter(id=mcp_tool_id).first() mcp_tool = QuerySet(Tool).filter(id=mcp_tool_id).first()
if mcp_tool: if mcp_tool and mcp_tool.is_active:
mcp_servers_config = json.loads(mcp_tool.code) mcp_servers_config = json.loads(mcp_tool.code)
if tool_enable: if tool_enable:
@ -288,6 +288,8 @@ class BaseChatNode(IChatNode):
self.context['execute_ids'] = [] self.context['execute_ids'] = []
for tool_id in tool_ids: for tool_id in tool_ids:
tool = QuerySet(Tool).filter(id=tool_id).first() tool = QuerySet(Tool).filter(id=tool_id).first()
if not tool.is_active:
continue
executor = ToolExecutor(CONFIG.get('SANDBOX')) executor = ToolExecutor(CONFIG.get('SANDBOX'))
if tool.init_params is not None: if tool.init_params is not None:
params = json.loads(rsa_long_decrypt(tool.init_params)) params = json.loads(rsa_long_decrypt(tool.init_params))