claude-memory-layer 1.0.0 → 1.0.2

package/dist/services/memory-service.js

@@ -12,7 +12,6 @@ import * as fs from "fs";
 import * as crypto2 from "crypto";
 
 // src/core/event-store.ts
-import { Database } from "duckdb";
 import { randomUUID } from "crypto";
 
 // src/core/canonical-key.ts
@@ -38,11 +37,92 @@ function makeDedupeKey(content, sessionId) {
   return `${sessionId}:${contentHash}`;
 }
 
+// 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;
+}
+function toDate(value) {
+  if (value instanceof Date)
+    return value;
+  if (typeof value === "string")
+    return new Date(value);
+  if (typeof value === "number")
+    return new Date(value);
+  return new Date(String(value));
+}
+function createDatabase(path2) {
+  return new duckdb.Database(path2);
+}
+function dbRun(db, sql, params = []) {
+  return new Promise((resolve2, reject) => {
+    if (params.length === 0) {
+      db.run(sql, (err) => {
+        if (err)
+          reject(err);
+        else
+          resolve2();
+      });
+    } else {
+      db.run(sql, ...params, (err) => {
+        if (err)
+          reject(err);
+        else
+          resolve2();
+      });
+    }
+  });
+}
+function dbAll(db, sql, params = []) {
+  return new Promise((resolve2, reject) => {
+    if (params.length === 0) {
+      db.all(sql, (err, rows) => {
+        if (err)
+          reject(err);
+        else
+          resolve2(convertBigInts(rows || []));
+      });
+    } else {
+      db.all(sql, ...params, (err, rows) => {
+        if (err)
+          reject(err);
+        else
+          resolve2(convertBigInts(rows || []));
+      });
+    }
+  });
+}
+function dbClose(db) {
+  return new Promise((resolve2, reject) => {
+    db.close((err) => {
+      if (err)
+        reject(err);
+      else
+        resolve2();
+    });
+  });
+}
+
 // src/core/event-store.ts
 var EventStore = class {
   constructor(dbPath) {
     this.dbPath = dbPath;
-    this.db = new Database(dbPath);
+    this.db = createDatabase(dbPath);
   }
   db;
   initialized = false;
@@ -52,7 +132,7 @@ var EventStore = class {
   async initialize() {
     if (this.initialized)
       return;
-    await this.db.run(`
+    await dbRun(this.db, `
       CREATE TABLE IF NOT EXISTS events (
         id VARCHAR PRIMARY KEY,
         event_type VARCHAR NOT NULL,
@@ -64,14 +144,14 @@ var EventStore = class {
         metadata JSON
       )
     `);
-    await this.db.run(`
+    await dbRun(this.db, `
       CREATE TABLE IF NOT EXISTS event_dedup (
         dedupe_key VARCHAR PRIMARY KEY,
         event_id VARCHAR NOT NULL,
         created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
       )
     `);
-    await this.db.run(`
+    await dbRun(this.db, `
       CREATE TABLE IF NOT EXISTS sessions (
         id VARCHAR PRIMARY KEY,
         started_at TIMESTAMP NOT NULL,
@@ -81,7 +161,7 @@ var EventStore = class {
         tags JSON
       )
     `);
-    await this.db.run(`
+    await dbRun(this.db, `
       CREATE TABLE IF NOT EXISTS insights (
         id VARCHAR PRIMARY KEY,
         insight_type VARCHAR NOT NULL,
@@ -93,7 +173,7 @@ var EventStore = class {
         last_updated TIMESTAMP
       )
     `);
-    await this.db.run(`
+    await dbRun(this.db, `
       CREATE TABLE IF NOT EXISTS embedding_outbox (
         id VARCHAR PRIMARY KEY,
         event_id VARCHAR NOT NULL,
@@ -105,7 +185,7 @@ var EventStore = class {
         error_message TEXT
       )
     `);
-    await this.db.run(`
+    await dbRun(this.db, `
       CREATE TABLE IF NOT EXISTS projection_offsets (
         projection_name VARCHAR PRIMARY KEY,
         last_event_id VARCHAR,
@@ -113,14 +193,14 @@ var EventStore = class {
         updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
       )
     `);
-    await this.db.run(`
+    await dbRun(this.db, `
       CREATE TABLE IF NOT EXISTS memory_levels (
         event_id VARCHAR PRIMARY KEY,
         level VARCHAR NOT NULL DEFAULT 'L0',
         promoted_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
       )
     `);
-    await this.db.run(`
+    await dbRun(this.db, `
       CREATE TABLE IF NOT EXISTS entries (
         entry_id VARCHAR PRIMARY KEY,
         created_ts TIMESTAMP NOT NULL,
@@ -136,7 +216,7 @@ var EventStore = class {
         created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
       )
     `);
-    await this.db.run(`
+    await dbRun(this.db, `
       CREATE TABLE IF NOT EXISTS entities (
         entity_id VARCHAR PRIMARY KEY,
         entity_type VARCHAR NOT NULL,
@@ -151,7 +231,7 @@ var EventStore = class {
         updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
       )
     `);
-    await this.db.run(`
+    await dbRun(this.db, `
       CREATE TABLE IF NOT EXISTS entity_aliases (
         entity_type VARCHAR NOT NULL,
         canonical_key VARCHAR NOT NULL,
@@ -161,7 +241,7 @@ var EventStore = class {
         PRIMARY KEY(entity_type, canonical_key)
       )
     `);
-    await this.db.run(`
+    await dbRun(this.db, `
       CREATE TABLE IF NOT EXISTS edges (
         edge_id VARCHAR PRIMARY KEY,
         src_type VARCHAR NOT NULL,
@@ -173,7 +253,7 @@ var EventStore = class {
         created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
       )
     `);
-    await this.db.run(`
+    await dbRun(this.db, `
       CREATE TABLE IF NOT EXISTS vector_outbox (
         job_id VARCHAR PRIMARY KEY,
         item_kind VARCHAR NOT NULL,
@@ -187,7 +267,7 @@ var EventStore = class {
         UNIQUE(item_kind, item_id, embedding_version)
       )
     `);
-    await this.db.run(`
+    await dbRun(this.db, `
       CREATE TABLE IF NOT EXISTS build_runs (
         build_id VARCHAR PRIMARY KEY,
         started_at TIMESTAMP NOT NULL,
@@ -202,7 +282,7 @@ var EventStore = class {
         error VARCHAR
       )
     `);
-    await this.db.run(`
+    await dbRun(this.db, `
       CREATE TABLE IF NOT EXISTS pipeline_metrics (
         id VARCHAR PRIMARY KEY,
         ts TIMESTAMP NOT NULL,
@@ -213,7 +293,7 @@ var EventStore = class {
         session_id VARCHAR
       )
     `);
-    await this.db.run(`
+    await dbRun(this.db, `
       CREATE TABLE IF NOT EXISTS working_set (
         id VARCHAR PRIMARY KEY,
         event_id VARCHAR NOT NULL,
@@ -223,7 +303,7 @@ var EventStore = class {
         expires_at TIMESTAMP
       )
     `);
-    await this.db.run(`
+    await dbRun(this.db, `
       CREATE TABLE IF NOT EXISTS consolidated_memories (
         memory_id VARCHAR PRIMARY KEY,
         summary TEXT NOT NULL,
@@ -235,7 +315,7 @@ var EventStore = class {
         access_count INTEGER DEFAULT 0
       )
     `);
-    await this.db.run(`
+    await dbRun(this.db, `
       CREATE TABLE IF NOT EXISTS continuity_log (
         log_id VARCHAR PRIMARY KEY,
         from_context_id VARCHAR,
@@ -245,26 +325,26 @@ var EventStore = class {
         created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
       )
     `);
-    await this.db.run(`
+    await dbRun(this.db, `
       CREATE TABLE IF NOT EXISTS endless_config (
         key VARCHAR PRIMARY KEY,
         value JSON,
         updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
       )
     `);
-    await this.db.run(`CREATE INDEX IF NOT EXISTS idx_entries_type ON entries(entry_type)`);
-    await this.db.run(`CREATE INDEX IF NOT EXISTS idx_entries_stage ON entries(stage)`);
-    await this.db.run(`CREATE INDEX IF NOT EXISTS idx_entries_canonical ON entries(canonical_key)`);
-    await this.db.run(`CREATE INDEX IF NOT EXISTS idx_entities_type_key ON entities(entity_type, canonical_key)`);
-    await this.db.run(`CREATE INDEX IF NOT EXISTS idx_entities_status ON entities(status)`);
-    await this.db.run(`CREATE INDEX IF NOT EXISTS idx_edges_src ON edges(src_id, rel_type)`);
-    await this.db.run(`CREATE INDEX IF NOT EXISTS idx_edges_dst ON edges(dst_id, rel_type)`);
-    await this.db.run(`CREATE INDEX IF NOT EXISTS idx_edges_rel ON edges(rel_type)`);
-    await this.db.run(`CREATE INDEX IF NOT EXISTS idx_outbox_status ON vector_outbox(status)`);
-    await this.db.run(`CREATE INDEX IF NOT EXISTS idx_working_set_expires ON working_set(expires_at)`);
-    await this.db.run(`CREATE INDEX IF NOT EXISTS idx_working_set_relevance ON working_set(relevance_score DESC)`);
-    await this.db.run(`CREATE INDEX IF NOT EXISTS idx_consolidated_confidence ON consolidated_memories(confidence DESC)`);
-    await this.db.run(`CREATE INDEX IF NOT EXISTS idx_continuity_created ON continuity_log(created_at)`);
+    await dbRun(this.db, `CREATE INDEX IF NOT EXISTS idx_entries_type ON entries(entry_type)`);
+    await dbRun(this.db, `CREATE INDEX IF NOT EXISTS idx_entries_stage ON entries(stage)`);
+    await dbRun(this.db, `CREATE INDEX IF NOT EXISTS idx_entries_canonical ON entries(canonical_key)`);
+    await dbRun(this.db, `CREATE INDEX IF NOT EXISTS idx_entities_type_key ON entities(entity_type, canonical_key)`);
+    await dbRun(this.db, `CREATE INDEX IF NOT EXISTS idx_entities_status ON entities(status)`);
+    await dbRun(this.db, `CREATE INDEX IF NOT EXISTS idx_edges_src ON edges(src_id, rel_type)`);
+    await dbRun(this.db, `CREATE INDEX IF NOT EXISTS idx_edges_dst ON edges(dst_id, rel_type)`);
+    await dbRun(this.db, `CREATE INDEX IF NOT EXISTS idx_edges_rel ON edges(rel_type)`);
+    await dbRun(this.db, `CREATE INDEX IF NOT EXISTS idx_outbox_status ON vector_outbox(status)`);
+    await dbRun(this.db, `CREATE INDEX IF NOT EXISTS idx_working_set_expires ON working_set(expires_at)`);
+    await dbRun(this.db, `CREATE INDEX IF NOT EXISTS idx_working_set_relevance ON working_set(relevance_score DESC)`);
+    await dbRun(this.db, `CREATE INDEX IF NOT EXISTS idx_consolidated_confidence ON consolidated_memories(confidence DESC)`);
+    await dbRun(this.db, `CREATE INDEX IF NOT EXISTS idx_continuity_created ON continuity_log(created_at)`);
     this.initialized = true;
   }
   /**
@@ -275,7 +355,8 @@ var EventStore = class {
     await this.initialize();
     const canonicalKey = makeCanonicalKey(input.content);
     const dedupeKey = makeDedupeKey(input.content, input.sessionId);
-    const existing = await this.db.all(
+    const existing = await dbAll(
+      this.db,
       `SELECT event_id FROM event_dedup WHERE dedupe_key = ?`,
       [dedupeKey]
     );
@@ -289,7 +370,8 @@ var EventStore = class {
     const id = randomUUID();
     const timestamp = input.timestamp.toISOString();
     try {
-      await this.db.run(
+      await dbRun(
+        this.db,
         `INSERT INTO events (id, event_type, session_id, timestamp, content, canonical_key, dedupe_key, metadata)
          VALUES (?, ?, ?, ?, ?, ?, ?, ?)`,
         [
@@ -303,11 +385,13 @@ var EventStore = class {
           JSON.stringify(input.metadata || {})
         ]
       );
-      await this.db.run(
+      await dbRun(
+        this.db,
         `INSERT INTO event_dedup (dedupe_key, event_id) VALUES (?, ?)`,
         [dedupeKey, id]
       );
-      await this.db.run(
+      await dbRun(
+        this.db,
         `INSERT INTO memory_levels (event_id, level) VALUES (?, 'L0')`,
         [id]
       );
@@ -324,7 +408,8 @@ var EventStore = class {
    */
   async getSessionEvents(sessionId) {
     await this.initialize();
-    const rows = await this.db.all(
+    const rows = await dbAll(
+      this.db,
       `SELECT * FROM events WHERE session_id = ? ORDER BY timestamp ASC`,
       [sessionId]
     );
@@ -335,7 +420,8 @@ var EventStore = class {
    */
   async getRecentEvents(limit = 100) {
     await this.initialize();
-    const rows = await this.db.all(
+    const rows = await dbAll(
+      this.db,
       `SELECT * FROM events ORDER BY timestamp DESC LIMIT ?`,
       [limit]
     );
@@ -346,7 +432,8 @@ var EventStore = class {
    */
   async getEvent(id) {
     await this.initialize();
-    const rows = await this.db.all(
+    const rows = await dbAll(
+      this.db,
       `SELECT * FROM events WHERE id = ?`,
       [id]
     );
@@ -359,12 +446,14 @@ var EventStore = class {
    */
   async upsertSession(session) {
     await this.initialize();
-    const existing = await this.db.all(
+    const existing = await dbAll(
+      this.db,
       `SELECT id FROM sessions WHERE id = ?`,
       [session.id]
     );
     if (existing.length === 0) {
-      await this.db.run(
+      await dbRun(
+        this.db,
         `INSERT INTO sessions (id, started_at, project_path, tags)
          VALUES (?, ?, ?, ?)`,
         [
@@ -391,7 +480,8 @@ var EventStore = class {
       }
       if (updates.length > 0) {
         values.push(session.id);
-        await this.db.run(
+        await dbRun(
+          this.db,
           `UPDATE sessions SET ${updates.join(", ")} WHERE id = ?`,
           values
         );
@@ -403,7 +493,8 @@ var EventStore = class {
    */
   async getSession(id) {
     await this.initialize();
-    const rows = await this.db.all(
+    const rows = await dbAll(
+      this.db,
       `SELECT * FROM sessions WHERE id = ?`,
       [id]
     );
@@ -412,8 +503,8 @@ var EventStore = class {
     const row = rows[0];
     return {
       id: row.id,
-      startedAt: new Date(row.started_at),
-      endedAt: row.ended_at ? new Date(row.ended_at) : void 0,
+      startedAt: toDate(row.started_at),
+      endedAt: row.ended_at ? toDate(row.ended_at) : void 0,
       projectPath: row.project_path,
       summary: row.summary,
       tags: row.tags ? JSON.parse(row.tags) : void 0
@@ -425,7 +516,8 @@ var EventStore = class {
   async enqueueForEmbedding(eventId, content) {
     await this.initialize();
     const id = randomUUID();
-    await this.db.run(
+    await dbRun(
+      this.db,
       `INSERT INTO embedding_outbox (id, event_id, content, status, retry_count)
        VALUES (?, ?, ?, 'pending', 0)`,
       [id, eventId, content]
@@ -437,25 +529,30 @@ var EventStore = class {
    */
   async getPendingOutboxItems(limit = 32) {
     await this.initialize();
-    const rows = await this.db.all(
-      `UPDATE embedding_outbox
-       SET status = 'processing'
-       WHERE id IN (
-         SELECT id FROM embedding_outbox
-         WHERE status = 'pending'
-         ORDER BY created_at
-         LIMIT ?
-       )
-       RETURNING *`,
+    const pending = await dbAll(
+      this.db,
+      `SELECT * FROM embedding_outbox
+       WHERE status = 'pending'
+       ORDER BY created_at
+       LIMIT ?`,
       [limit]
     );
-    return rows.map((row) => ({
+    if (pending.length === 0)
+      return [];
+    const ids = pending.map((r) => r.id);
+    const placeholders = ids.map(() => "?").join(",");
+    await dbRun(
+      this.db,
+      `UPDATE embedding_outbox SET status = 'processing' WHERE id IN (${placeholders})`,
+      ids
+    );
+    return pending.map((row) => ({
       id: row.id,
       eventId: row.event_id,
       content: row.content,
-      status: row.status,
+      status: "processing",
       retryCount: row.retry_count,
-      createdAt: new Date(row.created_at),
+      createdAt: toDate(row.created_at),
       errorMessage: row.error_message
     }));
   }
@@ -466,7 +563,8 @@ var EventStore = class {
     if (ids.length === 0)
       return;
     const placeholders = ids.map(() => "?").join(",");
-    await this.db.run(
+    await dbRun(
+      this.db,
       `DELETE FROM embedding_outbox WHERE id IN (${placeholders})`,
       ids
     );
@@ -478,7 +576,8 @@ var EventStore = class {
     if (ids.length === 0)
       return;
     const placeholders = ids.map(() => "?").join(",");
-    await this.db.run(
+    await dbRun(
+      this.db,
       `UPDATE embedding_outbox
        SET status = CASE WHEN retry_count >= 3 THEN 'failed' ELSE 'pending' END,
            retry_count = retry_count + 1,
@@ -492,7 +591,8 @@ var EventStore = class {
    */
   async updateMemoryLevel(eventId, level) {
     await this.initialize();
-    await this.db.run(
+    await dbRun(
+      this.db,
       `UPDATE memory_levels SET level = ?, promoted_at = CURRENT_TIMESTAMP WHERE event_id = ?`,
       [level, eventId]
     );
@@ -502,7 +602,8 @@ var EventStore = class {
    */
   async getLevelStats() {
     await this.initialize();
-    const rows = await this.db.all(
+    const rows = await dbAll(
+      this.db,
       `SELECT level, COUNT(*) as count FROM memory_levels GROUP BY level`
     );
     return rows;
@@ -521,7 +622,8 @@ var EventStore = class {
    */
   async getEndlessConfig(key) {
     await this.initialize();
-    const rows = await this.db.all(
+    const rows = await dbAll(
+      this.db,
       `SELECT value FROM endless_config WHERE key = ?`,
       [key]
     );
@@ -534,7 +636,8 @@ var EventStore = class {
    */
   async setEndlessConfig(key, value) {
     await this.initialize();
-    await this.db.run(
+    await dbRun(
+      this.db,
       `INSERT OR REPLACE INTO endless_config (key, value, updated_at)
        VALUES (?, ?, CURRENT_TIMESTAMP)`,
       [key, JSON.stringify(value)]
@@ -545,13 +648,14 @@ var EventStore = class {
    */
   async getAllSessions() {
     await this.initialize();
-    const rows = await this.db.all(
+    const rows = await dbAll(
+      this.db,
       `SELECT * FROM sessions ORDER BY started_at DESC`
     );
     return rows.map((row) => ({
       id: row.id,
-      startedAt: new Date(row.started_at),
-      endedAt: row.ended_at ? new Date(row.ended_at) : void 0,
+      startedAt: toDate(row.started_at),
+      endedAt: row.ended_at ? toDate(row.ended_at) : void 0,
       projectPath: row.project_path,
       summary: row.summary,
       tags: row.tags ? JSON.parse(row.tags) : void 0
@@ -561,7 +665,7 @@ var EventStore = class {
    * Close database connection
    */
   async close() {
-    await this.db.close();
+    await dbClose(this.db);
   }
   /**
    * Convert database row to MemoryEvent
@@ -571,7 +675,7 @@ var EventStore = class {
       id: row.id,
       eventType: row.event_type,
       sessionId: row.session_id,
-      timestamp: new Date(row.timestamp),
+      timestamp: toDate(row.timestamp),
       content: row.content,
       canonicalKey: row.canonical_key,
       dedupeKey: row.dedupe_key,
@@ -664,23 +768,28 @@ var VectorStore = class {
       return [];
     }
     const { limit = 5, minScore = 0.7, sessionId } = options;
-    let query = this.table.search(queryVector).limit(limit * 2);
+    let query = this.table.search(queryVector).distanceType("cosine").limit(limit * 2);
     if (sessionId) {
       query = query.where(`sessionId = '${sessionId}'`);
     }
     const results = await query.toArray();
     return results.filter((r) => {
-      const score = 1 - (r._distance || 0);
+      const distance = r._distance || 0;
+      const score = 1 - distance / 2;
       return score >= minScore;
-    }).slice(0, limit).map((r) => ({
-      id: r.id,
-      eventId: r.eventId,
-      content: r.content,
-      score: 1 - (r._distance || 0),
-      sessionId: r.sessionId,
-      eventType: r.eventType,
-      timestamp: r.timestamp
-    }));
+    }).slice(0, limit).map((r) => {
+      const distance = r._distance || 0;
+      const score = 1 - distance / 2;
+      return {
+        id: r.id,
+        eventId: r.eventId,
+        content: r.content,
+        score,
+        sessionId: r.sessionId,
+        eventType: r.eventType,
+        timestamp: r.timestamp
+      };
+    });
   }
   /**
    * Delete vector by event ID
@@ -1506,7 +1615,8 @@ var WorkingSetStore = class {
     const expiresAt = new Date(
       Date.now() + this.config.workingSet.timeWindowHours * 60 * 60 * 1e3
     );
-    await this.db.run(
+    await dbRun(
+      this.db,
       `INSERT OR REPLACE INTO working_set (id, event_id, added_at, relevance_score, topics, expires_at)
        VALUES (?, ?, CURRENT_TIMESTAMP, ?, ?, ?)`,
       [
@@ -1524,7 +1634,8 @@ var WorkingSetStore = class {
    */
   async get() {
     await this.cleanup();
-    const rows = await this.db.all(
+    const rows = await dbAll(
+      this.db,
       `SELECT ws.*, e.*
        FROM working_set ws
        JOIN events e ON ws.event_id = e.id
@@ -1536,7 +1647,7 @@ var WorkingSetStore = class {
       id: row.id,
       eventType: row.event_type,
       sessionId: row.session_id,
-      timestamp: new Date(row.timestamp),
+      timestamp: toDate(row.timestamp),
       content: row.content,
       canonicalKey: row.canonical_key,
       dedupeKey: row.dedupe_key,
@@ -1552,23 +1663,25 @@ var WorkingSetStore = class {
    * Get working set items (metadata only)
    */
   async getItems() {
-    const rows = await this.db.all(
+    const rows = await dbAll(
+      this.db,
       `SELECT * FROM working_set ORDER BY relevance_score DESC, added_at DESC`
     );
     return rows.map((row) => ({
       id: row.id,
       eventId: row.event_id,
-      addedAt: new Date(row.added_at),
+      addedAt: toDate(row.added_at),
       relevanceScore: row.relevance_score,
       topics: row.topics ? JSON.parse(row.topics) : void 0,
-      expiresAt: new Date(row.expires_at)
+      expiresAt: toDate(row.expires_at)
     }));
   }
   /**
    * Update relevance score for an event
    */
   async updateRelevance(eventId, score) {
-    await this.db.run(
+    await dbRun(
+      this.db,
       `UPDATE working_set SET relevance_score = ? WHERE event_id = ?`,
       [score, eventId]
     );
@@ -1580,7 +1693,8 @@ var WorkingSetStore = class {
     if (eventIds.length === 0)
       return;
     const placeholders = eventIds.map(() => "?").join(",");
-    await this.db.run(
+    await dbRun(
+      this.db,
       `DELETE FROM working_set WHERE event_id IN (${placeholders})`,
       eventIds
     );
@@ -1589,7 +1703,8 @@ var WorkingSetStore = class {
    * Get the count of items in working set
    */
   async count() {
-    const result = await this.db.all(
+    const result = await dbAll(
+      this.db,
       `SELECT COUNT(*) as count FROM working_set`
     );
     return result[0]?.count || 0;
@@ -1598,13 +1713,14 @@ var WorkingSetStore = class {
    * Clear the entire working set
    */
   async clear() {
-    await this.db.run(`DELETE FROM working_set`);
+    await dbRun(this.db, `DELETE FROM working_set`);
   }
   /**
    * Check if an event is in the working set
    */
   async contains(eventId) {
-    const result = await this.db.all(
+    const result = await dbAll(
+      this.db,
       `SELECT COUNT(*) as count FROM working_set WHERE event_id = ?`,
       [eventId]
     );
@@ -1617,7 +1733,8 @@ var WorkingSetStore = class {
     const newExpiresAt = new Date(
       Date.now() + this.config.workingSet.timeWindowHours * 60 * 60 * 1e3
     );
-    await this.db.run(
+    await dbRun(
+      this.db,
       `UPDATE working_set SET expires_at = ? WHERE event_id = ?`,
       [newExpiresAt.toISOString(), eventId]
     );
@@ -1626,7 +1743,8 @@ var WorkingSetStore = class {
    * Clean up expired items
    */
   async cleanup() {
-    await this.db.run(
+    await dbRun(
+      this.db,
       `DELETE FROM working_set WHERE expires_at < datetime('now')`
     );
   }
@@ -1636,7 +1754,8 @@ var WorkingSetStore = class {
    */
   async enforceLimit() {
     const maxEvents = this.config.workingSet.maxEvents;
-    const keepIds = await this.db.all(
+    const keepIds = await dbAll(
+      this.db,
       `SELECT id FROM working_set
        ORDER BY relevance_score DESC, added_at DESC
        LIMIT ?`,
@@ -1646,7 +1765,8 @@ var WorkingSetStore = class {
       return;
     const keepIdList = keepIds.map((r) => r.id);
     const placeholders = keepIdList.map(() => "?").join(",");
-    await this.db.run(
+    await dbRun(
+      this.db,
       `DELETE FROM working_set WHERE id NOT IN (${placeholders})`,
       keepIdList
     );
@@ -1655,7 +1775,8 @@ var WorkingSetStore = class {
    * Calculate continuity score based on recent context transitions
    */
   async calculateContinuityScore() {
-    const result = await this.db.all(
+    const result = await dbAll(
+      this.db,
       `SELECT AVG(continuity_score) as avg_score
        FROM continuity_log
        WHERE created_at > datetime('now', '-1 hour')`
@@ -1666,7 +1787,8 @@ var WorkingSetStore = class {
    * Get topics from current working set for context matching
    */
   async getActiveTopics() {
-    const rows = await this.db.all(
+    const rows = await dbAll(
+      this.db,
       `SELECT topics FROM working_set WHERE topics IS NOT NULL`
     );
     const allTopics = /* @__PURE__ */ new Set();
@@ -1695,7 +1817,8 @@ var ConsolidatedStore = class {
    */
   async create(input) {
     const memoryId = randomUUID3();
-    await this.db.run(
+    await dbRun(
+      this.db,
       `INSERT INTO consolidated_memories
         (memory_id, summary, topics, source_events, confidence, created_at)
        VALUES (?, ?, ?, ?, ?, CURRENT_TIMESTAMP)`,
@@ -1713,7 +1836,8 @@ var ConsolidatedStore = class {
    * Get a consolidated memory by ID
    */
   async get(memoryId) {
-    const rows = await this.db.all(
+    const rows = await dbAll(
+      this.db,
       `SELECT * FROM consolidated_memories WHERE memory_id = ?`,
       [memoryId]
     );
@@ -1726,7 +1850,8 @@ var ConsolidatedStore = class {
    */
   async search(query, options) {
     const topK = options?.topK || 5;
-    const rows = await this.db.all(
+    const rows = await dbAll(
+      this.db,
       `SELECT * FROM consolidated_memories
        WHERE summary LIKE ?
        ORDER BY confidence DESC
@@ -1742,7 +1867,8 @@ var ConsolidatedStore = class {
     const topK = options?.topK || 5;
     const topicConditions = topics.map(() => `topics LIKE ?`).join(" OR ");
     const topicParams = topics.map((t) => `%"${t}"%`);
-    const rows = await this.db.all(
+    const rows = await dbAll(
+      this.db,
       `SELECT * FROM consolidated_memories
        WHERE ${topicConditions}
        ORDER BY confidence DESC
@@ -1756,7 +1882,8 @@ var ConsolidatedStore = class {
    */
   async getAll(options) {
     const limit = options?.limit || 100;
-    const rows = await this.db.all(
+    const rows = await dbAll(
+      this.db,
       `SELECT * FROM consolidated_memories
        ORDER BY confidence DESC, created_at DESC
        LIMIT ?`,
@@ -1770,7 +1897,8 @@ var ConsolidatedStore = class {
   async getRecent(options) {
     const limit = options?.limit || 10;
     const hours = options?.hours || 24;
-    const rows = await this.db.all(
+    const rows = await dbAll(
+      this.db,
       `SELECT * FROM consolidated_memories
        WHERE created_at > datetime('now', '-${hours} hours')
        ORDER BY created_at DESC
@@ -1783,7 +1911,8 @@ var ConsolidatedStore = class {
    * Mark a memory as accessed (tracks usage for importance scoring)
    */
   async markAccessed(memoryId) {
-    await this.db.run(
+    await dbRun(
+      this.db,
       `UPDATE consolidated_memories
        SET accessed_at = CURRENT_TIMESTAMP,
            access_count = access_count + 1
@@ -1795,7 +1924,8 @@ var ConsolidatedStore = class {
    * Update confidence score for a memory
    */
   async updateConfidence(memoryId, confidence) {
-    await this.db.run(
+    await dbRun(
+      this.db,
       `UPDATE consolidated_memories
        SET confidence = ?
        WHERE memory_id = ?`,
@@ -1806,7 +1936,8 @@ var ConsolidatedStore = class {
    * Delete a consolidated memory
    */
   async delete(memoryId) {
-    await this.db.run(
+    await dbRun(
+      this.db,
       `DELETE FROM consolidated_memories WHERE memory_id = ?`,
       [memoryId]
     );
@@ -1815,7 +1946,8 @@ var ConsolidatedStore = class {
    * Get count of consolidated memories
    */
   async count() {
-    const result = await this.db.all(
+    const result = await dbAll(
+      this.db,
       `SELECT COUNT(*) as count FROM consolidated_memories`
     );
     return result[0]?.count || 0;
@@ -1824,7 +1956,8 @@ var ConsolidatedStore = class {
    * Get most accessed memories (for importance scoring)
    */
   async getMostAccessed(limit = 10) {
-    const rows = await this.db.all(
+    const rows = await dbAll(
+      this.db,
       `SELECT * FROM consolidated_memories
        WHERE access_count > 0
        ORDER BY access_count DESC
@@ -1838,11 +1971,13 @@ var ConsolidatedStore = class {
    */
   async getStats() {
     const total = await this.count();
-    const avgResult = await this.db.all(
+    const avgResult = await dbAll(
+      this.db,
       `SELECT AVG(confidence) as avg FROM consolidated_memories`
     );
     const averageConfidence = avgResult[0]?.avg || 0;
-    const recentResult = await this.db.all(
+    const recentResult = await dbAll(
+      this.db,
       `SELECT COUNT(*) as count FROM consolidated_memories
        WHERE created_at > datetime('now', '-24 hours')`
     );
@@ -1866,7 +2001,8 @@ var ConsolidatedStore = class {
    */
   async isAlreadyConsolidated(eventIds) {
     for (const eventId of eventIds) {
-      const result = await this.db.all(
+      const result = await dbAll(
+        this.db,
         `SELECT COUNT(*) as count FROM consolidated_memories
          WHERE source_events LIKE ?`,
         [`%"${eventId}"%`]
@@ -1880,7 +2016,8 @@ var ConsolidatedStore = class {
    * Get the last consolidation time
    */
   async getLastConsolidationTime() {
-    const result = await this.db.all(
+    const result = await dbAll(
+      this.db,
       `SELECT created_at FROM consolidated_memories
        ORDER BY created_at DESC
        LIMIT 1`
@@ -1899,8 +2036,8 @@ var ConsolidatedStore = class {
       topics: JSON.parse(row.topics || "[]"),
       sourceEvents: JSON.parse(row.source_events || "[]"),
       confidence: row.confidence,
-      createdAt: new Date(row.created_at),
-      accessedAt: row.accessed_at ? new Date(row.accessed_at) : void 0,
+      createdAt: toDate(row.created_at),
+      accessedAt: row.accessed_at ? toDate(row.accessed_at) : void 0,
       accessCount: row.access_count || 0
     };
   }
@@ -2266,7 +2403,8 @@ var ContinuityManager = class {
    * Get recent continuity logs
    */
   async getRecentLogs(limit = 10) {
-    const rows = await this.db.all(
+    const rows = await dbAll(
+      this.db,
       `SELECT * FROM continuity_log
        ORDER BY created_at DESC
        LIMIT ?`,
@@ -2278,14 +2416,15 @@ var ContinuityManager = class {
       toContextId: row.to_context_id,
       continuityScore: row.continuity_score,
       transitionType: row.transition_type,
-      createdAt: new Date(row.created_at)
+      createdAt: toDate(row.created_at)
     }));
   }
   /**
    * Get average continuity score over time period
    */
   async getAverageScore(hours = 1) {
-    const result = await this.db.all(
+    const result = await dbAll(
+      this.db,
       `SELECT AVG(continuity_score) as avg_score
        FROM continuity_log
        WHERE created_at > datetime('now', '-${hours} hours')`
@@ -2296,7 +2435,8 @@ var ContinuityManager = class {
    * Get transition type distribution
    */
   async getTransitionStats(hours = 24) {
-    const rows = await this.db.all(
+    const rows = await dbAll(
+      this.db,
       `SELECT transition_type, COUNT(*) as count
        FROM continuity_log
        WHERE created_at > datetime('now', '-${hours} hours')
@@ -2316,7 +2456,8 @@ var ContinuityManager = class {
    * Clear old continuity logs
    */
   async cleanup(olderThanDays = 7) {
-    const result = await this.db.all(
+    const result = await dbAll(
+      this.db,
       `DELETE FROM continuity_log
        WHERE created_at < datetime('now', '-${olderThanDays} days')
        RETURNING COUNT(*) as changes`
@@ -2351,7 +2492,8 @@ var ContinuityManager = class {
    * Log a context transition
    */
   async logTransition(current, previous, score, type) {
-    await this.db.run(
+    await dbRun(
+      this.db,
       `INSERT INTO continuity_log
         (log_id, from_context_id, to_context_id, continuity_score, transition_type, created_at)
        VALUES (?, ?, ?, ?, ?, CURRENT_TIMESTAMP)`,
@@ -2464,6 +2606,63 @@ function createContinuityManager(eventStore, config) {
 }
 
 // src/services/memory-service.ts
+function normalizePath(projectPath) {
+  const expanded = projectPath.startsWith("~") ? path.join(os.homedir(), projectPath.slice(1)) : projectPath;
+  try {
+    return fs.realpathSync(expanded);
+  } catch {
+    return path.resolve(expanded);
+  }
+}
+function hashProjectPath(projectPath) {
+  const normalizedPath = normalizePath(projectPath);
+  return crypto2.createHash("sha256").update(normalizedPath).digest("hex").slice(0, 8);
+}
+function getProjectStoragePath(projectPath) {
+  const hash = hashProjectPath(projectPath);
+  return path.join(os.homedir(), ".claude-code", "memory", "projects", hash);
+}
+var REGISTRY_PATH = path.join(os.homedir(), ".claude-code", "memory", "session-registry.json");
+function loadSessionRegistry() {
+  try {
+    if (fs.existsSync(REGISTRY_PATH)) {
+      const data = fs.readFileSync(REGISTRY_PATH, "utf-8");
+      return JSON.parse(data);
+    }
+  } catch (error) {
+    console.error("Failed to load session registry:", error);
+  }
+  return { version: 1, sessions: {} };
+}
+function saveSessionRegistry(registry) {
+  const dir = path.dirname(REGISTRY_PATH);
+  if (!fs.existsSync(dir)) {
+    fs.mkdirSync(dir, { recursive: true });
+  }
+  const tempPath = REGISTRY_PATH + ".tmp";
+  fs.writeFileSync(tempPath, JSON.stringify(registry, null, 2));
+  fs.renameSync(tempPath, REGISTRY_PATH);
+}
+function registerSession(sessionId, projectPath) {
+  const registry = loadSessionRegistry();
+  registry.sessions[sessionId] = {
+    projectPath: normalizePath(projectPath),
+    projectHash: hashProjectPath(projectPath),
+    registeredAt: (/* @__PURE__ */ new Date()).toISOString()
+  };
+  const entries = Object.entries(registry.sessions);
+  if (entries.length > 1e3) {
+    const sorted = entries.sort(
+      (a, b) => new Date(b[1].registeredAt).getTime() - new Date(a[1].registeredAt).getTime()
+    );
+    registry.sessions = Object.fromEntries(sorted.slice(0, 1e3));
+  }
+  saveSessionRegistry(registry);
+}
+function getSessionProject(sessionId) {
+  const registry = loadSessionRegistry();
+  return registry.sessions[sessionId] || null;
+}
 var MemoryService = class {
   eventStore;
   vectorStore;
@@ -2911,14 +3110,30 @@ var MemoryService = class {
     return p;
   }
 };
-var defaultService = null;
+var serviceCache = /* @__PURE__ */ new Map();
+var GLOBAL_KEY = "__global__";
 function getDefaultMemoryService() {
-  if (!defaultService) {
-    defaultService = new MemoryService({
+  if (!serviceCache.has(GLOBAL_KEY)) {
+    serviceCache.set(GLOBAL_KEY, new MemoryService({
       storagePath: "~/.claude-code/memory"
-    });
+    }));
+  }
+  return serviceCache.get(GLOBAL_KEY);
+}
+function getMemoryServiceForProject(projectPath) {
+  const hash = hashProjectPath(projectPath);
+  if (!serviceCache.has(hash)) {
+    const storagePath = getProjectStoragePath(projectPath);
+    serviceCache.set(hash, new MemoryService({ storagePath }));
+  }
+  return serviceCache.get(hash);
+}
+function getMemoryServiceForSession(sessionId) {
+  const projectInfo = getSessionProject(sessionId);
+  if (projectInfo) {
+    return getMemoryServiceForProject(projectInfo.projectPath);
   }
-  return defaultService;
+  return getDefaultMemoryService();
 }
 function createMemoryService(config) {
   return new MemoryService(config);
@@ -2926,6 +3141,12 @@ function createMemoryService(config) {
 export {
   MemoryService,
   createMemoryService,
-  getDefaultMemoryService
+  getDefaultMemoryService,
+  getMemoryServiceForProject,
+  getMemoryServiceForSession,
+  getProjectStoragePath,
+  getSessionProject,
+  hashProjectPath,
+  registerSession
 };
 //# sourceMappingURL=memory-service.js.map