@mastra/temporal 0.4.19 → 0.5.0-alpha.0

package/dist/worker.js

@@ -1,9 +1,14 @@
 import { t as toWorkflowType } from "./utils-R6cB86BM.js";
 import { builtinModules } from "module";
+import { validateStepInput, validateStepRequestContext } from "@mastra/core/workflows";
 import { readFileSync } from "fs";
 import { writeFile } from "fs/promises";
 import path, { basename, join } from "path";
 import { fileURLToPath, pathToFileURL } from "url";
+import { RequestContext } from "@mastra/core/di";
+import { MastraError } from "@mastra/core/error";
+import { toStandardSchema } from "@mastra/core/schema";
+import { ApplicationFailure } from "@temporalio/activity";
 import { generate } from "@babel/generator";
 import { parse } from "@babel/parser";
 import * as t from "@babel/types";
@@ -407,26 +412,18 @@ function createTemporalActivitiesHelperStatements(mastraImportPath, hasMastraBin
       };
     }
 
-    ${mastraImportPath ? `
-          function createStep(args) {
-            return async (params) => {
-              const { mastra } = await import(${JSON.stringify(mastraImportPath)});
-              return args.execute({ ...withRequestContext(params), mastra });
-            };
-          }
-        ` : hasMastraBinding ? `
-            function createStep(args) {
-              return async (params) => {
-                return args.execute({ ...withRequestContext(params), mastra });
-              };
-            }
-          ` : `
-            function createStep(args) {
-              return async (params) => {
-                return args.execute(withRequestContext(params));
-              };
-            }
-          `}
+    function createStep(args) {
+      return {
+        inputSchema: args.inputSchema,
+        requestContextSchema: args.requestContextSchema,
+        execute: async (params) => {
+          ${mastraImportPath ? `
+      const { mastra } = await import(${JSON.stringify(mastraImportPath)});
+      return args.execute({ ...withRequestContext(params), mastra });
+    ` : hasMastraBinding ? "return args.execute({ ...withRequestContext(params), mastra });" : "return args.execute(withRequestContext(params));"}
+        },
+      };
+    }
   `;
 	return parse(helperSource, {
 		sourceType: "module",
@@ -669,6 +666,43 @@ function getWorkflowIdMetadata(workflowConfig, workflowName, filePath) {
 	}
 	throw new Error(`Unable to determine workflow id for ${workflowName} in ${filePath}`);
 }
+function getStaticKey(property) {
+	if (property.computed && t.isTemplateLiteral(property.key) && property.key.expressions.length === 0) return property.key.quasis[0]?.value.cooked ?? null;
+	if (property.computed && t.isStringLiteral(property.key)) return property.key.value;
+	return getObjectPropertyName(property);
+}
+function isDynamicKey(property) {
+	return property.computed && !t.isStringLiteral(property.key) && !(t.isTemplateLiteral(property.key) && property.key.expressions.length === 0);
+}
+function findStaticProperty(object, name) {
+	let found = { readable: true };
+	for (const property of object.properties) {
+		if (t.isSpreadElement(property) || isDynamicKey(property)) {
+			found = { readable: false };
+			continue;
+		}
+		if (getStaticKey(property) === name) found = t.isObjectProperty(property) ? {
+			readable: true,
+			value: property.value
+		} : { readable: false };
+	}
+	return found;
+}
+/**
+* Read at build time so runs started outside `TemporalRun` (Schedules, the Temporal CLI) use the
+* workflow's setting too. `TemporalRun` also sends the setting, which covers `options` the build
+* cannot read.
+*/
+function getWorkflowValidateInputs(workflowConfig, workflowName, filePath) {
+	const options = findStaticProperty(workflowConfig, "options");
+	if (options.readable && !options.value) return;
+	if (options.readable && t.isObjectExpression(options.value)) {
+		const validateInputs = findStaticProperty(options.value, "validateInputs");
+		if (validateInputs.readable && !validateInputs.value) return;
+		if (validateInputs.readable && t.isBooleanLiteral(validateInputs.value)) return validateInputs.value.value;
+	}
+	console.warn(`[@mastra/temporal] Could not read options.validateInputs of ${workflowName} in ${filePath} at build time. Runs started with run.start() or run.startAsync() use the workflow setting. Runs started another way, such as from a Temporal Schedule, check step input.`);
+}
 /**
 * The helper runtime lives in its own `.mjs` module so it can be linted and unit-tested
 * like normal code. We parse that file directly here instead of using `Function#toString()`,
