@mastra/spanner 1.4.0 → 1.5.0-alpha.1

package/dist/index.js

@@ -1,6 +1,6 @@
 import { Spanner } from "@google-cloud/spanner";
 import { ErrorCategory, ErrorDomain, MastraError } from "@mastra/core/error";
-import { AGENTS_SCHEMA, AGENT_VERSIONS_SCHEMA, AgentsStorage, BackgroundTasksStorage, BlobStore, ChannelsStorage, DatasetsStorage, ExperimentsStorage, FavoritesStorage, MCPClientsStorage, MCPServersStorage, MCP_CLIENTS_SCHEMA, MCP_CLIENT_VERSIONS_SCHEMA, MCP_SERVERS_SCHEMA, MCP_SERVER_VERSIONS_SCHEMA, METRIC_DISTINCT_COLUMNS, MastraCompositeStore, MemoryStorage, ObservabilityStorage, PROMPT_BLOCKS_SCHEMA, PROMPT_BLOCK_VERSIONS_SCHEMA, PromptBlocksStorage, SCORER_DEFINITIONS_SCHEMA, SCORER_DEFINITION_VERSIONS_SCHEMA, SKILLS_SCHEMA, SKILL_BLOBS_SCHEMA, SKILL_VERSIONS_SCHEMA, SchedulesStorage, ScorerDefinitionsStorage, ScoresStorage, SkillsStorage, TABLE_AGENTS, TABLE_AGENT_VERSIONS, TABLE_BACKGROUND_TASKS, TABLE_CHANNEL_CONFIG, TABLE_CHANNEL_INSTALLATIONS, TABLE_CONFIGS, TABLE_DATASETS, TABLE_DATASET_ITEMS, TABLE_DATASET_VERSIONS, TABLE_EXPERIMENTS, TABLE_EXPERIMENT_RESULTS, TABLE_FAVORITES, TABLE_MCP_CLIENTS, TABLE_MCP_CLIENT_VERSIONS, TABLE_MCP_SERVERS, TABLE_MCP_SERVER_VERSIONS, TABLE_MESSAGES, TABLE_PROMPT_BLOCKS, TABLE_PROMPT_BLOCK_VERSIONS, TABLE_RESOURCES, TABLE_SCHEDULES, TABLE_SCHEDULE_TRIGGERS, TABLE_SCHEMAS, TABLE_SCORERS, TABLE_SCORER_DEFINITIONS, TABLE_SCORER_DEFINITION_VERSIONS, TABLE_SKILLS, TABLE_SKILL_BLOBS, TABLE_SKILL_VERSIONS, TABLE_SPANS, TABLE_THREADS, TABLE_WORKFLOW_SNAPSHOT, TABLE_WORKSPACES, TABLE_WORKSPACE_VERSIONS, TraceStatus, WORKSPACES_SCHEMA, WORKSPACE_VERSIONS_SCHEMA, WorkflowsStorage, WorkspacesStorage, calculatePagination, createStorageErrorId, getDefaultValue, hasErrorCode, listMetricsArgsSchema, listTracesArgsSchema, normalizePerPage, normalizeScheduleTarget, toTraceSpans, transformScoreRow, validateStorageMetadataFilter } from "@mastra/core/storage";
+import { AGENTS_SCHEMA, AGENT_VERSIONS_SCHEMA, AgentsStorage, BackgroundTasksStorage, BlobStore, ChannelsStorage, DatasetsStorage, ExperimentsStorage, FavoritesStorage, MCPClientsStorage, MCPServersStorage, MCP_CLIENTS_SCHEMA, MCP_CLIENT_VERSIONS_SCHEMA, MCP_SERVERS_SCHEMA, MCP_SERVER_VERSIONS_SCHEMA, METRIC_DISTINCT_COLUMNS, MastraCompositeStore, MemoryStorage, ObservabilityStorage, PROMPT_BLOCKS_SCHEMA, PROMPT_BLOCK_VERSIONS_SCHEMA, PromptBlocksStorage, SCORER_DEFINITIONS_SCHEMA, SCORER_DEFINITION_VERSIONS_SCHEMA, SKILLS_SCHEMA, SKILL_BLOBS_SCHEMA, SKILL_VERSIONS_SCHEMA, SchedulesStorage, ScorerDefinitionsStorage, ScoresStorage, SkillsStorage, TABLE_AGENTS, TABLE_AGENT_VERSIONS, TABLE_BACKGROUND_TASKS, TABLE_CHANNEL_CONFIG, TABLE_CHANNEL_INSTALLATIONS, TABLE_CONFIGS, TABLE_DATASETS, TABLE_DATASET_ITEMS, TABLE_DATASET_VERSIONS, TABLE_EXPERIMENTS, TABLE_EXPERIMENT_RESULTS, TABLE_FAVORITES, TABLE_MCP_CLIENTS, TABLE_MCP_CLIENT_VERSIONS, TABLE_MCP_SERVERS, TABLE_MCP_SERVER_VERSIONS, TABLE_MESSAGES, TABLE_PROMPT_BLOCKS, TABLE_PROMPT_BLOCK_VERSIONS, TABLE_RESOURCES, TABLE_SCHEDULES, TABLE_SCHEDULE_TRIGGERS, TABLE_SCHEMAS, TABLE_SCORERS, TABLE_SCORER_DEFINITIONS, TABLE_SCORER_DEFINITION_VERSIONS, TABLE_SKILLS, TABLE_SKILL_BLOBS, TABLE_SKILL_VERSIONS, TABLE_SPANS, TABLE_THREADS, TABLE_WORKFLOW_DEFINITIONS, TABLE_WORKFLOW_SNAPSHOT, TABLE_WORKSPACES, TABLE_WORKSPACE_VERSIONS, TraceStatus, WORKFLOW_DEFINITIONS_SCHEMA, WORKSPACES_SCHEMA, WORKSPACE_VERSIONS_SCHEMA, WorkflowDefinitionsStorage, WorkflowsStorage, WorkspacesStorage, calculatePagination, createStorageErrorId, getDefaultValue, hasErrorCode, listMetricsArgsSchema, listTracesArgsSchema, normalizePerPage, normalizeScheduleTarget, toTraceSpans, transformScoreRow, validateStorageMetadataFilter } from "@mastra/core/storage";
 import { MastraBase } from "@mastra/core/base";
 import { parseSqlIdentifier } from "@mastra/core/utils";
 import { randomUUID } from "crypto";
