@mastra/spanner 1.9.6-alpha.0 → 1.10.0

package/dist/docs/SKILL.md

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

package/dist/docs/assets/SOURCE_MAP.json

@@ -1,5 +1,5 @@
 {
-  "version": "1.9.6-alpha.0",
+  "version": "1.10.0",
   "package": "@mastra/spanner",
   "exports": {},
   "modules": {}

package/dist/index.cjs

@@ -839,24 +839,27 @@ var SpannerDB = class extends _mastra_core_base.MastraBase {
 				result[key] = value;
 				continue;
 			}
-			if (colDef?.type === "jsonb") if (typeof value === "string") try {
-				result[key] = JSON.parse(value);
-			} catch {
-				result[key] = value;
-			}
-			else result[key] = value;
-			else if (colDef?.type === "timestamp") if (value instanceof Date) result[key] = new Date(value.getTime());
-			else if (typeof value === "string") result[key] = new Date(value);
-			else if (typeof value === "object" && typeof value.value === "string") result[key] = new Date(value.value);
-			else result[key] = value;
-			else if (colDef?.type === "integer" || colDef?.type === "bigint") if (typeof value === "string") {
-				const n = Number(value);
-				result[key] = Number.isSafeInteger(n) ? n : value;
-			} else if (typeof value === "bigint") {
-				const n = Number(value);
-				result[key] = Number.isSafeInteger(n) ? n : value.toString();
-			} else result[key] = value;
-			else if (colDef?.type === "boolean") result[key] = Boolean(value);
+			if (colDef?.type === "jsonb") {
+				if (typeof value === "string") try {
+					result[key] = JSON.parse(value);
+				} catch {
+					result[key] = value;
+				}
+				else result[key] = value;
+			} else if (colDef?.type === "timestamp") {
+				if (value instanceof Date) result[key] = new Date(value.getTime());
+				else if (typeof value === "string") result[key] = new Date(value);
+				else if (typeof value === "object" && typeof value.value === "string") result[key] = new Date(value.value);
+				else result[key] = value;
+			} else if (colDef?.type === "integer" || colDef?.type === "bigint") {
+				if (typeof value === "string") {
+					const n = Number(value);
+					result[key] = Number.isSafeInteger(n) ? n : value;
+				} else if (typeof value === "bigint") {
+					const n = Number(value);
+					result[key] = Number.isSafeInteger(n) ? n : value.toString();
+				} else result[key] = value;
+			} else if (colDef?.type === "boolean") result[key] = Boolean(value);
 			else result[key] = value;
 		}
 		return result;
@@ -905,16 +908,17 @@ var SpannerDB = class extends _mastra_core_base.MastraBase {
 					for (const item of inValues) paramNames.push(`@${bind(key, item)}`);
 					conditions.push(`${quoteIdent(key, "field name")} IN (${paramNames.join(", ")})`);
 				}
-			} else if (Array.isArray(value)) if (value.length === 0) conditions.push("1 = 0");
-			else if (value.length === 1) {
-				const param = bind(key, value[0]);
-				conditions.push(`${quoteIdent(key, "field name")} = @${param}`);
+			} else if (Array.isArray(value)) {
+				if (value.length === 0) conditions.push("1 = 0");
+				else if (value.length === 1) {
+					const param = bind(key, value[0]);
+					conditions.push(`${quoteIdent(key, "field name")} = @${param}`);
+				} else {
+					const paramNames = [];
+					for (const item of value) paramNames.push(`@${bind(key, item)}`);
+					conditions.push(`${quoteIdent(key, "field name")} IN (${paramNames.join(", ")})`);
+				}
 			} else {
-				const paramNames = [];
-				for (const item of value) paramNames.push(`@${bind(key, item)}`);
-				conditions.push(`${quoteIdent(key, "field name")} IN (${paramNames.join(", ")})`);
-			}
-			else {
 				const param = bind(key, value);
 				conditions.push(`${quoteIdent(key, "field name")} = @${param}`);
 			}
