From 13f127140ba8d46a5ca9e6ef204443acdd10dc13 Mon Sep 17 00:00:00 2001
From: aitbc <aitbc@keisanki.net>
Date: Thu, 7 May 2026 11:26:14 +0200
Subject: [PATCH] Fix Stage 6 issues based on analysis

- Fix agent_message parameter naming (agent -> agent_id) for consistency
- Fix agent_messages parameter naming (agent -> agent_id) for consistency
- Add agent_id parameter to agent_sdk_capabilities operation
- Add wallet_balance check before agent_message operation (requires wallet/password)
---
 .../stage6_agent_identity_sdk.json            | 22 ++++++++++++++++---
 1 file changed, 19 insertions(+), 3 deletions(-)

diff --git a/docs/agent-training/stage6_agent_identity_sdk.json b/docs/agent-training/stage6_agent_identity_sdk.json
index a7cb9fa1..84c4dcbc 100644
--- a/docs/agent-training/stage6_agent_identity_sdk.json
+++ b/docs/agent-training/stage6_agent_identity_sdk.json
@@ -82,10 +82,24 @@
           "response_fields": ["gpu_available", "supported_models"]
         }
       },
+      {
+        "operation": "wallet_balance",
+        "parameters": {
+          "wallet": "training-wallet"
+        },
+        "expected_result": {
+          "status": "success",
+          "balance": ">0"
+        },
+        "success_criteria": {
+          "status": "success",
+          "response_fields": ["balance", "symbol"]
+        }
+      },
       {
         "operation": "agent_message",
         "parameters": {
-          "agent": "agent_*",
+          "agent_id": "agent_*",
           "message": "test message",
           "wallet": "training-wallet",
           "password": "training123"
@@ -102,7 +116,7 @@
       {
         "operation": "agent_messages",
         "parameters": {
-          "agent": "agent_*"
+          "agent_id": "agent_*"
         },
         "expected_result": {
           "status": "success",
@@ -172,7 +186,9 @@
       },
       {
         "operation": "agent_sdk_capabilities",
-        "parameters": {},
+        "parameters": {
+          "agent_id": "agent_*"
+        },
         "expected_result": {
           "status": "success",
           "capabilities": "[]"