@mastra/mongodb 1.18.7 → 1.18.8-alpha.1

package/dist/index.js

@@ -9,7 +9,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.18.7";
+var version = "1.18.8-alpha.1";
 //#endregion
 //#region src/vector/filter.ts
 /**
@@ -3096,42 +3096,65 @@ var MongoDBDatasetsStorage = class extends DatasetsStorage {
 	}
 	async _doUpdateItem(args) {
 		try {
-			const existing = await this.getItemById({ id: args.id });
-			if (!existing) throw new MastraError({
-				id: createStorageErrorId("MONGODB", "UPDATE_ITEM", "NOT_FOUND"),
-				domain: ErrorDomain.STORAGE,
-				category: ErrorCategory.USER,
-				details: { itemId: args.id }
-			});
-			if (existing.datasetId !== args.datasetId) throw new MastraError({
-				id: createStorageErrorId("MONGODB", "UPDATE_ITEM", "DATASET_MISMATCH"),
-				domain: ErrorDomain.STORAGE,
-				category: ErrorCategory.USER,
-				details: {
-					itemId: args.id,
-					expectedDatasetId: args.datasetId,
-					actualDatasetId: existing.datasetId
-				}
-			});
-			if (!(args.input !== void 0 || args.groundTruth !== void 0 || args.expectedTrajectory !== void 0 || args.toolMocks !== void 0 || args.unmockedToolPolicy !== void 0 || args.scorerIds !== void 0 || args.requestContext !== void 0 || args.metadata !== void 0 || args.source !== void 0)) return existing;
+			if (!(args.input !== void 0 || args.groundTruth !== void 0 || args.expectedTrajectory !== void 0 || args.toolMocks !== void 0 || args.unmockedToolPolicy !== void 0 || args.scorerIds !== void 0 || args.requestContext !== void 0 || args.metadata !== void 0 || args.source !== void 0)) {
+				const existing = await this.getItemById({ id: args.id });
+				if (!existing) throw new MastraError({
+					id: createStorageErrorId("MONGODB", "UPDATE_ITEM", "NOT_FOUND"),
+					domain: ErrorDomain.STORAGE,
+					category: ErrorCategory.USER,
+					details: { itemId: args.id }
+				});
+				if (existing.datasetId !== args.datasetId) throw new MastraError({
+					id: createStorageErrorId("MONGODB", "UPDATE_ITEM", "DATASET_MISMATCH"),
+					domain: ErrorDomain.STORAGE,
+					category: ErrorCategory.USER,
+					details: {
+						itemId: args.id,
+						expectedDatasetId: args.datasetId,
+						actualDatasetId: existing.datasetId
+					}
+				});
+				return existing;
+			}
 			const now = /* @__PURE__ */ new Date();
 			const versionId = randomUUID();
-			const mergedInput = args.input !== void 0 ? args.input : existing.input;
-			const mergedGroundTruth = args.groundTruth !== void 0 ? args.groundTruth : existing.groundTruth;
-			const mergedExpectedTrajectory = args.expectedTrajectory !== void 0 ? args.expectedTrajectory : existing.expectedTrajectory;
-			const mergedToolMocks = args.toolMocks !== void 0 ? args.toolMocks : existing.toolMocks;
-			const mergedUnmockedToolPolicy = args.unmockedToolPolicy !== void 0 ? args.unmockedToolPolicy : existing.unmockedToolPolicy;
-			const mergedScorerIds = args.scorerIds !== void 0 ? args.scorerIds ?? void 0 : existing.scorerIds;
-			const mergedRequestContext = args.requestContext !== void 0 ? args.requestContext : existing.requestContext;
-			const mergedMetadata = args.metadata !== void 0 ? args.metadata : existing.metadata;
-			const mergedSource = args.source !== void 0 ? args.source : existing.source;
 			const datasetsCollection = await this.getCollection(TABLE_DATASETS);
 			const itemsCollection = await this.getCollection(TABLE_DATASET_ITEMS);
 			const versionsCollection = await this.getCollection(TABLE_DATASET_VERSIONS);
-			let newVersion = 0;
-			let parentOrganizationId = null;
-			let parentProjectId = null;
+			let updated;
 			await this.#connector.withTransaction(async (session) => {
