@mastra/spanner 1.6.6-alpha.0 → 1.6.7-alpha.0

package/dist/index.js

@@ -3325,8 +3325,8 @@ var DatasetsSpanner = class DatasetsSpanner extends DatasetsStorage {
 		});
 	}
 	/** Reads the current live row for an item (validTo IS NULL, not deleted). */
-	async loadCurrentItemRow(itemId) {
-		const [rows] = await this.database.run({
+	async loadCurrentItemRow(tx, itemId) {
+		const [rows] = await tx.run({
 			sql: `SELECT * FROM ${quoteIdent(TABLE_DATASET_ITEMS, "table name")}
             WHERE ${quoteIdent("id", "column name")} = @id
               AND ${quoteIdent("validTo", "column name")} IS NULL
@@ -3409,39 +3409,49 @@ var DatasetsSpanner = class DatasetsSpanner extends DatasetsStorage {
 	}
 	async _doUpdateItem(args) {
 		try {
-			const existing = await this.loadCurrentItemRow(args.id);
-			if (!existing) throw new MastraError({
-				id: createStorageErrorId("SPANNER", "UPDATE_DATASET_ITEM", "NOT_FOUND"),
-				domain: ErrorDomain.STORAGE,
-				category: ErrorCategory.USER,
-				text: `Dataset item not found: ${args.id}`,
-				details: { id: args.id }
-			});
-			if (existing.datasetId !== args.datasetId) throw new MastraError({
-				id: createStorageErrorId("SPANNER", "UPDATE_DATASET_ITEM", "DATASET_MISMATCH"),
-				domain: ErrorDomain.STORAGE,
-				category: ErrorCategory.USER,
-				text: `Dataset item ${args.id} does not belong to dataset ${args.datasetId}`,
-				details: {
-					id: args.id,
-					datasetId: args.datasetId
-				}
-			});
-			const merged = {
-				input: args.input !== void 0 ? args.input : existing.input,
-				groundTruth: args.groundTruth !== void 0 ? args.groundTruth : existing.groundTruth,
-				expectedTrajectory: args.expectedTrajectory !== void 0 ? args.expectedTrajectory : existing.expectedTrajectory,
-				toolMocks: args.toolMocks !== void 0 ? args.toolMocks : existing.toolMocks,
-				unmockedToolPolicy: args.unmockedToolPolicy !== void 0 ? args.unmockedToolPolicy : existing.unmockedToolPolicy,
-				scorerIds: args.scorerIds !== void 0 ? args.scorerIds ?? void 0 : existing.scorerIds,
-				requestContext: args.requestContext !== void 0 ? args.requestContext : existing.requestContext,
-				metadata: args.metadata !== void 0 ? args.metadata : existing.metadata,
-				source: args.source !== void 0 ? args.source : existing.source
-			};
 			const now = /* @__PURE__ */ new Date();
 			let updated = null;
 			await this.db.runWithAbortRetry(() => this.database.runTransactionAsync(async (tx) => {
 				try {
+					const existing = await this.loadCurrentItemRow(tx, args.id);
+					if (!existing) throw new MastraError({
+						id: createStorageErrorId("SPANNER", "UPDATE_DATASET_ITEM", "NOT_FOUND"),
+						domain: ErrorDomain.STORAGE,
+						category: ErrorCategory.USER,
+						text: `Dataset item not found: ${args.id}`,
+						details: { id: args.id }
+					});
+					if (existing.datasetId !== args.datasetId) throw new MastraError({
+						id: createStorageErrorId("SPANNER", "UPDATE_DATASET_ITEM", "DATASET_MISMATCH"),
+						domain: ErrorDomain.STORAGE,
+						category: ErrorCategory.USER,
+						text: `Dataset item ${args.id} does not belong to dataset ${args.datasetId}`,
+						details: {
+							id: args.id,
+							datasetId: args.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 merged = {
+						input: args.input !== void 0 ? args.input : existing.input,
+						groundTruth: args.groundTruth !== void 0 ? args.groundTruth : existing.groundTruth,
+						expectedTrajectory: args.expectedTrajectory !== void 0 ? args.expectedTrajectory : existing.expectedTrajectory,
+						toolMocks: args.toolMocks !== void 0 ? args.toolMocks : existing.toolMocks,
+						unmockedToolPolicy: args.unmockedToolPolicy !== void 0 ? args.unmockedToolPolicy : existing.unmockedToolPolicy,
+						scorerIds: args.scorerIds !== void 0 ? args.scorerIds ?? void 0 : existing.scorerIds,
+						requestContext: args.requestContext !== void 0 ? args.requestContext : existing.requestContext,
+						metadata: args.metadata !== void 0 ? args.metadata : existing.metadata,
+						source: args.source !== void 0 ? args.source : existing.source
+					};
 					const { version: newVersion, organizationId, projectId } = await this.bumpVersion(tx, args.datasetId, now);
 					await this.closeCurrentRow(tx, args.id, newVersion);
 					await this.db.insert({
@@ -3510,21 +3520,24 @@ var DatasetsSpanner = class DatasetsSpanner extends DatasetsStorage {
 	}
 	async _doDeleteItem(args) {
 		try {
-			const existing = await this.loadCurrentItemRow(args.id);
-			if (!existing) return;
-			if (existing.datasetId !== args.datasetId) throw new MastraError({
-				id: createStorageErrorId("SPANNER", "DELETE_DATASET_ITEM", "DATASET_MISMATCH"),
-				domain: ErrorDomain.STORAGE,
-				category: ErrorCategory.USER,
-				text: `Dataset item ${args.id} does not belong to dataset ${args.datasetId}`,
-				details: {
-					id: args.id,
-					datasetId: args.datasetId
-				}
-			});
 			const now = /* @__PURE__ */ new Date();
 			await this.db.runWithAbortRetry(() => this.database.runTransactionAsync(async (tx) => {
 				try {
+					const existing = await this.loadCurrentItemRow(tx, args.id);
+					if (!existing) {
+						await tx.commit();
+						return;
+					}
+					if (existing.datasetId !== args.datasetId) throw new MastraError({
+						id: createStorageErrorId("SPANNER", "DELETE_DATASET_ITEM", "DATASET_MISMATCH"),
+						domain: ErrorDomain.STORAGE,
+						category: ErrorCategory.USER,
+						text: `Dataset item ${args.id} does not belong to dataset ${args.datasetId}`,
+						details: {
+							id: args.id,
+							datasetId: args.datasetId
+						}
+					});
 					const { version: newVersion, organizationId, projectId } = await this.bumpVersion(tx, args.datasetId, now);
 					await this.closeCurrentRow(tx, args.id, newVersion);
 					await this.db.insert({
@@ -3991,15 +4004,18 @@ var DatasetsSpanner = class DatasetsSpanner extends DatasetsStorage {
 	}
 	async _doBatchDeleteItems(input) {
 		try {
-			const current = [];
-			for (const itemId of input.itemIds) {
-				const row = await this.loadCurrentItemRow(itemId);
-				if (row && row.datasetId === input.datasetId) current.push(row);
-			}
-			if (current.length === 0) return;
 			const now = /* @__PURE__ */ new Date();
 			await this.db.runWithAbortRetry(() => this.database.runTransactionAsync(async (tx) => {
 				try {
+					const current = [];
+					for (const itemId of input.itemIds) {
+						const row = await this.loadCurrentItemRow(tx, itemId);
+						if (row && row.datasetId === input.datasetId) current.push(row);
+					}
+					if (current.length === 0) {
+						await tx.commit();
+						return;
+					}
 					const { version: newVersion, organizationId, projectId } = await this.bumpVersion(tx, input.datasetId, now);
 					for (const existing of current) {
 						await this.closeCurrentRow(tx, existing.id, newVersion);