@mastra/mysql 0.14.0-alpha.1 → 0.14.1-alpha.0

package/dist/index.js

@@ -5,7 +5,7 @@ import { parseSqlIdentifier } from "@mastra/core/utils";
 import { createHash } from "crypto";
 import { MessageList } from "@mastra/core/agent";
 import { skillSnapshotFieldValuesEqual } from "@mastra/core/storage/domains/skills";
-//#region src/storage/db/schema-snapshot.ts
+// src/storage/db/schema-snapshot.ts
 /** Builds the `table.index` presence key the snapshot uses. */
 function indexKey(table, index) {
 	return `${table.toLowerCase()}.${index.toLowerCase()}`;
@@ -44,8 +44,7 @@ async function loadSchemaSnapshot(pool, schemaName) {
 		indexes
 	};
 }
-//#endregion
-//#region src/storage/domains/utils.ts
+// src/storage/domains/utils.ts
 function quoteIdentifier(value, context) {
 	return `\`${parseSqlIdentifier(value, context)}\``;
 }
@@ -85,13 +84,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));
 		}
@@ -156,13 +156,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;
 	});
@@ -178,8 +179,7 @@ function parseDateTime(value) {
 	const withZone = /[zZ]|[+-]\d\d:?\d\d$/.test(normalized) ? normalized : `${normalized}Z`;
 	return new Date(withZone);
 }
-//#endregion
-//#region src/storage/domains/operations/index.ts
+// src/storage/domains/operations/index.ts
 var StoreOperationsMySQL = class extends StoreOperations {
 	pool;
 	database;
@@ -773,8 +773,7 @@ function mapToMySqlType(type, isKey = false) {
 		default: return "TEXT";
 	}
 }
-//#endregion
-//#region src/storage/domains/agents/index.ts
+// src/storage/domains/agents/index.ts
 var AgentsMySQL = class AgentsMySQL extends AgentsStorage {
 	pool;
 	operations;
@@ -1371,8 +1370,7 @@ var AgentsMySQL = class AgentsMySQL extends AgentsStorage {
 		};
 	}
 };
-//#endregion
-//#region src/storage/domains/background-tasks/index.ts
+// src/storage/domains/background-tasks/index.ts
 function serializeJson(v) {
 	if (typeof v === "object" && v != null) return JSON.stringify(v);
 	return v ?? null;
@@ -1564,15 +1562,19 @@ var BackgroundTasksMySQL = class BackgroundTasksMySQL extends BackgroundTasksSto
 			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(TABLE_BACKGROUND_TASKS)} SET ${setClauses.join(", ")} WHERE ${quoteIdentifier("id", "column name")} = ?${predicateSql}`, params);
@@ -1686,8 +1688,7 @@ var BackgroundTasksMySQL = class BackgroundTasksMySQL extends BackgroundTasksSto
 		return Number(rows[0]?.count ?? 0);
 	}
 };
-//#endregion
-//#region src/storage/domains/blobs/index.ts
+// src/storage/domains/blobs/index.ts
 const INSERT_IGNORE_SQL = `INSERT IGNORE INTO ${formatTableName(TABLE_SKILL_BLOBS)} (${[
 	"hash",
 	"content",
@@ -1768,8 +1769,7 @@ var BlobsMySQL = class extends BlobStore {
 		};
 	}
 };
-//#endregion
-//#region src/storage/domains/channels/index.ts
+// src/storage/domains/channels/index.ts
 var ChannelsMySQL = class ChannelsMySQL extends ChannelsStorage {
 	pool;
 	operations;
@@ -1903,8 +1903,7 @@ var ChannelsMySQL = class ChannelsMySQL extends ChannelsStorage {
 		};
 	}
 };
-//#endregion
-//#region src/storage/domains/datasets/index.ts
+// src/storage/domains/datasets/index.ts
 function parseJSON$3(value) {
 	if (value === null || value === void 0) return void 0;
 	if (typeof value === "string") try {
@@ -3055,8 +3054,7 @@ var DatasetsMySQL = class DatasetsMySQL extends DatasetsStorage {
 		}
 	}
 };
