From df5a5c088188544008004aba286dc5ad778ca1fa Mon Sep 17 00:00:00 2001
From: zxstty <zhaojiaqi18@huawei.com>
Date: Fri, 5 Sep 2025 16:17:26 +0800
Subject: [PATCH] =?UTF-8?q?=E5=AE=8C=E5=96=84=E5=8F=82=E6=95=B0=E8=A1=A5?=
 =?UTF-8?q?=E5=85=85=E6=9C=BA=E5=88=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/scheduler/executor/agent.py   | 1 +
 apps/scheduler/mcp_agent/host.py   | 3 +--
 apps/scheduler/mcp_agent/prompt.py | 8 ++++----
 3 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/apps/scheduler/executor/agent.py b/apps/scheduler/executor/agent.py
index 42289433..743a052e 100644
--- a/apps/scheduler/executor/agent.py
+++ b/apps/scheduler/executor/agent.py
@@ -333,6 +333,7 @@ class MCPAgentExecutor(BaseExecutor):
             if self.task.state.step_status == StepStatus.PARAM:
                 if len(self.task.context) and self.task.context[-1].step_id == self.task.state.step_id:
                     del self.task.context[-1]
+                await self.get_tool_input_param(is_first=False)
             elif self.task.state.step_status == StepStatus.WAITING:
                 if self.params:
                     if len(self.task.context) and self.task.context[-1].step_id == self.task.state.step_id:
diff --git a/apps/scheduler/mcp_agent/host.py b/apps/scheduler/mcp_agent/host.py
index b7827278..2f8048c6 100644
--- a/apps/scheduler/mcp_agent/host.py
+++ b/apps/scheduler/mcp_agent/host.py
@@ -94,12 +94,11 @@ class MCPHost(MCPBase):
             current_goal=current_goal,
             tool_description=mcp_tool.description,
             input_schema=mcp_tool.input_schema,
-            current_input=current_input,
+            input_params=current_input,
             error_message=error_message,
             params=params,
             params_description=params_description,
         )
-
         json_generator = JsonGenerator(
             llm_query,
             [
diff --git a/apps/scheduler/mcp_agent/prompt.py b/apps/scheduler/mcp_agent/prompt.py
index de9d2b1f..6d8bf028 100644
--- a/apps/scheduler/mcp_agent/prompt.py
+++ b/apps/scheduler/mcp_agent/prompt.py
@@ -2303,8 +2303,8 @@ REPAIR_PARAMS: dict[LanguageType, str] = {
     {{current_goal}}
     # 工具入参scheme
     {{input_schema}}
-    # 工具入参
-    {{input_param}}
+    # 工具当前的入参
+    {{input_params}}
     # 运行报错
     {{error_message}}
     # 补充的参数
@@ -2380,8 +2380,8 @@ REPAIR_PARAMS: dict[LanguageType, str] = {
     </tool>
     # Tool input schema
     {{input_schema}}
-    # Tool input parameters
-    {{input_param}}
+    # Tool current input parameters
+    {{input_params}}
     # Runtime error
     {{error_message}}
     # Supplementary parameters
-- 
Gitee