mem0ai 2.1.15 → 2.1.16-patch.1

package/dist/oss/index.d.ts

@@ -35,6 +35,7 @@ interface LLMConfig {
     config?: Record<string, any>;
     apiKey?: string;
     model?: string;
+    modelProperties?: Record<string, any>;
 }
 interface Neo4jConfig {
     url: string;
@@ -153,24 +154,29 @@ declare const MemoryConfigSchema: z.ZodObject<{
         config: z.ZodObject<{
             apiKey: z.ZodString;
             model: z.ZodOptional<z.ZodString>;
+            modelProperties: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodAny>>;
         }, "strip", z.ZodTypeAny, {
             apiKey: string;
             model?: string | undefined;
+            modelProperties?: Record<string, any> | undefined;
         }, {
             apiKey: string;
             model?: string | undefined;
+            modelProperties?: Record<string, any> | undefined;
         }>;
     }, "strip", z.ZodTypeAny, {
         provider: string;
         config: {
             apiKey: string;
             model?: string | undefined;
+            modelProperties?: Record<string, any> | undefined;
         };
     }, {
         provider: string;
         config: {
             apiKey: string;
             model?: string | undefined;
+            modelProperties?: Record<string, any> | undefined;
         };
     }>;
     historyDbPath: z.ZodOptional<z.ZodString>;
@@ -260,6 +266,7 @@ declare const MemoryConfigSchema: z.ZodObject<{
         config: {
             apiKey: string;
             model?: string | undefined;
+            modelProperties?: Record<string, any> | undefined;
         };
     };
     version?: string | undefined;
@@ -306,6 +313,7 @@ declare const MemoryConfigSchema: z.ZodObject<{
         config: {
             apiKey: string;
             model?: string | undefined;
+            modelProperties?: Record<string, any> | undefined;
         };
     };
     version?: string | undefined;
@@ -501,6 +509,17 @@ declare class OllamaLLM implements LLM {
     private ensureModelExists;
 }
 
+declare class MistralLLM implements LLM {
+    private client;
+    private model;
+    constructor(config: LLMConfig);
+    private contentToString;
+    generateResponse(messages: Message[], responseFormat?: {
+        type: string;
+    }, tools?: any[]): Promise<string | LLMResponse>;
+    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[]>;
@@ -615,4 +634,4 @@ declare class HistoryManagerFactory {
     static create(provider: string, config: HistoryStoreConfig): HistoryManager;
 }
 
-export { type AddMemoryOptions, AnthropicLLM, 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, Memory, type MemoryConfig, MemoryConfigSchema, type MemoryItem, MemoryVectorStore, type Message, type MultiModalMessages, type Neo4jConfig, OllamaEmbedder, OllamaLLM, 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, GoogleEmbedder, GoogleLLM, type GraphStoreConfig, GroqLLM, HistoryManagerFactory, type HistoryStoreConfig, type LLM, type LLMConfig, LLMFactory, type LLMResponse, 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, type VectorStore, type VectorStoreConfig, VectorStoreFactory, type VectorStoreResult };

package/dist/oss/index.js

@@ -40,6 +40,7 @@ __export(index_exports, {
   Memory: () => Memory,
   MemoryConfigSchema: () => MemoryConfigSchema,
   MemoryVectorStore: () => MemoryVectorStore,
+  MistralLLM: () => MistralLLM,
   OllamaEmbedder: () => OllamaEmbedder,
   OllamaLLM: () => OllamaLLM,
   OpenAIEmbedder: () => OpenAIEmbedder,
@@ -77,7 +78,8 @@ var MemoryConfigSchema = import_zod.z.object({
     provider: import_zod.z.string(),
     config: import_zod.z.object({
       apiKey: import_zod.z.string(),
-      model: import_zod.z.string().optional()
+      model: import_zod.z.string().optional(),
+      modelProperties: import_zod.z.record(import_zod.z.string(), import_zod.z.any()).optional()
     })
   }),
   historyDbPath: import_zod.z.string().optional(),
@@ -363,6 +365,86 @@ var GroqLLM = class {
   }
 };
 
+// src/oss/src/llms/mistral.ts
+var import_mistralai = require("@mistralai/mistralai");
+var MistralLLM = class {
+  constructor(config) {
+    if (!config.apiKey) {
+      throw new Error("Mistral API key is required");
+    }
+    this.client = new import_mistralai.Mistral({
+      apiKey: config.apiKey
+    });
+    this.model = config.model || "mistral-tiny-latest";
+  }
+  // Helper function to convert content to string
+  contentToString(content) {
+    if (typeof content === "string") {
+      return content;
+    }
+    if (Array.isArray(content)) {
+      return content.map((chunk) => {
+        if (chunk.type === "text") {
+          return chunk.text;
+        } else {
+          return JSON.stringify(chunk);
+        }
+      }).join("");
+    }
+    return String(content || "");
+  }
+  async generateResponse(messages, responseFormat, tools) {
+    const response = await this.client.chat.complete({
+      model: this.model,
+      messages: messages.map((msg) => ({
+        role: msg.role,
+        content: typeof msg.content === "string" ? msg.content : JSON.stringify(msg.content)
+      })),
+      ...tools && { tools },
+      ...responseFormat && { response_format: responseFormat }
+    });
+    if (!response || !response.choices || response.choices.length === 0) {
+      return "";
+    }
+    const message = response.choices[0].message;
+    if (!message) {
+      return "";
+    }
+    if (message.toolCalls && message.toolCalls.length > 0) {
+      return {
+        content: this.contentToString(message.content),
+        role: message.role || "assistant",
+        toolCalls: message.toolCalls.map((call) => ({
+          name: call.function.name,
+          arguments: typeof call.function.arguments === "string" ? call.function.arguments : JSON.stringify(call.function.arguments)
+        }))
+      };
+    }
+    return this.contentToString(message.content);
+  }
+  async generateChat(messages) {
+    const formattedMessages = messages.map((msg) => ({
+      role: msg.role,
+      content: typeof msg.content === "string" ? msg.content : JSON.stringify(msg.content)
+    }));
+    const response = await this.client.chat.complete({
+      model: this.model,
+      messages: formattedMessages
+    });
+    if (!response || !response.choices || response.choices.length === 0) {
+      return {
+        content: "",
+        role: "assistant"
+      };
+    }
+    const message = response.choices[0].message;
+    return {
+      content: this.contentToString(message.content),
+      role: message.role || "assistant"
+    };
+  }
+};
+
 // src/oss/src/vector_stores/memory.ts
 var import_sqlite3 = __toESM(require("sqlite3"));
 var import_path = __toESM(require("path"));
@@ -1823,6 +1905,67 @@ var GoogleLLM = class {
   }
 };
 
+// src/oss/src/llms/azure.ts
+var import_openai4 = require("openai");
+var AzureOpenAILLM = class {
+  constructor(config) {
+    var _a2;
+    if (!config.apiKey || !((_a2 = config.modelProperties) == null ? void 0 : _a2.endpoint)) {
+      throw new Error("Azure OpenAI requires both API key and endpoint");
+    }
+    const { endpoint, ...rest } = config.modelProperties;
+    this.client = new import_openai4.AzureOpenAI({
+      apiKey: config.apiKey,
+      endpoint,
+      ...rest
+    });
+    this.model = config.model || "gpt-4";
+  }
+  async generateResponse(messages, responseFormat, tools) {
+    const completion = await this.client.chat.completions.create({
+      messages: messages.map((msg) => {
+        const role = msg.role;
+        return {
+          role,
+          content: typeof msg.content === "string" ? msg.content : JSON.stringify(msg.content)
+        };
+      }),
+      model: this.model,
+      response_format: responseFormat,
+      ...tools && { tools, tool_choice: "auto" }
+    });
+    const response = completion.choices[0].message;
+    if (response.tool_calls) {
+      return {
+        content: response.content || "",
+        role: response.role,
+        toolCalls: response.tool_calls.map((call) => ({
+          name: call.function.name,
+          arguments: call.function.arguments
+        }))
+      };
+    }
+    return response.content || "";
+  }
+  async generateChat(messages) {
+    const completion = await this.client.chat.completions.create({
+      messages: messages.map((msg) => {
+        const role = msg.role;
+        return {
+          role,
+          content: typeof msg.content === "string" ? msg.content : JSON.stringify(msg.content)
+        };
+      }),
+      model: this.model
+    });
+    const response = completion.choices[0].message;
+    return {
+      content: response.content || "",
+      role: response.role
+    };
+  }
+};
+
 // src/oss/src/utils/factory.ts
 var EmbedderFactory = class {
   static create(provider, config) {
@@ -1853,6 +1996,10 @@ var LLMFactory = class {
         return new OllamaLLM(config);
       case "google":
         return new GoogleLLM(config);
+      case "azure_openai":
+        return new AzureOpenAILLM(config);
+      case "mistral":
+        return new MistralLLM(config);
       default:
         throw new Error(`Unsupported LLM provider: ${provider}`);
     }
@@ -2196,7 +2343,7 @@ var DEFAULT_MEMORY_CONFIG = {
 // src/oss/src/config/manager.ts
 var ConfigManager = class {
   static mergeConfig(userConfig = {}) {
-    var _a2, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o, _p;
+    var _a2, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o, _p, _q, _r;
     const mergedConfig = {
       version: userConfig.version || DEFAULT_MEMORY_CONFIG.version,
       embedder: {
@@ -2218,7 +2365,8 @@ var ConfigManager = class {
         provider: ((_l = userConfig.llm) == null ? void 0 : _l.provider) || DEFAULT_MEMORY_CONFIG.llm.provider,
         config: {
           apiKey: ((_n = (_m = userConfig.llm) == null ? void 0 : _m.config) == null ? void 0 : _n.apiKey) || DEFAULT_MEMORY_CONFIG.llm.config.apiKey,
-          model: ((_p = (_o = userConfig.llm) == null ? void 0 : _o.config) == null ? void 0 : _p.model) || DEFAULT_MEMORY_CONFIG.llm.config.model
+          model: ((_p = (_o = userConfig.llm) == null ? void 0 : _o.config) == null ? void 0 : _p.model) || DEFAULT_MEMORY_CONFIG.llm.config.model,
+          modelProperties: ((_r = (_q = userConfig.llm) == null ? void 0 : _q.config) == null ? void 0 : _r.modelProperties) || DEFAULT_MEMORY_CONFIG.llm.config.modelProperties
         }
       },
       historyDbPath: userConfig.historyDbPath || DEFAULT_MEMORY_CONFIG.historyDbPath,
@@ -2994,7 +3142,7 @@ var parse_vision_messages = async (messages) => {
 };
 
 // src/oss/src/utils/telemetry.ts
-var version = "2.1.15";
+var version = "2.1.16";
 var MEM0_TELEMETRY = true;
 var _a;
 try {
@@ -3561,6 +3709,7 @@ ${parsedMessages}`] : getFactRetrievalMessages(parsedMessages);
   Memory,
   MemoryConfigSchema,
   MemoryVectorStore,
+  MistralLLM,
   OllamaEmbedder,
   OllamaLLM,
   OpenAIEmbedder,