aws-runtime-bridge 1.9.40 → 1.9.45

package/package/acode/dist/compaction.js

@@ -15,8 +15,39 @@
 // ============ 常量（与 OpenCode compaction.ts 对齐） ============
 /** 压缩缓冲区：保留这么多 token 的空闲空间，避免压缩过程中溢出 */
 export const COMPACTION_BUFFER = 20_000;
-/** 基于实际累积 token 的触发比例：当累积量达到上下文窗口的此比例时触发压缩 */
+/**
+ * 基础压缩触发比例（兜底默认值，仅用于 contextLimit <= 0 的异常场景）。
+ * 正常情况下通过 {@link getCompactionTriggerRatio} 按窗口大小动态分级。
+ */
 export const COMPACTION_TRIGGER_RATIO = 0.55;
+/**
+ * 压缩触发比例的窗口分级表。
+ * 大窗口下压缩成本高（LLM 摘要调用 + 上下文丢失风险），应提高比例充分利用窗口；
+ * 小窗口下压缩成本低，可适当降低比例以及时释放空间。
+ *
+ * | 窗口大小       | 比例  |
+ * |---------------|-------|
+ * | < 32K         | 0.70  |
+ * | 32K - 128K    | 0.75  |
+ * | 128K - 512K   | 0.80  |
+ * | ≥ 512K        | 0.85  |
+ */
+const COMPACTION_TRIGGER_RATIO_TIERS = [
+    [32_000, 0.70],
+    [128_000, 0.75],
+    [512_000, 0.80],
+    [Number.POSITIVE_INFINITY, 0.85],
+];
+/** 按上下文窗口大小返回压缩触发比例（大窗口更高，充分利用空间）。 */
+export function getCompactionTriggerRatio(contextLimit) {
+    if (contextLimit <= 0)
+        return COMPACTION_TRIGGER_RATIO;
+    for (const [limit, ratio] of COMPACTION_TRIGGER_RATIO_TIERS) {
+        if (contextLimit < limit)
+            return ratio;
+    }
+    return 0.85;
+}
 /** Prune 保护线默认值：保留最近 PRUNE_PROTECT token 的工具输出完整。
  *  注意：实际使用时会按 contextLimit 比例调整（见 pruneToolOutputs），
  *  避免在小上下文窗口（如 60k）下保护线过高导致 prune 永不触发。
@@ -32,6 +63,57 @@ const PRUNE_PROTECT_RATIO = 0.5;
 const PRUNE_MINIMUM_RATIO = 0.1;
 /** Prune 白名单：这些工具的输出永远不会被裁剪 */
 const PRUNE_PROTECTED_TOOLS = new Set(["skill", "todo_write", "todo_read"]);
+/**
+ * 工具重要性等级（数字越小越重要，越优先保护）。
+ * 用于 prune 时按重要性排序：裁剪时优先移除低重要性的工具输出。
+ */
+const TOOL_IMPORTANCE = {
+    // 最高保护：文件读取和搜索类（结果包含关键上下文）
+    "read_file": 0,
+    "Read": 0,
+    "grep": 0,
+    "Grep": 0,
+    "glob": 0,
+    "Glob": 0,
+    "contextweaver_codebase-retrieval": 0,
+    // 高保护：文件写入和编辑（确认信息）
+    "Edit": 1,
+    "edit": 1,
+    "Write": 1,
+    "write": 1,
+    "write_file": 1,
+    // 中保护：网络请求
+    "webfetch": 2,
+    "WebFetch": 2,
+    "WebSearch": 2,
+    "websearch_web_search_exa": 2,
+    // 低保护：命令行执行（输出通常可丢弃）
+    "bash": 3,
+    "Bash": 3,
+    "run_terminal_command": 3,
+    // 最低保护：UI 操作
+    "playwright_browser_": 4,
+};
+/** 默认重要性（不在 TOOL_IMPORTANCE 中的工具） */
+const DEFAULT_TOOL_IMPORTANCE = 2;
+/**
+ * 获取工具的重要性等级。
+ * 支持前缀匹配（如 "playwright_browser_navigate" 匹配 "playwright_browser_"）。
+ */
+function getToolImportance(toolName) {
+    if (!toolName)
+        return DEFAULT_TOOL_IMPORTANCE;
+    if (TOOL_IMPORTANCE[toolName] !== undefined)
+        return TOOL_IMPORTANCE[toolName];
+    // 前缀匹配（按长度降序）
+    const matches = Object.keys(TOOL_IMPORTANCE)
+        .filter((key) => toolName.startsWith(key))
+        .sort((a, b) => b.length - a.length);
+    if (matches.length > 0) {
+        return TOOL_IMPORTANCE[matches[0]];
+    }
+    return DEFAULT_TOOL_IMPORTANCE;
+}
 /** 默认上下文窗口上限（查找不到模型时使用） */
 const DEFAULT_CONTEXT_LIMIT = 128_000;
 /** 默认最大输出 token */
@@ -76,19 +158,31 @@ export const COMPACTION_LLM_PROMPT_TEMPLATE = [
     "",
     "Output only the summary, no preamble.",
 ].join("\n");