@@ -1174,18 +1178,20 @@ function transformFromSpannerRow({ tableName, row }) {
 		} catch {
 			result[key] = value;
 		}
-		else if (colDef?.type === "timestamp") if (value instanceof Date) result[key] = new Date(value.getTime());
-		else if (typeof value === "string") result[key] = new Date(value);
-		else if (typeof value === "object" && typeof value.value === "string") result[key] = new Date(value.value);
-		else result[key] = value;
-		else if (colDef?.type === "integer" || colDef?.type === "bigint") if (typeof value === "string") {
-			const n = Number(value);
-			result[key] = Number.isSafeInteger(n) ? n : value;
-		} else if (typeof value === "bigint") {
-			const n = Number(value);
-			result[key] = Number.isSafeInteger(n) ? n : value.toString();
-		} else result[key] = value;
-		else if (colDef?.type === "boolean") result[key] = Boolean(value);
+		else if (colDef?.type === "timestamp") {
+			if (value instanceof Date) result[key] = new Date(value.getTime());
+			else if (typeof value === "string") result[key] = new Date(value);
+			else if (typeof value === "object" && typeof value.value === "string") result[key] = new Date(value.value);
+			else result[key] = value;
+		} else if (colDef?.type === "integer" || colDef?.type === "bigint") {
+			if (typeof value === "string") {
+				const n = Number(value);
+				result[key] = Number.isSafeInteger(n) ? n : value;
+			} else if (typeof value === "bigint") {
+				const n = Number(value);
+				result[key] = Number.isSafeInteger(n) ? n : value.toString();
+			} else result[key] = value;
+		} else if (colDef?.type === "boolean") result[key] = Boolean(value);
 		else result[key] = value;
 	}
 	return result;
@@ -2049,15 +2055,19 @@ var BackgroundTasksSpanner = class BackgroundTasksSpanner extends _mastra_core_s
 			predicates.push("status = @expectedStatus");
 			params.expectedStatus = options.expectedStatus;
 		}
