@mastra/mysql 0.13.3-alpha.0 → 0.14.0

package/dist/docs/SKILL.md

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

package/dist/docs/assets/SOURCE_MAP.json

@@ -1,5 +1,5 @@
 {
-  "version": "0.13.3-alpha.0",
+  "version": "0.14.0",
   "package": "@mastra/mysql",
   "exports": {},
   "modules": {}

package/dist/index.cjs

@@ -86,13 +86,14 @@ function prepareWhereClause(filters) {
 			if (value === true) conditions.push(`${quoteIdentifier(fieldName, "column name")} IS NULL`);
 			else if (value === false) conditions.push(`${quoteIdentifier(fieldName, "column name")} IS NOT NULL`);
 		} else if (value === null) conditions.push(`${quoteIdentifier(key, "column name")} IS NULL`);
-		else if (Array.isArray(value)) if (value.length === 0) conditions.push("FALSE");
-		else {
-			const placeholders = value.map(() => "?").join(", ");
-			conditions.push(`${quoteIdentifier(key, "column name")} IN (${placeholders})`);
-			args.push(...value.map(transformToSqlValue));
-		}
-		else {
+		else if (Array.isArray(value)) {
+			if (value.length === 0) conditions.push("FALSE");
+			else {
+				const placeholders = value.map(() => "?").join(", ");
+				conditions.push(`${quoteIdentifier(key, "column name")} IN (${placeholders})`);
+				args.push(...value.map(transformToSqlValue));
+			}
+		} else {
 			conditions.push(`${quoteIdentifier(key, "column name")} = ?`);
 			args.push(transformToSqlValue(value));
 		}
@@ -157,13 +158,14 @@ function transformFromSqlRow({ tableName, sqlRow }) {
 	const result = {};
 	Object.entries(sqlRow).forEach(([key, value]) => {
 		const columnSchema = schema?.[key];
-		if (columnSchema?.type === "jsonb") if (typeof value === "string") try {
-			result[key] = JSON.parse(value);
-		} catch {
-			result[key] = value;
-		}
-		else result[key] = value;
-		else if (columnSchema?.type === "timestamp") result[key] = value === null ? null : parseDateTime(value);
+		if (columnSchema?.type === "jsonb") {
+			if (typeof value === "string") try {
+				result[key] = JSON.parse(value);
+			} catch {
+				result[key] = value;
+			}
+			else result[key] = value;
+		} else if (columnSchema?.type === "timestamp") result[key] = value === null ? null : parseDateTime(value);
 		else if (columnSchema?.type === "boolean") result[key] = value === true || value === 1 || value === "1";
 		else result[key] = value;
 	});
@@ -1565,15 +1567,19 @@ var BackgroundTasksMySQL = class BackgroundTasksMySQL extends _mastra_core_stora
 			predicates.push(`${quoteIdentifier("status", "column name")} = ?`);
 			params.push(options.expectedStatus);
 		}
-		if (options?.expectedOwnerId !== void 0) if (options.expectedOwnerId === null) predicates.push(`${quoteIdentifier("ownerId", "column name")} IS NULL`);
-		else {
-			predicates.push(`${quoteIdentifier("ownerId", "column name")} = ?`);
-			params.push(options.expectedOwnerId);
+		if (options?.expectedOwnerId !== void 0) {
+			if (options.expectedOwnerId === null) predicates.push(`${quoteIdentifier("ownerId", "column name")} IS NULL`);
+			else {
+				predicates.push(`${quoteIdentifier("ownerId", "column name")} = ?`);
+				params.push(options.expectedOwnerId);
+			}
 		}
-		if (options?.expectedLeaseExpiresAt !== void 0) if (options.expectedLeaseExpiresAt === null) predicates.push(`${quoteIdentifier("leaseExpiresAt", "column name")} IS NULL`);
-		else {
-			predicates.push(`${quoteIdentifier("leaseExpiresAt", "column name")} = ?`);
-			params.push(transformToSqlValue(options.expectedLeaseExpiresAt));
+		if (options?.expectedLeaseExpiresAt !== void 0) {
+			if (options.expectedLeaseExpiresAt === null) predicates.push(`${quoteIdentifier("leaseExpiresAt", "column name")} IS NULL`);
+			else {
+				predicates.push(`${quoteIdentifier("leaseExpiresAt", "column name")} = ?`);
+				params.push(transformToSqlValue(options.expectedLeaseExpiresAt));
+			}
 		}
 		const predicateSql = predicates.length > 0 ? ` AND ${predicates.join(" AND ")}` : "";
 		const [result] = await this.pool.execute(`UPDATE ${formatTableName(_mastra_core_storage.TABLE_BACKGROUND_TASKS)} SET ${setClauses.join(", ")} WHERE ${quoteIdentifier("id", "column name")} = ?${predicateSql}`, params);
@@ -2340,6 +2346,10 @@ var DatasetsMySQL = class DatasetsMySQL extends _mastra_core_storage.DatasetsSto
 			const { page, perPage: perPageInput } = args.pagination;
 			const filterParts = [];
 			const filterArgs = [];
+			if (args.filters?.datasetIds !== void 0) {
+				filterParts.push(args.filters.datasetIds.length > 0 ? `\`id\` IN (${args.filters.datasetIds.map(() => "?").join(",")})` : "1 = 0");
+				filterArgs.push(...args.filters.datasetIds);
+			}
 			if (args.filters?.organizationId !== void 0) {
 				filterParts.push(`${quoteIdentifier("organizationId", "column name")} = ?`);
 				filterArgs.push(args.filters.organizationId);
@@ -6130,10 +6140,11 @@ async function assertRunFence(q, claimsTable, fence, operation) {
 	const [rows] = await q.execute(`SELECT generation, ${quoteIdentifier("ownerId", "column name")} AS ownerId FROM ${claimsTable}
      WHERE ${quoteIdentifier("runId", "column name")} = ? LOCK IN SHARE MODE`, [fence.runId]);
 	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 transaction on its own pooled connection. */
 async function inTransaction(pool, fn) {
@@ -6467,13 +6478,14 @@ var MemoryMySQL = class MemoryMySQL extends _mastra_core_storage.MemoryStorage {
 		for (const indexDef of this.#indexes) await this.operations.createIndex(indexDef);
 	}
 	async prune(policies, options) {
+		const targets = (0, _mastra_core_storage.resolveRetentionTargets)({
+			policies,
+			descriptor: MemoryMySQL.retentionTables,
+			order: ["runFences"]
+		});
 		return (0, _mastra_core_storage.executeRetentionPrune)({
 			domain: "memory",
-			targets: (0, _mastra_core_storage.resolveRetentionTargets)({
-				policies,
-				descriptor: MemoryMySQL.retentionTables,
-				order: ["runFences"]
-			}),
+			targets,
 			options,
 			cutoffFor: (target, now) => (0, _mastra_core_storage.retentionCutoffMs)(target.policy, now),
 			deleteBatch: (target, cutoff, limit) => this.operations.pruneBatch({
@@ -8401,13 +8413,14 @@ var ObservabilityMySQL = class ObservabilityMySQL extends _mastra_core_storage.O
 		this.#indexes = indexes?.filter((idx) => ObservabilityMySQL.MANAGED_TABLES.includes(idx.table));
 	}
 	async prune(policies, options) {
+		const targets = (0, _mastra_core_storage.resolveRetentionTargets)({
+			policies,
+			descriptor: ObservabilityMySQL.retentionTables,
+			order: ["spans"]
+		});
 		return (0, _mastra_core_storage.executeRetentionPrune)({
 			domain: "observability",
-			targets: (0, _mastra_core_storage.resolveRetentionTargets)({
-				policies,
-				descriptor: ObservabilityMySQL.retentionTables,
-				order: ["spans"]
-			}),
+			targets,
 			options,
 			cutoffFor: (target, now) => new Date((0, _mastra_core_storage.retentionCutoffMs)(target.policy, now)),
 			deleteBatch: (target, cutoff, limit) => this.operations.pruneBatch({
@@ -8774,22 +8787,22 @@ var ObservabilityMySQL = class ObservabilityMySQL extends _mastra_core_storage.O
 					case _mastra_core_storage.TraceStatus.RUNNING:
 						conditions.push(`${quoteIdentifier("endedAt", "column name")} IS NULL AND ${quoteIdentifier("error", "column name")} IS NULL`);
 						break;
-					case _mastra_core_storage.TraceStatus.SUCCESS:
-						conditions.push(`${quoteIdentifier("endedAt", "column name")} IS NOT NULL AND ${quoteIdentifier("error", "column name")} IS NULL`);
-						break;
+					case _mastra_core_storage.TraceStatus.SUCCESS: conditions.push(`${quoteIdentifier("endedAt", "column name")} IS NOT NULL AND ${quoteIdentifier("error", "column name")} IS NULL`);
 				}
-				if (filters.hasChildError !== void 0) if (filters.hasChildError) conditions.push(`EXISTS (
+				if (filters.hasChildError !== void 0) {
+					if (filters.hasChildError) conditions.push(`EXISTS (
               SELECT 1 FROM ${tbl} c
               WHERE c.${quoteIdentifier("traceId", "column name")} = ${tbl}.${quoteIdentifier("traceId", "column name")}
                 AND c.${quoteIdentifier("parentSpanId", "column name")} IS NOT NULL
                 AND c.${quoteIdentifier("error", "column name")} IS NOT NULL
             )`);
-				else conditions.push(`NOT EXISTS (
+					else conditions.push(`NOT EXISTS (
               SELECT 1 FROM ${tbl} c
               WHERE c.${quoteIdentifier("traceId", "column name")} = ${tbl}.${quoteIdentifier("traceId", "column name")}
                 AND c.${quoteIdentifier("parentSpanId", "column name")} IS NOT NULL
                 AND c.${quoteIdentifier("error", "column name")} IS NOT NULL
             )`);
+				}
 			}
 			const whereClause = conditions.length > 0 ? ` WHERE ${conditions.join(" AND ")}` : "";
 			const sortField = quoteIdentifier(orderBy.field, "column name");
@@ -9525,15 +9538,19 @@ var SchedulesMySQL = class SchedulesMySQL extends _mastra_core_storage.Schedules
 			conditions.push(`JSON_EXTRACT(${quoteIdentifier("target", "column name")}, '$.workflowId') = ?`);
 			params.push(filter.workflowId);
 		}
-		if (filter?.ownerType !== void 0) if (filter.ownerType === null) conditions.push(`${quoteIdentifier("owner_type", "column name")} IS NULL`);
-		else {
-			conditions.push(`${quoteIdentifier("owner_type", "column name")} = ?`);
-			params.push(filter.ownerType);
+		if (filter?.ownerType !== void 0) {
+			if (filter.ownerType === null) conditions.push(`${quoteIdentifier("owner_type", "column name")} IS NULL`);
+			else {
+				conditions.push(`${quoteIdentifier("owner_type", "column name")} = ?`);
+				params.push(filter.ownerType);
+			}
 		}
-		if (filter?.ownerId !== void 0) if (filter.ownerId === null) conditions.push(`${quoteIdentifier("owner_id", "column name")} IS NULL`);
-		else {
-			conditions.push(`${quoteIdentifier("owner_id", "column name")} = ?`);
-			params.push(filter.ownerId);
+		if (filter?.ownerId !== void 0) {
+			if (filter.ownerId === null) conditions.push(`${quoteIdentifier("owner_id", "column name")} IS NULL`);
+			else {
+				conditions.push(`${quoteIdentifier("owner_id", "column name")} = ?`);
+				params.push(filter.ownerId);
+			}
 		}
 		const where = conditions.length > 0 ? `WHERE ${conditions.join(" AND ")}` : "";
 		const [rows] = await this.pool.execute(`SELECT * FROM ${formatTableName(_mastra_core_storage.TABLE_SCHEDULES)} ${where} ORDER BY ${quoteIdentifier("created_at", "column name")} ASC`, params);
@@ -11468,13 +11485,14 @@ var WorkflowsMySQL = class WorkflowsMySQL extends _mastra_core_storage.Workflows
 		await this.createCustomIndexes();
 	}
 	async prune(policies, options) {
+		const targets = (0, _mastra_core_storage.resolveRetentionTargets)({
+			policies,
+			descriptor: WorkflowsMySQL.retentionTables,
+			order: ["runOwnership"]
+		});
 		return (0, _mastra_core_storage.executeRetentionPrune)({
 			domain: "workflows",
-			targets: (0, _mastra_core_storage.resolveRetentionTargets)({
-				policies,
-				descriptor: WorkflowsMySQL.retentionTables,
-				order: ["runOwnership"]
-			}),
+			targets,
 			options,
 			cutoffFor: (target, now) => (0, _mastra_core_storage.retentionCutoffMs)(target.policy, now),
 			deleteBatch: (target, cutoff, limit) => this.operations.pruneBatch({