-//#endregion
-//#region src/storage/domains/experiments/index.ts
+// src/storage/domains/experiments/index.ts
 function parseJSON$2(value) {
 	if (value === null || value === void 0) return void 0;
 	if (typeof value === "string") {
@@ -3928,8 +3926,7 @@ var ExperimentsMySQL = class ExperimentsMySQL extends ExperimentsStorage {
 		}
 	}
 };
-//#endregion
-//#region src/storage/domains/favorites/index.ts
+// src/storage/domains/favorites/index.ts
 /**
 * Maps a favorite entity type to its parent entity table.
 */
@@ -4165,8 +4162,7 @@ var FavoritesMySQL = class FavoritesMySQL extends FavoritesStorage {
 		}
 	}
 };
-//#endregion
-//#region src/storage/domains/knowledge/index.ts
+// src/storage/domains/knowledge/index.ts
 let assertDescriptionWithinBound;
 async function assertKnowledgeDescriptionWithinBoundCompat(description) {
 	let assertWithinBound = assertDescriptionWithinBound;
@@ -5061,8 +5057,7 @@ var KnowledgeMySQL = class extends KnowledgeStorage {
 		});
 	}
 };
-//#endregion
-//#region src/storage/domains/mcp-clients/index.ts
+// src/storage/domains/mcp-clients/index.ts
 var MCPClientsMySQL = class MCPClientsMySQL extends MCPClientsStorage {
 	pool;
 	operations;
@@ -5535,8 +5530,7 @@ var MCPClientsMySQL = class MCPClientsMySQL extends MCPClientsStorage {
 		}
 	}
 };
-//#endregion
-//#region src/storage/domains/mcp-servers/index.ts
+// src/storage/domains/mcp-servers/index.ts
 var MCPServersMySQL = class MCPServersMySQL extends MCPServersStorage {
 	pool;
 	operations;
@@ -6025,8 +6019,7 @@ var MCPServersMySQL = class MCPServersMySQL extends MCPServersStorage {
 		}
 	}
 };
-//#endregion
-//#region ../../packages/_internal-core/dist/memory/index.js
+// ../../packages/_internal-core/dist/memory/index.js
 /**
 * Shared rules for observational memory lifecycle writes. Every storage adapter applies
 * these inside its own atomic section (lock, transaction, or conditional update), so the
@@ -6111,8 +6104,7 @@ function isAppendOnlySince(storedObservations, snapshotObservations) {
 	const snapshot = snapshotObservations ?? "";
 	return stored === snapshot || stored.startsWith(snapshot);
 }
-//#endregion
-//#region src/storage/db/run-fencing.ts
+// src/storage/db/run-fencing.ts
 /**
 * Epoch milliseconds on the database clock. `UTC_TIMESTAMP` is the time the
 * statement started and doesn't depend on the session time zone. Every lease
@@ -6133,10 +6125,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 (!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 transaction on its own pooled connection. */
 async function inTransaction(pool, fn) {
@@ -6179,8 +6172,7 @@ async function claimTransaction(pool, fn) {
 		if (attempt >= MAX_CLAIM_ATTEMPTS || errno !== ER_DUP_ENTRY && errno !== ER_LOCK_DEADLOCK) throw error;
 	}
 }
-//#endregion
-//#region src/storage/domains/memory/index.ts
+// src/storage/domains/memory/index.ts
 const OM_TABLE = "mastra_observational_memory";
 const OM_TABLE_QUOTED = quoteIdentifier(OM_TABLE, "table name");
 const RUN_FENCES_TABLE = formatTableName(TABLE_MEMORY_RUN_FENCES);
