@mastra/spanner 1.8.0-alpha.1 → 1.8.1-alpha.0

package/dist/index.js

@@ -2754,6 +2754,17 @@ var ChannelsSpanner = class ChannelsSpanner extends ChannelsStorage {
 function toDate$1(value) {
 	return value instanceof Date ? value : new Date(value);
 }
+function jsonDataArg(value) {
+	return value === void 0 ? null : JSON.stringify(value);
+}
+const ITEM_SELECT_COLUMNS = Object.keys(TABLE_SCHEMAS[TABLE_DATASET_ITEMS]).map((column) => {
+	const name = quoteIdent(column, "column name");
+	return [
+		"input",
+		"groundTruth",
+		"expectedTrajectory"
+	].includes(column) ? `CASE WHEN ${name} IS NULL THEN NULL ELSE TO_JSON_STRING(${name}) END AS ${name}` : name;
+}).join(", ");
 function rowToDataset(row) {
 	const t = transformFromSpannerRow({
 		tableName: TABLE_DATASETS,
@@ -2767,10 +2778,10 @@ function rowToDataset(row) {
 		inputSchema: t.inputSchema ?? void 0,
 		groundTruthSchema: t.groundTruthSchema ?? void 0,
 		requestContextSchema: t.requestContextSchema ?? void 0,
-		tags: t.tags ?? null,
-		targetType: t.targetType ?? null,
-		targetIds: t.targetIds ?? null,
-		scorerIds: t.scorerIds ?? null,
+		tags: t.tags ?? void 0,
+		targetType: t.targetType ?? void 0,
+		targetIds: t.targetIds ?? void 0,
+		scorerIds: t.scorerIds ?? void 0,
 		version: Number(t.version ?? 0),
 		organizationId: t.organizationId ?? null,
 		projectId: t.projectId ?? null,
@@ -2794,8 +2805,8 @@ function rowToItem(row) {
 		organizationId: t.organizationId ?? null,
 		projectId: t.projectId ?? null,
 		input: t.input,
-		groundTruth: t.groundTruth ?? emptyValue,
-		expectedTrajectory: t.expectedTrajectory ?? emptyValue,
+		groundTruth: row.groundTruth == null ? emptyValue : t.groundTruth,
+		expectedTrajectory: row.expectedTrajectory == null ? emptyValue : t.expectedTrajectory,
 		toolMocks: t.toolMocks ?? emptyValue,
 		unmockedToolPolicy: t.unmockedToolPolicy ?? emptyValue,
 		scorerIds: t.scorerIds ?? emptyValue,
@@ -2972,10 +2983,10 @@ var DatasetsSpanner = class DatasetsSpanner extends DatasetsStorage {
 				inputSchema: input.inputSchema ?? void 0,
 				groundTruthSchema: input.groundTruthSchema ?? void 0,
 				requestContextSchema: input.requestContextSchema ?? void 0,
-				tags: null,
-				targetType: input.targetType ?? null,
-				targetIds: input.targetIds ?? null,
-				scorerIds: input.scorerIds ?? null,
+				tags: void 0,
+				targetType: input.targetType ?? void 0,
+				targetIds: input.targetIds ?? void 0,
+				scorerIds: input.scorerIds ?? void 0,
 				version: 0,
 				organizationId: input.organizationId ?? null,
 				projectId: input.projectId ?? null,
@@ -3335,7 +3346,7 @@ var DatasetsSpanner = class DatasetsSpanner extends DatasetsStorage {
 	/** Reads the current live row for an item (validTo IS NULL, not deleted). */
 	async loadCurrentItemRow(tx, itemId) {
 		const [rows] = await tx.run({
-			sql: `SELECT * FROM ${quoteIdent(TABLE_DATASET_ITEMS, "table name")}
+			sql: `SELECT ${ITEM_SELECT_COLUMNS} FROM ${quoteIdent(TABLE_DATASET_ITEMS, "table name")}
             WHERE ${quoteIdent("id", "column name")} = @id
               AND ${quoteIdent("validTo", "column name")} IS NULL
               AND ${quoteIdent("isDeleted", "column name")} = FALSE LIMIT 1`,
@@ -3363,9 +3374,9 @@ var DatasetsSpanner = class DatasetsSpanner extends DatasetsStorage {
 							projectId,
 							validTo: null,
 							isDeleted: false,
-							input: args.input,
-							groundTruth: args.groundTruth ?? null,
-							expectedTrajectory: args.expectedTrajectory ?? null,
+							input: jsonDataArg(args.input),
+							groundTruth: jsonDataArg(args.groundTruth),
+							expectedTrajectory: jsonDataArg(args.expectedTrajectory),
 							toolMocks: args.toolMocks ?? null,
 							unmockedToolPolicy: args.unmockedToolPolicy ?? null,
 							scorerIds: args.scorerIds ?? null,
@@ -3473,9 +3484,9 @@ var DatasetsSpanner = class DatasetsSpanner extends DatasetsStorage {
 							projectId,
 							validTo: null,
 							isDeleted: false,
-							input: merged.input,
-							groundTruth: merged.groundTruth ?? null,
-							expectedTrajectory: merged.expectedTrajectory ?? null,
+							input: jsonDataArg(merged.input),
+							groundTruth: jsonDataArg(merged.groundTruth),
+							expectedTrajectory: jsonDataArg(merged.expectedTrajectory),
 							toolMocks: merged.toolMocks ?? null,
 							unmockedToolPolicy: merged.unmockedToolPolicy ?? null,
 							scorerIds: merged.scorerIds ?? null,
@@ -3559,9 +3570,9 @@ var DatasetsSpanner = class DatasetsSpanner extends DatasetsStorage {
 							projectId,
 							validTo: null,
 							isDeleted: true,
-							input: existing.input,
-							groundTruth: existing.groundTruth ?? null,
-							expectedTrajectory: existing.expectedTrajectory ?? null,
+							input: jsonDataArg(existing.input),
+							groundTruth: jsonDataArg(existing.groundTruth),
+							expectedTrajectory: jsonDataArg(existing.expectedTrajectory),
 							toolMocks: existing.toolMocks ?? null,
 							unmockedToolPolicy: existing.unmockedToolPolicy ?? null,
 							scorerIds: existing.scorerIds ?? null,
@@ -3728,7 +3739,7 @@ var DatasetsSpanner = class DatasetsSpanner extends DatasetsStorage {
 			};
 			const limit = perPageInput === false ? total : perPage;
 			const [rows] = await this.database.run({
-				sql: `SELECT * FROM ${tableName} ${whereSql}
+				sql: `SELECT ${ITEM_SELECT_COLUMNS} FROM ${tableName} ${whereSql}
               ORDER BY ${quoteIdent(orderBy.field, "column name")} ${orderBy.direction}, ${quoteIdent("id", "column name")} ASC
               LIMIT @limit OFFSET @offset`,
 				params: {
@@ -3762,7 +3773,7 @@ var DatasetsSpanner = class DatasetsSpanner extends DatasetsStorage {
 			let sql;
 			const params = { id: args.id };
 			if (args.datasetVersion !== void 0) {
-				sql = `SELECT * FROM ${tableName}
+				sql = `SELECT ${ITEM_SELECT_COLUMNS} FROM ${tableName}
                WHERE ${quoteIdent("id", "column name")} = @id
                  AND ${quoteIdent("datasetVersion", "column name")} <= @datasetVersion
                  AND (${quoteIdent("validTo", "column name")} IS NULL OR ${quoteIdent("validTo", "column name")} > @datasetVersion)
@@ -3770,7 +3781,7 @@ var DatasetsSpanner = class DatasetsSpanner extends DatasetsStorage {
                ORDER BY ${quoteIdent("datasetVersion", "column name")} DESC
                LIMIT 1`;
 				params.datasetVersion = args.datasetVersion;
-			} else sql = `SELECT * FROM ${tableName}
+			} else sql = `SELECT ${ITEM_SELECT_COLUMNS} FROM ${tableName}
                WHERE ${quoteIdent("id", "column name")} = @id
                  AND ${quoteIdent("validTo", "column name")} IS NULL
                  AND ${quoteIdent("isDeleted", "column name")} = FALSE LIMIT 1`;
@@ -3794,7 +3805,7 @@ var DatasetsSpanner = class DatasetsSpanner extends DatasetsStorage {
 		try {
 			const tableName = quoteIdent(TABLE_DATASET_ITEMS, "table name");
 			const [rows] = await this.database.run({
-				sql: `SELECT * FROM ${tableName}
+				sql: `SELECT ${ITEM_SELECT_COLUMNS} FROM ${tableName}
               WHERE ${quoteIdent("datasetId", "column name")} = @datasetId
                 AND ${quoteIdent("datasetVersion", "column name")} <= @version
                 AND (${quoteIdent("validTo", "column name")} IS NULL OR ${quoteIdent("validTo", "column name")} > @version)
@@ -3823,7 +3834,7 @@ var DatasetsSpanner = class DatasetsSpanner extends DatasetsStorage {
 		try {
 			const tableName = quoteIdent(TABLE_DATASET_ITEMS, "table name");
 			const [rows] = await this.database.run({
-				sql: `SELECT * FROM ${tableName}
+				sql: `SELECT ${ITEM_SELECT_COLUMNS} FROM ${tableName}
               WHERE ${quoteIdent("id", "column name")} = @id
               ORDER BY ${quoteIdent("datasetVersion", "column name")} DESC`,
 				params: { id: itemId },
@@ -3943,7 +3954,7 @@ var DatasetsSpanner = class DatasetsSpanner extends DatasetsStorage {
 					let historyRows = [];
 					if (externalIds.length > 0) {
 						const [rows] = await tx.run({
-							sql: `SELECT * FROM ${quoteIdent(TABLE_DATASET_ITEMS, "table name")} WHERE ${quoteIdent("datasetId", "column name")} = @datasetId AND ${quoteIdent("externalId", "column name")} IN UNNEST(@externalIds) ORDER BY ${quoteIdent("datasetVersion", "column name")}`,
+							sql: `SELECT ${ITEM_SELECT_COLUMNS} FROM ${quoteIdent(TABLE_DATASET_ITEMS, "table name")} WHERE ${quoteIdent("datasetId", "column name")} = @datasetId AND ${quoteIdent("externalId", "column name")} IN UNNEST(@externalIds) ORDER BY ${quoteIdent("datasetVersion", "column name")}`,
 							params: {
 								datasetId: input.datasetId,
 								externalIds
@@ -3985,6 +3996,9 @@ var DatasetsSpanner = class DatasetsSpanner extends DatasetsStorage {
 								tableName: TABLE_DATASET_ITEMS,
 								record: {
 									...item,
+									input: jsonDataArg(item.input),
+									groundTruth: jsonDataArg(item.groundTruth),
+									expectedTrajectory: jsonDataArg(item.expectedTrajectory),
 									validTo: null,
 									isDeleted: false
 								},
@@ -4042,9 +4056,9 @@ var DatasetsSpanner = class DatasetsSpanner extends DatasetsStorage {
 								projectId,
 								validTo: null,
 								isDeleted: true,
-								input: existing.input,
-								groundTruth: existing.groundTruth ?? null,
-								expectedTrajectory: existing.expectedTrajectory ?? null,
+								input: jsonDataArg(existing.input),
+								groundTruth: jsonDataArg(existing.groundTruth),
+								expectedTrajectory: jsonDataArg(existing.expectedTrajectory),
 								toolMocks: existing.toolMocks ?? null,
 								unmockedToolPolicy: existing.unmockedToolPolicy ?? null,
 								scorerIds: existing.scorerIds ?? null,