mem0ai 3.0.3 → 3.0.4

package/dist/oss/index.mjs

@@ -4414,6 +4414,108 @@ function escapeFilterKey(key) {
   }
   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 = validateIdentifier(
@@ -4539,18 +4641,13 @@ 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)) {
-          const safeKey = escapeFilterKey(key);
-          filterConditions.push(`payload->>'${safeKey}' = $${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.col()}
@@ -4571,19 +4668,14 @@ 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)) {
-        const safeKey = escapeFilterKey(key);
-        filterConditions.push(`payload->>'${safeKey}' = $${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.col()}
@@ -4637,18 +4729,12 @@ 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)) {
-        const safeKey = escapeFilterKey(key);
-        filterConditions.push(`payload->>'${safeKey}' = $${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.col()}
@@ -4660,11 +4746,10 @@ var PGVector = class {
       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,
@@ -4988,7 +5073,7 @@ var parse_vision_messages = async (messages) => {
 };
 
 // src/oss/src/utils/telemetry.ts
-var version = true ? "3.0.3" : "dev";
+var version = true ? "3.0.4" : "dev";
 var MEM0_TELEMETRY = true;
 var _a;
 try {
@@ -7256,6 +7341,7 @@ export {
   RedisDB,
   SupabaseDB,
   VectorStoreFactory,
-  VectorizeDB
+  VectorizeDB,
+  buildFilterConditions
 };
 //# sourceMappingURL=index.mjs.map