@a9i5k4/dsh-auto-memory 2.4.0 → 2.4.1

package/lib/index.js

@@ -1947,6 +1947,11 @@ class MemoryEngine {
       const armRatio = hard ? Math.max(ratio, threshold) : ratio
       rt.waterLevel = armRatio
       rt.waterLevelTokens = estTokens
+      // 2026-09-10:pre-step 的 arm 走的是 runtime 字段(见 checkWaterLevelAtStep → armAutoContinue),
+      // 这里只写了 ratio/tokens、漏写 window/source,于是确认卡显示「512,311 / 0 token」(实测)。
+      // 与 state 同源同值,一处漏写两处都缺 —— 必须成对维护。
+      rt.waterLevelWindow = effectiveWin
+      rt.waterLevelSource = winSource
       this.state.waterLevelRatio = armRatio
       this.state.waterLevelTokens = estTokens
       this.state.waterLevelWindow = effectiveWin
@@ -1956,6 +1961,14 @@ class MemoryEngine {
       this.state.waterLevelAt = Date.now()
       this.state.waterLevelMeasuredTokens = estTokens
       this.state.waterLevelReserve = reserve
+      // 双口径(2026-09-10):官方小圈的分母是**声明窗口**(win),我们的分母是**可用额度**(effectiveWin),
+      // 两者在本机差 384,000(预留输出)。用户看到「才 50% 就接续」正是因为只看到小圈那个数 ——
+      // 把 ring 一并上报,界面就能同时显示「占可用额度 x% · 官方小圈 y%」,不再各说各话。
+      this.state.waterLevelRing = (win > 0 && Number.isFinite(estTokens)) ? (estTokens / win) : 0
+      // provider 自报的硬上限只有撞过墙(400 文本)才拿得到;拿得到就把「真实可写上限」也算出来,
+      // 让界面能显示「距硬墙还剩多少 token」(本机 = 1,048,576 − 384,000 = 664,576)。
+      const hardWin = Number(sig.overflow && sig.overflow.windowTokens) || 0
+      this.state.waterLevelWall = hardWin > 0 ? Math.max(0, hardWin - reserve) : 0
       this.state.waterLevelHardTrigger = hard ? (compacted ? 'compaction' : 'overflow') : ''
       // 会话级记录:切会话时 /handoff-state?sessionId= 直接取这份,不再显示别的会话的数
       this.rememberWaterRecord(agent && agent.session && agent.session.id ? String(agent.session.id) : '', {
@@ -2027,6 +2040,54 @@ class MemoryEngine {
     } catch (e) { diag('checkWaterLevelAtStep error: ' + (e && e.message)) }
   }
 
+  /** 已接续闩锁(2026-09-10):同一个会话**只允许被接续一次**。
+   *  旧实现只有 30 分钟冷却:冷却一过,用户切回那个旧会话、水位仍然高 → 再次 arm → 再次建会话,
+   *  表现为「切回旧窗口,它还想接续」(实测用户报障)。闩锁以会话 id 为键、落盘到
+   *  ~/.dsh/memory/auto-continue-done.json(重启后依旧生效),接续**失败不落闩**(允许重试)。 */
+  continuedSessionsFile() { return path.join(dshHome(), 'memory', 'auto-continue-done.json') }
+
+  loadContinuedSessions() {
+    if (this._continuedSessions instanceof Set) return this._continuedSessions
+    const set = new Set()
+    try {
+      const obj = JSON.parse(readFileSync(this.continuedSessionsFile(), 'utf8'))
+      const arr = Array.isArray(obj) ? obj : ((obj && Array.isArray(obj.sessions)) ? obj.sessions : [])
+      for (const it of arr) {
+        const k = this.waterKey(typeof it === 'string' ? it : ((it && it.from) || ''))
+        if (k) set.add(k)
+      }
+    } catch (e) {}
+    this._continuedSessions = set
+    return set
+  }
+
+  isContinuedSession(sid) {
+    const k = this.waterKey(sid)
+    return k ? this.loadContinuedSessions().has(k) : false
+  }
+
+  markContinuedSession(sid, toSid) {
+    const k = this.waterKey(sid)
+    if (!k) return false
+    const set = this.loadContinuedSessions()
+    if (set.has(k)) return false
+    set.add(k)
+    try {
+      const file = this.continuedSessionsFile()
+      mkdirSync(path.dirname(file), { recursive: true })
+      let arr = []
+      try {
+        const obj = JSON.parse(readFileSync(file, 'utf8'))
+        arr = Array.isArray(obj) ? obj : ((obj && Array.isArray(obj.sessions)) ? obj.sessions : [])
+      } catch (e0) {}
+      arr.push({ from: k, to: this.waterKey(toSid), at: Date.now() })
+      if (arr.length > 200) arr = arr.slice(-200)
+      writeFileSync(file, JSON.stringify({ sessions: arr, updatedAt: Date.now() }, null, 2), 'utf8')
+    } catch (e) { diag('markContinuedSession write failed: ' + ((e && e.message) || e)) }
+    diag('auto-continue latch: session ' + k + ' marked continued' + (toSid ? ' -> ' + this.waterKey(toSid) : ''))
+    return true
+  }
+
   /** M-CM6-C·宿主兜底自动接续(2026-09-08,2.2.6):浏览器端 AutoContinueHost 被后台节流/关闭时,
    *  宿主自身在倒计时到期后直接建新会话。协议:turn-stopping 且水位≥阈值 → arm 倒计时(confirmSeconds,
    *  默认 35s,写入 _autoContState)→ 浏览器轮询 GET auto-continue-state 显示确认卡(同意=POST decide
@@ -2043,6 +2104,8 @@ class MemoryEngine {
       const cooldownMin = Number(this.config.autoContinueCooldownMinutes) || 30
       if (st.lastRunAt && now - st.lastRunAt < cooldownMin * 60000) return
       const sid = String(agent.session.id || agent.session.header?.id || '')
+      // 已接续闩锁(2026-09-10):该会话已经从它接续过一次 → 永远不再 arm(与 30 分钟冷却无关)
+      if (this.isContinuedSession(sid)) return
       // 拒绝语义:时间窗 10 分钟内不再 arm(近似客户端「拒绝=跳过本边界」;窗口过后按新边界重新评估)
       if (st.rejectedEdgeAt && now - st.rejectedEdgeAt < 10 * 60 * 1000) return
       if (st.armed && Date.now() < st.armed.expiresAt) return         // 已有进行中的倒计时
@@ -2057,6 +2120,10 @@ class MemoryEngine {
         tokens: Number(wl.tokens) || 0,
         window: Number(wl.window) || 0,
         source: String(wl.source || ''),
+        // 双口径(2026-09-10):小圈读数(声明窗口为分母)与距硬墙剩余,供确认卡显示
+        // (state 可能尚未建立 —— 例如测试夹具或极早的 arm,取不到就不显示,不得因此让 arm 失败)
+        ring: Number(this.state && this.state.waterLevelRing) || 0,
+        wall: Number(this.state && this.state.waterLevelWall) || 0,
         // 2.2.7:pre-step 侧 arm 的倒计时到期时,若回合仍活跃则推迟(避免打断进行中的对话)
         awaitIdle: !!(opts && opts.awaitIdle),
         deferCount: 0,
@@ -2112,6 +2179,13 @@ class MemoryEngine {
       })
       const newId = String((created && (created.sessionId || created.id)) || '')
       if (!newId) throw new Error('create returned no sessionId')
+      // 接续序号标题(2026-09-10):浏览器路径一直有(「接续 #N · 工作区」),宿主兜底路径漏了 ——
+      // 兜底接续出来的会话在侧栏是无名会话,用户无法一眼看出它是「延续会话」还是新开的
+      // (实测 18:07 建出的 session-9cc01f76 日志里根本没有 session/title 事件)。
+      if (d.contSeq && typeof sc.rename === 'function') {
+        const title = '接续 #' + String(d.contSeq) + (d.wsBase ? ' · ' + d.wsBase : '')
+        try { await sc.rename({ sessionId: newId, title }) } catch (eT) { diag('host auto-continue rename: ' + ((eT && eT.message) || eT)) }
+      }
       if (d.provider && d.model && typeof sc.selectModel === 'function') {
         try { await sc.selectModel({ sessionId: newId, provider: d.provider, model: d.model, ...(d.reasoningEffort ? { reasoningEffort: d.reasoningEffort } : {}) }) } catch (eM) { diag('host auto-continue selectModel: ' + (eM && eM.message)) }
       }
@@ -2130,6 +2204,8 @@ class MemoryEngine {
       // 旧会话优先按 id 从 agents 注册表取(_lastAgent 重启后可能为空),fail-soft 不影响接续结果。
       const perm = await this.inheritPermissionForContinue(oldSid, newId)
       const ritualNote = (ritual && ritual.ok) ? 'updated' : ((ritual && ritual.reason) || 'skipped')
+      // 落闩:旧会话已接续过,此后不再被 arm(即使 30 分钟冷却已过)
+      try { this.markContinuedSession(oldSid, newId) } catch (eL) {}
       st.lastRunAt = Date.now()
       st.lastOk = { sessionId: newId, model: d.model || '', reasoningEffort: d.reasoningEffort || '', workspaceId: d.workspaceId || '', permissionPreset: (perm && perm.ok && perm.preset) ? perm.preset : '', refreshRitual: ritualNote }
       diag('auto-continue host-executed: new session ' + newId + ' model=' + (d.model || 'default') + ' perm=' + ((perm && perm.ok && perm.preset) || ('(' + ((perm && perm.reason) || 'n/a') + ')')) + ' ritual=' + ritualNote)
@@ -2153,26 +2229,34 @@ class MemoryEngine {
     }
   }
 
-  /** 浏览器轮询视图:armed 倒计时 / 最近一次执行结果 / 拒绝边界 / 错误。 */
-  autoContinueState() {
+  /** 浏览器轮询视图:armed 倒计时 / 最近一次执行结果 / 拒绝边界 / 错误。
+   *  selfSid(2026-09-10):本状态是**全局单值**,而 armed 只属于某一个会话 —— 不过滤的话,
+   *  别的会话的窗口也会弹「本会话水位已达 x%」(实测:用户切回旧窗口看到属于它的卡片后就以为清不掉)。
+   *  传了 selfSid 且不匹配 → armed 视为 null(取不到 selfSid 时不过滤,fail-open,卡片照常显示)。 */
+  autoContinueState(selfSid) {
     const st = this._autoContState || (this._autoContState = {})
-    const armed = st.armed ? {
+    const selfKey = this.waterKey(selfSid || '')
+    const armedRaw = st.armed ? {
       edgeAt: st.armed.edgeAt,
       sessionId: st.armed.sessionId,
       ratio: st.armed.ratio,
       tokens: st.armed.tokens,
       window: st.armed.window,
+      ring: Number(st.armed.ring) || 0,
+      wall: Number(st.armed.wall) || 0,
       expiresAt: st.armed.expiresAt,
       leftMs: Math.max(0, st.armed.expiresAt - Date.now()),
     } : null
+    const armedMine = !selfKey || !armedRaw || !armedRaw.sessionId || this.waterKey(armedRaw.sessionId) === selfKey
     return {
       enabled: this.config.autoContinueEnabled !== false && this.config.handoffEnabled !== false,
       threshold: Number(this.config.autoContinueThreshold) || 0.75,
-      armed,
+      armed: armedMine ? armedRaw : null,
       executing: !!st.executing,
       rejectedEdgeAt: st.rejectedEdgeAt || 0,
       lastRunAt: st.lastRunAt || 0,
-      lastOk: st.lastOk || null,
+      // at:客户端据此给「已完成」提示加 10 分钟有效期 —— 旧实现没有时间戳,提示会一直挂在所有窗口上
+      lastOk: st.lastOk ? { sessionId: st.lastOk.sessionId, model: st.lastOk.model, reasoningEffort: st.lastOk.reasoningEffort, workspaceId: st.lastOk.workspaceId, permissionPreset: st.lastOk.permissionPreset, refreshRitual: st.lastOk.refreshRitual, at: st.lastRunAt || 0 } : null,
       error: st.error || '',
     }
   }
@@ -7198,6 +7282,11 @@ export function apply(ctx, config) {
         try {
           const body = await readJsonBody(req).catch(() => ({}))
           const r = await engine.inheritPermissionForContinue(body && body.fromSessionId, body && body.toSessionId, { attempts: (body && body.attempts) || undefined })
+          // 落闩(2026-09-10):本端点由 client 在建好新会话后回调,带两个 sid → 正是「接续已发生」的事实点。
+          // 放在这里而不是 handoff-continue(取材料,可能只是预览)是为了只在**真建成**时才关掉旧会话的自动接续。
+          if (body && body.fromSessionId && body.toSessionId) {
+            try { engine.markContinuedSession(body.fromSessionId, body.toSessionId) } catch (eL) {}
+          }
           return writeJson(res, 200, r || { ok: false, reason: 'no result' })
         } catch (e) { return writeJson(res, 500, { error: String(e && e.message ? e.message : e) }) }
       },
@@ -7208,7 +7297,10 @@ export function apply(ctx, config) {
       handler: async (req, res) => {
         // M-CM6-C·宿主兜底状态(GET;loopback):浏览器 AutoContinueHost 轮询此端点展示确认卡/倒计时/执行结果。
         if (!isLoopbackRequest(req)) return writeJson(res, 403, { error: 'forbidden: loopback-only' })
-        try { return writeJson(res, 200, engine.autoContinueState()) } catch (e) { return writeJson(res, 500, { error: String(e && e.message ? e.message : e) }) }
+        try {
+          const url = new URL(req.url, 'http://127.0.0.1')
+          return writeJson(res, 200, engine.autoContinueState(url.searchParams.get('sessionId')))
+        } catch (e) { return writeJson(res, 500, { error: String(e && e.message ? e.message : e) }) }
       },
     },
     {

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@a9i5k4/dsh-auto-memory",
   "description": "Proactive associative memory for DSH: zero-prompt recall injected before the model speaks, three-layer auto-consolidation, skill crystallization, and Astra-style context management - handoff ledgers, PLAN whiteboard, water-level sensing. Local-first, model-agnostic, zero deps. 主动联想记忆+Astra 式上下文管理:自动唤回/自动沉淀/技能固化/交接账本与白板跨窗口续命/水位感知。",
-  "version": "2.4.0",
+  "version": "2.4.1",
   "type": "module",
   "main": "lib/index.js",
   "exports": {