@mastra/mongodb 0.13.0-alpha.2 → 0.13.0

package/dist/index.js

@@ -324,11 +324,12 @@ var MongoDBVector = class extends MastraVector {
       const indexNameInternal = `${indexName}_vector_index`;
       const mongoFilter = this.transformFilter(filter);
       const documentMongoFilter = documentFilter ? { [this.documentFieldName]: documentFilter } : {};
+      const transformedMongoFilter = this.transformMetadataFilter(mongoFilter);
       let combinedFilter = {};
-      if (Object.keys(mongoFilter).length > 0 && Object.keys(documentMongoFilter).length > 0) {
-        combinedFilter = { $and: [mongoFilter, documentMongoFilter] };
-      } else if (Object.keys(mongoFilter).length > 0) {
-        combinedFilter = mongoFilter;
+      if (Object.keys(transformedMongoFilter).length > 0 && Object.keys(documentMongoFilter).length > 0) {
+        combinedFilter = { $and: [transformedMongoFilter, documentMongoFilter] };
+      } else if (Object.keys(transformedMongoFilter).length > 0) {
+        combinedFilter = transformedMongoFilter;
       } else if (Object.keys(documentMongoFilter).length > 0) {
         combinedFilter = documentMongoFilter;
       }
@@ -343,6 +344,8 @@ var MongoDBVector = class extends MastraVector {
         const candidateIds = await collection.aggregate([{ $match: combinedFilter }, { $project: { _id: 1 } }]).map((doc) => doc._id).toArray();
         if (candidateIds.length > 0) {
           vectorSearch.filter = { _id: { $in: candidateIds } };
+        } else {
+          return [];
         }
       }
       const pipeline = [
@@ -565,6 +568,45 @@ var MongoDBVector = class extends MastraVector {
     if (!filter) return {};
     return translator.translate(filter);
   }
+  /**
+   * Transform metadata field filters to use MongoDB dot notation.
+   * Fields that are stored in the metadata subdocument need to be prefixed with 'metadata.'
+   * This handles filters from the Memory system which expects direct field access.
+   *
+   * @param filter - The filter object to transform
+   * @returns Transformed filter with metadata fields properly prefixed
+   */
+  transformMetadataFilter(filter) {
+    if (!filter || typeof filter !== "object") return filter;
+    const transformed = {};
+    for (const [key, value] of Object.entries(filter)) {
+      if (key.startsWith("$")) {
+        if (Array.isArray(value)) {
+          transformed[key] = value.map((item) => this.transformMetadataFilter(item));
+        } else {
+          transformed[key] = this.transformMetadataFilter(value);
+        }
+      } else if (key.startsWith("metadata.")) {
+        transformed[key] = value;
+      } else if (this.isMetadataField(key)) {
+        transformed[`metadata.${key}`] = value;
+      } else {
+        transformed[key] = value;
+      }
+    }
+    return transformed;
+  }
+  /**
+   * Determine if a field should be treated as a metadata field.
+   * Common metadata fields include thread_id, resource_id, message_id, and any field
+   * that doesn't start with underscore (MongoDB system fields).
+   */
+  isMetadataField(key) {
+    if (key.startsWith("_")) return false;
+    const documentFields = ["_id", this.embeddingFieldName, this.documentFieldName];
+    if (documentFields.includes(key)) return false;
+    return true;
+  }
 };
 var MongoDBConnector = class _MongoDBConnector {
   #client;