mem0ai 2.0.1 → 2.1.0

package/dist/oss/index.d.mts

@@ -15,11 +15,21 @@ interface VectorStoreConfig {
     [key: string]: any;
 }
 interface LLMConfig {
-    apiKey: string;
+    provider?: string;
+    config?: Record<string, any>;
+    apiKey?: string;
     model?: string;
 }
+interface Neo4jConfig {
+    url: string;
+    username: string;
+    password: string;
+}
 interface GraphStoreConfig {
-    config?: any;
+    provider: string;
+    config: Neo4jConfig;
+    llm?: LLMConfig;
+    customPrompt?: string;
 }
 interface MemoryConfig {
     version?: string;
@@ -38,6 +48,7 @@ interface MemoryConfig {
     historyDbPath?: string;
     customPrompt?: string;
     graphStore?: GraphStoreConfig;
+    enableGraph?: boolean;
 }
 interface MemoryItem {
     id: string;
@@ -146,12 +157,57 @@ declare const MemoryConfigSchema: z.ZodObject<{
     }>;
     historyDbPath: z.ZodOptional<z.ZodString>;
     customPrompt: z.ZodOptional<z.ZodString>;
+    enableGraph: z.ZodOptional<z.ZodBoolean>;
     graphStore: z.ZodOptional<z.ZodObject<{
-        config: z.ZodOptional<z.ZodAny>;
+        provider: z.ZodString;
+        config: z.ZodObject<{
+            url: z.ZodString;
+            username: z.ZodString;
+            password: z.ZodString;
+        }, "strip", z.ZodTypeAny, {
+            url: string;
+            username: string;
+            password: string;
+        }, {
+            url: string;
+            username: string;
+            password: string;
+        }>;
+        llm: z.ZodOptional<z.ZodObject<{
+            provider: z.ZodString;
+            config: z.ZodRecord<z.ZodString, z.ZodAny>;
+        }, "strip", z.ZodTypeAny, {
+            provider: string;
+            config: Record<string, any>;
+        }, {
+            provider: string;
+            config: Record<string, any>;
+        }>>;
+        customPrompt: z.ZodOptional<z.ZodString>;
     }, "strip", z.ZodTypeAny, {
-        config?: any;
+        provider: string;
+        config: {
+            url: string;
+            username: string;
+            password: string;
+        };
+        llm?: {
+            provider: string;
+            config: Record<string, any>;
+        } | undefined;
+        customPrompt?: string | undefined;
     }, {
-        config?: any;
+        provider: string;
+        config: {
+            url: string;
+            username: string;
+            password: string;
+        };
+        llm?: {
+            provider: string;
+            config: Record<string, any>;
+        } | undefined;
+        customPrompt?: string | undefined;
     }>>;
 }, "strip", z.ZodTypeAny, {
     embedder: {
@@ -180,8 +236,19 @@ declare const MemoryConfigSchema: z.ZodObject<{
     version?: string | undefined;
     historyDbPath?: string | undefined;
     customPrompt?: string | undefined;
+    enableGraph?: boolean | undefined;
     graphStore?: {
-        config?: any;
+        provider: string;
+        config: {
+            url: string;
+            username: string;
+            password: string;
+        };
+        llm?: {
+            provider: string;
+            config: Record<string, any>;
+        } | undefined;
+        customPrompt?: string | undefined;
     } | undefined;
 }, {
     embedder: {
@@ -210,8 +277,19 @@ declare const MemoryConfigSchema: z.ZodObject<{
     version?: string | undefined;
     historyDbPath?: string | undefined;
     customPrompt?: string | undefined;
+    enableGraph?: boolean | undefined;
     graphStore?: {
-        config?: any;
+        provider: string;
+        config: {
+            url: string;
+            username: string;
+            password: string;
+        };
+        llm?: {
+            provider: string;
+            config: Record<string, any>;
+        } | undefined;
+        customPrompt?: string | undefined;
     } | undefined;
 }>;
 
@@ -226,7 +304,6 @@ interface AddMemoryOptions extends Entity {
     prompt?: string;
 }
 interface SearchMemoryOptions extends Entity {
-    query: string;
     limit?: number;
     filters?: SearchFilters;
 }
@@ -245,6 +322,8 @@ declare class Memory {
     private db;
     private collectionName;
     private apiVersion;
+    private graphMemory?;
+    private enableGraph;
     constructor(config?: Partial<MemoryConfig>);
     static fromConfig(configDict: Record<string, any>): Memory;
     add(messages: string | Message[], config: AddMemoryOptions): Promise<SearchResult>;
@@ -284,11 +363,18 @@ declare class OpenAIEmbedder implements Embedder {
 interface LLMResponse {
     content: string;
     role: string;
+    toolCalls?: Array<{
+        name: string;
+        arguments: string;
+    }>;
 }
 interface LLM {
-    generateResponse(messages: Message[], responseFormat?: {
+    generateResponse(messages: Array<{
+        role: string;
+        content: string;
+    }>, response_format: {
         type: string;
-    }): Promise<string>;
+    }, tools?: any[]): Promise<any>;
     generateChat(messages: Message[]): Promise<LLMResponse>;
 }
 
@@ -298,17 +384,17 @@ declare class OpenAILLM implements LLM {
     constructor(config: LLMConfig);
     generateResponse(messages: Message[], responseFormat?: {
         type: string;
-    }): Promise<string>;
+    }, tools?: any[]): Promise<string | LLMResponse>;
     generateChat(messages: Message[]): Promise<LLMResponse>;
 }
 
 declare class OpenAIStructuredLLM implements LLM {
-    private client;
+    private openai;
     private model;
     constructor(config: LLMConfig);
     generateResponse(messages: Message[], responseFormat?: {
         type: string;
-    }): Promise<string>;
+    } | null, tools?: any[]): Promise<string | LLMResponse>;
     generateChat(messages: Message[]): Promise<LLMResponse>;
 }
 
@@ -343,9 +429,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>;
@@ -418,4 +509,4 @@ declare class VectorStoreFactory {
     static create(provider: string, config: VectorStoreConfig): VectorStore;
 }
 
-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 };
+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, type Neo4jConfig, OpenAIEmbedder, OpenAILLM, OpenAIStructuredLLM, Qdrant, RedisDB, type SearchFilters, type SearchMemoryOptions, type SearchResult, type VectorStore, type VectorStoreConfig, VectorStoreFactory, type VectorStoreResult };

package/dist/oss/index.d.ts

@@ -15,11 +15,21 @@ interface VectorStoreConfig {
     [key: string]: any;
 }
 interface LLMConfig {
-    apiKey: string;
+    provider?: string;
+    config?: Record<string, any>;
+    apiKey?: string;
     model?: string;
 }
+interface Neo4jConfig {
+    url: string;
+    username: string;
+    password: string;
+}
 interface GraphStoreConfig {
-    config?: any;
+    provider: string;
+    config: Neo4jConfig;
+    llm?: LLMConfig;
+    customPrompt?: string;
 }
 interface MemoryConfig {
     version?: string;
@@ -38,6 +48,7 @@ interface MemoryConfig {
     historyDbPath?: string;
     customPrompt?: string;
     graphStore?: GraphStoreConfig;
+    enableGraph?: boolean;
 }
 interface MemoryItem {
     id: string;
@@ -146,12 +157,57 @@ declare const MemoryConfigSchema: z.ZodObject<{
     }>;
     historyDbPath: z.ZodOptional<z.ZodString>;
     customPrompt: z.ZodOptional<z.ZodString>;
+    enableGraph: z.ZodOptional<z.ZodBoolean>;
     graphStore: z.ZodOptional<z.ZodObject<{
-        config: z.ZodOptional<z.ZodAny>;
+        provider: z.ZodString;
+        config: z.ZodObject<{
+            url: z.ZodString;
+            username: z.ZodString;
+            password: z.ZodString;
+        }, "strip", z.ZodTypeAny, {
+            url: string;
+            username: string;
+            password: string;
+        }, {
+            url: string;
+            username: string;
+            password: string;
+        }>;
+        llm: z.ZodOptional<z.ZodObject<{
+            provider: z.ZodString;
+            config: z.ZodRecord<z.ZodString, z.ZodAny>;
+        }, "strip", z.ZodTypeAny, {
+            provider: string;
+            config: Record<string, any>;
+        }, {
+            provider: string;
+            config: Record<string, any>;
+        }>>;
+        customPrompt: z.ZodOptional<z.ZodString>;
     }, "strip", z.ZodTypeAny, {
-        config?: any;
+        provider: string;
+        config: {
+            url: string;
+            username: string;
+            password: string;
+        };
+        llm?: {
+            provider: string;
+            config: Record<string, any>;
+        } | undefined;
+        customPrompt?: string | undefined;
     }, {
-        config?: any;
+        provider: string;
+        config: {
+            url: string;
+            username: string;
+            password: string;
+        };
+        llm?: {
+            provider: string;
+            config: Record<string, any>;
+        } | undefined;
+        customPrompt?: string | undefined;
     }>>;
 }, "strip", z.ZodTypeAny, {
     embedder: {
@@ -180,8 +236,19 @@ declare const MemoryConfigSchema: z.ZodObject<{
     version?: string | undefined;
     historyDbPath?: string | undefined;
     customPrompt?: string | undefined;
+    enableGraph?: boolean | undefined;
     graphStore?: {
-        config?: any;
+        provider: string;
+        config: {
+            url: string;
+            username: string;
+            password: string;
+        };
+        llm?: {
+            provider: string;
+            config: Record<string, any>;
+        } | undefined;
+        customPrompt?: string | undefined;
     } | undefined;
 }, {
     embedder: {
@@ -210,8 +277,19 @@ declare const MemoryConfigSchema: z.ZodObject<{
     version?: string | undefined;
     historyDbPath?: string | undefined;
     customPrompt?: string | undefined;
+    enableGraph?: boolean | undefined;
     graphStore?: {
-        config?: any;
+        provider: string;
+        config: {
+            url: string;
+            username: string;
+            password: string;
+        };
+        llm?: {
+            provider: string;
+            config: Record<string, any>;
+        } | undefined;
+        customPrompt?: string | undefined;
     } | undefined;
 }>;
 
@@ -226,7 +304,6 @@ interface AddMemoryOptions extends Entity {
     prompt?: string;
 }
 interface SearchMemoryOptions extends Entity {
-    query: string;
     limit?: number;
     filters?: SearchFilters;
 }
@@ -245,6 +322,8 @@ declare class Memory {
     private db;
     private collectionName;
     private apiVersion;
+    private graphMemory?;
+    private enableGraph;
     constructor(config?: Partial<MemoryConfig>);
     static fromConfig(configDict: Record<string, any>): Memory;
     add(messages: string | Message[], config: AddMemoryOptions): Promise<SearchResult>;
@@ -284,11 +363,18 @@ declare class OpenAIEmbedder implements Embedder {
 interface LLMResponse {
     content: string;
     role: string;
+    toolCalls?: Array<{
+        name: string;
+        arguments: string;
+    }>;
 }
 interface LLM {
-    generateResponse(messages: Message[], responseFormat?: {
+    generateResponse(messages: Array<{
+        role: string;
+        content: string;
+    }>, response_format: {
         type: string;
-    }): Promise<string>;
+    }, tools?: any[]): Promise<any>;
     generateChat(messages: Message[]): Promise<LLMResponse>;
 }
 
@@ -298,17 +384,17 @@ declare class OpenAILLM implements LLM {
     constructor(config: LLMConfig);
     generateResponse(messages: Message[], responseFormat?: {
         type: string;
-    }): Promise<string>;
+    }, tools?: any[]): Promise<string | LLMResponse>;
     generateChat(messages: Message[]): Promise<LLMResponse>;
 }
 
 declare class OpenAIStructuredLLM implements LLM {
-    private client;
+    private openai;
     private model;
     constructor(config: LLMConfig);
     generateResponse(messages: Message[], responseFormat?: {
         type: string;
-    }): Promise<string>;
+    } | null, tools?: any[]): Promise<string | LLMResponse>;
     generateChat(messages: Message[]): Promise<LLMResponse>;
 }
 
@@ -343,9 +429,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>;
@@ -418,4 +509,4 @@ declare class VectorStoreFactory {
     static create(provider: string, config: VectorStoreConfig): VectorStore;
 }
 
-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 };
+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, type Neo4jConfig, OpenAIEmbedder, OpenAILLM, OpenAIStructuredLLM, Qdrant, RedisDB, type SearchFilters, type SearchMemoryOptions, type SearchResult, type VectorStore, type VectorStoreConfig, VectorStoreFactory, type VectorStoreResult };