mem0ai 2.2.3 → 2.3.0

package/dist/oss/index.mjs

@@ -112,9 +112,10 @@ var OllamaEmbedder = class {
     } catch (err) {
       logger.error(`Error ensuring model exists: ${err}`);
     }
+    const prompt = typeof text === "string" ? text : JSON.stringify(text);
     const response = await this.ollama.embeddings({
       model: this.model,
-      prompt: text
+      prompt
     });
     return response.embedding;
   }
@@ -408,7 +409,7 @@ var MistralLLM = class {
 };
 
 // src/oss/src/vector_stores/memory.ts
-import sqlite3 from "sqlite3";
+import Database from "better-sqlite3";
 import path from "path";
 var MemoryVectorStore = class {
   constructor(config) {
@@ -417,48 +418,24 @@ var MemoryVectorStore = class {
     if (config.dbPath) {
       this.dbPath = config.dbPath;
     }
-    this.db = new sqlite3.Database(this.dbPath);
-    this.init().catch(console.error);
+    this.db = new Database(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;
@@ -477,18 +454,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) {
@@ -496,10 +478,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,
@@ -519,9 +505,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 {
@@ -536,27 +520,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)) {
@@ -569,26 +555,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();
   }
 };
 
@@ -2010,14 +1990,14 @@ See the SQL migration instructions in the code comments.`
 };
 
 // src/oss/src/storage/SQLiteManager.ts
-import sqlite32 from "sqlite3";
+import Database2 from "better-sqlite3";
 var SQLiteManager = class {
   constructor(dbPath) {
-    this.db = new sqlite32.Database(dbPath);
-    this.init().catch(console.error);
+    this.db = new Database2(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,
@@ -2029,48 +2009,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();
@@ -2346,8 +2310,13 @@ import {
 
 // src/oss/src/prompts/index.ts
 import { z as z2 } from "zod";
+var factItem = z2.union([
+  z2.string(),
+  z2.object({ fact: z2.string() }).transform((o) => o.fact),
+  z2.object({ text: z2.string() }).transform((o) => o.text)
+]);
 var FactRetrievalSchema = z2.object({
-  facts: z2.array(z2.string()).describe("An array of distinct facts extracted from the conversation.")
+  facts: z2.array(factItem).transform((arr) => arr.filter((s) => s.length > 0)).describe("An array of distinct facts extracted from the conversation.")
 });
 var MemoryUpdateSchema = z2.object({
   memory: z2.array(
@@ -4659,7 +4628,8 @@ ${parsedMessages}`
     const cleanResponse = removeCodeBlocks(response);
     let facts = [];
     try {
-      facts = JSON.parse(cleanResponse).facts || [];
+      const parsed = FactRetrievalSchema.parse(JSON.parse(cleanResponse));
+      facts = parsed.facts;
     } catch (e) {
       console.error(
         "Failed to parse facts from LLM response:",