th-memory-mcp 2.2.8 → 2.3.0

package/dist/tools/recall.js

@@ -1,6 +1,7 @@
 import { z } from "zod";
 import { db, buildFtsMatch, escapeLike, truncate, getAllEmbeddings, ok, err, } from "../db/index.js";
 import { embed, cosine, deserialize } from "../lib/embed.js";
+import { wrapMemoryReference } from "../lib/memory-format.js";
 export const recallInput = {
     topic: z.string().min(1).max(500).describe("Topic to recall from memory"),
     limit: z
@@ -121,7 +122,8 @@ export async function recallHandler(args) {
         if (parts.length === 0) {
             return ok(`no memory found for "${truncate(args.topic, 100)}"`);
         }
-        return ok(truncate(parts.join("\n\n"), RECALL_BUDGET));
+        // Batch B-2: recalled memory is reference data, not instructions.
+        return ok(wrapMemoryReference(truncate(parts.join("\n\n"), RECALL_BUDGET)));
     }
     catch (e) {
         return err(e instanceof Error ? e.message : String(e));

package/dist/tools/update_memory.js

@@ -2,6 +2,7 @@ import { z } from "zod";
 import { db, nowISO, ok, err, } from "../db/index.js";
 import { getMemoryById, createMemory, syncMemoryIndex, } from "../db/repositories/memories.js";
 import { supersede } from "../core/lifecycle-engine.js";
+import { isoDateTimeSchema, isIsoDateString } from "../lib/iso.js";
 export const updateMemoryInput = {
     id: z.number().int().describe("Memory id to update"),
     content: z
@@ -12,9 +13,12 @@ export const updateMemoryInput = {
     summary: z.string().max(2000).nullable().optional().describe("New summary"),
     importance: z.number().min(0).max(1).optional(),
     confidence: z.number().min(0).max(1).optional(),
+    validFrom: z
+        .union([isoDateTimeSchema, z.null()])
+        .optional()
+        .describe("ISO timestamp or null to clear"),
     validUntil: z
-        .string()
-        .nullable()
+        .union([isoDateTimeSchema, z.null()])
         .optional()
         .describe("ISO timestamp or null to clear"),
     metadata: z.unknown().optional().describe("New metadata object (replaces)"),
@@ -30,6 +34,24 @@ export function updateMemoryHandler(args) {
             return err(`memory ${args.id} not found`);
         if (mem.status === "deleted")
             return err("cannot update a deleted memory");
+        if (args.validFrom !== undefined &&
+            args.validFrom !== null &&
+            !isIsoDateString(args.validFrom)) {
+            return err(`validFrom must be a full ISO datetime with timezone offset (e.g. 2024-01-01T00:00:00.000Z), got '${args.validFrom}'`);
+        }
+        if (args.validUntil !== undefined &&
+            args.validUntil !== null &&
+            !isIsoDateString(args.validUntil)) {
+            return err(`validUntil must be a full ISO datetime with timezone offset (e.g. 2024-01-01T00:00:00.000Z), got '${args.validUntil}'`);
+        }
+        // Effective range covers new values and values mixed with the existing row.
+        const effectiveValidFrom = args.validFrom !== undefined ? args.validFrom : mem.valid_from;
+        const effectiveValidUntil = args.validUntil !== undefined ? args.validUntil : mem.valid_until;
+        if (effectiveValidFrom != null &&
+            effectiveValidUntil != null &&
+            new Date(effectiveValidFrom) > new Date(effectiveValidUntil)) {
+            return err(`validFrom (${effectiveValidFrom}) must not be later than validUntil (${effectiveValidUntil})`);
+        }
         const supersedeContent = args.content !== undefined && (args.supersede ?? true);
         if (supersedeContent) {
             const externalId = mem.user_id
@@ -48,7 +70,7 @@ export function updateMemoryHandler(args) {
                 projectId: mem.project_id,
                 sessionId: mem.session_id,
                 userId: externalId,
-                validFrom: mem.valid_from,
+                validFrom: args.validFrom !== undefined ? args.validFrom : mem.valid_from,
                 validUntil: args.validUntil !== undefined ? args.validUntil : mem.valid_until,
                 metadata: args.metadata !== undefined
                     ? args.metadata
@@ -73,6 +95,10 @@ export function updateMemoryHandler(args) {
             sets.push("confidence = ?");
             vals.push(args.confidence);
         }
+        if (args.validFrom !== undefined) {
+            sets.push("valid_from = ?");
+            vals.push(args.validFrom);
+        }
         if (args.validUntil !== undefined) {
             sets.push("valid_until = ?");
             vals.push(args.validUntil);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "th-memory-mcp",
-  "version": "2.2.8",
+  "version": "2.3.0",
   "mcpName": "io.github.worakorn-prince/th-memory-mcp",
   "description": "Adaptive Memory MCP server - SQLite-backed memory for OpenCode",
   "author": "worakorn-prince",
@@ -12,7 +12,8 @@
   "type": "module",
   "main": "dist/index.js",
   "bin": {
-    "th-memory-mcp": "dist/index.js"
+    "th-memory-mcp": "dist/index.js",
+    "th-memory": "dist/cli.js"
   },
   "files": [
     "dist",
@@ -29,9 +30,10 @@
     "prepublishOnly": "npm run version:sync && npm run build",
     "version:sync": "node scripts/sync-version.mjs",
     "version:check": "node scripts/sync-version.mjs --check",
+    "check:capture-sync": "node scripts/check-capture-sync.mjs",
     "start": "node dist/index.js",
     "inspect": "npx @modelcontextprotocol/inspector node dist/index.js",
-    "test": "npm run build && node --test test/*.test.mjs",
+    "test": "npm run build && node --test test/*.test.mjs test/smoke.mjs",
     "distill": "node dist/distill.js",
     "quickstart": "npm run build && node scripts/quickstart.mjs",
     "benchmark": "node repro/run.mjs",

package/dist/db.js

@@ -1,111 +0,0 @@
-import Database from "better-sqlite3";
-import { mkdirSync } from "node:fs";
-import { dirname } from "node:path";
-import { truncate } from "./lib/capture-core.js";
-import { serialize } from "./lib/embed.js";
-import { DEFAULT_DB_PATH } from "./lib/config.js";
-import { runMigrations } from "./migrations.js";
-export const DB_PATH = process.env.MEMORY_DB_PATH ?? DEFAULT_DB_PATH;
-function initDb() {
-    try {
-        mkdirSync(dirname(DB_PATH), { recursive: true });
-        return new Database(DB_PATH);
-    }
-    catch (e) {
-        const msg = e instanceof Error ? e.message : String(e);
-        console.error(`[th-memory-mcp] cannot open DB at ${DB_PATH}: ${msg}`);
-        process.exit(1);
-    }
-}
-export const db = initDb();
-db.pragma("journal_mode = WAL");
-db.pragma("busy_timeout = 5000");
-db.exec(`
-CREATE TABLE IF NOT EXISTS interactions (
-  id INTEGER PRIMARY KEY AUTOINCREMENT,
-  ts TEXT NOT NULL,
-  session_id TEXT,
-  kind TEXT NOT NULL,
-  content TEXT NOT NULL,
-  meta TEXT
-);
-
-CREATE TABLE IF NOT EXISTS preferences (
-  id INTEGER PRIMARY KEY AUTOINCREMENT,
-  category TEXT NOT NULL,
-  key TEXT NOT NULL,
-  value TEXT NOT NULL,
-  confidence REAL DEFAULT 0.5,
-  source TEXT DEFAULT 'explicit',
-  updated_at TEXT NOT NULL,
-  UNIQUE(category, key)
-);
-
-CREATE TABLE IF NOT EXISTS lessons (
-  id INTEGER PRIMARY KEY AUTOINCREMENT,
-  situation TEXT NOT NULL,
-  mistake TEXT NOT NULL,
-  correction TEXT NOT NULL,
-  created_at TEXT NOT NULL
-);
-
-CREATE TABLE IF NOT EXISTS profile (
-  section TEXT PRIMARY KEY,
-  content TEXT NOT NULL,
-  updated_at TEXT NOT NULL
-);
-
-CREATE VIRTUAL TABLE IF NOT EXISTS search_index USING fts5(
-  ref_table, ref_id, title, body
-);
-
-CREATE TABLE IF NOT EXISTS embeddings (
-  ref_table TEXT NOT NULL,
-  ref_id INTEGER NOT NULL,
-  vec BLOB NOT NULL,
-  PRIMARY KEY (ref_table, ref_id)
-);
-`);
-// v2 migration engine (non-destructive: only adds new tables + backfills from v1)
-runMigrations(db);
-export function nowISO() {
-    return new Date().toISOString();
-}
-// re-exported from lib/capture-core.js (single source of truth)
-export { truncate };
-export function escapeLike(text) {
-    return text.replace(/[\\%_]/g, (c) => "\\" + c);
-}
-export function buildFtsMatch(query) {
-    const tokens = query.trim().split(/\s+/).filter(Boolean).slice(0, 8);
-    return tokens.map((t) => `"${t.replace(/"/g, "")}"`).join(" ");
-}
-const insertSearchIndex = db.prepare("INSERT INTO search_index (ref_table, ref_id, title, body) VALUES (?, ?, ?, ?)");
-const deleteSearchIndex = db.prepare("DELETE FROM search_index WHERE ref_table = ? AND ref_id = ?");
-export function syncSearchIndex(refTable, refId, title, body) {
-    deleteSearchIndex.run(refTable, refId);
-    insertSearchIndex.run(refTable, refId, title, body);
-}
-export function removeSearchIndex(refTable, refId) {
-    deleteSearchIndex.run(refTable, refId);
-}
-// --- vector embeddings (lightweight local semantic search) ---
-const upsertEmbed = db.prepare(`INSERT INTO embeddings (ref_table, ref_id, vec) VALUES (?, ?, ?)
-   ON CONFLICT(ref_table, ref_id) DO UPDATE SET vec = excluded.vec`);
-const deleteEmbed = db.prepare("DELETE FROM embeddings WHERE ref_table = ? AND ref_id = ?");
-const allEmbeds = db.prepare("SELECT ref_table, ref_id, vec FROM embeddings");
-export function upsertEmbedding(refTable, refId, vec) {
-    upsertEmbed.run(refTable, refId, serialize(vec));
-}
-export function removeEmbedding(refTable, refId) {
-    deleteEmbed.run(refTable, refId);
-}
-export function getAllEmbeddings() {
-    return allEmbeds.all();
-}
-export function ok(text) {
-    return { content: [{ type: "text", text }] };
-}
-export function err(text) {
-    return ok(`error: ${truncate(text, 300)}`);
-}