From 91f0cafefb554f33dd4cbcb570730e8df57b2c92 Mon Sep 17 00:00:00 2001
From: Rasic2 <1051987201@qq.com>
Date: Sun, 8 Feb 2026 15:58:40 +0800
Subject: [PATCH] =?UTF-8?q?refactor=EF=BC=9A=E5=B0=86=E6=AD=A5=E9=AA=A4?=
 =?UTF-8?q?=E6=8F=8F=E8=BF=B0=E9=94=AE=E6=94=B9=E4=B8=BAStepKey=E6=9E=9A?=
 =?UTF-8?q?=E4=B8=BE=E4=BE=BF=E4=BA=8E=E7=AE=A1=E7=90=86=E5=92=8C=E7=BB=B4?=
 =?UTF-8?q?=E6=8A=A4?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 agents/matmaster_agent/flow_agents/execution_agent/agent.py | 6 +++---
 agents/matmaster_agent/state.py                             | 4 ++++
 2 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/agents/matmaster_agent/flow_agents/execution_agent/agent.py b/agents/matmaster_agent/flow_agents/execution_agent/agent.py
index 13228441..5285d035 100644
--- a/agents/matmaster_agent/flow_agents/execution_agent/agent.py
+++ b/agents/matmaster_agent/flow_agents/execution_agent/agent.py
@@ -36,7 +36,7 @@
 from agents.matmaster_agent.locales import i18n
 from agents.matmaster_agent.logger import PrefixFilter
 from agents.matmaster_agent.prompt import MatMasterCheckTransferPrompt
-from agents.matmaster_agent.state import PLAN, STEP_DESCRIPTION
+from agents.matmaster_agent.state import PLAN, STEP_DESCRIPTION, StepKey
 from agents.matmaster_agent.sub_agents.mapping import (
     MatMasterSubAgentsEnum,
 )
@@ -289,10 +289,10 @@ async def _prepare_retry_failed_result(
             )
             # 在重试时更新步骤描述，包含校验失败的原因
             original_description = ctx.session.state[PLAN]['steps'][index][
-                'description'
+                StepKey.STEP_DESCRIPTION
             ]
             update_plan['steps'][index][
-                'description'
+                StepKey.STEP_DESCRIPTION
             ] = f"{original_description}\n\n注意：上次执行因以下原因校验失败，请改进：{validation_reason}"
         else:
             logger.info(
diff --git a/agents/matmaster_agent/state.py b/agents/matmaster_agent/state.py
index 35cc9384..7d241be5 100644
--- a/agents/matmaster_agent/state.py
+++ b/agents/matmaster_agent/state.py
@@ -19,3 +19,7 @@
 
 class StateKey(StrEnum):
     pass
+
+
+class StepKey(StrEnum):
+    STEP_DESCRIPTION = 'step_description'