aws-runtime-bridge 1.7.50 → 1.8.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (63) hide show
  1. package/dist/adapter/AcodeSdkAdapter.d.ts +30 -3
  2. package/dist/adapter/AcodeSdkAdapter.d.ts.map +1 -1
  3. package/dist/adapter/AcodeSdkAdapter.js +427 -105
  4. package/dist/adapter/AcodeSdkAdapter.test.js +1093 -86
  5. package/dist/adapter/ClaudeSdkAdapter.test.js +31 -8
  6. package/dist/adapter/CodexSdkAdapter.d.ts.map +1 -1
  7. package/dist/adapter/CodexSdkAdapter.js +2 -0
  8. package/dist/adapter/OpencodeSdkAdapter.d.ts.map +1 -1
  9. package/dist/adapter/OpencodeSdkAdapter.js +2 -0
  10. package/dist/adapter/SdkProviderSpi.d.ts.map +1 -1
  11. package/dist/adapter/adapter.test.js +2 -2
  12. package/dist/adapter/idle-poll-loop.d.ts +4 -4
  13. package/dist/adapter/idle-poll-loop.d.ts.map +1 -1
  14. package/dist/adapter/idle-poll-loop.js +3 -3
  15. package/dist/adapter/idle-poll-loop.test.js +15 -1
  16. package/dist/adapter/types.d.ts +25 -6
  17. package/dist/adapter/types.d.ts.map +1 -1
  18. package/dist/adapter/types.js +221 -110
  19. package/dist/adapter/types.test.js +10 -3
  20. package/dist/routes/instance.d.ts.map +1 -1
  21. package/dist/routes/instance.js +6 -7
  22. package/dist/routes/runtime-binding.js +2 -2
  23. package/dist/routes/terminal.d.ts +20 -5
  24. package/dist/routes/terminal.d.ts.map +1 -1
  25. package/dist/routes/terminal.js +420 -170
  26. package/dist/routes/terminal.test.js +380 -208
  27. package/dist/services/mcp-launch-binding-queue.d.ts +1 -0
  28. package/dist/services/mcp-launch-binding-queue.d.ts.map +1 -1
  29. package/dist/services/mcp-launch-binding-queue.js +8 -2
  30. package/dist/services/mcp-launch-binding-queue.test.js +47 -4
  31. package/dist/services/session-output.d.ts +10 -6
  32. package/dist/services/session-output.d.ts.map +1 -1
  33. package/dist/services/session-output.js +29 -14
  34. package/dist/services/session-output.test.js +93 -39
  35. package/package/acode/dist/built-in-file-tools.d.ts +1 -1
  36. package/package/acode/dist/built-in-file-tools.d.ts.map +1 -1
  37. package/package/acode/dist/built-in-file-tools.js +388 -50
  38. package/package/acode/dist/runtime.d.ts +11 -3
  39. package/package/acode/dist/runtime.d.ts.map +1 -1
  40. package/package/acode/dist/runtime.js +902 -190
  41. package/package/acode/dist/types.d.ts +18 -5
  42. package/package/acode/dist/types.d.ts.map +1 -1
  43. package/package/aws-client-agent-mcp/dist/mcp-server.d.ts.map +1 -1
  44. package/package/aws-client-agent-mcp/dist/mcp-server.js +3 -2
  45. package/package/aws-client-agent-mcp/dist/mcp-server.js.map +1 -1
  46. package/package/aws-client-agent-mcp/dist/mcp-server.test.js +11 -9
  47. package/package/aws-client-agent-mcp/dist/mcp-server.test.js.map +1 -1
  48. package/package/aws-client-agent-mcp/dist/runtime-launch-binding.d.ts.map +1 -1
  49. package/package/aws-client-agent-mcp/dist/runtime-launch-binding.js +1 -0
  50. package/package/aws-client-agent-mcp/dist/runtime-launch-binding.js.map +1 -1
  51. package/package/aws-client-agent-mcp/dist/status-reporter.d.ts +2 -0
  52. package/package/aws-client-agent-mcp/dist/status-reporter.d.ts.map +1 -1
  53. package/package/aws-client-agent-mcp/dist/status-reporter.js +31 -3
  54. package/package/aws-client-agent-mcp/dist/status-reporter.js.map +1 -1
  55. package/package/aws-client-agent-mcp/dist/types.d.ts +1 -1
  56. package/package/aws-client-agent-mcp/dist/types.d.ts.map +1 -1
  57. package/package/aws-client-agent-mcp/dist/types.js.map +1 -1
  58. package/package/aws-client-agent-mcp/dist/websocket-client.d.ts.map +1 -1
  59. package/package/aws-client-agent-mcp/dist/websocket-client.js +2 -0
  60. package/package/aws-client-agent-mcp/dist/websocket-client.js.map +1 -1
  61. package/package/aws-client-agent-mcp/dist/websocket-client.test.js +43 -0
  62. package/package/aws-client-agent-mcp/dist/websocket-client.test.js.map +1 -1
  63. package/package.json +1 -1
@@ -4,19 +4,35 @@
4
4
  * 主流程:直接加载随 aws-runtime-bridge 内置发布的 ACode headless runtime,