@@ -6470,13 +6462,14 @@ var MemoryMySQL = class MemoryMySQL extends MemoryStorage {
 		for (const indexDef of this.#indexes) await this.operations.createIndex(indexDef);
 	}
 	async prune(policies, options) {
+		const targets = resolveRetentionTargets({
+			policies,
+			descriptor: MemoryMySQL.retentionTables,
+			order: ["runFences"]
+		});
 		return executeRetentionPrune({
 			domain: "memory",
-			targets: resolveRetentionTargets({
-				policies,
-				descriptor: MemoryMySQL.retentionTables,
-				order: ["runFences"]
-			}),
+			targets,
 			options,
 			cutoffFor: (target, now) => retentionCutoffMs(target.policy, now),
 			deleteBatch: (target, cutoff, limit) => this.operations.pruneBatch({
@@ -8370,8 +8363,7 @@ var MemoryMySQL = class MemoryMySQL extends MemoryStorage {
 		};
 	}
 };
-//#endregion
-//#region src/storage/domains/observability/index.ts
+// src/storage/domains/observability/index.ts
 const JSON_SPAN_FIELDS = [
 	"input",
 	"output",
@@ -8404,13 +8396,14 @@ var ObservabilityMySQL = class ObservabilityMySQL extends ObservabilityStorage {
 		this.#indexes = indexes?.filter((idx) => ObservabilityMySQL.MANAGED_TABLES.includes(idx.table));
 	}
 	async prune(policies, options) {
+		const targets = resolveRetentionTargets({
+			policies,
+			descriptor: ObservabilityMySQL.retentionTables,
+			order: ["spans"]
+		});
 		return executeRetentionPrune({
 			domain: "observability",
-			targets: resolveRetentionTargets({
-				policies,
-				descriptor: ObservabilityMySQL.retentionTables,
-				order: ["spans"]
-			}),
+			targets,
 			options,
 			cutoffFor: (target, now) => new Date(retentionCutoffMs(target.policy, now)),
 			deleteBatch: (target, cutoff, limit) => this.operations.pruneBatch({
@@ -8777,22 +8770,22 @@ var ObservabilityMySQL = class ObservabilityMySQL extends ObservabilityStorage {
 					case TraceStatus.RUNNING:
 						conditions.push(`${quoteIdentifier("endedAt", "column name")} IS NULL AND ${quoteIdentifier("error", "column name")} IS NULL`);
 						break;
-					case TraceStatus.SUCCESS:
-						conditions.push(`${quoteIdentifier("endedAt", "column name")} IS NOT NULL AND ${quoteIdentifier("error", "column name")} IS NULL`);
-						break;
+					case 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");
@@ -8904,8 +8897,7 @@ var ObservabilityMySQL = class ObservabilityMySQL extends ObservabilityStorage {
 		}
 	}
 };
-//#endregion
-//#region src/storage/domains/prompt-blocks/index.ts
+// src/storage/domains/prompt-blocks/index.ts
 var PromptBlocksMySQL = class PromptBlocksMySQL extends PromptBlocksStorage {
 	pool;
 	operations;
@@ -9377,8 +9369,7 @@ var PromptBlocksMySQL = class PromptBlocksMySQL extends PromptBlocksStorage {
 		}
 	}
 };
-//#endregion
-//#region src/storage/domains/schedules/index.ts
+// src/storage/domains/schedules/index.ts
 function parseJson(val) {
 	if (val == null) return void 0;
 	if (typeof val === "string") try {
@@ -9528,15 +9519,19 @@ var SchedulesMySQL = class SchedulesMySQL extends SchedulesStorage {
 			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(TABLE_SCHEDULES)} ${where} ORDER BY ${quoteIdentifier("created_at", "column name")} ASC`, params);
@@ -9643,8 +9638,7 @@ var SchedulesMySQL = class SchedulesMySQL extends SchedulesStorage {
 		return rows.map((r) => rowToTrigger(r));
 	}
 };
-//#endregion
-//#region src/storage/domains/scorer-definitions/index.ts
+// src/storage/domains/scorer-definitions/index.ts
 var ScorerDefinitionsMySQL = class ScorerDefinitionsMySQL extends ScorerDefinitionsStorage {
 	pool;
 	operations;
@@ -10119,8 +10113,7 @@ var ScorerDefinitionsMySQL = class ScorerDefinitionsMySQL extends ScorerDefiniti
 		}
 	}
 };
-//#endregion
-//#region src/storage/domains/scores/index.ts
+// src/storage/domains/scores/index.ts
 function parseJSON(value) {
 	if (value === null || value === void 0) return void 0;
 	if (typeof value === "string") {
@@ -10415,8 +10408,7 @@ var ScoresMySQL = class ScoresMySQL extends ScoresStorage {
 		}), pagination);
 	}
 };
-//#endregion
-//#region src/storage/domains/skills/index.ts
+// src/storage/domains/skills/index.ts
 const SNAPSHOT_FIELDS$1 = [
 	"name",
 	"description",
@@ -10976,8 +10968,7 @@ var SkillsMySQL = class SkillsMySQL extends SkillsStorage {
 		}
 	}
 };
-//#endregion
-//#region src/storage/domains/tool-provider-connections/index.ts
+// src/storage/domains/tool-provider-connections/index.ts
 function normaliseScope(raw) {
 	const value = raw == null ? "per-author" : String(raw);
 	if (value === "shared") return "shared";
@@ -11202,8 +11193,7 @@ var ToolProviderConnectionsMySQL = class ToolProviderConnectionsMySQL extends To
 		}
 	}
 };
-//#endregion
-//#region src/storage/domains/workflow-definitions/index.ts
+// src/storage/domains/workflow-definitions/index.ts
 function rowToDefinition(row) {
 	const transformed = transformFromSqlRow({
 		tableName: TABLE_WORKFLOW_DEFINITIONS,
@@ -11344,8 +11334,7 @@ var WorkflowDefinitionsMySQL = class extends WorkflowDefinitionsStorage {
 		await this.pool.execute(`DELETE FROM ${formatTableName(TABLE_WORKFLOW_DEFINITIONS, this.database)} WHERE ${quoteIdentifier("id", "column name")} = ?`, [id]);
 	}
 };
-//#endregion
-//#region src/storage/domains/workflows/index.ts
+// src/storage/domains/workflows/index.ts
 function parseSnapshot(snapshot) {
 	if (typeof snapshot === "string") try {
 		return JSON.parse(snapshot);
@@ -11471,13 +11460,14 @@ var WorkflowsMySQL = class WorkflowsMySQL extends WorkflowsStorage {
 		await this.createCustomIndexes();
 	}
 	async prune(policies, options) {
+		const targets = resolveRetentionTargets({
+			policies,
+			descriptor: WorkflowsMySQL.retentionTables,
+			order: ["runOwnership"]
+		});
 		return executeRetentionPrune({
 			domain: "workflows",
-			targets: resolveRetentionTargets({
-				policies,
-				descriptor: WorkflowsMySQL.retentionTables,
-				order: ["runOwnership"]
-			}),
+			targets,
 			options,
 			cutoffFor: (target, now) => retentionCutoffMs(target.policy, now),
 			deleteBatch: (target, cutoff, limit) => this.operations.pruneBatch({
@@ -11863,8 +11853,7 @@ var WorkflowsMySQL = class WorkflowsMySQL extends WorkflowsStorage {
 		}
 	}
 };
-//#endregion
-//#region src/storage/domains/workspaces/index.ts
+// src/storage/domains/workspaces/index.ts
 const SNAPSHOT_FIELDS = [
 	"name",
 	"description",
@@ -12382,8 +12371,7 @@ var WorkspacesMySQL = class WorkspacesMySQL extends WorkspacesStorage {
 		}
 	}
 };
-//#endregion
-//#region src/storage/index.ts
+// src/storage/index.ts
 function validateConfig(config) {
 	if ("connectionString" in config) {
 		if (!config.connectionString || typeof config.connectionString !== "string") throw new Error("MySQLStore: connectionString must be a non-empty string.");
@@ -12656,7 +12644,6 @@ var MySQLStore = class extends MastraCompositeStore {
 		await this.pool.end();
 	}
 };
-//#endregion
 export { AgentsMySQL, BlobsMySQL, DatasetsMySQL, ExperimentsMySQL, MCPClientsMySQL, MCPServersMySQL, MemoryMySQL, MySQLStore, ObservabilityMySQL, PromptBlocksMySQL, ScorerDefinitionsMySQL, ScoresMySQL, SkillsMySQL, StoreOperationsMySQL, WorkflowsMySQL, WorkspacesMySQL };
 
 //# sourceMappingURL=index.js.map