+				const row = await itemsCollection.findOne({
+					id: args.id,
+					validTo: null,
+					isDeleted: false
+				}, { session });
+				const existing = row ? this.transformItemRow(row) : null;
+				if (!existing) throw new MastraError({
+					id: createStorageErrorId("MONGODB", "UPDATE_ITEM", "NOT_FOUND"),
+					domain: ErrorDomain.STORAGE,
+					category: ErrorCategory.USER,
+					details: { itemId: args.id }
+				});
+				if (existing.datasetId !== args.datasetId) throw new MastraError({
+					id: createStorageErrorId("MONGODB", "UPDATE_ITEM", "DATASET_MISMATCH"),
+					domain: ErrorDomain.STORAGE,
+					category: ErrorCategory.USER,
+					details: {
+						itemId: args.id,
+						expectedDatasetId: args.datasetId,
+						actualDatasetId: existing.datasetId
+					}
+				});
+				if (existing.metadata?.__purged === true) throw new MastraError({
+					id: "DATASET_ITEM_PURGED",
+					domain: ErrorDomain.STORAGE,
+					category: ErrorCategory.USER,
+					details: {
+						datasetId: args.datasetId,
+						itemId: args.id
+					},
+					text: `Purged dataset item cannot be updated: ${args.id}`
+				});
 				const result = await datasetsCollection.findOneAndUpdate({ id: args.datasetId }, { $inc: { version: 1 } }, {
 					session,
 					returnDocument: "after"
@@ -3142,9 +3165,18 @@ var MongoDBDatasetsStorage = class extends DatasetsStorage {
 					category: ErrorCategory.USER,
 					details: { datasetId: args.datasetId }
 				});
-				newVersion = result.version;
-				parentOrganizationId = result.organizationId ?? null;
-				parentProjectId = result.projectId ?? null;
+				const mergedInput = args.input !== void 0 ? args.input : existing.input;
+				const mergedGroundTruth = args.groundTruth !== void 0 ? args.groundTruth : existing.groundTruth;
+				const mergedExpectedTrajectory = args.expectedTrajectory !== void 0 ? args.expectedTrajectory : existing.expectedTrajectory;
+				const mergedToolMocks = args.toolMocks !== void 0 ? args.toolMocks : existing.toolMocks;
+				const mergedUnmockedToolPolicy = args.unmockedToolPolicy !== void 0 ? args.unmockedToolPolicy : existing.unmockedToolPolicy;
+				const mergedScorerIds = args.scorerIds !== void 0 ? args.scorerIds ?? void 0 : existing.scorerIds;
+				const mergedRequestContext = args.requestContext !== void 0 ? args.requestContext : existing.requestContext;
+				const mergedMetadata = args.metadata !== void 0 ? args.metadata : existing.metadata;
+				const mergedSource = args.source !== void 0 ? args.source : existing.source;
+				const newVersion = result.version;
+				const parentOrganizationId = result.organizationId ?? null;
+				const parentProjectId = result.projectId ?? null;
 				await itemsCollection.updateOne({
 					id: args.id,
 					validTo: null,
@@ -3177,23 +3209,24 @@ var MongoDBDatasetsStorage = class extends DatasetsStorage {
 					version: newVersion,
 					createdAt: now
 				}, { session });
+				updated = {
+					...existing,
+					datasetVersion: newVersion,
+					organizationId: parentOrganizationId,
+					projectId: parentProjectId,
+					input: mergedInput,
+					groundTruth: mergedGroundTruth,
+					expectedTrajectory: mergedExpectedTrajectory,
+					toolMocks: mergedToolMocks,
+					unmockedToolPolicy: mergedUnmockedToolPolicy,
+					scorerIds: mergedScorerIds,
+					requestContext: mergedRequestContext,
+					metadata: mergedMetadata,
+					source: mergedSource,
+					updatedAt: now
+				};
 			});
