@raolin2025/claude-code-node 2.8.0 → 2.8.2

package/README.md

@@ -111,11 +111,51 @@ printf '{"jsonrpc":"2.0","id":1,"method":"initialize"}\n' | cc-node --stdio
 | `/clear` | 清空当前对话 |
 | `/config KEY` | 查看配置（支持点号路径如 `tools.bash.timeout`） |
 | `/budget` | 查看 Token 预算使用情况 |
+| `/window [N]` | 查看/设置上下文窗口（如 `/window 128k`、`/window auto`） |
 | `/exit` `/quit` | 退出（Ctrl+C 也可以） |
 
 [![npm version](https://img.shields.io/npm/v/@raolin2025/claude-code-node.svg)](https://www.npmjs.com/package/@raolin2025/claude-code-node) [![GitHub](https://img.shields.io/badge/GitHub-bg1avd%2Fclaude--code--node-blue)](https://github.com/bg1avd/claude-code-node) [![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
 ---
 
+## 🧠 上下文窗口感知（自动压缩）
+
+cc-node 会自动感知当前所用模型的**上下文窗口长度**，并在接近上限时**自动压缩**对话历史，
+确保上下文**永不超出**模型窗口。
+
+### 窗口来源优先级
+
+| 优先级 | 来源 | 说明 |
+|--------|------|------|
+| 1 | **手动指定**（`/window N`） | 持久化到 config，每次启动优先采用 |
+| 2 | **API 探测** | 从 `GET /models` 响应提取（vLLM `context_length`、Ollama `model_info` 等） |
+| 3 | **内置模型表** | 内置 40+ 常见模型上下文表（DeepSeek/OpenAI/Qwen/GLM/Kimi 等） |
+| 4 | **安全兜底** | 64K（探测不到且表里没有时） |
+
+> 探测结果**不落盘**，每次启动重新探测；只有 `/window N` 手动指定才会持久化（方案 A）。
+> 探测不精确时，用 `/window N` 手动纠正即可。
+
+### 自动压缩机制
+
+- 当已用 token 达到窗口的 **80%** 时自动触发压缩（`autoCompact`）。
+- 压缩策略：保留最近 4 轮完整对话，早期历史压缩为摘要（用户意图、用到的工具、关键结果），
+  并将过长工具结果截断，目标压到窗口的 60%。
+- **发送前硬校验**：每次发请求前估算消息量，超窗先压缩再发，作为最终兜底保险。
+
+### `/window` 命令用法
+
+```
+/window              → 查看当前窗口 + 来源 + 用量
+/window 128k         → 手动指定 128K（持久化到 config）
+/window 64k          → 手动指定 64K
+/window 1m           → 手动指定 100 万
+/window auto         → 清除手动指定，回到自动探测
+/window reset        → 清除手动指定并立即重探测
+```
+
+> 切换模型（`/model`）后会自动重新探测窗口。`/budget` 也会显示当前窗口与 80% 触发阈值。
+
+---
+
 ## 🛠️ 内置工具（10 个）
 
 | 工具 | 说明 | 权限级别 | 安全检查 |
@@ -259,7 +299,7 @@ cc-node --api-base http://localhost:11434/v1
 {
   "model": "deepseek-chat",
   "maxTurns": 100,
-  "maxBudgetTokens": 1000000,
+  "maxBudgetTokens": 128000,
   "permissionMode": "ask",
   "tools": {
     "bash": { "timeout": 120 },
@@ -272,6 +312,10 @@ cc-node --api-base http://localhost:11434/v1
 }
 ```
 
+> **`maxBudgetTokens`**：手动指定的上下文窗口上限（token 数）。
+> 为 `0` 或未设置时，自动探测模型真实窗口；手动指定后优先于自动探测，
+> 等价于 `/window N` 的效果，并持久化保存。
+
 [![npm version](https://img.shields.io/npm/v/@raolin2025/claude-code-node.svg)](https://www.npmjs.com/package/@raolin2025/claude-code-node) [![GitHub](https://img.shields.io/badge/GitHub-bg1avd%2Fclaude--code--node-blue)](https://github.com/bg1avd/claude-code-node) [![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
 ---
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@raolin2025/claude-code-node",
-  "version": "2.8.0",
+  "version": "2.8.2",
   "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/core/cli.js

@@ -17,6 +17,13 @@ import { TokenBudget } from './token-budget.js'
 import { ChannelManager } from '../channel/index.js'
 import { CostTracker } from './cost-tracker.js'
 import { autoCompact } from './compact.js'
+import {
+  detectContextWindow,
+  parseWindowArg,
+  formatTokens,
+  windowSourceLabel,
+  WINDOW_SOURCE,
+} from './context-window.js'
 import { isLocalLlmServer } from '../utils/index.js'
 import { SOCK_DIR, SOCK_PATH, CC_NODE_PID } from './paths.js'
 import { renderHeadpiece } from './headpiece.js'
@@ -254,6 +261,7 @@ Commands:
   /stop          — Stop current AI work (when stuck/long)
   /config KEY    — Show config value
   /budget        — Show token budget
+  /window [N]    — Show/set context window (e.g. /window 128k, /window auto)
   /channel CMD   — Manage notification channels (list|send|test)
   /cost          — Show API cost report
   /compact       — Manually compact conversation context
@@ -292,6 +300,8 @@ const DETAILED_HELP = {
 
   budget:  "/budget\n  Show token budget usage for the current session.\n  Displays how many tokens have been used vs the limit.",
 
+  window:  "/window [N|auto|reset]\n  Show or set the LLM context window (max tokens allowed in conversation).\n  This is the hard ceiling for auto-compression — context will never exceed it.\n\n  Subcommands:\n    (no arg)   — Show current window + its source (manual/probe/table/fallback) + usage\n    /window 128k — Manually set window (persisted to config). Accepts K/M suffix:\n                  64k, 128k, 200k, 1m, or a raw number like 131072.\n    /window auto  — Clear manual override, return to auto-detection (next message re-detects).\n    /window reset — Clear manual override AND re-detect now.\n\n  Manual setting takes priority over auto-detection and survives restart.\n  Auto-detection itself is NOT persisted (re-runs at each startup).\n\n  Examples:\n    /window\n    /window 128k\n    /window 64k\n    /window auto\n    /window reset",
+
   channel: "/channel <list|send|test>\n  Manage notification channels.\n\n  Subcommands:\n    list       — List all configured notification channels\n    send <msg> — Send a message via all channels\n    test       — Send a test message to verify channels\n\n  Requires channel environment variables to be set at startup.",
 
   cost:    "/cost\n  Show API cost report.\n  Displays total tokens used and estimated cost in USD.\n  Supports pricing for: DeepSeek, OpenAI, Qwen, GLM, Kimi.",
@@ -481,7 +491,31 @@ const systemPrompt = cliArgs.systemPrompt || DEFAULT_SYSTEM_PROMPT
   }
 
   // M1 fix: tokenBudget 必须在 engineConfig 之前定义，否则 TDZ ReferenceError
-  const tokenBudget = new TokenBudget({ maxTokens: config.get('maxBudgetTokens') || 200_000 })
+  // 上下文窗口来源：手动指定 > API 探测 > 内置表 > 安全兜底
+  //   - 手动指定（config.maxBudgetTokens > 0）优先，且持久化
+  //   - 自动探测结果不落盘，每次启动重新探测（方案 A）
+  let windowSource = WINDOW_SOURCE.FALLBACK
+  const tokenBudget = new TokenBudget({
+    maxTokens: config.get('maxBudgetTokens') || 0,
+  })
+
+  /** 重新探测并应用上下文窗口（供启动 & /window reset & /model 切换后调用） */
+  async function reapplyWindow() {
+    const manual = config.get('maxBudgetTokens') || 0
+    const { window: win, source } = await detectContextWindow({
+      model,
+      apiBase,
+      apiKey,
+      manualWindow: manual,
+    })
+    tokenBudget.setWindow(win, source)
+    windowSource = source
+    return { window: win, source }
+  }
+
+  // 启动即探测一次（若已手动指定，则直接应用手动值）
+  await reapplyWindow()
+
   const costTracker = new CostTracker({ model })
 
   const engineConfig = new QueryEngineConfig({
@@ -684,10 +718,48 @@ const systemPrompt = cliArgs.systemPrompt || DEFAULT_SYSTEM_PROMPT
             } else {
               engine.config.model = rest.join(' ')
             }
+            model = engine.config.model
             console.log(`Model → ${engine.config.model}`)
+            // 切换模型后重新探测上下文窗口（仅当非手动指定时自动更新）
+            if ((config.get('maxBudgetTokens') || 0) <= 0) {
+              const { window: win, source } = await reapplyWindow()
+              console.log(`  ↳ Context window → ${formatTokens(win)} (${windowSourceLabel(source)})`)
+            }
           }
           else console.log(`Model: ${engine.config.model}`)
           break
+        case 'window': {
+          const arg = rest[0] ? rest.join(' ').trim() : ''
+          if (!arg) {
+            // 无参数：显示当前窗口
+            console.log(`Context window: ${formatTokens(tokenBudget.maxTokens)} (${windowSourceLabel(windowSource)})`)
+            console.log(`  Input used: ${tokenBudget.inputTokens.toLocaleString()} / ${tokenBudget.maxTokens.toLocaleString()} (${tokenBudget.usagePercent}%)`)
+            const manual = config.get('maxBudgetTokens') || 0
+            if (manual > 0) console.log(`  Manual override: ${formatTokens(manual)} (persisted in config)`)
+            break
+          }
+          const lower = arg.toLowerCase()
+          if (lower === 'auto' || lower === 'reset') {
+            // 清除手动指定并重新探测
+            config.set('maxBudgetTokens', 0)
+            try { await config.saveToUser() } catch {}
+            const { window: win, source } = await reapplyWindow()
+            console.log(`Cleared manual override. Context window → ${formatTokens(win)} (${windowSourceLabel(source)})`)
+            break
+          }
+          const win = parseWindowArg(arg)
+          if (!win) {
+            console.log(`❌ Invalid window: "${arg}". Use a number or K/M suffix, e.g. /window 128k, /window 64k, /window 1m`)
+            break
+          }
+          // 手动指定：立即生效 + 持久化
+          tokenBudget.setWindow(win, WINDOW_SOURCE.MANUAL)
+          windowSource = WINDOW_SOURCE.MANUAL
+          config.set('maxBudgetTokens', win)
+          try { await config.saveToUser() } catch (e) { console.log(`⚠️  Failed to persist to config: ${e.message}`) }
+          console.log(`Context window → ${formatTokens(win)} (${windowSourceLabel(WINDOW_SOURCE.MANUAL)}, persisted)`)
+          break
+        }
         case 'models': {
           const apiBase = engine.config.apiBase
           const apiKey = engine.config.apiKey
@@ -818,7 +890,11 @@ const systemPrompt = cliArgs.systemPrompt || DEFAULT_SYSTEM_PROMPT
             console.log(JSON.stringify(config.toJSON(), null, 2))
           }
           break
-        case 'budget': console.log(tokenBudget.format()); break
+        case 'budget': {
+          console.log(tokenBudget.format())
+          console.log(`Context window: ${formatTokens(tokenBudget.maxTokens)} (${windowSourceLabel(windowSource)}) | trigger: ${Math.round((tokenBudget.maxTokens * 0.8)).toLocaleString()} (80%)`)
+          break
+        }
         case 'channel': {
           const subCmd = rest.join(' ')
           if (subCmd === 'list' || subCmd === '') {

package/src/core/context-window.js

@@ -0,0 +1,262 @@
+/**
+ * 上下文窗口感知 (Context Window) — 自动探测 + 手动指定
+ *
+ * 解决的问题：TokenBudget.maxTokens 之前是一个"静态配置"（默认 100 万，
+ * CLI 兜底 20 万），并不知道所用模型真实上下文窗口。若预算 > 模型真实窗口，
+ * 自动压缩永远不会触发，上下文一路涨到模型报错。
+ *
+ * 本模块提供三层"窗口来源"，优先级：
+ *   1. manual  — 用户通过 /window 手动指定（持久化到 config）
+ *   2. probe   — 从 API /models 探测到的窗口（本次进程生效，不落盘）
+ *   3. table   — 内置常见模型上下文表（无法探测时的静态兜底）
+ *   4. fallback— 安全默认值（如 64K）
+ *
+ * 探测结果默认不落盘，手动指定才落盘（方案 A）。
+ */
+
+// ---- 内置常见模型上下文表（token）----
+// 缺失的厂商/模型会回落到 fallback；不精确属预期，可用 /window 手动纠正。
+export const MODEL_CONTEXT_TABLE = {
+  // DeepSeek
+  'deepseek-chat': 128_000,
+  'deepseek-reasoner': 128_000,
+  'deepseek-coder': 128_000,
+  'deepseek-v3': 128_000,
+  'deepseek-v2': 128_000,
+  'deepseek-v2.5': 128_000,
+  'deepseek-v4': 128_000,
+  // OpenAI
+  'gpt-4o': 128_000,
+  'gpt-4o-mini': 128_000,
+  'gpt-4-turbo': 128_000,
+  'gpt-4': 32_768,
+  'gpt-3.5-turbo': 16_384,
+  'o1': 200_000,
+  'o1-mini': 128_000,
+  'o3-mini': 200_000,
+  // Anthropic (经兼容网关时)
+  'claude-3-5-sonnet': 200_000,
+  'claude-3-5-haiku': 200_000,
+  'claude-sonnet-4': 200_000,
+  'claude-haiku-4': 200_000,
+  // Qwen (通义千问)
+  'qwen-max': 32_768,
+  'qwen-plus': 131_072,
+  'qwen-turbo': 1_000_000,
+  'qwen2.5-72b-instruct': 131_072,
+  'qwen2.5-32b-instruct': 131_072,
+  'qwen2.5-7b-instruct': 131_072,
+  'qwen2.5-14b-instruct': 131_072,
+  'qwen2.5-coder-32b-instruct': 131_072,
+  // GLM (智谱)
+  'glm-4': 128_000,
+  'glm-4-plus': 128_000,
+  'glm-4-air': 128_000,
+  'glm-4-flash': 128_000,
+  'glm-4v': 8_192,
+  'glm-4.5': 128_000,
+  'glm-4.5-air': 128_000,
+  // Moonshot (Kimi)
+  'moonshot-v1-8k': 8_192,
+  'moonshot-v1-32k': 32_768,
+  'moonshot-v1-128k': 131_072,
+  'moonshot-v1-auto': 131_072,
+  'kimi-k2': 128_000,
+  // 本地/开源 (Ollama / vLLM 常见)
+  'llama3.1': 131_072,
+  'llama3': 8_192,
+  'llama2': 4_096,
+  'mistral': 32_768,
+  'mixtral': 32_768,
+  'gemma2': 8_192,
+  'codellama': 16_384,
+  'qwen2.5': 131_072,
+  'yi-34b': 4_096,
+}
+
+// 安全兜底默认值 — 探测不到、表里也没有时使用
+export const FALLBACK_CONTEXT_WINDOW = 64_000
+
+// ---- 窗口来源标签 ----
+export const WINDOW_SOURCE = {
+  MANUAL: 'manual',
+  PROBE: 'probe',
+  TABLE: 'table',
+  FALLBACK: 'fallback',
+}
+
+const SOURCE_LABEL = {
+  [WINDOW_SOURCE.MANUAL]: '手动指定',
+  [WINDOW_SOURCE.PROBE]: 'API 探测',
+  [WINDOW_SOURCE.TABLE]: '内置模型表',
+  [WINDOW_SOURCE.FALLBACK]: '安全兜底',
+}
+
+/**
+ * 从 /models 响应中尝试提取单个模型的上下文窗口
+ *
+ * 不同服务字段名不一，尽量覆盖常见命名：
+ *   - OpenAI 兼容 vLLM: context_length / max_model_len / max_context_length
+ *   - Ollama: (来自 /api/show 的 model_info.llama.context_length)
+ *
+ * @param {object} modelObj — /models 返回数组中的单个模型对象
+ * @returns {number|null} 窗口 token 数；未知返回 null
+ */
+export function extractContextFromModelObj(modelObj) {
+  if (!modelObj || typeof modelObj !== 'object') return null
+  const keys = [
+    'context_length',
+    'context_window',
+    'max_context_length',
+    'max_model_len',
+    'max_sequence_length',
+    'model_max_length',
+    'n_ctx',
+    'contextWindow',
+    'contextSize',
+    'maxContext',
+    'maxTokens',
+    'context_tokens',
+  ]
+  for (const k of keys) {
+    const v = modelObj[k]
+    if (typeof v === 'number' && Number.isFinite(v) && v > 0) return v
+  }
+  // 某些服务把窗口放在 meta / details 子对象
+  for (const sub of ['meta', 'details', 'model_info', 'capabilities']) {
+    const nested = modelObj[sub]
+    if (nested && typeof nested === 'object') {
+      const found = extractContextFromModelObj(nested)
+      if (found) return found
+    }
+  }
+  // Ollama model_info 专用字段：llama.context_length
+  const modelInfo = modelObj.model_info
+  if (modelInfo && typeof modelInfo === 'object') {
+    const ctxKey = Object.keys(modelInfo).find(k => k.endsWith('.context_length'))
+    if (ctxKey && typeof modelInfo[ctxKey] === 'number' && modelInfo[ctxKey] > 0) {
+      return modelInfo[ctxKey]
+    }
+  }
+  return null
+}
+
+/**
+ * 探测当前模型的上下文窗口
+ *
+ * 顺序：
+ *   1. 若已手动指定（manualWindow > 0）→ 直接返回手动值
+ *   2. 尝试从 GET {apiBase}/models 的响应中找匹配模型并提取窗口
+ *   3. 查内置表
+ *   4. 兜底 fallback
+ *
+ * @param {object} opts
+ * @param {string} opts.model — 模型名
+ * @param {string} opts.apiBase — API Base URL
+ * @param {string} [opts.apiKey]
+ * @param {number} [opts.manualWindow] — 用户手动指定的窗口（0 表示未指定）
+ * @param {number} [opts.fetchTimeoutMs] — 探测超时（默认 3000ms）
+ * @returns {Promise<{window:number, source:string}>}
+ */
+export async function detectContextWindow({ model, apiBase, apiKey, manualWindow = 0, fetchTimeoutMs = 3000 }) {
+  // 1. 手动指定最高优先级
+  if (manualWindow > 0) {
+    return { window: manualWindow, source: WINDOW_SOURCE.MANUAL }
+  }
+
+  // 2. 探测 API
+  if (apiBase) {
+    try {
+      const controller = new AbortController()
+      const timer = setTimeout(() => controller.abort(), fetchTimeoutMs)
+      const url = apiBase.replace(/\/+$/, '') + '/models'
+      const res = await fetch(url, {
+        headers: {
+          'Content-Type': 'application/json',
+          ...(apiKey ? { Authorization: `Bearer ${apiKey}` } : {}),
+        },
+        signal: controller.signal,
+      })
+      clearTimeout(timer)
+      if (res.ok) {
+        const data = await res.json()
+        const models = data.data || []
+        // 优先精确匹配当前 model
+        let target = models.find(m => (m.id || '') === model)
+        // 其次尝试包含匹配（如 model 是 'qwen2.5' 而列表是 'qwen2.5:14b'）
+        if (!target) {
+          target = models.find(m => (m.id || '').startsWith(model) || model.startsWith(m.id || ''))
+        }
+        if (target) {
+          const win = extractContextFromModelObj(target)
+          if (win) return { window: win, source: WINDOW_SOURCE.PROBE }
+        }
+      }
+    } catch {
+      // 探测失败（超时/网络/无权限）→ 继续走表/兜底
+    }
+  }
+
+  // 3. 内置表
+  const normalized = (model || '').toLowerCase()
+  if (normalized && MODEL_CONTEXT_TABLE[normalized]) {
+    return { window: MODEL_CONTEXT_TABLE[normalized], source: WINDOW_SOURCE.TABLE }
+  }
+  // 表内做前缀匹配（如 'deepseek-chat:latest'、'qwen2.5-coder:7b'）
+  for (const key of Object.keys(MODEL_CONTEXT_TABLE)) {
+    if (normalized.startsWith(key) || key.startsWith(normalized)) {
+      return { window: MODEL_CONTEXT_TABLE[key], source: WINDOW_SOURCE.TABLE }
+    }
+  }
+
+  // 4. 兜底
+  return { window: FALLBACK_CONTEXT_WINDOW, source: WINDOW_SOURCE.FALLBACK }
+}
+
+/**
+ * 解析 /window 命令的参数为窗口 token 数
+ *
+ * 支持：
+ *   "131072"  → 131072
+ *   "64k"     → 65536
+ *   "128k"    → 131072
+ *   "200k"    → 204800
+ *   "1m"      → 1_000_000
+ *
+ * @param {string} str
+ * @returns {number|null} token 数；非法返回 null
+ */
+export function parseWindowArg(str) {
+  if (typeof str !== 'string') return null
+  const s = str.trim().toLowerCase().replace(/,/g, '')
+  if (!s) return null
+  const m = s.match(/^(\d+)\s*(k|m|kb|mb)?$/)
+  if (!m) return null
+  const num = parseInt(m[1], 10)
+  const unit = m[2]
+  if (!Number.isFinite(num) || num <= 0) return null
+  if (unit === 'k' || unit === 'kb') return num * 1000
+  if (unit === 'm' || unit === 'mb') return num * 1_000_000
+  return num
+}
+
+/**
+ * 将 token 数格式化为人类可读（十进制，与 token 常见记法一致：128K = 128,000）
+ * @param {number} tokens
+ * @returns {string} 如 "128.0K"、"1.00M"
+ */
+export function formatTokens(tokens) {
+  if (!Number.isFinite(tokens) || tokens <= 0) return '?'
+  if (tokens >= 1_000_000) return (tokens / 1_000_000).toFixed(2) + 'M'
+  if (tokens >= 1000) return (tokens / 1000).toFixed(1) + 'K'
+  return String(tokens)
+}
+
+/**
+ * 获取窗口来源的中文标签
+ * @param {string} source
+ * @returns {string}
+ */
+export function windowSourceLabel(source) {
+  return SOURCE_LABEL[source] || source || '未知'
+}

package/src/core/headpiece.js

@@ -4,10 +4,13 @@
 //  只负责「头标」的绘制与配色，与 CLI 框架解耦。
 //  以后要替换/升级头标，只需修改本文件并保持接口不变即可。
 //
-//  头标内容: CC-NODE 艺术字
-//    CC   : 两个 C, 10列宽 x 4行高 (扁平, 蓝渐变)
-//    NODE : 四个字母, 5列宽 x 4行高 (绿渐变)
-//    上下严格对齐 (20列 x 8行)
+//  头标内容: CC-NODE 机械臂机器人艺术字（v2，取自 log/headpiece.js）
+//    CC = 一对机械眼睛（带 ◉◉ 瞳孔，蓝渐变 5 行）
+//    N  = 左臂，向左侧伸展 ──┼
+//    O  = 圆形身体/躯干（白色）
+//    D  = 右臂，向右侧伸展 ┼──
+//    E  = 手持工具箱/锤子（黄色 ╔═══╗）
+//    上下总宽 20 字符，总高 10 行，严格对齐
 //
 //  接口:
 //    renderHeadpiece({ colWidth = 24 } = {})
@@ -17,39 +20,38 @@
 //         width   头标原始可见宽度
 // ============================================================
 
-// ---- 头标原始数据（20列 x 8行）----
-// CC 部分（蓝渐变, 4行）
+// ---- 头标原始数据（20列 x 10行）----
+// CC 部分（机械眼睛，蓝渐变，5 行；每行统一右填充至 20 列保证对齐）
 const CC = [
-  '╔════════╗╔════════╗',
-  '██║       ██║       ',
-  '██║       ██║       ',
-  '╚════════╝╚════════╝',
+  '\x1b[38;5;81m    ╭────╮╭────╮\x1b[0m',
+  '\x1b[38;5;75m    ╱ ◉◉ ╲╱ ◉◉ ╲\x1b[0m',
+  '\x1b[38;5;69m   │ ◉◉  ││ ◉◉  │\x1b[0m',
+  '\x1b[38;5;63m   │      ││      │\x1b[0m',
+  '\x1b[38;5;57m    ╰────╯╰────╯\x1b[0m',
 ]
 
-// NODE 部分（绿渐变, 4行）
+// NODE 部分（左臂绿 + 身体白 + 右臂绿 + 工具箱黄，5 行）
 const NODE = [
-  '╔╗╔═╗╔═══╗╔═══╗╔═══╗',
-  '║║║ ║║   ║║   ║║   ║',
-  '║╚╝ ║║   ║║   ║╠═══╣',
-  '╚═══╝╚═══╝╚═══╝╚═══╝',
+  '\x1b[38;5;82m╱───┐\x1b[0m\x1b[38;5;15m┌───┐\x1b[0m\x1b[38;5;82m┌───╲\x1b[0m\x1b[38;5;220m╔═══╗\x1b[0m',
+  '\x1b[38;5;118m│N  │\x1b[0m\x1b[38;5;15m│   │\x1b[0m\x1b[38;5;118m│D  │\x1b[0m\x1b[38;5;220m║   ║\x1b[0m',
+  '\x1b[38;5;154m│   │\x1b[0m\x1b[38;5;15m│   │\x1b[0m\x1b[38;5;154m│   │\x1b[0m\x1b[38;5;220m╠═══╣\x1b[0m',
+  '\x1b[38;5;190m──┼ │\x1b[0m\x1b[38;5;15m│   │\x1b[0m\x1b[38;5;190m│  ┼─\x1b[0m\x1b[38;5;220m║   ║\x1b[0m',
+  '\x1b[38;5;226m╲───┘\x1b[0m\x1b[38;5;15m└───┘\x1b[0m\x1b[38;5;226m└───╱\x1b[0m\x1b[38;5;220m╚═══╝\x1b[0m',
 ]
 
-// ---- 配色（渐变）----
-// CC 蓝色渐变（深 → 浅）
-const BLUE = [81, 75, 69, 63]
-// NODE 绿色渐变（深 → 浅）
-const GREEN = [82, 118, 154, 190]
+// 合并全部行
+const RAW_LINES = [...CC, ...NODE]
 
-// 给每一行按对应色号上色
-function colorize(lines, colors) {
-  return lines.map((line, i) => `\x1b[38;5;${colors[i]}m${line}\x1b[0m`)
-}
-
-// 去掉 ANSI 转义码，返回真实可见长度
+// ---- 配色工具 ----
 function stripAnsi(s) {
   return s.replace(/\x1b\[[0-9;]*m/g, '')
 }
 
+/** 去掉尾部 ANSI reset 码，供拼接填充空格时保持颜色连续 */
+function trimTrailingReset(s) {
+  return s.replace(/\x1b\[0m\s*$/, '')
+}
+
 /**
  * 渲染头标。
  * @param {Object} [opts]
@@ -57,23 +59,33 @@ function stripAnsi(s) {
  * @returns {{ lines: string[], height: number, width: number }}
  */
 export function renderHeadpiece({ colWidth = 24 } = {}) {
-  // 1. 原始可见宽度
-  const width = stripAnsi(CC[0]).length // 20
+  // 1. 原始可见宽度（取最宽一行）
+  const width = Math.max(...RAW_LINES.map(stripAnsi).map(s => s.length))
+  const height = RAW_LINES.length // 10
 
-  // 2. 合并 CC + NODE，分别上色
-  const rawLines = [...colorize(CC, BLUE), ...colorize(NODE, GREEN)]
-  const height = rawLines.length // 8
+  // 2. 逐行补齐到统一宽度 width（右填充空格，保证右边缘对齐；空格放 ANSI reset 之前）
+  const padded = RAW_LINES.map((line) => {
+    const plain = stripAnsi(line)
+    const missing = width - plain.length
+    if (missing <= 0) return line
+    // 保留前导 ANSI 与尾部 reset：在 reset 前补空格
+    const leadingAnsi = line.match(/^(\x1b\[[0-9;]*m)+/)?.[0] || ''
+    const body = line.slice(leadingAnsi.length)
+    const trailingAnsi = body.match(/(\x1b\[0m)+$/)?.[0] || ''
+    const inner = body.slice(0, body.length - trailingAnsi.length)
+    return leadingAnsi + inner + ' '.repeat(missing) + trailingAnsi
+  })
 
-  // 3. 在 colWidth 内居中填充（若 colWidth < width 则不做填充，原样返回）
+  // 3. 在 colWidth 内居中填充（若 colWidth <= width 则不做填充，原样返回）
   if (colWidth <= width) {
-    return { lines: rawLines, height, width }
+    return { lines: padded, height, width }
   }
 
   const totalPad = colWidth - width
   const leftPad = Math.floor(totalPad / 2)
   const rightPad = totalPad - leftPad
 
-  const lines = rawLines.map((line) => {
+  const lines = padded.map((line) => {
     const leadingAnsi = line.match(/^(\x1b\[[0-9;]*m)+/)?.[0] || ''
     const trailingAnsi = line.match(/(\x1b\[0m)+$/)?.[0] || '\x1b[0m'
     const plain = stripAnsi(line)

package/src/core/index.js

@@ -5,6 +5,14 @@ export { UserMessage, AssistantMessage, ToolCall } from "../types/index.js"
  */
 export { QueryEngine, QueryEngineConfig } from './query-engine.js'
 export { TokenBudget, estimateTokens } from './token-budget.js'
+export {
+  detectContextWindow,
+  parseWindowArg,
+  formatTokens,
+  windowSourceLabel,
+  WINDOW_SOURCE,
+  MODEL_CONTEXT_TABLE,
+} from './context-window.js'
 export { SessionManager } from './session.js'
 export { Config } from './config.js'
 export { parseStream, parseNonStreamResponse } from './streaming.js'

package/src/core/query-engine.js

@@ -111,6 +111,20 @@ export class QueryEngine {
     let finalResponse = ''
 
     for (let turn = 0; turn < this.config.maxTurns; turn++) {
+      // 发送前硬校验：估算即将发送的消息是否超出窗口，超限则先压缩（最终兜底，防止溢出）
+      if (this.tokenBudget) {
+        const est = this.tokenBudget.estimateMessages(this.state.messages)
+        if (est > this.tokenBudget.maxTokens - this.tokenBudget.reservedForOutput) {
+          const { compacted, messages } = autoCompact(this.state.messages, this.tokenBudget, {
+            maxTokens: Math.floor(this.tokenBudget.maxTokens * 0.6),
+          })
+          if (compacted) {
+            this.state.messages = messages
+            if (this.config.verbose) console.error('[compact] Pre-send hard check: compressed to stay within window')
+          }
+        }
+      }
+
       const requestMessages = this._buildRequest(this.state.messages)
       const response = await this._callLLM(requestMessages, this.state.messages)
 

package/src/core/token-budget.js

@@ -24,6 +24,20 @@ export class TokenBudget {
     this.used = 0
     this.inputTokens = 0
     this.outputTokens = 0
+    // 当前窗口来源：'manual' | 'probe' | 'table' | 'fallback'（见 context-window.js）
+    this.windowSource = options.windowSource || 'fallback'
+  }
+
+  /**
+   * 更新上下文窗口上限（运行时生效，立即反映到 usagePercent）
+   * @param {number} maxTokens
+   * @param {string} [source] — 窗口来源标签
+   */
+  setWindow(maxTokens, source) {
+    if (Number.isFinite(maxTokens) && maxTokens > 0) {
+      this.maxTokens = maxTokens
+    }
+    if (source) this.windowSource = source
   }
 
   /** 可用于上下文的最大 token 数 */