claude-memory-layer 1.0.23 → 1.0.25

package/dist/core/index.js

@@ -124,6 +124,7 @@ var ConfigSchema = z.object({
   toolObservation: z.object({
     enabled: z.boolean().default(true),
     excludedTools: z.array(z.string()).default(["TodoWrite", "TodoRead"]),
+    minOutputLength: z.number().default(100),
     maxOutputLength: z.number().default(1e4),
     maxOutputLines: z.number().default(100),
     storeOnlyOnSuccess: z.boolean().default(false)
@@ -620,25 +621,7 @@ function parseEntityCanonicalKey(canonicalKey) {
 import { randomUUID } from "crypto";
 
 // src/core/db-wrapper.ts
-import duckdb from "duckdb";
-function convertBigInts(obj) {
-  if (obj === null || obj === void 0)
-    return obj;
-  if (typeof obj === "bigint")
-    return Number(obj);
-  if (obj instanceof Date)
-    return obj;
-  if (Array.isArray(obj))
-    return obj.map(convertBigInts);
-  if (typeof obj === "object") {
-    const result = {};
-    for (const [key, value] of Object.entries(obj)) {
-      result[key] = convertBigInts(value);
-    }
-    return result;
-  }
-  return obj;
-}
+import BetterSqlite3 from "better-sqlite3";
 function toDate(value) {
   if (value instanceof Date)
     return value;
@@ -648,59 +631,19 @@ function toDate(value) {
     return new Date(value);
   return new Date(String(value));
 }
-function createDatabase(path2, options) {
-  if (options?.readOnly) {
-    return new duckdb.Database(path2, { access_mode: "READ_ONLY" });
-  }
-  return new duckdb.Database(path2);
+function createDatabase(dbPath, options) {
+  return new BetterSqlite3(dbPath, { readonly: options?.readOnly });
 }
 function dbRun(db, sql, params = []) {
-  return new Promise((resolve, reject) => {
-    if (params.length === 0) {
-      db.run(sql, (err) => {
-        if (err)
-          reject(err);
-        else
-          resolve();
-      });
-    } else {
-      db.run(sql, ...params, (err) => {
-        if (err)
-          reject(err);
-        else
-          resolve();
-      });
-    }
-  });
+  db.prepare(sql).run(...params);
+  return Promise.resolve();
 }
 function dbAll(db, sql, params = []) {
-  return new Promise((resolve, reject) => {
-    if (params.length === 0) {
-      db.all(sql, (err, rows) => {
-        if (err)
-          reject(err);
-        else
-          resolve(convertBigInts(rows || []));
-      });
-    } else {
-      db.all(sql, ...params, (err, rows) => {
-        if (err)
-          reject(err);
-        else
-          resolve(convertBigInts(rows || []));
-      });
-    }
-  });
+  return Promise.resolve(db.prepare(sql).all(...params));
 }
 function dbClose(db) {
-  return new Promise((resolve, reject) => {
-    db.close((err) => {
-      if (err)
-        reject(err);
-      else
-        resolve();
-    });
-  });
+  db.close();
+  return Promise.resolve();
 }
 
 // src/core/event-store.ts
@@ -1902,6 +1845,29 @@ var SQLiteEventStore = class {
       };
     }
   }
+  /**
+   * Get session IDs that have events but no session_summary event.
+   * Used to backfill summaries for sessions that ended without Stop hook.
+   */
+  async getSessionsWithoutSummary(currentSessionId, limit = 5) {
+    await this.initialize();
+    const rows = sqliteAll(
+      this.db,
+      `SELECT DISTINCT e.session_id
+       FROM events e
+       WHERE e.session_id != ?
+         AND e.event_type != 'session_summary'
+         AND e.session_id NOT IN (
+           SELECT DISTINCT session_id FROM events WHERE event_type = 'session_summary'
+         )
+       GROUP BY e.session_id
+       HAVING COUNT(*) >= 3
+       ORDER BY MAX(e.timestamp) DESC
+       LIMIT ?`,
+      [currentSessionId, limit]
+    );
+    return rows.map((r) => r.session_id);
+  }
   /**
    * Get events by session ID
    */
@@ -2418,6 +2384,21 @@ var SQLiteEventStore = class {
       [id, eventId, sessionId, score, query.slice(0, 100)]
     );
   }
+  /**
+   * Get session IDs that have unevaluated retrievals (measured_at IS NULL).
+   * Excludes the current session. Used to backfill sessions that ended without Stop hook.
+   */
+  async getUnevaluatedSessions(currentSessionId, limit = 5) {
+    await this.initialize();
+    const rows = sqliteAll(
+      this.db,
+      `SELECT DISTINCT session_id FROM memory_helpfulness
+       WHERE measured_at IS NULL AND session_id != ?
+       ORDER BY created_at DESC LIMIT ?`,
+      [currentSessionId, limit]
+    );
+    return rows.map((r) => r.session_id);
+  }
   /**
    * Evaluate helpfulness for all retrievals in a session
    * Called at session end - uses behavioral signals to compute score
@@ -3997,7 +3978,7 @@ var VectorStore = class {
 
 // src/core/embedder.ts
 import { pipeline } from "@huggingface/transformers";
-var Embedder = class {
+var Embedder = class _Embedder {
   pipeline = null;
   modelName;
   activeModelName;
@@ -4028,6 +4009,11 @@ var Embedder = class {
       this.initialized = true;
     }
   }
+  // ~4 chars per token; 512 tokens * 4 = 2048, use 2000 to be safe
+  static MAX_CHARS = 2e3;
+  truncate(text) {
+    return text.length > _Embedder.MAX_CHARS ? text.slice(0, _Embedder.MAX_CHARS) : text;
+  }
   /**
    * Generate embedding for a single text
    */
@@ -4036,10 +4022,11 @@ var Embedder = class {
     if (!this.pipeline) {
       throw new Error("Embedding pipeline not initialized");
     }
-    const output = await this.pipeline(text, {
+    const output = await this.pipeline(this.truncate(text), {
       pooling: "mean",
       normalize: true,
-      truncation: true
+      truncation: true,
+      max_length: 512
     });
     const vector = Array.from(output.data);
     return {
@@ -4061,10 +4048,11 @@ var Embedder = class {
     for (let i = 0; i < texts.length; i += batchSize) {
       const batch = texts.slice(i, i + batchSize);
       for (const text of batch) {
-        const output = await this.pipeline(text, {
+        const output = await this.pipeline(this.truncate(text), {
           pooling: "mean",
           normalize: true,
-          truncation: true
+          truncation: true,
+          max_length: 512
         });
         const vector = Array.from(output.data);
         results.push({