@raolin2025/claude-code-node 2.8.17 → 2.8.19

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@raolin2025/claude-code-node",
-  "version": "2.8.17",
+  "version": "2.8.19",
   "description": "Node.js AI Code Agent CLI - Zero dependencies, pure JavaScript, security hardened, multi-channel notifications, Telegram & QQ Bot remote programming, rich media upload, multi-account management",
   "type": "module",
   "main": "src/core/index.js",

package/src/__tests__/compact-window.test.js

@@ -284,3 +284,39 @@ test('工具结果截断：非 tool 消息与空内容不受影响', () => {
   assert.equal(r[0].content.length, 10000, 'user 消息不应被截断')
   assert.equal(r[1].content, '')
 })
+
+test('折叠/裁剪：多次折叠后所有 system 仍在开头（防 llama.cpp 500）', () => {
+  // 模拟：一次折叠后 body 里有摘要 system 残留的场景
+  const msgs = [
+    { role: 'system', content: 'ROOT SYS' },
+    { role: 'system', content: '[Context Summary — 旧的折叠摘要]' }, // 已存在第二条 system
+    { role: 'user', content: '任务1' },
+    { role: 'assistant', content: '结果1' },
+    { role: 'user', content: '任务2' },
+    { role: 'assistant', content: '结果2' },
+  ]
+  // foldHistoryByCount 折叠后，所有 system 都应在最前，body 里无 system
+  const r = foldHistoryByCount(msgs, { maxMessages: 5, keepRecentTurns: 1 })
+  const roles = r.messages.map(m => m.role)
+  // 找到第一个非 system 的索引，之后不应再有 system
+  const firstBody = roles.findIndex(x => x !== 'system')
+  const afterBody = roles.slice(firstBody + 1)
+  assert.ok(!afterBody.includes('system'), `折叠后 body 中不应有 system（实际 roles=${roles.join(',')}）`)
+})
+
+test('trimToWindow：所有 system 均在开头，body 中无 system', () => {
+  const msgs = [
+    { role: 'system', content: 'ROOT SYS' },
+    { role: 'system', content: '[Context Summary — 旧摘要]' },
+    { role: 'user', content: 'a'.repeat(50) },
+    { role: 'assistant', content: 'b'.repeat(50) },
+    { role: 'user', content: 'c'.repeat(50) },
+    { role: 'assistant', content: 'd'.repeat(50) },
+  ]
+  const tb = makeBudget(200, 0)
+  const r = trimToWindow(msgs, { tokenBudget: tb, maxTokens: tb.maxTokens })
+  const roles = r.messages.map(m => m.role)
+  const firstBody = roles.findIndex(x => x !== 'system')
+  const afterBody = roles.slice(firstBody + 1)
+  assert.ok(!afterBody.includes('system'), `trim 后 body 中不应有 system（实际 roles=${roles.join(',')}）`)
+})

package/src/core/cli.js

@@ -333,6 +333,7 @@ function parseArgs(argv) {
     maxMessages: 0,
     smallModel: false,
     maxOutputTokens: 0,
+    smallModelMaxTurns: 0,
   }
 
   let i = 2
@@ -351,6 +352,7 @@ function parseArgs(argv) {
       case '--max-messages': args.maxMessages = parseInt(argv[++i], 10); break
       case '--small-model': args.smallModel = true; break
       case '--max-output-tokens': args.maxOutputTokens = parseInt(argv[++i], 10); break
+      case '--small-model-max-turns': args.smallModelMaxTurns = parseInt(argv[++i], 10); break
       case '--stdio': args.stdio = true; break
       case '--with-notify': args.withNotify = true; break
       case '--version':
@@ -374,6 +376,7 @@ Options:
   --max-messages N          Fold history when message count exceeds N (default: 0 = off)
   --small-model             Enable small-model adaptation (tool-call enforcement, filler retry, intent guidance)
   --max-output-tokens N     Override max single-response output tokens (default: computed from window size)
+  --small-model-max-turns N Small-model tool-loop cap (default: 8, prevents infinite loops)
   --with-notify             Start built-in channel listener (Telegram)
                             (replaces cc-notify daemon — no external script needed)
   -h, --help                Show this help
@@ -540,6 +543,8 @@ const systemPrompt = cliArgs.systemPrompt || DEFAULT_SYSTEM_PROMPT
     smallModel: cliArgs.smallModel || config.get('smallModel') || false,
     // 单次输出上限覆盖（默认根据窗口动态计算）
     maxOutputTokens: cliArgs.maxOutputTokens || config.get('maxOutputTokens') || 0,
+    // 小模型工具循环轮数上限（默认 8，防无限循环）
+    smallModelMaxTurns: cliArgs.smallModelMaxTurns || config.get('smallModelMaxTurns') || 0,
   })
   const engine = new QueryEngine(engineConfig)
 

package/src/core/compact.js

@@ -197,11 +197,14 @@ export function foldHistoryByCount(messages, options = {}) {
     return { folded: false, messages, removed: 0, summary: null }
   }
 
