@mastra/libsql 1.19.0 → 1.20.0-alpha.1

package/dist/index.js

@@ -4591,7 +4591,13 @@ var ExperimentsLibSQL = class extends ExperimentsStorage {
 			ifNotExists: [
 				"agentVersion",
 				"organizationId",
-				"projectId"
+				"projectId",
+				"provenance",
+				"runnerAttestation",
+				"experimentSetId",
+				"comparisonId",
+				"variantId",
+				"trialIndex"
 			]
 		});
 		await this.#db.alterTable({
@@ -4611,6 +4617,10 @@ var ExperimentsLibSQL = class extends ExperimentsStorage {
 				sql: `CREATE INDEX IF NOT EXISTS idx_experiments_datasetid ON "${TABLE_EXPERIMENTS}" ("datasetId")`,
 				args: []
 			},
+			{
+				sql: `CREATE INDEX IF NOT EXISTS idx_experiments_grouping ON "${TABLE_EXPERIMENTS}" ("experimentSetId", "comparisonId", "variantId", "trialIndex")`,
+				args: []
+			},
 			{
 				sql: `CREATE INDEX IF NOT EXISTS idx_experiment_results_experimentid ON "${TABLE_EXPERIMENT_RESULTS}" ("experimentId")`,
 				args: []
@@ -4710,6 +4720,12 @@ var ExperimentsLibSQL = class extends ExperimentsStorage {
 			name: row.name ?? void 0,
 			description: row.description ?? void 0,
 			metadata: row.metadata ? safelyParseJSON(row.metadata) : void 0,
+			provenance: row.provenance ? safelyParseJSON(row.provenance) : null,
+			runnerAttestation: row.runnerAttestation ? safelyParseJSON(row.runnerAttestation) : null,
+			experimentSetId: row.experimentSetId ?? null,
+			comparisonId: row.comparisonId ?? null,
+			variantId: row.variantId ?? null,
+			trialIndex: row.trialIndex != null ? row.trialIndex : null,
 			status: row.status,
 			totalItems: row.totalItems,
 			succeededCount: row.succeededCount,
@@ -4763,6 +4779,12 @@ var ExperimentsLibSQL = class extends ExperimentsStorage {
 					name: input.name ?? null,
 					description: input.description ?? null,
 					metadata: input.metadata ?? null,
+					provenance: input.provenance ?? null,
+					runnerAttestation: input.runnerAttestation ?? null,
+					experimentSetId: input.experimentSetId ?? null,
+					comparisonId: input.comparisonId ?? null,
+					variantId: input.variantId ?? null,
+					trialIndex: input.trialIndex ?? null,
 					status: "pending",
 					totalItems: input.totalItems,
 					succeededCount: 0,
@@ -4786,6 +4808,12 @@ var ExperimentsLibSQL = class extends ExperimentsStorage {
 				name: input.name,
 				description: input.description,
 				metadata: input.metadata,
+				provenance: input.provenance ?? null,
+				runnerAttestation: input.runnerAttestation ?? null,
+				experimentSetId: input.experimentSetId ?? null,
+				comparisonId: input.comparisonId ?? null,
+				variantId: input.variantId ?? null,
+				trialIndex: input.trialIndex ?? null,
 				status: "pending",
 				totalItems: input.totalItems,
 				succeededCount: 0,
@@ -4911,6 +4939,22 @@ var ExperimentsLibSQL = class extends ExperimentsStorage {
 				conditions.push("status = ?");
 				queryParams.push(args.status);
 			}
+			if (args.experimentSetId !== void 0) {
+				conditions.push("experimentSetId = ?");
+				queryParams.push(args.experimentSetId);
+			}
+			if (args.comparisonId !== void 0) {
+				conditions.push("comparisonId = ?");
+				queryParams.push(args.comparisonId);
+			}
+			if (args.variantId !== void 0) {
+				conditions.push("variantId = ?");
+				queryParams.push(args.variantId);
+			}
+			if (args.trialIndex !== void 0) {
+				conditions.push("trialIndex = ?");
+				queryParams.push(args.trialIndex);
+			}
 			if (args.filters) {
 				const { organizationId, projectId } = args.filters;
 				if (organizationId !== void 0) {
@@ -6677,14 +6721,21 @@ var MemoryLibSQL = class MemoryLibSQL extends MemoryStorage {
 			return direction === "ASC" ? String(aValue).localeCompare(String(bValue)) : String(bValue).localeCompare(String(aValue));
 		});
 	}
-	async _getIncludedMessages({ include }) {
+	/**
+	* Fetches included messages by ID, discovering their thread automatically.
+	* This handles cross-thread includes where the include item doesn't specify a threadId.
+	* When a resourceId is given, both the target lookup and the surrounding window stay
+	* inside that resource, so an include never leaks another resource's messages.
+	*/
+	async _getIncludedMessages({ include, resourceId }) {
 		if (!include || include.length === 0) return null;
 		const targetIds = include.map((inc) => inc.id).filter(Boolean);
 		if (targetIds.length === 0) return null;
+		const resourceCondition = resourceId ? ` AND "resourceId" = ?` : "";
 		const idPlaceholders = targetIds.map(() => "?").join(", ");
 		const targetResult = await this.#client.execute({
-			sql: `SELECT id, thread_id, "createdAt" FROM "${TABLE_MESSAGES}" WHERE id IN (${idPlaceholders})`,
-			args: targetIds
+			sql: `SELECT id, thread_id, "createdAt" FROM "${TABLE_MESSAGES}" WHERE id IN (${idPlaceholders})${resourceCondition}`,
+			args: resourceId ? [...targetIds, resourceId] : targetIds
 		});
 		if (!targetResult.rows || targetResult.rows.length === 0) return null;
 		const targetMap = new Map(targetResult.rows.map((r) => [r.id, {
@@ -6701,21 +6752,25 @@ var MemoryLibSQL = class MemoryLibSQL extends MemoryStorage {
         SELECT id, content, role, type, "createdAt", thread_id, "resourceId"
         FROM "${TABLE_MESSAGES}"
         WHERE thread_id = ?
-          AND "createdAt" <= ?
+          AND "createdAt" <= ?${resourceCondition}
         ORDER BY "createdAt" DESC, id DESC
         LIMIT ?
       )`);
-			params.push(target.threadId, target.createdAt, withPreviousMessages + 1);
+			params.push(target.threadId, target.createdAt);
+			if (resourceId) params.push(resourceId);
+			params.push(withPreviousMessages + 1);
 			if (withNextMessages > 0) {
 				unionQueries.push(`SELECT * FROM (
           SELECT id, content, role, type, "createdAt", thread_id, "resourceId"
           FROM "${TABLE_MESSAGES}"
           WHERE thread_id = ?
-            AND "createdAt" > ?
+            AND "createdAt" > ?${resourceCondition}
           ORDER BY "createdAt" ASC, id ASC
           LIMIT ?
         )`);
-				params.push(target.threadId, target.createdAt, withNextMessages);
+				params.push(target.threadId, target.createdAt);
+				if (resourceId) params.push(resourceId);
+				params.push(withNextMessages);
 			}
 		}
 		if (unionQueries.length === 0) return null;
@@ -6811,7 +6866,10 @@ var MemoryLibSQL = class MemoryLibSQL extends MemoryStorage {
 				hasMore: false
 			};
 			if (perPage === 0 && include && include.length > 0) {
-				const includeMessages = await this._getIncludedMessages({ include });
+				const includeMessages = await this._getIncludedMessages({
+					include,
+					resourceId
+				});
 				if (!includeMessages || includeMessages.length === 0) return {
 					messages: [],
 					total: 0,
@@ -6852,7 +6910,10 @@ var MemoryLibSQL = class MemoryLibSQL extends MemoryStorage {
 			};
 			const messageIds = new Set(messages.map((m) => m.id));
 			if (include && include.length > 0) {
-				const includeMessages = await this._getIncludedMessages({ include });
+				const includeMessages = await this._getIncludedMessages({
+					include,
+					resourceId
+				});
 				if (includeMessages) {
 					for (const includeMsg of includeMessages) if (!messageIds.has(includeMsg.id)) {
 						messages.push(includeMsg);
@@ -6872,6 +6933,7 @@ var MemoryLibSQL = class MemoryLibSQL extends MemoryStorage {
 				hasMore: metadataFilter ? perPageInput !== false && offset + paginatedCount < total : perPageInput !== false && !allThreadMessagesReturned && offset + perPage < total
 			};
 		} catch (error) {
+			if (error instanceof MastraError && error.category === ErrorCategory.USER) throw error;
 			const mastraError = new MastraError({
 				id: createStorageErrorId("LIBSQL", "LIST_MESSAGES", "FAILED"),
 				domain: ErrorDomain.STORAGE,
@@ -6883,25 +6945,19 @@ var MemoryLibSQL = class MemoryLibSQL extends MemoryStorage {
 			}, error);
 			this.logger?.error?.(mastraError.toString());
 			this.logger?.trackException?.(mastraError);
-			return {
-				messages: [],
-				total: 0,
-				page,
-				perPage: perPageForResponse,
-				hasMore: false
-			};
+			throw mastraError;
 		}
 	}
 	async listMessagesByResourceId(args) {
 		const { resourceId, include, filter, perPage: perPageInput, page = 0, orderBy } = args;
 		if (!resourceId || typeof resourceId !== "string" || resourceId.trim().length === 0) throw new MastraError({
-			id: createStorageErrorId("LIBSQL", "LIST_MESSAGES", "INVALID_QUERY"),
+			id: createStorageErrorId("LIBSQL", "LIST_MESSAGES_BY_RESOURCE_ID", "INVALID_QUERY"),
 			domain: ErrorDomain.STORAGE,
 			category: ErrorCategory.USER,
 			details: { resourceId: resourceId ?? "" }
 		}, /* @__PURE__ */ new Error("resourceId is required"));
 		if (page < 0) throw new MastraError({
-			id: createStorageErrorId("LIBSQL", "LIST_MESSAGES", "INVALID_PAGE"),
+			id: createStorageErrorId("LIBSQL", "LIST_MESSAGES_BY_RESOURCE_ID", "INVALID_PAGE"),
 			domain: ErrorDomain.STORAGE,
 			category: ErrorCategory.USER,
 			details: { page }
@@ -6936,7 +6992,10 @@ var MemoryLibSQL = class MemoryLibSQL extends MemoryStorage {
 				hasMore: false
 			};
 			if (perPage === 0 && include && include.length > 0) {
-				const includeMessages = await this._getIncludedMessages({ include });
+				const includeMessages = await this._getIncludedMessages({
+					include,
+					resourceId
+				});
 				if (!includeMessages || includeMessages.length === 0) return {
 					messages: [],
 					total: 0,
@@ -6976,7 +7035,10 @@ var MemoryLibSQL = class MemoryLibSQL extends MemoryStorage {
 			};
 			const messageIds = new Set(messages.map((m) => m.id));
 			if (include && include.length > 0) {
-				const includeMessages = await this._getIncludedMessages({ include });
+				const includeMessages = await this._getIncludedMessages({
+					include,
+					resourceId
+				});
 				if (includeMessages) {
 					for (const includeMsg of includeMessages) if (!messageIds.has(includeMsg.id)) {
 						messages.push(includeMsg);
@@ -6993,21 +7055,16 @@ var MemoryLibSQL = class MemoryLibSQL extends MemoryStorage {
 				hasMore: perPageInput !== false && offset + perPage < total
 			};
 		} catch (error) {
+			if (error instanceof MastraError && error.category === ErrorCategory.USER) throw error;
 			const mastraError = new MastraError({
-				id: createStorageErrorId("LIBSQL", "LIST_MESSAGES", "FAILED"),
+				id: createStorageErrorId("LIBSQL", "LIST_MESSAGES_BY_RESOURCE_ID", "FAILED"),
 				domain: ErrorDomain.STORAGE,
 				category: ErrorCategory.THIRD_PARTY,
 				details: { resourceId }
 			}, error);
 			this.logger?.error?.(mastraError.toString());
 			this.logger?.trackException?.(mastraError);
-			return {
-				messages: [],
-				total: 0,
-				page,
-				perPage: perPageForResponse,
-				hasMore: false
-			};
+			throw mastraError;
 		}
 	}
 	async saveMessages({ messages }) {
@@ -7355,13 +7412,7 @@ var MemoryLibSQL = class MemoryLibSQL extends MemoryStorage {
 			}, error);
 			this.logger?.trackException?.(mastraError);
 			this.logger?.error?.(mastraError.toString());
-			return {
-				threads: [],
-				total: 0,
-				page,
-				perPage: perPageForResponse,
-				hasMore: false
-			};
+			throw mastraError;
 		}
 	}
 	async saveThread({ thread }) {