@mastra/mysql 0.5.0 → 0.6.0-alpha.0

package/dist/index.js

@@ -1906,6 +1906,7 @@ var DatasetsMySQL = class DatasetsMySQL extends DatasetsStorage {
 				"source",
 				"expectedTrajectory",
 				"toolMocks",
+				"unmockedToolPolicy",
 				"externalId"
 			]
 		});
@@ -1961,6 +1962,7 @@ var DatasetsMySQL = class DatasetsMySQL extends DatasetsStorage {
 			groundTruth: row.groundTruth ? parseJSON$3(row.groundTruth) : void 0,
 			expectedTrajectory: row.expectedTrajectory ? parseJSON$3(row.expectedTrajectory) : void 0,
 			toolMocks: row.toolMocks ? parseJSON$3(row.toolMocks) : void 0,
+			unmockedToolPolicy: row.unmockedToolPolicy ?? void 0,
 			requestContext: row.requestContext ? parseJSON$3(row.requestContext) : void 0,
 			metadata: row.metadata ? parseJSON$3(row.metadata) : void 0,
 			source: row.source ? parseJSON$3(row.source) : void 0,
@@ -1982,6 +1984,7 @@ var DatasetsMySQL = class DatasetsMySQL extends DatasetsStorage {
 			groundTruth: row.groundTruth ? parseJSON$3(row.groundTruth) : void 0,
 			expectedTrajectory: row.expectedTrajectory ? parseJSON$3(row.expectedTrajectory) : void 0,
 			toolMocks: row.toolMocks ? parseJSON$3(row.toolMocks) : void 0,
+			unmockedToolPolicy: row.unmockedToolPolicy ?? void 0,
 			requestContext: row.requestContext ? parseJSON$3(row.requestContext) : void 0,
 			metadata: row.metadata ? parseJSON$3(row.metadata) : void 0,
 			source: row.source ? parseJSON$3(row.source) : void 0,
@@ -2263,7 +2266,7 @@ var DatasetsMySQL = class DatasetsMySQL extends DatasetsStorage {
 			const newVersion = parentRow?.version;
 			const parentOrganizationId = parentRow?.organizationId ?? null;
 			const parentProjectId = parentRow?.projectId ?? null;
-			await connection.execute(`INSERT INTO ${tableItemsName} (\`id\`, \`datasetId\`, \`datasetVersion\`, \`organizationId\`, \`projectId\`, \`validTo\`, \`isDeleted\`, \`input\`, \`groundTruth\`, \`metadata\`, \`createdAt\`, \`updatedAt\`) VALUES (?, ?, ?, ?, ?, NULL, 0, ?, ?, ?, ?, ?)`, [
+			await connection.execute(`INSERT INTO ${tableItemsName} (\`id\`, \`datasetId\`, \`datasetVersion\`, \`organizationId\`, \`projectId\`, \`validTo\`, \`isDeleted\`, \`input\`, \`groundTruth\`, \`unmockedToolPolicy\`, \`metadata\`, \`createdAt\`, \`updatedAt\`) VALUES (?, ?, ?, ?, ?, NULL, 0, ?, ?, ?, ?, ?, ?)`, [
 				id,
 				args.datasetId,
 				newVersion,
@@ -2271,6 +2274,7 @@ var DatasetsMySQL = class DatasetsMySQL extends DatasetsStorage {
 				parentProjectId,
 				jsonArg(args.input),
 				jsonArg(args.groundTruth),
+				args.unmockedToolPolicy ?? null,
 				jsonArg(args.metadata),
 				transformToSqlValue(now),
 				transformToSqlValue(now)
@@ -2290,6 +2294,7 @@ var DatasetsMySQL = class DatasetsMySQL extends DatasetsStorage {
 				projectId: parentProjectId,
 				input: args.input,
 				groundTruth: args.groundTruth,
+				unmockedToolPolicy: args.unmockedToolPolicy,
 				metadata: args.metadata,
 				createdAt: now,
 				updatedAt: now
@@ -2337,6 +2342,7 @@ var DatasetsMySQL = class DatasetsMySQL extends DatasetsStorage {
 			const mergedGroundTruth = args.groundTruth ?? existing.groundTruth;
 			const mergedExpectedTrajectory = args.expectedTrajectory ?? existing.expectedTrajectory;
 			const mergedToolMocks = args.toolMocks ?? existing.toolMocks;
+			const mergedUnmockedToolPolicy = args.unmockedToolPolicy ?? existing.unmockedToolPolicy;
 			const mergedRequestContext = args.requestContext ?? existing.requestContext;
 			const mergedMetadata = args.metadata ?? existing.metadata;
 			const mergedSource = args.source ?? existing.source;
@@ -2347,7 +2353,7 @@ var DatasetsMySQL = class DatasetsMySQL extends DatasetsStorage {
 			const parentOrganizationId = parentRow?.organizationId ?? null;
 			const parentProjectId = parentRow?.projectId ?? null;
 			await connection.execute(`UPDATE ${tableItemsName} SET \`validTo\` = ? WHERE \`id\` = ? AND \`validTo\` IS NULL AND \`isDeleted\` = 0`, [newVersion, args.id]);
-			await connection.execute(`INSERT INTO ${tableItemsName} (\`id\`, \`datasetId\`, \`datasetVersion\`, \`externalId\`, \`organizationId\`, \`projectId\`, \`validTo\`, \`isDeleted\`, \`input\`, \`groundTruth\`, \`expectedTrajectory\`, \`toolMocks\`, \`requestContext\`, \`metadata\`, \`source\`, \`createdAt\`, \`updatedAt\`) VALUES (?, ?, ?, ?, ?, ?, NULL, 0, ?, ?, ?, ?, ?, ?, ?, ?, ?)`, [
+			await connection.execute(`INSERT INTO ${tableItemsName} (\`id\`, \`datasetId\`, \`datasetVersion\`, \`externalId\`, \`organizationId\`, \`projectId\`, \`validTo\`, \`isDeleted\`, \`input\`, \`groundTruth\`, \`expectedTrajectory\`, \`toolMocks\`, \`unmockedToolPolicy\`, \`requestContext\`, \`metadata\`, \`source\`, \`createdAt\`, \`updatedAt\`) VALUES (?, ?, ?, ?, ?, ?, NULL, 0, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`, [
 				args.id,
 				args.datasetId,
 				newVersion,
@@ -2358,6 +2364,7 @@ var DatasetsMySQL = class DatasetsMySQL extends DatasetsStorage {
 				jsonArg(mergedGroundTruth),
 				jsonArg(mergedExpectedTrajectory),
 				jsonArg(mergedToolMocks),
+				mergedUnmockedToolPolicy ?? null,
 				jsonArg(mergedRequestContext),
 				jsonArg(mergedMetadata),
 				jsonArg(mergedSource),
@@ -2380,6 +2387,7 @@ var DatasetsMySQL = class DatasetsMySQL extends DatasetsStorage {
 				groundTruth: mergedGroundTruth,
 				expectedTrajectory: mergedExpectedTrajectory,
 				toolMocks: mergedToolMocks,
+				unmockedToolPolicy: mergedUnmockedToolPolicy,
 				requestContext: mergedRequestContext,
 				metadata: mergedMetadata,
 				source: mergedSource,
@@ -2425,7 +2433,7 @@ var DatasetsMySQL = class DatasetsMySQL extends DatasetsStorage {
 			const parentOrganizationId = parentRow?.organizationId ?? null;
 			const parentProjectId = parentRow?.projectId ?? null;
 			await connection.execute(`UPDATE ${tableItemsName} SET \`validTo\` = ? WHERE \`id\` = ? AND \`validTo\` IS NULL AND \`isDeleted\` = 0`, [newVersion, id]);
-			await connection.execute(`INSERT INTO ${tableItemsName} (\`id\`, \`datasetId\`, \`datasetVersion\`, \`externalId\`, \`organizationId\`, \`projectId\`, \`validTo\`, \`isDeleted\`, \`input\`, \`groundTruth\`, \`expectedTrajectory\`, \`toolMocks\`, \`requestContext\`, \`metadata\`, \`source\`, \`createdAt\`, \`updatedAt\`) VALUES (?, ?, ?, ?, ?, ?, NULL, 1, ?, ?, ?, ?, ?, ?, ?, ?, ?)`, [
+			await connection.execute(`INSERT INTO ${tableItemsName} (\`id\`, \`datasetId\`, \`datasetVersion\`, \`externalId\`, \`organizationId\`, \`projectId\`, \`validTo\`, \`isDeleted\`, \`input\`, \`groundTruth\`, \`expectedTrajectory\`, \`toolMocks\`, \`unmockedToolPolicy\`, \`requestContext\`, \`metadata\`, \`source\`, \`createdAt\`, \`updatedAt\`) VALUES (?, ?, ?, ?, ?, ?, NULL, 1, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`, [
 				id,
 				datasetId,
 				newVersion,
@@ -2436,6 +2444,7 @@ var DatasetsMySQL = class DatasetsMySQL extends DatasetsStorage {
 				jsonArg(existing.groundTruth),
 				jsonArg(existing.expectedTrajectory),
 				jsonArg(existing.toolMocks),
+				existing.unmockedToolPolicy ?? null,
 				jsonArg(existing.requestContext),
 				jsonArg(existing.metadata),
 				jsonArg(existing.source),
@@ -2674,7 +2683,7 @@ var DatasetsMySQL = class DatasetsMySQL extends DatasetsStorage {
 			await connection.execute(`UPDATE ${tableDatasetsName} SET \`version\` = ? WHERE id = ?`, [newVersion, input.datasetId]);
 			const inserted = /* @__PURE__ */ new Map();
 			for (const { id, item } of plan.inserts) {
-				await connection.execute(`INSERT INTO ${tableItemsName} (\`id\`, \`datasetId\`, \`datasetVersion\`, \`externalId\`, \`organizationId\`, \`projectId\`, \`validTo\`, \`isDeleted\`, \`input\`, \`groundTruth\`, \`expectedTrajectory\`, \`toolMocks\`, \`requestContext\`, \`metadata\`, \`source\`, \`createdAt\`, \`updatedAt\`) VALUES (?, ?, ?, ?, ?, ?, NULL, 0, ?, ?, ?, ?, ?, ?, ?, ?, ?)`, [
+				await connection.execute(`INSERT INTO ${tableItemsName} (\`id\`, \`datasetId\`, \`datasetVersion\`, \`externalId\`, \`organizationId\`, \`projectId\`, \`validTo\`, \`isDeleted\`, \`input\`, \`groundTruth\`, \`expectedTrajectory\`, \`toolMocks\`, \`unmockedToolPolicy\`, \`requestContext\`, \`metadata\`, \`source\`, \`createdAt\`, \`updatedAt\`) VALUES (?, ?, ?, ?, ?, ?, NULL, 0, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`, [
 					id,
 					input.datasetId,
 					newVersion,
@@ -2685,6 +2694,7 @@ var DatasetsMySQL = class DatasetsMySQL extends DatasetsStorage {
 					jsonArg(item.groundTruth),
 					jsonArg(item.expectedTrajectory),
 					jsonArg(item.toolMocks),
+					item.unmockedToolPolicy ?? null,
 					jsonArg(item.requestContext),
 					jsonArg(item.metadata),
 					jsonArg(item.source),
@@ -2752,7 +2762,7 @@ var DatasetsMySQL = class DatasetsMySQL extends DatasetsStorage {
 			const parentProjectId = dataset.projectId ?? null;
 			for (const item of currentItems) {
 				await connection.execute(`UPDATE ${tableItemsName} SET \`validTo\` = ? WHERE \`id\` = ? AND \`validTo\` IS NULL AND \`isDeleted\` = 0`, [newVersion, item.id]);
-				await connection.execute(`INSERT INTO ${tableItemsName} (\`id\`, \`datasetId\`, \`datasetVersion\`, \`externalId\`, \`organizationId\`, \`projectId\`, \`validTo\`, \`isDeleted\`, \`input\`, \`groundTruth\`, \`expectedTrajectory\`, \`toolMocks\`, \`requestContext\`, \`metadata\`, \`source\`, \`createdAt\`, \`updatedAt\`) VALUES (?, ?, ?, ?, ?, ?, NULL, 1, ?, ?, ?, ?, ?, ?, ?, ?, ?)`, [
+				await connection.execute(`INSERT INTO ${tableItemsName} (\`id\`, \`datasetId\`, \`datasetVersion\`, \`externalId\`, \`organizationId\`, \`projectId\`, \`validTo\`, \`isDeleted\`, \`input\`, \`groundTruth\`, \`expectedTrajectory\`, \`toolMocks\`, \`unmockedToolPolicy\`, \`requestContext\`, \`metadata\`, \`source\`, \`createdAt\`, \`updatedAt\`) VALUES (?, ?, ?, ?, ?, ?, NULL, 1, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`, [
 					item.id,
 					input.datasetId,
 					newVersion,
@@ -2763,6 +2773,7 @@ var DatasetsMySQL = class DatasetsMySQL extends DatasetsStorage {
 					jsonArg(item.groundTruth),
 					jsonArg(item.expectedTrajectory),
 					jsonArg(item.toolMocks),
+					item.unmockedToolPolicy ?? null,
 					jsonArg(item.requestContext),
 					jsonArg(item.metadata),
 					jsonArg(item.source),
@@ -2885,7 +2896,11 @@ var ExperimentsMySQL = class ExperimentsMySQL extends ExperimentsStorage {
 		await this.operations.alterTable({
 			tableName: TABLE_EXPERIMENT_RESULTS,
 			schema: EXPERIMENT_RESULTS_SCHEMA,
-			ifNotExists: ["organizationId", "projectId"]
+			ifNotExists: [
+				"comment",
+				"organizationId",
+				"projectId"
+			]
 		});
 		await this.createDefaultIndexes();
 		await this.createCustomIndexes();
@@ -2936,6 +2951,7 @@ var ExperimentsMySQL = class ExperimentsMySQL extends ExperimentsStorage {
 			traceId: row.traceId ?? null,
 			status: row.status ?? null,
 			tags: row.tags ? parseJSON$2(row.tags) ?? null : null,
+			comment: row.comment ?? null,
 			createdAt: parseDateTime(row.createdAt) ?? /* @__PURE__ */ new Date()
 		};
 	}
@@ -3267,6 +3283,7 @@ var ExperimentsMySQL = class ExperimentsMySQL extends ExperimentsStorage {
 			const updateData = {};
 			if (input.status !== void 0) updateData.status = input.status;
 			if (input.tags !== void 0) updateData.tags = input.tags ? JSON.stringify(input.tags) : null;
+			if (input.comment !== void 0) updateData.comment = input.comment;
 			if (Object.keys(updateData).length > 0) await this.operations.update({
 				tableName: TABLE_EXPERIMENT_RESULTS,
 				keys: { id: input.id },