mem0ai 3.0.8 → 3.0.9

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,54 @@ 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";
+    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 +449,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 +5076,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 +5086,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
           };
         })()
       },
@@ -5125,7 +5164,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.9" : "dev";
 var MEM0_TELEMETRY = true;
 var _a, _b;
 try {
@@ -8424,20 +8463,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(