@mastra/mongodb 0.0.0-zod-v4-compat-part-2-20250820135355 → 0.0.0-zod-v4-stuff-20250825154219

package/dist/index.d.ts

@@ -1,5 +1,5 @@
-export * from './vector';
-export * from './storage';
-export * from './storage/connectors/base';
-export { MONGODB_PROMPT } from './vector/prompt';
+export * from './vector/index.js';
+export * from './storage/index.js';
+export * from './storage/connectors/base.js';
+export { MONGODB_PROMPT } from './vector/prompt.js';
 //# sourceMappingURL=index.d.ts.map

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;
@@ -881,6 +923,29 @@ var MemoryStorageMongoDB = class extends MemoryStorage {
       );
     }
   }
+  async getMessagesById({
+    messageIds,
+    format
+  }) {
+    if (messageIds.length === 0) return [];
+    try {
+      const collection = await this.operations.getCollection(TABLE_MESSAGES);
+      const rawMessages = await collection.find({ id: { $in: messageIds } }).sort({ createdAt: -1 }).toArray();
+      const list = new MessageList().add(rawMessages.map(this.parseRow), "memory");
+      if (format === `v1`) return list.get.all.v1();
+      return list.get.all.v2();
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "MONGODB_STORE_GET_MESSAGES_BY_ID_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { messageIds: JSON.stringify(messageIds) }
+        },
+        error
+      );
+    }
+  }
   async getMessagesPaginated(args) {
     const { threadId, format, selectBy } = args;
     const { page = 0, perPage: perPageInput, dateRange } = selectBy?.pagination || {};
@@ -1490,12 +1555,12 @@ function transformScoreRow(row) {
       console.warn("Failed to parse scorer:", e);
     }
   }
-  let extractStepResultValue = null;
-  if (row.extractStepResult) {
+  let preprocessStepResultValue = null;
+  if (row.preprocessStepResult) {
     try {
-      extractStepResultValue = typeof row.extractStepResult === "string" ? safelyParseJSON(row.extractStepResult) : row.extractStepResult;
+      preprocessStepResultValue = typeof row.preprocessStepResult === "string" ? safelyParseJSON(row.preprocessStepResult) : row.preprocessStepResult;
     } catch (e) {
-      console.warn("Failed to parse extractStepResult:", e);
+      console.warn("Failed to parse preprocessStepResult:", e);
     }
   }
   let analyzeStepResultValue = null;
@@ -1546,7 +1611,7 @@ function transformScoreRow(row) {
     traceId: row.traceId,
     runId: row.runId,
     scorer: scorerValue,
-    extractStepResult: extractStepResultValue,
+    preprocessStepResult: preprocessStepResultValue,
     analyzeStepResult: analyzeStepResultValue,
     score: row.score,
     reason: row.reason,
@@ -1603,11 +1668,13 @@ var ScoresStorageMongoDB = class extends ScoresStorage {
         traceId: score.traceId || "",
         runId: score.runId,
         scorer: typeof score.scorer === "string" ? safelyParseJSON(score.scorer) : score.scorer,
-        extractStepResult: typeof score.extractStepResult === "string" ? safelyParseJSON(score.extractStepResult) : score.extractStepResult,
+        preprocessStepResult: typeof score.preprocessStepResult === "string" ? safelyParseJSON(score.preprocessStepResult) : score.preprocessStepResult,
         analyzeStepResult: typeof score.analyzeStepResult === "string" ? safelyParseJSON(score.analyzeStepResult) : score.analyzeStepResult,
         score: score.score,
         reason: score.reason,
-        extractPrompt: score.extractPrompt,
+        preprocessPrompt: score.preprocessPrompt,
+        generateScorePrompt: score.generateScorePrompt,
+        generateReasonPrompt: score.generateReasonPrompt,
         analyzePrompt: score.analyzePrompt,
         reasonPrompt: score.reasonPrompt,
         input: typeof score.input === "string" ? safelyParseJSON(score.input) : score.input,
@@ -1646,7 +1713,8 @@ var ScoresStorageMongoDB = class extends ScoresStorage {
     scorerId,
     pagination,
     entityId,
-    entityType
+    entityType,
+    source
   }) {
     try {
       const query = { scorerId };
@@ -1656,6 +1724,9 @@ var ScoresStorageMongoDB = class extends ScoresStorage {
       if (entityType) {
         query.entityType = entityType;
       }
+      if (source) {
+        query.source = source;
+      }
       const collection = await this.operations.getCollection(TABLE_SCORERS);
       const total = await collection.countDocuments(query);
       const currentOffset = pagination.page * pagination.perPage;
@@ -2185,6 +2256,12 @@ var MongoDBStore = class extends MastraStorage {
   }) {
     return this.stores.memory.getMessages({ threadId, selectBy, format });
   }
+  async getMessagesById({
+    messageIds,
+    format
+  }) {
+    return this.stores.memory.getMessagesById({ messageIds, format });
+  }
   async saveMessages(args) {
     return this.stores.memory.saveMessages(args);
   }
@@ -2271,9 +2348,10 @@ var MongoDBStore = class extends MastraStorage {
     scorerId,
     pagination,
     entityId,
-    entityType
+    entityType,
+    source
   }) {
-    return this.stores.scores.getScoresByScorerId({ scorerId, pagination, entityId, entityType });
+    return this.stores.scores.getScoresByScorerId({ scorerId, pagination, entityId, entityType, source });
   }
   /**
    * RESOURCES