5
5
  * 不依赖外部 CLI 安装;将 ACode 的结构化事件映射为 runtime-bridge 统一事件流。
6
6
  */
7
- import { EventEmitter } from 'node:events';
8
- import path from 'node:path';
9
- import { pathToFileURL } from 'node:url';
10
- import { v4 as uuidv4 } from 'uuid';
11
- import { getAcodeConfigFile, getAcodeSkillsDir, getRuntimeHomeDir } from '../config.js';
12
- import { getReleasedAcodeEntryPath } from '../services/acode-package.js';
7
+ import { EventEmitter } from "node:events";
8
+ import path from "node:path";
9
+ import { pathToFileURL } from "node:url";
10
+ import { v4 as uuidv4 } from "uuid";
11
+ import { getAcodeConfigFile, getAcodeSkillsDir, getRuntimeHomeDir, } from "../config.js";
12
+ import { getReleasedAcodeEntryPath } from "../services/acode-package.js";
13
+ import { IdlePollLoop } from "./idle-poll-loop.js";
14
+ import { getCanonicalMcpDetail, getToolActionInfo, } from "./types.js";
13
15
  const ACODE_AUTO_POLL_PROMPT = [
14
- 'Continue the AgentsWork message loop now.',
15
- 'You MUST call the exact MCP tool named aws-mcp__poll_message as your next action and wait for the next assigned message or task.',
16
- 'Do not answer in plain text before calling aws-mcp__poll_message.',
17
- 'After handling any received work, reply through the appropriate aws-mcp messaging tool, then continue listening again.',
18
- ].join(' ');
16
+ "Continue the AgentsWork message loop now.",
17
+ "You MUST call the exact MCP tool named aws-mcp__poll_message as your next action and wait for the next assigned message or task.",
18
+ "Do not answer in plain text before calling aws-mcp__poll_message.",
19
+ "After handling any received work, reply through the appropriate aws-mcp messaging tool, then continue listening again.",
20
+ ].join(" ");
19
21
  const ACODE_AUTO_POLL_DELAY_MS = 250;
22
+ // Backoff used when a polling-driven turn fails (e.g., 60s MCP tool timeout,
23
+ // model 5xx, provider unreachable). Larger than ACODE_AUTO_POLL_DELAY_MS to
24
+ // avoid busy-looping when the same failure recurs on every retry.
25
+ const ACODE_AUTO_POLL_FAILURE_DELAY_MS = 2000;
26
+ const ACODE_MESSAGE_POLLING_IDLE_PROMPT = "系统提醒:当前工作已完成或运行时已进入空闲状态。请立即调用 poll_message 工具阻塞等待新消息和流程线任务提醒;收到流程任务提醒后调用 my_task 查看已由后台静默抢占的任务详情并直接处理;完成任务后调用 submit_task_result 提交结果,然后继续 poll_message。";
27
+ function isMessagePollingIdleCommand(command) {
28
+ const normalized = command.trim();
29
+ if (!normalized)
30
+ return false;
31
+ const lower = normalized.toLowerCase();
32
+ return (lower.includes("poll_message") ||
33
+ lower.includes("get_message") ||
34
+ normalized.includes("取消息"));
35
+ }
20
36
  function getAcodePackageEntryPath() {
21
37
  return path.resolve(getReleasedAcodeEntryPath());
22
38
  }
@@ -24,15 +40,26 @@ export function toAcodeEsmImportSpecifier(modulePath) {
24
40
  return pathToFileURL(path.resolve(modulePath)).href;
25
41
  }
