@mastra/pg 1.18.1 → 1.19.0-alpha.0

package/dist/index.js

@@ -5003,6 +5003,7 @@ var DatasetsPG = class DatasetsPG extends DatasetsStorage {
 		await this.#addColumnIfNotExists(TABLE_DATASET_ITEMS, "organizationId", "TEXT");
 		await this.#addColumnIfNotExists(TABLE_DATASET_ITEMS, "projectId", "TEXT");
 		await this.#addColumnIfNotExists(TABLE_DATASET_ITEMS, "toolMocks", "JSONB");
+		await this.#addColumnIfNotExists(TABLE_DATASET_ITEMS, "unmockedToolPolicy", "TEXT");
 		await this.#addColumnIfNotExists(TABLE_DATASET_ITEMS, "externalId", "TEXT");
 		await this.createDefaultIndexes();
 		await this.createCustomIndexes();
@@ -5124,6 +5125,7 @@ var DatasetsPG = class DatasetsPG extends DatasetsStorage {
 			groundTruth: row.groundTruth ? safelyParseJSON(row.groundTruth) : void 0,
 			expectedTrajectory: row.expectedTrajectory ? safelyParseJSON(row.expectedTrajectory) : void 0,
 			toolMocks: row.toolMocks ? safelyParseJSON(row.toolMocks) : void 0,
+			unmockedToolPolicy: row.unmockedToolPolicy ?? void 0,
 			requestContext: row.requestContext ? safelyParseJSON(row.requestContext) : void 0,
 			metadata: row.metadata ? safelyParseJSON(row.metadata) : void 0,
 			source: row.source ? safelyParseJSON(row.source) : void 0,
@@ -5145,6 +5147,7 @@ var DatasetsPG = class DatasetsPG extends DatasetsStorage {
 			groundTruth: row.groundTruth ? safelyParseJSON(row.groundTruth) : void 0,
 			expectedTrajectory: row.expectedTrajectory ? safelyParseJSON(row.expectedTrajectory) : void 0,
 			toolMocks: row.toolMocks ? safelyParseJSON(row.toolMocks) : void 0,
+			unmockedToolPolicy: row.unmockedToolPolicy ?? void 0,
 			requestContext: row.requestContext ? safelyParseJSON(row.requestContext) : void 0,
 			metadata: row.metadata ? safelyParseJSON(row.metadata) : void 0,
 			source: row.source ? safelyParseJSON(row.source) : void 0,
@@ -5476,7 +5479,7 @@ var DatasetsPG = class DatasetsPG extends DatasetsStorage {
 				newVersion = row.version;
 				parentOrganizationId = row.organizationId ?? null;
 				parentProjectId = row.projectId ?? null;
-				await t.none(`INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","externalId","organizationId","projectId","validTo","isDeleted","input","groundTruth","expectedTrajectory","toolMocks","requestContext","metadata","source","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,$4,$5,$6,NULL,false,$7,$8,$9,$10,$11,$12,$13,$14,$15,$16,$17)`, [
+				await t.none(`INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","externalId","organizationId","projectId","validTo","isDeleted","input","groundTruth","expectedTrajectory","toolMocks","unmockedToolPolicy","requestContext","metadata","source","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,$4,$5,$6,NULL,false,$7,$8,$9,$10,$11,$12,$13,$14,$15,$16,$17,$18)`, [
 					id,
 					args.datasetId,
 					newVersion,
@@ -5487,6 +5490,7 @@ var DatasetsPG = class DatasetsPG extends DatasetsStorage {
 					jsonbArg(args.groundTruth),
 					jsonbArg(args.expectedTrajectory),
 					jsonbArg(args.toolMocks),
+					args.unmockedToolPolicy ?? null,
 					jsonbArg(args.requestContext),
 					jsonbArg(args.metadata),
 					jsonbArg(args.source),
@@ -5513,6 +5517,7 @@ var DatasetsPG = class DatasetsPG extends DatasetsStorage {
 				groundTruth: args.groundTruth,
 				expectedTrajectory: args.expectedTrajectory,
 				toolMocks: args.toolMocks,
+				unmockedToolPolicy: args.unmockedToolPolicy,
 				requestContext: args.requestContext,
 				metadata: args.metadata,
 				source: args.source,
@@ -5566,6 +5571,7 @@ var DatasetsPG = class DatasetsPG extends DatasetsStorage {
 			const mergedGroundTruth = args.groundTruth !== void 0 ? args.groundTruth : existing.groundTruth;
 			const mergedExpectedTrajectory = args.expectedTrajectory !== void 0 ? args.expectedTrajectory : existing.expectedTrajectory;
 			const mergedToolMocks = args.toolMocks !== void 0 ? args.toolMocks : existing.toolMocks;
+			const mergedUnmockedToolPolicy = args.unmockedToolPolicy !== void 0 ? args.unmockedToolPolicy : existing.unmockedToolPolicy;
 			const mergedRequestContext = args.requestContext !== void 0 ? args.requestContext : existing.requestContext;
 			const mergedMetadata = args.metadata !== void 0 ? args.metadata : existing.metadata;
 			const mergedSource = args.source !== void 0 ? args.source : existing.source;
@@ -5578,7 +5584,7 @@ var DatasetsPG = class DatasetsPG extends DatasetsStorage {
 				parentOrganizationId = row.organizationId ?? null;
 				parentProjectId = row.projectId ?? null;
 				await t.none(`UPDATE ${itemsTable} SET "validTo" = $1 WHERE "id" = $2 AND "validTo" IS NULL AND "isDeleted" = false`, [newVersion, args.id]);
-				await t.none(`INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","externalId","organizationId","projectId","validTo","isDeleted","input","groundTruth","expectedTrajectory","toolMocks","requestContext","metadata","source","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,$4,$5,$6,NULL,false,$7,$8,$9,$10,$11,$12,$13,$14,$15,$16,$17)`, [
+				await t.none(`INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","externalId","organizationId","projectId","validTo","isDeleted","input","groundTruth","expectedTrajectory","toolMocks","unmockedToolPolicy","requestContext","metadata","source","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,$4,$5,$6,NULL,false,$7,$8,$9,$10,$11,$12,$13,$14,$15,$16,$17,$18)`, [
 					args.id,
 					args.datasetId,
 					newVersion,
@@ -5589,6 +5595,7 @@ var DatasetsPG = class DatasetsPG extends DatasetsStorage {
 					jsonbArg(mergedGroundTruth),
 					jsonbArg(mergedExpectedTrajectory),
 					jsonbArg(mergedToolMocks),
+					mergedUnmockedToolPolicy ?? null,
 					jsonbArg(mergedRequestContext),
 					jsonbArg(mergedMetadata),
 					jsonbArg(mergedSource),
@@ -5614,6 +5621,7 @@ var DatasetsPG = class DatasetsPG extends DatasetsStorage {
 				groundTruth: mergedGroundTruth,
 				expectedTrajectory: mergedExpectedTrajectory,
 				toolMocks: mergedToolMocks,
+				unmockedToolPolicy: mergedUnmockedToolPolicy,
 				requestContext: mergedRequestContext,
 				metadata: mergedMetadata,
 				source: mergedSource,
@@ -5662,7 +5670,7 @@ var DatasetsPG = class DatasetsPG extends DatasetsStorage {
 				const parentOrganizationId = row.organizationId ?? null;
 				const parentProjectId = row.projectId ?? null;
 				await t.none(`UPDATE ${itemsTable} SET "validTo" = $1 WHERE "id" = $2 AND "validTo" IS NULL AND "isDeleted" = false`, [newVersion, id]);
-				await t.none(`INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","externalId","organizationId","projectId","validTo","isDeleted","input","groundTruth","expectedTrajectory","toolMocks","requestContext","metadata","source","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,$4,$5,$6,NULL,true,$7,$8,$9,$10,$11,$12,$13,$14,$15,$16,$17)`, [
+				await t.none(`INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","externalId","organizationId","projectId","validTo","isDeleted","input","groundTruth","expectedTrajectory","toolMocks","unmockedToolPolicy","requestContext","metadata","source","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,$4,$5,$6,NULL,true,$7,$8,$9,$10,$11,$12,$13,$14,$15,$16,$17,$18)`, [
 					id,
 					datasetId,
 					newVersion,
@@ -5673,6 +5681,7 @@ var DatasetsPG = class DatasetsPG extends DatasetsStorage {
 					jsonbArg(existing.groundTruth),
 					jsonbArg(existing.expectedTrajectory),
 					jsonbArg(existing.toolMocks),
+					existing.unmockedToolPolicy ?? null,
 					jsonbArg(existing.requestContext),
 					jsonbArg(existing.metadata),
 					jsonbArg(existing.source),
@@ -5732,7 +5741,7 @@ var DatasetsPG = class DatasetsPG extends DatasetsStorage {
 					const nowIso = now.toISOString();
 					await t.none(`UPDATE ${datasetsTable} SET "version" = $2 WHERE "id" = $1`, [input.datasetId, newVersion]);
 					for (const { id, item } of plan.inserts) {
-						await t.none(`INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","externalId","organizationId","projectId","validTo","isDeleted","input","groundTruth","expectedTrajectory","toolMocks","requestContext","metadata","source","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,$4,$5,$6,NULL,false,$7,$8,$9,$10,$11,$12,$13,$14,$15,$16,$17)`, [
+						await t.none(`INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","externalId","organizationId","projectId","validTo","isDeleted","input","groundTruth","expectedTrajectory","toolMocks","unmockedToolPolicy","requestContext","metadata","source","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,$4,$5,$6,NULL,false,$7,$8,$9,$10,$11,$12,$13,$14,$15,$16,$17,$18)`, [
 							id,
 							input.datasetId,
 							newVersion,
@@ -5743,6 +5752,7 @@ var DatasetsPG = class DatasetsPG extends DatasetsStorage {
 							jsonbArg(item.groundTruth),
 							jsonbArg(item.expectedTrajectory),
 							jsonbArg(item.toolMocks),
+							item.unmockedToolPolicy ?? null,
 							jsonbArg(item.requestContext),
 							jsonbArg(item.metadata),
 							jsonbArg(item.source),
@@ -5762,6 +5772,7 @@ var DatasetsPG = class DatasetsPG extends DatasetsStorage {
 							groundTruth: item.groundTruth,
 							expectedTrajectory: item.expectedTrajectory,
 							toolMocks: item.toolMocks,
+							unmockedToolPolicy: item.unmockedToolPolicy,
 							requestContext: item.requestContext,
 							metadata: item.metadata,
 							source: item.source,
@@ -5824,7 +5835,7 @@ var DatasetsPG = class DatasetsPG extends DatasetsStorage {
 				const newVersion = (await t.one(`UPDATE ${datasetsTable} SET "version" = "version" + 1 WHERE "id" = $1 RETURNING "version"`, [input.datasetId])).version;
 				for (const item of currentItems) {
 					await t.none(`UPDATE ${itemsTable} SET "validTo" = $1 WHERE "id" = $2 AND "validTo" IS NULL AND "isDeleted" = false`, [newVersion, item.id]);
-					await t.none(`INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","externalId","organizationId","projectId","validTo","isDeleted","input","groundTruth","expectedTrajectory","toolMocks","requestContext","metadata","source","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,$4,$5,$6,NULL,true,$7,$8,$9,$10,$11,$12,$13,$14,$15,$16,$17)`, [
+					await t.none(`INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","externalId","organizationId","projectId","validTo","isDeleted","input","groundTruth","expectedTrajectory","toolMocks","unmockedToolPolicy","requestContext","metadata","source","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,$4,$5,$6,NULL,true,$7,$8,$9,$10,$11,$12,$13,$14,$15,$16,$17,$18)`, [
 						item.id,
 						input.datasetId,
 						newVersion,
@@ -5835,6 +5846,7 @@ var DatasetsPG = class DatasetsPG extends DatasetsStorage {
 						jsonbArg(item.groundTruth),
 						jsonbArg(item.expectedTrajectory),
 						jsonbArg(item.toolMocks),
+						item.unmockedToolPolicy ?? null,
 						jsonbArg(item.requestContext),
 						jsonbArg(item.metadata),
 						jsonbArg(item.source),
@@ -6129,6 +6141,7 @@ var ExperimentsPG = class ExperimentsPG extends ExperimentsStorage {
 			ifNotExists: [
 				"status",
 				"tags",
+				"comment",
 				"toolMockReport",
 				"organizationId",
 				"projectId"
@@ -6306,6 +6319,7 @@ var ExperimentsPG = class ExperimentsPG extends ExperimentsStorage {
 			traceId: row.traceId ?? null,
 			status: row.status ?? null,
 			tags: row.tags ? safelyParseJSON(row.tags) : null,
+			comment: row.comment ?? null,
 			toolMockReport: row.toolMockReport ? safelyParseJSON(row.toolMockReport) : null,
 			createdAt: ensureDate(row.createdAtZ || row.createdAt)
 		};
@@ -6631,6 +6645,10 @@ var ExperimentsPG = class ExperimentsPG extends ExperimentsStorage {
 				setClauses.push(`"tags" = $${paramIndex++}`);
 				values.push(JSON.stringify(input.tags));
 			}
+			if (input.comment !== void 0) {
+				setClauses.push(`"comment" = $${paramIndex++}`);
+				values.push(input.comment);
+			}
 			if (setClauses.length === 0) {
 				const existing = await this.getExperimentResultById({ id: input.id });
 				if (!existing) throw new MastraError({