@mastra/mongodb 1.18.6-alpha.0 → 1.18.6

package/dist/docs/SKILL.md

@@ -3,7 +3,7 @@ name: mastra-mongodb
 description: Documentation for @mastra/mongodb. Use when working with @mastra/mongodb APIs, configuration, or implementation.
 metadata:
   package: "@mastra/mongodb"
-  version: "1.18.6-alpha.0"
+  version: "1.18.6"
 ---
 
 ## When to use

package/dist/docs/assets/SOURCE_MAP.json

@@ -1,5 +1,5 @@
 {
-  "version": "1.18.6-alpha.0",
+  "version": "1.18.6",
   "package": "@mastra/mongodb",
   "exports": {},
   "modules": {}

package/dist/index.cjs

@@ -10,7 +10,7 @@ let _mastra_core_agent = require("@mastra/core/agent");
 let _mastra_core_evals = require("@mastra/core/evals");
 let _mastra_core_storage_domains_skills = require("@mastra/core/storage/domains/skills");
 //#region package.json
-var version = "1.18.6-alpha.0";
+var version = "1.18.6";
 //#endregion
 //#region src/vector/filter.ts
 /**
@@ -1396,6 +1396,10 @@ var MongoDBConnector = class MongoDBConnector {
 		if (this.#handler) return this.#handler.getCollection(collectionName);
 		return (await this.getConnection()).collection(collectionName);
 	}
+	async collectionExists(collectionName) {
+		if (this.#handler) return true;
+		return (await this.getConnection()).listCollections({ name: collectionName }).hasNext();
+	}
 	/**
 	* Returns true when the deployment supports multi-document transactions
 	* (replica set or sharded cluster). Standalone servers and custom connector
@@ -2792,6 +2796,8 @@ var MongoDBDatasetsStorage = class extends _mastra_core_storage.DatasetsStorage
 		};
 	}
 	transformItemRow(row) {
+		const metadata = (typeof row.metadata === "string" ? (0, _mastra_core_storage.safelyParseJSON)(row.metadata) : row.metadata) ?? void 0;
+		const emptyValue = metadata?.__purged === true ? null : void 0;
 		return {
 			id: row.id,
 			datasetId: row.datasetId,
@@ -2802,12 +2808,12 @@ var MongoDBDatasetsStorage = class extends _mastra_core_storage.DatasetsStorage
 			input: typeof row.input === "string" ? (0, _mastra_core_storage.safelyParseJSON)(row.input) : row.input,
 			groundTruth: typeof row.groundTruth === "string" ? (0, _mastra_core_storage.safelyParseJSON)(row.groundTruth) : row.groundTruth,
 			expectedTrajectory: typeof row.expectedTrajectory === "string" ? (0, _mastra_core_storage.safelyParseJSON)(row.expectedTrajectory) : row.expectedTrajectory,
-			toolMocks: (typeof row.toolMocks === "string" ? (0, _mastra_core_storage.safelyParseJSON)(row.toolMocks) : row.toolMocks) ?? void 0,
-			unmockedToolPolicy: row.unmockedToolPolicy ?? void 0,
-			scorerIds: (typeof row.scorerIds === "string" ? (0, _mastra_core_storage.safelyParseJSON)(row.scorerIds) : row.scorerIds) ?? void 0,
-			requestContext: typeof row.requestContext === "string" ? (0, _mastra_core_storage.safelyParseJSON)(row.requestContext) : row.requestContext,
-			metadata: typeof row.metadata === "string" ? (0, _mastra_core_storage.safelyParseJSON)(row.metadata) : row.metadata,
-			source: typeof row.source === "string" ? (0, _mastra_core_storage.safelyParseJSON)(row.source) : row.source,
+			toolMocks: (typeof row.toolMocks === "string" ? (0, _mastra_core_storage.safelyParseJSON)(row.toolMocks) : row.toolMocks) ?? emptyValue,
+			unmockedToolPolicy: row.unmockedToolPolicy ?? emptyValue,
+			scorerIds: (typeof row.scorerIds === "string" ? (0, _mastra_core_storage.safelyParseJSON)(row.scorerIds) : row.scorerIds) ?? emptyValue,
+			requestContext: (typeof row.requestContext === "string" ? (0, _mastra_core_storage.safelyParseJSON)(row.requestContext) : row.requestContext) ?? emptyValue,
+			metadata,
+			source: (typeof row.source === "string" ? (0, _mastra_core_storage.safelyParseJSON)(row.source) : row.source) ?? emptyValue,
 			createdAt: (0, _mastra_core_storage.ensureDate)(row.createdAt),
 			updatedAt: (0, _mastra_core_storage.ensureDate)(row.updatedAt)
 		};
@@ -3273,6 +3279,79 @@ var MongoDBDatasetsStorage = class extends _mastra_core_storage.DatasetsStorage
 			}, error);
 		}
 	}
+	async _doPurgeItem({ id, datasetId }) {
+		try {
+			if (!await this.#connector.supportsTransactions()) throw new _mastra_core_error.MastraError({
+				id: "MONGODB_DATASET_ITEM_PURGE_REQUIRES_TRANSACTIONS",
+				domain: _mastra_core_error.ErrorDomain.STORAGE,
+				category: _mastra_core_error.ErrorCategory.USER,
+				text: "Dataset item purge requires a MongoDB replica set or sharded deployment with transaction support.",
+				details: {
+					datasetId,
+					itemId: id
+				}
+			});
+			const itemsCollection = await this.getCollection(_mastra_core_storage.TABLE_DATASET_ITEMS);
+			const experimentsCollection = await this.getCollection(_mastra_core_storage.TABLE_EXPERIMENTS);
+			const experimentResultsCollection = await this.getCollection(_mastra_core_storage.TABLE_EXPERIMENT_RESULTS);
+			const experimentCollectionsExist = await this.#connector.collectionExists(_mastra_core_storage.TABLE_EXPERIMENTS) && await this.#connector.collectionExists(_mastra_core_storage.TABLE_EXPERIMENT_RESULTS);
+			const metadata = {
+				__purged: true,
+				purgedAt: (/* @__PURE__ */ new Date()).toISOString()
+			};
+			await this.#connector.withTransaction(async (session) => {
+				if (!await itemsCollection.findOneAndUpdate({
+					id,
+					datasetId
+				}, { $inc: { purgeBarrierRevision: 1 } }, {
+					projection: { id: 1 },
+					returnDocument: "after",
+					session,
+					sort: { datasetVersion: -1 }
+				})) return;
+				await itemsCollection.updateMany({
+					id,
+					datasetId
+				}, { $set: {
+					input: null,
+					groundTruth: null,
+					expectedTrajectory: null,
+					toolMocks: null,
+					unmockedToolPolicy: null,
+					scorerIds: null,
+					requestContext: null,
+					metadata,
+					source: null
+				} }, { session });
+				if (experimentCollectionsExist) {
+					const experimentIds = await experimentsCollection.find({ datasetId }, {
+						projection: { id: 1 },
+						session
+					}).map((experiment) => experiment.id).toArray();
+					if (experimentIds.length > 0) await experimentResultsCollection.updateMany({
+						itemId: id,
+						experimentId: { $in: experimentIds }
+					}, { $set: {
+						input: null,
+						output: null,
+						groundTruth: null,
+						error: null,
+						toolMockReport: null,
+						tags: null,
+						comment: null,
+						metadata
+					} }, { session });
+				}
+			});
+		} catch (error) {
+			if (error instanceof _mastra_core_error.MastraError) throw error;
+			throw new _mastra_core_error.MastraError({
+				id: (0, _mastra_core_storage.createStorageErrorId)("MONGODB", "PURGE_ITEM", "FAILED"),
+				domain: _mastra_core_error.ErrorDomain.STORAGE,
+				category: _mastra_core_error.ErrorCategory.THIRD_PARTY
+			}, error);
+		}
+	}
 	async _doBatchInsertItems(input) {
 		try {
 			if (input.items.length === 0) return [];
@@ -3704,7 +3783,7 @@ function transformExperimentResultRow(row) {
 		itemDatasetVersion: row.itemDatasetVersion != null ? Number(row.itemDatasetVersion) : null,
 		organizationId: row.organizationId ?? null,
 		projectId: row.projectId ?? null,
-		input: parseJsonField(row.input),
+		input: row.input === null ? null : parseJsonField(row.input),
 		output: parseJsonField(row.output) ?? null,
 		groundTruth: parseJsonField(row.groundTruth) ?? null,
 		metadata: parseJsonField(row.metadata) ?? null,
@@ -3746,6 +3825,27 @@ var MongoDBExperimentsStorage = class MongoDBExperimentsStorage extends _mastra_
 	async getCollection(name) {
 		return this.#connector.getCollection(name);
 	}
+	async #withPurgeBarrier(experimentId, itemId, fn) {
+		const experiments = await this.getCollection(_mastra_core_storage.TABLE_EXPERIMENTS);
+		const items = await this.getCollection(_mastra_core_storage.TABLE_DATASET_ITEMS);
+		return this.#connector.withTransaction(async (session) => {
+			const datasetId = (await experiments.findOne({ id: experimentId }, { session }))?.datasetId ?? null;
+			let purgeMetadata = null;
+			if (datasetId) {
+				const item = await items.findOneAndUpdate({
+					id: itemId,
+					datasetId
+				}, { $inc: { purgeBarrierRevision: 1 } }, {
+					projection: { metadata: 1 },
+					returnDocument: "after",
+					session,
+					sort: { datasetVersion: -1 }
+				});
+				purgeMetadata = item?.metadata?.__purged === true ? item.metadata : null;
+			}
+			return fn(session, purgeMetadata);
+		});
+	}
 	/**
 	* Prune whole experiments older than the `experiments` policy's `maxAge`.
 	*
@@ -4081,52 +4181,34 @@ var MongoDBExperimentsStorage = class MongoDBExperimentsStorage extends _mastra_
 	async addExperimentResult(input) {
 		const id = input.id ?? (0, crypto$1.randomUUID)();
 		const now = /* @__PURE__ */ new Date();
