mem0ai 2.1.4 → 2.1.6

package/dist/oss/index.mjs

@@ -609,6 +609,24 @@ var EXCLUDED_KEYS = /* @__PURE__ */ new Set([
   "created_at",
   "updated_at"
 ]);
+function toSnakeCase(obj) {
+  if (typeof obj !== "object" || obj === null) return obj;
+  return Object.fromEntries(
+    Object.entries(obj).map(([key, value]) => [
+      key.replace(/[A-Z]/g, (letter) => `_${letter.toLowerCase()}`),
+      value
+    ])
+  );
+}
+function toCamelCase(obj) {
+  if (typeof obj !== "object" || obj === null) return obj;
+  return Object.fromEntries(
+    Object.entries(obj).map(([key, value]) => [
+      key.replace(/_([a-z])/g, (_, letter) => letter.toUpperCase()),
+      value
+    ])
+  );
+}
 var RedisDB = class {
   constructor(config) {
     this.indexName = config.collectionName;
@@ -740,7 +758,7 @@ var RedisDB = class {
   }
   async insert(vectors, ids, payloads) {
     const data = vectors.map((vector, idx) => {
-      const payload = payloads[idx];
+      const payload = toSnakeCase(payloads[idx]);
       const id = ids[idx];
       const entry = {
         memory_id: id,
@@ -776,7 +794,8 @@ var RedisDB = class {
     }
   }
   async search(query, limit = 5, filters) {
-    const filterExpr = filters ? Object.entries(filters).filter(([_, value]) => value !== null).map(([key, value]) => `@${key}:{${value}}`).join(" ") : "*";
+    const snakeFilters = filters ? toSnakeCase(filters) : void 0;
+    const filterExpr = snakeFilters ? Object.entries(snakeFilters).filter(([_, value]) => value !== null).map(([key, value]) => `@${key}:{${value}}`).join(" ") : "*";
     const queryVector = new Float32Array(query).buffer;
     const searchOptions = {
       PARAMS: {
@@ -790,9 +809,10 @@ var RedisDB = class {
         "user_id",
         "memory",
         "metadata",
-        "created_at"
+        "created_at",
+        "__vector_score"
       ],
-      SORTBY: "vector_score",
+      SORTBY: "__vector_score",
       DIALECT: 2,
       LIMIT: {
         from: 0,
@@ -802,11 +822,12 @@ var RedisDB = class {
     try {
       const results = await this.client.ft.search(
         this.indexName,
-        `${filterExpr} =>[KNN ${limit} @embedding $vec AS vector_score]`,
+        `${filterExpr} =>[KNN ${limit} @embedding $vec AS __vector_score]`,
         searchOptions
       );
       return results.documents.map((doc) => {
-        const payload = {
+        var _a;
+        const resultPayload = {
           hash: doc.value.hash,
           data: doc.value.memory,
           created_at: new Date(parseInt(doc.value.created_at)).toISOString(),
@@ -820,8 +841,8 @@ var RedisDB = class {
         };
         return {
           id: doc.value.memory_id,
-          payload,
-          score: doc.value.vector_score
+          payload: toCamelCase(resultPayload),
+          score: (_a = Number(doc.value.__vector_score)) != null ? _a : 0
         };
       });
     } catch (error) {
@@ -919,22 +940,23 @@ var RedisDB = class {
     }
   }
   async update(vectorId, vector, payload) {
+    const snakePayload = toSnakeCase(payload);
     const entry = {
       memory_id: vectorId,
-      hash: payload.hash,
-      memory: payload.data,
-      created_at: new Date(payload.created_at).getTime(),
-      updated_at: new Date(payload.updated_at).getTime(),
+      hash: snakePayload.hash,
+      memory: snakePayload.data,
+      created_at: new Date(snakePayload.created_at).getTime(),
+      updated_at: new Date(snakePayload.updated_at).getTime(),
       embedding: Buffer.from(new Float32Array(vector).buffer)
     };
     ["agent_id", "run_id", "user_id"].forEach((field) => {
-      if (field in payload) {
-        entry[field] = payload[field];
+      if (field in snakePayload) {
+        entry[field] = snakePayload[field];
       }
     });
     entry.metadata = JSON.stringify(
       Object.fromEntries(
-        Object.entries(payload).filter(([key]) => !EXCLUDED_KEYS.has(key))
+        Object.entries(snakePayload).filter(([key]) => !EXCLUDED_KEYS.has(key))
       )
     );
     try {
@@ -966,7 +988,8 @@ var RedisDB = class {
     await this.client.ft.dropIndex(this.indexName);
   }
   async list(filters, limit = 100) {
-    const filterExpr = filters ? Object.entries(filters).filter(([_, value]) => value !== null).map(([key, value]) => `@${key}:{${value}}`).join(" ") : "*";
+    const snakeFilters = filters ? toSnakeCase(filters) : void 0;
+    const filterExpr = snakeFilters ? Object.entries(snakeFilters).filter(([_, value]) => value !== null).map(([key, value]) => `@${key}:{${value}}`).join(" ") : "*";
     const searchOptions = {
       SORTBY: "created_at",
       SORTDIR: "DESC",
@@ -982,7 +1005,7 @@ var RedisDB = class {
     );
     const items = results.documents.map((doc) => ({
       id: doc.value.memory_id,
-      payload: {
+      payload: toCamelCase({
         hash: doc.value.hash,
         data: doc.value.memory,
         created_at: new Date(parseInt(doc.value.created_at)).toISOString(),
@@ -993,7 +1016,7 @@ var RedisDB = class {
         ...doc.value.run_id && { run_id: doc.value.run_id },
         ...doc.value.user_id && { user_id: doc.value.user_id },
         ...JSON.parse(doc.value.metadata || "{}")
-      }
+      })
     }));
     return [items, results.total];
   }