@jiugeqqqqq/dsh-house2 0.2.0 → 0.2.2

package/CHANGELOG.md

@@ -3,6 +3,15 @@
 > 每次发版都必须在这里写一条：改了什么、为什么改、别人要不要跟着动。
 > 版本号跟 `package.json` 走，`npm i` 的人看这里就知道升了什么。
 
+## 0.2.2 — 2026-10-30
+
+- 压缩改成**按 token 触发**（不再挂在「任务完成」上）：house2 自己统计累计 token（口径同 dsh-session-rotate，监听 `assistant/message` 的 `usage`），到阈值 80% 就在注入里提醒「该准备交接/换会话」。阈值 `thresholdTokens`，默认 1 亿。
+- 不自己调压缩（DSH 没给插件公开的手动压缩接口），只按 token 提醒；工段只负责记任务。
+
+## 0.2.1 — 2026-10-30
+
+- `house_search` 加 `stage` 参数：先按工段名找到它挂的积木文件（`stage_done` 时写的「积木：xxx」），只在那一个文件里搜，比全目录翻更快更准。不传就还是全目录搜。
+
 ## 0.2.0 — 2026-10-30
 
 - 新功能：**工段（任务段）**。不再按轮数切，按「这件事做完了没有」切——`stage_start` 开一座、`stage_done` 收掉（写结果 + 积木文件名）。

package/lib/host/index.js

