sage-mem 0.4.3 → 0.5.0

package/README.md

@@ -25,7 +25,8 @@ sage-mem 换成文件式：每条记忆一个 markdown 文件，直接用编辑
 
 ## 功能
 
-- **按问题检索注入**：每次提问，插件扫记忆目录，按相关性把记忆注入 system prompt，agent 第一轮就「想起来」
+- **按问题检索注入**：每次提问，插件扫记忆目录，用 frontmatter 的 `description` 和文件名做匹配，把相关记忆注入上下文，agent 第一轮就「想起来」
+- **每次会话必读**（v0.5）：frontmatter 标了 `baseline: true` 的记忆，在新会话第一步无条件注入——给「待办清单」「你是谁」这类每次开场都该在场的记忆用（上限 5 条）
 - **四类记忆**：`user`（用户是谁）/ `feedback`（工作方式指导）/ `project`（项目状态）/ `reference`（外部信息指针）
 - **文件透明**：记忆就是 `*.md` 文件，frontmatter 存元数据，正文存内容
 - **防膨胀**：写入靠规则引导 agent 判断「存不存」，无实质内容、能靠代码/git 推导的不存
@@ -86,6 +87,7 @@ sage-mem 是纯 DSH 插件，在 `package.json` 的 `dsh.compatibility.dshReleas
 | 0.1.2-rc.1 | compatible |
 | 0.1.3-alpha.1 | compatible |
 | 0.1.3-alpha.2 | compatible |
+| 0.1.5-rc.1 | compatible |
 
 Node.js 要求 `>= 18`（见 `engines.node`）。
 
@@ -107,9 +109,10 @@ SAGE_MEM_DIR=/path/to/your/memory
 ```markdown
 ---
 name: 可选的短名
-description: 一句话说清这条记忆是什么（检索与列表都靠它）
+description: 一句话说清这条记忆是什么（检索靠它，务必写准）
 metadata:
   type: user        # user / feedback / project / reference
+baseline: true      # 可选：新会话第一步无条件注入（上限 5 条）
 ---
 
 记忆正文。

package/lib/index.js

@@ -21,6 +21,7 @@ import { join, basename } from 'node:path'
 const MEMORY_DIR = process.env.SAGE_MEM_DIR || join(homedir(), '.sage-mem', 'memory')
 const MAX_RESULTS = 5
 const MAX_CHARS_PER_FILE = 1500
