mem0ai 2.0.0 → 2.0.2

package/dist/oss/index.d.mts

@@ -1,4 +1,5 @@
 import { z } from 'zod';
+import { QdrantClient } from '@qdrant/js-client-rest';
 
 interface Message {
     role: string;
@@ -214,6 +215,26 @@ declare const MemoryConfigSchema: z.ZodObject<{
     } | undefined;
 }>;
 
+interface Entity {
+    userId?: string;
+    agentId?: string;
+    runId?: string;
+}
+interface AddMemoryOptions extends Entity {
+    metadata?: Record<string, any>;
+    filters?: SearchFilters;
+    prompt?: string;
+}
+interface SearchMemoryOptions extends Entity {
+    limit?: number;
+    filters?: SearchFilters;
+}
+interface GetAllMemoryOptions extends Entity {
+    limit?: number;
+}
+interface DeleteAllMemoryOptions extends Entity {
+}
+
 declare class Memory {
     private config;
     private customPrompt;
@@ -225,22 +246,22 @@ declare class Memory {
     private apiVersion;
     constructor(config?: Partial<MemoryConfig>);
     static fromConfig(configDict: Record<string, any>): Memory;
-    add(messages: string | Message[], userId?: string, agentId?: string, runId?: string, metadata?: Record<string, any>, filters?: SearchFilters, prompt?: string): Promise<SearchResult>;
+    add(messages: string | Message[], config: AddMemoryOptions): Promise<SearchResult>;
     private addToVectorStore;
     get(memoryId: string): Promise<MemoryItem | null>;
-    search(query: string, userId?: string, agentId?: string, runId?: string, limit?: number, filters?: SearchFilters): Promise<SearchResult>;
+    search(query: string, config: SearchMemoryOptions): Promise<SearchResult>;
     update(memoryId: string, data: string): Promise<{
         message: string;
     }>;
     delete(memoryId: string): Promise<{
         message: string;
     }>;
-    deleteAll(userId?: string, agentId?: string, runId?: string): Promise<{
+    deleteAll(config: DeleteAllMemoryOptions): Promise<{
         message: string;
     }>;
     history(memoryId: string): Promise<any[]>;
     reset(): Promise<void>;
-    getAll(userId?: string, agentId?: string, runId?: string, limit?: number): Promise<SearchResult>;
+    getAll(config: GetAllMemoryOptions): Promise<SearchResult>;
     private createMemory;
     private updateMemory;
     private deleteMemory;
@@ -321,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>;
@@ -335,6 +361,57 @@ declare class MemoryVectorStore implements VectorStore {
     list(filters?: SearchFilters, limit?: number): Promise<[VectorStoreResult[], number]>;
 }
 
+interface QdrantConfig extends VectorStoreConfig {
+    client?: QdrantClient;
+    host?: string;
+    port?: number;
+    path?: string;
+    url?: string;
+    apiKey?: string;
+    onDisk?: boolean;
+    collectionName: string;
+    embeddingModelDims: number;
+}
+declare class Qdrant implements VectorStore {
+    private client;
+    private readonly collectionName;
+    constructor(config: QdrantConfig);
+    private createCol;
+    private createFilter;
+    insert(vectors: number[][], ids: string[], payloads: Record<string, any>[]): Promise<void>;
+    search(query: number[], limit?: number, filters?: SearchFilters): Promise<VectorStoreResult[]>;
+    get(vectorId: string): Promise<VectorStoreResult | null>;
+    update(vectorId: string, vector: number[], payload: Record<string, any>): Promise<void>;
+    delete(vectorId: string): Promise<void>;
+    deleteCol(): Promise<void>;
+    list(filters?: SearchFilters, limit?: number): Promise<[VectorStoreResult[], number]>;
+}
+
+interface RedisConfig extends VectorStoreConfig {
+    redisUrl: string;
+    collectionName: string;
+    embeddingModelDims: number;
+    username?: string;
+    password?: string;
+}
+declare class RedisDB implements VectorStore {
+    private client;
+    private readonly indexName;
+    private readonly indexPrefix;
+    private readonly schema;
+    constructor(config: RedisConfig);
+    private initialize;
+    private createIndex;
+    insert(vectors: number[][], ids: string[], payloads: Record<string, any>[]): Promise<void>;
+    search(query: number[], limit?: number, filters?: SearchFilters): Promise<VectorStoreResult[]>;
+    get(vectorId: string): Promise<VectorStoreResult | null>;
+    update(vectorId: string, vector: number[], payload: Record<string, any>): Promise<void>;
+    delete(vectorId: string): Promise<void>;
+    deleteCol(): Promise<void>;
+    list(filters?: SearchFilters, limit?: number): Promise<[VectorStoreResult[], number]>;
+    close(): Promise<void>;
+}
+
 declare class EmbedderFactory {
     static create(provider: string, config: EmbeddingConfig): Embedder;
 }
@@ -345,4 +422,4 @@ declare class VectorStoreFactory {
     static create(provider: string, config: VectorStoreConfig): VectorStore;
 }
 
-export { AnthropicLLM, type Embedder, EmbedderFactory, type EmbeddingConfig, type GraphStoreConfig, GroqLLM, type LLM, type LLMConfig, LLMFactory, type LLMResponse, Memory, type MemoryConfig, MemoryConfigSchema, type MemoryItem, MemoryVectorStore, type Message, OpenAIEmbedder, OpenAILLM, OpenAIStructuredLLM, type SearchFilters, type SearchResult, type VectorStore, type VectorStoreConfig, VectorStoreFactory, type VectorStoreResult };
+export { type AddMemoryOptions, AnthropicLLM, type DeleteAllMemoryOptions, type Embedder, EmbedderFactory, type EmbeddingConfig, type Entity, type GetAllMemoryOptions, type GraphStoreConfig, GroqLLM, type LLM, type LLMConfig, LLMFactory, type LLMResponse, Memory, type MemoryConfig, MemoryConfigSchema, type MemoryItem, MemoryVectorStore, type Message, OpenAIEmbedder, OpenAILLM, OpenAIStructuredLLM, Qdrant, RedisDB, type SearchFilters, type SearchMemoryOptions, type SearchResult, type VectorStore, type VectorStoreConfig, VectorStoreFactory, type VectorStoreResult };

package/dist/oss/index.d.ts

@@ -1,4 +1,5 @@
 import { z } from 'zod';
+import { QdrantClient } from '@qdrant/js-client-rest';
 
 interface Message {
     role: string;
@@ -214,6 +215,26 @@ declare const MemoryConfigSchema: z.ZodObject<{
     } | undefined;
 }>;
 
+interface Entity {
+    userId?: string;
+    agentId?: string;
+    runId?: string;
+}
+interface AddMemoryOptions extends Entity {
+    metadata?: Record<string, any>;
+    filters?: SearchFilters;
+    prompt?: string;
+}
+interface SearchMemoryOptions extends Entity {
+    limit?: number;
+    filters?: SearchFilters;
+}
+interface GetAllMemoryOptions extends Entity {
+    limit?: number;
+}
+interface DeleteAllMemoryOptions extends Entity {
+}
+
 declare class Memory {
     private config;
     private customPrompt;
@@ -225,22 +246,22 @@ declare class Memory {
     private apiVersion;
     constructor(config?: Partial<MemoryConfig>);
     static fromConfig(configDict: Record<string, any>): Memory;
-    add(messages: string | Message[], userId?: string, agentId?: string, runId?: string, metadata?: Record<string, any>, filters?: SearchFilters, prompt?: string): Promise<SearchResult>;
+    add(messages: string | Message[], config: AddMemoryOptions): Promise<SearchResult>;
     private addToVectorStore;
     get(memoryId: string): Promise<MemoryItem | null>;
-    search(query: string, userId?: string, agentId?: string, runId?: string, limit?: number, filters?: SearchFilters): Promise<SearchResult>;
+    search(query: string, config: SearchMemoryOptions): Promise<SearchResult>;
     update(memoryId: string, data: string): Promise<{
         message: string;
     }>;
     delete(memoryId: string): Promise<{
         message: string;
     }>;
-    deleteAll(userId?: string, agentId?: string, runId?: string): Promise<{
+    deleteAll(config: DeleteAllMemoryOptions): Promise<{
         message: string;
     }>;
     history(memoryId: string): Promise<any[]>;
     reset(): Promise<void>;
-    getAll(userId?: string, agentId?: string, runId?: string, limit?: number): Promise<SearchResult>;
+    getAll(config: GetAllMemoryOptions): Promise<SearchResult>;
     private createMemory;
     private updateMemory;
     private deleteMemory;
@@ -321,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>;
@@ -335,6 +361,57 @@ declare class MemoryVectorStore implements VectorStore {
     list(filters?: SearchFilters, limit?: number): Promise<[VectorStoreResult[], number]>;
 }
 
+interface QdrantConfig extends VectorStoreConfig {
+    client?: QdrantClient;
+    host?: string;
+    port?: number;
+    path?: string;
+    url?: string;
+    apiKey?: string;
+    onDisk?: boolean;
+    collectionName: string;
+    embeddingModelDims: number;
+}
+declare class Qdrant implements VectorStore {
+    private client;
+    private readonly collectionName;
+    constructor(config: QdrantConfig);
+    private createCol;
+    private createFilter;
+    insert(vectors: number[][], ids: string[], payloads: Record<string, any>[]): Promise<void>;
+    search(query: number[], limit?: number, filters?: SearchFilters): Promise<VectorStoreResult[]>;
+    get(vectorId: string): Promise<VectorStoreResult | null>;
+    update(vectorId: string, vector: number[], payload: Record<string, any>): Promise<void>;
+    delete(vectorId: string): Promise<void>;
+    deleteCol(): Promise<void>;
+    list(filters?: SearchFilters, limit?: number): Promise<[VectorStoreResult[], number]>;
+}
+
+interface RedisConfig extends VectorStoreConfig {
+    redisUrl: string;
+    collectionName: string;
+    embeddingModelDims: number;
+    username?: string;
+    password?: string;
+}
+declare class RedisDB implements VectorStore {
+    private client;
+    private readonly indexName;
+    private readonly indexPrefix;
+    private readonly schema;
+    constructor(config: RedisConfig);
+    private initialize;
+    private createIndex;
+    insert(vectors: number[][], ids: string[], payloads: Record<string, any>[]): Promise<void>;
+    search(query: number[], limit?: number, filters?: SearchFilters): Promise<VectorStoreResult[]>;
+    get(vectorId: string): Promise<VectorStoreResult | null>;
+    update(vectorId: string, vector: number[], payload: Record<string, any>): Promise<void>;
+    delete(vectorId: string): Promise<void>;
+    deleteCol(): Promise<void>;
+    list(filters?: SearchFilters, limit?: number): Promise<[VectorStoreResult[], number]>;
+    close(): Promise<void>;
+}
+
 declare class EmbedderFactory {
     static create(provider: string, config: EmbeddingConfig): Embedder;
 }
@@ -345,4 +422,4 @@ declare class VectorStoreFactory {
     static create(provider: string, config: VectorStoreConfig): VectorStore;
 }
 
-export { AnthropicLLM, type Embedder, EmbedderFactory, type EmbeddingConfig, type GraphStoreConfig, GroqLLM, type LLM, type LLMConfig, LLMFactory, type LLMResponse, Memory, type MemoryConfig, MemoryConfigSchema, type MemoryItem, MemoryVectorStore, type Message, OpenAIEmbedder, OpenAILLM, OpenAIStructuredLLM, type SearchFilters, type SearchResult, type VectorStore, type VectorStoreConfig, VectorStoreFactory, type VectorStoreResult };
+export { type AddMemoryOptions, AnthropicLLM, type DeleteAllMemoryOptions, type Embedder, EmbedderFactory, type EmbeddingConfig, type Entity, type GetAllMemoryOptions, type GraphStoreConfig, GroqLLM, type LLM, type LLMConfig, LLMFactory, type LLMResponse, Memory, type MemoryConfig, MemoryConfigSchema, type MemoryItem, MemoryVectorStore, type Message, OpenAIEmbedder, OpenAILLM, OpenAIStructuredLLM, Qdrant, RedisDB, type SearchFilters, type SearchMemoryOptions, type SearchResult, type VectorStore, type VectorStoreConfig, VectorStoreFactory, type VectorStoreResult };

package/dist/oss/index.js

@@ -40,6 +40,8 @@ __export(index_exports, {
   OpenAIEmbedder: () => OpenAIEmbedder,
   OpenAILLM: () => OpenAILLM,
   OpenAIStructuredLLM: () => OpenAIStructuredLLM,
+  Qdrant: () => Qdrant,
+  RedisDB: () => RedisDB,
   VectorStoreFactory: () => VectorStoreFactory
 });
 module.exports = __toCommonJS(index_exports);
@@ -249,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;
@@ -278,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) {
@@ -291,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
         });
       }
@@ -306,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) {
@@ -319,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
         });
       }
     }