-		if (options?.expectedOwnerId !== void 0) if (options.expectedOwnerId === null) predicates.push("ownerId IS NULL");
-		else {
-			predicates.push("ownerId = @expectedOwnerId");
-			params.expectedOwnerId = options.expectedOwnerId;
+		if (options?.expectedOwnerId !== void 0) {
+			if (options.expectedOwnerId === null) predicates.push("ownerId IS NULL");
+			else {
+				predicates.push("ownerId = @expectedOwnerId");
+				params.expectedOwnerId = options.expectedOwnerId;
+			}
 		}
-		if (options?.expectedLeaseExpiresAt !== void 0) if (options.expectedLeaseExpiresAt === null) predicates.push("leaseExpiresAt IS NULL");
-		else {
-			predicates.push("leaseExpiresAt = @expectedLeaseExpiresAt");
-			params.expectedLeaseExpiresAt = options.expectedLeaseExpiresAt;
+		if (options?.expectedLeaseExpiresAt !== void 0) {
+			if (options.expectedLeaseExpiresAt === null) predicates.push("leaseExpiresAt IS NULL");
+			else {
+				predicates.push("leaseExpiresAt = @expectedLeaseExpiresAt");
+				params.expectedLeaseExpiresAt = options.expectedLeaseExpiresAt;
+			}
 		}
 		const predicateSql = predicates.length > 0 ? ` AND ${predicates.join(" AND ")}` : "";
 		const [rowCount] = await this.database.runUpdate({
@@ -3274,6 +3284,10 @@ var DatasetsSpanner = class DatasetsSpanner extends _mastra_core_storage.Dataset
 			const tableName = quoteIdent(_mastra_core_storage.TABLE_DATASETS, "table name");
 			const filterConditions = [];
 			const filterParams = {};
+			if (args.filters?.datasetIds !== void 0) {
+				filterConditions.push(`${quoteIdent("id", "column name")} IN UNNEST(@datasetIds)`);
+				filterParams.datasetIds = args.filters.datasetIds;
+			}
 			if (args.filters?.organizationId !== void 0) {
 				filterConditions.push(`${quoteIdent("organizationId", "column name")} = @organizationId`);
 				filterParams.organizationId = args.filters.organizationId;
@@ -3303,10 +3317,15 @@ var DatasetsSpanner = class DatasetsSpanner extends _mastra_core_storage.Dataset
 				filterParams.nameSubstring = `%${args.filters.name}%`;
 			}
 			const whereClause = filterConditions.length > 0 ? `WHERE ${filterConditions.join(" AND ")}` : "";
-			const filterTypes = args.filters?.targetIds !== void 0 && args.filters.targetIds.length > 0 ? { targetIds: {
+			const filterTypes = {};
+			if (args.filters?.targetIds !== void 0 && args.filters.targetIds.length > 0) filterTypes.targetIds = {
 				type: "array",
 				child: { type: "string" }
-			} } : {};
+			};
+			if (args.filters?.datasetIds !== void 0) filterTypes.datasetIds = {
+				type: "array",
+				child: { type: "string" }
+			};
 			const [countRows] = await this.database.run({
 				sql: `SELECT COUNT(*) AS count FROM ${tableName} ${whereClause}`,
 				params: filterParams,
@@ -6720,10 +6739,11 @@ async function assertRunFence(tx, claimsTable, fence, operation) {
 		json: true
 	});
 	const row = rows[0];
-	if (!(0, _mastra_core_storage.matchesRunFence)(row ? {
+	const current = row ? {
 		generation: Number(row.generation),
 		ownerId: String(row.ownerId)
-	} : null, fence)) throw new _mastra_core_storage.RunFenceConflictError(fence, operation);
+	} : null;
+	if (!(0, _mastra_core_storage.matchesRunFence)(current, fence)) throw new _mastra_core_storage.RunFenceConflictError(fence, operation);
 }
 /**
 * Runs `fn` in a read-write transaction and commits it, retrying the whole
@@ -6874,13 +6894,14 @@ var MemorySpanner = class MemorySpanner extends _mastra_core_storage.MemoryStora
 	}
 	/** Removes every row from threads/messages/resources and run fence tables. Intended for tests. */
 	async prune(policies, options) {
+		const targets = (0, _mastra_core_storage.resolveRetentionTargets)({
+			policies,
+			descriptor: MemorySpanner.retentionTables,
+			order: ["runFences"]
+		});
 		return (0, _mastra_core_storage.executeRetentionPrune)({
 			domain: "memory",
-			targets: (0, _mastra_core_storage.resolveRetentionTargets)({
-				policies,
-				descriptor: MemorySpanner.retentionTables,
-				order: ["runFences"]
-			}),
+			targets,
 			options,
 			cutoffFor: (target, now) => (0, _mastra_core_storage.retentionCutoffMs)(target.policy, now),
 			deleteBatch: (target, cutoff, limit) => this.db.pruneRunRows({
@@ -8483,18 +8504,15 @@ function buildWhereClause(filters, startIndex = 0) {
 			conditions.push("1 = 0");
 			return;
 		}
-		const placeholders = [];
-		for (const v of values) {
-			const param = `f${i++}`;
-			placeholders.push(`@${param}`);
-			params[param] = v;
-		}
-		conditions.push(`${quoteIdent(col, "column name")} IN (${placeholders.join(", ")})`);
+		const param = `f${i++}`;
+		params[param] = [...values];
+		conditions.push(`${quoteIdent(col, "column name")} IN UNNEST(@${param})`);
 	};
 	const ts = filters.timestamp;
 	if (ts?.start) bindScalar("timestamp", ts.start, ts.startExclusive ? ">" : ">=");
 	if (ts?.end) bindScalar("timestamp", ts.end, ts.endExclusive ? "<" : "<=");
 	if (Array.isArray(filters.traceIds)) bindIn("traceId", filters.traceIds);
+	if (Array.isArray(filters.experimentIds)) bindIn("experimentId", filters.experimentIds);
 	for (const k of [
 		"traceId",
 		"spanId",
@@ -8555,18 +8573,19 @@ function combineWhere(nameValue, filterClause, filterParams, filterTypes, nextIn
 	const params = { ...filterParams };
 	const types = { ...filterTypes };
 	const conditions = [];
-	if (Array.isArray(nameValue)) if (nameValue.length === 0) conditions.push("1 = 0");
-	else {
-		const placeholders = [];
-		let i = nextIndex;
-		for (const n of nameValue) {
-			const param = `n${i++}`;
-			placeholders.push(`@${param}`);
-			params[param] = n;
+	if (Array.isArray(nameValue)) {
+		if (nameValue.length === 0) conditions.push("1 = 0");
+		else {
+			const placeholders = [];
+			let i = nextIndex;
+			for (const n of nameValue) {
+				const param = `n${i++}`;
+				placeholders.push(`@${param}`);
+				params[param] = n;
+			}
+			conditions.push(`${quoteIdent("name", "column name")} IN (${placeholders.join(", ")})`);
 		}
-		conditions.push(`${quoteIdent("name", "column name")} IN (${placeholders.join(", ")})`);
-	}
-	else {
+	} else {
 		const param = `n${nextIndex}`;
 		params[param] = nameValue;
 		conditions.push(`${quoteIdent("name", "column name")} = @${param}`);
@@ -8785,7 +8804,6 @@ async function getMetricAggregate(database, args, options) {
 				default:
 					prevStart = new Date(ts.start.getTime() - duration);
 					prevEnd = new Date(ts.end.getTime() - duration);
-					break;
 			}
 			const prevFilterResult = buildWhereClause({
 				...args.filters,
@@ -9203,17 +9221,19 @@ var ObservabilitySpanner = class ObservabilitySpanner extends _mastra_core_stora
 		return this.disableMetrics ? [] : ["metrics", "metric-discovery"];
 	}
 	async prune(policies, options) {
+		const descriptor = this.disableMetrics ? { spans: {
+			table: _mastra_core_storage.TABLE_SPANS,
+			column: "startedAt",
+			indexed: false
+		} } : ObservabilitySpanner.retentionTables;
+		const targets = (0, _mastra_core_storage.resolveRetentionTargets)({
+			policies,
+			descriptor,
+			order: ["spans", "metrics"]
+		});
 		return (0, _mastra_core_storage.executeRetentionPrune)({
 			domain: "observability",
-			targets: (0, _mastra_core_storage.resolveRetentionTargets)({
-				policies,
-				descriptor: this.disableMetrics ? { spans: {
-					table: _mastra_core_storage.TABLE_SPANS,
-					column: "startedAt",
-					indexed: false
-				} } : ObservabilitySpanner.retentionTables,
-				order: ["spans", "metrics"]
-			}),
+			targets,
 			options,
 			cutoffFor: (target, now) => new Date((0, _mastra_core_storage.retentionCutoffMs)(target.policy, now)).toISOString(),
 			deleteBatch: async (target, cutoff, limit) => {
@@ -9613,9 +9633,7 @@ var ObservabilitySpanner = class ObservabilitySpanner extends _mastra_core_stora
 					case _mastra_core_storage.TraceStatus.RUNNING:
 						conditions.push(`${rAlias}.${quoteIdent("endedAt", "column name")} IS NULL AND ${rAlias}.${quoteIdent("error", "column name")} IS NULL`);
 						break;
-					case _mastra_core_storage.TraceStatus.SUCCESS:
-						conditions.push(`${rAlias}.${quoteIdent("endedAt", "column name")} IS NOT NULL AND ${rAlias}.${quoteIdent("error", "column name")} IS NULL`);
-						break;
+					case _mastra_core_storage.TraceStatus.SUCCESS: conditions.push(`${rAlias}.${quoteIdent("endedAt", "column name")} IS NOT NULL AND ${rAlias}.${quoteIdent("error", "column name")} IS NULL`);
 				}
 				if (filters.hasChildError !== void 0) {
 					const existsClause = `EXISTS (SELECT 1 FROM ${tableName} c WHERE c.${quoteIdent("traceId", "column name")} = ${rAlias}.${quoteIdent("traceId", "column name")} AND c.${quoteIdent("error", "column name")} IS NOT NULL)`;
@@ -10586,15 +10604,19 @@ var SchedulesSpanner = class SchedulesSpanner extends _mastra_core_storage.Sched
 				else conditions.push(`JSON_VALUE(${quoteIdent("target", "column name")}, '$.workflowId') = @workflowId`);
 				params.workflowId = filter.workflowId;
 			}
-			if (filter?.ownerType !== void 0) if (filter.ownerType === null) conditions.push(`${quoteIdent("owner_type", "column name")} IS NULL`);
-			else {
-				conditions.push(`${quoteIdent("owner_type", "column name")} = @ownerType`);
-				params.ownerType = filter.ownerType;
+			if (filter?.ownerType !== void 0) {
+				if (filter.ownerType === null) conditions.push(`${quoteIdent("owner_type", "column name")} IS NULL`);
+				else {
+					conditions.push(`${quoteIdent("owner_type", "column name")} = @ownerType`);
+					params.ownerType = filter.ownerType;
+				}
 			}
-			if (filter?.ownerId !== void 0) if (filter.ownerId === null) conditions.push(`${quoteIdent("owner_id", "column name")} IS NULL`);
-			else {
-				conditions.push(`${quoteIdent("owner_id", "column name")} = @ownerId`);
-				params.ownerId = filter.ownerId;
+			if (filter?.ownerId !== void 0) {
+				if (filter.ownerId === null) conditions.push(`${quoteIdent("owner_id", "column name")} IS NULL`);
+				else {
+					conditions.push(`${quoteIdent("owner_id", "column name")} = @ownerId`);
+					params.ownerId = filter.ownerId;
+				}
 			}
 			const where = conditions.length > 0 ? `WHERE ${conditions.join(" AND ")}` : "";
 			const sql = `SELECT * FROM ${quoteIdent(_mastra_core_storage.TABLE_SCHEDULES, "table name")}
@@ -11442,10 +11464,11 @@ var ScorerDefinitionsSpanner = class ScorerDefinitionsSpanner extends _mastra_co
 //#endregion
 //#region src/storage/domains/scores/index.ts
 function transformScoreRow(row) {
-	return (0, _mastra_core_storage.transformScoreRow)(transformFromSpannerRow({
+	const normalized = transformFromSpannerRow({
 		tableName: _mastra_core_storage.TABLE_SCORERS,
 		row
-	}), { convertTimestamps: true });
+	});
+	return (0, _mastra_core_storage.transformScoreRow)(normalized, { convertTimestamps: true });
 }
 /** Appends multi-tenant scope conditions and their named params. */
 function applyTenancyFilters(conditions, params, filters) {
@@ -12697,13 +12720,14 @@ var WorkflowsSpanner = class WorkflowsSpanner extends _mastra_core_storage.Workf
 	}
 	/** Removes every workflow snapshot and run ownership row. Intended for tests. */
 	async prune(policies, options) {
+		const targets = (0, _mastra_core_storage.resolveRetentionTargets)({
+			policies,
+			descriptor: WorkflowsSpanner.retentionTables,
+			order: ["runOwnership"]
+		});
 		return (0, _mastra_core_storage.executeRetentionPrune)({
 			domain: "workflows",
-			targets: (0, _mastra_core_storage.resolveRetentionTargets)({
-				policies,
-				descriptor: WorkflowsSpanner.retentionTables,
-				order: ["runOwnership"]
-			}),
+			targets,
 			options,
 			cutoffFor: (target, now) => (0, _mastra_core_storage.retentionCutoffMs)(target.policy, now),
 			deleteBatch: (target, cutoff, limit) => this.db.pruneRunRows({