26
42
  function toUsage(value) {
27
- if (!value || typeof value !== 'object') {
43
+ if (!value || typeof value !== "object") {
28
44
  return undefined;
29
45
  }
30
46
  const usage = value;
47
+ const inputTokens = toFiniteTokenCount(usage.inputTokens, usage.input_tokens, usage.promptTokens, usage.prompt_tokens);
48
+ const outputTokens = toFiniteTokenCount(usage.outputTokens, usage.output_tokens, usage.completionTokens, usage.completion_tokens);
31
49
  return {
32
- inputTokens: Number(usage.inputTokens) || 0,
33
- outputTokens: Number(usage.outputTokens) || 0,
50
+ inputTokens,
51
+ outputTokens,
34
52
  };
35
53
  }
54
+ function toFiniteTokenCount(...values) {
55
+ for (const value of values) {
56
+ const count = Number(value);
57
+ if (Number.isFinite(count) && count > 0) {
58
+ return count;
59
+ }
60
+ }
61
+ return 0;
62
+ }
36
63
  /**
37
64
  * Check whether an ACode turn_complete event represents a failed turn.
38
65
  * Main flow: only explicit failed=true prevents error status from being reset to waiting_input.
@@ -40,18 +67,28 @@ function toUsage(value) {
40
67
  export function isFailedAcodeTurnComplete(data) {
41
68
  return data.failed === true;
42
69
  }
70
+ function toAcodeMcpActionDetail(actionDetail, server, tool) {
71
+ if (!actionDetail) {
72
+ return tool ? `${server}/${tool}` : server;
73
+ }
74
+ const prefix = `${server}__`;
75
+ if (actionDetail.startsWith(prefix)) {
76
+ return getCanonicalMcpDetail(`${server}__${actionDetail.slice(prefix.length)}`);
77
+ }
78
+ return getCanonicalMcpDetail(actionDetail.replace("/", "__"));
79
+ }
43
80
  export class AcodeSdkAdapter extends EventEmitter {
44
81
  constructor() {
45
82
  super(...arguments);
46
- this.providerId = 'acode';
47
- this.displayName = 'ACode';
83
+ this.providerId = "acode";
84
+ this.displayName = "ACode";
48
85
  this.sessions = new Map();
49
86
  }
50
87
  async startSession(sessionId, config) {
51
88
  const runtimeHome = getRuntimeHomeDir();
52
89
  const session = {
53
90
  sessionId,
54
- status: 'starting',
91
+ status: "starting",
55
92
  messages: [],
56
93
  createdAt: new Date().toISOString(),
57
94
  totalUsage: { inputTokens: 0, outputTokens: 0 },
@@ -62,33 +99,58 @@ export class AcodeSdkAdapter extends EventEmitter {
62
99
  runtime: await this.createRuntime(sessionId, config, runtimeHome),
63
100
  autoPollActive: false,
64
101
  });
65
- this.emit('status-change', sessionId, 'starting');
102
+ this.emit("status-change", sessionId, "starting");
66
103
  const entry = this.sessions.get(sessionId);
67
104
  if (!entry) {
68
- throw new Error('ACode session was not initialized');
105
+ throw new Error("ACode session was not initialized");
69
106
  }
70
- await entry.runtime.start();
71
- session.status = 'waiting_input';
72
- this.emit('status-change', sessionId, 'waiting_input');
73
- if (config.initialPrompt?.trim()) {
74
- this.startInitialPromptTurn(sessionId, config.initialPrompt);
107
+ try {
108
+ await entry.runtime.start();
109
+ session.status = "waiting_input";
110
+ this.emit("status-change", sessionId, "waiting_input");
111
+ if (config.initialPrompt?.trim()) {
112
+ this.startInitialPromptTurn(sessionId, config.initialPrompt);
113
+ }
114
+ else {
115
+ // R1 fix: when no initial prompt is provided, explicitly arm the
116
+ // auto-poll loop. Without this, the session sits at 'waiting_input'
117
+ // forever if setIdleCommands hasn't been called yet (or is configured
118
+ // with a non-polling command). setIdleCommands will cancel this timer
119
+ // and replace it with the configured idle loop when invoked.
120
+ this.armIdleLoopOrAutoPoll(sessionId);
121
+ }
122
+ }
123
+ catch (error) {
124
+ session.status = "error";
125
+ this.emitRuntimeErrorEvent(sessionId, error);
126
+ this.emit("status-change", sessionId, "error");
127
+ this.sessions.delete(sessionId);
128
+ throw error;
75
129
  }
76
130
  }
77
131
  async sendMessage(sessionId, message) {
78
132
  const entry = this.requireSession(sessionId);
79
133
  this.cancelScheduledAutoPoll(entry);
134
+ entry.failedTurnRecoveryScheduled = false;
135
+ entry.idlePollLoop?.stop();
80
136
  const userMessage = {
81
137
  id: uuidv4(),
82
138
  sessionId,
83
- role: 'user',
139
+ role: "user",
84
140
  content: message,
85
141
  timestamp: new Date().toISOString(),
86
142
  };
87
143
  entry.adapterSession.messages.push(userMessage);
88
- this.emit('conversation-message', sessionId, userMessage);
89
- entry.adapterSession.status = 'thinking';
90
- this.emit('status-change', sessionId, 'thinking');
91
- await entry.runtime.submit(message);
144
+ this.emit("conversation-message", sessionId, userMessage);
145
+ entry.adapterSession.status = "thinking";
146
+ this.emit("status-change", sessionId, "thinking");
147
+ try {
148
+ await entry.runtime.submit(message);
149
+ this.recoverResolvedTurnWithoutTerminalEvent(sessionId);
150
+ }
151
+ catch (error) {
152
+ this.emitRuntimeError(sessionId, error, { recover: true });
153
+ }
92
154
  }
93
155
  async sendConfirmation(_sessionId, _accept) { }
94
156
  async abortCurrentTurn(sessionId) {
@@ -100,10 +162,12 @@ export class AcodeSdkAdapter extends EventEmitter {
100
162
  return;
101
163
  }
102
164
  this.cancelScheduledAutoPoll(entry);
165
+ entry.failedTurnRecoveryScheduled = false;
166
+ entry.idlePollLoop?.stop();
103
167
  await entry.runtime.stop();
104
- entry.adapterSession.status = 'terminated';
168
+ entry.adapterSession.status = "terminated";
105
169
  this.sessions.delete(sessionId);
106
- this.emit('status-change', sessionId, 'terminated');
170
+ this.emit("status-change", sessionId, "terminated");
107
171
  }
108
172
  getConversation(sessionId) {
109
173
  return this.sessions.get(sessionId)?.adapterSession.messages ?? [];
@@ -122,6 +186,23 @@ export class AcodeSdkAdapter extends EventEmitter {
122
186
  void this.terminateSession(sessionId);
123
187
  }
124
188
  }
189
+ setIdleCommands(sessionId, commands) {
190
+ const entry = this.sessions.get(sessionId);
191
+ if (!entry)
192
+ return;
193
+ entry.idleCommands = commands;
194
+ this.cancelScheduledAutoPoll(entry);
195
+ if (commands.idleInputCommand &&
196
+ isMessagePollingIdleCommand(commands.idleInputCommand)) {
197
+ const loop = this.getIdlePollLoop(entry);
198
+ loop.start();
199
+ return;
200
+ }
201
+ if (entry.adapterSession.status === "waiting_input" &&
202
+ (commands.idleInputCommand || commands.nonInputCommand)) {
203
+ this.getIdlePollLoop(entry).start();
204
+ }
205
+ }
125
206
  /** Injects a runtime constructor for adapter tests without loading the packaged ACode module. */
126
207
  setRuntimeConstructorForTest(runtimeConstructor) {
127
208
  this.runtimeConstructor = runtimeConstructor;
@@ -134,10 +215,12 @@ export class AcodeSdkAdapter extends EventEmitter {
134
215
  configPath: config.mcpConfigPath || getAcodeConfigFile(runtimeHome),
135
216
  skillsDir: getAcodeSkillsDir(runtimeHome),
136
217
  model: config.model,
137
- systemPrompt: typeof config.systemPrompt === 'string' ? config.systemPrompt : undefined,
218
+ systemPrompt: typeof config.systemPrompt === "string"
219
+ ? config.systemPrompt
220
+ : undefined,
138
221
  env: config.envOverrides,
139
222
  });
140
- runtime.on('event', (event) => this.forwardAcodeEvent(event));
223
+ runtime.on("event", (event) => this.forwardAcodeEvent(event));
141
224
  return runtime;
142
225
  }
