@mastra/temporal 0.4.18-alpha.4 → 0.4.18-alpha.6

package/dist/worker.js

@@ -764,7 +764,10 @@ function collectWorkflowBindings(program, filePath) {
 			const [workflowConfig] = workflowChain?.createWorkflowCall.arguments ?? [];
 			if (!workflowChain || !workflowConfig || !t.isObjectExpression(workflowConfig)) continue;
 			const { workflowId } = getWorkflowIdMetadata(workflowConfig, declaration.id.name, filePath);
-			workflowBindings.set(declaration.id.name, toWorkflowType(workflowId));
+			workflowBindings.set(declaration.id.name, {
+				workflowType: toWorkflowType(workflowId),
+				workflowId
+			});
 		}
 	}
 	return workflowBindings;
@@ -796,8 +799,8 @@ function rewriteChainMethod(method, filePath, workflowName, workflowId, mappingO
 		case "then": {
 			const arg = argNode(0);
 			if (t.isIdentifier(arg)) {
-				const workflowType = workflowBindings.get(arg.name);
-				if (workflowType) return rewritten([t.stringLiteral(workflowType)], "thenWorkflow");
+				const binding = workflowBindings.get(arg.name);
+				if (binding) return rewritten([t.stringLiteral(binding.workflowType), t.stringLiteral(binding.workflowId)], "thenWorkflow");
 			}
 			const name = getWorkflowStepName(arg, stepBindings);
 			if (!name) throw new Error(`.then() in ${workflowName} (${filePath}) must take a step or workflow identifier (inline createStep calls are not supported)`);
@@ -823,8 +826,12 @@ function rewriteChainMethod(method, filePath, workflowName, workflowId, mappingO
 			if (!t.isArrayExpression(arg)) throw new Error(`.parallel() in ${workflowName} (${filePath}) requires an array literal argument`);
 			const entries = arg.elements.map((element) => {
 				if (t.isIdentifier(element)) {
-					const workflowType = workflowBindings.get(element.name);
-					if (workflowType) return t.objectExpression([t.objectProperty(t.identifier("type"), t.stringLiteral("childWorkflow")), t.objectProperty(t.identifier("workflowType"), t.stringLiteral(workflowType))]);
+					const binding = workflowBindings.get(element.name);
+					if (binding) return t.objectExpression([
+						t.objectProperty(t.identifier("type"), t.stringLiteral("childWorkflow")),
+						t.objectProperty(t.identifier("workflowType"), t.stringLiteral(binding.workflowType)),
+						t.objectProperty(t.identifier("id"), t.stringLiteral(binding.workflowId))
+					]);
 				}
 				const stepId = getWorkflowStepName(element, stepBindings);
 				if (!stepId) throw new Error(`Unable to determine step names inside .parallel() in ${workflowName} (${filePath})`);
@@ -977,7 +984,7 @@ function rewriteWorkflowImportDeclaration(statement, state) {
 }
 function getNormalizedWorkflowBindingName(name, state) {
 	if (!state.workflowNames.has(name)) return null;
-	return state.workflowBindings.get(name) ?? name;
+	return state.workflowBindings.get(name)?.workflowType ?? name;
 }
 function rewriteWorkflowNamedExport(statement, state) {
 	if (statement.source != null) return;