mem0ai 2.0.0 → 2.0.1

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,27 @@ 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 {
+    query: string;
+    limit?: number;
+    filters?: SearchFilters;
+}
+interface GetAllMemoryOptions extends Entity {
+    limit?: number;
+}
+interface DeleteAllMemoryOptions extends Entity {
+}
+
 declare class Memory {
     private config;
     private customPrompt;
@@ -225,22 +247,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;
@@ -335,6 +357,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 +418,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,27 @@ 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 {
+    query: string;
+    limit?: number;
+    filters?: SearchFilters;
+}
+interface GetAllMemoryOptions extends Entity {
+    limit?: number;
+}
+interface DeleteAllMemoryOptions extends Entity {
+}
+
 declare class Memory {
     private config;
     private customPrompt;
@@ -225,22 +247,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;
@@ -335,6 +357,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 +418,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);
@@ -1377,7 +1379,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 +1530,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 +1578,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 +1606,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 +1719,8 @@ ${parsedMessages}`] : getFactRetrievalMessages(parsedMessages);
   OpenAIEmbedder,
   OpenAILLM,
   OpenAIStructuredLLM,
+  Qdrant,
+  RedisDB,
   VectorStoreFactory
 });
 //# sourceMappingURL=index.js.map