@mastra/mysql 0.6.0 → 0.7.0-alpha.0

package/dist/index.js

@@ -2837,15 +2837,28 @@ var ExperimentsMySQL = class ExperimentsMySQL extends ExperimentsStorage {
 	* Returns default index definitions for the experiments domain tables.
 	*/
 	static getDefaultIndexDefs(_prefix = "") {
-		return [{
-			name: "idx_experiments_org_project",
-			table: TABLE_EXPERIMENTS,
-			columns: ["organizationId", "projectId"]
-		}, {
-			name: "idx_experiment_results_org_project",
-			table: TABLE_EXPERIMENT_RESULTS,
-			columns: ["organizationId", "projectId"]
-		}];
+		return [
+			{
+				name: "idx_experiments_grouping",
+				table: TABLE_EXPERIMENTS,
+				columns: [
+					"experimentSetId",
+					"comparisonId",
+					"variantId",
+					"trialIndex"
+				]
+			},
+			{
+				name: "idx_experiments_org_project",
+				table: TABLE_EXPERIMENTS,
+				columns: ["organizationId", "projectId"]
+			},
+			{
+				name: "idx_experiment_results_org_project",
+				table: TABLE_EXPERIMENT_RESULTS,
+				columns: ["organizationId", "projectId"]
+			}
+		];
 	}
 	/**
 	* Exports DDL statements for all managed tables.
@@ -2901,7 +2914,13 @@ var ExperimentsMySQL = class ExperimentsMySQL extends ExperimentsStorage {
 			ifNotExists: [
 				"agentVersion",
 				"organizationId",
-				"projectId"
+				"projectId",
+				"provenance",
+				"runnerAttestation",
+				"experimentSetId",
+				"comparisonId",
+				"variantId",
+				"trialIndex"
 			]
 		});
 		await this.operations.alterTable({
@@ -2933,6 +2952,12 @@ var ExperimentsMySQL = class ExperimentsMySQL extends ExperimentsStorage {
 			name: row.name ?? void 0,
 			description: row.description ?? void 0,
 			metadata: parseJSON$2(row.metadata),
+			provenance: parseJSON$2(row.provenance) ?? null,
+			runnerAttestation: parseJSON$2(row.runnerAttestation) ?? null,
+			experimentSetId: row.experimentSetId ?? null,
+			comparisonId: row.comparisonId ?? null,
+			variantId: row.variantId ?? null,
+			trialIndex: row.trialIndex ?? null,
 			status: row.status,
 			totalItems: row.totalItems,
 			succeededCount: row.succeededCount,
@@ -2984,6 +3009,12 @@ var ExperimentsMySQL = class ExperimentsMySQL extends ExperimentsStorage {
 					name: input.name ?? null,
 					description: input.description ?? null,
 					metadata: input.metadata ? JSON.stringify(input.metadata) : null,
+					provenance: input.provenance ? JSON.stringify(input.provenance) : null,
+					runnerAttestation: input.runnerAttestation ? JSON.stringify(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,
@@ -3007,6 +3038,12 @@ var ExperimentsMySQL = class ExperimentsMySQL 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,
@@ -3103,6 +3140,22 @@ var ExperimentsMySQL = class ExperimentsMySQL extends ExperimentsStorage {
 				conditions.push(`${quoteIdentifier("status", "column name")} = ?`);
 				params.push(args.status);
 			}
+			if (args.experimentSetId !== void 0) {
+				conditions.push(`${quoteIdentifier("experimentSetId", "column name")} = ?`);
+				params.push(args.experimentSetId);
+			}
+			if (args.comparisonId !== void 0) {
+				conditions.push(`${quoteIdentifier("comparisonId", "column name")} = ?`);
+				params.push(args.comparisonId);
+			}
+			if (args.variantId !== void 0) {
+				conditions.push(`${quoteIdentifier("variantId", "column name")} = ?`);
+				params.push(args.variantId);
+			}
+			if (args.trialIndex !== void 0) {
+				conditions.push(`${quoteIdentifier("trialIndex", "column name")} = ?`);
+				params.push(args.trialIndex);
+			}
 			if (args.filters) {
 				const { organizationId, projectId } = args.filters;
 				if (organizationId !== void 0) {
@@ -5065,12 +5118,16 @@ var MemoryMySQL = class MemoryMySQL extends MemoryStorage {
 				hasMore: perPageInput === false ? false : offset + perPageNormalized < total
 			};
 		} catch (error) {
-			throw new MastraError({
-				id: "MYSQL_MEMORY_LIST_THREADS_FAILED",
+			if (error instanceof MastraError && error.category === ErrorCategory.USER) throw error;
+			const mastraError = new MastraError({
+				id: createStorageErrorId("MYSQL", "LIST_THREADS", "FAILED"),
 				domain: ErrorDomain.STORAGE,
 				category: ErrorCategory.THIRD_PARTY,
 				details: { filter: JSON.stringify(filter ?? {}) }
 			}, error);
+			this.logger?.error?.(mastraError.toString());
+			this.logger?.trackException?.(mastraError);
+			throw mastraError;
 		}
 	}
 	async getThreadsByResourceId(args) {
@@ -5596,8 +5653,9 @@ var MemoryMySQL = class MemoryMySQL extends MemoryStorage {
 				hasMore: metadataFilter || !include?.length || mainThreadMessageCount < total ? baseHasMore : false
 			};
 		} catch (error) {
+			if (error instanceof MastraError && error.category === ErrorCategory.USER) throw error;
 			const mastraError = new MastraError({
-				id: "MYSQL_MEMORY_GET_MESSAGES_PAGINATED_FAILED",
+				id: createStorageErrorId("MYSQL", "LIST_MESSAGES", "FAILED"),
 				domain: ErrorDomain.STORAGE,
 				category: ErrorCategory.THIRD_PARTY,
 				details: {
@@ -5608,13 +5666,7 @@ var MemoryMySQL = class MemoryMySQL extends MemoryStorage {
 			}, error);
 			this.logger?.error?.(mastraError.toString());
 			this.logger?.trackException?.(mastraError);
-			return {
-				messages: [],
-				total: 0,
-				page,
-				perPage,
-				hasMore: false
-			};
+			throw mastraError;
 		}
 	}
 	async listMessagesByResourceId(args) {
@@ -5706,21 +5758,16 @@ var MemoryMySQL = class MemoryMySQL extends MemoryStorage {
 				hasMore
 			};
 		} catch (error) {
+			if (error instanceof MastraError && error.category === ErrorCategory.USER) throw error;
 			const mastraError = new MastraError({
-				id: createStorageErrorId("MYSQL", "LIST_MESSAGES", "FAILED"),
+				id: createStorageErrorId("MYSQL", "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 getResourceById({ resourceId }) {