Skip to content

Commit 2294bd1

Browse files
committed
refactor: streamline server and parameter handling in base_mcp_node.py
1 parent d3333ec commit 2294bd1

File tree

1 file changed

+5
-8
lines changed

1 file changed

+5
-8
lines changed

apps/application/flow/step_node/mcp_node/impl/base_mcp_node.py

Lines changed: 5 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -29,16 +29,13 @@ def execute(self, mcp_servers, mcp_server, mcp_tool, mcp_tool_id, mcp_source, to
2929
if not tool.is_active:
3030
raise ValueError(f"Tool with ID {mcp_tool_id} is inactive.")
3131
servers = json.loads(tool.code)
32-
servers = self.handle_variables(servers) # 处理servers中的变量
33-
ToolExecutor().validate_mcp_transport(json.dumps(servers))
34-
params = json.loads(json.dumps(tool_params))
35-
params = self.handle_variables(params)
3632
else:
3733
servers = json.loads(mcp_servers)
38-
servers = self.handle_variables(servers) # 处理servers中的变量
39-
ToolExecutor().validate_mcp_transport(json.dumps(servers))
40-
params = json.loads(json.dumps(tool_params))
41-
params = self.handle_variables(params)
34+
35+
servers = self.handle_variables(servers) # 处理servers中的变量
36+
ToolExecutor().validate_mcp_transport(json.dumps(servers))
37+
params = json.loads(json.dumps(tool_params))
38+
params = self.handle_variables(params)
4239

4340
async def call_tool(t, a):
4441
client = MultiServerMCPClient(servers)

0 commit comments

Comments
 (0)