mem0ai 3.0.2 → 3.0.4

package/dist/oss/index.js

@@ -58,7 +58,8 @@ __export(index_exports, {
   RedisDB: () => RedisDB,
   SupabaseDB: () => SupabaseDB,
   VectorStoreFactory: () => VectorStoreFactory,
-  VectorizeDB: () => VectorizeDB
+  VectorizeDB: () => VectorizeDB,
+  buildFilterConditions: () => buildFilterConditions
 });
 module.exports = __toCommonJS(index_exports);
 
@@ -893,9 +894,9 @@ var _MemoryVectorStore = class _MemoryVectorStore {
     if (row) {
       return row.user_id;
     }
-    const randomUserId = Math.random().toString(36).substring(2, 15) + Math.random().toString(36).substring(2, 15);
-    this.db.prepare(`INSERT INTO memory_migrations (user_id) VALUES (?)`).run(randomUserId);
-    return randomUserId;
+    const randomUserId2 = Math.random().toString(36).substring(2, 15) + Math.random().toString(36).substring(2, 15);
+    this.db.prepare(`INSERT INTO memory_migrations (user_id) VALUES (?)`).run(randomUserId2);
+    return randomUserId2;
   }
   async setUserId(userId) {
     this.db.prepare(`DELETE FROM memory_migrations`).run();
@@ -1182,17 +1183,17 @@ var Qdrant = class {
       if (result.points.length > 0) {
         return (_a2 = result.points[0].payload) == null ? void 0 : _a2.user_id;
       }
-      const randomUserId = Math.random().toString(36).substring(2, 15) + Math.random().toString(36).substring(2, 15);
+      const randomUserId2 = Math.random().toString(36).substring(2, 15) + Math.random().toString(36).substring(2, 15);
       await this.client.upsert("memory_migrations", {
         points: [
           {
             id: this.generateUUID(),
             vector: [0],
-            payload: { user_id: randomUserId }
+            payload: { user_id: randomUserId2 }
           }
         ]
       });
-      return randomUserId;
+      return randomUserId2;
     } catch (error) {
       console.error("Error getting user ID:", error);
       throw error;
@@ -1496,11 +1497,11 @@ var VectorizeDB = class {
       if (result.matches.length > 0) {
         return result.matches[0].metadata.userId;
       }
-      const randomUserId = Math.random().toString(36).substring(2, 15) + Math.random().toString(36).substring(2, 15);
+      const randomUserId2 = Math.random().toString(36).substring(2, 15) + Math.random().toString(36).substring(2, 15);
       const data = {
         id: this.generateUUID(),
         values: [0],
-        metadata: { userId: randomUserId }
+        metadata: { userId: randomUserId2 }
       };
       await fetch(
         `https://api.cloudflare.com/client/v4/accounts/${this.accountId}/vectorize/v2/indexes/memory_migrations/upsert`,
@@ -1514,7 +1515,7 @@ var VectorizeDB = class {
           // ndjson format
         }
       );
-      return randomUserId;
+      return randomUserId2;
     } catch (error) {
       console.error("Error getting user ID:", error);
       throw new Error(
@@ -2129,9 +2130,9 @@ var RedisDB = class {
       if (userId) {
         return userId;
       }
-      const randomUserId = Math.random().toString(36).substring(2, 15) + Math.random().toString(36).substring(2, 15);
-      await this.client.set("memory_migrations:1", randomUserId);
-      return randomUserId;
+      const randomUserId2 = Math.random().toString(36).substring(2, 15) + Math.random().toString(36).substring(2, 15);
+      await this.client.set("memory_migrations:1", randomUserId2);
+      return randomUserId2;
     } catch (error) {
       console.error("Error getting user ID:", error);
       throw error;
@@ -2507,18 +2508,18 @@ See the SQL migration instructions in the code comments.`
     try {
       const { data: tableExists } = await this.client.from("memory_migrations").select("user_id").limit(1);
       if (!tableExists || tableExists.length === 0) {
-        const randomUserId = Math.random().toString(36).substring(2, 15) + Math.random().toString(36).substring(2, 15);
-        const { error: insertError } = await this.client.from("memory_migrations").insert({ user_id: randomUserId });
+        const randomUserId2 = Math.random().toString(36).substring(2, 15) + Math.random().toString(36).substring(2, 15);
+        const { error: insertError } = await this.client.from("memory_migrations").insert({ user_id: randomUserId2 });
         if (insertError) throw insertError;
-        return randomUserId;
+        return randomUserId2;
       }
       const { data, error } = await this.client.from("memory_migrations").select("user_id").limit(1);
       if (error) throw error;
       if (!data || data.length === 0) {
-        const randomUserId = Math.random().toString(36).substring(2, 15) + Math.random().toString(36).substring(2, 15);
-        const { error: insertError } = await this.client.from("memory_migrations").insert({ user_id: randomUserId });
+        const randomUserId2 = Math.random().toString(36).substring(2, 15) + Math.random().toString(36).substring(2, 15);
+        const { error: insertError } = await this.client.from("memory_migrations").insert({ user_id: randomUserId2 });
         if (insertError) throw insertError;
-        return randomUserId;
+        return randomUserId2;
       }
       return data[0].user_id;
     } catch (error) {
@@ -4390,14 +4391,14 @@ var AzureAISearch = class {
           return userId;
         }
       }
-      const randomUserId = Math.random().toString(36).substring(2, 15) + Math.random().toString(36).substring(2, 15);
+      const randomUserId2 = Math.random().toString(36).substring(2, 15) + Math.random().toString(36).substring(2, 15);
       await this.searchClient.uploadDocuments([
         {
           id: this.generateUUID(),
-          user_id: randomUserId
+          user_id: randomUserId2
         }
       ]);
-      return randomUserId;
+      return randomUserId2;
     } catch (error) {
       console.error("Error getting user ID:", error);
       throw error;
@@ -4445,13 +4446,135 @@ var AzureAISearch = class {
 
 // src/oss/src/vector_stores/pgvector.ts
 var import_pg = __toESM(require("pg"));
-var { Client } = import_pg.default;
+var { Client, escapeIdentifier } = import_pg.default;
+var SAFE_IDENTIFIER_RE = /^[a-zA-Z_][a-zA-Z0-9_]{0,127}$/;
+function validateIdentifier(name, label = "identifier") {
+  if (!SAFE_IDENTIFIER_RE.test(name)) {
+    throw new Error(
+      `Invalid ${label} '${name}': only letters, digits, and underscores are allowed, must start with a letter or underscore, and be at most 128 characters.`
+    );
+  }
+  return name;
+}
+function escapeFilterKey(key) {
+  if (!SAFE_IDENTIFIER_RE.test(key)) {
+    throw new Error(
+      `Invalid filter key '${key}': only letters, digits, and underscores are allowed.`
+    );
+  }
+  return key;
+}
+var OPERATOR_SQL_MAP = {
+  eq: { template: "payload->>'%KEY%' = $%IDX%", numeric: false },
+  ne: { template: "payload->>'%KEY%' != $%IDX%", numeric: false },
+  gt: { template: "(payload->>'%KEY%')::numeric > $%IDX%", numeric: true },
+  gte: { template: "(payload->>'%KEY%')::numeric >= $%IDX%", numeric: true },
+  lt: { template: "(payload->>'%KEY%')::numeric < $%IDX%", numeric: true },
+  lte: { template: "(payload->>'%KEY%')::numeric <= $%IDX%", numeric: true },
+  in: { template: "payload->>'%KEY%' = ANY($%IDX%::text[])", numeric: false },
+  nin: {
+    template: "NOT (payload->>'%KEY%' = ANY($%IDX%::text[]))",
+    numeric: false
+  },
+  contains: {
+    template: "payload->>'%KEY%' LIKE $%IDX% ESCAPE '\\'",
+    numeric: false
+  },
+  icontains: {
+    template: "payload->>'%KEY%' ILIKE $%IDX% ESCAPE '\\'",
+    numeric: false
+  }
+};
+function buildFilterConditions(filters, startIndex) {
+  const conditions = [];
+  const values = [];
+  let paramIndex = startIndex;
+  if (!filters) {
+    return { conditions, values, paramIndex };
+  }
+  for (const [key, value] of Object.entries(filters)) {
+    if (key === "$or") {
+      const orGroups = [];
+      for (const orFilter of value) {
+        const sub = buildFilterConditions(orFilter, paramIndex);
+        if (sub.conditions.length > 0) {
+          orGroups.push("(" + sub.conditions.join(" AND ") + ")");
+          values.push(...sub.values);
+          paramIndex = sub.paramIndex;
+        }
+      }
+      if (orGroups.length > 0) {
+        conditions.push("(" + orGroups.join(" OR ") + ")");
+      }
+      continue;
+    }
+    if (key === "$not") {
+      const notGroups = [];
+      for (const notFilter of value) {
+        const sub = buildFilterConditions(notFilter, paramIndex);
+        if (sub.conditions.length > 0) {
+          notGroups.push("(" + sub.conditions.join(" AND ") + ")");
+          values.push(...sub.values);
+          paramIndex = sub.paramIndex;
+        }
+      }
+      if (notGroups.length > 0) {
+        conditions.push("NOT (" + notGroups.join(" OR ") + ")");
+      }
+      continue;
+    }
+    const safeKey = escapeFilterKey(key);
+    if (value === "*") {
+      conditions.push(`payload ? $${paramIndex}`);
+      values.push(key);
+      paramIndex++;
+      continue;
+    }
+    if (typeof value === "object" && value !== null && !Array.isArray(value)) {
+      for (const [op, opValue] of Object.entries(value)) {
+        const mapping = OPERATOR_SQL_MAP[op];
+        if (!mapping) {
+          throw new Error(`Unsupported filter operator: ${op}`);
+        }
+        const clause = mapping.template.replace("%KEY%", safeKey).replace("%IDX%", String(paramIndex));
+        conditions.push(clause);
+        if (op === "in" || op === "nin") {
+          values.push(opValue.map(String));
+        } else if (op === "contains" || op === "icontains") {
+          const escaped = String(opValue).replace(/\\/g, "\\\\").replace(/%/g, "\\%").replace(/_/g, "\\_");
+          values.push(`%${escaped}%`);
+        } else if (mapping.numeric) {
+          values.push(Number(opValue));
+        } else {
+          values.push(String(opValue));
+        }
+        paramIndex++;
+      }
+    } else if (Array.isArray(value)) {
+      conditions.push(`payload->>'${safeKey}' = ANY($${paramIndex}::text[])`);
+      values.push(value.map(String));
+      paramIndex++;
+    } else {
+      conditions.push(`payload->>'${safeKey}' = $${paramIndex}`);
+      if (typeof value === "boolean") {
+        values.push(JSON.stringify(value));
+      } else {
+        values.push(String(value));
+      }
+      paramIndex++;
+    }
+  }
+  return { conditions, values, paramIndex };
+}
 var PGVector = class {
   constructor(config) {
-    this.collectionName = config.collectionName || "memories";
+    this.collectionName = validateIdentifier(
+      config.collectionName || "memories",
+      "collectionName"
+    );
     this.useDiskann = config.diskann || false;
     this.useHnsw = config.hnsw || false;
-    this.dbName = config.dbname || "vector_store";
+    this.dbName = validateIdentifier(config.dbname || "vector_store", "dbname");
     this.config = config;
     this.client = new Client({
       database: "postgres",
@@ -4463,6 +4586,9 @@ var PGVector = class {
     });
     this.initialize().catch(console.error);
   }
+  col() {
+    return escapeIdentifier(this.collectionName);
+  }
   async initialize() {
     if (!this._initPromise) {
       this._initPromise = this._doInitialize();
@@ -4509,13 +4635,14 @@ var PGVector = class {
     return result.rows.length > 0;
   }
   async createDatabase(dbName) {
-    await this.client.query(`CREATE DATABASE ${dbName}`);
+    await this.client.query(`CREATE DATABASE ${escapeIdentifier(dbName)}`);
   }
   async createCol(embeddingModelDims) {
+    const dims = Math.floor(embeddingModelDims);
     await this.client.query(`
-      CREATE TABLE IF NOT EXISTS ${this.collectionName} (
+      CREATE TABLE IF NOT EXISTS ${this.col()} (
         id UUID PRIMARY KEY,
-        vector vector(${embeddingModelDims}),
+        vector vector(${dims}),
         payload JSONB
       );
     `);
@@ -4526,8 +4653,8 @@ var PGVector = class {
         );
         if (result.rows.length > 0) {
           await this.client.query(`
-            CREATE INDEX IF NOT EXISTS ${this.collectionName}_diskann_idx
-            ON ${this.collectionName}
+            CREATE INDEX IF NOT EXISTS ${escapeIdentifier(this.collectionName + "_diskann_idx")}
+            ON ${this.col()}
             USING diskann (vector);
           `);
         }
@@ -4537,8 +4664,8 @@ var PGVector = class {
     } else if (this.useHnsw) {
       try {
         await this.client.query(`
-          CREATE INDEX IF NOT EXISTS ${this.collectionName}_hnsw_idx
-          ON ${this.collectionName}
+          CREATE INDEX IF NOT EXISTS ${escapeIdentifier(this.collectionName + "_hnsw_idx")}
+          ON ${this.col()}
           USING hnsw (vector vector_cosine_ops);
         `);
       } catch (error) {
@@ -4550,11 +4677,10 @@ var PGVector = class {
     const values = vectors.map((vector, i) => ({
       id: ids[i],
       vector: `[${vector.join(",")}]`,
-      // Format vector as string with square brackets
       payload: payloads[i]
     }));
     const query = `
-      INSERT INTO ${this.collectionName} (id, vector, payload)
+      INSERT INTO ${this.col()} (id, vector, payload)
       VALUES ($1, $2::vector, $3::jsonb)
     `;
     await Promise.all(
@@ -4565,20 +4691,16 @@ var PGVector = class {
   }
   async keywordSearch(query, topK = 5, filters) {
     try {
-      const filterConditions = [];
-      const filterValues = [query, topK];
-      let filterIndex = 3;
-      if (filters) {
-        for (const [key, value] of Object.entries(filters)) {
-          filterConditions.push(`payload->>'${key}' = $${filterIndex}`);
-          filterValues.push(value);
-          filterIndex++;
-        }
-      }
-      const filterClause = filterConditions.length > 0 ? "AND " + filterConditions.join(" AND ") : "";
+      const {
+        conditions,
+        values,
+        paramIndex: _
+      } = buildFilterConditions(filters, 3);
+      const filterValues = [query, topK, ...values];
+      const filterClause = conditions.length > 0 ? "AND " + conditions.join(" AND ") : "";
       const searchQuery = `
         SELECT id, ts_rank_cd(to_tsvector('simple', payload->>'textLemmatized'), plainto_tsquery('simple', $1)) AS score, payload
-        FROM ${this.collectionName}
+        FROM ${this.col()}
         WHERE to_tsvector('simple', payload->>'textLemmatized') @@ plainto_tsquery('simple', $1)
         ${filterClause}
         ORDER BY score DESC
@@ -4596,21 +4718,17 @@ var PGVector = class {
     }
   }
   async search(query, topK = 5, filters) {
-    const filterConditions = [];
     const queryVector = `[${query.join(",")}]`;
-    const filterValues = [queryVector, topK];
-    let filterIndex = 3;
-    if (filters) {
-      for (const [key, value] of Object.entries(filters)) {
-        filterConditions.push(`payload->>'${key}' = $${filterIndex}`);
-        filterValues.push(value);
-        filterIndex++;
-      }
-    }
-    const filterClause = filterConditions.length > 0 ? "WHERE " + filterConditions.join(" AND ") : "";
+    const {
+      conditions,
+      values,
+      paramIndex: _
+    } = buildFilterConditions(filters, 3);
+    const filterValues = [queryVector, topK, ...values];
+    const filterClause = conditions.length > 0 ? "WHERE " + conditions.join(" AND ") : "";
     const searchQuery = `
       SELECT id, vector <=> $1::vector AS distance, payload
-      FROM ${this.collectionName}
+      FROM ${this.col()}
       ${filterClause}
       ORDER BY distance
       LIMIT $2
@@ -4619,12 +4737,12 @@ var PGVector = class {
     return result.rows.map((row) => ({
       id: row.id,
       payload: row.payload,
-      score: row.distance
+      score: Math.max(0, Math.min(1, 1 - Number(row.distance)))
     }));
   }
   async get(vectorId) {
     const result = await this.client.query(
-      `SELECT id, payload FROM ${this.collectionName} WHERE id = $1`,
+      `SELECT id, payload FROM ${this.col()} WHERE id = $1`,
       [vectorId]
     );
     if (result.rows.length === 0) return null;
@@ -4637,7 +4755,7 @@ var PGVector = class {
     const vectorStr = `[${vector.join(",")}]`;
     await this.client.query(
       `
-      UPDATE ${this.collectionName}
+      UPDATE ${this.col()}
       SET vector = $1::vector, payload = $2::jsonb
       WHERE id = $3
       `,
@@ -4645,13 +4763,12 @@ var PGVector = class {
     );
   }
   async delete(vectorId) {
-    await this.client.query(
-      `DELETE FROM ${this.collectionName} WHERE id = $1`,
-      [vectorId]
-    );
+    await this.client.query(`DELETE FROM ${this.col()} WHERE id = $1`, [
+      vectorId
+    ]);
   }
   async deleteCol() {
-    await this.client.query(`DROP TABLE IF EXISTS ${this.collectionName}`);
+    await this.client.query(`DROP TABLE IF EXISTS ${this.col()}`);
   }
   async listCols() {
     const result = await this.client.query(`
@@ -4662,33 +4779,27 @@ var PGVector = class {
     return result.rows.map((row) => row.table_name);
   }
   async list(filters, topK = 100) {
-    const filterConditions = [];
-    const filterValues = [];
-    let paramIndex = 1;
-    if (filters) {
-      for (const [key, value] of Object.entries(filters)) {
-        filterConditions.push(`payload->>'${key}' = $${paramIndex}`);
-        filterValues.push(value);
-        paramIndex++;
-      }
-    }
-    const filterClause = filterConditions.length > 0 ? "WHERE " + filterConditions.join(" AND ") : "";
+    const {
+      conditions,
+      values: filterValues,
+      paramIndex
+    } = buildFilterConditions(filters, 1);
+    const filterClause = conditions.length > 0 ? "WHERE " + conditions.join(" AND ") : "";
     const listQuery = `
       SELECT id, payload
-      FROM ${this.collectionName}
+      FROM ${this.col()}
       ${filterClause}
       LIMIT $${paramIndex}
     `;
     const countQuery = `
       SELECT COUNT(*)
-      FROM ${this.collectionName}
+      FROM ${this.col()}
       ${filterClause}
     `;
-    filterValues.push(topK);
+    const listValues = [...filterValues, topK];
     const [listResult, countResult] = await Promise.all([
-      this.client.query(listQuery, filterValues),
-      this.client.query(countQuery, filterValues.slice(0, -1))
-      // Remove limit parameter for count query
+      this.client.query(listQuery, listValues),
+      this.client.query(countQuery, filterValues)
     ]);
     const results = listResult.rows.map((row) => ({
       id: row.id,
@@ -4706,12 +4817,12 @@ var PGVector = class {
     if (result.rows.length > 0) {
       return result.rows[0].user_id;
     }
-    const randomUserId = Math.random().toString(36).substring(2, 15) + Math.random().toString(36).substring(2, 15);
+    const randomUserId2 = Math.random().toString(36).substring(2, 15) + Math.random().toString(36).substring(2, 15);
     await this.client.query(
       "INSERT INTO memory_migrations (user_id) VALUES ($1)",
-      [randomUserId]
+      [randomUserId2]
     );
-    return randomUserId;
+    return randomUserId2;
   }
   async setUserId(userId) {
     await this.client.query("DELETE FROM memory_migrations");
@@ -5012,7 +5123,7 @@ var parse_vision_messages = async (messages) => {
 };
 
 // src/oss/src/utils/telemetry.ts
-var version = true ? "3.0.2" : "dev";
+var version = true ? "3.0.4" : "dev";
 var MEM0_TELEMETRY = true;
 var _a;
 try {
@@ -5914,6 +6025,69 @@ function scoreAndRank(semanticResults, bm25Scores, entityBoosts, threshold, topK
   return scored.slice(0, topK);
 }
 
+// src/client/config.ts
+async function getNodeFs() {
+  var _a2, _b, _c, _d, _e;
+  if (typeof process === "undefined" || !((_a2 = process.versions) == null ? void 0 : _a2.node)) return null;
+  try {
+    const [fs4, path3, os2, crypto] = await Promise.all([
+      import("fs"),
+      import("path"),
+      import("os"),
+      import("crypto")
+    ]);
+    const fsMod = (_b = fs4.default) != null ? _b : fs4;
+    const pathMod = (_c = path3.default) != null ? _c : path3;
+    const osMod = (_d = os2.default) != null ? _d : os2;
+    const cryptoMod = (_e = crypto.default) != null ? _e : crypto;
+    const dir = process.env.MEM0_DIR || pathMod.join(osMod.homedir(), ".mem0");
+    return {
+      fs: fsMod,
+      path: pathMod,
+      crypto: cryptoMod,
+      configPath: pathMod.join(dir, "config.json")
+    };
+  } catch (e) {
+    return null;
+  }
+}
+function loadConfig(node) {
+  try {
+    if (!node.fs.existsSync(node.configPath)) return null;
+    const parsed = JSON.parse(node.fs.readFileSync(node.configPath, "utf8"));
+    return parsed && typeof parsed === "object" ? parsed : null;
+  } catch (e) {
+    return null;
+  }
+}
+function writeConfig(node, config) {
+  node.fs.mkdirSync(node.path.dirname(node.configPath), { recursive: true });
+  node.fs.writeFileSync(node.configPath, JSON.stringify(config, null, 4));
+}
+function randomUserId(node) {
+  if (typeof node.crypto.randomUUID === "function") {
+    return node.crypto.randomUUID();
+  }
+  return Math.random().toString(36).substring(2, 15) + Math.random().toString(36).substring(2, 15);
+}
+async function getOrCreateMem0UserId() {
+  var _a2;
+  const node = await getNodeFs();
+  if (!node) return null;
+  try {
+    const config = (_a2 = loadConfig(node)) != null ? _a2 : {};
+    if (typeof config.user_id === "string" && config.user_id) {
+      return config.user_id;
+    }
+    const userId = randomUserId(node);
+    config.user_id = userId;
+    writeConfig(node, config);
+    return userId;
+  } catch (e) {
+    return null;
+  }
+}
+
 // src/oss/src/memory/index.ts
 var ENTITY_PARAMS = [
   "user_id",
@@ -6220,7 +6394,11 @@ var Memory = class _Memory {
   async _getTelemetryId() {
     try {
       if (!this.telemetryId || this.telemetryId === "anonymous" || this.telemetryId === "anonymous-supabase") {
-        this.telemetryId = await this.vectorStore.getUserId();
+        this.telemetryId = await getOrCreateMem0UserId() || await this.vectorStore.getUserId();
+        try {
+          await this.vectorStore.setUserId(this.telemetryId);
+        } catch (e) {
+        }
       }
       return this.telemetryId;
     } catch (error) {
@@ -7214,6 +7392,7 @@ var Memory = class _Memory {
   RedisDB,
   SupabaseDB,
   VectorStoreFactory,
-  VectorizeDB
+  VectorizeDB,
+  buildFilterConditions
 });
 //# sourceMappingURL=index.js.map