mem0ai 3.0.8 → 3.0.10

package/dist/oss/index.js

@@ -99,7 +99,10 @@ var MemoryConfigSchema = import_zod.z.object({
       modelProperties: import_zod.z.record(import_zod.z.string(), import_zod.z.any()).optional(),
       baseURL: import_zod.z.string().optional(),
       url: import_zod.z.string().optional(),
-      timeout: import_zod.z.number().optional()
+      timeout: import_zod.z.number().optional(),
+      temperature: import_zod.z.number().optional(),
+      topP: import_zod.z.number().optional(),
+      maxTokens: import_zod.z.number().optional()
     })
   }),
   historyDbPath: import_zod.z.string().optional(),
@@ -389,25 +392,58 @@ var OpenAIStructuredLLM = class {
 var import_sdk = __toESM(require("@anthropic-ai/sdk"));
 var AnthropicLLM = class {
   constructor(config) {
+    var _a2, _b2;
     const apiKey = config.apiKey || process.env.ANTHROPIC_API_KEY;
     if (!apiKey) {
       throw new Error("Anthropic API key is required");
     }
-    this.client = new import_sdk.default({ apiKey });
-    this.model = config.model || "claude-3-sonnet-20240229";
+    const clientArgs = { apiKey };
+    if (config.baseURL) {
+      clientArgs.baseURL = config.baseURL;
+    }
+    this.client = new import_sdk.default(clientArgs);
+    this.model = config.model || "claude-sonnet-4-6";
+    this.maxTokens = (_a2 = config.maxTokens) != null ? _a2 : 2e3;
+    this.temperature = (_b2 = config.temperature) != null ? _b2 : 0.1;
+    this.topP = config.topP;
   }
-  async generateResponse(messages, responseFormat) {
+  async generateResponse(messages, responseFormat, tools) {
     const systemMessage = messages.find((msg) => msg.role === "system");
     const otherMessages = messages.filter((msg) => msg.role !== "system");
-    const response = await this.client.messages.create({
+    const params = {
       model: this.model,
       messages: otherMessages.map((msg) => ({
         role: msg.role,
         content: typeof msg.content === "string" ? msg.content : msg.content.image_url.url
       })),
       system: typeof (systemMessage == null ? void 0 : systemMessage.content) === "string" ? systemMessage.content : void 0,
-      max_tokens: 4096
-    });
+      max_tokens: this.maxTokens
+    };
+    if (this.temperature !== void 0) {
+      params.temperature = this.temperature;
+    } else if (this.topP !== void 0) {
+      params.top_p = this.topP;
+    }
+    if (tools) {
+      params.tools = tools;
+      params.tool_choice = { type: "auto" };
+    }
+    const response = await this.client.messages.create(params);
+    if (tools) {
+      let content = "";
+      const toolCalls = [];
+      for (const block of response.content) {
+        if (block.type === "text") {
+          content = block.text;
+        } else if (block.type === "tool_use") {
+          toolCalls.push({
+            name: block.name,
+            arguments: JSON.stringify(block.input)
+          });
+        }
+      }
+      return { content, role: "assistant", toolCalls };
+    }
     const firstBlock = response.content[0];
     if (firstBlock.type === "text") {
       return firstBlock.text;
@@ -417,10 +453,10 @@ var AnthropicLLM = class {
   }
   async generateChat(messages) {
     const response = await this.generateResponse(messages);
-    return {
-      content: response,
-      role: "assistant"
-    };
+    if (typeof response === "string") {
+      return { content: response, role: "assistant" };
+    }
+    return response;
   }
 };
 
@@ -5044,7 +5080,7 @@ var ConfigManager = class {
       llm: {
         provider: ((_c = userConfig.llm) == null ? void 0 : _c.provider) || DEFAULT_MEMORY_CONFIG.llm.provider,
         config: (() => {
-          var _a3, _b3, _c2, _d2;
+          var _a3, _b3, _c2, _d2, _e2, _f2, _g2;
           const defaultConf = DEFAULT_MEMORY_CONFIG.llm.config;
           const userConf = (_a3 = userConfig.llm) == null ? void 0 : _a3.config;
           let finalModel = defaultConf.model;
@@ -5054,12 +5090,19 @@ var ConfigManager = class {
             finalModel = userConf.model;
           }
           const llmBaseURL = (_d2 = (_c2 = (_b3 = userConf == null ? void 0 : userConf.baseURL) != null ? _b3 : userConf == null ? void 0 : userConf.lmstudio_base_url) != null ? _c2 : userConf == null ? void 0 : userConf.url) != null ? _d2 : defaultConf.baseURL;
+          const llmRaw = userConf;
+          const temperature = (_e2 = userConf == null ? void 0 : userConf.temperature) != null ? _e2 : llmRaw == null ? void 0 : llmRaw.temperature;
+          const topP = (_f2 = userConf == null ? void 0 : userConf.topP) != null ? _f2 : llmRaw == null ? void 0 : llmRaw.top_p;
+          const maxTokens = (_g2 = userConf == null ? void 0 : userConf.maxTokens) != null ? _g2 : llmRaw == null ? void 0 : llmRaw.max_tokens;
           return {
             baseURL: llmBaseURL,
             url: userConf == null ? void 0 : userConf.url,
             apiKey: (userConf == null ? void 0 : userConf.apiKey) !== void 0 ? userConf.apiKey : defaultConf.apiKey,
             model: finalModel,
-            modelProperties: (userConf == null ? void 0 : userConf.modelProperties) !== void 0 ? userConf.modelProperties : defaultConf.modelProperties
+            modelProperties: (userConf == null ? void 0 : userConf.modelProperties) !== void 0 ? userConf.modelProperties : defaultConf.modelProperties,
+            temperature,
+            topP,
+            maxTokens
           };
         })()
       },
@@ -5105,6 +5148,7 @@ var get_image_description = async (image_url) => {
   return response;
 };
 var parse_vision_messages = async (messages) => {
+  var _a2;
   const parsed_messages = [];
   for (const message of messages) {
     let new_message = {
@@ -5113,9 +5157,11 @@ var parse_vision_messages = async (messages) => {
     };
     if (message.role !== "system") {
       if (typeof message.content === "object" && message.content.type === "image_url") {
-        const description = await get_image_description(
-          message.content.image_url.url
-        );
+        const imageUrl = (_a2 = message.content.image_url) == null ? void 0 : _a2.url;
+        if (!imageUrl) {
+          throw new Error("image_url content part is missing image_url.url");
+        }
+        const description = await get_image_description(imageUrl);
         new_message.content = typeof description === "string" ? description : JSON.stringify(description);
         parsed_messages.push(new_message);
       } else parsed_messages.push(message);
@@ -5125,7 +5171,7 @@ var parse_vision_messages = async (messages) => {
 };
 
 // src/oss/src/utils/telemetry.ts
-var version = true ? "3.0.8" : "dev";
+var version = true ? "3.0.10" : "dev";
 var MEM0_TELEMETRY = true;
 var _a, _b;
 try {
@@ -7549,6 +7595,25 @@ var Memory = class _Memory {
         "messages is required and cannot be undefined or null. Provide a string or array of messages."
       );
     }
+    if (Array.isArray(messages)) {
+      if (messages.length === 0) {
+        throw new Error(
+          "messages array cannot be empty. Provide at least one message with non-empty content."
+        );
+      }
+      const allBlank = messages.every(
+        (m) => typeof m.content === "string" && m.content.trim() === ""
+      );
+      if (allBlank) {
+        throw new Error(
+          "messages array cannot contain only blank content. Provide at least one message with non-empty content."
+        );
+      }
+    } else if (messages.trim() === "") {
+      throw new Error(
+        "messages string cannot be empty. Provide non-empty content."
+      );
+    }
     const temporalUsageNotice = detectTemporalUsageFromMetadata(
       config == null ? void 0 : config.metadata
     );
@@ -7608,7 +7673,7 @@ var Memory = class _Memory {
     if (!infer) {
       const returnedMemories = [];
       for (const message of messages) {
-        if (message.content === "system") {
+        if (message.role === "system") {
           continue;
         }
         const memoryId = await this.createMemory(
@@ -7632,7 +7697,7 @@ var Memory = class _Memory {
       } catch (e) {
       }
     }
-    const parsedMessages = messages.map((m) => m.content).join("\n");
+    const parsedMessages = messages.map((m) => `${m.role}: ${m.content}`).join("\n");
     const queryEmbedding = await this.embedder.embed(parsedMessages);
     const existingResults = await this.vectorStore.search(
       queryEmbedding,
@@ -7991,7 +8056,13 @@ var Memory = class _Memory {
         memoryItem.metadata[key] = value;
       }
     }
-    const result = { ...memoryItem, ...filters };
+    const result = {
+      ...memoryItem,
+      ...filters,
+      ...memory.payload.attributedTo && {
+        attributedTo: memory.payload.attributedTo
+      }
+    };
     await this._displayFirstRunNotice("get");
     return result;
   }
@@ -8189,6 +8260,7 @@ var Memory = class _Memory {
         ...payload.user_id && { user_id: payload.user_id },
         ...payload.agent_id && { agent_id: payload.agent_id },
         ...payload.run_id && { run_id: payload.run_id },
+        ...payload.attributedTo && { attributedTo: payload.attributedTo },
         ...scored.scoreDetails && { score_details: scored.scoreDetails }
       };
     });
@@ -8382,7 +8454,10 @@ var Memory = class _Memory {
       metadata: Object.entries(mem.payload).filter(([key]) => !excludedKeys.has(key)).reduce((acc, [key, value]) => ({ ...acc, [key]: value }), {}),
       ...mem.payload.user_id && { user_id: mem.payload.user_id },
       ...mem.payload.agent_id && { agent_id: mem.payload.agent_id },
-      ...mem.payload.run_id && { run_id: mem.payload.run_id }
+      ...mem.payload.run_id && { run_id: mem.payload.run_id },
+      ...mem.payload.attributedTo && {
+        attributedTo: mem.payload.attributedTo
+      }
     }));
     const result = { results };
     const scaleThresholdNotice = detectScaleThresholdFromTopK(topK);
@@ -8424,20 +8499,13 @@ var Memory = class _Memory {
     const prevValue = existingMemory.payload.data;
     const embedding = existingEmbeddings[data] || await this.embedder.embed(data);
     const newMetadata = {
+      ...existingMemory.payload,
       ...metadata,
       data,
       hash: (0, import_crypto2.createHash)("md5").update(data).digest("hex"),
+      textLemmatized: lemmatizeForBm25(data),
       createdAt: existingMemory.payload.createdAt,
-      updatedAt: (/* @__PURE__ */ new Date()).toISOString(),
-      ...existingMemory.payload.user_id && {
-        user_id: existingMemory.payload.user_id
-      },
-      ...existingMemory.payload.agent_id && {
-        agent_id: existingMemory.payload.agent_id
-      },
-      ...existingMemory.payload.run_id && {
-        run_id: existingMemory.payload.run_id
-      }
+      updatedAt: (/* @__PURE__ */ new Date()).toISOString()
     };
     await this.vectorStore.update(memoryId, embedding, newMetadata);
     await this.db.addHistory(