@@ -1217,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() {
@@ -1377,7 +1437,15 @@ var Memory = class _Memory {
       throw e;
     }
   }
-  async add(messages, userId, agentId, runId, metadata = {}, filters = {}, prompt) {
+  async add(messages, config) {
+    const {
+      userId,
+      agentId,
+      runId,
+      metadata = {},
+      filters = {},
+      prompt
+    } = config;
     if (userId) filters.userId = metadata.userId = userId;
     if (agentId) filters.agentId = metadata.agentId = agentId;
     if (runId) filters.runId = metadata.runId = runId;
@@ -1520,7 +1588,8 @@ ${parsedMessages}`] : getFactRetrievalMessages(parsedMessages);
     }
     return { ...memoryItem, ...filters };
   }
-  async search(query, userId, agentId, runId, limit = 100, filters = {}) {
+  async search(query, config) {
+    const { userId, agentId, runId, limit = 100, filters = {} } = config;
     if (userId) filters.userId = userId;
     if (agentId) filters.agentId = agentId;
     if (runId) filters.runId = runId;
@@ -1567,7 +1636,8 @@ ${parsedMessages}`] : getFactRetrievalMessages(parsedMessages);
     await this.deleteMemory(memoryId);
     return { message: "Memory deleted successfully!" };
   }
-  async deleteAll(userId, agentId, runId) {
+  async deleteAll(config) {
+    const { userId, agentId, runId } = config;
     const filters = {};
     if (userId) filters.userId = userId;
     if (agentId) filters.agentId = agentId;
@@ -1594,7 +1664,8 @@ ${parsedMessages}`] : getFactRetrievalMessages(parsedMessages);
       this.config.vectorStore.config
     );
   }
-  async getAll(userId, agentId, runId, limit = 100) {
+  async getAll(config) {
+    const { userId, agentId, runId, limit = 100 } = config;
     const filters = {};
     if (userId) filters.userId = userId;
     if (agentId) filters.agentId = agentId;
@@ -1706,6 +1777,8 @@ ${parsedMessages}`] : getFactRetrievalMessages(parsedMessages);
   OpenAIEmbedder,
   OpenAILLM,
   OpenAIStructuredLLM,
+  Qdrant,
+  RedisDB,
   VectorStoreFactory
 });
 //# sourceMappingURL=index.js.map