-		const doc = {
-			id,
-			experimentId: input.experimentId,
-			itemId: input.itemId,
-			itemDatasetVersion: input.itemDatasetVersion ?? null,
-			organizationId: input.organizationId ?? null,
-			projectId: input.projectId ?? null,
-			input: input.input,
-			output: input.output ?? null,
-			groundTruth: input.groundTruth ?? null,
-			metadata: input.metadata ?? null,
-			error: input.error ?? null,
-			startedAt: input.startedAt,
-			completedAt: input.completedAt,
-			retryCount: input.retryCount,
-			attempt: input.attempt ?? 0,
-			traceId: input.traceId ?? null,
-			status: input.status ?? null,
-			tags: input.tags ?? null,
-			toolMockReport: input.toolMockReport ?? null,
-			createdAt: now
-		};
 		try {
-			await (await this.getCollection(_mastra_core_storage.TABLE_EXPERIMENT_RESULTS)).insertOne(doc);
-			return {
-				id,
-				experimentId: input.experimentId,
-				itemId: input.itemId,
-				itemDatasetVersion: input.itemDatasetVersion ?? null,
-				organizationId: input.organizationId ?? null,
-				projectId: input.projectId ?? null,
-				input: input.input,
-				output: input.output ?? null,
-				groundTruth: input.groundTruth ?? null,
-				metadata: input.metadata ?? null,
-				error: input.error ?? null,
-				startedAt: input.startedAt,
-				completedAt: input.completedAt,
-				retryCount: input.retryCount,
-				attempt: input.attempt ?? 0,
-				traceId: input.traceId ?? null,
-				status: input.status ?? null,
-				tags: input.tags ?? null,
-				toolMockReport: input.toolMockReport ?? null,
-				createdAt: now
-			};
+			return await this.#withPurgeBarrier(input.experimentId, input.itemId, async (session, marker) => {
+				const collection = await this.getCollection(_mastra_core_storage.TABLE_EXPERIMENT_RESULTS);
+				const doc = {
+					id,
+					experimentId: input.experimentId,
+					itemId: input.itemId,
+					itemDatasetVersion: input.itemDatasetVersion ?? null,
+					organizationId: input.organizationId ?? null,
+					projectId: input.projectId ?? null,
+					input: marker ? null : input.input,
+					output: marker ? null : input.output ?? null,
+					groundTruth: marker ? null : input.groundTruth ?? null,
+					metadata: marker ?? input.metadata ?? null,
+					error: marker ? null : input.error ?? null,
+					startedAt: input.startedAt,
+					completedAt: input.completedAt,
+					retryCount: input.retryCount,
+					attempt: input.attempt ?? 0,
+					traceId: input.traceId ?? null,
+					status: input.status ?? null,
+					tags: marker ? null : input.tags ?? null,
+					toolMockReport: marker ? null : input.toolMockReport ?? null,
+					createdAt: now
+				};
+				await collection.insertOne(doc, { session });
+				return transformExperimentResultRow(doc);
+			});
 		} catch (error) {
 			throw new _mastra_core_error.MastraError({
 				id: (0, _mastra_core_storage.createStorageErrorId)("MONGODB", "ADD_EXPERIMENT_RESULT", "FAILED"),
@@ -4139,37 +4221,41 @@ var MongoDBExperimentsStorage = class MongoDBExperimentsStorage extends _mastra_
 	async upsertExperimentResult(input) {
 		const attempt = input.attempt ?? 0;
 		try {
-			return transformExperimentResultRow(await (await this.getCollection(_mastra_core_storage.TABLE_EXPERIMENT_RESULTS)).findOneAndUpdate({
-				experimentId: input.experimentId,
-				itemId: input.itemId,
-				$or: [{ attempt }, ...attempt === 0 ? [{ attempt: null }, { attempt: { $exists: false } }] : []]
-			}, {
-				$set: {
-					itemDatasetVersion: input.itemDatasetVersion ?? null,
-					organizationId: input.organizationId ?? null,
-					projectId: input.projectId ?? null,
-					input: input.input,
-					output: input.output ?? null,
-					groundTruth: input.groundTruth ?? null,
-					metadata: input.metadata ?? null,
-					error: input.error ?? null,
-					startedAt: input.startedAt,
-					completedAt: input.completedAt,
-					retryCount: input.retryCount,
-					attempt,
-					traceId: input.traceId ?? null,
-					status: input.status ?? null,
-					tags: input.tags ?? null,
-					toolMockReport: input.toolMockReport ?? null
-				},
-				$setOnInsert: {
-					id: (0, crypto$1.randomUUID)(),
-					createdAt: /* @__PURE__ */ new Date()
-				}
-			}, {
-				upsert: true,
-				returnDocument: "after"
-			}));
+			return await this.#withPurgeBarrier(input.experimentId, input.itemId, async (session, marker) => {
+				return transformExperimentResultRow(await (await this.getCollection(_mastra_core_storage.TABLE_EXPERIMENT_RESULTS)).findOneAndUpdate({
+					experimentId: input.experimentId,
+					itemId: input.itemId,
+					$or: [{ attempt }, ...attempt === 0 ? [{ attempt: null }, { attempt: { $exists: false } }] : []]
+				}, {
+					$set: {
+						itemDatasetVersion: input.itemDatasetVersion ?? null,
+						organizationId: input.organizationId ?? null,
+						projectId: input.projectId ?? null,
+						input: marker ? null : input.input,
+						output: marker ? null : input.output ?? null,
+						groundTruth: marker ? null : input.groundTruth ?? null,
+						metadata: marker ?? input.metadata ?? null,
+						error: marker ? null : input.error ?? null,
+						startedAt: input.startedAt,
+						completedAt: input.completedAt,
+						retryCount: input.retryCount,
+						attempt,
+						traceId: input.traceId ?? null,
+						status: input.status ?? null,
+						tags: marker ? null : input.tags ?? null,
+						toolMockReport: marker ? null : input.toolMockReport ?? null,
+						...marker ? { comment: null } : {}
+					},
+					$setOnInsert: {
+						id: (0, crypto$1.randomUUID)(),
+						createdAt: /* @__PURE__ */ new Date()
+					}
+				}, {
+					upsert: true,
+					returnDocument: "after",
+					session
+				}));
+			});
 		} catch (error) {
 			throw new _mastra_core_error.MastraError({
 				id: (0, _mastra_core_storage.createStorageErrorId)("MONGODB", "UPSERT_EXPERIMENT_RESULT", "FAILED"),
@@ -4198,14 +4284,30 @@ var MongoDBExperimentsStorage = class MongoDBExperimentsStorage extends _mastra_
 			const collection = await this.getCollection(_mastra_core_storage.TABLE_EXPERIMENT_RESULTS);
 			const filter = { id: input.id };
 			if (input.experimentId) filter.experimentId = input.experimentId;
-			const result = await collection.findOneAndUpdate(filter, { $set: updateFields }, { returnDocument: "after" });
-			if (!result) throw new _mastra_core_error.MastraError({
+			const existing = await collection.findOne(filter);
+			if (!existing) throw new _mastra_core_error.MastraError({
 				id: (0, _mastra_core_storage.createStorageErrorId)("MONGODB", "UPDATE_EXPERIMENT_RESULT", "NOT_FOUND"),
 				domain: _mastra_core_error.ErrorDomain.STORAGE,
 				category: _mastra_core_error.ErrorCategory.USER,
 				details: { resultId: input.id }
 			});
-			return transformExperimentResultRow(result);
+			return await this.#withPurgeBarrier(existing.experimentId, existing.itemId, async (session, marker) => {
+				if (marker) {
+					updateFields.tags = null;
+					updateFields.comment = null;
+				}
+				const result = await collection.findOneAndUpdate(filter, { $set: updateFields }, {
+					returnDocument: "after",
+					session
+				});
+				if (!result) throw new _mastra_core_error.MastraError({
+					id: (0, _mastra_core_storage.createStorageErrorId)("MONGODB", "UPDATE_EXPERIMENT_RESULT", "NOT_FOUND"),
+					domain: _mastra_core_error.ErrorDomain.STORAGE,
+					category: _mastra_core_error.ErrorCategory.USER,
+					details: { resultId: input.id }
+				});
+				return transformExperimentResultRow(result);
+			});
 		} catch (error) {
 			if (error instanceof _mastra_core_error.MastraError) throw error;
 			throw new _mastra_core_error.MastraError({