Skip to content

Commit 6492b9d

Browse files
committed
fix: refactor tool initialization parameters handling in base_chat_step and base_chat_node
--bug=1069435@tapd-62980211 --user=刘瑞斌 【工具】新创建的工具,启动参数必须要手动保存,才能使用默认值 https://www.tapd.cn/62980211/s/1925056
1 parent 238c627 commit 6492b9d

2 files changed

Lines changed: 4 additions & 6 deletions

File tree

apps/application/chat_pipeline/step/chat_step/impl/base_chat_step.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -279,11 +279,10 @@ def _handle_mcp_request(self, mcp_source, mcp_servers, mcp_tool_ids, tool_ids,
279279
continue
280280
executor = ToolExecutor()
281281
if tool.init_params is not None:
282-
params = json.loads(rsa_long_decrypt(tool.init_params))
283282
tool_init_params = json.loads(rsa_long_decrypt(tool.init_params))
284283
else:
285-
params = {}
286-
tool_config = executor.get_tool_mcp_config(tool, params)
284+
tool_init_params = {i["field"]: i.get('default_value') for i in tool.init_field_list}
285+
tool_config = executor.get_tool_mcp_config(tool, tool_init_params)
287286

288287
mcp_servers_config[str(tool.id)] = tool_config
289288

apps/application/flow/step_node/ai_chat_step_node/impl/base_chat_node.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -277,11 +277,10 @@ def _handle_mcp_request(self, mcp_source, mcp_servers, mcp_tool_id, mcp_tool_ids
277277
continue
278278
executor = ToolExecutor()
279279
if tool.init_params is not None:
280-
params = json.loads(rsa_long_decrypt(tool.init_params))
281280
tool_init_params = json.loads(rsa_long_decrypt(tool.init_params))
282281
else:
283-
params = {}
284-
tool_config = executor.get_tool_mcp_config(tool, params)
282+
tool_init_params = {i["field"]: i.get('default_value') for i in tool.init_field_list}
283+
tool_config = executor.get_tool_mcp_config(tool, tool_init_params)
285284

286285
mcp_servers_config[str(tool.id)] = tool_config
287286

0 commit comments

Comments
 (0)