@memrosetta/core 0.2.20 → 0.2.21

package/dist/index.d.ts

@@ -44,6 +44,8 @@ interface MemoryRow {
     readonly access_count: number | null;
     readonly last_accessed_at: string | null;
     readonly compressed_from: string | null;
+    readonly use_count: number | null;
+    readonly success_count: number | null;
 }
 declare function rowToMemory(row: MemoryRow): Memory;
 declare function serializeEmbedding(embedding: readonly number[] | Float32Array): Buffer;
@@ -242,6 +244,7 @@ declare class SqliteMemoryEngine implements IMemoryEngine {
     compress(userId: string): Promise<CompressResult>;
     maintain(userId: string): Promise<MaintenanceResult>;
     setTier(memoryId: string, tier: MemoryTier): Promise<void>;
+    feedback(memoryId: string, helpful: boolean): Promise<void>;
     quality(userId: string): Promise<MemoryQuality>;
     /**
      * Check the newly stored memory against existing similar memories

package/dist/index.js

@@ -24,7 +24,9 @@ CREATE TABLE memories (
   activation_score REAL DEFAULT 1.0,
   access_count     INTEGER DEFAULT 0,
   last_accessed_at TEXT,
-  compressed_from  TEXT
+  compressed_from  TEXT,
+  use_count        INTEGER DEFAULT 0,
+  success_count    INTEGER DEFAULT 0
 );
 
 CREATE INDEX idx_memories_user_id ON memories(user_id);
@@ -72,6 +74,10 @@ CREATE TRIGGER memories_au AFTER UPDATE ON memories BEGIN
   INSERT INTO memories_fts(rowid, content, keywords) VALUES (new.id, new.content, new.keywords);
 END;
 `;
+var SCHEMA_V5 = `
+ALTER TABLE memories ADD COLUMN use_count INTEGER DEFAULT 0;
+ALTER TABLE memories ADD COLUMN success_count INTEGER DEFAULT 0;
+`;
 function schemaV2(dim) {
   return `CREATE VIRTUAL TABLE IF NOT EXISTS vec_memories USING vec0(embedding float[${dim}]);`;
 }
@@ -106,7 +112,7 @@ function ensureSchema(db, options) {
       db.exec(schemaV2(dim));
       version = 2;
     }
-    version = 4;
+    version = 5;
     db.prepare("INSERT INTO schema_version (version, embedding_dimension) VALUES (?, ?)").run(version, dim);
     return;
   }
@@ -132,6 +138,12 @@ function ensureSchema(db, options) {
     }
     db.prepare("UPDATE schema_version SET version = ?").run(4);
   }
+  if (currentVersion < 5) {
+    if (currentVersion >= 1) {
+      db.exec(SCHEMA_V5);
+    }
+    db.prepare("UPDATE schema_version SET version = ?").run(5);
+  }
   if (options?.vectorEnabled) {
     const hasVecTable = db.prepare(
       "SELECT name FROM sqlite_master WHERE type='table' AND name='vec_memories'"
@@ -213,7 +225,9 @@ function rowToMemory(row) {
     activationScore: row.activation_score ?? 1,
     accessCount: row.access_count ?? 0,
     ...row.last_accessed_at != null ? { lastAccessedAt: row.last_accessed_at } : {},
-    ...row.compressed_from != null ? { compressedFrom: row.compressed_from } : {}
+    ...row.compressed_from != null ? { compressedFrom: row.compressed_from } : {},
+    useCount: row.use_count ?? 0,
+    successCount: row.success_count ?? 0
   };
 }
 function deserializeEmbedding(buf) {
@@ -229,8 +243,8 @@ function serializeEmbedding(embedding) {
 function createPreparedStatements(db) {
   return {
     insertMemory: db.prepare(`
-      INSERT INTO memories (memory_id, user_id, namespace, memory_type, content, raw_text, document_date, learned_at, source_id, confidence, salience, is_latest, embedding, keywords, event_date_start, event_date_end, invalidated_at, tier, activation_score, access_count, last_accessed_at, compressed_from)
-      VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+      INSERT INTO memories (memory_id, user_id, namespace, memory_type, content, raw_text, document_date, learned_at, source_id, confidence, salience, is_latest, embedding, keywords, event_date_start, event_date_end, invalidated_at, tier, activation_score, access_count, last_accessed_at, compressed_from, use_count, success_count)
+      VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
     `),
     getById: db.prepare("SELECT * FROM memories WHERE id = ?"),
     getByMemoryId: db.prepare("SELECT * FROM memories WHERE memory_id = ?"),
@@ -269,8 +283,12 @@ function storeMemory(db, stmts, input) {
     // access_count
     null,
     // last_accessed_at
-    null
+    null,
     // compressed_from
+    0,
+    // use_count
+    0
+    // success_count
   );
   const row = stmts.getByMemoryId.get(memoryId);
   return rowToMemory(row);
@@ -312,8 +330,12 @@ async function storeMemoryAsync(db, stmts, input, embedder) {
     // access_count
     null,
     // last_accessed_at
-    null
+    null,
     // compressed_from
+    0,
+    // use_count
+    0
+    // success_count
   );
   if (embeddingVec && info.lastInsertRowid) {
     const rowid = Number(info.lastInsertRowid);
@@ -379,8 +401,12 @@ async function storeBatchAsync(db, stmts, inputs, embedder) {
         // access_count
         null,
         // last_accessed_at
-        null
+        null,
         // compressed_from
+        0,
+        // use_count
+        0
+        // success_count
       );
       if (info.lastInsertRowid) {
         db.prepare("INSERT INTO vec_memories(rowid, embedding) VALUES (CAST(? AS INTEGER), ?)").run(
@@ -619,7 +645,7 @@ function buildSearchSql(query) {
       stateConditions.push("(m.is_latest = 1 AND m.invalidated_at IS NULL)");
     }
     if (filters.states.includes("superseded")) {
-      stateConditions.push("(m.is_latest = 0)");
+      stateConditions.push("(m.is_latest = 0 AND m.invalidated_at IS NULL)");
     }
     if (filters.states.includes("invalidated")) {
       stateConditions.push("(m.invalidated_at IS NOT NULL)");
@@ -703,7 +729,7 @@ function bruteForceVectorSearch(db, queryVec, userId, limit, filters) {
       stateConditions.push("(is_latest = 1 AND invalidated_at IS NULL)");
     }
     if (filters.states.includes("superseded")) {
-      stateConditions.push("(is_latest = 0)");
+      stateConditions.push("(is_latest = 0 AND invalidated_at IS NULL)");
     }
     if (filters.states.includes("invalidated")) {
       stateConditions.push("(invalidated_at IS NOT NULL)");
@@ -791,7 +817,7 @@ function vectorSearch(db, queryVec, userId, limit, filters, useVecTable = true)
       stateConditions.push("(is_latest = 1 AND invalidated_at IS NULL)");
     }
     if (filters.states.includes("superseded")) {
-      stateConditions.push("(is_latest = 0)");
+      stateConditions.push("(is_latest = 0 AND invalidated_at IS NULL)");
     }
     if (filters.states.includes("invalidated")) {
       stateConditions.push("(invalidated_at IS NOT NULL)");
@@ -1209,10 +1235,22 @@ var SqliteMemoryEngine = class {
     } else {
       memories = storeBatchInTransaction(this.db, this.stmts, inputs);
     }
-    if (this.options.contradictionDetector && this.options.embedder && memories.length <= 50) {
+    if (memories.length <= 50) {
       for (const memory of memories) {
+        if (this.options.embedder && this.options.contradictionDetector) {
+          try {
+            await this.checkContradictions(memory);
+          } catch {
+          }
+        }
+        if (this.options.embedder) {
+          try {
+            await this.checkDuplicates(memory);
+          } catch {
+          }
+        }
         try {
-          await this.checkContradictions(memory);
+          await this.autoRelate(memory);
         } catch {
         }
       }
@@ -1338,6 +1376,7 @@ var SqliteMemoryEngine = class {
     const coldMemories = db.prepare(`
       SELECT * FROM memories
       WHERE user_id = ? AND tier = 'cold' AND activation_score < 0.1 AND is_latest = 1
+      AND invalidated_at IS NULL
       ORDER BY namespace, learned_at
     `).all(userId);
     if (coldMemories.length === 0) return { compressed: 0, removed: 0 };
@@ -1398,8 +1437,12 @@ var SqliteMemoryEngine = class {
           // access_count
           null,
           // last_accessed_at
-          rows[0].memory_id
+          rows[0].memory_id,
           // compressed_from
+          0,
+          // use_count
+          0
+          // success_count
         );
         compressed++;
         for (const row of rows) {
@@ -1467,6 +1510,29 @@ var SqliteMemoryEngine = class {
     this.ensureInitialized();
     this.db.prepare("UPDATE memories SET tier = ? WHERE memory_id = ?").run(tier, memoryId);
   }
+  async feedback(memoryId, helpful) {
+    this.ensureInitialized();
+    const db = this.db;
+    db.transaction(() => {
+      if (helpful) {
+        db.prepare(
+          "UPDATE memories SET use_count = use_count + 1, success_count = success_count + 1 WHERE memory_id = ?"
+        ).run(memoryId);
+      } else {
+        db.prepare(
+          "UPDATE memories SET use_count = use_count + 1 WHERE memory_id = ?"
+        ).run(memoryId);
+      }
+      const row = db.prepare(
+        "SELECT salience, use_count, success_count FROM memories WHERE memory_id = ?"
+      ).get(memoryId);
+      if (row && row.use_count > 0) {
+        const successRate = row.success_count / row.use_count;
+        const newSalience = Math.min(1, Math.max(0.1, 0.5 + 0.5 * successRate));
+        db.prepare("UPDATE memories SET salience = ? WHERE memory_id = ?").run(newSalience, memoryId);
+      }
+    })();
+  }
   async quality(userId) {
     this.ensureInitialized();
     const db = this.db;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@memrosetta/core",
-  "version": "0.2.20",
+  "version": "0.2.21",
   "type": "module",
   "main": "./dist/index.js",
   "types": "./dist/index.d.ts",
@@ -20,10 +20,10 @@
     "better-sqlite3": "^11.0.0",
     "nanoid": "^5.0.0",
     "sqlite-vec": "^0.1.7",
-    "@memrosetta/types": "0.2.20"
+    "@memrosetta/types": "0.2.21"
   },
   "optionalDependencies": {
-    "@memrosetta/embeddings": "0.2.20"
+    "@memrosetta/embeddings": "0.2.21"
   },
   "devDependencies": {
     "@types/better-sqlite3": "^7.6.0",