@mastra/mongodb 1.20.1-alpha.1 → 1.21.0

package/dist/index.js

@@ -8,7 +8,7 @@ import { MessageList } from "@mastra/core/agent";
 import { saveScorePayloadSchema } from "@mastra/core/evals";
 import { skillSnapshotFieldValuesEqual } from "@mastra/core/storage/domains/skills";
 //#region package.json
-var version = "1.20.1-alpha.1";
+var version = "1.21.0";
 //#endregion
 //#region src/vector/filter.ts
 /**
@@ -6464,6 +6464,7 @@ const OM_TABLE = "mastra_observational_memory";
 var MemoryStorageMongoDB = class MemoryStorageMongoDB extends MemoryStorage {
 	supportsPartialThreadUpdate = true;
 	supportsObservationalMemory = true;
+	supportsObservationalMemoryHistorySearch = true;
 	#connector;
 	#skipDefaultIndexes;
 	#indexes;
@@ -7504,13 +7505,34 @@ var MemoryStorageMongoDB = class MemoryStorageMongoDB extends MemoryStorage {
 			const lookupKey = this.getOMKey(threadId, resourceId);
 			const collection = await this.getCollection(OM_TABLE);
 			const filter = { lookupKey };
+			if (options?.recordId !== void 0) filter["id"] = options.recordId;
 			if (options?.from || options?.to) {
 				const createdAtFilter = {};
 				if (options.from) createdAtFilter["$gte"] = options.from;
 				if (options.to) createdAtFilter["$lte"] = options.to;
 				filter["createdAt"] = createdAtFilter;
 			}
-			let cursor = collection.find(filter).sort({ generationCount: -1 });
+			if (options?.groupId !== void 0) {
+				const prefix = { $literal: `<observation-group id="${options.groupId}"` };
+				filter["$expr"] = { $or: [{ $gte: [{ $indexOfCP: [{ $ifNull: ["$activeObservations", ""] }, prefix] }, 0] }, { $anyElementTrue: [{ $map: {
+					input: { $cond: [
+						{ $isArray: "$bufferedObservationChunks" },
+						"$bufferedObservationChunks",
+						[]
+					] },
+					as: "chunk",
+					in: { $gte: [{ $indexOfCP: [{ $ifNull: ["$$chunk.observations", ""] }, prefix] }, 0] }
+				} }] }] };
+			}
+			if (options?.beforeGeneration !== void 0 || options?.afterGeneration !== void 0) filter["generationCount"] = {
+				...options.beforeGeneration !== void 0 ? { $lt: options.beforeGeneration } : {},
+				...options.afterGeneration !== void 0 ? { $gt: options.afterGeneration } : {}
+			};
+			let cursor = collection.find(filter).sort({
+				generationCount: options?.sortDirection === "ASC" ? 1 : -1,
+				createdAt: 1,
+				id: 1
+			});
 			if (options?.offset != null) cursor = cursor.skip(options.offset);
 			return (await cursor.limit(limit).toArray()).map((doc) => this.parseOMDocument(doc));
 		} catch (error) {