mem0ai 2.1.15 → 2.1.16-patch.1

package/dist/oss/index.mjs

@@ -24,7 +24,8 @@ var MemoryConfigSchema = z.object({
     provider: z.string(),
     config: z.object({
       apiKey: z.string(),
-      model: z.string().optional()
+      model: z.string().optional(),
+      modelProperties: z.record(z.string(), z.any()).optional()
     })
   }),
   historyDbPath: z.string().optional(),
@@ -310,6 +311,86 @@ var GroqLLM = class {
   }
 };
 
+// src/oss/src/llms/mistral.ts
+import { Mistral } from "@mistralai/mistralai";
+var MistralLLM = class {
+  constructor(config) {
+    if (!config.apiKey) {
+      throw new Error("Mistral API key is required");
+    }
+    this.client = new 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
 import sqlite3 from "sqlite3";
 import path from "path";
@@ -1770,6 +1851,67 @@ var GoogleLLM = class {
   }
 };
 
+// src/oss/src/llms/azure.ts
+import { AzureOpenAI } from "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 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) {
@@ -1800,6 +1942,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}`);
     }
@@ -2143,7 +2289,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: {
@@ -2165,7 +2311,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,
@@ -2941,7 +3088,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 {
@@ -3507,6 +3654,7 @@ export {
   Memory,
   MemoryConfigSchema,
   MemoryVectorStore,
+  MistralLLM,
   OllamaEmbedder,
   OllamaLLM,
   OpenAIEmbedder,