mem0ai 2.0.1 → 2.0.2

package/dist/oss/index.d.mts

@@ -226,7 +226,6 @@ interface AddMemoryOptions extends Entity {
     prompt?: string;
 }
 interface SearchMemoryOptions extends Entity {
-    query: string;
     limit?: number;
     filters?: SearchFilters;
 }
@@ -343,9 +342,14 @@ interface VectorStore {
 }
 
 declare class MemoryVectorStore implements VectorStore {
-    private vectors;
+    private db;
     private dimension;
+    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

@@ -226,7 +226,6 @@ interface AddMemoryOptions extends Entity {
     prompt?: string;
 }
 interface SearchMemoryOptions extends Entity {
-    query: string;
     limit?: number;
     filters?: SearchFilters;
 }
@@ -343,9 +342,14 @@ interface VectorStore {
 }
 
 declare class MemoryVectorStore implements VectorStore {
-    private vectors;
+    private db;
     private dimension;
+    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

@@ -251,10 +251,50 @@ var GroqLLM = class {
 };
 
 // src/oss/src/vector_stores/memory.ts
+var import_sqlite3 = __toESM(require("sqlite3"));
+var import_path = __toESM(require("path"));
 var MemoryVectorStore = class {
   constructor(config) {
-    this.vectors = /* @__PURE__ */ new Map();
     this.dimension = config.dimension || 1536;
+    this.dbPath = import_path.default.join(process.cwd(), "vector_store.db");
+    if (config.dbPath) {
+      this.dbPath = config.dbPath;
+    }
+    this.db = new import_sqlite3.default.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;
@@ -280,11 +320,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) {
@@ -293,13 +333,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
         });
       }
@@ -308,11 +356,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) {
@@ -321,27 +372,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
         });
       }
     }
@@ -1219,10 +1277,10 @@ function removeCodeBlocks(text) {
 }
 
 // src/oss/src/storage/SQLiteManager.ts
-var import_sqlite3 = __toESM(require("sqlite3"));
+var import_sqlite32 = __toESM(require("sqlite3"));
 var SQLiteManager = class {
   constructor(dbPath) {
-    this.db = new import_sqlite3.default.Database(dbPath);
+    this.db = new import_sqlite32.default.Database(dbPath);
     this.init().catch(console.error);
   }
   async init() {