mem0ai 3.0.5 → 3.0.7

package/dist/oss/index.mjs

@@ -68,6 +68,7 @@ var OpenAIEmbedder = class {
     const response = await this.openai.embeddings.create({
       model: this.model,
       input: text,
+      encoding_format: "float",
       ...this.embeddingDims !== void 0 && {
         dimensions: this.embeddingDims
       }
@@ -82,6 +83,7 @@ var OpenAIEmbedder = class {
       const response = await this.openai.embeddings.create({
         model: this.model,
         input: chunk,
+        encoding_format: "float",
         ...this.embeddingDims !== void 0 && {
           dimensions: this.embeddingDims
         }
@@ -198,7 +200,7 @@ var LMStudioEmbedder = class {
         input: normalized,
         encoding_format: "float"
       });
-      return response.data.map((item) => item.embedding);
+      return response.data.sort((a, b) => a.index - b.index).map((item) => item.embedding);
     } catch (err) {
       const message = err instanceof Error ? err.message : String(err);
       throw new Error(`LM Studio embedder failed: ${message}`);
@@ -1885,7 +1887,7 @@ var RedisDB = class {
         return {
           id: doc.value.memory_id,
           payload: toCamelCase(resultPayload),
-          score: (_a2 = Number(doc.value.__vector_score)) != null ? _a2 : 0
+          score: Math.max(0, 1 - ((_a2 = Number(doc.value.__vector_score)) != null ? _a2 : 0))
         };
       });
     } catch (error) {
@@ -5073,7 +5075,7 @@ var parse_vision_messages = async (messages) => {
 };
 
 // src/oss/src/utils/telemetry.ts
-var version = true ? "3.0.5" : "dev";
+var version = true ? "3.0.7" : "dev";
 var MEM0_TELEMETRY = true;
 var _a;
 try {
@@ -5939,7 +5941,7 @@ function getBm25Params(query, lemmatized) {
 function normalizeBm25(rawScore, midpoint, steepness) {
   return 1 / (1 + Math.exp(-steepness * (rawScore - midpoint)));
 }
-function scoreAndRank(semanticResults, bm25Scores, entityBoosts, threshold, topK) {
+function scoreAndRank(semanticResults, bm25Scores, entityBoosts, threshold, topK, explain = false) {
   var _a2, _b, _c;
   const hasBm25 = Object.keys(bm25Scores).length > 0;
   const hasEntity = Object.keys(entityBoosts).length > 0;
@@ -5965,11 +5967,23 @@ function scoreAndRank(semanticResults, bm25Scores, entityBoosts, threshold, topK
     const entityBoost = (_c = entityBoosts[memIdStr]) != null ? _c : 0;
     const rawCombined = semanticScore + bm25Score + entityBoost;
     const combined = Math.min(rawCombined / maxPossible, 1);
-    scored.push({
+    const entry = {
       id: memIdStr,
       score: combined,
       payload: result.payload
-    });
+    };
+    if (explain) {
+      entry.scoreDetails = {
+        semanticScore,
+        bm25Score,
+        entityBoost,
+        rawScore: rawCombined,
+        maxPossibleScore: maxPossible,
+        finalScore: combined,
+        threshold
+      };
+    }
+    scored.push(entry);
   }
   scored.sort((a, b) => b.score - a.score);
   return scored.slice(0, topK);
@@ -6817,7 +6831,7 @@ var Memory = class _Memory {
       }).filter(([, v]) => v !== void 0)
     ) : {};
     await this._ensureInitialized();
-    const { topK = 20, threshold = 0.1 } = config;
+    const { topK = 20, threshold = 0.1, explain = false } = config;
     await this._captureEvent("search", {
       query_length: query.length,
       topK,
@@ -6887,15 +6901,32 @@ var Memory = class _Memory {
         }
         if (deduped.length > 0) {
           const entityStore = await this.getEntityStore();
-          for (const entity of deduped) {
-            try {
-              const entityEmbedding = await this.embedder.embed(entity.text);
-              const matches = await entityStore.search(
-                entityEmbedding,
-                500,
-                effectiveFilters
-              );
-              for (const match of matches) {
+          const entitySearchFilters = {};
+          for (const k of ["user_id", "agent_id", "run_id"]) {
+            if (effectiveFilters[k])
+              entitySearchFilters[k] = effectiveFilters[k];
+          }
+          const entityTexts = deduped.map((e) => e.text);
+          const embeddings = await this.embedder.embedBatch(entityTexts);
+          if (embeddings.length !== entityTexts.length) {
+            console.warn(
+              `embedBatch returned ${embeddings.length} vectors for ${entityTexts.length} texts \u2014 skipping entity boost`
+            );
+          } else {
+            const searchResults = await Promise.allSettled(
+              deduped.map(
+                (_, i) => entityStore.search(embeddings[i], 500, entitySearchFilters)
+              )
+            );
+            for (const result of searchResults) {
+              if (result.status === "rejected") {
+                console.warn(
+                  "Entity boost search failed for one entity:",
+                  result.reason
+                );
+                continue;
+              }
+              for (const match of result.value) {
                 const similarity = (_c = match.score) != null ? _c : 0;
                 if (similarity < 0.5) continue;
                 const payload = match.payload || {};
@@ -6914,7 +6945,6 @@ var Memory = class _Memory {
                   }
                 }
               }
-            } catch (e) {
             }
           }
         }
@@ -6935,7 +6965,8 @@ var Memory = class _Memory {
       bm25Scores,
       entityBoosts,
       threshold != null ? threshold : 0.1,
-      topK
+      topK,
+      explain
     );
     const excludedKeys = /* @__PURE__ */ new Set([
       "user_id",
@@ -6963,7 +6994,8 @@ var Memory = class _Memory {
         metadata: Object.entries(payload).filter(([key]) => !excludedKeys.has(key)).reduce((acc, [key, value]) => ({ ...acc, [key]: value }), {}),
         ...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.run_id && { run_id: payload.run_id },
+        ...scored.scoreDetails && { score_details: scored.scoreDetails }
       };
     });
     return {