mem0ai 2.0.1 → 2.0.2

package/dist/oss/index.mjs

@@ -203,10 +203,50 @@ var GroqLLM = class {
 };
 
 // src/oss/src/vector_stores/memory.ts
+import sqlite3 from "sqlite3";
+import path from "path";
 var MemoryVectorStore = class {
   constructor(config) {
-    this.vectors = /* @__PURE__ */ new Map();
     this.dimension = config.dimension || 1536;
+    this.dbPath = path.join(process.cwd(), "vector_store.db");
+    if (config.dbPath) {
+      this.dbPath = config.dbPath;
+    }
+    this.db = new sqlite3.Database(this.dbPath);
+    this.init().catch(console.error);
+  }
+  async init() {
+    await this.run(`
+      CREATE TABLE IF NOT EXISTS vectors (
+        id TEXT PRIMARY KEY,
+        vector BLOB NOT NULL,
+        payload TEXT NOT NULL
+      )
+    `);
+  }
+  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;
@@ -232,11 +272,11 @@ var MemoryVectorStore = class {
           `Vector dimension mismatch. Expected ${this.dimension}, got ${vectors[i].length}`
         );
       }
-      this.vectors.set(ids[i], {
-        id: ids[i],
-        vector: vectors[i],
-        payload: payloads[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])]
+      );
     }
   }
   async search(query, limit = 10, filters) {
@@ -245,13 +285,21 @@ var MemoryVectorStore = class {
         `Query dimension mismatch. Expected ${this.dimension}, got ${query.length}`
       );
     }
+    const rows = await this.all(`SELECT * FROM vectors`);
     const results = [];
-    for (const vector of this.vectors.values()) {
-      if (this.filterVector(vector, filters)) {
-        const score = this.cosineSimilarity(query, vector.vector);
+    for (const row of rows) {
+      const vector = new Float32Array(row.vector.buffer);
+      const payload = JSON.parse(row.payload);
+      const memoryVector = {
+        id: row.id,
+        vector: Array.from(vector),
+        payload
+      };
+      if (this.filterVector(memoryVector, filters)) {
+        const score = this.cosineSimilarity(query, Array.from(vector));
         results.push({
-          id: vector.id,
-          payload: vector.payload,
+          id: memoryVector.id,
+          payload: memoryVector.payload,
           score
         });
       }
@@ -260,11 +308,14 @@ var MemoryVectorStore = class {
     return results.slice(0, limit);
   }
   async get(vectorId) {
-    const vector = this.vectors.get(vectorId);
-    if (!vector) return null;
+    const row = await this.getOne(`SELECT * FROM vectors WHERE id = ?`, [
+      vectorId
+    ]);
+    if (!row) return null;
+    const payload = JSON.parse(row.payload);
     return {
-      id: vector.id,
-      payload: vector.payload
+      id: row.id,
+      payload
     };
   }
   async update(vectorId, vector, payload) {
@@ -273,27 +324,34 @@ var MemoryVectorStore = class {
         `Vector dimension mismatch. Expected ${this.dimension}, got ${vector.length}`
       );
     }
-    const existing = this.vectors.get(vectorId);
-    if (!existing) throw new Error(`Vector with ID ${vectorId} not found`);
-    this.vectors.set(vectorId, {
-      id: vectorId,
-      vector,
-      payload
-    });
+    const vectorBuffer = Buffer.from(new Float32Array(vector).buffer);
+    await this.run(`UPDATE vectors SET vector = ?, payload = ? WHERE id = ?`, [
+      vectorBuffer,
+      JSON.stringify(payload),
+      vectorId
+    ]);
   }
   async delete(vectorId) {
-    this.vectors.delete(vectorId);
+    await this.run(`DELETE FROM vectors WHERE id = ?`, [vectorId]);
   }
   async deleteCol() {
-    this.vectors.clear();
+    await this.run(`DROP TABLE IF EXISTS vectors`);
+    await this.init();
   }
   async list(filters, limit = 100) {
+    const rows = await this.all(`SELECT * FROM vectors`);
     const results = [];
-    for (const vector of this.vectors.values()) {
-      if (this.filterVector(vector, filters)) {
+    for (const row of rows) {
+      const payload = JSON.parse(row.payload);
+      const memoryVector = {
+        id: row.id,
+        vector: Array.from(new Float32Array(row.vector.buffer)),
+        payload
+      };
+      if (this.filterVector(memoryVector, filters)) {
         results.push({
-          id: vector.id,
-          payload: vector.payload
+          id: memoryVector.id,
+          payload: memoryVector.payload
         });
       }
     }
@@ -1171,10 +1229,10 @@ function removeCodeBlocks(text) {
 }
 
 // src/oss/src/storage/SQLiteManager.ts
-import sqlite3 from "sqlite3";
+import sqlite32 from "sqlite3";
 var SQLiteManager = class {
   constructor(dbPath) {
-    this.db = new sqlite3.Database(dbPath);
+    this.db = new sqlite32.Database(dbPath);
     this.init().catch(console.error);
   }
   async init() {