@mastra/spanner 1.6.6 → 1.7.0-alpha.1

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);
@@ -6942,6 +6958,88 @@ var MemorySpanner = class MemorySpanner extends MemoryStorage {
 			}, error);
 		}
 	}
+	/**
+	* Atomically reassign a thread and all of its messages to a different resource.
+	*
+	* Runs inside a single read-write transaction: the thread row is read inside the
+	* transaction (acquiring a lock), then the thread and every message are moved together.
+	* Overlapping transfers of the same thread serialize on the transaction, so they can never
+	* interleave the thread update with the message update — either both move or neither does,
+	* with no split-ownership window. The thread's `createdAt` is preserved. Callers are
+	* responsible for authorizing the reassignment.
+	*/
+	async updateThreadResourceId({ threadId, resourceId }) {
+		const tableThreads = quoteIdent(TABLE_THREADS, "table name");
+		const messagesTable = quoteIdent(TABLE_MESSAGES, "table name");
+		const now = /* @__PURE__ */ new Date();
+		let existingThread = null;
+		let noop = false;
+		try {
+			await this.db.runWithAbortRetry(() => this.database.runTransactionAsync(async (tx) => {
+				try {
+					const [rows] = await tx.run({
+						sql: `SELECT * FROM ${tableThreads} WHERE id = @threadId LIMIT 1`,
+						params: { threadId },
+						json: true
+					});
+					const row = rows[0];
+					if (!row) throw new MastraError({
+						id: createStorageErrorId("SPANNER", "UPDATE_THREAD_RESOURCE_ID", "NOT_FOUND"),
+						domain: ErrorDomain.STORAGE,
+						category: ErrorCategory.USER,
+						text: `Thread ${threadId} not found`,
+						details: { threadId }
+					});
+					existingThread = this.formatThreadRow(row);
+					if (existingThread.resourceId === resourceId) {
+						noop = true;
+						await tx.commit();
+						return;
+					}
+					await this.db.update({
+						tableName: TABLE_THREADS,
+						keys: { id: threadId },
+						data: {
+							resourceId,
+							updatedAt: now
+						},
+						transaction: tx
+					});
+					await tx.runUpdate({
+						sql: `UPDATE ${messagesTable} SET ${quoteIdent("resourceId", "column name")} = @resourceId WHERE ${quoteIdent("thread_id", "column name")} = @threadId`,
+						params: {
+							resourceId,
+							threadId
+						}
+					});
+					await tx.commit();
+				} catch (err) {
+					await tx.rollback().catch((rollbackErr) => {
+						throw new AggregateError([err, rollbackErr], "Transaction and rollback both failed");
+					});
+					throw err;
+				}
+			}));
+			const thread = existingThread;
+			if (noop) return thread;
+			return {
+				...thread,
+				resourceId,
+				updatedAt: now
+			};
+		} catch (error) {
+			if (error instanceof MastraError) throw error;
+			throw new MastraError({
+				id: createStorageErrorId("SPANNER", "UPDATE_THREAD_RESOURCE_ID", "FAILED"),
+				domain: ErrorDomain.STORAGE,
+				category: ErrorCategory.THIRD_PARTY,
+				details: {
+					threadId,
+					resourceId
+				}
+			}, error);
+		}
+	}
 	/** Deletes a thread and all its messages atomically. */
 	async deleteThread({ threadId }) {
 		const messagesTable = quoteIdent(TABLE_MESSAGES, "table name");