claude-memory-layer 1.0.24 → 1.0.26

package/dist/core/index.js

@@ -621,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;
@@ -649,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
@@ -1903,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
    */
@@ -2489,7 +2454,8 @@ var SQLiteEventStore = class {
         }
       }
       const retrievalScore = retrieval.retrieval_score || 0;
-      const helpfulnessScore = 0.3 * Math.min(retrievalScore, 1) + 0.25 * (sessionContinued ? 1 : 0) + 0.25 * toolSuccessRatio + 0.2 * (wasReasked ? 0 : 1);
+      const promptNorm = Math.min(promptCountAfter / 2, 1);
+      const helpfulnessScore = 0.4 * Math.min(retrievalScore, 1) + 0.3 * promptNorm + 0.2 * toolSuccessRatio + 0.1 * (sessionContinued ? 1 : 0);
       sqliteRun(
         this.db,
         `UPDATE memory_helpfulness