refactor: consolidate agent wait methods and use in-memory MCP transport

packages/cli/src/commands/agent/send.ts

@@ -182,8 +182,8 @@ export async function runSendCommand(
       }
     }
 
-    // Wait for agent to become idle
-    await client.waitForAgentIdle(agentId, 600000) // 10 minute timeout
+    // Wait for agent to finish
+    const state = await client.waitForFinish(agentId, 600000) // 10 minute timeout
 
     await client.close()
 
@@ -192,7 +192,7 @@ export async function runSendCommand(
       data: {
         agentId,
         status: 'completed',
-        message: 'Agent completed processing the message',
+        message: state.status === 'error' ? 'Agent finished with error' : 'Agent completed processing the message',
       },
       schema: agentSendSchema,
     }

packages/cli/src/commands/agent/wait.ts

@@ -140,35 +140,33 @@ export async function runWaitCommand(
       throw error
     }
 
-    // Wait for agent to become idle OR request permission (whichever comes first)
+    // Wait for agent to finish (idle, error, or permission)
     try {
-      const idlePromise = client.waitForAgentIdle(agentId, timeoutMs).then(() => ({ type: 'idle' as const }))
-      const permissionPromise = client
-        .waitForPermission(agentId, timeoutMs)
-        .then((request) => ({ type: 'permission' as const, request }))
-
-      const result = await Promise.race([idlePromise, permissionPromise])
+      const state = await client.waitForFinish(agentId, timeoutMs)
 
       await client.close()
 
-      if (result.type === 'permission') {
+      // Check if agent has pending permissions
+      if (state.pendingPermissions && state.pendingPermissions.length > 0) {
+        const permission = state.pendingPermissions[0]
         return {
           type: 'single',
           data: {
             agentId,
             status: 'permission',
-            message: `Agent is waiting for permission: ${result.request.kind}`,
+            message: `Agent is waiting for permission: ${permission.kind}`,
           },
           schema: agentWaitSchema,
         }
       }
 
+      // Agent is idle or error
       return {
         type: 'single',
         data: {
           agentId,
           status: 'idle',
-          message: 'Agent is now idle',
+          message: state.status === 'error' ? 'Agent finished with error' : 'Agent is now idle',
         },
         schema: agentWaitSchema,
       }