@mastra/libsql 1.21.1 → 1.22.0-alpha.1

package/dist/index.js

@@ -13054,7 +13054,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 +13854,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 +14012,7 @@ var LibSQLFactoryStorage = class extends FactoryStorage {
 		});
 	}
 	async close() {
-		this.#client.close();
+		await this.#client.close();
 	}
 	authDatabase() {
 		return {
@@ -14250,7 +14259,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