mem0ai 2.4.4 → 2.4.5

package/dist/oss/index.mjs

@@ -68,19 +68,25 @@ var OpenAIEmbedder = class {
       baseURL: config.baseURL || config.url
     });
     this.model = config.model || "text-embedding-3-small";
-    this.embeddingDims = config.embeddingDims || 1536;
+    this.embeddingDims = config.embeddingDims;
   }
   async embed(text) {
     const response = await this.openai.embeddings.create({
       model: this.model,
-      input: text
+      input: text,
+      ...this.embeddingDims !== void 0 && {
+        dimensions: this.embeddingDims
+      }
     });
     return response.data[0].embedding;
   }
   async embedBatch(texts) {
     const response = await this.openai.embeddings.create({
       model: this.model,
-      input: texts
+      input: texts,
+      ...this.embeddingDims !== void 0 && {
+        dimensions: this.embeddingDims
+      }
     });
     return response.data.map((item) => item.embedding);
   }
@@ -2010,7 +2016,7 @@ See the SQL migration instructions in the code comments.`
   }
   async get(vectorId) {
     try {
-      const { data, error } = await this.client.from(this.tableName).select("*").eq("id", vectorId).single();
+      const { data, error } = await this.client.from(this.tableName).select("*").eq("id", vectorId).maybeSingle();
       if (error) throw error;
       if (!data) return null;
       return {
@@ -2270,13 +2276,15 @@ var GoogleEmbedder = class {
       apiKey: config.apiKey || process.env.GOOGLE_API_KEY
     });
     this.model = config.model || "gemini-embedding-001";
-    this.embeddingDims = config.embeddingDims || 1536;
+    this.embeddingDims = config.embeddingDims;
   }
   async embed(text) {
     const response = await this.google.models.embedContent({
       model: this.model,
       contents: text,
-      config: { outputDimensionality: this.embeddingDims }
+      ...this.embeddingDims !== void 0 && {
+        config: { outputDimensionality: this.embeddingDims }
+      }
     });
     return response.embeddings[0].values;
   }
@@ -2284,7 +2292,9 @@ var GoogleEmbedder = class {
     const response = await this.google.models.embedContent({
       model: this.model,
       contents: texts,
-      config: { outputDimensionality: this.embeddingDims }
+      ...this.embeddingDims !== void 0 && {
+        config: { outputDimensionality: this.embeddingDims }
+      }
     });
     return response.embeddings.map((item) => item.values);
   }
@@ -2426,19 +2436,25 @@ var AzureOpenAIEmbedder = class {
       ...rest
     });
     this.model = config.model || "text-embedding-3-small";
-    this.embeddingDims = config.embeddingDims || 1536;
+    this.embeddingDims = config.embeddingDims;
   }
   async embed(text) {
     const response = await this.client.embeddings.create({
       model: this.model,
-      input: text
+      input: text,
+      ...this.embeddingDims !== void 0 && {
+        dimensions: this.embeddingDims
+      }
     });
     return response.data[0].embedding;
   }
   async embedBatch(texts) {
     const response = await this.client.embeddings.create({
       model: this.model,
-      input: texts
+      input: texts,
+      ...this.embeddingDims !== void 0 && {
+        dimensions: this.embeddingDims
+      }
     });
     return response.data.map((item) => item.embedding);
   }
@@ -2704,7 +2720,23 @@ function getUpdateMemoryMessages(retrievedOldMemory, newRetrievedFacts) {
   Do not return anything except the JSON format.`;
 }
 function removeCodeBlocks(text) {
-  return text.replace(/```(?:\w+)?\n?([\s\S]*?)(?:```|$)/g, "$1").trim();
+  const stripped = text.replace(/```(?:\w+)?\n?([\s\S]*?)(?:```|$)/g, "$1").trim();
+  return stripped.replace(/<think>[\s\S]*?<\/think>/g, "").trim();
+}
+function extractJson(text) {
+  const cleaned = removeCodeBlocks(text);
+  const trimmed = cleaned.trim();
+  const firstBrace = trimmed.indexOf("{");
+  const lastBrace = trimmed.lastIndexOf("}");
+  if (firstBrace !== -1 && lastBrace > firstBrace) {
+    return trimmed.substring(firstBrace, lastBrace + 1);
+  }
+  const firstBracket = trimmed.indexOf("[");
+  const lastBracket = trimmed.lastIndexOf("]");
+  if (firstBracket !== -1 && lastBracket > firstBracket) {
+    return trimmed.substring(firstBracket, lastBracket + 1);
+  }
+  return trimmed;
 }
 
 // src/oss/src/graphs/tools.ts
@@ -5066,7 +5098,7 @@ ${parsedMessages}`
       ],
       { type: "json_object" }
     );
-    const cleanResponse = removeCodeBlocks(response);
+    const cleanResponse = extractJson(response);
     let facts = [];
     try {
       const parsed = FactRetrievalSchema.parse(JSON.parse(cleanResponse));
@@ -5106,7 +5138,7 @@ ${parsedMessages}`
       [{ role: "user", content: updatePrompt }],
       { type: "json_object" }
     );
-    const cleanUpdateResponse = removeCodeBlocks(updateResponse);
+    const cleanUpdateResponse = extractJson(updateResponse);
     let memoryActions = [];
     try {
       memoryActions = JSON.parse(cleanUpdateResponse).memory || [];