mem0ai 1.0.39 → 2.0.1

package/dist/oss/index.d.mts

@@ -0,0 +1,421 @@
+import { z } from 'zod';
+import { QdrantClient } from '@qdrant/js-client-rest';
+
+interface Message {
+    role: string;
+    content: string;
+}
+interface EmbeddingConfig {
+    apiKey: string;
+    model?: string;
+}
+interface VectorStoreConfig {
+    collectionName: string;
+    dimension?: number;
+    [key: string]: any;
+}
+interface LLMConfig {
+    apiKey: string;
+    model?: string;
+}
+interface GraphStoreConfig {
+    config?: any;
+}
+interface MemoryConfig {
+    version?: string;
+    embedder: {
+        provider: string;
+        config: EmbeddingConfig;
+    };
+    vectorStore: {
+        provider: string;
+        config: VectorStoreConfig;
+    };
+    llm: {
+        provider: string;
+        config: LLMConfig;
+    };
+    historyDbPath?: string;
+    customPrompt?: string;
+    graphStore?: GraphStoreConfig;
+}
+interface MemoryItem {
+    id: string;
+    memory: string;
+    hash?: string;
+    createdAt?: string;
+    updatedAt?: string;
+    score?: number;
+    metadata?: Record<string, any>;
+}
+interface SearchFilters {
+    userId?: string;
+    agentId?: string;
+    runId?: string;
+    [key: string]: any;
+}
+interface SearchResult {
+    results: MemoryItem[];
+    relations?: any[];
+}
+interface VectorStoreResult {
+    id: string;
+    payload: Record<string, any>;
+    score?: number;
+}
+declare const MemoryConfigSchema: z.ZodObject<{
+    version: z.ZodOptional<z.ZodString>;
+    embedder: z.ZodObject<{
+        provider: z.ZodString;
+        config: z.ZodObject<{
+            apiKey: z.ZodString;
+            model: z.ZodOptional<z.ZodString>;
+        }, "strip", z.ZodTypeAny, {
+            apiKey: string;
+            model?: string | undefined;
+        }, {
+            apiKey: string;
+            model?: string | undefined;
+        }>;
+    }, "strip", z.ZodTypeAny, {
+        provider: string;
+        config: {
+            apiKey: string;
+            model?: string | undefined;
+        };
+    }, {
+        provider: string;
+        config: {
+            apiKey: string;
+            model?: string | undefined;
+        };
+    }>;
+    vectorStore: z.ZodObject<{
+        provider: z.ZodString;
+        config: z.ZodObject<{
+            collectionName: z.ZodString;
+            dimension: z.ZodOptional<z.ZodNumber>;
+        }, "passthrough", z.ZodTypeAny, z.objectOutputType<{
+            collectionName: z.ZodString;
+            dimension: z.ZodOptional<z.ZodNumber>;
+        }, z.ZodTypeAny, "passthrough">, z.objectInputType<{
+            collectionName: z.ZodString;
+            dimension: z.ZodOptional<z.ZodNumber>;
+        }, z.ZodTypeAny, "passthrough">>;
+    }, "strip", z.ZodTypeAny, {
+        provider: string;
+        config: {
+            collectionName: string;
+            dimension?: number | undefined;
+        } & {
+            [k: string]: unknown;
+        };
+    }, {
+        provider: string;
+        config: {
+            collectionName: string;
+            dimension?: number | undefined;
+        } & {
+            [k: string]: unknown;
+        };
+    }>;
+    llm: z.ZodObject<{
+        provider: z.ZodString;
+        config: z.ZodObject<{
+            apiKey: z.ZodString;
+            model: z.ZodOptional<z.ZodString>;
+        }, "strip", z.ZodTypeAny, {
+            apiKey: string;
+            model?: string | undefined;
+        }, {
+            apiKey: string;
+            model?: string | undefined;
+        }>;
+    }, "strip", z.ZodTypeAny, {
+        provider: string;
+        config: {
+            apiKey: string;
+            model?: string | undefined;
+        };
+    }, {
+        provider: string;
+        config: {
+            apiKey: string;
+            model?: string | undefined;
+        };
+    }>;
+    historyDbPath: z.ZodOptional<z.ZodString>;
+    customPrompt: z.ZodOptional<z.ZodString>;
+    graphStore: z.ZodOptional<z.ZodObject<{
+        config: z.ZodOptional<z.ZodAny>;
+    }, "strip", z.ZodTypeAny, {
+        config?: any;
+    }, {
+        config?: any;
+    }>>;
+}, "strip", z.ZodTypeAny, {
+    embedder: {
+        provider: string;
+        config: {
+            apiKey: string;
+            model?: string | undefined;
+        };
+    };
+    vectorStore: {
+        provider: string;
+        config: {
+            collectionName: string;
+            dimension?: number | undefined;
+        } & {
+            [k: string]: unknown;
+        };
+    };
+    llm: {
+        provider: string;
+        config: {
+            apiKey: string;
+            model?: string | undefined;
+        };
+    };
+    version?: string | undefined;
+    historyDbPath?: string | undefined;
+    customPrompt?: string | undefined;
+    graphStore?: {
+        config?: any;
+    } | undefined;
+}, {
+    embedder: {
+        provider: string;
+        config: {
+            apiKey: string;
+            model?: string | undefined;
+        };
+    };
+    vectorStore: {
+        provider: string;
+        config: {
+            collectionName: string;
+            dimension?: number | undefined;
+        } & {
+            [k: string]: unknown;
+        };
+    };
+    llm: {
+        provider: string;
+        config: {
+            apiKey: string;
+            model?: string | undefined;
+        };
+    };
+    version?: string | undefined;
+    historyDbPath?: string | undefined;
+    customPrompt?: string | undefined;
+    graphStore?: {
+        config?: any;
+    } | 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;
+    private embedder;
+    private vectorStore;
+    private llm;
+    private db;
+    private collectionName;
+    private apiVersion;
+    constructor(config?: Partial<MemoryConfig>);
+    static fromConfig(configDict: Record<string, any>): Memory;
+    add(messages: string | Message[], config: AddMemoryOptions): Promise<SearchResult>;
+    private addToVectorStore;
+    get(memoryId: string): Promise<MemoryItem | null>;
+    search(query: string, config: SearchMemoryOptions): Promise<SearchResult>;
+    update(memoryId: string, data: string): Promise<{
+        message: string;
+    }>;
+    delete(memoryId: string): Promise<{
+        message: string;
+    }>;
+    deleteAll(config: DeleteAllMemoryOptions): Promise<{
+        message: string;
+    }>;
+    history(memoryId: string): Promise<any[]>;
+    reset(): Promise<void>;
+    getAll(config: GetAllMemoryOptions): Promise<SearchResult>;
+    private createMemory;
+    private updateMemory;
+    private deleteMemory;
+}
+
+interface Embedder {
+    embed(text: string): Promise<number[]>;
+    embedBatch(texts: string[]): Promise<number[][]>;
+}
+
+declare class OpenAIEmbedder implements Embedder {
+    private openai;
+    private model;
+    constructor(config: EmbeddingConfig);
+    embed(text: string): Promise<number[]>;
+    embedBatch(texts: string[]): Promise<number[][]>;
+}
+
+interface LLMResponse {
+    content: string;
+    role: string;
+}
+interface LLM {
+    generateResponse(messages: Message[], responseFormat?: {
+        type: string;
+    }): Promise<string>;
+    generateChat(messages: Message[]): Promise<LLMResponse>;
+}
+
+declare class OpenAILLM implements LLM {
+    private openai;
+    private model;
+    constructor(config: LLMConfig);
+    generateResponse(messages: Message[], responseFormat?: {
+        type: string;
+    }): Promise<string>;
+    generateChat(messages: Message[]): Promise<LLMResponse>;
+}
+
+declare class OpenAIStructuredLLM implements LLM {
+    private client;
+    private model;
+    constructor(config: LLMConfig);
+    generateResponse(messages: Message[], responseFormat?: {
+        type: string;
+    }): Promise<string>;
+    generateChat(messages: Message[]): Promise<LLMResponse>;
+}
+
+declare class AnthropicLLM implements LLM {
+    private client;
+    private model;
+    constructor(config: LLMConfig);
+    generateResponse(messages: Message[], responseFormat?: {
+        type: string;
+    }): Promise<string>;
+    generateChat(messages: Message[]): Promise<LLMResponse>;
+}
+
+declare class GroqLLM implements LLM {
+    private client;
+    private model;
+    constructor(config: LLMConfig);
+    generateResponse(messages: Message[], responseFormat?: {
+        type: string;
+    }): Promise<string>;
+    generateChat(messages: Message[]): Promise<LLMResponse>;
+}
+
+interface VectorStore {
+    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]>;
+}
+
+declare class MemoryVectorStore implements VectorStore {
+    private vectors;
+    private dimension;
+    constructor(config: VectorStoreConfig);
+    private cosineSimilarity;
+    private filterVector;
+    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 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;
+}
+declare class LLMFactory {
+    static create(provider: string, config: LLMConfig): LLM;
+}
+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 };