-/** token 估算系数：约 3 字符/token（混合中英文） */
-const ESTIMATED_CHARS_PER_TOKEN = 3;
+/** 中文字符 token 估算系数：约 1.5 字符/token */
+const ESTIMATION_CJK_CHARS_PER_TOKEN = 1.5;
+/** ASCII 字符 token 估算系数：约 4 字符/token */
+const ESTIMATION_ASCII_CHARS_PER_TOKEN = 4;
+/**
+ * Token 估算安全填充系数。
+ * 参考 Claude Code 的 ×4/3 策略：本地估算始终比实际 token 数偏小，
+ * 乘以安全系数确保估算值 >= 实际值，避免在溢出前来不及压缩。
+ */
+const ESTIMATION_SAFE_PADDING = 1.15;
+/** 图片/文档块固定 token 估算值（参考 Claude Code 的 2000 tokens/image） */
+const ESTIMATION_IMAGE_TOKENS = 2000;
+/** 工具调用 JSON 结构附加开销（tool_call_id + role 等） */
+const ESTIMATION_TOOL_OVERHEAD = 8;
 /**
- * 计算压缩触发阈值：取 min(contextLimit × 55%, hardLimit - reserved)。
+ * 计算压缩触发阈值：取 min(contextLimit × 动态比例, hardLimit - reserved)。
  *
- * - ratioBased = contextLimit × COMPACTION_TRIGGER_RATIO（基于窗口比例的软阈值）
+ * - ratioBased = contextLimit × getCompactionTriggerRatio(contextLimit)（按窗口大小动态分级）
  * - hardLimit = inputLimit ?? contextLimit（provider 显式 inputLimit 优先）
  * - reserved = min(COMPACTION_BUFFER, maxOutput)（buffer 不超过输出上限）
  *
  * 取两者较小者，既避免小窗口下过早触发，又避免接近硬上限时来不及压缩。
  */
 export function calculateCompactionThreshold(contextLimit, maxOutput, inputLimit) {
-    const ratioBased = Math.floor(contextLimit * COMPACTION_TRIGGER_RATIO);
+    const ratioBased = Math.floor(contextLimit * getCompactionTriggerRatio(contextLimit));
     const hardLimit = inputLimit ?? contextLimit;
     const reserved = Math.min(COMPACTION_BUFFER, maxOutput);
     const hardLimitBased = hardLimit - reserved;
@@ -172,14 +266,55 @@ export function lookupModelLimits(modelName, configOverrides) {
 }
 // ============ Token 估算 ============
 /**
- * 粗略估算文本的 token 数。
- * 不使用真实 tokenizer（太慢），基于字符数乘经验系数。
- * 对纯 ASCII 约 4 字符/token，对中文约 1.5-2 字符/token，混合取 3 字符/token。
+ * 中英文分离的文本 token 估算。
+ * 对纯 ASCII 约 4 字符/token，对中文约 1.5 字符/token。
+ * 参考：context-manager.ts 的同名方法。
+ */
+export function estimateTextTokens(text) {
+    if (!text)
+        return 0;
+    let cjkCount = 0;
+    let asciiCount = 0;
+    for (const ch of text) {
+        if (ch > "\u00ff") {
+            cjkCount++;
+        }
+        else {
+            asciiCount++;
+        }
+    }
+    return Math.ceil(cjkCount / ESTIMATION_CJK_CHARS_PER_TOKEN + asciiCount / ESTIMATION_ASCII_CHARS_PER_TOKEN);
+}
+/**
+ * 粗略估算文本的 token 数（保留向后兼容）。
+ * 使用 estimateTextTokens 的中英文分离 + 安全填充。
  */
 export function estimateTokens(text) {
     if (!text)
         return 0;
-    return Math.ceil(text.length / ESTIMATED_CHARS_PER_TOKEN);
+    // × ESTIMATION_SAFE_PADDING 确保估算始终偏大（安全侧）
+    // 参考 Claude Code 的 estimateMessageTokens 的 ×4/3 策略
+    return Math.ceil(estimateTextTokens(text) * ESTIMATION_SAFE_PADDING);
+}
+/**
+ * 块级内容 token 估算：识别 text/image_url 等不同块类型分别定价。
+ */
+function estimateContentBlock(block) {
+    if (typeof block === "object" && block !== null) {
+        const record = block;
+        if (record.type === "text" && typeof record.text === "string") {
+            return estimateTokens(record.text);
+        }
+        if (record.type === "image_url" || record.type === "document") {
+            return ESTIMATION_IMAGE_TOKENS;
+        }
+        // 其他类型（tool_result 嵌套等）：JSON 估算
+        return Math.ceil(JSON.stringify(block).length / ESTIMATION_ASCII_CHARS_PER_TOKEN);
+    }
+    if (typeof block === "string") {
+        return estimateTokens(block);
+    }
+    return 0;
 }
 /**
  * 估算单条消息的字节大小。
@@ -206,28 +341,25 @@ function estimateMessageBytes(message) {
 }
 /**
  * 估算整个对话的 token 数。
+ * 使用块级识别 + 中英文分离 + 安全填充。
  */
 export function estimateConversationTokens(messages) {
-    let totalChars = 0;
+    let total = 0;
     for (const msg of messages) {
         if (typeof msg.content === "string") {
-            totalChars += msg.content.length;
+            total += estimateTokens(msg.content);
         }
         else if (Array.isArray(msg.content)) {
             for (const part of msg.content) {
-                if (typeof part === "object" && part !== null) {
-                    totalChars += JSON.stringify(part).length;
-                }
+                total += estimateContentBlock(part);
             }
         }
+        // tool_calls 结构开销
         if (msg.tool_calls) {
-            totalChars += JSON.stringify(msg.tool_calls).length;
-        }
-        if (msg.tool_call_id) {
-            totalChars += msg.tool_call_id.length;
+            total += ESTIMATION_TOOL_OVERHEAD + Math.ceil(JSON.stringify(msg.tool_calls).length / ESTIMATION_ASCII_CHARS_PER_TOKEN);
         }
     }
-    return Math.ceil(totalChars / ESTIMATED_CHARS_PER_TOKEN);
+    return total;
 }
 /**
  * 计算对话的 KB 大小。
@@ -251,7 +383,7 @@ export function isOverflow(messages, contextLimit, _maxOutput) {
     if (contextLimit <= 0)
         return false;
     const totalUsed = estimateConversationTokens(messages);
-    const ratioThreshold = Math.floor(contextLimit * COMPACTION_TRIGGER_RATIO);
+    const ratioThreshold = Math.floor(contextLimit * getCompactionTriggerRatio(contextLimit));
     return totalUsed >= ratioThreshold;
 }
 /**
@@ -260,7 +392,7 @@ export function isOverflow(messages, contextLimit, _maxOutput) {
  * provider 每次请求都会返回实际 token 计数（input + output），
  * 这个计数比基于文本的估算（estimateConversationTokens）要准确得多。
  *
- * 触发条件：累积 tokens >= contextLimit * COMPACTION_TRIGGER_RATIO
+ * 触发条件：累积 tokens >= contextLimit * getCompactionTriggerRatio(contextLimit)
  * 使用显式传入的 contextLimit，不依赖模型名查找表。
  *
  * @param accumulatedInput  provider 实际累积的 input tokens
@@ -271,7 +403,7 @@ export function isOverflowByActualUsage(accumulatedInput, accumulatedOutput, con
     if (contextLimit <= 0)
         return false;
     const totalUsed = accumulatedInput + accumulatedOutput;
-    const ratioThreshold = Math.floor(contextLimit * COMPACTION_TRIGGER_RATIO);
+    const ratioThreshold = Math.floor(contextLimit * getCompactionTriggerRatio(contextLimit));
     return totalUsed >= ratioThreshold;
 }
 // ============ Prune 策略 ============
@@ -335,9 +467,12 @@ export function pruneToolOutputs(messages, contextLimit = DEFAULT_CONTEXT_LIMIT)
     // 按比例调整保护线和最小收益阈值
     const protectLine = Math.min(PRUNE_PROTECT, Math.floor(contextLimit * PRUNE_PROTECT_RATIO));
     const minimumGain = Math.min(PRUNE_MINIMUM, Math.floor(contextLimit * PRUNE_MINIMUM_RATIO));
-    let total = 0;
-    let freedTokens = 0;
-    const toPrune = [];
+    // 阶段 1：收集所有候选（按重要性分组）
+    // 在保护线内的工具输出（最近 N 轮），无论重要性如何都保留。
+    // 超过保护线的输出，按重要性排序，优先裁剪低重要性的。
+    let accumulatedTokens = 0;
+    const lowImportanceCandidates = [];
+    const highImportanceCandidates = [];
     let turns = 0;
     // 从最新往旧遍历
     for (let i = messages.length - 1; i >= 0; i--) {
@@ -353,10 +488,41 @@ export function pruneToolOutputs(messages, contextLimit = DEFAULT_CONTEXT_LIMIT)
             continue;
         const contentStr = typeof msg.content === "string" ? msg.content : "";
         const estimate = estimateTokens(contentStr);
-        total += estimate;
-        if (total > protectLine) {
-            freedTokens += estimate;
-            toPrune.push(msg);
+        accumulatedTokens += estimate;
+        if (accumulatedTokens > protectLine) {
+            // 超过保护线，按重要性分组
+            const importance = toolName ? getToolImportance(toolName) : DEFAULT_TOOL_IMPORTANCE;
+            const candidate = { msg, tokens: estimate, name: toolName ?? "" };
+            if (importance <= 1) {
+                // 高重要性（Read/Grep/Edit 等）：暂不优先裁剪
+                highImportanceCandidates.push(candidate);
+            }
+            else {
+                // 低重要性（Bash/WebSearch 等）：优先裁剪
+                lowImportanceCandidates.push(candidate);
+            }
+        }
+    }
+    // 阶段 2：按重要性排序，低重要性优先裁剪
+    // 低重要性按 token 数降序（先裁大的），高重要性按 token 数升序（尽量保留）
+    lowImportanceCandidates.sort((a, b) => b.tokens - a.tokens);
+    highImportanceCandidates.sort((a, b) => a.tokens - b.tokens);
+    const toPrune = [];
+    let freedTokens = 0;
+    // 先裁低重要性
+    for (const c of lowImportanceCandidates) {
+        if (freedTokens >= minimumGain)
+            break;
+        freedTokens += c.tokens;
+        toPrune.push(c.msg);
+    }
+    // 如果低重要性不够，再裁高重要性（从最小的开始）
+    if (freedTokens < minimumGain) {
+        for (const c of highImportanceCandidates) {
+            freedTokens += c.tokens;
+            toPrune.push(c.msg);
+            if (freedTokens >= minimumGain)
+                break;
         }
     }
     if (freedTokens < minimumGain) {
@@ -569,6 +735,69 @@ export async function summarizeConversation(messages, options) {
     messages.push(...keptMessages);
     return removableCount;
 }
+// ============ 配对不变量保护 ============
+/**
+ * Snip：删除最旧的一组消息（从第一个非 system 消息到下一个 user 消息之前）。
+ *
+ * 作为压缩的最终手段——当 prune 和 summarize 都无法释放足够空间时使用。
+ * 能确保：
+ *   1. 不删除 system 消息
+ *   2. 删除的组不跨 user 边界（保证一个完整轮次）
+ *   3. 删除后至少保留 1 组非 system 消息
+ *
+ * @param messages 对话消息数组（会被原地修改，通过 splice 删除）
+ * @returns 删除的消息数
+ */
+export function snipOldestMessageGroup(messages) {
+    if (messages.length <= 1)
+        return 0;
+    // 找到第一个非 system 消息的索引
+    let startIdx = 0;
+    while (startIdx < messages.length && messages[startIdx].role === "system") {
+        startIdx++;
+    }
+    if (startIdx >= messages.length)
+        return 0;
+    // 找到下一个 user 消息的索引（作为删除组的结束边界）
+    let groupEnd = messages.length;
+    for (let i = startIdx + 1; i < messages.length; i++) {
+        if (messages[i].role === "user") {
+            groupEnd = i;
+            break;
+        }
+    }
+    // 不能删除全部非 system 消息（至少保留一组）
+    if (groupEnd >= messages.length)
+        return 0;
+    // 收集删除范围内 assistant 发起的 tool_call_id
+    const removedToolCallIds = new Set();
+    for (let i = startIdx; i < groupEnd; i++) {
+        const msg = messages[i];
+        if (msg.role === "assistant" && Array.isArray(msg.tool_calls)) {
+            for (const tc of msg.tool_calls) {
+                if (typeof tc === "object" && tc !== null && typeof tc.id === "string") {
+                    removedToolCallIds.add(tc.id);
+                }
+            }
+        }
+    }
+    // 删除主组 [startIdx, groupEnd)
+    const removedCount = groupEnd - startIdx;
+    messages.splice(startIdx, removedCount);
+    // 清理孤儿 tool_result：其对应的 tool_call 已在上一步删除，但 tool_result 可能落在下一轮（groupEnd 之后）。
+    // 不能用连续 splice 扩展范围（会误删中间的 user 消息），故采用逆向遍历逐个删除孤儿 tool_result。
+    let orphanRemoved = 0;
+    if (removedToolCallIds.size > 0) {
+        for (let i = messages.length - 1; i >= startIdx; i--) {
+            const msg = messages[i];
+            if (msg.role === "tool" && msg.tool_call_id && removedToolCallIds.has(msg.tool_call_id)) {
+                messages.splice(i, 1);
+                orphanRemoved++;
+            }
+        }
+    }
+    return removedCount + orphanRemoved;
+}
 // ============ 完整压缩流程 ============
 /**
  * 执行完整压缩流程：Prune → 若仍溢出则 Summarize。
@@ -610,7 +839,16 @@ export async function performCompaction(messages, contextLimit, maxOutput = DEFA
         });
     }
     // 阶段 2: 若仍溢出，执行 Summarize
-    if (isOverflow(messages, contextLimit, maxOutput)) {
+    //
+    // ★ 使用 effectiveUsed（文本估算与 actual 的较大值）而非仅文本估算：
+    //    provider 实际 token 计数往往高于基于字符的估算，若仅依赖文本估算，
+    //    可能因估算值低于阈值而导致"假阴性"——实际已溢出但 isOverflow 返回 false，
+    //    summarize 被直接跳过，最终策略为 "none"。
+    const tokensAfterPrune = estimateConversationTokens(messages);
+    const postPruneEffective = options?.actualUsedTokens
+        ? Math.max(tokensAfterPrune, Math.max(options.actualUsedTokens, tokensBefore) - freedTokens)
+        : tokensAfterPrune;
+    if (postPruneEffective >= Math.floor(contextLimit * getCompactionTriggerRatio(contextLimit))) {
         if (options?.onPhase) {
             options.onPhase({ phase: "summarize_start", messageCount: messages.length });
         }
@@ -629,6 +867,18 @@ export async function performCompaction(messages, contextLimit, maxOutput = DEFA
                 m.content.includes("summarized by LLM"));
             usedLLM = Boolean(summaryMsg);
         }
+        else {
+            // ★ 阶段 2b: Summarize 失败降级 → Snip 最旧消息组（最终手段）
+            // 当 summarize 返回 0（无法找到干净边界或清理后 kept 为空），
+            // 尝试直接删除最旧的一组非 system 消息。
+            const sniped = snipOldestMessageGroup(messages);
+            if (sniped > 0) {
+                strategy = strategy === "prune" ? "prune+summarize" : "summarize";
+                summarizedMessages = sniped;
+                const tokensAfter = estimateConversationTokens(messages);
+                freedTokens = Math.max(freedTokens, tokensBefore - tokensAfter);
+            }
+        }
         if (options?.onPhase) {
             options.onPhase({
                 phase: "summarize_end",
@@ -639,12 +889,15 @@ export async function performCompaction(messages, contextLimit, maxOutput = DEFA
     }
     const sizeAfterKB = conversationSizeKB(messages);
     const messageCountAfter = messages.length;
+    const tokensAfter = estimateConversationTokens(messages);
     if (strategy === "none") {
         return {
             strategy: "none",
             freedTokens: 0,
             sizeBeforeKB,
             sizeAfterKB: sizeBeforeKB,
+            tokensBefore,
+            tokensAfter,
             messageCountBefore,
             messageCountAfter,
             prunedToolMessages: 0,
@@ -656,6 +909,8 @@ export async function performCompaction(messages, contextLimit, maxOutput = DEFA
         freedTokens,
         sizeBeforeKB,
         sizeAfterKB,
+        tokensBefore,
+        tokensAfter,
         messageCountBefore,
         messageCountAfter,
         prunedToolMessages,

package/package/acode/dist/runtime.d.ts

@@ -1,11 +1,42 @@
 import { EventEmitter } from "node:events";
+import { Client } from "@modelcontextprotocol/sdk/client/index.js";
+import type { Transport } from "@modelcontextprotocol/sdk/shared/transport.js";
+import { type JsonObject } from "./built-in-file-tools.js";
 import type { ACodeRuntimeConfig } from "./types.js";
+export interface ACodeMcpConnection {
+    serverName: string;
+    client: Client;
+    transport: Transport;
+    tools: ACodeMcpTool[];
+}
+interface ACodeMcpTool {
+    kind: "mcp";
+    serverName: string;
+    toolName: string;
+    exposedName: string;
+    description?: string;
+    inputSchema: JsonObject;
+}
 export declare class ACodeRuntime extends EventEmitter {
     private readonly config;
     private readonly hookBus;
     private initialConversation;
     private state;
     private abortController;
+    /**
+     * 正在处理中的 memory 子 agent 跟踪表。
+     * key = memory URI，value = subId（用于防止同一 memory 被重复委托）。
+     * 子 agent 完成后通过 SubAgentManager 事件回调移除对应 URI。
+     */
+    private readonly pendingMemorySubAgents;
+    /**
+     * memory 委托冷却表（「已解决/已处理」守卫，区别于 pendingMemorySubAgents 的「进行中」守卫）。
+     * key = memory URI，value = 冷却到期时间戳（ms）。
+     * 子 agent 进入终态时写入；冷却期内即使 memory 仍待处理也不再 spawn 新子任务。
+     * 这是切断「委托完成→复发」循环的关键：原实现只有进行中守卫，子 agent 一结束 URI 立即释放，
+     * 导致下一轮 poll 对 sticky 的 updating memory 再次委托。
+     */
+    private readonly memoryDelegateCooldownUntil;
     private submitQueue;
     private activeTurn;
     private abortRequested;
@@ -29,6 +60,8 @@ export declare class ACodeRuntime extends EventEmitter {
      */
     private readonly questionWaiters;
     constructor(config: ACodeRuntimeConfig);
+    /** 获取当前已暴露的工具名称列表（供 SubAgentManager 记录） */
+    getAvailableTools(): string[];
     private emitAcodeEvent;
     /**
      * 发射压缩子事件（phase 事件）到面板时间线。
@@ -168,6 +201,92 @@ export declare class ACodeRuntime extends EventEmitter {
     private readChatCompletionStream;
     private handleToolCalls;
     private callMcpToolWithTimeout;
+    /**
+     * 将子 agent 终态通知转为 poll_message 格式的合成 DirectMessage。
+     * 主流程：构建符合 MCP poll_message 返回格式的 JSON 字符串。
+     */
+    private static formatSubAgentCompletionAsPollResult;
+    /**
+     * 将子 agent 终态通知合并到已有的 MCP poll_message 返回结果中。
+     * 主流程：解析 MCP 返回的 JSON -> 将通知转为 DirectMessage 插入到 directMessages 前面 -> 重新序列化。
+     */
+    private static mergeSubAgentCompletionsIntoMcpResult;
+    /**
+     * Memory 自动委托 sub-agent 完成一轮后的外部重新检查。
+     *
+     * 查询服务端 memory 真实状态（通过 MCP 工具直接调用，非 LLM 自报），
+     * 判断 sub-agent 是否真正完成了处理：
+     * - memory_file_change：检查每条 memory 的 memory_status 是否已回到 normal
+     * - memory_expiration：检查每条 memory 是否已续期（expiresAt 在未来）或已清理（read 返回 null）
+     * - memory_cleanup：检查 get_memory_stats 中对应类型/域是否仍 exceeded
+     *
+     * 若仍有未完成项，返回 { done: false, continueMessage } 包含 "经检查，未完成对xxx的处理，请继续"
+     * 及具体未完成列表，由 SubAgentManager 注入同一 sub-agent 的下一轮 submit。
+     *
+     * recheck 自身出错（MCP 连接不可用、工具调用超时等）时返回 { done: true }，不阻塞 sub-agent 结束。
+     */
+    private recheckMemoryDelegate;
+    /**
+     * 通过 MCP read_memory 工具读取单条 memory 记录。
+     * 注入 _aws_sub_agent=true 避免状态上报污染父 agent 时间线。
+     * 返回 null 表示 memory 明确不存在（服务端返回 null/undefined 或非对象）。
+     * 读取/解析异常（网络抖动、超时、服务端错误）会向上抛出，交由 recheckMemoryDelegate
+     * 统一判定为「未完成」，避免被误当作「已处理」。
+     */
+    private readMemoryViaMcp;
+    /** recheck memory_file_change：检查 memory_status 是否仍为 updating */
+    private recheckMemoryFileChange;
+    /** recheck memory_expiration：检查 expiresAt 是否仍过期 或 memory_status 是否仍为 recalling */
+    private recheckMemoryExpiration;
+    /** recheck memory_cleanup：检查 get_memory_stats 中对应类型/域是否仍 exceeded */
+    private recheckMemoryCleanup;
+    /**
+     * 构造 memory 自动委托子 agent 的 prompt。
+     * 主流程：根据 delegate 类型（过期整理/文件变更更新）生成自包含指令。
+     */
+    private buildMemoryDelegatePrompt;
+    /**
+     * 处理 poll_message 结果中的 memory 自动委托标记 (_acodeAutoDelegate)。
+     * 主流程：解析 MCP 返回 -> 识别标记 -> spawn 子 agent -> 返回简化结果。
+     * 非 ACode 运行时（无 subAgentManager）直接透传原结果，不受影响。
+     * 返回 null 表示全部消息已被委托，调用方应重新 poll_message 继续等待真正的消息。
+     */
+    private processMemoryAutoDelegates;
+    /**
+     * poll_message MCP 工具调用的子 agent 完成事件 race 包装器。
+     *
+     * 主流程：
+     * 1. 先检查 completionQueue 中是否有已完成的子 agent -> 有则直接返回合成结果（不调 MCP）
+     * 2. 启动 MCP poll_message 调用与 waitForAnyCompletion 的 race
+     * 3. 子 agent 先完成 -> 返回合成结果（MCP 调用被放弃但仍在运行，下次 poll_message 会复用 pendingPollMessage）
+     * 4. MCP 先返回 -> 合并新完成的子 agent 事件后返回
+     *
+     * 设计说明：
+     * - MCP server 的 poll_message 内部有 while(true) 永久阻塞，只在有消息时返回
+     * - 子 agent 完成事件由 SubAgentManager 在同进程内产生，通过 completionQueue 传递
+      * - 当子 agent 先完成时，MCP 调用被放弃（Promise 不再被 await），MCP server 侧的
+      *   pendingPollMessage 仍在运行，下次 poll_message 调用会通过 pendingPollMessage 复用
+      */
+    private callPollMessageWithSubAgentRace;
+    /**
+     * 检查 poll_message 结果是否包含实际消息（directMessages 或 groupMessages 非空）。
+     * 用于循环中过滤空结果，确保 poll_message 只在有消息时返回。
+     */
+    private pollResultHasMessages;
+    /**
+     * 在 memory 委托消息被全部处理后，等待一段时间再重新 poll_message。
+     *
+     * 背景：MCP doPollMessage 对 updating memory 每次都返回通知（无去重），
+     * 而 processMemoryAutoDelegates 因 URI 已在 pendingMemorySubAgents 中而返回 null，
+     * 导致 while(true) 无延迟重 poll，每轮触发 handleToolCall 的 reportToolUse/reportThinking
+     * 状态上报，引发"思考→MCP→跳过"快速循环，最终连接丢失。
+     *
+     * 等待策略：
+     * 1. 固定延迟 DELEGATE_REPOLL_DELAY_MS（5s），给子 agent 时间完成 memory 更新
+     * 2. 可被 runtime stop 中断（通过 createRuntimeStopWaiter）
+     * 3. 不消费 completionQueue（下一轮 takeCompletedSubAgents 会取到）
+     */
+    private waitForDelegateRepoll;
     /**
      * Build a one-shot waiter that rejects long-running MCP calls when the runtime stops.
      * Main flow: register a stop callback -> race it with the SDK call -> remove it after either side settles.
@@ -180,6 +299,16 @@ export declare class ACodeRuntime extends EventEmitter {
     private notifyRuntimeStop;
     private appendSyntheticMultimodalUserMessage;
     private appendToolFailureResult;
+    /**
+     * 为子 agent 过滤共享 MCP 连接：只保留 memory 工具。
+     *
+     * 原 MCP server 进程通过 AWS_SUB_AGENT=true 限制子 agent 只能调用 memory 工具，
+     * 共享连接后绕过了该机制。此方法在 ACode runtime 层重新施加限制：
+     *   - 复用同一 client/transport（不创建新连接）
+     *   - 仅暴露 MEMORY_TOOL_NAMES 中的工具
+     *   - 防止子 agent 调用 poll_message（与父 agent 冲突）、send_group 等
+     */
+    private filterSharedMcpConnectionsForSubAgent;
     private connectMcpServers;
     private createMcpTransport;
     /**
@@ -196,4 +325,5 @@ export declare class ACodeRuntime extends EventEmitter {
     private extractProfileFromToolResult;
     private disconnectMcpServers;
 }
+export {};
 //# sourceMappingURL=runtime.d.ts.map

package/package/acode/dist/runtime.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"runtime.d.ts","sourceRoot":"","sources":["../src/runtime.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AA+C3C,OAAO,KAAK,EAOV,kBAAkB,EAGnB,MAAM,YAAY,CAAC;AA4oCpB,qBAAa,YAAa,SAAQ,YAAY;IAgEhC,OAAO,CAAC,QAAQ,CAAC,MAAM;IA/DnC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAe;IACvC,OAAO,CAAC,mBAAmB,CAAqB;IAEhD,OAAO,CAAC,KAAK,CAsBX;IAEF,OAAO,CAAC,eAAe,CAAgC;IAEvD,OAAO,CAAC,WAAW,CAAoC;IAEvD,OAAO,CAAC,UAAU,CAAS;IAE3B,OAAO,CAAC,cAAc,CAAS;IAE/B,OAAO,CAAC,sBAAsB,CAAS;IACvC,OAAO,CAAC,cAAc,CAAK;IAC3B,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAyB;IACrD,4BAA4B;IAC5B,OAAO,CAAC,oBAAoB,CAA+C;IAC3E,0DAA0D;IAC1D,OAAO,CAAC,SAAS,CAA+B;IAChD,mCAAmC;IACnC,OAAO,CAAC,aAAa,CAAsB;IAC3C,oCAAoC;IACpC,OAAO,CAAC,iBAAiB,CAA8C;IACvE,uBAAuB;IACvB,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,yBAAyB,CAAO;IAExD;;;;OAIG;IACH,OAAO,CAAC,QAAQ,CAAC,eAAe,CAOlB;gBAEe,MAAM,EAAE,kBAAkB;IAevD,OAAO,CAAC,cAAc;IAatB;;;;;;;;OAQG;IACH,OAAO,CAAC,mBAAmB;IAgB3B;;OAEG;IACH,OAAO,CAAC,qBAAqB;IAqB7B;;;;OAIG;IACG,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAsK5B;;;;;;;;;;OAUG;YACW,aAAa;IA+C3B;;;;;;;;OAQG;IACH,OAAO,CAAC,kBAAkB;IA2B1B;;OAEG;IACH,OAAO,CAAC,oBAAoB;IAU5B;;;;;;;;OAQG;YACW,aAAa;IAuC3B;;OAEG;IACH,OAAO,CAAC,iBAAiB;IAezB;;;OAGG;IACH,OAAO,CAAC,uBAAuB;IAqD/B;;OAEG;IACH,OAAO,CAAC,sBAAsB;IAO9B;;;;OAIG;IACG,MAAM,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAS5C;;;OAGG;IACG,YAAY,IAAI,OAAO,CAAC,IAAI,CAAC;IAanC,OAAO,CAAC,gBAAgB;YAIV,aAAa;IAgP3B,oDAAoD;IAC9C,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAQ5B;;;;OAIG;IACH,OAAO,CAAC,wBAAwB;IAuBhC;;;;;;;;;OASG;IACH,OAAO,CAAC,qBAAqB;IA4E7B;;;OAGG;IACH,oBAAoB,CAClB,UAAU,EAAE,MAAM,EAClB,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAC9B,OAAO;IAeV;;;OAGG;IACH,OAAO,CAAC,wBAAwB;IAQhC,uCAAuC;IACjC,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IA4C3B,OAAO,CAAC,kBAAkB;YAaZ,YAAY;IA4K1B;;;;;;;;;;;;;;;;;OAiBG;YACW,4BAA4B;YAwE5B,qBAAqB;YA2TrB,wBAAwB;YAsFxB,eAAe;YAwSf,sBAAsB;IA6CpC;;;OAGG;IACH,OAAO,CAAC,uBAAuB;IAiC/B;;;OAGG;IACH,OAAO,CAAC,iBAAiB;YAQX,oCAAoC;IA0DlD,OAAO,CAAC,uBAAuB;YA0BjB,iBAAiB;IA2C/B,OAAO,CAAC,kBAAkB;IA2B1B;;;;;OAKG;YACW,kBAAkB;IAgChC;;;OAGG;IACH,OAAO,CAAC,4BAA4B;YA4BtB,oBAAoB;CAiBnC"}
+{"version":3,"file":"runtime.d.ts","sourceRoot":"","sources":["../src/runtime.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAK3C,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AAInE,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,+CAA+C,CAAC;AAE/E,OAAO,EAIL,KAAK,UAAU,EAChB,MAAM,0BAA0B,CAAC;AA+BlC,OAAO,KAAK,EAOV,kBAAkB,EAGnB,MAAM,YAAY,CAAC;AAEpB,MAAM,WAAW,kBAAkB;IACjC,UAAU,EAAE,MAAM,CAAC;IACnB,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,SAAS,CAAC;IACrB,KAAK,EAAE,YAAY,EAAE,CAAC;CACvB;AAED,UAAU,YAAY;IACpB,IAAI,EAAE,KAAK,CAAC;IACZ,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,WAAW,EAAE,MAAM,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,WAAW,EAAE,UAAU,CAAC;CACzB;AAiyCD,qBAAa,YAAa,SAAQ,YAAY;IAkFhC,OAAO,CAAC,QAAQ,CAAC,MAAM;IAjFnC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAe;IACvC,OAAO,CAAC,mBAAmB,CAAqB;IAEhD,OAAO,CAAC,KAAK,CAwBX;IAEF,OAAO,CAAC,eAAe,CAAgC;IAEvD;;;;OAIG;IACH,OAAO,CAAC,QAAQ,CAAC,sBAAsB,CAAkC;IAEzE;;;;;;OAMG;IACH,OAAO,CAAC,QAAQ,CAAC,2BAA2B,CAAkC;IAE9E,OAAO,CAAC,WAAW,CAAoC;IAEvD,OAAO,CAAC,UAAU,CAAS;IAE3B,OAAO,CAAC,cAAc,CAAS;IAE/B,OAAO,CAAC,sBAAsB,CAAS;IACvC,OAAO,CAAC,cAAc,CAAK;IAC3B,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAyB;IACrD,4BAA4B;IAC5B,OAAO,CAAC,oBAAoB,CAA+C;IAC3E,0DAA0D;IAC1D,OAAO,CAAC,SAAS,CAA+B;IAChD,mCAAmC;IACnC,OAAO,CAAC,aAAa,CAAsB;IAC3C,oCAAoC;IACpC,OAAO,CAAC,iBAAiB,CAA8C;IACvE,uBAAuB;IACvB,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,yBAAyB,CAAO;IAExD;;;;OAIG;IACH,OAAO,CAAC,QAAQ,CAAC,eAAe,CAOlB;gBAEe,MAAM,EAAE,kBAAkB;IAevD,2CAA2C;IAC3C,iBAAiB,IAAI,MAAM,EAAE;IAI7B,OAAO,CAAC,cAAc;IAatB;;;;;;;;OAQG;IACH,OAAO,CAAC,mBAAmB;IAgB3B;;OAEG;IACH,OAAO,CAAC,qBAAqB;IAqB7B;;;;OAIG;IACG,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAwL5B;;;;;;;;;;OAUG;YACW,aAAa;IA+C3B;;;;;;;;OAQG;IACH,OAAO,CAAC,kBAAkB;IA2B1B;;OAEG;IACH,OAAO,CAAC,oBAAoB;IAU5B;;;;;;;;OAQG;YACW,aAAa;IAuC3B;;OAEG;IACH,OAAO,CAAC,iBAAiB;IAezB;;;OAGG;IACH,OAAO,CAAC,uBAAuB;IA6C/B;;OAEG;IACH,OAAO,CAAC,sBAAsB;IAO9B;;;;OAIG;IACG,MAAM,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAwB5C;;;OAGG;IACG,YAAY,IAAI,OAAO,CAAC,IAAI,CAAC;IAanC,OAAO,CAAC,gBAAgB;YAIV,aAAa;IA8O3B,oDAAoD;IAC9C,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAsB5B;;;;OAIG;IACH,OAAO,CAAC,wBAAwB;IAuBhC;;;;;;;;;OASG;IACH,OAAO,CAAC,qBAAqB;IAiF7B;;;OAGG;IACH,oBAAoB,CAClB,UAAU,EAAE,MAAM,EAClB,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAC9B,OAAO;IAeV;;;OAGG;IACH,OAAO,CAAC,wBAAwB;IAQhC,uCAAuC;IACjC,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IA4C3B,OAAO,CAAC,kBAAkB;YAaZ,YAAY;IA6K1B;;;;;;;;;;;;;;;;;OAiBG;YACW,4BAA4B;YAqF5B,qBAAqB;YAmWrB,wBAAwB;YAsFxB,eAAe;YA0Vf,sBAAsB;IAmDpC;;;OAGG;IACH,OAAO,CAAC,MAAM,CAAC,oCAAoC;IAiCnD;;;OAGG;IACH,OAAO,CAAC,MAAM,CAAC,qCAAqC;IAiDpD;;;;;;;;;;;;;OAaG;YACW,qBAAqB;IAwCnC;;;;;;OAMG;YACW,gBAAgB;IA8B9B,gEAAgE;YAClD,uBAAuB;IAkCrC,kFAAkF;YACpE,uBAAuB;IAyCrC,qEAAqE;YACvD,oBAAoB;IAwElC;;;OAGG;IACH,OAAO,CAAC,yBAAyB;IAyEjC;;;;;OAKG;YACW,0BAA0B;IA2JxC;;;;;;;;;;;;;;QAcI;YACU,+BAA+B;IAkG7C;;;OAGG;IACH,OAAO,CAAC,qBAAqB;IAY7B;;;;;;;;;;;;OAYG;YACW,qBAAqB;IAkBnC;;;OAGG;IACH,OAAO,CAAC,uBAAuB;IAiC/B;;;OAGG;IACH,OAAO,CAAC,iBAAiB;YAQX,oCAAoC;IA0DlD,OAAO,CAAC,uBAAuB;IA8B/B;;;;;;;;OAQG;IACH,OAAO,CAAC,qCAAqC;YAW/B,iBAAiB;IA2C/B,OAAO,CAAC,kBAAkB;IA2B1B;;;;;OAKG;YACW,kBAAkB;IAgChC;;;OAGG;IACH,OAAO,CAAC,4BAA4B;YA4BtB,oBAAoB;CAuBnC"}