autosnippet 2.8.3 → 2.10.0

package/lib/service/chat/ContextWindow.js

@@ -218,7 +218,7 @@ export class ContextWindow {
     for (const m of removed) {
       if (m.role === 'assistant' && m.toolCalls) {
         for (const tc of m.toolCalls) {
-          if (tc.name === 'submit_candidate' || tc.name === 'submit_with_check') {
+          if (tc.name === 'submit_knowledge' || tc.name === 'submit_with_check') {
             this.#compactedSubmits.add(tc.args?.title || tc.args?.category || 'untitled');
           }
         }
@@ -339,7 +339,7 @@ export class ContextWindow {
         const m = this.#messages[i];
         if (m.role === 'assistant' && m.toolCalls) {
           for (const tc of m.toolCalls) {
-            if (tc.name === 'submit_candidate' || tc.name === 'submit_with_check') {
+            if (tc.name === 'submit_knowledge' || tc.name === 'submit_with_check') {
               this.#compactedSubmits.add(tc.args?.title || tc.args?.category || 'untitled');
             }
           }
@@ -393,19 +393,33 @@ export class ContextWindow {
 export function limitToolResult(toolName, result, quota) {
   const { maxChars = 4000, maxMatches = 10 } = quota;
 
-  // submit_candidate / submit_with_check 结果很短，不截断
-  if (toolName === 'submit_candidate' || toolName === 'submit_with_check') {
+  // submit_knowledge / submit_with_check 结果很短，不截断
+  if (toolName === 'submit_knowledge' || toolName === 'submit_with_check') {
     const raw = typeof result === 'string' ? result : JSON.stringify(result);
     return raw.length > 500 ? raw.substring(0, 500) : raw;
   }
 
-  // search_project_code: 限制匹配数 + 截断上下文
+  // search_project_code: 限制匹配数 + 截断上下文（支持批量模式）
   if (toolName === 'search_project_code') {
+    if (result && typeof result === 'object' && result.batchResults) {
+      // 批量模式：对每个 pattern 的结果独立限制（直接操作对象，避免 stringify→parse 往返）
+      const limited = { ...result };
+      const perKeyChars = Math.floor(maxChars / Object.keys(limited.batchResults).length);
+      for (const [key, sub] of Object.entries(limited.batchResults)) {
+        limited.batchResults[key] = limitSearchResultObj(sub, Math.min(maxMatches, 3), perKeyChars);
+      }
+      const raw = JSON.stringify(limited);
+      return raw.length > maxChars ? raw.substring(0, maxChars) + '\n... [batch truncated]' : raw;
+    }
     return limitSearchResult(result, maxMatches, maxChars);
   }
 
-  // read_project_file: 限制字符数
+  // read_project_file: 限制字符数（支持批量模式）
   if (toolName === 'read_project_file') {
+    if (result && typeof result === 'object' && result.batchResults) {
+      const raw = JSON.stringify(result);
+      return raw.length > maxChars ? raw.substring(0, maxChars) + '\n... [batch truncated]' : raw;
+    }
     return limitFileContent(result, maxChars);
   }
 
@@ -465,6 +479,41 @@ function limitSearchResult(result, maxMatches, maxChars) {
   return str;
 }
 
+/**
+ * 限制搜索结果（返回对象） — 用于批量模式，避免 JSON.stringify → JSON.parse 往返
+ * 当源码含控制字符时，stringify→substring 截断会破坏 JSON 结构导致 parse 失败
+ */
+function limitSearchResultObj(result, maxMatches, maxChars) {
+  if (!result || typeof result !== 'object') return result || {};
+  if (typeof result === 'string') return { _raw: result.substring(0, maxChars) };
+
+  const limited = { ...result };
+  if (Array.isArray(limited.matches)) {
+    limited.matches = limited.matches.slice(0, maxMatches).map(m => {
+      const copy = { ...m };
+      if (copy.context && typeof copy.context === 'string') {
+        const contextLines = copy.context.split('\n');
+        if (contextLines.length > 7) {
+          copy.context = contextLines.slice(0, 7).join('\n') + '\n... [truncated]';
+        }
+        // 按字符上限截断 context（防止单个代码块过大）
+        if (copy.context.length > 500) {
+          copy.context = copy.context.substring(0, 500) + '\n... [truncated]';
+        }
+      }
+      if (Array.isArray(copy.lines) && copy.lines.length > 5) {
+        copy.lines = copy.lines.slice(0, 5);
+        copy._truncated = true;
+      }
+      return copy;
+    });
+    if (result.matches.length > maxMatches) {
+      limited._note = `Showing ${maxMatches} of ${result.matches.length} matches`;
+    }
+  }
+  return limited;
+}
+
 /**
  * 限制文件内容 — 截断 content 字段
  */
@@ -526,6 +575,9 @@ export class PhaseRouter {
   /** @type {Object} 日志器 */
   #logger;
 
+  /** @type {boolean} 是否因 maxIterations 强制进入 SUMMARIZE */
+  #forcedSummarize = false;
+
   /**
    * @param {Object} budget — 预算配置
    * @param {boolean} isSkillOnly — 是否为 skill-only 维度
@@ -561,6 +613,19 @@ export class PhaseRouter {
     return this.#totalSubmits;
   }
 
+  /**
+   * 是否因 maxIterations 触顶而强制进入 SUMMARIZE（一次性标记）
+   * 调用后自动复位
+   * @returns {boolean}
+   */
+  consumeForcedSummarize() {
+    if (this.#forcedSummarize) {
+      this.#forcedSummarize = false;
+      return true;
+    }
+    return false;
+  }
+
   /**
    * 获取当前阶段的 toolChoice
    * @returns {'required'|'auto'|'none'}
@@ -585,8 +650,21 @@ export class PhaseRouter {
    * @returns {boolean}
    */
   shouldExit() {
-    if (this.#totalIterations >= this.#budget.maxIterations) return true;
+    // 已在 SUMMARIZE 阶段 → 给 2 轮输出总结后退出
     if (this.#phase === 'SUMMARIZE' && this.#phaseRounds >= 2) return true;
+
+    // 达到 maxIterations → 不硬退出，而是强制转入 SUMMARIZE 让 AI 在完整上下文中收尾
+    if (this.#totalIterations >= this.#budget.maxIterations && this.#phase !== 'SUMMARIZE') {
+      this.#logger.info(`[PhaseRouter] maxIterations reached (${this.#totalIterations}/${this.#budget.maxIterations}), forcing → SUMMARIZE for graceful exit`);
+      this.#forcedSummarize = true;
+      this.#transitionTo('SUMMARIZE');
+      // 返回 false 让主循环继续运行 SUMMARIZE 阶段的 2 轮收尾
+      return false;
+    }
+
+    // SUMMARIZE 阶段超限兜底（maxIterations + 2 轮 grace）
+    if (this.#totalIterations >= this.#budget.maxIterations + 2) return true;
+
     return false;
   }
 
@@ -688,6 +766,12 @@ export class PhaseRouter {
    * @returns {string|null}
    */
   getPhaseHint() {
+    // 接近 maxIterations 上限时，无论处于哪个阶段都警告 AI
+    const remaining = this.#budget.maxIterations - this.#totalIterations;
+    if (remaining <= 2 && remaining > 0 && this.#phase !== 'SUMMARIZE') {
+      return `⚠️ 仅剩 ${remaining} 轮次即达上限，请尽快完成当前工作并准备输出总结。`;
+    }
+
     switch (this.#phase) {
       case 'EXPLORE':
         if (this.#phaseRounds >= this.#budget.searchBudget - 2) {
@@ -699,7 +783,7 @@ export class PhaseRouter {
         if (this.#totalSubmits === 0 && this.#phaseRounds >= 1) {
           return this.#isSkillOnly
             ? '你已收集足够信息，请在回复中直接输出 dimensionDigest JSON。'
-            : '⚠️ 探索阶段已结束。你已收集了足够的项目信息，请 **立即** 调用 submit_candidate 提交候选。不要继续搜索，直接提交。';
+            : '⚠️ 探索阶段已结束。你已收集了足够的项目信息，请 **立即** 调用 submit_knowledge 提交候选。不要继续搜索，直接提交。';
         }
         if (this.#totalSubmits >= this.#budget.softSubmitLimit && this.#budget.softSubmitLimit > 0) {
           const remaining = this.#budget.maxSubmits - this.#totalSubmits;

package/lib/service/chat/HandoffProtocol.js

@@ -13,6 +13,31 @@
 // AnalysisReport 构建
 // ──────────────────────────────────────────────────────────────────
 
+/**
+ * 清理 Analyst 分析文本中可能泄漏的系统 nudge / graceful exit 指令。
+ * 这些内容如果传给 Producer，会干扰其正常工作流。
+ */
+function sanitizeAnalysisText(text) {
+  if (!text) return '';
+  // 移除 graceful exit nudge 及 digest 模板指令
+  const patterns = [
+    /\*{0,2}⚠️?\s*(?:你已使用|轮次即将耗尽|仅剩|请立即停止|必须立即结束)[^\n]*\n?/gi,
+    /\*{0,2}请立即停止所有工具调用[^\n]*\*{0,2}\n?/gi,
+    /请在回复中直接输出\s*dimensionDigest\s*JSON[^\n]*\n?/gi,
+    /> ?(?:remainingTasks|如果所有信号都已覆盖)[^\n]*\n?/gi,
+    /> ?⚠️ 严禁输出任何非 JSON 内容[^\n]*\n?/gi,
+    // 移除 AI 回显的 dimensionDigest JSON 块（对 Producer 无价值且会干扰）
+    /```json\s*\n\s*\{\s*"dimensionDigest"\s*:[\s\S]*?\n```/g,
+  ];
+  let cleaned = text;
+  for (const pat of patterns) {
+    cleaned = cleaned.replace(pat, '');
+  }
+  // 移除可能残留的空行堆积
+  cleaned = cleaned.replace(/\n{3,}/g, '\n\n').trim();
+  return cleaned;
+}
+
 /**
  * 从 Analyst 的执行结果构建 AnalysisReport
  *
@@ -73,7 +98,7 @@ export function buildAnalysisReport(analystResult, dimensionId, projectGraph = n
   }
 
   // 从分析文本中提取文件路径
-  const text = analystResult.reply || '';
+  const text = sanitizeAnalysisText(analystResult.reply || '');
   const textFileRefs = text.match(/[\w/.-]+\.[mhswift]+/g);
   if (textFileRefs) {
     for (const f of textFileRefs) {

package/lib/service/chat/ProducerAgent.js

@@ -2,7 +2,7 @@
  * ProducerAgent.js — v3.0 生产者 Agent
  *
  * 职责:
- * - 将 Analyst 的分析文本转换为结构化的 submit_candidate 调用
+ * - 将 Analyst 的分析文本转换为结构化的 submit_knowledge 调用
  * - 遵循 PROJECT_SNAPSHOT_STYLE_GUIDE 格式
  * - 使用 read_project_file 获取代码片段
  * - CandidateGuardrail 验证每次提交
@@ -21,7 +21,7 @@ import Logger from '../../infrastructure/logging/Logger.js';
 
 const PRODUCER_SYSTEM_PROMPT = `你是知识管理专家。你会收到一段代码分析文本，需要将其中的知识点转化为结构化的知识候选。
 
-核心原则: 分析文本已经包含了所有发现，你的唯一工作是将它们格式化为 submit_candidate 调用。
+核心原则: 分析文本已经包含了所有发现，你的唯一工作是将它们格式化为 submit_knowledge 调用。
 
 每个候选必须:
 1. 有清晰的标题 (描述知识点的核心，使用项目真实类名)
@@ -31,12 +31,14 @@ const PRODUCER_SYSTEM_PROMPT = `你是知识管理专家。你会收到一段代
 
 工作流程:
 1. 阅读分析文本，识别每个独立的知识点/发现
-2. 对每个知识点，用 read_project_file 获取关键代码片段（读取 30-80 行，不要只读 5 行头部）
-3. 立刻调用 submit_candidate 提交
+2. 用 read_project_file 批量获取关键代码片段:
+   read_project_file({ filePaths: ["FileA.m", "FileB.m"], maxLines: 80 })
+3. 立刻调用 submit_knowledge 提交
 4. 重复直到分析中的所有知识点都已提交
 
 关键规则:
 - 分析中的每个要点/段落都应转化为至少一个候选
+- read_project_file 支持 filePaths 数组批量读取多个文件，一次调用完成
 - read_project_file 时读取足够多的行数（startLine + maxLines 至少 30 行）
 - reasoning.sources 必须是非空数组，填写相关文件路径如 ["FileName.m"]
 - 如果分析提到了 3 个模式，就应该提交 3 个候选，不要合并
@@ -53,7 +55,7 @@ const PRODUCER_SYSTEM_PROMPT = `你是知识管理专家。你会收到一段代
 // ──────────────────────────────────────────────────────────────────
 
 const PRODUCER_TOOLS = [
-  'submit_candidate',
+  'submit_knowledge',
   'submit_with_check',
   'read_project_file',
 ];
@@ -63,7 +65,7 @@ const PRODUCER_TOOLS = [
 // ──────────────────────────────────────────────────────────────────
 
 const PRODUCER_BUDGET = {
-  maxIterations: 15,
+  maxIterations: 24,      // was 18 — 与 Analyst 统一
   searchBudget: 4,
   searchBudgetGrace: 3,
   maxSubmits: 10,
@@ -77,7 +79,7 @@ const PRODUCER_BUDGET = {
 
 const STYLE_GUIDE = `# 「项目特写」写作要求
 
-submit_candidate 的 code 字段必须是「项目特写」。
+submit_knowledge 的 code 字段必须是「项目特写」。
 
 ## 什么是「项目特写」
 将一种技术的**基本用法**与**本项目的具体特征**融合为一体。
@@ -205,7 +207,7 @@ export class ProducerAgent {
 
       // 统计提交 (区分成功/失败)
       const submitCalls = (result.toolCalls || []).filter(
-        tc => (tc.tool || tc.name) === 'submit_candidate' || (tc.tool || tc.name) === 'submit_with_check'
+        tc => (tc.tool || tc.name) === 'submit_knowledge' || (tc.tool || tc.name) === 'submit_with_check'
       );
       const successCount = submitCalls.filter(tc => {
         const res = tc.result;
@@ -236,7 +238,7 @@ export class ProducerAgent {
           });
 
           const retrySubmits = (retryResult.toolCalls || []).filter(
-            tc => (tc.tool || tc.name) === 'submit_candidate' || (tc.tool || tc.name) === 'submit_with_check'
+            tc => (tc.tool || tc.name) === 'submit_knowledge' || (tc.tool || tc.name) === 'submit_with_check'
           );
           const retrySuccess = retrySubmits.filter(tc => {
             const res = tc.result;