@mastra/libsql 1.24.1 → 1.25.1-alpha.0

package/dist/index.js

@@ -7826,6 +7826,7 @@ function addSqliteMetadataValuePredicate(conditions, params, path, value) {
 var MemoryLibSQL = class MemoryLibSQL extends MemoryStorage {
 	supportsPartialThreadUpdate = true;
 	supportsObservationalMemory = true;
+	supportsObservationalMemoryHistorySearch = true;
 	/**
 	* Retention-eligible tables. `threads`, `messages`, and `resources` all anchor
 	* on `createdAt` and are indexed for fast batched deletes. Cascade order is
@@ -8997,6 +8998,10 @@ var MemoryLibSQL = class MemoryLibSQL extends MemoryStorage {
 			const lookupKey = this.getOMKey(threadId, resourceId);
 			const conditions = [`"lookupKey" = ?`];
 			const args = [lookupKey];
+			if (options?.recordId !== void 0) {
+				conditions.push(`id = ?`);
+				args.push(options.recordId);
+			}
 			if (options?.from) {
 				conditions.push(`"createdAt" >= ?`);
 				args.push(options.from.toISOString());
@@ -9005,8 +9010,25 @@ var MemoryLibSQL = class MemoryLibSQL extends MemoryStorage {
 				conditions.push(`"createdAt" <= ?`);
 				args.push(options.to.toISOString());
 			}
+			if (options?.groupId !== void 0) {
+				conditions.push(`(instr("activeObservations", ?) > 0 OR EXISTS (
+          SELECT 1 FROM json_each("bufferedObservationChunks") AS chunk
+          WHERE instr(json_extract(chunk.value, '$.observations'), ?) > 0
+        ))`);
+				const prefix = `<observation-group id="${options.groupId}"`;
+				args.push(prefix, prefix);
+			}
+			if (options?.beforeGeneration !== void 0) {
+				conditions.push(`"generationCount" < ?`);
+				args.push(options.beforeGeneration);
+			}
+			if (options?.afterGeneration !== void 0) {
+				conditions.push(`"generationCount" > ?`);
+				args.push(options.afterGeneration);
+			}
+			const direction = options?.sortDirection === "ASC" ? "ASC" : "DESC";
 			args.push(limit);
-			let sql = `SELECT * FROM "${OM_TABLE}" WHERE ${conditions.join(" AND ")} ORDER BY "generationCount" DESC LIMIT ?`;
+			let sql = `SELECT * FROM "${OM_TABLE}" WHERE ${conditions.join(" AND ")} ORDER BY "generationCount" ${direction}, "createdAt" ASC, id ASC LIMIT ?`;
 			if (options?.offset != null) {
 				args.push(options.offset);
 				sql += ` OFFSET ?`;
@@ -11432,16 +11454,17 @@ var SchedulesLibSQL = class SchedulesLibSQL extends SchedulesStorage {
 		if (!updated) throw new Error(`Schedule ${id} not found`);
 		return updated;
 	}
-	async updateScheduleNextFire(id, expectedNextFireAt, newNextFireAt, lastFireAt, lastRunId) {
+	async updateScheduleNextFire(id, expectedNextFireAt, newNextFireAt, lastFireAt, lastRunId, newStatus) {
 		return ((await this.#client.execute({
 			sql: `UPDATE ${TABLE_SCHEDULES}
-            SET next_fire_at = ?, last_fire_at = ?, last_run_id = ?, updated_at = ?
+            SET next_fire_at = ?, last_fire_at = ?, last_run_id = ?, updated_at = ?, status = COALESCE(?, status)
             WHERE id = ? AND next_fire_at = ? AND status = ?`,
 			args: [
 				newNextFireAt,
 				lastFireAt,
 				lastRunId,
 				Date.now(),
+				newStatus ?? null,
 				id,
 				expectedNextFireAt,
 				"active"