@mastra/libsql 1.21.1 → 1.22.0-alpha.2

package/dist/index.js

@@ -2966,7 +2966,7 @@ var BackgroundTasksLibSQL = class BackgroundTasksLibSQL extends BackgroundTasksS
 			}
 		});
 	}
-	async updateTask(taskId, update) {
+	async updateTask(taskId, update, options) {
 		const setClauses = [];
 		const params = [];
 		if ("status" in update) {
@@ -3001,12 +3001,17 @@ var BackgroundTasksLibSQL = class BackgroundTasksLibSQL extends BackgroundTasksS
 			setClauses.push("completedAt = ?");
 			params.push(update.completedAt?.toISOString() ?? null);
 		}
-		if (setClauses.length === 0) return;
+		if (setClauses.length === 0) return false;
 		params.push(taskId);
-		await this.#client.execute({
-			sql: `UPDATE ${TABLE_BACKGROUND_TASKS} SET ${setClauses.join(", ")} WHERE id = ?`,
+		let where = "id = ?";
+		if (options?.expectedStatus) {
+			where += " AND status = ?";
+			params.push(options.expectedStatus);
+		}
+		return (await this.#client.execute({
+			sql: `UPDATE ${TABLE_BACKGROUND_TASKS} SET ${setClauses.join(", ")} WHERE ${where}`,
 			args: params
-		});
+		})).rowsAffected > 0;
 	}
 	async getTask(taskId) {
 		const row = (await this.#client.execute({
@@ -4616,6 +4621,7 @@ var ExperimentsLibSQL = class extends ExperimentsStorage {
 				"tags",
 				"comment",
 				"toolMockReport",
+				"metadata",
 				"organizationId",
 				"projectId",
 				"attempt"
@@ -4762,6 +4768,7 @@ var ExperimentsLibSQL = class extends ExperimentsStorage {
 			input: safelyParseJSON(row.input),
 			output: row.output ? safelyParseJSON(row.output) : null,
 			groundTruth: row.groundTruth ? safelyParseJSON(row.groundTruth) : null,
+			metadata: row.metadata ? safelyParseJSON(row.metadata) : null,
 			error: row.error ? safelyParseJSON(row.error) : null,
 			startedAt: ensureDate(row.startedAt),
 			completedAt: ensureDate(row.completedAt),
@@ -5064,6 +5071,7 @@ var ExperimentsLibSQL = class extends ExperimentsStorage {
 					input: input.input,
 					output: input.output,
 					groundTruth: input.groundTruth,
+					metadata: input.metadata ?? null,
 					error: input.error ?? null,
 					startedAt: input.startedAt.toISOString(),
 					completedAt: input.completedAt.toISOString(),
@@ -5086,6 +5094,7 @@ var ExperimentsLibSQL = class extends ExperimentsStorage {
 				input: input.input,
 				output: input.output,
 				groundTruth: input.groundTruth,
+				metadata: input.metadata ?? null,
 				error: input.error,
 				startedAt: input.startedAt,
 				completedAt: input.completedAt,
@@ -5140,7 +5149,7 @@ var ExperimentsLibSQL = class extends ExperimentsStorage {
 			await this.#client.execute({
 				sql: `UPDATE ${TABLE_EXPERIMENT_RESULTS} SET
           "itemDatasetVersion" = ?, "organizationId" = ?, "projectId" = ?,
-          "input" = ?, "output" = ?, "groundTruth" = ?, "error" = ?,
+          "input" = ?, "output" = ?, "groundTruth" = ?, "metadata" = ?, "error" = ?,
           "startedAt" = ?, "completedAt" = ?, "retryCount" = ?, "attempt" = ?,
           "traceId" = ?, "status" = ?, "tags" = ?, "toolMockReport" = ?
         WHERE "id" = ?`,
@@ -5151,6 +5160,7 @@ var ExperimentsLibSQL = class extends ExperimentsStorage {
 					JSON.stringify(input.input),
 					input.output != null ? JSON.stringify(input.output) : null,
 					input.groundTruth != null ? JSON.stringify(input.groundTruth) : null,
+					input.metadata != null ? JSON.stringify(input.metadata) : null,
 					input.error != null ? JSON.stringify(input.error) : null,
 					input.startedAt.toISOString(),
 					input.completedAt.toISOString(),
@@ -13054,7 +13064,7 @@ var WorkflowsLibSQL = class WorkflowsLibSQL extends WorkflowsStorage {
 			sql: `INSERT INTO ${TABLE_WORKFLOW_SNAPSHOT} (workflow_name, run_id, resourceId, snapshot, createdAt, updatedAt)
                 VALUES (?, ?, ?, jsonb(?), ?, ?)
                 ON CONFLICT(workflow_name, run_id)
-                DO UPDATE SET resourceId = excluded.resourceId, snapshot = excluded.snapshot, updatedAt = excluded.updatedAt`,
+                DO UPDATE SET resourceId = COALESCE(excluded.resourceId, ${TABLE_WORKFLOW_SNAPSHOT}.resourceId), snapshot = excluded.snapshot, updatedAt = excluded.updatedAt`,
 			args: [
 				workflowName,
 				runId,
@@ -13854,6 +13864,15 @@ var LibSQLFactoryStorageOps = class {
 	async findMany(collection, where, opts) {
 		return this.#select(collection, where, opts);
 	}
+	async count(collection, where) {
+		const schema = this.#schema(collection);
+		const filter = this.#buildWhere(schema, where);
+		const result = await this.#client.execute({
+			sql: `SELECT COUNT(*) AS count FROM "${schema.name}" WHERE ${filter.sql}`,
+			args: filter.args
+		});
+		return Number(result.rows[0]?.count ?? 0);
+	}
 	async #insertOne(collection, row) {
 		const schema = this.#schema(collection);
 		const pk = primaryKeyOf(schema);
@@ -14003,7 +14022,7 @@ var LibSQLFactoryStorage = class extends FactoryStorage {
 		});
 	}
 	async close() {
-		this.#client.close();
+		await this.#client.close();
 	}
 	authDatabase() {
 		return {
@@ -14123,10 +14142,12 @@ var LibSQLStore = class extends MastraCompositeStore {
 		};
 		if ("url" in config) {
 			if (config.url.includes(":memory:")) this.shouldCacheInit = false;
-			this.isLocalDb = config.url.startsWith("file:") || config.url.includes(":memory:");
+			this.isLocalDb = (config.url.startsWith("file:") || config.url.includes(":memory:")) && !config.syncUrl;
 			this.client = createClient({
 				url: config.url,
 				...config.authToken ? { authToken: config.authToken } : {},
+				...config.syncUrl ? { syncUrl: config.syncUrl } : {},
+				...config.syncInterval !== void 0 ? { syncInterval: config.syncInterval } : {},
 				...this.isLocalDb ? { timeout: this.connectionTimeoutMs } : {}
 			});
 			this.pragmasReady = this.isLocalDb ? this.applyLocalPragmas() : Promise.resolve();
@@ -14250,7 +14271,7 @@ var LibSQLStore = class extends MastraCompositeStore {
 	* Safe to call more than once; subsequent calls are no-ops.
 	*/
 	async close() {
-		if (!this.client.closed) this.client.close();
+		if (!this.client.closed) await this.client.close();
 	}
 };
 //#endregion