@mastra/libsql 1.26.3-alpha.0 → 1.27.0-alpha.2

package/dist/index.js

@@ -1644,6 +1644,27 @@ function buildSelectColumnsWithAlias(tableName, alias) {
 function isLockError(error) {
 	return error.code === "SQLITE_BUSY" || error.code === "SQLITE_LOCKED" || error.message?.toLowerCase().includes("database is locked") || error.message?.toLowerCase().includes("database table is locked") || error.message?.toLowerCase().includes("table is locked") || error.constructor.name === "SqliteError" && error.message?.toLowerCase().includes("locked");
 }
+const TRANSIENT_NETWORK_PATTERNS = [
+	"fetch failed",
+	"econnreset",
+	"etimedout",
+	"eai_again",
+	"socket hang up",
+	"und_err_"
+];
+/**
+* Checks if an error (or any error in its `cause` chain) is a transient network failure
+* such as `fetch failed` from a remote libsql client.
+*/
+function isTransientNetworkError(error) {
+	let current = error;
+	for (let depth = 0; current && depth < 5; depth++) {
+		const text = `${current.message ?? ""} ${current.code ?? ""}`.toLowerCase();
+		if (TRANSIENT_NETWORK_PATTERNS.some((pattern) => text.includes(pattern))) return true;
+		current = current.cause;
+	}
+	return false;
+}
 function createExecuteWriteOperationWithRetry({ logger, maxRetries, initialBackoffMs }) {
 	return async function executeWriteOperationWithRetry(operationFn, operationDescription) {
 		let attempts = 0;
@@ -4352,6 +4373,10 @@ var DatasetsLibSQL = class extends DatasetsStorage {
 			const { page, perPage: perPageInput } = args.pagination;
 			const filterConditions = [];
 			const filterParams = [];
+			if (args.filters?.datasetIds !== void 0) {
+				filterConditions.push(args.filters.datasetIds.length > 0 ? `id IN (${args.filters.datasetIds.map(() => "?").join(",")})` : "1 = 0");
+				filterParams.push(...args.filters.datasetIds);
+			}
 			if (args.filters?.organizationId !== void 0) {
 				filterConditions.push("organizationId = ?");
 				filterParams.push(args.filters.organizationId);
@@ -16579,6 +16604,8 @@ var LibSQLFactoryStorage = class extends FactoryStorage {
 };
 //#endregion
 //#region src/storage/index.ts
+const REMOTE_INIT_CONCURRENCY = 5;
+const REMOTE_INIT_MAX_ATTEMPTS = 3;
 const DEFAULT_LOCAL_CACHE_SIZE = -16e3;
 const DEFAULT_LOCAL_MMAP_SIZE = 134217728;
 /**
@@ -16725,8 +16752,31 @@ var LibSQLStore = class extends MastraCompositeStore {
 		for (const store of this.getStoresToInit()) await store.init();
 		return true;
 	}
+	async initStoreWithRetry(store) {
+		for (let attempt = 1;; attempt++) try {
+			await store.init();
+			return;
+		} catch (error) {
+			if (attempt >= REMOTE_INIT_MAX_ATTEMPTS || !isTransientNetworkError(error)) throw error;
+			const backoff = this.initialBackoffMs * 2 ** (attempt - 1);
+			this.logger.warn(`LibSQLStore: Transient network error during domain init, retrying in ${backoff}ms (attempt ${attempt}/${REMOTE_INIT_MAX_ATTEMPTS}).`, error);
+			await new Promise((resolve) => setTimeout(resolve, backoff));
+		}
+	}
 	async initDomainsInParallel() {
-		await Promise.all(this.getStoresToInit().map((store) => store.init()));
+		const stores = this.getStoresToInit();
+		let next = 0;
+		let failed = false;
+		const worker = async () => {
+			while (!failed && next < stores.length) try {
+				await this.initStoreWithRetry(stores[next++]);
+			} catch (error) {
+				failed = true;
+				throw error;
+			}
+		};
+		const rejected = (await Promise.allSettled(Array.from({ length: Math.min(REMOTE_INIT_CONCURRENCY, stores.length) }, worker))).find((r) => r.status === "rejected");
+		if (rejected) throw rejected.reason;
 		return true;
 	}
 	async init() {
@@ -16737,7 +16787,10 @@ var LibSQLStore = class extends MastraCompositeStore {
 					await this.hasInitialized;
 					return;
 				}
-				this.hasInitialized = this.initDomainsInParallel();
+				this.hasInitialized = this.initDomainsInParallel().catch((error) => {
+					this.hasInitialized = null;
+					throw error;
+				});
 				await this.hasInitialized;
 				return;
 			}
@@ -16749,7 +16802,10 @@ var LibSQLStore = class extends MastraCompositeStore {
 				await this.hasInitialized;
 				return;
 			}
-			this.hasInitialized = this.initDomainsSequentially();
+			this.hasInitialized = this.initDomainsSequentially().catch((error) => {
+				this.hasInitialized = null;
+				throw error;
+			});
 			await this.hasInitialized;
 			return;
 		}