@mastra/spanner 1.9.6-alpha.0 → 1.10.0

package/dist/index.js

@@ -838,24 +838,27 @@ var SpannerDB = class extends 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;
@@ -904,16 +907,17 @@ var SpannerDB = class extends 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}`);
 			}
@@ -1173,18 +1177,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;
@@ -2048,15 +2054,19 @@ var BackgroundTasksSpanner = class BackgroundTasksSpanner extends BackgroundTask
 			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({
@@ -3273,6 +3283,10 @@ var DatasetsSpanner = class DatasetsSpanner extends DatasetsStorage {
 			const tableName = quoteIdent(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;
@@ -3302,10 +3316,15 @@ var DatasetsSpanner = class DatasetsSpanner extends DatasetsStorage {
 				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,
@@ -6719,10 +6738,11 @@ async function assertRunFence(tx, claimsTable, fence, operation) {
 		json: true
 	});
 	const row = rows[0];
-	if (!matchesRunFence(row ? {
+	const current = row ? {
 		generation: Number(row.generation),
 		ownerId: String(row.ownerId)
-	} : null, fence)) throw new RunFenceConflictError(fence, operation);
+	} : null;
+	if (!matchesRunFence(current, fence)) throw new RunFenceConflictError(fence, operation);
 }
 /**
 * Runs `fn` in a read-write transaction and commits it, retrying the whole
@@ -6873,13 +6893,14 @@ var MemorySpanner = class MemorySpanner extends MemoryStorage {
 	}
 	/** Removes every row from threads/messages/resources and run fence tables. Intended for tests. */
 	async prune(policies, options) {
+		const targets = resolveRetentionTargets({
+			policies,
+			descriptor: MemorySpanner.retentionTables,
+			order: ["runFences"]
+		});
 		return executeRetentionPrune({
 			domain: "memory",
-			targets: resolveRetentionTargets({
-				policies,
-				descriptor: MemorySpanner.retentionTables,
-				order: ["runFences"]
-			}),
+			targets,
 			options,
 			cutoffFor: (target, now) => retentionCutoffMs(target.policy, now),
 			deleteBatch: (target, cutoff, limit) => this.db.pruneRunRows({
@@ -8482,18 +8503,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",
@@ -8554,18 +8572,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}`);
@@ -8784,7 +8803,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,
@@ -9202,17 +9220,19 @@ var ObservabilitySpanner = class ObservabilitySpanner extends ObservabilityStora
 		return this.disableMetrics ? [] : ["metrics", "metric-discovery"];
 	}
 	async prune(policies, options) {
+		const descriptor = this.disableMetrics ? { spans: {
+			table: TABLE_SPANS,
+			column: "startedAt",
+			indexed: false
+		} } : ObservabilitySpanner.retentionTables;
+		const targets = resolveRetentionTargets({
+			policies,
+			descriptor,
+			order: ["spans", "metrics"]
+		});
 		return executeRetentionPrune({
 			domain: "observability",
-			targets: resolveRetentionTargets({
-				policies,
-				descriptor: this.disableMetrics ? { spans: {
-					table: TABLE_SPANS,
-					column: "startedAt",
-					indexed: false
-				} } : ObservabilitySpanner.retentionTables,
-				order: ["spans", "metrics"]
-			}),
+			targets,
 			options,
 			cutoffFor: (target, now) => new Date(retentionCutoffMs(target.policy, now)).toISOString(),
 			deleteBatch: async (target, cutoff, limit) => {
@@ -9612,9 +9632,7 @@ var ObservabilitySpanner = class ObservabilitySpanner extends ObservabilityStora
 					case TraceStatus.RUNNING:
 						conditions.push(`${rAlias}.${quoteIdent("endedAt", "column name")} IS NULL AND ${rAlias}.${quoteIdent("error", "column name")} IS NULL`);
 						break;
-					case TraceStatus.SUCCESS:
-						conditions.push(`${rAlias}.${quoteIdent("endedAt", "column name")} IS NOT NULL AND ${rAlias}.${quoteIdent("error", "column name")} IS NULL`);
-						break;
+					case 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)`;
@@ -10585,15 +10603,19 @@ var SchedulesSpanner = class SchedulesSpanner extends SchedulesStorage {
 				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(TABLE_SCHEDULES, "table name")}
@@ -11441,10 +11463,11 @@ var ScorerDefinitionsSpanner = class ScorerDefinitionsSpanner extends ScorerDefi
 //#endregion
 //#region src/storage/domains/scores/index.ts
 function transformScoreRow$1(row) {
-	return transformScoreRow(transformFromSpannerRow({
+	const normalized = transformFromSpannerRow({
 		tableName: TABLE_SCORERS,
 		row
-	}), { convertTimestamps: true });
+	});
+	return transformScoreRow(normalized, { convertTimestamps: true });
 }
 /** Appends multi-tenant scope conditions and their named params. */
 function applyTenancyFilters(conditions, params, filters) {
@@ -12696,13 +12719,14 @@ var WorkflowsSpanner = class WorkflowsSpanner extends WorkflowsStorage {
 	}
 	/** Removes every workflow snapshot and run ownership row. Intended for tests. */
 	async prune(policies, options) {
+		const targets = resolveRetentionTargets({
+			policies,
+			descriptor: WorkflowsSpanner.retentionTables,
+			order: ["runOwnership"]
+		});
 		return executeRetentionPrune({
 			domain: "workflows",
-			targets: resolveRetentionTargets({
-				policies,
-				descriptor: WorkflowsSpanner.retentionTables,
-				order: ["runOwnership"]
-			}),
+			targets,
 			options,
 			cutoffFor: (target, now) => retentionCutoffMs(target.policy, now),
 			deleteBatch: (target, cutoff, limit) => this.db.pruneRunRows({