@ztorchan/mem0ai 2.4.2 → 2.4.3

package/dist/oss/index.mjs

@@ -414,7 +414,6 @@ var MistralLLM = class {
 };
 
 // src/oss/src/vector_stores/memory.ts
-import Database from "better-sqlite3";
 import fs2 from "fs";
 import path2 from "path";
 
@@ -435,6 +434,7 @@ function ensureSQLiteDirectory(dbPath) {
 // src/oss/src/vector_stores/memory.ts
 var MemoryVectorStore = class {
   constructor(config) {
+    this.initPromise = null;
     this.dimension = config.dimension || 1536;
     this.dbPath = config.dbPath || getDefaultVectorStoreDbPath();
     if (!config.dbPath) {
@@ -445,11 +445,21 @@ var MemoryVectorStore = class {
         );
       }
     }
+  }
+  async ensureInit() {
+    if (this.db) return;
+    if (this.initPromise) return this.initPromise;
+    this.initPromise = this._initDb();
+    return this.initPromise;
+  }
+  async _initDb() {
     ensureSQLiteDirectory(this.dbPath);
+    const mod = await import("better-sqlite3");
+    const Database = mod.default;
     this.db = new Database(this.dbPath);
-    this.init();
+    this.setupTables();
   }
-  init() {
+  setupTables() {
     this.db.exec(`
       CREATE TABLE IF NOT EXISTS vectors (
         id TEXT PRIMARY KEY,
@@ -482,6 +492,7 @@ var MemoryVectorStore = class {
     );
   }
   async insert(vectors, ids, payloads) {
+    await this.ensureInit();
     const stmt = this.db.prepare(
       `INSERT OR REPLACE INTO vectors (id, vector, payload) VALUES (?, ?, ?)`
     );
@@ -501,6 +512,7 @@ var MemoryVectorStore = class {
     insertMany(vectors, ids, payloads);
   }
   async search(query, limit = 10, filters) {
+    await this.ensureInit();
     if (query.length !== this.dimension) {
       throw new Error(
         `Query dimension mismatch. Expected ${this.dimension}, got ${query.length}`
@@ -533,6 +545,7 @@ var MemoryVectorStore = class {
     return results.slice(0, limit);
   }
   async get(vectorId) {
+    await this.ensureInit();
     const row = this.db.prepare(`SELECT * FROM vectors WHERE id = ?`).get(vectorId);
     if (!row) return null;
     const payload = JSON.parse(row.payload);
@@ -542,6 +555,7 @@ var MemoryVectorStore = class {
     };
   }
   async update(vectorId, vector, payload) {
+    await this.ensureInit();
     if (vector.length !== this.dimension) {
       throw new Error(
         `Vector dimension mismatch. Expected ${this.dimension}, got ${vector.length}`
@@ -551,13 +565,16 @@ var MemoryVectorStore = class {
     this.db.prepare(`UPDATE vectors SET vector = ?, payload = ? WHERE id = ?`).run(vectorBuffer, JSON.stringify(payload), vectorId);
   }
   async delete(vectorId) {
+    await this.ensureInit();
     this.db.prepare(`DELETE FROM vectors WHERE id = ?`).run(vectorId);
   }
   async deleteCol() {
+    await this.ensureInit();
     this.db.exec(`DROP TABLE IF EXISTS vectors`);
-    this.init();
+    this.setupTables();
   }
   async list(filters, limit = 100) {
+    await this.ensureInit();
     const rows = this.db.prepare(`SELECT * FROM vectors`).all();
     const results = [];
     for (const row of rows) {
@@ -583,6 +600,7 @@ var MemoryVectorStore = class {
     return [results.slice(0, limit), results.length];
   }
   async getUserId() {
+    await this.ensureInit();
     const row = this.db.prepare(`SELECT user_id FROM memory_migrations LIMIT 1`).get();
     if (row) {
       return row.user_id;
@@ -592,11 +610,12 @@ var MemoryVectorStore = class {
     return randomUserId;
   }
   async setUserId(userId) {
+    await this.ensureInit();
     this.db.prepare(`DELETE FROM memory_migrations`).run();
     this.db.prepare(`INSERT INTO memory_migrations (user_id) VALUES (?)`).run(userId);
   }
   async initialize() {
-    this.init();
+    await this.ensureInit();
   }
 };
 
@@ -2018,14 +2037,25 @@ See the SQL migration instructions in the code comments.`
 };
 
 // src/oss/src/storage/SQLiteManager.ts
-import Database2 from "better-sqlite3";
 var SQLiteManager = class {
   constructor(dbPath) {
-    ensureSQLiteDirectory(dbPath);
-    this.db = new Database2(dbPath);
-    this.init();
+    this.initPromise = null;
+    this.dbPath = dbPath;
   }
-  init() {
+  async ensureInit() {
+    if (this.db) return;
+    if (this.initPromise) return this.initPromise;
+    this.initPromise = this._init();
+    return this.initPromise;
+  }
+  async _init() {
+    ensureSQLiteDirectory(this.dbPath);
+    const mod = await import("better-sqlite3");
+    const Database = mod.default;
+    this.db = new Database(this.dbPath);
+    this.setupStatements();
+  }
+  setupStatements() {
     this.db.exec(`
       CREATE TABLE IF NOT EXISTS memory_history (
         id INTEGER PRIMARY KEY AUTOINCREMENT,
@@ -2048,6 +2078,7 @@ var SQLiteManager = class {
     );
   }
   async addHistory(memoryId, previousValue, newValue, action, createdAt, updatedAt, isDeleted = 0) {
+    await this.ensureInit();
     this.stmtInsert.run(
       memoryId,
       previousValue,
@@ -2059,14 +2090,18 @@ var SQLiteManager = class {
     );
   }
   async getHistory(memoryId) {
+    await this.ensureInit();
     return this.stmtSelect.all(memoryId);
   }
   async reset() {
+    await this.ensureInit();
     this.db.exec("DROP TABLE IF EXISTS memory_history");
-    this.init();
+    this.setupStatements();
   }
   close() {
-    this.db.close();
+    if (this.db) {
+      this.db.close();
+    }
   }
 };