@mastra/mysql 0.11.1 → 0.12.1-alpha.0

package/dist/index.js

@@ -6120,6 +6120,7 @@ function addMySQLMessageMetadataFilter(conditions, params, metadataFilter) {
 var MemoryMySQL = class MemoryMySQL extends MemoryStorage {
 	supportsPartialThreadUpdate = true;
 	supportsObservationalMemory = true;
+	supportsObservationalMemoryHistorySearch = true;
 	pool;
 	operations;
 	#skipDefaultIndexes;
@@ -7322,6 +7323,10 @@ var MemoryMySQL = class MemoryMySQL extends MemoryStorage {
 			const safeLimit = Math.max(1, Math.floor(Number(limit)) || 10);
 			const conditions = [`${omCol("lookupKey")} = ?`];
 			const params = [lookupKey];
+			if (options?.recordId !== void 0) {
+				conditions.push(`${omCol("id")} = ?`);
+				params.push(options.recordId);
+			}
 			if (options?.from) {
 				conditions.push(`${omCol("createdAt")} >= ?`);
 				params.push(transformToSqlValue(options.from));
@@ -7330,8 +7335,26 @@ var MemoryMySQL = class MemoryMySQL extends MemoryStorage {
 				conditions.push(`${omCol("createdAt")} <= ?`);
 				params.push(transformToSqlValue(options.to));
 			}
+			if (options?.groupId !== void 0) {
+				conditions.push(`(LOCATE(CAST(? AS BINARY), CAST(${omCol("activeObservations")} AS BINARY)) > 0 OR EXISTS (
+          SELECT 1 FROM JSON_TABLE(${omCol("bufferedObservationChunks")}, '$[*]'
+            COLUMNS (observations LONGTEXT PATH '$.observations')) AS chunk
+          WHERE LOCATE(CAST(? AS BINARY), CAST(chunk.observations AS BINARY)) > 0
+        ))`);
+				const prefix = `<observation-group id="${options.groupId}"`;
+				params.push(prefix, prefix);
+			}
+			if (options?.beforeGeneration !== void 0) {
+				conditions.push(`${omCol("generationCount")} < ?`);
+				params.push(options.beforeGeneration);
+			}
+			if (options?.afterGeneration !== void 0) {
+				conditions.push(`${omCol("generationCount")} > ?`);
+				params.push(options.afterGeneration);
+			}
+			const direction = options?.sortDirection === "ASC" ? "ASC" : "DESC";
 			const whereClause = conditions.join(" AND ");
-			let sql = `SELECT * FROM ${OM_TABLE_QUOTED} WHERE ${whereClause} ORDER BY ${omCol("generationCount")} DESC LIMIT ${safeLimit}`;
+			let sql = `SELECT * FROM ${OM_TABLE_QUOTED} WHERE ${whereClause} ORDER BY ${omCol("generationCount")} ${direction}, ${omCol("createdAt")} ASC, id ASC LIMIT ${safeLimit}`;
 			if (options?.offset != null && options.offset > 0) sql += ` OFFSET ${options.offset}`;
 			const [rows] = await this.pool.execute(sql, params);
 			if (!rows) return [];
@@ -9103,12 +9126,13 @@ var SchedulesMySQL = class SchedulesMySQL extends SchedulesStorage {
 		if (!updated) throw new Error(`Schedule ${id} not found`);
 		return updated;
 	}
-	async updateScheduleNextFire(id, expectedNextFireAt, newNextFireAt, lastFireAt, lastRunId) {
-		const [result] = await this.pool.execute(`UPDATE ${formatTableName(TABLE_SCHEDULES)} SET ${quoteIdentifier("next_fire_at", "column name")} = ?, ${quoteIdentifier("last_fire_at", "column name")} = ?, ${quoteIdentifier("last_run_id", "column name")} = ?, ${quoteIdentifier("updated_at", "column name")} = ? WHERE ${quoteIdentifier("id", "column name")} = ? AND ${quoteIdentifier("next_fire_at", "column name")} = ? AND ${quoteIdentifier("status", "column name")} = ?`, [
+	async updateScheduleNextFire(id, expectedNextFireAt, newNextFireAt, lastFireAt, lastRunId, newStatus) {
+		const [result] = await this.pool.execute(`UPDATE ${formatTableName(TABLE_SCHEDULES)} SET ${quoteIdentifier("next_fire_at", "column name")} = ?, ${quoteIdentifier("last_fire_at", "column name")} = ?, ${quoteIdentifier("last_run_id", "column name")} = ?, ${quoteIdentifier("updated_at", "column name")} = ?, ${quoteIdentifier("status", "column name")} = COALESCE(?, ${quoteIdentifier("status", "column name")}) WHERE ${quoteIdentifier("id", "column name")} = ? AND ${quoteIdentifier("next_fire_at", "column name")} = ? AND ${quoteIdentifier("status", "column name")} = ?`, [
 			newNextFireAt,
 			lastFireAt,
 			lastRunId,
 			Date.now(),
+			newStatus ?? null,
 			id,
 			expectedNextFireAt,
 			"active"