@mastra/lance 1.5.2-alpha.1 → 1.6.0-alpha.0

package/dist/index.js

@@ -411,7 +411,7 @@ var LanceDB = class extends MastraBase {
 					const quotedKey = /^[a-z][a-zA-Z]*$/.test(key) && /[A-Z]/.test(key) ? `\`${key}\`` : key;
 					if (typeof value === "string") return `${quotedKey} = '${value}'`;
 					else if (value === null) return `${quotedKey} IS NULL`;
-					else return `${quotedKey} = ${value}`;
+					return `${quotedKey} = ${value}`;
 				}).join(" AND ");
 				this.logger.debug("where clause generated: " + filterConditions);
 				query.where(filterConditions);
@@ -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];
@@ -1985,13 +1997,13 @@ var LanceFilterTranslator = class extends BaseFilterTranslator {
 			if (keys.length > 1) return keys.map((key) => {
 				const value = obj[key];
 				if (this.isNestedObject(value) && !this.isDateObject(value)) return this.processNestedObject(key, value);
-				else return this.processField(key, value);
+				return this.processField(key, value);
 			}).join(" AND ");
 			if (keys.length === 1) {
 				const key = keys[0];
 				const value = obj[key];
 				if (this.isNestedObject(value) && !this.isDateObject(value)) return this.processNestedObject(key, value);
-				else return this.processField(key, value);
+				return this.processField(key, value);
 			}
 		}
 		return "";
@@ -2006,7 +2018,7 @@ var LanceFilterTranslator = class extends BaseFilterTranslator {
 			const keys = Object.keys(condObj);
 			if (keys.length === 1 && this.isOperator(keys[0])) {
 				if (this.isLogicalOperator(keys[0])) return `(${this.processLogicalOperator(keys[0], condObj[keys[0]])})`;
-				else throw new Error(BaseFilterTranslator.ErrorMessages.UNSUPPORTED_OPERATOR(keys[0]));
+				throw new Error(BaseFilterTranslator.ErrorMessages.UNSUPPORTED_OPERATOR(keys[0]));
 			}
 			if (keys.length > 1) return `(${this.processFilter(condition)})`;
 			return this.processFilter(condition);
@@ -2021,7 +2033,7 @@ var LanceFilterTranslator = class extends BaseFilterTranslator {
 		return keys.map((key) => {
 			const nestedPath = key.includes(".") ? `${path}.${key}` : `${path}.${key}`;
 			if (this.isNestedObject(obj[key]) && !this.isDateObject(obj[key])) return this.processNestedObject(nestedPath, obj[key]);
-			else return this.processField(nestedPath, obj[key]);
+			return this.processField(nestedPath, obj[key]);
 		}).join(" AND ");
 	}
 	processField(field, value) {
@@ -2593,7 +2605,8 @@ var LanceVectorStore = class LanceVectorStore extends MastraVector {
 				}
 				this.logger.debug(`Table ${resolvedTableName} created. Index creation deferred until data is available.`);
 				return;
-			} else table = await this.lanceClient.openTable(resolvedTableName);
+			}
+			table = await this.lanceClient.openTable(resolvedTableName);
 			const metricType = this.mastraMetricToLance(metric);
 			const rowCount = await table.countRows();
 			if (rowCount < 256) {