@mastra/spanner 1.4.0 → 1.5.0-alpha.0

package/dist/index.js

@@ -2787,6 +2787,7 @@ function rowToItem(row) {
 		groundTruth: t.groundTruth ?? void 0,
 		expectedTrajectory: t.expectedTrajectory ?? void 0,
 		toolMocks: t.toolMocks ?? void 0,
+		unmockedToolPolicy: t.unmockedToolPolicy ?? void 0,
 		requestContext: t.requestContext ?? void 0,
 		metadata: t.metadata ?? void 0,
 		source: t.source ?? void 0,
@@ -2884,7 +2885,8 @@ var DatasetsSpanner = class DatasetsSpanner extends DatasetsStorage {
 			ifNotExists: [
 				"organizationId",
 				"projectId",
-				"externalId"
+				"externalId",
+				"unmockedToolPolicy"
 			]
 		});
 		await this.createDefaultIndexes();
@@ -3349,6 +3351,7 @@ var DatasetsSpanner = class DatasetsSpanner extends DatasetsStorage {
 							groundTruth: args.groundTruth ?? null,
 							expectedTrajectory: args.expectedTrajectory ?? null,
 							toolMocks: args.toolMocks ?? null,
+							unmockedToolPolicy: args.unmockedToolPolicy ?? null,
 							requestContext: args.requestContext ?? null,
 							metadata: args.metadata ?? null,
 							source: args.source ?? null,
@@ -3369,6 +3372,7 @@ var DatasetsSpanner = class DatasetsSpanner extends DatasetsStorage {
 						groundTruth: args.groundTruth,
 						expectedTrajectory: args.expectedTrajectory,
 						toolMocks: args.toolMocks,
+						unmockedToolPolicy: args.unmockedToolPolicy,
 						requestContext: args.requestContext,
 						metadata: args.metadata,
 						source: args.source,
@@ -3418,6 +3422,7 @@ var DatasetsSpanner = class DatasetsSpanner extends DatasetsStorage {
 				groundTruth: args.groundTruth !== void 0 ? args.groundTruth : existing.groundTruth,
 				expectedTrajectory: args.expectedTrajectory !== void 0 ? args.expectedTrajectory : existing.expectedTrajectory,
 				toolMocks: args.toolMocks !== void 0 ? args.toolMocks : existing.toolMocks,
+				unmockedToolPolicy: args.unmockedToolPolicy !== void 0 ? args.unmockedToolPolicy : existing.unmockedToolPolicy,
 				requestContext: args.requestContext !== void 0 ? args.requestContext : existing.requestContext,
 				metadata: args.metadata !== void 0 ? args.metadata : existing.metadata,
 				source: args.source !== void 0 ? args.source : existing.source
@@ -3443,6 +3448,7 @@ var DatasetsSpanner = class DatasetsSpanner extends DatasetsStorage {
 							groundTruth: merged.groundTruth ?? null,
 							expectedTrajectory: merged.expectedTrajectory ?? null,
 							toolMocks: merged.toolMocks ?? null,
+							unmockedToolPolicy: merged.unmockedToolPolicy ?? null,
 							requestContext: merged.requestContext ?? null,
 							metadata: merged.metadata ?? null,
 							source: merged.source ?? null,
@@ -3464,6 +3470,7 @@ var DatasetsSpanner = class DatasetsSpanner extends DatasetsStorage {
 						groundTruth: merged.groundTruth,
 						expectedTrajectory: merged.expectedTrajectory,
 						toolMocks: merged.toolMocks,
+						unmockedToolPolicy: merged.unmockedToolPolicy,
 						requestContext: merged.requestContext,
 						metadata: merged.metadata,
 						source: merged.source,
@@ -3522,6 +3529,7 @@ var DatasetsSpanner = class DatasetsSpanner extends DatasetsStorage {
 							groundTruth: existing.groundTruth ?? null,
 							expectedTrajectory: existing.expectedTrajectory ?? null,
 							toolMocks: existing.toolMocks ?? null,
+							unmockedToolPolicy: existing.unmockedToolPolicy ?? null,
 							requestContext: existing.requestContext ?? null,
 							metadata: existing.metadata ?? null,
 							source: existing.source ?? null,
@@ -3838,6 +3846,7 @@ var DatasetsSpanner = class DatasetsSpanner extends DatasetsStorage {
 								groundTruth: insert.item.groundTruth,
 								expectedTrajectory: insert.item.expectedTrajectory,
 								toolMocks: insert.item.toolMocks,
+								unmockedToolPolicy: insert.item.unmockedToolPolicy,
 								requestContext: insert.item.requestContext,
 								metadata: insert.item.metadata,
 								source: insert.item.source,
@@ -3906,6 +3915,7 @@ var DatasetsSpanner = class DatasetsSpanner extends DatasetsStorage {
 								groundTruth: existing.groundTruth ?? null,
 								expectedTrajectory: existing.expectedTrajectory ?? null,
 								toolMocks: existing.toolMocks ?? null,
+								unmockedToolPolicy: existing.unmockedToolPolicy ?? null,
 								requestContext: existing.requestContext ?? null,
 								metadata: existing.metadata ?? null,
 								source: existing.source ?? null,
@@ -3990,6 +4000,7 @@ function rowToExperimentResult(row) {
 		traceId: t.traceId ?? null,
 		status: t.status ?? null,
 		tags: t.tags ?? null,
+		comment: t.comment ?? null,
 		toolMockReport: t.toolMockReport ?? null,
 		createdAt: toDate(t.createdAt)
 	};
@@ -4034,7 +4045,11 @@ var ExperimentsSpanner = class ExperimentsSpanner extends ExperimentsStorage {
 		await this.db.alterTable({
 			tableName: TABLE_EXPERIMENT_RESULTS,
 			schema: TABLE_SCHEMAS[TABLE_EXPERIMENT_RESULTS],
-			ifNotExists: ["organizationId", "projectId"]
+			ifNotExists: [
+				"comment",
+				"organizationId",
+				"projectId"
+			]
 		});
 		await this.createDefaultIndexes();
 		await this.createCustomIndexes();
@@ -4427,7 +4442,7 @@ var ExperimentsSpanner = class ExperimentsSpanner extends ExperimentsStorage {
 	}
 	async updateExperimentResult(input) {
 		try {
-			if (input.status === void 0 && input.tags === void 0) {
+			if (input.status === void 0 && input.tags === void 0 && input.comment === void 0) {
 				const existing = await this.getExperimentResultById({ id: input.id });
 				if (!existing || input.experimentId !== void 0 && existing.experimentId !== input.experimentId) throw new MastraError({
 					id: createStorageErrorId("SPANNER", "UPDATE_EXPERIMENT_RESULT", "NOT_FOUND"),
@@ -4451,6 +4466,11 @@ var ExperimentsSpanner = class ExperimentsSpanner extends ExperimentsStorage {
 				params.tags = input.tags === null ? null : JSON.stringify(input.tags);
 				types.tags = "json";
 			}
+			if (input.comment !== void 0) {
+				setClauses.push(`${quoteIdent("comment", "column name")} = @comment`);
+				params.comment = input.comment;
+				if (input.comment === null) types.comment = "string";
+			}
 			const whereClauses = [`${quoteIdent("id", "column name")} = @id`];
 			if (input.experimentId !== void 0) {
 				whereClauses.push(`${quoteIdent("experimentId", "column name")} = @experimentId`);