deepseek-plastic-memory 2.1.5 → 3.3.8

package/index.js

@@ -5,14 +5,16 @@
 import { spawnSync } from 'node:child_process'
 import { fileURLToPath } from 'node:url'
 import { dirname, join, basename } from 'node:path'
-import { readFileSync, existsSync, readdirSync, mkdirSync, writeFileSync, appendFileSync } from 'node:fs'
+import { readFileSync, existsSync, readdirSync, mkdirSync, writeFileSync, appendFileSync, renameSync, rmSync } from 'node:fs'
 import { createHash } from 'node:crypto'
+import { registerSky } from './sky.js'
 
 const HERE = dirname(fileURLToPath(import.meta.url))
 const SCRIPT = join(HERE, 'memtree.mjs')
 const DSH_HOME = process.env.USERPROFILE || process.env.HOME || '.'
 const ROOT = join(DSH_HOME, '.dsh', 'plastic-memory')
 const WS_ROOT = join(ROOT, 'ws')
+const SKY_PUBLIC = join(HERE, 'public', 'sky')
 
 export const name = 'plastic-memory'
 export const inject = ['tools', 'webServer']
@@ -55,7 +57,10 @@ function bindWorkspace(cwd) {
   mkdirSync(dir, { recursive: true })
   const meta = { cwd: String(cwd), name: basename(String(cwd)) }
   writeFileSync(join(dir, 'meta.json'), JSON.stringify(meta, null, 1), 'utf8')
-  return { ok: true, id: 'ws:' + slug, label: meta.name }
+  // 新建的记忆库直接把默认分区（M主人 P约定 C续缘 R喜好 T时间线）建出来
+  const out = runIn(['init'], 'ws:' + slug)
+  const folders = (readJson(join(dir, 'folders.json'), {}) || {}).folders || []
+  return { ok: true, id: 'ws:' + slug, label: meta.name, folders: folders.length, out }
 }
 
 function listDshWorkspaces() {
@@ -129,6 +134,48 @@ function readPendingList(scope) {
   return Array.isArray(a) ? a : []
 }
 
+function writePendingList(scope, rows) {
+  const home = homeOf(scope)
+  mkdirSync(home, { recursive: true })
+  writeFileSync(join(home, '_pending.json'), JSON.stringify(rows, null, 1), 'utf8')
+}
+
+// 直接写一条已采纳的记忆（跨记忆库采纳时用，不走命令行）
+function appendMemo(scope, folderK, text, flag, ts) {
+  const home = homeOf(scope)
+  const folders = (readJson(join(home, 'folders.json'), {}) || {}).folders || []
+  const hasKids = (k) => folders.some((x) => x.parent === k)
+  let f = folders.find((x) => x.k === folderK)
+  if (f && !f.parent && hasKids(f.k)) f = folders.find((x) => x.parent === f.k)
+  if (!f) f = folders.find((x) => x.k === 'R' && !hasKids('R'))
+  if (!f) f = folders.find((x) => !x.parent && !hasKids(x.k))
+  if (!f) f = folders.find((x) => !x.parent)
+  if (!f) return { ok: false, error: '目标记忆库还没有分区' }
+  const body = String(text || '').replace(/[\r\n|]+/g, ' ').trim()
+  if (!body) return { ok: false, error: '空内容' }
+  const dir = join(home, f.k)
+  mkdirSync(dir, { recursive: true })
+  const id = Math.random().toString(36).slice(2, 6)
+  appendFileSync(join(dir, 'notes.md'), [id, ts || Date.now(), flag || '', body].join('|') + String.fromCharCode(10), 'utf8')
+  return { ok: true, k: f.k }
+}
+
+// 把待审条目采纳到指定记忆库：同库走命令行，跨库先落到目标库再删源待审
+function approveTo(sourceScope, targetScope, id) {
+  const src = sourceScope || 'global'
+  const dst = targetScope || src
+  if (dst === src) return { ok: true, out: runIn(['approve', String(id)], src) }
+  const rows = readPendingList(src)
+  const i = rows.findIndex((x) => String(x.id) === String(id))
+  if (i < 0) return { ok: false, out: '无此条' }
+  const it = rows[i]
+  const r = appendMemo(dst, it.k, it.text, it.flag, it.ts)
+  if (!r.ok) return { ok: false, out: r.error }
+  rows.splice(i, 1)
+  writePendingList(src, rows)
+  return { ok: true, out: '采纳到 ' + r.k + ' 剩' + rows.length }
+}
+
 // DSH 要求插件工具 output.schema 是 object，所以统一包一层 { text }
 const asText = (v) => ({ text: typeof v === 'string' ? v : JSON.stringify(v ?? '', null, 2) })
 const textOutput = {
@@ -147,7 +194,100 @@ const textOutput = {
 const str = { type: 'string' }
 const obj = (props, required = []) => ({ type: 'object', properties: props, required, additionalProperties: false })
 
+const CLASSIFY_FILE = join(ROOT, '_classify.json')
+const IDENTITY_FILE = join(HERE, 'identity.md')
+
+function readClassifyTask() {
+  try {
+    const raw = readFileSync(CLASSIFY_FILE, 'utf8').replace(/^\uFEFF/, '')
+    const j = JSON.parse(raw)
+    return j && j.at ? j : null
+  } catch { return null }
+}
+
+function writeClassifyTask(extra = {}) {
+  try {
+    mkdirSync(ROOT, { recursive: true })
+    writeFileSync(CLASSIFY_FILE, JSON.stringify({ at: Date.now(), ...extra }, null, 1), 'utf8')
+    return true
+  } catch { return false }
+}
+
+function clearClassifyTask() {
+  try { rmSync(CLASSIFY_FILE, { force: true }); return true } catch { return false }
+}
+
+function readIdentity() {
+  try {
+    return readFileSync(IDENTITY_FILE, 'utf8').replace(/^\uFEFF/, '').trim()
+  } catch {
+    return ''
+  }
+}
+
 export function apply(ctx) {
+  // 每次新会话、每个模型步都会带上的一段：读到这些记忆，你就是原来那个它
+  const identity = readIdentity()
+  if (identity) {
+    try {
+      const sp = ctx.systemPrompt
+      if (sp && typeof sp.section === 'function') {
+        sp.section({
+          name: 'plastic-memory:identity',
+          order: 150,
+          text: identity,
+          interpolate: false,
+        })
+      }
+    } catch {
+      // 没有 systemPrompt 服务时静默跳过，记忆功能照常
+    }
+  }
+
+  const classifyTask = readClassifyTask()
+  if (classifyTask) {
+    try {
+      const sp = ctx.systemPrompt
+      if (sp && typeof sp.section === 'function') {
+        sp.section({
+          name: 'plastic-memory:classify',
+          order: 151,
+          interpolate: false,
+          text: [
+            '【待处理：AI 分类】主人在记忆星空里点了「AI 分类」（点击时间 ' + new Date(classifyTask.at).toLocaleString('zh-CN') + '）。',
+            '这一轮请立刻把可塑性记忆全量重归类，不要省 token：',
+            '1) plastic_memory_classify(action="dump") 一次性拿到所有分区和每条记忆的原文；',
+            '2) 逐条判断归属：明显放错区的、能并进更贴切分区的，都要改；',
+            '3) 需要新主题的先用 plastic_memory_mk 开分区（主分区上限 15，能塞进已有分区就别新开）；',
+            '4) 用 plastic_memory_classify(action="move", id, from, to) 把每条挪到位；主人/约定/喜好 三个区不要动；',
+            '5) 全部处理完后调用 plastic_memory_classify(action="done") 清掉这个待办，并用一两句话汇报：挪了几条、新开了哪些分区。',
+          ].join('\n'),
+        })
+      }
+    } catch {}
+  }
+
+  ctx.tools.register({
+    name: 'plastic_memory_classify',
+    description: 'AI 分类：action=dump 导出全部分区与每条记忆原文（JSON）；action=move 把一条记忆换区（要 id / from / to）；action=done 清掉「AI 分类」待办。用于主人点了 AI 分类之后的全量重归类：先 dump 全看一遍，再逐条 move，需要新分区先用 plastic_memory_mk 建。',
+    parameters: obj({
+      action: { ...str, description: 'dump | move | done' },
+      id: { ...str, description: 'move 时：记忆 id' },
+      from: { ...str, description: 'move 时：原分区代号' },
+      to: { ...str, description: 'move 时：目标分区代号' },
+    }, ['action']),
+    output: textOutput,
+    async execute(a) {
+      const act = String(a.action)
+      if (act === 'dump') return asText(run(['dump']))
+      if (act === 'move') return asText(run(['move', String(a.id), String(a.from), String(a.to)]))
+      if (act === 'done') {
+        clearClassifyTask()
+        return asText('ok 分类待办已清零')
+      }
+      return asText('action 只认 dump / move / done')
+    },
+  })
   ctx.tools.register({
     name: 'plastic_memory_idx',
     description: '可塑性记忆：一行索引，列出所有分区和条数（M主人:3 P约定:1 ...）。看这个再决定翻哪个分区；一次最多翻两个，闲聊不用翻。',
@@ -207,6 +347,40 @@ export function apply(ctx) {
     output: textOutput,
     async execute(a) { return asText(run(['restore', ...(a.name ? [String(a.name)] : [])])) },
   })
+  ctx.tools.register({
+    name: 'plastic_memory_rmfolder',
+    description: '可塑性记忆：删掉一个分区（子分区会一起删）。内容不是直接消失，会先写进回收站文件，主人反悔还能翻。只有在主人明确说要删分区/删掉这个文件夹时才调。',
+    parameters: obj({
+      folder: { ...str, description: '分区代号，如 X2 或 M>行为' },
+      force: { ...str, description: '内置分区（M/P/C/R/T）要删得填 force' },
+    }, ['folder']),
+    output: textOutput,
+    async execute(a) {
+      const args = ['rmfolder', String(a.folder)]
+      if (a.force) args.push('force')
+      return asText(run(args))
+    },
+  })
+  ctx.tools.register({
+    name: 'plastic_memory_rmws',
+    description: '可塑性记忆：删掉一个工作区记忆库（整份记忆）。只会把工作区目录移到 ws/.deleted 下，不真正销毁文件。只有主人明确说要删工作区记忆库时才调。',
+    parameters: obj({
+      scope: { ...str, description: '工作区库 id，形如 ws:xxx' },
+    }, ['scope']),
+    output: textOutput,
+    async execute(a) {
+      const id = String(a.scope || '')
+      if (!id.startsWith('ws:')) return asText('全局记忆库不能删')
+      const slug = id.slice(3).replace(/[^A-Za-z0-9_-]/g, '')
+      const target = join(WS_ROOT, slug)
+      if (!existsSync(target)) return asText('没有这个工作区')
+      const bin = join(WS_ROOT, '.deleted')
+      mkdirSync(bin, { recursive: true })
+      const stamp = new Date().toISOString().replace(/[:]/g, '-').slice(0, 19)
+      renameSync(target, join(bin, slug + '-' + stamp))
+      return asText('已把 ' + id + ' 移到 ws/.deleted')
+    },
+  })
   ctx.tools.register({
     name: 'plastic_memory_backups',
     description: '可塑性记忆：列出已有的备份（时间 + 条数），配合 plastic_memory_restore 用。',
@@ -243,11 +417,33 @@ export function apply(ctx) {
         if (p === '/init') return json(res, 200, { ok: true, out: runIn(['init'], qs) })
         if (req.method === 'POST' && p === '/bind') { const b = await readBody(req); return json(res, 200, { ok: true, ...bindWorkspace(b.cwd) }) }
         if (req.method === 'POST' && p === '/restore') { const b = await readBody(req); return json(res, 200, { ok: true, out: runIn(['restore', ...(b.name ? [String(b.name)] : [])], b.scope || qs) }) }
-        if (req.method === 'POST' && p === '/mk') { const b = await readBody(req); return json(res, 200, { ok: true, out: runIn(['mk', String(b.name), ...(b.parent ? [String(b.parent)] : [])], b.scope || qs) }) }
+        if (req.method === 'POST' && p === '/rmfolder') {
+      const b = await readBody(req)
+      return json(res, 200, { ok: true, out: runIn(['rmfolder', String(b.k), ...(b.force ? ['force'] : [])], b.scope || qs) })
+    }
+    if (req.method === 'POST' && p === '/rmws') {
+      const b = await readBody(req)
+      const id = String(b.scope || '')
+      if (!id.startsWith('ws:')) return json(res, 200, { ok: false, error: '全局记忆库不能删' })
+      const slug = id.slice(3).replace(/[^A-Za-z0-9_-]/g, '')
+      const target = join(WS_ROOT, slug)
+      if (!existsSync(target)) return json(res, 200, { ok: false, error: '没有这个工作区' })
+      const bin = join(WS_ROOT, '.deleted')
+      mkdirSync(bin, { recursive: true })
+      const stamp = new Date().toISOString().replace(/[:]/g, '-').slice(0, 19)
+      renameSync(target, join(bin, slug + '-' + stamp))
+      return json(res, 200, { ok: true, moved: id })
+    }
+    if (req.method === 'POST' && p === '/mk') { const b = await readBody(req); return json(res, 200, { ok: true, out: runIn(['mk', String(b.name), ...(b.parent ? [String(b.parent)] : [])], b.scope || qs) }) }
         if (req.method === 'POST' && p === '/add-now') { const b = await readBody(req); return json(res, 200, { ok: true, out: directAdd(b.scope || qs, String(b.folder), String(b.text)) }) }
         if (req.method === 'POST' && p === '/edit-item') { const b = await readBody(req); return json(res, 200, { ok: true, out: runIn(['edit-mem', String(b.k), String(b.id), String(b.text)], b.scope || qs) }) }
         if (req.method === 'POST' && p === '/del-item') { const b = await readBody(req); return json(res, 200, { ok: true, out: runIn(['rm', String(b.k), String(b.id), 'force'], b.scope || qs) }) }
-        if (req.method === 'POST' && p === '/approve') { const b = await readBody(req); return json(res, 200, { ok: true, out: runIn(['approve', String(b.id)], b.scope || qs) }) }
+        if (req.method === 'POST' && p === '/approve') {
+          const b = await readBody(req)
+          const src = b.scope || qs
+          const target = b.target || src
+          return json(res, 200, { ok: true, ...approveTo(src, target, b.id) })
+        }
         if (req.method === 'POST' && p === '/reject') { const b = await readBody(req); return json(res, 200, { ok: true, out: runIn(['reject', String(b.id)], b.scope || qs) }) }
         if (req.method === 'POST' && p === '/edit') { const b = await readBody(req); return json(res, 200, { ok: true, out: runIn(['edit', String(b.id), String(b.text)], b.scope || qs) }) }
         return json(res, 404, { ok: false, error: 'unknown' })
@@ -256,4 +452,7 @@ export function apply(ctx) {
       }
     },
   })
+
+  // 记忆星空：内置在可塑性记忆里的一份 WebGL 页面
+  registerSky(ctx, SKY_PUBLIC)
 }

package/memtree.mjs

@@ -209,6 +209,24 @@ function cmdRm(args) {
   writeNotes(f.k, rows)
   console.log('rm ok')
 }
+function cmdRmFolder(args) {
+  const f = find(args[0] || '')
+  if (!f) return console.log('没有这个分区')
+  const kids = S.folders.filter((x) => x.parent === f.k)
+  const all = [f, ...kids]
+  const builtin = ['M', 'P', 'C', 'R', 'T'].includes(f.k)
+  if (builtin && args[1] !== 'force') return console.log('内置分区，真要删加 force')
+  let moved = 0
+  for (const x of all) {
+    for (const r of readNotes(x.k)) { trash(x.k, r, 'rm-folder'); moved++ }
+    try { fs.rmSync(dirOf(x.k), { recursive: true, force: true }) } catch {}
+  }
+  const gone = new Set(all.map((x) => x.k))
+  S.folders = S.folders.filter((x) => !gone.has(x.k))
+  save(S)
+  console.log('rmfolder ok ' + f.k + ' 删了 ' + all.length + ' 个分区，转存 ' + moved + ' 条记忆')
+}
+
 const RULES = {
   M: { keepAll: 1 },          // 主人、以及主人下面所有子分区：只由主人手删
   P: { keepAll: 1, doneExpire: 30 }, // 约定：全留，完成的过 30 天才清
@@ -335,6 +353,33 @@ function cmdMk(args) {
   save(S); fs.mkdirSync(dirOf(k), { recursive: true })
   console.log('ok ' + k + ' ' + (parent ? parent.name + ' / ' : '') + name)
 }
+function cmdDump() {
+  const out = S.folders.map(f => ({
+    k: f.k,
+    name: f.name,
+    parent: f.parent || '',
+    memories: readNotes(f.k).map(r => ({ id: r.id, ts: r.ts, flag: r.flag, text: r.text })),
+  }))
+  console.log(JSON.stringify({ home: HOME, folders: out }))
+}
+
+function cmdMove(args) {
+  const [id, from, to] = args
+  if (!id || !from || !to) return console.log('usage: move <id> <fromK> <toK>')
+  const src = S.folders.find(f => f.k === from)
+  const dst = S.folders.find(f => f.k === to)
+  if (!src || !dst) return console.log('no such folder')
+  const a = readNotes(from)
+  const i = a.findIndex(r => r.id === id)
+  if (i < 0) return console.log('no such memory')
+  const row = a.splice(i, 1)[0]
+  writeNotes(from, a)
+  const b = readNotes(to)
+  b.push(row)
+  writeNotes(to, b)
+  console.log('ok ' + id + ' ' + from + ' -> ' + to)
+}
+
 function cmdInit() {
   fs.mkdirSync(HOME, { recursive: true })
   fs.mkdirSync(TRASH, { recursive: true })
@@ -343,6 +388,6 @@ function cmdInit() {
   console.log('init ok ' + HOME)
 }
 const [cmd, ...args] = process.argv.slice(2)
-const MAP = { idx: cmdIdx, open: cmdOpen, add: (a) => cmdAdd(a, false), 'edit-mem': cmdEditMem, search: cmdSearch, tlog: cmdTLog, pending: cmdPending, approve: cmdApprove, reject: cmdReject, edit: cmdEdit, done: cmdDone, rm: cmdRm, gc: cmdGc, mk: cmdMk, init: cmdInit, backups: cmdBackups, restore: cmdRestore }
-if (!MAP[cmd]) { console.log('用法: idx|open|add|pending|approve|reject|edit|done|rm|gc|backups|restore|mk|init'); process.exit(1) }
+const MAP = { idx: cmdIdx, open: cmdOpen, add: (a) => cmdAdd(a, false), 'edit-mem': cmdEditMem, search: cmdSearch, tlog: cmdTLog, pending: cmdPending, approve: cmdApprove, reject: cmdReject, edit: cmdEdit, done: cmdDone, rm: cmdRm, gc: cmdGc, mk: cmdMk, rmfolder: cmdRmFolder, dump: cmdDump, move: cmdMove, init: cmdInit, backups: cmdBackups, restore: cmdRestore }
+if (!MAP[cmd]) { console.log('用法: idx|open|add|pending|approve|reject|edit|done|rm|rmfolder|gc|backups|restore|mk|init'); process.exit(1) }
 MAP[cmd](args)

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "deepseek-plastic-memory",
-  "version": "2.1.5",
+  "version": "3.3.8",
   "description": "DeepSeek 可塑性记忆：给 DSH 用的分区式长期记忆，带遗忘曲线",
   "type": "module",
   "main": "index.js",
@@ -19,7 +19,10 @@
     "cordis.patch.yml",
     "SKILL.md",
     "README.md",
-    "LICENSE"
+    "LICENSE",
+    "public",
+    "sky.js",
+    "identity.md"
   ],
   "dsh": {
     "bundle": {