-  // 分离 system 提示（首条 system 永不折叠）与普通消息
+  // 分离 system 提示与普通消息
+  // 注意：把所有 system 都收集到 systemMsgs（不只是第一条）——多次折叠后 body 里
+  // 可能残留旧的摘要 system，若不全部隔离，折叠结果会把 system 挤到对话中间，
+  // 触发 llama.cpp "System message must be at the beginning" (500)。
   const systemMsgs = []
   const body = []
   for (const m of messages) {
-    if (m.role === 'system' && systemMsgs.length === 0) {
+    if (m.role === 'system') {
       systemMsgs.push(m)
     } else {
       body.push(m)
@@ -400,11 +403,13 @@ export function trimToWindow(messages, options = {}) {
     return { trimmed: false, messages, removed: 0, summary: null }
   }
 
-  // 分离 system 提示（首条 system 永不裁剪）与普通消息
+  // 分离 system 提示与普通消息
+  // 注意：收集【所有】 system（不只第一条），避免多次裁剪后旧的摘要 system
+  // 残留在 body 里、被挤到对话中间，触发 llama.cpp "System message must be at the beginning" (500)。
   const systemMsgs = []
   const body = []
   for (const m of messages) {
-    if (m.role === 'system' && systemMsgs.length === 0) {
+    if (m.role === 'system') {
       systemMsgs.push(m)
     } else {
       body.push(m)

package/src/core/query-engine.js

@@ -53,6 +53,9 @@ export class QueryEngineConfig {
     //   - 工具数量精简 + 意图引导
     // 默认关闭，通过 config.smallModel=true 或 --small-model 开启
     this.smallModel = options.smallModel || false
+    // 小模型模式下的工具循环轮数上限（默认 8）：
+    // 小模型常陷入"写一个又写一个"的无限工具循环，需更小的上限 + 收尾引导
+    this.smallModelMaxTurns = options.smallModelMaxTurns || 8
     // 单次输出上限动态计算：
     //   outputRatio — 输出占窗口比例（默认 1/16），窗口越大输出越大
     //   maxOutputTokens — 输出下限（默认 4096），也可作为硬性覆盖
@@ -228,8 +231,14 @@ export class QueryEngine {
     // 敷衍重试次数（层 A）：模型没调工具只回空话时，追加强引导重试
     let fillerRetries = 0
     const MAX_FILLER_RETRIES = 1
-
-    for (let turn = 0; turn < this.config.maxTurns; turn++) {
+    // 工具循环轮数上限：
+    //   - 小模型模式：用更小的上限（默认 8），避免模型陷入"写一个又写一个"的无限循环
+    //   - 普通模式：用 config.maxTurns
+    const toolLoopLimit = smallModel
+      ? (this.config.smallModelMaxTurns || 8)
+      : this.config.maxTurns
+
+    for (let turn = 0; turn < toolLoopLimit; turn++) {
       // 发送前硬校验：工具结果可能已使上下文超窗，确保 ≤ 窗口（摘要优先 + 滑动窗口裁剪兜底）
       this._ensureFitWindow()
 
@@ -259,6 +268,22 @@ export class QueryEngine {
         intentGuided = true
       }
 
+      // 小模型模式：接近工具循环上限时，注入"总结收尾"引导，防止模型无限循环
+      if (smallModel && turn === toolLoopLimit - 1) {
+        const stopGuidance = `[系统提示] 你已经完成了大部分工作。现在请【停止调用新工具】，把已经完成的内容整理成一段总结回复给用户。如果确实还有关键步骤未完成，只再调用一次工具完成它，然后立即总结。不要再继续无休止地调用工具。`
+        const sysIdx = this.state.messages.findIndex(m => m.role === 'system')
+        if (sysIdx !== -1) {
+          const sysMsg = this.state.messages[sysIdx]
+          this.state.messages[sysIdx] = {
+            ...sysMsg,
+            content: (typeof sysMsg.content === 'string' ? sysMsg.content : '') + '\n\n' + stopGuidance,
+          }
+        } else {
+          this.state.messages.push({ role: 'user', content: stopGuidance })
+        }
+        if (this.config.verbose) console.error(`[small-model] 已到工具循环第 ${turn + 1} 轮（上限 ${toolLoopLimit}），注入收尾引导`)
+      }
+
       const requestMessages = this._buildRequest(this.state.messages)
       const response = await this._callLLM(requestMessages, this.state.messages)
 
@@ -306,8 +331,9 @@ export class QueryEngine {
       }
     }
 
-    if (!finalResponse && this.state.turnCount >= this.config.maxTurns) {
-      finalResponse = `[达到最大回合数限制 (${this.config.maxTurns})，停止响应]`
+    // 工具循环达到上限但仍未收尾（模型一直调用工具）→ 强制收尾
+    if (!finalResponse) {
+      finalResponse = `[已达到工具循环上限 (${toolLoopLimit})，已停止进一步调用工具。请查看上方工具执行结果，确认任务完成情况。]`
     }
 
     return {