@mastra/lance 1.5.2 → 1.6.0-alpha.0

package/dist/index.js

@@ -464,7 +464,9 @@ function toRecord(task) {
 		suspendedAt: task.suspendedAt ?? /* @__PURE__ */ new Date(0),
 		completedAt: task.completedAt ?? /* @__PURE__ */ new Date(0),
 		ownerId: task.ownerId ?? null,
-		leaseExpiresAt: task.leaseExpiresAt ?? null
+		leaseExpiresAt: task.leaseExpiresAt ?? null,
+		expiresAt: task.expiresAt ?? null,
+		metadata: serializeJson(task.metadata) ?? null
 	};
 }
 function fromRecord(row) {
@@ -481,6 +483,7 @@ function fromRecord(row) {
 	const suspendedAt = row.suspendedAt instanceof Date ? row.suspendedAt : row.suspendedAt ? new Date(row.suspendedAt) : void 0;
 	const completedAt = row.completedAt instanceof Date ? row.completedAt : row.completedAt ? new Date(row.completedAt) : void 0;
 	const leaseExpiresAt = row.leaseExpiresAt instanceof Date ? row.leaseExpiresAt : row.leaseExpiresAt ? new Date(row.leaseExpiresAt) : void 0;
+	const expiresAt = row.expiresAt instanceof Date ? row.expiresAt : row.expiresAt ? new Date(row.expiresAt) : void 0;
 	return {
 		id: String(row.id),
 		status: String(row.status),
@@ -502,13 +505,16 @@ function fromRecord(row) {
 		suspendedAt: suspendedAt && suspendedAt.getTime() > 0 ? suspendedAt : void 0,
 		completedAt: completedAt && completedAt.getTime() > 0 ? completedAt : void 0,
 		ownerId: row.ownerId != null && row.ownerId !== "" ? String(row.ownerId) : void 0,
-		leaseExpiresAt
+		leaseExpiresAt,
+		expiresAt,
+		metadata: parseJson(row.metadata)
 	};
 }
 function escapeStr(val) {
 	return val.replace(/'/g, "''");
 }
 var StoreBackgroundTasksLance = class extends BackgroundTasksStorage {
+	supportsTaskExpiry = true;
 	client;
 	#db;
 	constructor(config) {
@@ -529,7 +535,9 @@ var StoreBackgroundTasksLance = class extends BackgroundTasksStorage {
 				"suspend_payload",
 				"suspendedAt",
 				"ownerId",
-				"leaseExpiresAt"
+				"leaseExpiresAt",
+				"expiresAt",
+				"metadata"
 			]
 		});
 	}
@@ -551,11 +559,14 @@ var StoreBackgroundTasksLance = class extends BackgroundTasksStorage {
 		if ("completedAt" in update) values.completedAt = update.completedAt?.getTime() ?? 0;
 		if ("ownerId" in update) values.ownerId = update.ownerId ?? null;
 		if ("leaseExpiresAt" in update) values.leaseExpiresAt = update.leaseExpiresAt?.getTime() ?? null;
+		if ("expiresAt" in update) values.expiresAt = update.expiresAt?.getTime() ?? null;
+		if ("metadata" in update) values.metadata = serializeJson(update.metadata) ?? null;
 		if (Object.keys(values).length === 0) return false;
 		const conditions = [`id = '${escapeStr(taskId)}'`];
 		if (options?.expectedStatus) conditions.push(`status = '${escapeStr(options.expectedStatus)}'`);
 		if (options?.expectedOwnerId !== void 0) conditions.push(options.expectedOwnerId === null ? "ownerId IS NULL" : `ownerId = '${escapeStr(options.expectedOwnerId)}'`);
 		if (options?.expectedLeaseExpiresAt !== void 0) conditions.push(options.expectedLeaseExpiresAt === null ? "leaseExpiresAt IS NULL" : `leaseExpiresAt = ${options.expectedLeaseExpiresAt.getTime()}`);
+		if (options?.expectedSuspendedAt !== void 0) conditions.push(options.expectedSuspendedAt === null ? "(suspendedAt IS NULL OR suspendedAt = 0)" : `suspendedAt = ${options.expectedSuspendedAt.getTime()}`);
 		return (await (await this.client.openTable(TABLE_BACKGROUND_TASKS)).update({
 			where: conditions.join(" AND "),
 			values
@@ -587,6 +598,7 @@ var StoreBackgroundTasksLance = class extends BackgroundTasksStorage {
 			let query = table.query();
 			if (conditions.length > 0) query = query.where(conditions.join(" AND "));
 			let tasks = (await query.toArray()).map(fromRecord);
+			if (filter.hasExpiresAt !== void 0) tasks = tasks.filter((t) => t.expiresAt != null === filter.hasExpiresAt);
 			const dateCol = filter.dateFilterBy ?? "createdAt";
 			if (filter.fromDate) tasks = tasks.filter((t) => {
 				const val = t[dateCol];