+const MAX_BASELINE = 5
 
 /** 提取消息文本，兼容 content 为字符串或 text-block 数组两种形态。 */
 function extractText(content) {
@@ -51,7 +52,7 @@ function score(query, text) {
   return hit / q.size
 }
 
-/** 扫 memory 目录，读回全部记忆文件的正文（跳过索引与子目录）。 */
+/** 扫 memory 目录，读回全部记忆文件（跳过索引与子目录），并解析 frontmatter 摘要。 */
 async function scanMemoryFiles() {
   try {
     const entries = await readdir(MEMORY_DIR, { withFileTypes: true })
@@ -61,7 +62,13 @@ async function scanMemoryFiles() {
     const loaded = await Promise.all(names.map(async (name) => {
       try {
         const content = await readFile(join(MEMORY_DIR, name), 'utf8')
-        return { file: name, content }
+        const meta = parseFrontmatter(content)
+        return {
+          file: name,
+          content,
+          description: meta.description || '',
+          baseline: parseBaseline(content),
+        }
       } catch {
         return null
       }
@@ -72,10 +79,22 @@ async function scanMemoryFiles() {
   }
 }
 
-/** 按得分选相关文件，降序取 top N。 */
+/**
+ * 相关性打分只用 frontmatter 的 description + 文件名，不扫全文。
+ *
+ * 旧版拿 query 的双字去全文里找，得分是「覆盖率」，分母是 query，不惩罚
+ * 文件长度 —— 越长的记忆越容易蒙中。实测 8 个典型问题：project_heartscape
+ * (5.9KB) 与 project_self-cognition (6.8KB) 命中 7 个，连「今天天气不错」都
+ * 注入 2300 token；而「论文写得怎么样了」该命中的 project_position-paper
+ * 反被挤掉。description 是「一句话说清这条是什么」的精准摘要，用它当检索
+ * 信号，噪音大幅下降。
+ */
 function selectRelevant(query, files) {
   return files
-    .map(f => ({ ...f, score: score(query, f.content) }))
+    .map(f => {
+      const hay = f.description + ' ' + f.file.replace(/\.md$/, '').replace(/[_-]/g, ' ')
+      return { ...f, score: score(query, hay) }
+    })
     .filter(f => f.score > 0)
     .sort((a, b) => b.score - a.score)
     .slice(0, MAX_RESULTS)
@@ -99,6 +118,15 @@ function parseFrontmatter(content) {
   return { name: grab('name'), description: grab('description'), type }
 }
 
+/** 解析 frontmatter 的 `baseline` 标记：会话第一回合无条件注入。 */
+function parseBaseline(content) {
+  const m = content.match(/^---\r?\n([\s\S]*?)\r?\n---/)
+  if (!m) return false
+  const hit = m[1].match(/^baseline:\s*(.+)$/m)
+  if (!hit) return false
+  return /^(true|yes|1)$/i.test(hit[1].replace(/^["']|["']$/g, '').trim())
+}
+
 /** 防路径穿越：只取 basename，且必须在 memory 目录内（.md，排除索引）。 */
 function safeName(raw) {
   const name = basename(String(raw || ''))
@@ -128,8 +156,12 @@ export class MemoryGateway extends TypertRemoteService {
     // 一并挂载星图 remote（同一记忆目录，纯只读）—— starmap.* 服务
     new StarmapGateway(ctx)
 
+    // 同一轮对话里问题不变、注入内容也不变，但 DSH 每一步都会往 session 追加
+    // 一条 context 消息——不去重的话同一份记忆会被反复追加，撑爆上下文。
+    const recallCache = new Map()
+
     // 按问题检索注入：system-prompt/assemble 是异步 waterfall，监听器可 await。
-    // 从 agent.session 拿当前 user message，扫 memory 目录按双字匹配选相关
+    // 从 agent.session 拿当前 user message，扫 memory 目录按摘要匹配选相关
     // 文件，读全文作为动态 context 段注入——让 agent 第一轮就「想起」。
     ctx.on('system-prompt/assemble', async (assembly, context, next) => {
       const assembled = await next()
@@ -138,15 +170,44 @@ export class MemoryGateway extends TypertRemoteService {
         const session = agent?.session
         if (!session || typeof session.deriveMessages !== 'function') return assembled
         const messages = [...session.deriveMessages()]
-        const lastUser = messages.reverse().find(m => m?.role === 'user')
-        const text = extractText(lastUser?.content)
-        if (!text || text.length < 2) return assembled
+
+        // 关键：DSH 0.1.5 把工具结果（source.kind === 'tool'）和注入内容
+        // （source.kind === 'plugin'）也存成 role: 'user' 的消息。只按 role 取
+        // 「最后一条」会取到 tool-result，extractText 得到空串，下面的长度检查
+        // 直接返回——表现就是全程静默不注入。只有 source.kind === 'user' 是真人输入。
+        const userMessages = messages.filter(m => m?.role === 'user' && m?.source?.kind === 'user')
+        const text = extractText(userMessages[userMessages.length - 1]?.content)
 
         const files = await scanMemoryFiles()
-        const relevant = selectRelevant(text, files)
-        if (relevant.length === 0) return assembled
+        const picked = []
+        const seen = new Set()
+        const take = (f) => {
+          if (seen.has(f.file)) return
+          seen.add(f.file)
+          picked.push(f)
+        }
+
+        // 新会话首步：注入 frontmatter 标了 `baseline: true` 的记忆。当前这条
+        // 用户消息在 assemble 时还没写进 session，所以首步这里恰好是 0 条。
+        // 这一层补「开场白失忆」——「继续吧」这类短话在 description 里找不到
+        // 匹配，纯检索会返回 0 条，而那恰恰是最需要上下文的时刻。
+        if (userMessages.length === 0) {
+          files
+            .filter(f => f.baseline)
+            .sort((a, b) => a.file.localeCompare(b.file))
+            .slice(0, MAX_BASELINE)
+            .forEach(take)
+        }
+
+        if (text && text.length >= 2) selectRelevant(text, files).forEach(take)
+        if (picked.length === 0) return assembled
+
+        const signature = text + '\u0000' + picked.map(f => f.file).join(',')
+        if (recallCache.get(session.id) === signature) return assembled
+        recallCache.set(session.id, signature)
+        if (recallCache.size > 8) recallCache.delete(recallCache.keys().next().value)
 
-        const parts = relevant.map(f => {
+        const parts = picked.map(f => {
           const body = f.content.length > MAX_CHARS_PER_FILE
             ? f.content.slice(0, MAX_CHARS_PER_FILE) + '\n…（截断）'
             : f.content

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "sage-mem",
   "description": "File-based memory plugin for DeepSeek Harness (DSH) — cross-session memory in transparent markdown files, with a memory star-map view",
-  "version": "0.4.3",
+  "version": "0.5.0",
   "type": "module",
   "main": "lib/index.js",
   "exports": {
@@ -36,7 +36,8 @@
       "dshReleases": {
         "0.1.2-rc.1": "compatible",
         "0.1.3-alpha.1": "compatible",
-        "0.1.3-alpha.2": "compatible"
+        "0.1.3-alpha.2": "compatible",
+        "0.1.5-rc.1": "compatible"
       }
     }
   },
@@ -51,7 +52,7 @@
   },
   "repository": {
     "type": "git",
-    "url": "https://github.com/gezi-wen/sage-mem.git"
+    "url": "git+https://github.com/gezi-wen/sage-mem.git"
   },
   "keywords": [
     "memory",