@@ -889,9 +923,11 @@ function getExportedName(node) {
 * we emit a deterministic exported function whose name matches Temporal's runtime
 * lookup and whose body delegates into the injected helper runtime.
 */
-function createTemporalWorkflowStatements(exportName, workflowId, workflowIdValue, methods, filePath, includeCommit, stepBindings, workflowBindings, runtimeOptions) {
+function createTemporalWorkflowStatements(exportName, workflowId, workflowIdValue, methods, filePath, includeCommit, stepBindings, workflowBindings, runtimeOptions, validateInputs) {
 	const createWorkflowArgs = [t.cloneNode(workflowId, true)];
-	if (runtimeOptions) createWorkflowArgs.push(t.cloneNode(runtimeOptions, true));
+	const optionProperties = runtimeOptions ? runtimeOptions.properties.map((property) => t.cloneNode(property, true)) : [];
+	if (validateInputs !== void 0) optionProperties.push(t.objectProperty(t.identifier("validateInputs"), t.booleanLiteral(validateInputs)));
+	if (optionProperties.length > 0) createWorkflowArgs.push(t.objectExpression(optionProperties));
 	let expression = t.callExpression(t.identifier("createWorkflow"), createWorkflowArgs);
 	let mappingOrdinal = 0;
 	for (const method of methods) {
@@ -1026,7 +1062,7 @@ function rewriteWorkflowVariableDeclaration(statement, filePath, state) {
 		if (!workflowExport) throw new Error(`Unable to determine workflow export for ${declaration.id.name} in ${filePath}`);
 		const { exportName } = workflowExport;
 		state.workflowExports.push(workflowExport);
-		state.statements.push(...createTemporalWorkflowStatements(exportName, workflowId, workflowIdValue, workflowChain.methods, filePath, state.committedWorkflowNames.has(declaration.id.name), state.stepBindings, state.workflowBindings, state.runtimeOptions));
+		state.statements.push(...createTemporalWorkflowStatements(exportName, workflowId, workflowIdValue, workflowChain.methods, filePath, state.committedWorkflowNames.has(declaration.id.name), state.stepBindings, state.workflowBindings, state.runtimeOptions, getWorkflowValidateInputs(workflowConfig, declaration.id.name, filePath)));
 	}
 	if (declarations.length > 0) {
 		const cloned = declarations.map((declaration) => t.cloneNode(declaration, true));
@@ -1120,6 +1156,36 @@ function getGeneratedActivitiesModulePath(outputDir) {
 function getActivityBindingsPath(outputDir) {
 	return path.join(outputDir, ACTIVITY_BINDINGS_FILE_NAME);
 }
+/**
+* Activities queued by a Workflow built before `validateInputs` existed carry no flag and run
+* unchecked. A mismatch is non-retryable because a retry gets the same input.
+*/
+async function checkStepParams(stepId, { inputSchema, requestContextSchema }, { validateInputs, ...params }) {
+	if (validateInputs !== true) return params;
+	const [input, requestContext] = await Promise.all([validateStepInput({
+		prevOutput: params.inputData,
+		step: { inputSchema: inputSchema && toStandardSchema(inputSchema) },
+		validateInputs: true
+	}), validateStepRequestContext({
+		requestContext: new RequestContext(Object.entries(params.requestContext ?? {})),
+		step: {
+			id: stepId,
+			requestContextSchema: requestContextSchema && toStandardSchema(requestContextSchema)
+		},
+		validateInputs: true
+	})]);
+	const error = input.validationError ?? requestContext.validationError;
+	if (error) throw ApplicationFailure.create({
+		type: error instanceof MastraError ? error.id : error.name,
+		message: error.message,
+		nonRetryable: true,
+		cause: error
+	});
+	return {
+		...params,
+		inputData: input.inputData
+	};
+}
 var MastraPlugin = class {
 	entryFile;
 	projectRoot;
@@ -1175,10 +1241,11 @@ var MastraPlugin = class {
 		const generatedActivities = {};
 		for (const binding of activityBindings) {
 			if (generatedActivities[binding.stepId]) continue;
-			generatedActivities[binding.stepId] = async (...args) => {
+			generatedActivities[binding.stepId] = async (params) => {
 				const activity = (await this.#loadCompiledActivitiesModule(compiledActivitiesPath))[binding.exportName];
-				if (typeof activity !== "function") throw new Error(`Unable to load activity '${binding.exportName}' from ${compiledActivitiesPath}`);
-				return activity(...args);
+				if (typeof activity === "function") return activity(params);
+				if (typeof activity?.execute !== "function") throw new Error(`Unable to load activity '${binding.exportName}' from ${compiledActivitiesPath}`);
+				return activity.execute(await checkStepParams(binding.stepId, activity, params));
 			};
 		}
 		return generatedActivities;