mem0ai 2.2.4 → 2.3.0

package/dist/oss/index.d.mts

@@ -617,9 +617,6 @@ declare class MemoryVectorStore implements VectorStore {
     private dbPath;
     constructor(config: VectorStoreConfig);
     private init;
-    private run;
-    private all;
-    private getOne;
     private cosineSimilarity;
     private filterVector;
     insert(vectors: number[][], ids: string[], payloads: Record<string, any>[]): Promise<void>;

package/dist/oss/index.d.ts

@@ -617,9 +617,6 @@ declare class MemoryVectorStore implements VectorStore {
     private dbPath;
     constructor(config: VectorStoreConfig);
     private init;
-    private run;
-    private all;
-    private getOne;
     private cosineSimilarity;
     private filterVector;
     insert(vectors: number[][], ids: string[], payloads: Record<string, any>[]): Promise<void>;

package/dist/oss/index.js

@@ -470,7 +470,7 @@ var MistralLLM = class {
 };
 
 // src/oss/src/vector_stores/memory.ts
-var import_sqlite3 = __toESM(require("sqlite3"));
+var import_better_sqlite3 = __toESM(require("better-sqlite3"));
 var import_path = __toESM(require("path"));
 var MemoryVectorStore = class {
   constructor(config) {
@@ -479,48 +479,24 @@ var MemoryVectorStore = class {
     if (config.dbPath) {
       this.dbPath = config.dbPath;
     }
-    this.db = new import_sqlite3.default.Database(this.dbPath);
-    this.init().catch(console.error);
+    this.db = new import_better_sqlite3.default(this.dbPath);
+    this.init();
   }
-  async init() {
-    await this.run(`
+  init() {
+    this.db.exec(`
       CREATE TABLE IF NOT EXISTS vectors (
         id TEXT PRIMARY KEY,
         vector BLOB NOT NULL,
         payload TEXT NOT NULL
       )
     `);
-    await this.run(`
+    this.db.exec(`
       CREATE TABLE IF NOT EXISTS memory_migrations (
         id INTEGER PRIMARY KEY AUTOINCREMENT,
         user_id TEXT NOT NULL UNIQUE
       )
     `);
   }
-  async run(sql, params = []) {
-    return new Promise((resolve, reject) => {
-      this.db.run(sql, params, (err) => {
-        if (err) reject(err);
-        else resolve();
-      });
-    });
-  }
-  async all(sql, params = []) {
-    return new Promise((resolve, reject) => {
-      this.db.all(sql, params, (err, rows) => {
-        if (err) reject(err);
-        else resolve(rows);
-      });
-    });
-  }
-  async getOne(sql, params = []) {
-    return new Promise((resolve, reject) => {
-      this.db.get(sql, params, (err, row) => {
-        if (err) reject(err);
-        else resolve(row);
-      });
-    });
-  }
   cosineSimilarity(a, b) {
     let dotProduct = 0;
     let normA = 0;
@@ -539,18 +515,23 @@ var MemoryVectorStore = class {
     );
   }
   async insert(vectors, ids, payloads) {
-    for (let i = 0; i < vectors.length; i++) {
-      if (vectors[i].length !== this.dimension) {
-        throw new Error(
-          `Vector dimension mismatch. Expected ${this.dimension}, got ${vectors[i].length}`
-        );
+    const stmt = this.db.prepare(
+      `INSERT OR REPLACE INTO vectors (id, vector, payload) VALUES (?, ?, ?)`
+    );
+    const insertMany = this.db.transaction(
+      (vecs, vIds, vPayloads) => {
+        for (let i = 0; i < vecs.length; i++) {
+          if (vecs[i].length !== this.dimension) {
+            throw new Error(
+              `Vector dimension mismatch. Expected ${this.dimension}, got ${vecs[i].length}`
+            );
+          }
+          const vectorBuffer = Buffer.from(new Float32Array(vecs[i]).buffer);
+          stmt.run(vIds[i], vectorBuffer, JSON.stringify(vPayloads[i]));
+        }
       }
-      const vectorBuffer = Buffer.from(new Float32Array(vectors[i]).buffer);
-      await this.run(
-        `INSERT OR REPLACE INTO vectors (id, vector, payload) VALUES (?, ?, ?)`,
-        [ids[i], vectorBuffer, JSON.stringify(payloads[i])]
-      );
-    }
+    );
+    insertMany(vectors, ids, payloads);
   }
   async search(query, limit = 10, filters) {
     if (query.length !== this.dimension) {
@@ -558,10 +539,14 @@ var MemoryVectorStore = class {
         `Query dimension mismatch. Expected ${this.dimension}, got ${query.length}`
       );
     }
-    const rows = await this.all(`SELECT * FROM vectors`);
+    const rows = this.db.prepare(`SELECT * FROM vectors`).all();
     const results = [];
     for (const row of rows) {
-      const vector = new Float32Array(row.vector.buffer);
+      const vector = new Float32Array(
+        row.vector.buffer,
+        row.vector.byteOffset,
+        row.vector.byteLength / 4
+      );
       const payload = JSON.parse(row.payload);
       const memoryVector = {
         id: row.id,
@@ -581,9 +566,7 @@ var MemoryVectorStore = class {
     return results.slice(0, limit);
   }
   async get(vectorId) {
-    const row = await this.getOne(`SELECT * FROM vectors WHERE id = ?`, [
-      vectorId
-    ]);
+    const row = this.db.prepare(`SELECT * FROM vectors WHERE id = ?`).get(vectorId);
     if (!row) return null;
     const payload = JSON.parse(row.payload);
     return {
@@ -598,27 +581,29 @@ var MemoryVectorStore = class {
       );
     }
     const vectorBuffer = Buffer.from(new Float32Array(vector).buffer);
-    await this.run(`UPDATE vectors SET vector = ?, payload = ? WHERE id = ?`, [
-      vectorBuffer,
-      JSON.stringify(payload),
-      vectorId
-    ]);
+    this.db.prepare(`UPDATE vectors SET vector = ?, payload = ? WHERE id = ?`).run(vectorBuffer, JSON.stringify(payload), vectorId);
   }
   async delete(vectorId) {
-    await this.run(`DELETE FROM vectors WHERE id = ?`, [vectorId]);
+    this.db.prepare(`DELETE FROM vectors WHERE id = ?`).run(vectorId);
   }
   async deleteCol() {
-    await this.run(`DROP TABLE IF EXISTS vectors`);
-    await this.init();
+    this.db.exec(`DROP TABLE IF EXISTS vectors`);
+    this.init();
   }
   async list(filters, limit = 100) {
-    const rows = await this.all(`SELECT * FROM vectors`);
+    const rows = this.db.prepare(`SELECT * FROM vectors`).all();
     const results = [];
     for (const row of rows) {
       const payload = JSON.parse(row.payload);
       const memoryVector = {
         id: row.id,
-        vector: Array.from(new Float32Array(row.vector.buffer)),
+        vector: Array.from(
+          new Float32Array(
+            row.vector.buffer,
+            row.vector.byteOffset,
+            row.vector.byteLength / 4
+          )
+        ),
         payload
       };
       if (this.filterVector(memoryVector, filters)) {
@@ -631,26 +616,20 @@ var MemoryVectorStore = class {
     return [results.slice(0, limit), results.length];
   }
   async getUserId() {
-    const row = await this.getOne(
-      `SELECT user_id FROM memory_migrations LIMIT 1`
-    );
+    const row = this.db.prepare(`SELECT user_id FROM memory_migrations LIMIT 1`).get();
     if (row) {
       return row.user_id;
     }
     const randomUserId = Math.random().toString(36).substring(2, 15) + Math.random().toString(36).substring(2, 15);
-    await this.run(`INSERT INTO memory_migrations (user_id) VALUES (?)`, [
-      randomUserId
-    ]);
+    this.db.prepare(`INSERT INTO memory_migrations (user_id) VALUES (?)`).run(randomUserId);
     return randomUserId;
   }
   async setUserId(userId) {
-    await this.run(`DELETE FROM memory_migrations`);
-    await this.run(`INSERT INTO memory_migrations (user_id) VALUES (?)`, [
-      userId
-    ]);
+    this.db.prepare(`DELETE FROM memory_migrations`).run();
+    this.db.prepare(`INSERT INTO memory_migrations (user_id) VALUES (?)`).run(userId);
   }
   async initialize() {
-    await this.init();
+    this.init();
   }
 };
 
@@ -2072,14 +2051,14 @@ See the SQL migration instructions in the code comments.`
 };
 
 // src/oss/src/storage/SQLiteManager.ts
-var import_sqlite32 = __toESM(require("sqlite3"));
+var import_better_sqlite32 = __toESM(require("better-sqlite3"));
 var SQLiteManager = class {
   constructor(dbPath) {
-    this.db = new import_sqlite32.default.Database(dbPath);
-    this.init().catch(console.error);
+    this.db = new import_better_sqlite32.default(dbPath);
+    this.init();
   }
-  async init() {
-    await this.run(`
+  init() {
+    this.db.exec(`
       CREATE TABLE IF NOT EXISTS memory_history (
         id INTEGER PRIMARY KEY AUTOINCREMENT,
         memory_id TEXT NOT NULL,
@@ -2091,48 +2070,32 @@ var SQLiteManager = class {
         is_deleted INTEGER DEFAULT 0
       )
     `);
-  }
-  async run(sql, params = []) {
-    return new Promise((resolve, reject) => {
-      this.db.run(sql, params, (err) => {
-        if (err) reject(err);
-        else resolve();
-      });
-    });
-  }
-  async all(sql, params = []) {
-    return new Promise((resolve, reject) => {
-      this.db.all(sql, params, (err, rows) => {
-        if (err) reject(err);
-        else resolve(rows);
-      });
-    });
+    this.stmtInsert = this.db.prepare(
+      `INSERT INTO memory_history
+      (memory_id, previous_value, new_value, action, created_at, updated_at, is_deleted)
+      VALUES (?, ?, ?, ?, ?, ?, ?)`
+    );
+    this.stmtSelect = this.db.prepare(
+      "SELECT * FROM memory_history WHERE memory_id = ? ORDER BY id DESC"
+    );
   }
   async addHistory(memoryId, previousValue, newValue, action, createdAt, updatedAt, isDeleted = 0) {
-    await this.run(
-      `INSERT INTO memory_history 
-      (memory_id, previous_value, new_value, action, created_at, updated_at, is_deleted)
-      VALUES (?, ?, ?, ?, ?, ?, ?)`,
-      [
-        memoryId,
-        previousValue,
-        newValue,
-        action,
-        createdAt,
-        updatedAt,
-        isDeleted
-      ]
+    this.stmtInsert.run(
+      memoryId,
+      previousValue,
+      newValue,
+      action,
+      createdAt != null ? createdAt : null,
+      updatedAt != null ? updatedAt : null,
+      isDeleted
     );
   }
   async getHistory(memoryId) {
-    return this.all(
-      "SELECT * FROM memory_history WHERE memory_id = ? ORDER BY id DESC",
-      [memoryId]
-    );
+    return this.stmtSelect.all(memoryId);
   }
   async reset() {
-    await this.run("DROP TABLE IF EXISTS memory_history");
-    await this.init();
+    this.db.exec("DROP TABLE IF EXISTS memory_history");
+    this.init();
   }
   close() {
     this.db.close();