sage-mem 0.4.0 → 0.4.2

package/lib/index.js

@@ -1,212 +1,212 @@
-/**
- * sage-mem — 文件式记忆插件（CC 原生方式）。
- *
- * 记忆存在本地 markdown 文件目录（frontmatter + 正文，4 类），不是数据库。
- * host 侧两件事：
- *   1. 按问题检索注入：system-prompt/assemble 时扫 memory 目录，按双字匹配
- *      选相关文件，读全文注入 system prompt，让 agent 第一轮就「想起」
- *   2. TypertRemoteService：给设置页「记忆管理」提供文件列表/读/写/删
- *
- * 写入靠 AGENTS.md 的「记忆使用」规则引导 agent 自己判断 + 用文件工具写
- * memory 目录——透明、可检查、防膨胀。无 worker、无 SQLite、无端口。
- *
- * 旧 worker 版见 git 分支 `sqlite-worker`。
- */
-
-import { Remote, TypertRemoteService } from '@deepseek-ai/dsh-typert-protocol'
-import { homedir } from 'node:os'
-import { readFile, readdir, writeFile, unlink } from 'node:fs/promises'
-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
-
-/** 提取消息文本，兼容 content 为字符串或 text-block 数组两种形态。 */
-function extractText(content) {
-  if (typeof content === 'string') return content.trim()
-  if (!Array.isArray(content)) return ''
-  return content
-    .filter(b => b && b.type === 'text' && typeof b.text === 'string')
-    .map(b => b.text)
-    .join('\n')
-    .trim()
-}
-
-/** 提取双字（2-gram）集合，中英文数字通用。 */
-function bigrams(s) {
-  const clean = String(s).toLowerCase().replace(/[^\p{L}\p{N}]+/gu, '')
-  const set = new Set()
-  for (let i = 0; i < clean.length - 1; i++) set.add(clean.slice(i, i + 2))
-  return set
-}
-
-/** query 的双字在 text 中的覆盖率（0~1），作为相关性得分。 */
-function score(query, text) {
-  const q = bigrams(query)
-  if (q.size === 0) return 0
-  const t = bigrams(text)
-  let hit = 0
-  for (const g of q) if (t.has(g)) hit++
-  return hit / q.size
-}
-
-/** 扫 memory 目录，读回全部记忆文件的正文（跳过索引与子目录）。 */
-async function scanMemoryFiles() {
-  try {
-    const entries = await readdir(MEMORY_DIR, { withFileTypes: true })
-    const names = entries
-      .filter(e => e.isFile() && e.name.endsWith('.md') && e.name !== 'MEMORY.md')
-      .map(e => e.name)
-    const loaded = await Promise.all(names.map(async (name) => {
-      try {
-        const content = await readFile(join(MEMORY_DIR, name), 'utf8')
-        return { file: name, content }
-      } catch {
-        return null
-      }
-    }))
-    return loaded.filter(Boolean)
-  } catch {
-    return []
-  }
-}
-
-/** 按得分选相关文件，降序取 top N。 */
-function selectRelevant(query, files) {
-  return files
-    .map(f => ({ ...f, score: score(query, f.content) }))
-    .filter(f => f.score > 0)
-    .sort((a, b) => b.score - a.score)
-    .slice(0, MAX_RESULTS)
-}
-
-/** 解析 frontmatter 的 name/description/type（兼容顶层与 metadata 嵌套两种 type）。 */
-function parseFrontmatter(content) {
-  const m = content.match(/^---\n([\s\S]*?)\n---/)
-  if (!m) return { name: '', description: '', type: '' }
-  const body = m[1]
-  const grab = (key) => {
-    const hit = body.match(new RegExp(`^${key}:\\s*(.+)$`, 'm'))
-    if (!hit) return ''
-    return hit[1].replace(/^["']|["']$/g, '').trim()
-  }
-  let type = grab('type')
-  if (!type) {
-    const nested = body.match(/^metadata:\s*\n(?:\s+[^\n]+\n)*?\s+type:\s*(\S+)/m)
-    if (nested) type = nested[1]
-  }
-  return { name: grab('name'), description: grab('description'), type }
-}
-
-/** 防路径穿越：只取 basename，且必须在 memory 目录内（.md，排除索引）。 */
-function safeName(raw) {
-  const name = basename(String(raw || ''))
-  if (!name || !name.endsWith('.md') || name === 'MEMORY.md') return null
-  return name
-}
-
-/**
- * 手动 @Remote marker 注册（Node 24 不解析装饰器语法，手写 decorator context）。
- */
-function markRemote(cls, method, exportName) {
-  const instance = Object.create(cls.prototype)
-  Remote(exportName)(undefined, {
-    kind: 'method',
-    name: method,
-    private: false,
-    static: false,
-    addInitializer(fn) {
-      fn.call(instance)
-    },
-  })
-}
-
-export class MemoryGateway extends TypertRemoteService {
-  constructor(ctx) {
-    super(ctx, 'memory')
-
-    // 按问题检索注入：system-prompt/assemble 是异步 waterfall，监听器可 await。
-    // 从 agent.session 拿当前 user message，扫 memory 目录按双字匹配选相关
-    // 文件，读全文作为动态 context 段注入——让 agent 第一轮就「想起」。
-    ctx.on('system-prompt/assemble', async (assembly, context, next) => {
-      const assembled = await next()
-      try {
-        const agent = context?.agent
-        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
-
-        const files = await scanMemoryFiles()
-        const relevant = selectRelevant(text, files)
-        if (relevant.length === 0) return assembled
-
-        const parts = relevant.map(f => {
-          const body = f.content.length > MAX_CHARS_PER_FILE
-            ? f.content.slice(0, MAX_CHARS_PER_FILE) + '\n…（截断）'
-            : f.content
-          return `### ${f.file}\n${body}`
-        })
-
-        return {
-          ...assembled,
-          contexts: [...(assembled.contexts ?? []), {
-            name: 'sage-mem:recall',
-            order: 1000,
-            text: `以下是与你当前问题相关的历史记忆（文件式，来自 ${MEMORY_DIR}）：\n\n${parts.join('\n\n')}`,
-          }],
-        }
-      } catch {
-        return assembled
-      }
-    })
-  }
-
-  /** 列出 memory 目录全部记忆文件（文件名 + 类型 + 描述 + 大小）。 */
-  async listFiles() {
-    const files = await scanMemoryFiles()
-    return files.map(f => {
-      const meta = parseFrontmatter(f.content)
-      return {
-        file: f.file,
-        type: meta.type || 'reference',
-        description: meta.description || '',
-        size: f.content.length,
-      }
-    })
-  }
-
-  /** 读单个记忆文件全文。 */
-  async readFile(name) {
-    const safe = safeName(name)
-    if (!safe) throw new Error('sage-mem: invalid file name')
-    const content = await readFile(join(MEMORY_DIR, safe), 'utf8')
-    return { name: safe, content }
-  }
-
-  /** 写（新增或覆盖）单个记忆文件。 */
-  async writeFile(name, content) {
-    const safe = safeName(name)
-    if (!safe) throw new Error('sage-mem: invalid file name')
-    await writeFile(join(MEMORY_DIR, safe), String(content ?? ''), 'utf8')
-    return { ok: true, file: safe }
-  }
-
-  /** 删除单个记忆文件。 */
-  async deleteFile(name) {
-    const safe = safeName(name)
-    if (!safe) throw new Error('sage-mem: invalid file name')
-    await unlink(join(MEMORY_DIR, safe))
-    return { ok: true }
-  }
-}
-
-markRemote(MemoryGateway, 'listFiles', 'listFiles')
-markRemote(MemoryGateway, 'readFile', 'readFile')
-markRemote(MemoryGateway, 'writeFile', 'writeFile')
-markRemote(MemoryGateway, 'deleteFile', 'deleteFile')
-
-export default MemoryGateway
+/**
+ * sage-mem — 文件式记忆插件（CC 原生方式）。
+ *
+ * 记忆存在本地 markdown 文件目录（frontmatter + 正文，4 类），不是数据库。
+ * host 侧两件事：
+ *   1. 按问题检索注入：system-prompt/assemble 时扫 memory 目录，按双字匹配
+ *      选相关文件，读全文注入 system prompt，让 agent 第一轮就「想起」
+ *   2. TypertRemoteService：给设置页「记忆管理」提供文件列表/读/写/删
+ *
+ * 写入靠 AGENTS.md 的「记忆使用」规则引导 agent 自己判断 + 用文件工具写
+ * memory 目录——透明、可检查、防膨胀。无 worker、无 SQLite、无端口。
+ *
+ * 旧 worker 版见 git 分支 `sqlite-worker`。
+ */
+
+import { Remote, TypertRemoteService } from '@deepseek-ai/dsh-typert-protocol'
+import { homedir } from 'node:os'
+import { readFile, readdir, writeFile, unlink } from 'node:fs/promises'
+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
+
+/** 提取消息文本，兼容 content 为字符串或 text-block 数组两种形态。 */
+function extractText(content) {
+  if (typeof content === 'string') return content.trim()
+  if (!Array.isArray(content)) return ''
+  return content
+    .filter(b => b && b.type === 'text' && typeof b.text === 'string')
+    .map(b => b.text)
+    .join('\n')
+    .trim()
+}
+
+/** 提取双字（2-gram）集合，中英文数字通用。 */
+function bigrams(s) {
+  const clean = String(s).toLowerCase().replace(/[^\p{L}\p{N}]+/gu, '')
+  const set = new Set()
+  for (let i = 0; i < clean.length - 1; i++) set.add(clean.slice(i, i + 2))
+  return set
+}
+
+/** query 的双字在 text 中的覆盖率（0~1），作为相关性得分。 */
+function score(query, text) {
+  const q = bigrams(query)
+  if (q.size === 0) return 0
+  const t = bigrams(text)
+  let hit = 0
+  for (const g of q) if (t.has(g)) hit++
+  return hit / q.size
+}
+
+/** 扫 memory 目录，读回全部记忆文件的正文（跳过索引与子目录）。 */
+async function scanMemoryFiles() {
+  try {
+    const entries = await readdir(MEMORY_DIR, { withFileTypes: true })
+    const names = entries
+      .filter(e => e.isFile() && e.name.endsWith('.md') && e.name !== 'MEMORY.md' && e.name !== 'session-log.md')
+      .map(e => e.name)
+    const loaded = await Promise.all(names.map(async (name) => {
+      try {
+        const content = await readFile(join(MEMORY_DIR, name), 'utf8')
+        return { file: name, content }
+      } catch {
+        return null
+      }
+    }))
+    return loaded.filter(Boolean)
+  } catch {
+    return []
+  }
+}
+
+/** 按得分选相关文件，降序取 top N。 */
+function selectRelevant(query, files) {
+  return files
+    .map(f => ({ ...f, score: score(query, f.content) }))
+    .filter(f => f.score > 0)
+    .sort((a, b) => b.score - a.score)
+    .slice(0, MAX_RESULTS)
+}
+
+/** 解析 frontmatter 的 name/description/type（兼容顶层与 metadata 嵌套两种 type）。 */
+function parseFrontmatter(content) {
+  const m = content.match(/^---\n([\s\S]*?)\n---/)
+  if (!m) return { name: '', description: '', type: '' }
+  const body = m[1]
+  const grab = (key) => {
+    const hit = body.match(new RegExp(`^${key}:\\s*(.+)$`, 'm'))
+    if (!hit) return ''
+    return hit[1].replace(/^["']|["']$/g, '').trim()
+  }
+  let type = grab('type')
+  if (!type) {
+    const nested = body.match(/^metadata:\s*\n(?:\s+[^\n]+\n)*?\s+type:\s*(\S+)/m)
+    if (nested) type = nested[1]
+  }
+  return { name: grab('name'), description: grab('description'), type }
+}
+
+/** 防路径穿越：只取 basename，且必须在 memory 目录内（.md，排除索引）。 */
+function safeName(raw) {
+  const name = basename(String(raw || ''))
+  if (!name || !name.endsWith('.md') || name === 'MEMORY.md') return null
+  return name
+}
+
+/**
+ * 手动 @Remote marker 注册（Node 24 不解析装饰器语法，手写 decorator context）。
+ */
+function markRemote(cls, method, exportName) {
+  const instance = Object.create(cls.prototype)
+  Remote(exportName)(undefined, {
+    kind: 'method',
+    name: method,
+    private: false,
+    static: false,
+    addInitializer(fn) {
+      fn.call(instance)
+    },
+  })
+}
+
+export class MemoryGateway extends TypertRemoteService {
+  constructor(ctx) {
+    super(ctx, 'memory')
+
+    // 按问题检索注入：system-prompt/assemble 是异步 waterfall，监听器可 await。
+    // 从 agent.session 拿当前 user message，扫 memory 目录按双字匹配选相关
+    // 文件，读全文作为动态 context 段注入——让 agent 第一轮就「想起」。
+    ctx.on('system-prompt/assemble', async (assembly, context, next) => {
+      const assembled = await next()
+      try {
+        const agent = context?.agent
+        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
+
+        const files = await scanMemoryFiles()
+        const relevant = selectRelevant(text, files)
+        if (relevant.length === 0) return assembled
+
+        const parts = relevant.map(f => {
+          const body = f.content.length > MAX_CHARS_PER_FILE
+            ? f.content.slice(0, MAX_CHARS_PER_FILE) + '\n…（截断）'
+            : f.content
+          return `### ${f.file}\n${body}`
+        })
+
+        return {
+          ...assembled,
+          contexts: [...(assembled.contexts ?? []), {
+            name: 'sage-mem:recall',
+            order: 1000,
+            text: `以下是与你当前问题相关的历史记忆（文件式，来自 ${MEMORY_DIR}）：\n\n${parts.join('\n\n')}`,
+          }],
+        }
+      } catch {
+        return assembled
+      }
+    })
+  }
+
+  /** 列出 memory 目录全部记忆文件（文件名 + 类型 + 描述 + 大小）。 */
+  async listFiles() {
+    const files = await scanMemoryFiles()
+    return files.map(f => {
+      const meta = parseFrontmatter(f.content)
+      return {
+        file: f.file,
+        type: meta.type || 'reference',
+        description: meta.description || '',
+        size: f.content.length,
+      }
+    })
+  }
+
+  /** 读单个记忆文件全文。 */
+  async readFile(name) {
+    const safe = safeName(name)
+    if (!safe) throw new Error('sage-mem: invalid file name')
+    const content = await readFile(join(MEMORY_DIR, safe), 'utf8')
+    return { name: safe, content }
+  }
+
+  /** 写（新增或覆盖）单个记忆文件。 */
+  async writeFile(name, content) {
+    const safe = safeName(name)
+    if (!safe) throw new Error('sage-mem: invalid file name')
+    await writeFile(join(MEMORY_DIR, safe), String(content ?? ''), 'utf8')
+    return { ok: true, file: safe }
+  }
+
+  /** 删除单个记忆文件。 */
+  async deleteFile(name) {
+    const safe = safeName(name)
+    if (!safe) throw new Error('sage-mem: invalid file name')
+    await unlink(join(MEMORY_DIR, safe))
+    return { ok: true }
+  }
+}
+
+markRemote(MemoryGateway, 'listFiles', 'listFiles')
+markRemote(MemoryGateway, 'readFile', 'readFile')
+markRemote(MemoryGateway, 'writeFile', 'writeFile')
+markRemote(MemoryGateway, 'deleteFile', 'deleteFile')
+
+export default MemoryGateway

package/lib/typert.host.js

@@ -1,90 +1,90 @@
-/**
- * sage-mem — 手写 strict typert host manifest（对齐官方生成文件格式）。
- * typert-loader 在插件挂载时读 package.json 的 exports["./typert"]，
- * import 本文件并把 TYPERT 注册进 ctx.typert.local（strict 定义）。
- * schema 必须是 zod v4 实例（typert-loader 校验 "_zod" in schema）。
- */
-import { z } from 'zod'
-
-const fileNameSchema = z.string()
-const fileContentSchema = z.string()
-
-const fileSchema = z.object({
-	'file': z.string().readonly(),
-	'type': z.string().readonly(),
-	'description': z.string().readonly(),
-	'size': z.number().readonly(),
-}).readonly()
-
-const listResultSchema = z.array(fileSchema)
-const readResultSchema = z.object({
-	'name': z.string().readonly(),
-	'content': z.string().readonly(),
-}).readonly()
-const writeResultSchema = z.object({
-	'ok': z.boolean().readonly(),
-	'file': z.string().readonly(),
-}).readonly()
-const deleteResultSchema = z.object({
-	'ok': z.boolean().readonly(),
-}).readonly()
-
-export const TYPERT = {
-	package: 'sage-mem',
-	face: 'host',
-	schemas: [],
-	invocations: [
-		{
-			id: 'sage-mem#memory/listFiles',
-			service: 'memory',
-			namespace: 'memory',
-			method: 'listFiles',
-			invocation: { kind: 'direct' },
-			parameters: [],
-			result: { mode: 'strict', typeSymbol: 'sage-mem/types#FileList', schema: listResultSchema },
-			sourceLocation: { "file": "workspace/sage-mem/lib/index.js", "line": 1, "column": 1 },
-		},
-		{
-			id: 'sage-mem#memory/readFile',
-			service: 'memory',
-			namespace: 'memory',
-			method: 'readFile',
-			invocation: { kind: 'direct' },
-			parameters: [
-				{ name: 'name', wire: 'name', source: 'json', codec: { mode: 'strict', typeSymbol: 'sage-mem/types#FileName', schema: fileNameSchema } },
-			],
-			result: { mode: 'strict', typeSymbol: 'sage-mem/types#FileContent', schema: readResultSchema },
-			sourceLocation: { "file": "workspace/sage-mem/lib/index.js", "line": 1, "column": 1 },
-		},
-		{
-			id: 'sage-mem#memory/writeFile',
-			service: 'memory',
-			namespace: 'memory',
-			method: 'writeFile',
-			invocation: { kind: 'direct' },
-			parameters: [
-				{ name: 'name', wire: 'name', source: 'json', codec: { mode: 'strict', typeSymbol: 'sage-mem/types#FileName', schema: fileNameSchema } },
-				{ name: 'content', wire: 'content', source: 'json', codec: { mode: 'strict', typeSymbol: 'sage-mem/types#FileContent', schema: fileContentSchema } },
-			],
-			result: { mode: 'strict', typeSymbol: 'sage-mem/types#WriteResult', schema: writeResultSchema },
-			sourceLocation: { "file": "workspace/sage-mem/lib/index.js", "line": 1, "column": 1 },
-		},
-		{
-			id: 'sage-mem#memory/deleteFile',
-			service: 'memory',
-			namespace: 'memory',
-			method: 'deleteFile',
-			invocation: { kind: 'direct' },
-			parameters: [
-				{ name: 'name', wire: 'name', source: 'json', codec: { mode: 'strict', typeSymbol: 'sage-mem/types#FileName', schema: fileNameSchema } },
-			],
-			result: { mode: 'strict', typeSymbol: 'sage-mem/types#DeleteResult', schema: deleteResultSchema },
-			sourceLocation: { "file": "workspace/sage-mem/lib/index.js", "line": 1, "column": 1 },
-		},
-	],
-	model: {
-		"services": [],
-		"events": [],
-		"objects": [],
-	},
-}
+/**
+ * sage-mem — 手写 strict typert host manifest（对齐官方生成文件格式）。
+ * typert-loader 在插件挂载时读 package.json 的 exports["./typert"]，
+ * import 本文件并把 TYPERT 注册进 ctx.typert.local（strict 定义）。
+ * schema 必须是 zod v4 实例（typert-loader 校验 "_zod" in schema）。
+ */
+import { z } from 'zod'
+
+const fileNameSchema = z.string()
+const fileContentSchema = z.string()
+
+const fileSchema = z.object({
+	'file': z.string().readonly(),
+	'type': z.string().readonly(),
+	'description': z.string().readonly(),
+	'size': z.number().readonly(),
+}).readonly()
+
+const listResultSchema = z.array(fileSchema)
+const readResultSchema = z.object({
+	'name': z.string().readonly(),
+	'content': z.string().readonly(),
+}).readonly()
+const writeResultSchema = z.object({
+	'ok': z.boolean().readonly(),
+	'file': z.string().readonly(),
+}).readonly()
+const deleteResultSchema = z.object({
+	'ok': z.boolean().readonly(),
+}).readonly()
+
+export const TYPERT = {
+	package: 'sage-mem',
+	face: 'host',
+	schemas: [],
+	invocations: [
+		{
+			id: 'sage-mem#memory/listFiles',
+			service: 'memory',
+			namespace: 'memory',
+			method: 'listFiles',
+			invocation: { kind: 'direct' },
+			parameters: [],
+			result: { mode: 'strict', typeSymbol: 'sage-mem/types#FileList', schema: listResultSchema },
+			sourceLocation: { "file": "workspace/sage-mem/lib/index.js", "line": 1, "column": 1 },
+		},
+		{
+			id: 'sage-mem#memory/readFile',
+			service: 'memory',
+			namespace: 'memory',
+			method: 'readFile',
+			invocation: { kind: 'direct' },
+			parameters: [
+				{ name: 'name', wire: 'name', source: 'json', codec: { mode: 'strict', typeSymbol: 'sage-mem/types#FileName', schema: fileNameSchema } },
+			],
+			result: { mode: 'strict', typeSymbol: 'sage-mem/types#FileContent', schema: readResultSchema },
+			sourceLocation: { "file": "workspace/sage-mem/lib/index.js", "line": 1, "column": 1 },
+		},
+		{
+			id: 'sage-mem#memory/writeFile',
+			service: 'memory',
+			namespace: 'memory',
+			method: 'writeFile',
+			invocation: { kind: 'direct' },
+			parameters: [
+				{ name: 'name', wire: 'name', source: 'json', codec: { mode: 'strict', typeSymbol: 'sage-mem/types#FileName', schema: fileNameSchema } },
+				{ name: 'content', wire: 'content', source: 'json', codec: { mode: 'strict', typeSymbol: 'sage-mem/types#FileContent', schema: fileContentSchema } },
+			],
+			result: { mode: 'strict', typeSymbol: 'sage-mem/types#WriteResult', schema: writeResultSchema },
+			sourceLocation: { "file": "workspace/sage-mem/lib/index.js", "line": 1, "column": 1 },
+		},
+		{
+			id: 'sage-mem#memory/deleteFile',
+			service: 'memory',
+			namespace: 'memory',
+			method: 'deleteFile',
+			invocation: { kind: 'direct' },
+			parameters: [
+				{ name: 'name', wire: 'name', source: 'json', codec: { mode: 'strict', typeSymbol: 'sage-mem/types#FileName', schema: fileNameSchema } },
+			],
+			result: { mode: 'strict', typeSymbol: 'sage-mem/types#DeleteResult', schema: deleteResultSchema },
+			sourceLocation: { "file": "workspace/sage-mem/lib/index.js", "line": 1, "column": 1 },
+		},
+	],
+	model: {
+		"services": [],
+		"events": [],
+		"objects": [],
+	},
+}