mem0ai 2.1.38 → 2.2.0

package/dist/oss/index.d.mts

@@ -16,6 +16,7 @@ interface EmbeddingConfig {
     apiKey?: string;
     model?: string | any;
     url?: string;
+    embeddingDims?: number;
     modelProperties?: Record<string, any>;
 }
 interface VectorStoreConfig {
@@ -107,16 +108,22 @@ declare const MemoryConfigSchema: z.ZodObject<{
             apiKey: z.ZodOptional<z.ZodString>;
             model: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodAny]>>;
             baseURL: z.ZodOptional<z.ZodString>;
+            embeddingDims: z.ZodOptional<z.ZodNumber>;
+            url: z.ZodOptional<z.ZodString>;
         }, "strip", z.ZodTypeAny, {
             modelProperties?: Record<string, any> | undefined;
             apiKey?: string | undefined;
             model?: any;
             baseURL?: string | undefined;
+            embeddingDims?: number | undefined;
+            url?: string | undefined;
         }, {
             modelProperties?: Record<string, any> | undefined;
             apiKey?: string | undefined;
             model?: any;
             baseURL?: string | undefined;
+            embeddingDims?: number | undefined;
+            url?: string | undefined;
         }>;
     }, "strip", z.ZodTypeAny, {
         provider: string;
@@ -125,6 +132,8 @@ declare const MemoryConfigSchema: z.ZodObject<{
             apiKey?: string | undefined;
             model?: any;
             baseURL?: string | undefined;
+            embeddingDims?: number | undefined;
+            url?: string | undefined;
         };
     }, {
         provider: string;
@@ -133,6 +142,8 @@ declare const MemoryConfigSchema: z.ZodObject<{
             apiKey?: string | undefined;
             model?: any;
             baseURL?: string | undefined;
+            embeddingDims?: number | undefined;
+            url?: string | undefined;
         };
     }>;
     vectorStore: z.ZodObject<{
@@ -277,6 +288,8 @@ declare const MemoryConfigSchema: z.ZodObject<{
             apiKey?: string | undefined;
             model?: any;
             baseURL?: string | undefined;
+            embeddingDims?: number | undefined;
+            url?: string | undefined;
         };
     };
     vectorStore: {
@@ -328,6 +341,8 @@ declare const MemoryConfigSchema: z.ZodObject<{
             apiKey?: string | undefined;
             model?: any;
             baseURL?: string | undefined;
+            embeddingDims?: number | undefined;
+            url?: string | undefined;
         };
     };
     vectorStore: {
@@ -439,6 +454,7 @@ interface Embedder {
 declare class OpenAIEmbedder implements Embedder {
     private openai;
     private model;
+    private embeddingDims?;
     constructor(config: EmbeddingConfig);
     embed(text: string): Promise<number[]>;
     embedBatch(texts: string[]): Promise<number[][]>;
@@ -447,6 +463,7 @@ declare class OpenAIEmbedder implements Embedder {
 declare class OllamaEmbedder implements Embedder {
     private ollama;
     private model;
+    private embeddingDims?;
     private initialized;
     constructor(config: EmbeddingConfig);
     embed(text: string): Promise<number[]>;
@@ -457,6 +474,7 @@ declare class OllamaEmbedder implements Embedder {
 declare class GoogleEmbedder implements Embedder {
     private google;
     private model;
+    private embeddingDims?;
     constructor(config: EmbeddingConfig);
     embed(text: string): Promise<number[]>;
     embedBatch(texts: string[]): Promise<number[][]>;
@@ -465,6 +483,7 @@ declare class GoogleEmbedder implements Embedder {
 declare class AzureOpenAIEmbedder implements Embedder {
     private client;
     private model;
+    private embeddingDims?;
     constructor(config: EmbeddingConfig);
     embed(text: string): Promise<number[]>;
     embedBatch(texts: string[]): Promise<number[][]>;
@@ -742,6 +761,144 @@ declare class VectorizeDB implements VectorStore {
     initialize(): Promise<void>;
 }
 
+/**
+ * Configuration interface for Azure AI Search vector store
+ */
+interface AzureAISearchConfig extends VectorStoreConfig {
+    /**
+     * Azure AI Search service name (e.g., "my-search-service")
+     */
+    serviceName: string;
+    /**
+     * Index/collection name to use
+     */
+    collectionName: string;
+    /**
+     * API key for authentication (if not provided, uses DefaultAzureCredential)
+     */
+    apiKey?: string;
+    /**
+     * Vector embedding dimensions
+     */
+    embeddingModelDims: number;
+    /**
+     * Compression type: 'none', 'scalar', or 'binary'
+     * @default 'none'
+     */
+    compressionType?: "none" | "scalar" | "binary";
+    /**
+     * Use half precision (float16) instead of full precision (float32)
+     * @default false
+     */
+    useFloat16?: boolean;
+    /**
+     * Enable hybrid search (combines vector + text search)
+     * @default false
+     */
+    hybridSearch?: boolean;
+    /**
+     * Vector filter mode: 'preFilter' or 'postFilter'
+     * @default 'preFilter'
+     */
+    vectorFilterMode?: string;
+}
+/**
+ * Azure AI Search vector store implementation
+ * Supports vector search with hybrid search, compression, and filtering
+ */
+declare class AzureAISearch implements VectorStore {
+    private searchClient;
+    private indexClient;
+    private readonly serviceName;
+    private readonly indexName;
+    private readonly embeddingModelDims;
+    private readonly compressionType;
+    private readonly useFloat16;
+    private readonly hybridSearch;
+    private readonly vectorFilterMode;
+    private readonly apiKey;
+    constructor(config: AzureAISearchConfig);
+    /**
+     * Initialize the Azure AI Search index if it doesn't exist
+     */
+    initialize(): Promise<void>;
+    /**
+     * Create a new index in Azure AI Search
+     */
+    private createCol;
+    /**
+     * Generate a document for insertion
+     */
+    private generateDocument;
+    /**
+     * Insert vectors into the index
+     */
+    insert(vectors: number[][], ids: string[], payloads: Record<string, any>[]): Promise<void>;
+    /**
+     * Sanitize filter keys to remove non-alphanumeric characters
+     */
+    private sanitizeKey;
+    /**
+     * Build OData filter expression from SearchFilters
+     */
+    private buildFilterExpression;
+    /**
+     * Extract JSON from payload string
+     * Handles cases where payload might have extra text
+     */
+    private extractJson;
+    /**
+     * Search for similar vectors
+     */
+    search(query: number[], limit?: number, filters?: SearchFilters): Promise<VectorStoreResult[]>;
+    /**
+     * Delete a vector by ID
+     */
+    delete(vectorId: string): Promise<void>;
+    /**
+     * Update a vector and its payload
+     */
+    update(vectorId: string, vector: number[], payload: Record<string, any>): Promise<void>;
+    /**
+     * Retrieve a vector by ID
+     */
+    get(vectorId: string): Promise<VectorStoreResult | null>;
+    /**
+     * List all collections (indexes)
+     */
+    private listCols;
+    /**
+     * Delete the index
+     */
+    deleteCol(): Promise<void>;
+    /**
+     * Get information about the index
+     */
+    private colInfo;
+    /**
+     * List all vectors in the index
+     */
+    list(filters?: SearchFilters, limit?: number): Promise<[VectorStoreResult[], number]>;
+    /**
+     * Generate a random user ID
+     */
+    private generateUUID;
+    /**
+     * Get user ID from memory_migrations collection
+     * Required by VectorStore interface
+     */
+    getUserId(): Promise<string>;
+    /**
+     * Set user ID in memory_migrations collection
+     * Required by VectorStore interface
+     */
+    setUserId(userId: string): Promise<void>;
+    /**
+     * Reset the index by deleting and recreating it
+     */
+    reset(): Promise<void>;
+}
+
 interface HistoryManager {
     addHistory(memoryId: string, previousValue: string | null, newValue: string | null, action: string, createdAt?: string, updatedAt?: string, isDeleted?: number): Promise<void>;
     getHistory(memoryId: string): Promise<any[]>;
@@ -762,4 +919,4 @@ declare class HistoryManagerFactory {
     static create(provider: string, config: HistoryStoreConfig): HistoryManager;
 }
 
-export { type AddMemoryOptions, AnthropicLLM, AzureOpenAIEmbedder, type DeleteAllMemoryOptions, type Embedder, EmbedderFactory, type EmbeddingConfig, type Entity, type GetAllMemoryOptions, GoogleEmbedder, GoogleLLM, type GraphStoreConfig, GroqLLM, HistoryManagerFactory, type HistoryStoreConfig, type LLM, type LLMConfig, LLMFactory, type LLMResponse, LangchainEmbedder, LangchainLLM, LangchainVectorStore, Memory, type MemoryConfig, MemoryConfigSchema, type MemoryItem, MemoryVectorStore, type Message, MistralLLM, type MultiModalMessages, type Neo4jConfig, OllamaEmbedder, OllamaLLM, OpenAIEmbedder, OpenAILLM, OpenAIStructuredLLM, Qdrant, RedisDB, type SearchFilters, type SearchMemoryOptions, type SearchResult, SupabaseDB, type VectorStore, type VectorStoreConfig, VectorStoreFactory, type VectorStoreResult, VectorizeDB };
+export { type AddMemoryOptions, AnthropicLLM, AzureAISearch, AzureOpenAIEmbedder, type DeleteAllMemoryOptions, type Embedder, EmbedderFactory, type EmbeddingConfig, type Entity, type GetAllMemoryOptions, GoogleEmbedder, GoogleLLM, type GraphStoreConfig, GroqLLM, HistoryManagerFactory, type HistoryStoreConfig, type LLM, type LLMConfig, LLMFactory, type LLMResponse, LangchainEmbedder, LangchainLLM, LangchainVectorStore, Memory, type MemoryConfig, MemoryConfigSchema, type MemoryItem, MemoryVectorStore, type Message, MistralLLM, type MultiModalMessages, type Neo4jConfig, OllamaEmbedder, OllamaLLM, OpenAIEmbedder, OpenAILLM, OpenAIStructuredLLM, Qdrant, RedisDB, type SearchFilters, type SearchMemoryOptions, type SearchResult, SupabaseDB, type VectorStore, type VectorStoreConfig, VectorStoreFactory, type VectorStoreResult, VectorizeDB };

package/dist/oss/index.d.ts

@@ -16,6 +16,7 @@ interface EmbeddingConfig {
     apiKey?: string;
     model?: string | any;
     url?: string;
+    embeddingDims?: number;
     modelProperties?: Record<string, any>;
 }
 interface VectorStoreConfig {
@@ -107,16 +108,22 @@ declare const MemoryConfigSchema: z.ZodObject<{
             apiKey: z.ZodOptional<z.ZodString>;
             model: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodAny]>>;
             baseURL: z.ZodOptional<z.ZodString>;
+            embeddingDims: z.ZodOptional<z.ZodNumber>;
+            url: z.ZodOptional<z.ZodString>;
         }, "strip", z.ZodTypeAny, {
             modelProperties?: Record<string, any> | undefined;
             apiKey?: string | undefined;
             model?: any;
             baseURL?: string | undefined;
+            embeddingDims?: number | undefined;
+            url?: string | undefined;
         }, {
             modelProperties?: Record<string, any> | undefined;
             apiKey?: string | undefined;
             model?: any;
             baseURL?: string | undefined;
+            embeddingDims?: number | undefined;
+            url?: string | undefined;
         }>;
     }, "strip", z.ZodTypeAny, {
         provider: string;
@@ -125,6 +132,8 @@ declare const MemoryConfigSchema: z.ZodObject<{
             apiKey?: string | undefined;
             model?: any;
             baseURL?: string | undefined;
+            embeddingDims?: number | undefined;
+            url?: string | undefined;
         };
     }, {
         provider: string;
@@ -133,6 +142,8 @@ declare const MemoryConfigSchema: z.ZodObject<{
             apiKey?: string | undefined;
             model?: any;
             baseURL?: string | undefined;
+            embeddingDims?: number | undefined;
+            url?: string | undefined;
         };
     }>;
     vectorStore: z.ZodObject<{
@@ -277,6 +288,8 @@ declare const MemoryConfigSchema: z.ZodObject<{
             apiKey?: string | undefined;
             model?: any;
             baseURL?: string | undefined;
+            embeddingDims?: number | undefined;
+            url?: string | undefined;
         };
     };
     vectorStore: {
@@ -328,6 +341,8 @@ declare const MemoryConfigSchema: z.ZodObject<{
             apiKey?: string | undefined;
             model?: any;
             baseURL?: string | undefined;
+            embeddingDims?: number | undefined;
+            url?: string | undefined;
         };
     };
     vectorStore: {
@@ -439,6 +454,7 @@ interface Embedder {
 declare class OpenAIEmbedder implements Embedder {
     private openai;
     private model;
+    private embeddingDims?;
     constructor(config: EmbeddingConfig);
     embed(text: string): Promise<number[]>;
     embedBatch(texts: string[]): Promise<number[][]>;
@@ -447,6 +463,7 @@ declare class OpenAIEmbedder implements Embedder {
 declare class OllamaEmbedder implements Embedder {
     private ollama;
     private model;
+    private embeddingDims?;
     private initialized;
     constructor(config: EmbeddingConfig);
     embed(text: string): Promise<number[]>;
@@ -457,6 +474,7 @@ declare class OllamaEmbedder implements Embedder {
 declare class GoogleEmbedder implements Embedder {
     private google;
     private model;
+    private embeddingDims?;
     constructor(config: EmbeddingConfig);
     embed(text: string): Promise<number[]>;
     embedBatch(texts: string[]): Promise<number[][]>;
@@ -465,6 +483,7 @@ declare class GoogleEmbedder implements Embedder {
 declare class AzureOpenAIEmbedder implements Embedder {
     private client;
     private model;
+    private embeddingDims?;
     constructor(config: EmbeddingConfig);
     embed(text: string): Promise<number[]>;
     embedBatch(texts: string[]): Promise<number[][]>;
@@ -742,6 +761,144 @@ declare class VectorizeDB implements VectorStore {
     initialize(): Promise<void>;
 }
 
+/**
+ * Configuration interface for Azure AI Search vector store
+ */
+interface AzureAISearchConfig extends VectorStoreConfig {
+    /**
+     * Azure AI Search service name (e.g., "my-search-service")
+     */
+    serviceName: string;
+    /**
+     * Index/collection name to use
+     */
+    collectionName: string;
+    /**
+     * API key for authentication (if not provided, uses DefaultAzureCredential)
+     */
+    apiKey?: string;
+    /**
+     * Vector embedding dimensions
+     */
+    embeddingModelDims: number;
+    /**
+     * Compression type: 'none', 'scalar', or 'binary'
+     * @default 'none'
+     */
+    compressionType?: "none" | "scalar" | "binary";
+    /**
+     * Use half precision (float16) instead of full precision (float32)
+     * @default false
+     */
+    useFloat16?: boolean;
+    /**
+     * Enable hybrid search (combines vector + text search)
+     * @default false
+     */
+    hybridSearch?: boolean;
+    /**
+     * Vector filter mode: 'preFilter' or 'postFilter'
+     * @default 'preFilter'
+     */
+    vectorFilterMode?: string;
+}
+/**
+ * Azure AI Search vector store implementation
+ * Supports vector search with hybrid search, compression, and filtering
+ */
+declare class AzureAISearch implements VectorStore {
+    private searchClient;
+    private indexClient;
+    private readonly serviceName;
+    private readonly indexName;
+    private readonly embeddingModelDims;
+    private readonly compressionType;
+    private readonly useFloat16;
+    private readonly hybridSearch;
+    private readonly vectorFilterMode;
+    private readonly apiKey;
+    constructor(config: AzureAISearchConfig);
+    /**
+     * Initialize the Azure AI Search index if it doesn't exist
+     */
+    initialize(): Promise<void>;
+    /**
+     * Create a new index in Azure AI Search
+     */
+    private createCol;
+    /**
+     * Generate a document for insertion
+     */
+    private generateDocument;
+    /**
+     * Insert vectors into the index
+     */
+    insert(vectors: number[][], ids: string[], payloads: Record<string, any>[]): Promise<void>;
+    /**
+     * Sanitize filter keys to remove non-alphanumeric characters
+     */
+    private sanitizeKey;
+    /**
+     * Build OData filter expression from SearchFilters
+     */
+    private buildFilterExpression;
+    /**
+     * Extract JSON from payload string
+     * Handles cases where payload might have extra text
+     */
+    private extractJson;
+    /**
+     * Search for similar vectors
+     */
+    search(query: number[], limit?: number, filters?: SearchFilters): Promise<VectorStoreResult[]>;
+    /**
+     * Delete a vector by ID
+     */
+    delete(vectorId: string): Promise<void>;
+    /**
+     * Update a vector and its payload
+     */
+    update(vectorId: string, vector: number[], payload: Record<string, any>): Promise<void>;
+    /**
+     * Retrieve a vector by ID
+     */
+    get(vectorId: string): Promise<VectorStoreResult | null>;
+    /**
+     * List all collections (indexes)
+     */
+    private listCols;
+    /**
+     * Delete the index
+     */
+    deleteCol(): Promise<void>;
+    /**
+     * Get information about the index
+     */
+    private colInfo;
+    /**
+     * List all vectors in the index
+     */
+    list(filters?: SearchFilters, limit?: number): Promise<[VectorStoreResult[], number]>;
+    /**
+     * Generate a random user ID
+     */
+    private generateUUID;
+    /**
+     * Get user ID from memory_migrations collection
+     * Required by VectorStore interface
+     */
+    getUserId(): Promise<string>;
+    /**
+     * Set user ID in memory_migrations collection
+     * Required by VectorStore interface
+     */
+    setUserId(userId: string): Promise<void>;
+    /**
+     * Reset the index by deleting and recreating it
+     */
+    reset(): Promise<void>;
+}
+
 interface HistoryManager {
     addHistory(memoryId: string, previousValue: string | null, newValue: string | null, action: string, createdAt?: string, updatedAt?: string, isDeleted?: number): Promise<void>;
     getHistory(memoryId: string): Promise<any[]>;
@@ -762,4 +919,4 @@ declare class HistoryManagerFactory {
     static create(provider: string, config: HistoryStoreConfig): HistoryManager;
 }
 
-export { type AddMemoryOptions, AnthropicLLM, AzureOpenAIEmbedder, type DeleteAllMemoryOptions, type Embedder, EmbedderFactory, type EmbeddingConfig, type Entity, type GetAllMemoryOptions, GoogleEmbedder, GoogleLLM, type GraphStoreConfig, GroqLLM, HistoryManagerFactory, type HistoryStoreConfig, type LLM, type LLMConfig, LLMFactory, type LLMResponse, LangchainEmbedder, LangchainLLM, LangchainVectorStore, Memory, type MemoryConfig, MemoryConfigSchema, type MemoryItem, MemoryVectorStore, type Message, MistralLLM, type MultiModalMessages, type Neo4jConfig, OllamaEmbedder, OllamaLLM, OpenAIEmbedder, OpenAILLM, OpenAIStructuredLLM, Qdrant, RedisDB, type SearchFilters, type SearchMemoryOptions, type SearchResult, SupabaseDB, type VectorStore, type VectorStoreConfig, VectorStoreFactory, type VectorStoreResult, VectorizeDB };
+export { type AddMemoryOptions, AnthropicLLM, AzureAISearch, AzureOpenAIEmbedder, type DeleteAllMemoryOptions, type Embedder, EmbedderFactory, type EmbeddingConfig, type Entity, type GetAllMemoryOptions, GoogleEmbedder, GoogleLLM, type GraphStoreConfig, GroqLLM, HistoryManagerFactory, type HistoryStoreConfig, type LLM, type LLMConfig, LLMFactory, type LLMResponse, LangchainEmbedder, LangchainLLM, LangchainVectorStore, Memory, type MemoryConfig, MemoryConfigSchema, type MemoryItem, MemoryVectorStore, type Message, MistralLLM, type MultiModalMessages, type Neo4jConfig, OllamaEmbedder, OllamaLLM, OpenAIEmbedder, OpenAILLM, OpenAIStructuredLLM, Qdrant, RedisDB, type SearchFilters, type SearchMemoryOptions, type SearchResult, SupabaseDB, type VectorStore, type VectorStoreConfig, VectorStoreFactory, type VectorStoreResult, VectorizeDB };