diff --git a/tools/test/test_minimal_agent.py b/tools/test/test_minimal_agent.py
index 393c56f..cf5fbcf 100644
--- a/tools/test/test_minimal_agent.py
+++ b/tools/test/test_minimal_agent.py
@@ -125,9 +125,9 @@ async def run_single_test(graph, test_case: dict) -> dict:
         print("开始执行图...")
         result = await graph.ainvoke(input_state, config=config)
         
-        # 提取最终回复
-        reply = ""
-        if result.get("messages"):
+        # 提取最终回复（优先使用 final_reply）
+        reply = result.get("final_reply", "")
+        if not reply and result.get("messages"):
             reply = result["messages"][-1].content
         
         print(f"\n✓ 执行完成")