143
226
  async loadRuntimeConstructor() {
@@ -145,7 +228,7 @@ export class AcodeSdkAdapter extends EventEmitter {
145
228
  return this.runtimeConstructor;
146
229
  }
147
230
  const modulePath = getAcodePackageEntryPath();
148
- const module = await import(toAcodeEsmImportSpecifier(modulePath));
231
+ const module = (await import(toAcodeEsmImportSpecifier(modulePath)));
149
232
  if (!module.ACodeRuntime) {
150
233
  throw new Error(`ACode runtime export not found: ${modulePath}`);
151
234
  }
@@ -155,7 +238,7 @@ export class AcodeSdkAdapter extends EventEmitter {
155
238
  requireSession(sessionId) {
156
239
  const entry = this.sessions.get(sessionId);
157
240
  if (!entry) {
158
- throw new Error('ACode session not found');
241
+ throw new Error("ACode session not found");
159
242
  }
160
243
  return entry;
161
244
  }
@@ -169,22 +252,25 @@ export class AcodeSdkAdapter extends EventEmitter {
169
252
  if (!entry) {
170
253
  return;
171
254
  }
172
- entry.adapterSession.status = 'thinking';
173
- this.emit('status-change', sessionId, 'thinking');
174
- void entry.runtime.submit(initialPrompt).catch((error) => {
255
+ entry.adapterSession.status = "thinking";
256
+ this.emit("status-change", sessionId, "thinking");
257
+ void entry.runtime
258
+ .submit(initialPrompt)
259
+ .then(() => {
260
+ this.recoverResolvedTurnWithoutTerminalEvent(sessionId);
261
+ })
262
+ .catch((error) => {
175
263
  const currentEntry = this.sessions.get(sessionId);
176
264
  if (!currentEntry) {
177
265
  return;
178
266
  }
179
- currentEntry.adapterSession.status = 'error';
180
- const timestamp = new Date().toISOString();
181
- this.emit('event', {
182
- type: 'error',
183
- sessionId,
184
- timestamp,
185
- data: { text: error instanceof Error ? error.message : String(error) },
186
- });
187
- this.emit('status-change', sessionId, 'error');
267
+ // R4 fix: a failed initial prompt turn used to leave the session
268
+ // permanently in 'error' because autoPollActive was false at this
269
+ // point (only set by submitAutoPollPrompt), so the R2 recovery in
270
+ // forwardAcodeEvent skipped it, and no idle loop was armed.
271
+ // Now we always recover to 'waiting_input' and arm the idle loop
272
+ // so the agent can retry via poll_message.
273
+ this.emitRuntimeError(sessionId, error, { recover: true });
188
274
  });
189
275
  }
190
276
  forwardAcodeEvent(event) {
@@ -192,107 +278,280 @@ export class AcodeSdkAdapter extends EventEmitter {
192
278
  if (!entry) {
193
279
  return;
194
280
  }
195
- if (event.type === 'assistant_delta') {
196
- const text = typeof event.data.text === 'string' ? event.data.text : '';
281
+ if (event.type === "assistant_delta") {
282
+ const text = typeof event.data.text === "string" ? event.data.text : "";
197
283
  entry.adapterSession.messages.push({
198
284
  id: uuidv4(),
199
285
  sessionId: event.sessionId,
200
- role: 'assistant',
286
+ role: "assistant",
201
287
  content: text,
202
288
  timestamp: event.timestamp,
203
289
  });
204
- this.emitProviderEvent(event, 'text_delta', { text });
290
+ this.emitProviderEvent(event, "text_delta", { text });
291
+ return;
292
+ }
293
+ if (event.type === "thinking_delta") {
294
+ const text = typeof event.data.text === "string" ? event.data.text : "";
295
+ this.emitProviderEvent(event, "thinking", { text });
296
+ return;
297
+ }
298
+ if (event.type === "ai_request") {
299
+ this.emitProviderEvent(event, "ai_request", {
300
+ model: typeof event.data.model === "string" ? event.data.model : undefined,
301
+ endpoint: typeof event.data.endpoint === "string"
302
+ ? event.data.endpoint
303
+ : undefined,
304
+ stream: typeof event.data.stream === "boolean"
305
+ ? event.data.stream
306
+ : undefined,
307
+ messageCount: typeof event.data.messageCount === "number"
308
+ ? event.data.messageCount
309
+ : undefined,
310
+ toolCount: typeof event.data.toolCount === "number"
311
+ ? event.data.toolCount
312
+ : undefined,
313
+ lastUserMessagePreview: typeof event.data.lastUserMessagePreview === "string"
314
+ ? event.data.lastUserMessagePreview
315
+ : undefined,
316
+ requestBody: typeof event.data.requestBody === "string"
317
+ ? event.data.requestBody
318
+ : undefined,
319
+ });
205
320
  return;
206
321
  }
207
- if (event.type === 'turn_complete') {
322
+ if (event.type === "turn_complete") {
208
323
  const usage = toUsage(event.data.usage);
209
324
  if (usage) {
210
325
  entry.adapterSession.totalUsage.inputTokens += usage.inputTokens;
211
326
  entry.adapterSession.totalUsage.outputTokens += usage.outputTokens;
212
327
  }
213
328
  if (isFailedAcodeTurnComplete(event.data)) {
329
+ // R2 fix: a failed polling-driven turn (e.g., 60s MCP tool timeout,
330
+ // model 5xx, provider unreachable) used to leave the session parked
331
+ // in 'error' or 'thinking' forever, so the agent stopped calling
332
+ // poll_message permanently. Recover to 'waiting_input' and re-arm
333
+ // the auto-poll with a longer backoff so a transient failure
334
+ // doesn't break the message loop.
335
+ //
336
+ // R4 fix: previously this only recovered when autoPollActive was
337
+ // true (i.e. during a submitAutoPollPrompt-driven turn). When the
338
+ // initial prompt turn failed, autoPollActive was false and the
339
+ // session got permanently stuck in 'error'. Now we always recover
340
+ // for non-terminal sessions so the idle loop can retry.
341
+ this.recoverTurnFailure(event.sessionId, entry);
214
342
  return;
215
343
  }
216
- this.emitProviderEvent(event, 'turn_complete', { usage: entry.adapterSession.totalUsage });
217
- entry.adapterSession.status = 'waiting_input';
218
- this.emit('status-change', event.sessionId, 'waiting_input', { usage: entry.adapterSession.totalUsage });
219
- this.scheduleAutoPoll(event.sessionId);
344
+ this.emitProviderEvent(event, "turn_complete", {
345
+ usage: entry.adapterSession.totalUsage,
346
+ responseBody: typeof event.data.responseBody === "string"
347
+ ? event.data.responseBody
348
+ : undefined,
349
+ finishReason: typeof event.data.finishReason === "string"
350
+ ? event.data.finishReason
351
+ : undefined,
352
+ });
353
+ entry.adapterSession.status = "waiting_input";
354
+ this.emit("status-change", event.sessionId, "waiting_input", {
355
+ usage: entry.adapterSession.totalUsage,
356
+ });
357
+ this.armIdleLoopOrAutoPoll(event.sessionId);
220
358
  return;
221
359
  }
222
- if (event.type === 'session_complete') {
223
- entry.adapterSession.status = 'completed';
224
- this.emitProviderEvent(event, 'session_complete', { exitCode: Number(event.data.exitCode) || 0 });
225
- this.emit('status-change', event.sessionId, 'completed');
360
+ if (event.type === "session_complete") {
361
+ this.cancelScheduledAutoPoll(entry);
362
+ entry.failedTurnRecoveryScheduled = false;
363
+ entry.idlePollLoop?.stop();
364
+ entry.adapterSession.status = "completed";
365
+ this.emitProviderEvent(event, "session_complete", {
366
+ exitCode: Number(event.data.exitCode) || 0,
367
+ });
368
+ this.emit("status-change", event.sessionId, "completed");
226
369
  return;
227
370
  }
228
- if (event.type === 'error') {
229
- entry.adapterSession.status = 'error';
230
- this.emitProviderEvent(event, 'error', { text: String(event.data.message || 'ACode runtime error') });
231
- this.emit('status-change', event.sessionId, 'error');
371
+ if (event.type === "error") {
372
+ entry.adapterSession.status = "error";
373
+ this.emitProviderEvent(event, "error", {
374
+ text: String(event.data.message || "ACode runtime error"),
375
+ });
376
+ this.emit("status-change", event.sessionId, "error");
377
+ if (!entry.autoPollActive) {
378
+ this.recoverTurnFailure(event.sessionId, entry);
379
+ }
232
380
  return;
233
381
  }
234
- if (event.type === 'mcp_status') {
235
- const server = typeof event.data.server === 'string' ? event.data.server : 'mcp';
236
- const tool = typeof event.data.tool === 'string' ? event.data.tool : '';
237
- const actionDetail = tool ? `${server}/${tool}` : server;
238
- entry.adapterSession.status = 'tool_using';
239
- this.emitProviderEvent(event, 'tool_use_start', {
382
+ if (event.type === "mcp_status") {
383
+ // R3 fix: the ACode runtime emits a startup mcp_status event with no
384
+ // `action` field but containing a `tools` array describing what the
385
+ // LLM can actually call. Validate that a poll_message tool is exposed,
386
+ // otherwise the message loop cannot continue.
387
+ //
388
+ // The previous version of this check matched the literal string
389
+ // 'aws-mcp' against the user-configured MCP server keys in the
390
+ // `servers` field, which produced false positives when the operator
391
+ // named the entry differently (e.g. 'aws-mcp-server' or 'agent-mcp')
392
+ // while still exposing aws-mcp's tools to the LLM. Worse, on a
393
+ // mismatch it emitted a generic `type: 'error'` ProviderEvent, which
394
+ // the rest of the bridge promotes to a fatal session-level 'error'
395
+ // status that the dashboard cannot recover from — the LLM keeps
396
+ // working but the timeline header gets stuck on "发生错误" forever.
397
+ //
398
+ // The fix below validates against the actual `tools` list (the same
399
+ // tools the LLM sees) and on a missing poll_message it only logs a
400
+ // warning. The runtime will naturally emit a real 'error' event the
401
+ // first time the LLM tries to call a missing tool, which the existing
402
+ // error handling already routes correctly without permanently locking
403
+ // the session status.
404
+ if (!event.data.action && Array.isArray(event.data.tools)) {
405
+ const tools = event.data.tools;
406
+ const hasPollMessage = tools.some((tool) => typeof tool?.name === "string" &&
407
+ tool.name.includes("poll_message"));
408
+ if (!hasPollMessage) {
409
+ const message = "ACode runtime is not exposing a poll_message tool. The LLM cannot call poll_message. " +
410
+ "Please configure an MCP server that provides poll_message (e.g. aws-mcp) in the dashboard panel for this runtime.";
411
+ console.warn(`[${this.displayName}] ${message}`);
412
+ }
413
+ return;
414
+ }
415
+ const server = typeof event.data.server === "string" ? event.data.server : "mcp";
416
+ const tool = typeof event.data.tool === "string" ? event.data.tool : "";
417
+ const toolName = server && tool ? `${server}__${tool}` : tool || server;
418
+ const actionInfo = getToolActionInfo(toolName);
419
+ const actionDetail = toAcodeMcpActionDetail(actionInfo.actionDetail, server, tool);
420
+ const toolUseId = typeof event.data.toolUseId === "string"
421
+ ? event.data.toolUseId
422
+ : undefined;
423
+ const eventData = {
240
424
  toolName: actionDetail,
241
- actionType: 'mcp',
242
- actionLabel: 'Calling MCP',
425
+ actionType: actionInfo.actionType,
426
+ actionLabel: actionInfo.actionLabel,
243
427
  actionDetail,
244
- });
245
- this.emit('status-change', event.sessionId, 'tool_using', {
246
- actionType: 'mcp',
247
- actionLabel: 'Calling MCP',
248
- actionDetail,
249
- });
428
+ ...(toolUseId ? { toolUseId, actionId: toolUseId } : {}),
429
+ };
430
+ if (event.data.action === "tool_result") {
431
+ this.emitProviderEvent(event, "tool_use_end", {
432
+ ...eventData,
433
+ toolResult: typeof event.data.result === "string"
434
+ ? event.data.result
435
+ : JSON.stringify(event.data.result ?? ""),
436
+ });
437
+ entry.adapterSession.status = "thinking";
438
+ this.emit("status-change", event.sessionId, "thinking");
439
+ return;
440
+ }
441
+ const status = actionInfo.actionType === "idle" ? "waiting_input" : "tool_using";
442
+ entry.adapterSession.status = status;
443
+ this.emitProviderEvent(event, "tool_use_start", eventData);
444
+ this.emit("status-change", event.sessionId, status, eventData);
250
445
  return;
251
446
  }
252
- this.emitProviderEvent(event, 'thinking', {
447
+ this.emitProviderEvent(event, "thinking", {
253
448
  text: JSON.stringify(event.data),
254
449
  });
255
450
  }
256
451
  emitProviderEvent(event, type, data) {
257
- this.emit('event', {
452
+ this.emit("event", {
258
453
  type,
259
454
  sessionId: event.sessionId,
260
455
  timestamp: event.timestamp,
261
456
  data,
262
457
  });
263
458
  }
459
+ /**
460
+ * After a completed turn, prefer the configured IdlePollLoop (set via setIdleCommands).
461
+ * Fall back to the legacy auto-poll mechanism when no idle commands are configured.
462
+ */
463
+ armIdleLoopOrAutoPoll(sessionId) {
464
+ const entry = this.sessions.get(sessionId);
465
+ if (!entry || this.isTerminalStatus(entry.adapterSession.status)) {
466
+ return;
467
+ }
468
+ if (entry.idleCommands?.idleInputCommand) {
469
+ entry.idlePollLoop?.start();
470
+ return;
471
+ }
472
+ this.scheduleAutoPoll(sessionId);
473
+ }
474
+ getIdlePollLoop(entry) {
475
+ if (entry.idlePollLoop)
476
+ return entry.idlePollLoop;
477
+ const sessionId = entry.adapterSession.sessionId;
478
+ entry.idlePollLoop = new IdlePollLoop({
479
+ getStatus: () => this.sessions.get(sessionId)?.adapterSession.status,
480
+ getCommand: () => this.sessions.get(sessionId)?.idleCommands?.idleInputCommand,
481
+ isPollingCommand: (command) => isMessagePollingIdleCommand(command),
482
+ dispatchPollingPrompt: () => this.submitAutoPollPrompt(sessionId, ACODE_MESSAGE_POLLING_IDLE_PROMPT),
483
+ dispatchRegularIdleCommand: (command) => this.submitAutoPollPrompt(sessionId, command),
484
+ });
485
+ return entry.idlePollLoop;
486
+ }
487
+ submitAutoPollPrompt(sessionId, prompt) {
488
+ const entry = this.sessions.get(sessionId);
489
+ if (!entry || this.isTerminalStatus(entry.adapterSession.status))
490
+ return;
491
+ entry.autoPollActive = true;
492
+ entry.adapterSession.status = "thinking";
493
+ this.emit("status-change", sessionId, "thinking");
494
+ void entry.runtime
495
+ .submit(prompt)
496
+ .catch((error) => {
497
+ this.emitRuntimeError(sessionId, error);
498
+ })
499
+ .finally(() => {
500
+ const latestEntry = this.sessions.get(sessionId);
501
+ if (!latestEntry)
502
+ return;
503
+ latestEntry.autoPollActive = false;
504
+ // If the R2 recovery path already scheduled a delayed re-arm for a
505
+ // failed polling-driven turn, skip armIdleLoopOrAutoPoll so we do
506
+ // not double-arm (immediate fire via IdlePollLoop AND a 2s timer).
507
+ // The recovery timer is the source of truth here: the flag is just
508
+ // a defensive marker; the timer itself is what matters.
509
+ if (latestEntry.failedTurnRecoveryScheduled ||
510
+ latestEntry.autoPollTimer) {
511
+ latestEntry.failedTurnRecoveryScheduled = false;
512
+ return;
513
+ }
514
+ if (latestEntry.adapterSession.status === "error") {
515
+ this.recoverTurnFailure(sessionId, latestEntry);
516
+ return;
517
+ }
518
+ if (latestEntry.adapterSession.status === "thinking" ||
519
+ latestEntry.adapterSession.status === "tool_using") {
520
+ this.recoverResolvedTurnWithoutTerminalEvent(sessionId);
521
+ return;
522
+ }
523
+ if (latestEntry.adapterSession.status === "waiting_input") {
524
+ this.armIdleLoopOrAutoPoll(sessionId);
525
+ }
526
+ });
527
+ }
264
528
  /**
265
529
  * Continue ACode's headless agent loop after a completed turn.
266
530
  * Main flow: debounce duplicate scheduling -> mark busy -> submit an internal
267
531
  * poll instruction -> let normal ACode events publish MCP/tool/turn states.
532
+ *
533
+ * @param delayMs Optional override for the default 250ms backoff. The R2
534
+ * recovery path uses ACODE_AUTO_POLL_FAILURE_DELAY_MS (2s) to avoid
535
+ * busy-looping when the same failure recurs on every retry.
268
536
  */
269
- scheduleAutoPoll(sessionId) {
537
+ scheduleAutoPoll(sessionId, delayMs = ACODE_AUTO_POLL_DELAY_MS) {
270
538
  const entry = this.sessions.get(sessionId);
271
- if (!entry || entry.autoPollActive || entry.autoPollTimer || this.isTerminalStatus(entry.adapterSession.status)) {
539
+ if (!entry ||
540
+ entry.autoPollActive ||
541
+ entry.autoPollTimer ||
542
+ this.isTerminalStatus(entry.adapterSession.status)) {
272
543
  return;
273
544
  }
274
545
  entry.autoPollTimer = setTimeout(() => {
275
546
  const currentEntry = this.sessions.get(sessionId);
276
- if (!currentEntry || currentEntry.autoPollActive || this.isTerminalStatus(currentEntry.adapterSession.status)) {
547
+ if (!currentEntry ||
548
+ currentEntry.autoPollActive ||
549
+ this.isTerminalStatus(currentEntry.adapterSession.status)) {
277
550
  return;
278
551
  }
279
552
  currentEntry.autoPollTimer = undefined;
280
- currentEntry.autoPollActive = true;
281
- currentEntry.adapterSession.status = 'thinking';
282
- this.emit('status-change', sessionId, 'thinking');
283
- void currentEntry.runtime.submit(ACODE_AUTO_POLL_PROMPT).catch((error) => {
284
- this.emitRuntimeError(sessionId, error);
285
- }).finally(() => {
286
- const latestEntry = this.sessions.get(sessionId);
287
- if (!latestEntry) {
288
- return;
289
- }
290
- latestEntry.autoPollActive = false;
291
- if (latestEntry.adapterSession.status === 'waiting_input') {
292
- this.scheduleAutoPoll(sessionId);
293
- }
294
- });
295
- }, ACODE_AUTO_POLL_DELAY_MS);
553
+ this.submitAutoPollPrompt(sessionId, ACODE_AUTO_POLL_PROMPT);
554
+ }, delayMs);
296
555
  }
297
556
  cancelScheduledAutoPoll(entry) {
298
557
  if (!entry.autoPollTimer) {
@@ -302,21 +561,84 @@ export class AcodeSdkAdapter extends EventEmitter {
302
561
  entry.autoPollTimer = undefined;
303
562
  }
304
563
  isTerminalStatus(status) {
305
- return status === 'completed' || status === 'terminated' || status === 'error';
564
+ return status === "completed" || status === "terminated";
306
565
  }
307
- emitRuntimeError(sessionId, error) {
566
+ emitRuntimeError(sessionId, error, options) {
308
567
  const currentEntry = this.sessions.get(sessionId);
309
568
  if (!currentEntry) {
310
569
  return;
311
570
  }
312
- currentEntry.adapterSession.status = 'error';
571
+ currentEntry.adapterSession.status = "error";
572
+ this.emitRuntimeErrorEvent(sessionId, error);
573
+ this.emit("status-change", sessionId, "error");
574
+ if (options?.recover) {
575
+ this.recoverTurnFailure(sessionId, currentEntry);
576
+ }
577
+ }
578
+ emitRuntimeErrorEvent(sessionId, error) {
313
579
  const timestamp = new Date().toISOString();
314
- this.emit('event', {
315
- type: 'error',
580
+ this.emit("event", {
581
+ type: "error",
316
582
  sessionId,
317
583
  timestamp,
318
584
  data: { text: error instanceof Error ? error.message : String(error) },
319
585
  });
320
- this.emit('status-change', sessionId, 'error');
586
+ }
587
+ recoverTurnFailure(sessionId, entry) {
588
+ if (this.isTerminalStatus(entry.adapterSession.status)) {
589
+ return;
590
+ }
591
+ if (entry.autoPollTimer || entry.failedTurnRecoveryScheduled) {
592
+ if (entry.adapterSession.status !== "waiting_input") {
593
+ entry.adapterSession.status = "waiting_input";
594
+ this.emit("status-change", sessionId, "waiting_input");
595
+ }
596
+ return;
597
+ }
598
+ this.cancelScheduledAutoPoll(entry);
599
+ entry.adapterSession.status = "waiting_input";
600
+ this.emit("status-change", sessionId, "waiting_input");
601
+ entry.failedTurnRecoveryScheduled = true;
602
+ entry.autoPollTimer = setTimeout(() => {
603
+ const currentEntry = this.sessions.get(sessionId);
604
+ if (!currentEntry ||
605
+ currentEntry.autoPollActive ||
606
+ this.isTerminalStatus(currentEntry.adapterSession.status)) {
607
+ if (currentEntry) {
608
+ currentEntry.failedTurnRecoveryScheduled = false;
609
+ }
610
+ return;
611
+ }
612
+ currentEntry.autoPollTimer = undefined;
613
+ currentEntry.failedTurnRecoveryScheduled = false;
614
+ if (currentEntry.idleCommands?.idleInputCommand) {
615
+ this.getIdlePollLoop(currentEntry).start();
616
+ }
617
+ else {
618
+ this.submitAutoPollPrompt(sessionId, ACODE_AUTO_POLL_PROMPT);
619
+ }
620
+ }, ACODE_AUTO_POLL_FAILURE_DELAY_MS);
621
+ }
622
+ /**
623
+ * Recover turns where runtime.submit() resolved but no terminal turn event arrived.
624
+ *
625
+ * Main flow: ACode should normally emit turn_complete/session_complete/error before
626
+ * submit resolves. Some edge paths can emit only intermediate events (for example
627
+ * command_status or tool_result), then resolve. At that point there is no active
628
+ * turn anymore, so keeping the adapter at thinking/tool_using makes the dashboard
629
+ * look permanently stuck. Recover to waiting_input and arm the listening loop.
630
+ */
631
+ recoverResolvedTurnWithoutTerminalEvent(sessionId) {
632
+ const entry = this.sessions.get(sessionId);
633
+ if (!entry || this.isTerminalStatus(entry.adapterSession.status)) {
634
+ return;
635
+ }
636
+ if (entry.adapterSession.status !== "thinking" &&
637
+ entry.adapterSession.status !== "tool_using") {
638
+ return;
639
+ }
640
+ entry.adapterSession.status = "waiting_input";
641
+ this.emit("status-change", sessionId, "waiting_input");
642
+ this.armIdleLoopOrAutoPoll(sessionId);
321
643
  }
322
644
  }