-			return {
-				...existing,
-				datasetVersion: newVersion,
-				organizationId: parentOrganizationId,
-				projectId: parentProjectId,
-				input: mergedInput,
-				groundTruth: mergedGroundTruth,
-				expectedTrajectory: mergedExpectedTrajectory,
-				toolMocks: mergedToolMocks,
-				unmockedToolPolicy: mergedUnmockedToolPolicy,
-				scorerIds: mergedScorerIds,
-				requestContext: mergedRequestContext,
-				metadata: mergedMetadata,
-				source: mergedSource,
-				updatedAt: now
-			};
+			return updated;
 		} catch (error) {
 			if (error instanceof MastraError) throw error;
 			throw new MastraError({
@@ -3205,24 +3238,29 @@ var MongoDBDatasetsStorage = class extends DatasetsStorage {
 	}
 	async _doDeleteItem({ id, datasetId }) {
 		try {
-			const existing = await this.getItemById({ id });
-			if (!existing) return;
-			if (existing.datasetId !== datasetId) throw new MastraError({
-				id: createStorageErrorId("MONGODB", "DELETE_ITEM", "DATASET_MISMATCH"),
-				domain: ErrorDomain.STORAGE,
-				category: ErrorCategory.USER,
-				details: {
-					itemId: id,
-					expectedDatasetId: datasetId,
-					actualDatasetId: existing.datasetId
-				}
-			});
 			const now = /* @__PURE__ */ new Date();
 			const versionId = randomUUID();
 			const datasetsCollection = await this.getCollection(TABLE_DATASETS);
 			const itemsCollection = await this.getCollection(TABLE_DATASET_ITEMS);
 			const versionsCollection = await this.getCollection(TABLE_DATASET_VERSIONS);
 			await this.#connector.withTransaction(async (session) => {
+				const row = await itemsCollection.findOne({
+					id,
+					validTo: null,
+					isDeleted: false
+				}, { session });
+				const existing = row ? this.transformItemRow(row) : null;
+				if (!existing) return;
+				if (existing.datasetId !== datasetId) throw new MastraError({
+					id: createStorageErrorId("MONGODB", "DELETE_ITEM", "DATASET_MISMATCH"),
+					domain: ErrorDomain.STORAGE,
+					category: ErrorCategory.USER,
+					details: {
+						itemId: id,
+						expectedDatasetId: datasetId,
+						actualDatasetId: existing.datasetId
+					}
+				});
 				const result = await datasetsCollection.findOneAndUpdate({ id: datasetId }, { $inc: { version: 1 } }, {
 					session,
 					returnDocument: "after"
@@ -3444,19 +3482,19 @@ var MongoDBDatasetsStorage = class extends DatasetsStorage {
 				details: { datasetId: input.datasetId }
 			});
 			const itemsCollection = await this.getCollection(TABLE_DATASET_ITEMS);
-			const currentItems = (await itemsCollection.find({
-				id: { $in: input.itemIds },
-				datasetId: input.datasetId,
-				validTo: null,
-				isDeleted: false
-			}).toArray()).map((row) => this.transformItemRow(row));
-			if (currentItems.length === 0) return;
 			const now = /* @__PURE__ */ new Date();
 			const versionId = randomUUID();
 			const datasetsCollection = await this.getCollection(TABLE_DATASETS);
 			const versionsCollection = await this.getCollection(TABLE_DATASET_VERSIONS);
-			const currentIds = currentItems.map((i) => i.id);
 			await this.#connector.withTransaction(async (session) => {
+				const currentItems = (await itemsCollection.find({
+					id: { $in: input.itemIds },
+					datasetId: input.datasetId,
+					validTo: null,
+					isDeleted: false
+				}, { session }).toArray()).map((row) => this.transformItemRow(row));
+				if (currentItems.length === 0) return;
+				const currentIds = currentItems.map((i) => i.id);
 				const result = await datasetsCollection.findOneAndUpdate({ id: input.datasetId }, { $inc: { version: 1 } }, {
 					session,
 					returnDocument: "after"
@@ -7937,6 +7975,8 @@ var MemoryStorageMongoDB = class MemoryStorageMongoDB extends MemoryStorage {
 };
 //#endregion
 //#region src/storage/domains/notifications/index.ts
+/** Ids per updateMany/find command — keeps the `$in` filter far below MongoDB's 16 MiB BSON limit. */
+const BULK_ID_BATCH_SIZE = 500;
 const statusTimestamp = (status, now) => {
 	if (status === "delivered") return { deliveredAt: now };
 	if (status === "seen") return { seenAt: now };
@@ -8267,6 +8307,31 @@ var NotificationsMongoDB = class NotificationsMongoDB extends NotificationsStora
 		if (!updated) throw new Error(`Notification ${input.id} was not found for thread ${input.threadId}`);
 		return updated;
 	}
+	async updateNotificationsStatus(input) {
+		const ids = Array.from(new Set(input.ids));
+		if (ids.length === 0) return [];
+		const now = /* @__PURE__ */ new Date();
+		const update = { $set: {
+			status: input.status,
+			...statusTimestamp(input.status, now),
+			updatedAt: now
+		} };
+		const collection = await this.getCollection();
+		const updated = [];
+		for (let offset = 0; offset < ids.length; offset += BULK_ID_BATCH_SIZE) {
+			const filter = {
+				threadId: input.threadId,
+				id: { $in: ids.slice(offset, offset + BULK_ID_BATCH_SIZE) }
+			};
+			await collection.updateMany(filter, update);
+			const rows = await collection.find({
+				...filter,
+				status: input.status
+			}).toArray();
+			for (const row of rows) updated.push(rowToNotification(row));
+		}
+		return updated;
+	}
 	async findCoalescable(input) {
 		if (!input.dedupeKey && !input.coalesceKey) return void 0;
 		const row = await (await this.getCollection()).findOne({