@@ -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`);
@@ -11480,6 +11500,169 @@ var SkillsSpanner = class SkillsSpanner extends SkillsStorage {
 	}
 };
 //#endregion
+//#region src/storage/domains/workflow-definitions/index.ts
+function rowToDefinition(row) {
+	const parsed = transformFromSpannerRow({
+		tableName: TABLE_WORKFLOW_DEFINITIONS,
+		row
+	});
+	if (parsed.inputSchema == null || parsed.outputSchema == null || parsed.graph == null) throw new Error(`Workflow definition row "${String(parsed.id)}" is missing required JSON columns.`);
+	const def = {
+		id: String(parsed.id),
+		inputSchema: parsed.inputSchema,
+		outputSchema: parsed.outputSchema,
+		graph: parsed.graph,
+		status: parsed.status,
+		source: parsed.source,
+		createdAt: parsed.createdAt instanceof Date ? parsed.createdAt : new Date(parsed.createdAt),
+		updatedAt: parsed.updatedAt instanceof Date ? parsed.updatedAt : new Date(parsed.updatedAt)
+	};
+	if (parsed.description != null) def.description = parsed.description;
+	if (parsed.metadata != null) def.metadata = parsed.metadata;
+	if (parsed.stateSchema != null) def.stateSchema = parsed.stateSchema;
+	if (parsed.requestContextSchema != null) def.requestContextSchema = parsed.requestContextSchema;
+	if (parsed.authorId != null) def.authorId = parsed.authorId;
+	return def;
+}
+var WorkflowDefinitionsSpanner = class WorkflowDefinitionsSpanner extends WorkflowDefinitionsStorage {
+	database;
+	db;
+	skipDefaultIndexes;
+	indexes;
+	static MANAGED_TABLES = [TABLE_WORKFLOW_DEFINITIONS];
+	constructor(config) {
+		super();
+		const { database, indexes, skipDefaultIndexes, initMode } = resolveSpannerConfig(config);
+		this.database = database;
+		this.db = new SpannerDB({
+			database,
+			skipDefaultIndexes,
+			initMode
+		});
+		this.skipDefaultIndexes = skipDefaultIndexes;
+		this.indexes = indexes?.filter((idx) => WorkflowDefinitionsSpanner.MANAGED_TABLES.includes(idx.table));
+	}
+	async init() {
+		await this.db.createTable({
+			tableName: TABLE_WORKFLOW_DEFINITIONS,
+			schema: WORKFLOW_DEFINITIONS_SCHEMA
+		});
+		await this.createDefaultIndexes();
+		await this.createCustomIndexes();
+	}
+	getDefaultIndexDefinitions() {
+		return [{
+			name: "mastra_workflow_definitions_status_idx",
+			table: TABLE_WORKFLOW_DEFINITIONS,
+			columns: ["status"]
+		}];
+	}
+	async createDefaultIndexes() {
+		if (this.skipDefaultIndexes) return;
+		await this.db.createIndexes(this.getDefaultIndexDefinitions());
+	}
+	async createCustomIndexes() {
+		if (!this.indexes || this.indexes.length === 0) return;
+		await this.db.createIndexes(this.indexes);
+	}
+	async dangerouslyClearAll() {
+		await this.db.clearTable({ tableName: TABLE_WORKFLOW_DEFINITIONS });
+	}
+	async upsert(input) {
+		const now = /* @__PURE__ */ new Date();
+		if (!await this.get(input.id)) {
+			if (!("inputSchema" in input) || input.inputSchema === void 0) throw new Error(`Cannot create workflow definition "${input.id}": inputSchema is required.`);
+			if (!("outputSchema" in input) || input.outputSchema === void 0) throw new Error(`Cannot create workflow definition "${input.id}": outputSchema is required.`);
+			if (!("graph" in input) || input.graph === void 0) throw new Error(`Cannot create workflow definition "${input.id}": graph is required.`);
+			const record = {
+				id: input.id,
+				description: input.description ?? null,
+				metadata: input.metadata ?? null,
+				inputSchema: input.inputSchema,
+				outputSchema: input.outputSchema,
+				stateSchema: input.stateSchema ?? null,
+				requestContextSchema: input.requestContextSchema ?? null,
+				graph: input.graph,
+				status: "active",
+				source: "storage",
+				authorId: "authorId" in input ? input.authorId ?? null : null,
+				createdAt: now,
+				updatedAt: now
+			};
+			try {
+				await this.db.insert({
+					tableName: TABLE_WORKFLOW_DEFINITIONS,
+					record
+				});
+			} catch (error) {
+				if (!await this.get(input.id)) throw error;
+				return this.applyUpdate(input, now);
+			}
+			const created = await this.get(input.id);
+			if (!created) throw new Error(`Failed to persist workflow definition "${input.id}".`);
+			return created;
+		}
+		return this.applyUpdate(input, now);
+	}
+	async applyUpdate(input, now) {
+		const data = { updatedAt: now };
+		if ("description" in input && input.description !== void 0) data.description = input.description;
+		if ("metadata" in input && input.metadata !== void 0) data.metadata = input.metadata;
+		if ("inputSchema" in input && input.inputSchema !== void 0) data.inputSchema = input.inputSchema;
+		if ("outputSchema" in input && input.outputSchema !== void 0) data.outputSchema = input.outputSchema;
+		if ("stateSchema" in input && input.stateSchema !== void 0) data.stateSchema = input.stateSchema;
+		if ("requestContextSchema" in input && input.requestContextSchema !== void 0) data.requestContextSchema = input.requestContextSchema;
+		if ("graph" in input && input.graph !== void 0) data.graph = input.graph;
+		if ("status" in input && input.status !== void 0) data.status = input.status;
+		if ("authorId" in input && input.authorId !== void 0) data.authorId = input.authorId;
+		await this.db.update({
+			tableName: TABLE_WORKFLOW_DEFINITIONS,
+			keys: { id: input.id },
+			data
+		});
+		const updated = await this.get(input.id);
+		if (!updated) throw new Error(`Failed to update workflow definition "${input.id}".`);
+		return updated;
+	}
+	async get(id) {
+		const row = await this.db.load({
+			tableName: TABLE_WORKFLOW_DEFINITIONS,
+			keys: { id }
+		});
+		return row ? rowToDefinition(row) : null;
+	}
+	async list(args) {
+		const params = {};
+		const conditions = [];
+		if (args?.status) {
+			params.status = args.status;
+			conditions.push(`${quoteIdent("status", "column name")} = @status`);
+		}
+		if (args?.authorId !== void 0) {
+			params.authorId = args.authorId;
+			conditions.push(`${quoteIdent("authorId", "column name")} = @authorId`);
+		}
+		const where = conditions.length ? `WHERE ${conditions.join(" AND ")}` : "";
+		const sql = `SELECT * FROM ${quoteIdent(TABLE_WORKFLOW_DEFINITIONS, "table name")} ${where} ORDER BY ${quoteIdent("updatedAt", "column name")} DESC`;
+		const [rows] = await this.database.run({
+			sql,
+			params,
+			json: true
+		});
+		const definitions = rows.map(rowToDefinition);
+		return {
+			definitions,
+			total: definitions.length
+		};
+	}
+	async delete(id) {
+		await this.db.runDml({
+			sql: `DELETE FROM ${quoteIdent(TABLE_WORKFLOW_DEFINITIONS, "table name")} WHERE id = @id`,
+			params: { id }
+		});
+	}
+};
+//#endregion
 //#region src/storage/domains/workflows/index.ts
 /**
 * Spanner-backed storage for workflow run snapshots, including persistence,
@@ -12648,7 +12831,8 @@ const SPANNER_DOMAIN_KEYS = [
 	"datasets",
 	"experiments",
 	"favorites",
-	"workspaces"
+	"workspaces",
+	"workflowDefinitions"
 ];
 const isPreConfiguredDatabase = (config) => "database" in config && !!config.database;
 /**
@@ -12736,7 +12920,8 @@ var SpannerStore = class extends MastraCompositeStore {
 				...wants("datasets") && { datasets: new DatasetsSpanner(domainConfig) },
 				...wants("experiments") && { experiments: new ExperimentsSpanner(domainConfig) },
 				...wants("favorites") && { favorites: new FavoritesSpanner(domainConfig) },
-				...wants("workspaces") && { workspaces: new WorkspacesSpanner(domainConfig) }
+				...wants("workspaces") && { workspaces: new WorkspacesSpanner(domainConfig) },
+				...wants("workflowDefinitions") && { workflowDefinitions: new WorkflowDefinitionsSpanner(domainConfig) }
 			};
 		} catch (e) {
 			throw new MastraError({
@@ -12792,7 +12977,8 @@ var SpannerStore = class extends MastraCompositeStore {
 				"datasets",
 				"experiments",
 				"workspaces",
-				"favorites"
+				"favorites",
+				"workflowDefinitions"
 			]) {
 				const store = this.stores?.[key];
 				if (store) await store.init();
@@ -12824,6 +13010,6 @@ var SpannerStore = class extends MastraCompositeStore {
 	}
 };
 //#endregion
-export { AgentsSpanner, BackgroundTasksSpanner, BlobsSpanner, ChannelsSpanner, DatasetsSpanner, ExperimentsSpanner, FavoritesSpanner, MCPClientsSpanner, MCPServersSpanner, MemorySpanner, ObservabilitySpanner, PromptBlocksSpanner, SPANNER_DOMAIN_KEYS, SchedulesSpanner, ScorerDefinitionsSpanner, ScoresSpanner, SkillsSpanner, SpannerStore, WorkflowsSpanner, WorkspacesSpanner };
+export { AgentsSpanner, BackgroundTasksSpanner, BlobsSpanner, ChannelsSpanner, DatasetsSpanner, ExperimentsSpanner, FavoritesSpanner, MCPClientsSpanner, MCPServersSpanner, MemorySpanner, ObservabilitySpanner, PromptBlocksSpanner, SPANNER_DOMAIN_KEYS, SchedulesSpanner, ScorerDefinitionsSpanner, ScoresSpanner, SkillsSpanner, SpannerStore, WorkflowDefinitionsSpanner, WorkflowsSpanner, WorkspacesSpanner };
 
 //# sourceMappingURL=index.js.map