@@ -125,6 +125,56 @@ export function apply(ctx, config = {}) {
   stages.ensure();
   ctx.logger?.info('dsh-house2: 工段文件 = %s', stagesPath);
 
+  // 2026-10-30：压缩改成按 token 触发（不再挂在「任务完成」上）。
+  // 口径跟 dsh-session-rotate 一样：监听 session/event 里 assistant/message 的 data.usage。
+  // 到阈值就在注入里提醒「该交接/换会话」，不自己调压缩（DSH 没给插件公开的手动压缩接口）。
+  const tokenStatePath = path.join(dshHome, 'house2-token.json');
+  const tokenState = { sessionId: '', usedTokens: 0 };
+  try {
+    Object.assign(tokenState, JSON.parse(fs.readFileSync(tokenStatePath, 'utf8')));
+  } catch {}
+  let tokenSaveTimer = null;
+  const saveTokenState = () => {
+    if (tokenSaveTimer) return;
+    tokenSaveTimer = setTimeout(() => {
+      tokenSaveTimer = null;
+      try {
+        fs.writeFileSync(tokenStatePath, JSON.stringify(tokenState), 'utf8');
+      } catch {}
+    }, 1000);
+    tokenSaveTimer.unref?.();
+  };
+  const tokenThreshold = Math.max(0, Number(config.thresholdTokens) || 100000000);
+  try {
+    ctx.on('session/event', (session, event) => {
+      try {
+        const sid = String(session?.id ?? session?.sessionId ?? '');
+        if (sid && sid !== tokenState.sessionId) {
+          tokenState.sessionId = sid;
+          tokenState.usedTokens = 0;
+          saveTokenState();
+        }
+        if (event?.type !== 'assistant/message') return;
+        const u = event?.data?.usage;
+        if (!u) return;
+        const total = Number(u.totalTokens);
+        const add =
+          Number.isFinite(total) && total > 0
+            ? total
+            : (Number(u.inputTokens) || 0) +
+              (Number(u.outputTokens) || 0) +
+              (Number(u.cacheReadTokens) || 0) +
+              (Number(u.cacheWriteTokens) || 0);
+        if (add > 0) {
+          tokenState.usedTokens += add;
+          saveTokenState();
+        }
+      } catch {}
+    });
+  } catch (e) {
+    ctx.logger?.warn('dsh-house2: 挂 token 统计失败：' + (e && e.message ? e.message : e));
+  }
+
   // 版本自检：只查 npm 有没有新版，不自动安装。
   // 注意：本地开发目录的 package.json name 是 dsh-house2，而 npm 上是 scoped 包，所以写死 npmName。
   const updateChecker = createUpdateChecker({
@@ -528,11 +578,12 @@ export function apply(ctx, config = {}) {
     return out;
   }
 
-  function searchKnowledge(query, limit) {
+  function searchKnowledge(query, limit, onlyFiles) {
     const q = String(query ?? '').trim().toLowerCase();
     if (!q) return { ok: false, error: { code: 'invalid', message: '要给个关键词。' } };
     const max = Math.max(1, Math.min(30, Number(limit) || 8));
-    const files = walkKnowledge(knowledgeDir);
+    // 2026-10-30：给了 onlyFiles 就只在这些积木里搜（工段定向检索）。
+    const files = Array.isArray(onlyFiles) && onlyFiles.length ? onlyFiles : walkKnowledge(knowledgeDir);
     const hits = [];
     for (const f of files) {
       if (hits.length >= max) break;
@@ -557,6 +608,22 @@ export function apply(ctx, config = {}) {
     return { ok: true, data: { query: q, hits, scanned: files.length } };
   }
 
+  /** 按工段名找它挂的积木文件（stage_done 时写的「积木：xxx」）。 */
+  function stageBrickFiles(stageName) {
+    const name = String(stageName ?? '').trim();
+    if (!name) return null;
+    const { doing, done } = stages.parse();
+    const all = [...doing, ...done];
+    const hit = all.find((x) => x.name === name) || all.find((x) => x.name.includes(name) || name.includes(x.name));
+    if (!hit) return { error: `没有叫「${name}」的工段` };
+    const m = hit.line.match(/积木[:：]\s*([^\s|]+)/);
+    if (!m) return { error: `工段「${hit.name}」没有记积木文件名` };
+    const base = m[1].replace(/\.md$/i, '');
+    const file = path.join(knowledgeDir, `${base}.md`);
+    if (!fs.existsSync(file)) return { error: `积木文件不存在：${base}.md` };
+    return { files: [file], brick: base };
+  }
+
 
   // 2026-10-27：以前「房子清单能不能进模型上下文」要靠**另一个插件**
   // (dsh-session-rotate 的 assetsPath) 配一下才行 —— 等于装了这个插件的人也
@@ -569,7 +636,8 @@ export function apply(ctx, config = {}) {
     '写积木分两种：按能力写（`手机-接入与操作`、`画图-Agnes`，回答「怎么做」）、按任务写（`任务-调通一直发`，回答「那段对话干了什么」，一段任务一篇，按时间轴切，切完要覆盖全程、不重不漏）；' +
     '细节、命令、踩过的坑，用 house_write 写进积木（knowledge 目录，默认追加，别一个坑建一个文件）；' +
     '一件事真做完了（跑通、验证过），用 house_done 把它从「正在长出来的」搬到「已经完工的」——别让旧条目一直挂着；' +
-    '清单和最近对话里找不到的东西，用 house_search 去积木里按关键词翻，翻到再答，别猜。';
+    '清单和最近对话里找不到的东西，用 house_search 去积木里按关键词翻，翻到再答，别猜（先按工段定位，传 stage 参数只搜它挂的那一篇）；' +
+    '压缩和换会话不归这里管——那是按累计 token 阈值来的（session-rotate），工段只负责把任务记清楚。';
 
   function assetsAsPrompt() {
     let text = '';
@@ -578,7 +646,15 @@ export function apply(ctx, config = {}) {
     } catch {}
     const stageText = stages.renderPrompt();
     const base = text ? `【资产清单】\n${text}\n\n${HOUSE_GUIDE}` : HOUSE_GUIDE;
-    return stageText ? `${base}\n\n${stageText}` : base;
+    // 2026-10-30：压缩按 token 走，到 80% 就在这里提醒一句。
+    const used = tokenState.usedTokens;
+    let tokenLine = '';
+    if (used > 0 && tokenThreshold > 0) {
+      const pct = Math.round((used / tokenThreshold) * 100);
+      tokenLine = `【token】已用 ${(used / 10000).toFixed(0)} 万 / 阈值 ${(tokenThreshold / 10000).toFixed(0)} 万（${pct}%）`;
+      if (pct >= 80) tokenLine += ' —— 该准备交接、收尾换会话了（压缩按 token 触发，不归工段管）';
+    }
+    return [base, stageText, tokenLine].filter(Boolean).join('\n\n');
   }
 
   // 2026-10-30：改成直接 ctx.systemPrompt.section（跟 dsh-affection 一样）。
@@ -659,13 +735,20 @@ export function apply(ctx, config = {}) {
             properties: {
               query: { type: 'string', description: '关键词，比如 抖音、画图、报错信息里的词' },
               limit: { type: 'number', description: '最多返回几处，默认 8' },
+              stage: { type: 'string', description: '可选：先定位到这个工段，只搜它挂的积木文件（更快更准）' },
             },
             required: ['query'],
             additionalProperties: false,
           },
           output,
           async execute(args) {
-            const r = searchKnowledge(args?.query, args?.limit);
+            let onlyFiles = null;
+            if (args?.stage) {
+              const sb = stageBrickFiles(args.stage);
+              if (!sb || sb.error) return { ok: false, message: sb?.error || '找不到这个工段' };
+              onlyFiles = sb.files;
+            }
+            const r = searchKnowledge(args?.query, args?.limit, onlyFiles);
             if (!r.ok) return { ok: false, message: r.error?.message || '没搜成' };
             if (!r.data.hits.length) return { ok: true, message: `积木里没搜到「${r.data.query}」（翻过 ${r.data.scanned} 个文件）` };
             const body = r.data.hits.map((h) => `${h.file}:${h.line}  ${h.text}`).join('\n');

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@jiugeqqqqq/dsh-house2",
-  "version": "0.2.0",
+  "version": "0.2.2",
   "description": "房子 2 —— 资产/能力清单 + 工段，配侧边栏页面。0.1.0 先把安装跑通（保守兼容写法），功能后续迭代。",
   "type": "module",
   "license": "MIT",