@mastra/inngest 0.0.0-chore-core-swap-aiv5-ai-package-naming-20251009203931 → 0.0.0-ci-trigger-cloud-tests-on-stable-release-20260120235652

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
package/dist/index.cjs CHANGED
@@ -1,434 +1,1504 @@
1
1
  'use strict';
2
2
 
3
- var crypto = require('crypto');
4
- var realtime = require('@inngest/realtime');
5
- var aiTracing = require('@mastra/core/ai-tracing');
6
- var di = require('@mastra/core/di');
3
+ var agent = require('@mastra/core/agent');
4
+ var error = require('@mastra/core/error');
5
+ var observability = require('@mastra/core/observability');
6
+ var processors = require('@mastra/core/processors');
7
7
  var tools = require('@mastra/core/tools');
8
8
  var workflows = require('@mastra/core/workflows');
9
9
  var _constants = require('@mastra/core/workflows/_constants');
10
+ var zod = require('zod');
11
+ var crypto$1 = require('crypto');
12
+ var di = require('@mastra/core/di');
10
13
  var inngest = require('inngest');
14
+ var realtime = require('@inngest/realtime');
15
+ var events = require('@mastra/core/events');
16
+ var web = require('stream/web');
17
+ var stream = require('@mastra/core/stream');
11
18
  var hono = require('inngest/hono');
12
- var zod = require('zod');
13
19
 
14
20
  // src/index.ts
15
- function serve({
16
- mastra,
17
- inngest,
18
- functions: userFunctions = [],
19
- registerOptions
20
- }) {
21
- const wfs = mastra.getWorkflows();
22
- const workflowFunctions = Array.from(
23
- new Set(
24
- Object.values(wfs).flatMap((wf) => {
25
- if (wf instanceof InngestWorkflow) {
26
- wf.__registerMastra(mastra);
27
- return wf.getFunctions();
28
- }
29
- return [];
30
- })
31
- )
32
- );
33
- return hono.serve({
34
- ...registerOptions,
35
- client: inngest,
36
- functions: [...workflowFunctions, ...userFunctions]
37
- });
38
- }
39
- var InngestRun = class extends workflows.Run {
40
- inngest;
41
- serializedStepGraph;
42
- #mastra;
43
- constructor(params, inngest) {
44
- super(params);
45
- this.inngest = inngest;
46
- this.serializedStepGraph = params.serializedStepGraph;
47
- this.#mastra = params.mastra;
21
+ var InngestExecutionEngine = class extends workflows.DefaultExecutionEngine {
22
+ inngestStep;
23
+ inngestAttempts;
24
+ constructor(mastra, inngestStep, inngestAttempts = 0, options) {
25
+ super({ mastra, options });
26
+ this.inngestStep = inngestStep;
27
+ this.inngestAttempts = inngestAttempts;
48
28
  }
49
- async getRuns(eventId) {
50
- const response = await fetch(`${this.inngest.apiBaseUrl ?? "https://api.inngest.com"}/v1/events/${eventId}/runs`, {
51
- headers: {
52
- Authorization: `Bearer ${process.env.INNGEST_SIGNING_KEY}`
53
- }
29
+ // =============================================================================
30
+ // Hook Overrides
31
+ // =============================================================================
32
+ /**
33
+ * Format errors while preserving Error instances and their custom properties.
34
+ * Uses getErrorFromUnknown to ensure all error properties are preserved.
35
+ */
36
+ formatResultError(error$1, lastOutput) {
37
+ const outputError = lastOutput?.error;
38
+ const errorSource = error$1 || outputError;
39
+ const errorInstance = error.getErrorFromUnknown(errorSource, {
40
+ serializeStack: true,
41
+ // Include stack in JSON for better debugging in Inngest
42
+ fallbackMessage: "Unknown workflow error"
54
43
  });
55
- const json = await response.json();
56
- return json.data;
44
+ return errorInstance.toJSON();
57
45
  }
58
- async getRunOutput(eventId) {
59
- let runs = await this.getRuns(eventId);
60
- while (runs?.[0]?.status !== "Completed" || runs?.[0]?.event_id !== eventId) {
61
- await new Promise((resolve) => setTimeout(resolve, 1e3));
62
- runs = await this.getRuns(eventId);
63
- if (runs?.[0]?.status === "Failed") {
64
- const snapshot = await this.#mastra?.storage?.loadWorkflowSnapshot({
65
- workflowName: this.workflowId,
66
- runId: this.runId
67
- });
68
- return {
69
- output: { result: { steps: snapshot?.context, status: "failed", error: runs?.[0]?.output?.message } }
70
- };
46
+ /**
47
+ * Detect InngestWorkflow instances for special nested workflow handling
48
+ */
49
+ isNestedWorkflowStep(step) {
50
+ return step instanceof InngestWorkflow;
51
+ }
52
+ /**
53
+ * Inngest requires requestContext serialization for memoization.
54
+ * When steps are replayed, the original function doesn't re-execute,
55
+ * so requestContext modifications must be captured and restored.
56
+ */
57
+ requiresDurableContextSerialization() {
58
+ return true;
59
+ }
60
+ /**
61
+ * Execute a step with retry logic for Inngest.
62
+ * Retries are handled via step-level retry (RetryAfterError thrown INSIDE step.run()).
63
+ * After retries exhausted, error propagates here and we return a failed result.
64
+ */
65
+ async executeStepWithRetry(stepId, runStep, params) {
66
+ for (let i = 0; i < params.retries + 1; i++) {
67
+ if (i > 0 && params.delay) {
68
+ await new Promise((resolve) => setTimeout(resolve, params.delay));
71
69
  }
72
- if (runs?.[0]?.status === "Cancelled") {
73
- const snapshot = await this.#mastra?.storage?.loadWorkflowSnapshot({
74
- workflowName: this.workflowId,
75
- runId: this.runId
76
- });
77
- return { output: { result: { steps: snapshot?.context, status: "canceled" } } };
70
+ try {
71
+ const result = await this.wrapDurableOperation(stepId, runStep);
72
+ return { ok: true, result };
73
+ } catch (e) {
74
+ if (i === params.retries) {
75
+ const cause = e?.cause;
76
+ if (cause?.status === "failed") {
77
+ params.stepSpan?.error({
78
+ error: e,
79
+ attributes: { status: "failed" }
80
+ });
81
+ if (cause.error && !(cause.error instanceof Error)) {
82
+ cause.error = error.getErrorFromUnknown(cause.error, { serializeStack: false });
83
+ }
84
+ return { ok: false, error: cause };
85
+ }
86
+ const errorInstance = error.getErrorFromUnknown(e, {
87
+ serializeStack: false,
88
+ fallbackMessage: "Unknown step execution error"
89
+ });
90
+ params.stepSpan?.error({
91
+ error: errorInstance,
92
+ attributes: { status: "failed" }
93
+ });
94
+ return {
95
+ ok: false,
96
+ error: {
97
+ status: "failed",
98
+ error: errorInstance,
99
+ endedAt: Date.now()
100
+ }
101
+ };
102
+ }
78
103
  }
79
104
  }
80
- return runs?.[0];
105
+ return { ok: false, error: { status: "failed", error: new Error("Unknown error"), endedAt: Date.now() } };
81
106
  }
82
- async sendEvent(event, data) {
83
- await this.inngest.send({
84
- name: `user-event-${event}`,
85
- data
86
- });
107
+ /**
108
+ * Use Inngest's sleep primitive for durability
109
+ */
110
+ async executeSleepDuration(duration, sleepId, workflowId) {
111
+ await this.inngestStep.sleep(`workflow.${workflowId}.sleep.${sleepId}`, duration < 0 ? 0 : duration);
87
112
  }
88
- async cancel() {
89
- await this.inngest.send({
90
- name: `cancel.workflow.${this.workflowId}`,
91
- data: {
92
- runId: this.runId
113
+ /**
114
+ * Use Inngest's sleepUntil primitive for durability
115
+ */
116
+ async executeSleepUntilDate(date, sleepUntilId, workflowId) {
117
+ await this.inngestStep.sleepUntil(`workflow.${workflowId}.sleepUntil.${sleepUntilId}`, date);
118
+ }
119
+ /**
120
+ * Wrap durable operations in Inngest step.run() for durability.
121
+ *
122
+ * IMPORTANT: Errors are wrapped with a cause structure before throwing.
123
+ * This is necessary because Inngest's error serialization (serialize-error-cjs)
124
+ * only captures standard Error properties (message, name, stack, code, cause).
125
+ * Custom properties like statusCode, responseHeaders from AI SDK errors would
126
+ * be lost. By putting our serialized error (via getErrorFromUnknown with toJSON())
127
+ * in the cause property, we ensure custom properties survive serialization.
128
+ * The cause property is in serialize-error-cjs's allowlist, and when the cause
129
+ * object is finally JSON.stringify'd, our error's toJSON() is called.
130
+ */
131
+ async wrapDurableOperation(operationId, operationFn) {
132
+ return this.inngestStep.run(operationId, async () => {
133
+ try {
134
+ return await operationFn();
135
+ } catch (e) {
136
+ const errorInstance = error.getErrorFromUnknown(e, {
137
+ serializeStack: false,
138
+ fallbackMessage: "Unknown step execution error"
139
+ });
140
+ throw new Error(errorInstance.message, {
141
+ cause: {
142
+ status: "failed",
143
+ error: errorInstance,
144
+ endedAt: Date.now()
145
+ }
146
+ });
93
147
  }
94
148
  });
95
- const snapshot = await this.#mastra?.storage?.loadWorkflowSnapshot({
96
- workflowName: this.workflowId,
97
- runId: this.runId
98
- });
99
- if (snapshot) {
100
- await this.#mastra?.storage?.persistWorkflowSnapshot({
101
- workflowName: this.workflowId,
102
- runId: this.runId,
103
- resourceId: this.resourceId,
104
- snapshot: {
105
- ...snapshot,
106
- status: "canceled"
107
- }
149
+ }
150
+ /**
151
+ * Provide Inngest step primitive in engine context
152
+ */
153
+ getEngineContext() {
154
+ return { step: this.inngestStep };
155
+ }
156
+ /**
157
+ * For Inngest, lifecycle callbacks are invoked in the workflow's finalize step
158
+ * (wrapped in step.run for durability), not in execute(). Override to skip.
159
+ */
160
+ async invokeLifecycleCallbacks(_result) {
161
+ }
162
+ /**
163
+ * Actually invoke the lifecycle callbacks. Called from workflow.ts finalize step.
164
+ */
165
+ async invokeLifecycleCallbacksInternal(result) {
166
+ return super.invokeLifecycleCallbacks(result);
167
+ }
168
+ // =============================================================================
169
+ // Durable Span Lifecycle Hooks
170
+ // =============================================================================
171
+ /**
172
+ * Create a step span durably - on first execution, creates and exports span.
173
+ * On replay, returns cached span data without re-creating.
174
+ */
175
+ async createStepSpan(params) {
176
+ const { executionContext, operationId, options, parentSpan } = params;
177
+ const parentSpanId = parentSpan?.id ?? executionContext.tracingIds?.workflowSpanId;
178
+ const exportedSpan = await this.wrapDurableOperation(operationId, async () => {
179
+ const observability = this.mastra?.observability?.getSelectedInstance({});
180
+ if (!observability) return void 0;
181
+ const span = observability.startSpan({
182
+ ...options,
183
+ entityType: options.entityType,
184
+ traceId: executionContext.tracingIds?.traceId,
185
+ parentSpanId
108
186
  });
187
+ return span?.exportSpan();
188
+ });
189
+ if (exportedSpan) {
190
+ const observability = this.mastra?.observability?.getSelectedInstance({});
191
+ return observability?.rebuildSpan(exportedSpan);
109
192
  }
193
+ return void 0;
110
194
  }
111
- async start({
112
- inputData,
113
- initialState
114
- }) {
115
- await this.#mastra.getStorage()?.persistWorkflowSnapshot({
116
- workflowName: this.workflowId,
117
- runId: this.runId,
118
- resourceId: this.resourceId,
119
- snapshot: {
120
- runId: this.runId,
121
- serializedStepGraph: this.serializedStepGraph,
122
- value: {},
123
- context: {},
124
- activePaths: [],
125
- suspendedPaths: {},
126
- waitingPaths: {},
127
- timestamp: Date.now(),
128
- status: "running"
129
- }
195
+ /**
196
+ * End a step span durably.
197
+ */
198
+ async endStepSpan(params) {
199
+ const { span, operationId, endOptions } = params;
200
+ if (!span) return;
201
+ await this.wrapDurableOperation(operationId, async () => {
202
+ span.end(endOptions);
130
203
  });
131
- const inputDataToUse = await this._validateInput(inputData);
132
- const initialStateToUse = await this._validateInitialState(initialState ?? {});
133
- const eventOutput = await this.inngest.send({
134
- name: `workflow.${this.workflowId}`,
135
- data: {
136
- inputData: inputDataToUse,
137
- initialState: initialStateToUse,
138
- runId: this.runId,
139
- resourceId: this.resourceId
140
- }
204
+ }
205
+ /**
206
+ * Record error on step span durably.
207
+ */
208
+ async errorStepSpan(params) {
209
+ const { span, operationId, errorOptions } = params;
210
+ if (!span) return;
211
+ await this.wrapDurableOperation(operationId, async () => {
212
+ span.error(errorOptions);
141
213
  });
142
- const eventId = eventOutput.ids[0];
143
- if (!eventId) {
144
- throw new Error("Event ID is not set");
145
- }
146
- const runOutput = await this.getRunOutput(eventId);
147
- const result = runOutput?.output?.result;
148
- if (result.status === "failed") {
149
- result.error = new Error(result.error);
150
- }
151
- if (result.status !== "suspended") {
152
- this.cleanup?.();
153
- }
154
- return result;
155
214
  }
156
- async resume(params) {
157
- const p = this._resume(params).then((result) => {
158
- if (result.status !== "suspended") {
159
- this.closeStreamAction?.().catch(() => {
160
- });
161
- }
162
- return result;
215
+ /**
216
+ * Create a generic child span durably (for control-flow operations).
217
+ * On first execution, creates and exports span. On replay, returns cached span data.
218
+ */
219
+ async createChildSpan(params) {
220
+ const { executionContext, operationId, options, parentSpan } = params;
221
+ const parentSpanId = parentSpan?.id ?? executionContext.tracingIds?.workflowSpanId;
222
+ const exportedSpan = await this.wrapDurableOperation(operationId, async () => {
223
+ const observability = this.mastra?.observability?.getSelectedInstance({});
224
+ if (!observability) return void 0;
225
+ const span = observability.startSpan({
226
+ ...options,
227
+ traceId: executionContext.tracingIds?.traceId,
228
+ parentSpanId
229
+ });
230
+ return span?.exportSpan();
163
231
  });
164
- this.executionResults = p;
165
- return p;
232
+ if (exportedSpan) {
233
+ const observability = this.mastra?.observability?.getSelectedInstance({});
234
+ return observability?.rebuildSpan(exportedSpan);
235
+ }
236
+ return void 0;
166
237
  }
167
- async _resume(params) {
168
- const steps = (Array.isArray(params.step) ? params.step : [params.step]).map(
169
- (step) => typeof step === "string" ? step : step?.id
170
- );
171
- const snapshot = await this.#mastra?.storage?.loadWorkflowSnapshot({
172
- workflowName: this.workflowId,
173
- runId: this.runId
238
+ /**
239
+ * End a generic child span durably (for control-flow operations).
240
+ */
241
+ async endChildSpan(params) {
242
+ const { span, operationId, endOptions } = params;
243
+ if (!span) return;
244
+ await this.wrapDurableOperation(operationId, async () => {
245
+ span.end(endOptions);
174
246
  });
175
- const suspendedStep = this.workflowSteps[steps?.[0] ?? ""];
176
- const resumeDataToUse = await this._validateResumeData(params.resumeData, suspendedStep);
177
- const eventOutput = await this.inngest.send({
178
- name: `workflow.${this.workflowId}`,
179
- data: {
180
- inputData: resumeDataToUse,
181
- initialState: snapshot?.value ?? {},
182
- runId: this.runId,
183
- workflowId: this.workflowId,
184
- stepResults: snapshot?.context,
185
- resume: {
186
- steps,
187
- stepResults: snapshot?.context,
188
- resumePayload: resumeDataToUse,
189
- // @ts-ignore
190
- resumePath: snapshot?.suspendedPaths?.[steps?.[0]]
191
- }
192
- }
247
+ }
248
+ /**
249
+ * Record error on a generic child span durably (for control-flow operations).
250
+ */
251
+ async errorChildSpan(params) {
252
+ const { span, operationId, errorOptions } = params;
253
+ if (!span) return;
254
+ await this.wrapDurableOperation(operationId, async () => {
255
+ span.error(errorOptions);
193
256
  });
194
- const eventId = eventOutput.ids[0];
195
- if (!eventId) {
196
- throw new Error("Event ID is not set");
197
- }
198
- const runOutput = await this.getRunOutput(eventId);
199
- const result = runOutput?.output?.result;
200
- if (result.status === "failed") {
201
- result.error = new Error(result.error);
202
- }
203
- return result;
204
257
  }
205
- watch(cb, type = "watch") {
206
- let active = true;
207
- const streamPromise = realtime.subscribe(
208
- {
209
- channel: `workflow:${this.workflowId}:${this.runId}`,
210
- topics: [type],
211
- app: this.inngest
212
- },
213
- (message) => {
214
- if (active) {
215
- cb(message.data);
216
- }
258
+ /**
259
+ * Execute nested InngestWorkflow using inngestStep.invoke() for durability.
260
+ * This MUST be called directly (not inside step.run()) due to Inngest constraints.
261
+ */
262
+ async executeWorkflowStep(params) {
263
+ if (!(params.step instanceof InngestWorkflow)) {
264
+ return null;
265
+ }
266
+ const {
267
+ step,
268
+ stepResults,
269
+ executionContext,
270
+ resume,
271
+ timeTravel,
272
+ prevOutput,
273
+ inputData,
274
+ pubsub,
275
+ startedAt,
276
+ perStep,
277
+ stepSpan
278
+ } = params;
279
+ const nestedTracingContext = executionContext.tracingIds?.traceId ? {
280
+ traceId: executionContext.tracingIds.traceId,
281
+ parentSpanId: stepSpan?.id
282
+ } : void 0;
283
+ const isResume = !!resume?.steps?.length;
284
+ let result;
285
+ let runId;
286
+ const isTimeTravel = !!(timeTravel && timeTravel.steps?.length > 1 && timeTravel.steps[0] === step.id);
287
+ try {
288
+ if (isResume) {
289
+ runId = stepResults[resume?.steps?.[0] ?? ""]?.suspendPayload?.__workflow_meta?.runId ?? crypto$1.randomUUID();
290
+ const workflowsStore = await this.mastra?.getStorage()?.getStore("workflows");
291
+ const snapshot = await workflowsStore?.loadWorkflowSnapshot({
292
+ workflowName: step.id,
293
+ runId
294
+ });
295
+ const invokeResp = await this.inngestStep.invoke(`workflow.${executionContext.workflowId}.step.${step.id}`, {
296
+ function: step.getFunction(),
297
+ data: {
298
+ inputData,
299
+ initialState: executionContext.state ?? snapshot?.value ?? {},
300
+ runId,
301
+ resume: {
302
+ runId,
303
+ steps: resume.steps.slice(1),
304
+ stepResults: snapshot?.context,
305
+ resumePayload: resume.resumePayload,
306
+ resumePath: resume.steps?.[1] ? snapshot?.suspendedPaths?.[resume.steps?.[1]] : void 0
307
+ },
308
+ outputOptions: { includeState: true },
309
+ perStep,
310
+ tracingOptions: nestedTracingContext
311
+ }
312
+ });
313
+ result = invokeResp.result;
314
+ runId = invokeResp.runId;
315
+ executionContext.state = invokeResp.result.state;
316
+ } else if (isTimeTravel) {
317
+ const workflowsStoreForTimeTravel = await this.mastra?.getStorage()?.getStore("workflows");
318
+ const snapshot = await workflowsStoreForTimeTravel?.loadWorkflowSnapshot({
319
+ workflowName: step.id,
320
+ runId: executionContext.runId
321
+ }) ?? { context: {} };
322
+ const timeTravelParams = workflows.createTimeTravelExecutionParams({
323
+ steps: timeTravel.steps.slice(1),
324
+ inputData: timeTravel.inputData,
325
+ resumeData: timeTravel.resumeData,
326
+ context: timeTravel.nestedStepResults?.[step.id] ?? {},
327
+ nestedStepsContext: timeTravel.nestedStepResults ?? {},
328
+ snapshot,
329
+ graph: step.buildExecutionGraph()
330
+ });
331
+ const invokeResp = await this.inngestStep.invoke(`workflow.${executionContext.workflowId}.step.${step.id}`, {
332
+ function: step.getFunction(),
333
+ data: {
334
+ timeTravel: timeTravelParams,
335
+ initialState: executionContext.state ?? {},
336
+ runId: executionContext.runId,
337
+ outputOptions: { includeState: true },
338
+ perStep,
339
+ tracingOptions: nestedTracingContext
340
+ }
341
+ });
342
+ result = invokeResp.result;
343
+ runId = invokeResp.runId;
344
+ executionContext.state = invokeResp.result.state;
345
+ } else {
346
+ const invokeResp = await this.inngestStep.invoke(`workflow.${executionContext.workflowId}.step.${step.id}`, {
347
+ function: step.getFunction(),
348
+ data: {
349
+ inputData,
350
+ initialState: executionContext.state ?? {},
351
+ outputOptions: { includeState: true },
352
+ perStep,
353
+ tracingOptions: nestedTracingContext
354
+ }
355
+ });
356
+ result = invokeResp.result;
357
+ runId = invokeResp.runId;
358
+ executionContext.state = invokeResp.result.state;
217
359
  }
218
- );
219
- return () => {
220
- active = false;
221
- streamPromise.then(async (stream) => {
222
- return stream.cancel();
223
- }).catch((err) => {
224
- console.error(err);
225
- });
226
- };
227
- }
228
- stream({ inputData, runtimeContext } = {}) {
229
- const { readable, writable } = new TransformStream();
230
- const writer = writable.getWriter();
231
- const unwatch = this.watch(async (event) => {
232
- try {
233
- const e = {
234
- ...event,
235
- type: event.type.replace("workflow-", "")
360
+ } catch (e) {
361
+ const errorCause = e?.cause;
362
+ if (errorCause && typeof errorCause === "object") {
363
+ result = errorCause;
364
+ runId = errorCause.runId || crypto$1.randomUUID();
365
+ } else {
366
+ runId = crypto$1.randomUUID();
367
+ result = {
368
+ status: "failed",
369
+ error: e instanceof Error ? e : new Error(String(e)),
370
+ steps: {},
371
+ input: inputData
236
372
  };
237
- await writer.write(e);
238
- } catch {
239
373
  }
240
- }, "watch-v2");
241
- this.closeStreamAction = async () => {
242
- unwatch();
243
- try {
244
- await writer.close();
245
- } catch (err) {
246
- console.error("Error closing stream:", err);
247
- } finally {
248
- writer.releaseLock();
249
- }
250
- };
251
- this.executionResults = this.start({ inputData, runtimeContext }).then((result) => {
252
- if (result.status !== "suspended") {
253
- this.closeStreamAction?.().catch(() => {
374
+ }
375
+ const res = await this.inngestStep.run(
376
+ `workflow.${executionContext.workflowId}.step.${step.id}.nestedwf-results`,
377
+ async () => {
378
+ if (result.status === "failed") {
379
+ await pubsub.publish(`workflow.events.v2.${executionContext.runId}`, {
380
+ type: "watch",
381
+ runId: executionContext.runId,
382
+ data: {
383
+ type: "workflow-step-result",
384
+ payload: {
385
+ id: step.id,
386
+ status: "failed",
387
+ error: result?.error,
388
+ payload: prevOutput
389
+ }
390
+ }
391
+ });
392
+ return { executionContext, result: { status: "failed", error: result?.error, endedAt: Date.now() } };
393
+ } else if (result.status === "suspended") {
394
+ const suspendedSteps = Object.entries(result.steps).filter(([_stepName, stepResult]) => {
395
+ const stepRes = stepResult;
396
+ return stepRes?.status === "suspended";
397
+ });
398
+ for (const [stepName, stepResult] of suspendedSteps) {
399
+ const suspendPath = [stepName, ...stepResult?.suspendPayload?.__workflow_meta?.path ?? []];
400
+ executionContext.suspendedPaths[step.id] = executionContext.executionPath;
401
+ await pubsub.publish(`workflow.events.v2.${executionContext.runId}`, {
402
+ type: "watch",
403
+ runId: executionContext.runId,
404
+ data: {
405
+ type: "workflow-step-suspended",
406
+ payload: {
407
+ id: step.id,
408
+ status: "suspended"
409
+ }
410
+ }
411
+ });
412
+ return {
413
+ executionContext,
414
+ result: {
415
+ status: "suspended",
416
+ suspendedAt: Date.now(),
417
+ payload: stepResult.payload,
418
+ suspendPayload: {
419
+ ...stepResult?.suspendPayload,
420
+ __workflow_meta: { runId, path: suspendPath }
421
+ }
422
+ }
423
+ };
424
+ }
425
+ return {
426
+ executionContext,
427
+ result: {
428
+ status: "suspended",
429
+ suspendedAt: Date.now(),
430
+ payload: {}
431
+ }
432
+ };
433
+ } else if (result.status === "tripwire") {
434
+ await pubsub.publish(`workflow.events.v2.${executionContext.runId}`, {
435
+ type: "watch",
436
+ runId: executionContext.runId,
437
+ data: {
438
+ type: "workflow-step-result",
439
+ payload: {
440
+ id: step.id,
441
+ status: "tripwire",
442
+ error: result?.tripwire?.reason,
443
+ payload: prevOutput
444
+ }
445
+ }
446
+ });
447
+ return {
448
+ executionContext,
449
+ result: {
450
+ status: "tripwire",
451
+ tripwire: result?.tripwire,
452
+ endedAt: Date.now()
453
+ }
454
+ };
455
+ } else if (perStep || result.status === "paused") {
456
+ await pubsub.publish(`workflow.events.v2.${executionContext.runId}`, {
457
+ type: "watch",
458
+ runId: executionContext.runId,
459
+ data: {
460
+ type: "workflow-step-result",
461
+ payload: {
462
+ id: step.id,
463
+ status: "paused"
464
+ }
465
+ }
466
+ });
467
+ await pubsub.publish(`workflow.events.v2.${executionContext.runId}`, {
468
+ type: "watch",
469
+ runId: executionContext.runId,
470
+ data: {
471
+ type: "workflow-step-finish",
472
+ payload: {
473
+ id: step.id,
474
+ metadata: {}
475
+ }
476
+ }
477
+ });
478
+ return { executionContext, result: { status: "paused" } };
479
+ }
480
+ await pubsub.publish(`workflow.events.v2.${executionContext.runId}`, {
481
+ type: "watch",
482
+ runId: executionContext.runId,
483
+ data: {
484
+ type: "workflow-step-result",
485
+ payload: {
486
+ id: step.id,
487
+ status: "success",
488
+ output: result?.result
489
+ }
490
+ }
491
+ });
492
+ await pubsub.publish(`workflow.events.v2.${executionContext.runId}`, {
493
+ type: "watch",
494
+ runId: executionContext.runId,
495
+ data: {
496
+ type: "workflow-step-finish",
497
+ payload: {
498
+ id: step.id,
499
+ metadata: {}
500
+ }
501
+ }
254
502
  });
503
+ return { executionContext, result: { status: "success", output: result?.result, endedAt: Date.now() } };
255
504
  }
256
- return result;
257
- });
505
+ );
506
+ Object.assign(executionContext, res.executionContext);
258
507
  return {
259
- stream: readable,
260
- getWorkflowState: () => this.executionResults
508
+ ...res.result,
509
+ startedAt,
510
+ payload: inputData,
511
+ resumedAt: resume?.steps[0] === step.id ? startedAt : void 0,
512
+ resumePayload: resume?.steps[0] === step.id ? resume?.resumePayload : void 0
261
513
  };
262
514
  }
263
515
  };
264
- var InngestWorkflow = class _InngestWorkflow extends workflows.Workflow {
265
- #mastra;
516
+ var InngestPubSub = class extends events.PubSub {
266
517
  inngest;
267
- function;
268
- flowControlConfig;
269
- constructor(params, inngest) {
270
- const { concurrency, rateLimit, throttle, debounce, priority, ...workflowParams } = params;
271
- super(workflowParams);
272
- const flowControlEntries = Object.entries({ concurrency, rateLimit, throttle, debounce, priority }).filter(
273
- ([_, value]) => value !== void 0
274
- );
275
- this.flowControlConfig = flowControlEntries.length > 0 ? Object.fromEntries(flowControlEntries) : void 0;
276
- this.#mastra = params.mastra;
518
+ workflowId;
519
+ publishFn;
520
+ subscriptions = /* @__PURE__ */ new Map();
521
+ constructor(inngest, workflowId, publishFn) {
522
+ super();
277
523
  this.inngest = inngest;
524
+ this.workflowId = workflowId;
525
+ this.publishFn = publishFn;
278
526
  }
279
- async getWorkflowRuns(args) {
280
- const storage = this.#mastra?.getStorage();
281
- if (!storage) {
282
- this.logger.debug("Cannot get workflow runs. Mastra engine is not initialized");
283
- return { runs: [], total: 0 };
527
+ /**
528
+ * Publish an event to Inngest's realtime system.
529
+ *
530
+ * Topic format: "workflow.events.v2.{runId}"
531
+ * Maps to Inngest channel: "workflow:{workflowId}:{runId}"
532
+ */
533
+ async publish(topic, event) {
534
+ if (!this.publishFn) {
535
+ return;
284
536
  }
285
- return storage.getWorkflowRuns({ workflowName: this.id, ...args ?? {} });
286
- }
287
- async getWorkflowRunById(runId) {
288
- const storage = this.#mastra?.getStorage();
289
- if (!storage) {
290
- this.logger.debug("Cannot get workflow runs. Mastra engine is not initialized");
291
- return this.runs.get(runId) ? { ...this.runs.get(runId), workflowName: this.id } : null;
537
+ const match = topic.match(/^workflow\.events\.v2\.(.+)$/);
538
+ if (!match) {
539
+ return;
292
540
  }
293
- const run = await storage.getWorkflowRunById({ runId, workflowName: this.id });
294
- return run ?? (this.runs.get(runId) ? { ...this.runs.get(runId), workflowName: this.id } : null);
295
- }
296
- __registerMastra(mastra) {
297
- this.#mastra = mastra;
298
- this.executionEngine.__registerMastra(mastra);
299
- const updateNested = (step) => {
300
- if ((step.type === "step" || step.type === "loop" || step.type === "foreach") && step.step instanceof _InngestWorkflow) {
301
- step.step.__registerMastra(mastra);
302
- } else if (step.type === "parallel" || step.type === "conditional") {
303
- for (const subStep of step.steps) {
304
- updateNested(subStep);
305
- }
306
- }
307
- };
308
- if (this.executionGraph.steps.length) {
309
- for (const step of this.executionGraph.steps) {
310
- updateNested(step);
311
- }
541
+ const runId = match[1];
542
+ try {
543
+ await this.publishFn({
544
+ channel: `workflow:${this.workflowId}:${runId}`,
545
+ topic: "watch",
546
+ data: event.data
547
+ });
548
+ } catch (err) {
549
+ console.error("InngestPubSub publish error:", err?.message ?? err);
312
550
  }
313
551
  }
314
552
  /**
315
- * @deprecated Use createRunAsync() instead.
316
- * @throws {Error} Always throws an error directing users to use createRunAsync()
553
+ * Subscribe to events from Inngest's realtime system.
554
+ *
555
+ * Topic format: "workflow.events.v2.{runId}"
556
+ * Maps to Inngest channel: "workflow:{workflowId}:{runId}"
317
557
  */
318
- createRun(_options) {
319
- throw new Error(
320
- "createRun() has been deprecated. Please use createRunAsync() instead.\n\nMigration guide:\n Before: const run = workflow.createRun();\n After: const run = await workflow.createRunAsync();\n\nNote: createRunAsync() is an async method, so make sure your calling function is async."
321
- );
322
- }
323
- async createRunAsync(options) {
324
- const runIdToUse = options?.runId || crypto.randomUUID();
325
- const run = this.runs.get(runIdToUse) ?? new InngestRun(
558
+ async subscribe(topic, cb) {
559
+ const match = topic.match(/^workflow\.events\.v2\.(.+)$/);
560
+ if (!match || !match[1]) {
561
+ return;
562
+ }
563
+ const runId = match[1];
564
+ if (this.subscriptions.has(topic)) {
565
+ this.subscriptions.get(topic).callbacks.add(cb);
566
+ return;
567
+ }
568
+ const callbacks = /* @__PURE__ */ new Set([cb]);
569
+ const channel = `workflow:${this.workflowId}:${runId}`;
570
+ const streamPromise = realtime.subscribe(
326
571
  {
327
- workflowId: this.id,
328
- runId: runIdToUse,
329
- resourceId: options?.resourceId,
330
- executionEngine: this.executionEngine,
331
- executionGraph: this.executionGraph,
332
- serializedStepGraph: this.serializedStepGraph,
333
- mastra: this.#mastra,
334
- retryConfig: this.retryConfig,
335
- cleanup: () => this.runs.delete(runIdToUse),
336
- workflowSteps: this.steps
572
+ channel,
573
+ topics: ["watch"],
574
+ app: this.inngest
337
575
  },
338
- this.inngest
576
+ (message) => {
577
+ const event = {
578
+ id: crypto.randomUUID(),
579
+ type: "watch",
580
+ runId,
581
+ data: message.data,
582
+ createdAt: /* @__PURE__ */ new Date()
583
+ };
584
+ for (const callback of callbacks) {
585
+ callback(event);
586
+ }
587
+ }
339
588
  );
340
- this.runs.set(runIdToUse, run);
341
- const shouldPersistSnapshot = this.options.shouldPersistSnapshot({
342
- workflowStatus: run.workflowRunStatus,
343
- stepResults: {}
589
+ this.subscriptions.set(topic, {
590
+ unsubscribe: () => {
591
+ streamPromise.then((stream) => stream.cancel()).catch((err) => {
592
+ console.error("InngestPubSub unsubscribe error:", err);
593
+ });
594
+ },
595
+ callbacks
344
596
  });
345
- const workflowSnapshotInStorage = await this.getWorkflowRunExecutionResult(runIdToUse, false);
346
- if (!workflowSnapshotInStorage && shouldPersistSnapshot) {
347
- await this.mastra?.getStorage()?.persistWorkflowSnapshot({
348
- workflowName: this.id,
349
- runId: runIdToUse,
350
- resourceId: options?.resourceId,
351
- snapshot: {
352
- runId: runIdToUse,
353
- status: "pending",
354
- value: {},
355
- context: {},
356
- activePaths: [],
357
- waitingPaths: {},
358
- serializedStepGraph: this.serializedStepGraph,
359
- suspendedPaths: {},
360
- result: void 0,
361
- error: void 0,
362
- // @ts-ignore
363
- timestamp: Date.now()
364
- }
365
- });
597
+ }
598
+ /**
599
+ * Unsubscribe a callback from a topic.
600
+ * If no callbacks remain, the underlying Inngest subscription is cancelled.
601
+ */
602
+ async unsubscribe(topic, cb) {
603
+ const sub = this.subscriptions.get(topic);
604
+ if (!sub) {
605
+ return;
606
+ }
607
+ sub.callbacks.delete(cb);
608
+ if (sub.callbacks.size === 0) {
609
+ sub.unsubscribe();
610
+ this.subscriptions.delete(topic);
366
611
  }
367
- return run;
368
612
  }
369
- getFunction() {
370
- if (this.function) {
371
- return this.function;
613
+ /**
614
+ * Flush any pending operations. No-op for Inngest.
615
+ */
616
+ async flush() {
617
+ }
618
+ /**
619
+ * Clean up all subscriptions during graceful shutdown.
620
+ */
621
+ async close() {
622
+ for (const [, sub] of this.subscriptions) {
623
+ sub.unsubscribe();
372
624
  }
373
- this.function = this.inngest.createFunction(
374
- {
375
- id: `workflow.${this.id}`,
376
- // @ts-ignore
377
- retries: this.retryConfig?.attempts ?? 0,
378
- cancelOn: [{ event: `cancel.workflow.${this.id}` }],
379
- // Spread flow control configuration
380
- ...this.flowControlConfig
381
- },
382
- { event: `workflow.${this.id}` },
383
- async ({ event, step, attempt, publish }) => {
384
- let { inputData, initialState, runId, resourceId, resume, outputOptions } = event.data;
385
- if (!runId) {
386
- runId = await step.run(`workflow.${this.id}.runIdGen`, async () => {
387
- return crypto.randomUUID();
388
- });
389
- }
390
- const emitter = {
391
- emit: async (event2, data) => {
392
- if (!publish) {
393
- return;
625
+ this.subscriptions.clear();
626
+ }
627
+ };
628
+ var InngestRun = class extends workflows.Run {
629
+ inngest;
630
+ serializedStepGraph;
631
+ #mastra;
632
+ constructor(params, inngest) {
633
+ super(params);
634
+ this.inngest = inngest;
635
+ this.serializedStepGraph = params.serializedStepGraph;
636
+ this.#mastra = params.mastra;
637
+ }
638
+ async getRuns(eventId) {
639
+ const maxRetries = 3;
640
+ let lastError = null;
641
+ for (let attempt = 0; attempt < maxRetries; attempt++) {
642
+ try {
643
+ const response = await fetch(
644
+ `${this.inngest.apiBaseUrl ?? "https://api.inngest.com"}/v1/events/${eventId}/runs`,
645
+ {
646
+ headers: {
647
+ Authorization: `Bearer ${process.env.INNGEST_SIGNING_KEY}`
394
648
  }
395
- try {
396
- await publish({
397
- channel: `workflow:${this.id}:${runId}`,
398
- topic: event2,
399
- data
400
- });
401
- } catch (err) {
402
- this.logger.error("Error emitting event: " + (err?.stack ?? err?.message ?? err));
649
+ }
650
+ );
651
+ if (response.status === 429) {
652
+ const retryAfter = parseInt(response.headers.get("retry-after") || "2", 10);
653
+ await new Promise((resolve) => setTimeout(resolve, retryAfter * 1e3));
654
+ continue;
655
+ }
656
+ if (!response.ok) {
657
+ throw new Error(`Inngest API error: ${response.status} ${response.statusText}`);
658
+ }
659
+ const text = await response.text();
660
+ if (!text) {
661
+ await new Promise((resolve) => setTimeout(resolve, 1e3 * (attempt + 1)));
662
+ continue;
663
+ }
664
+ const json = JSON.parse(text);
665
+ return json.data;
666
+ } catch (error) {
667
+ lastError = error;
668
+ if (attempt < maxRetries - 1) {
669
+ await new Promise((resolve) => setTimeout(resolve, 1e3 * Math.pow(2, attempt)));
670
+ }
671
+ }
672
+ }
673
+ throw new inngest.NonRetriableError(`Failed to get runs after ${maxRetries} attempts: ${lastError?.message}`);
674
+ }
675
+ async getRunOutput(eventId, maxWaitMs = 3e5) {
676
+ const startTime = Date.now();
677
+ const storage = this.#mastra?.getStorage();
678
+ const workflowsStore = await storage?.getStore("workflows");
679
+ while (Date.now() - startTime < maxWaitMs) {
680
+ let runs;
681
+ try {
682
+ runs = await this.getRuns(eventId);
683
+ } catch (error) {
684
+ if (error instanceof inngest.NonRetriableError) {
685
+ throw error;
686
+ }
687
+ throw new inngest.NonRetriableError(
688
+ `Failed to poll workflow status: ${error instanceof Error ? error.message : String(error)}`
689
+ );
690
+ }
691
+ if (runs?.[0]?.status === "Completed" && runs?.[0]?.event_id === eventId) {
692
+ return runs[0];
693
+ }
694
+ if (runs?.[0]?.status === "Failed") {
695
+ const snapshot = await workflowsStore?.loadWorkflowSnapshot({
696
+ workflowName: this.workflowId,
697
+ runId: this.runId
698
+ });
699
+ if (snapshot?.context) {
700
+ snapshot.context = workflows.hydrateSerializedStepErrors(snapshot.context);
701
+ }
702
+ return {
703
+ output: {
704
+ result: {
705
+ steps: snapshot?.context,
706
+ status: "failed",
707
+ // Get the original error from NonRetriableError's cause (which contains the workflow result)
708
+ error: error.getErrorFromUnknown(runs?.[0]?.output?.cause?.error, { serializeStack: false })
403
709
  }
404
- },
405
- on: (_event, _callback) => {
406
- },
407
- off: (_event, _callback) => {
408
- },
409
- once: (_event, _callback) => {
410
710
  }
411
711
  };
412
- const engine = new InngestExecutionEngine(this.#mastra, step, attempt, this.options);
413
- const result = await engine.execute({
414
- workflowId: this.id,
415
- runId,
416
- resourceId,
417
- graph: this.executionGraph,
418
- serializedStepGraph: this.serializedStepGraph,
419
- input: inputData,
420
- initialState,
421
- emitter,
422
- retryConfig: this.retryConfig,
423
- runtimeContext: new di.RuntimeContext(),
424
- // TODO
425
- resume,
426
- abortController: new AbortController(),
427
- currentSpan: void 0,
428
- // TODO: Pass actual parent AI span from workflow execution context
429
- outputOptions
712
+ }
713
+ if (runs?.[0]?.status === "Cancelled") {
714
+ const snapshot = await workflowsStore?.loadWorkflowSnapshot({
715
+ workflowName: this.workflowId,
716
+ runId: this.runId
430
717
  });
431
- await step.run(`workflow.${this.id}.finalize`, async () => {
718
+ return { output: { result: { steps: snapshot?.context, status: "canceled" } } };
719
+ }
720
+ await new Promise((resolve) => setTimeout(resolve, 1e3 + Math.random() * 1e3));
721
+ }
722
+ throw new inngest.NonRetriableError(`Workflow did not complete within ${maxWaitMs}ms`);
723
+ }
724
+ async cancel() {
725
+ const storage = this.#mastra?.getStorage();
726
+ await this.inngest.send({
727
+ name: `cancel.workflow.${this.workflowId}`,
728
+ data: {
729
+ runId: this.runId
730
+ }
731
+ });
732
+ const workflowsStore = await storage?.getStore("workflows");
733
+ const snapshot = await workflowsStore?.loadWorkflowSnapshot({
734
+ workflowName: this.workflowId,
735
+ runId: this.runId
736
+ });
737
+ if (snapshot) {
738
+ await workflowsStore?.persistWorkflowSnapshot({
739
+ workflowName: this.workflowId,
740
+ runId: this.runId,
741
+ resourceId: this.resourceId,
742
+ snapshot: {
743
+ ...snapshot,
744
+ status: "canceled",
745
+ value: snapshot.value
746
+ }
747
+ });
748
+ }
749
+ }
750
+ async start(args) {
751
+ return this._start(args);
752
+ }
753
+ /**
754
+ * Starts the workflow execution without waiting for completion (fire-and-forget).
755
+ * Returns immediately with the runId after sending the event to Inngest.
756
+ * The workflow executes independently in Inngest.
757
+ * Use this when you don't need to wait for the result or want to avoid polling failures.
758
+ */
759
+ async startAsync(args) {
760
+ const workflowsStore = await this.#mastra.getStorage()?.getStore("workflows");
761
+ await workflowsStore?.persistWorkflowSnapshot({
762
+ workflowName: this.workflowId,
763
+ runId: this.runId,
764
+ resourceId: this.resourceId,
765
+ snapshot: {
766
+ runId: this.runId,
767
+ serializedStepGraph: this.serializedStepGraph,
768
+ status: "running",
769
+ value: {},
770
+ context: {},
771
+ activePaths: [],
772
+ suspendedPaths: {},
773
+ activeStepsPath: {},
774
+ resumeLabels: {},
775
+ waitingPaths: {},
776
+ timestamp: Date.now()
777
+ }
778
+ });
779
+ const inputDataToUse = await this._validateInput(args.inputData);
780
+ const initialStateToUse = await this._validateInitialState(args.initialState ?? {});
781
+ const eventOutput = await this.inngest.send({
782
+ name: `workflow.${this.workflowId}`,
783
+ data: {
784
+ inputData: inputDataToUse,
785
+ initialState: initialStateToUse,
786
+ runId: this.runId,
787
+ resourceId: this.resourceId,
788
+ outputOptions: args.outputOptions,
789
+ tracingOptions: args.tracingOptions,
790
+ requestContext: args.requestContext ? Object.fromEntries(args.requestContext.entries()) : {},
791
+ perStep: args.perStep
792
+ }
793
+ });
794
+ const eventId = eventOutput.ids[0];
795
+ if (!eventId) {
796
+ throw new Error("Event ID is not set");
797
+ }
798
+ return { runId: this.runId };
799
+ }
800
+ async _start({
801
+ inputData,
802
+ initialState,
803
+ outputOptions,
804
+ tracingOptions,
805
+ format,
806
+ requestContext,
807
+ perStep
808
+ }) {
809
+ const workflowsStore = await this.#mastra.getStorage()?.getStore("workflows");
810
+ await workflowsStore?.persistWorkflowSnapshot({
811
+ workflowName: this.workflowId,
812
+ runId: this.runId,
813
+ resourceId: this.resourceId,
814
+ snapshot: {
815
+ runId: this.runId,
816
+ serializedStepGraph: this.serializedStepGraph,
817
+ status: "running",
818
+ value: {},
819
+ context: {},
820
+ activePaths: [],
821
+ suspendedPaths: {},
822
+ activeStepsPath: {},
823
+ resumeLabels: {},
824
+ waitingPaths: {},
825
+ timestamp: Date.now()
826
+ }
827
+ });
828
+ const inputDataToUse = await this._validateInput(inputData);
829
+ const initialStateToUse = await this._validateInitialState(initialState ?? {});
830
+ const eventOutput = await this.inngest.send({
831
+ name: `workflow.${this.workflowId}`,
832
+ data: {
833
+ inputData: inputDataToUse,
834
+ initialState: initialStateToUse,
835
+ runId: this.runId,
836
+ resourceId: this.resourceId,
837
+ outputOptions,
838
+ tracingOptions,
839
+ format,
840
+ requestContext: requestContext ? Object.fromEntries(requestContext.entries()) : {},
841
+ perStep
842
+ }
843
+ });
844
+ const eventId = eventOutput.ids[0];
845
+ if (!eventId) {
846
+ throw new Error("Event ID is not set");
847
+ }
848
+ const runOutput = await this.getRunOutput(eventId);
849
+ const result = runOutput?.output?.result;
850
+ this.hydrateFailedResult(result);
851
+ if (result.status !== "suspended") {
852
+ this.cleanup?.();
853
+ }
854
+ return result;
855
+ }
856
+ async resume(params) {
857
+ const p = this._resume(params).then((result) => {
858
+ if (result.status !== "suspended") {
859
+ this.closeStreamAction?.().catch(() => {
860
+ });
861
+ }
862
+ return result;
863
+ });
864
+ this.executionResults = p;
865
+ return p;
866
+ }
867
+ async _resume(params) {
868
+ const storage = this.#mastra?.getStorage();
869
+ let steps = [];
870
+ if (typeof params.step === "string") {
871
+ steps = params.step.split(".");
872
+ } else {
873
+ steps = (Array.isArray(params.step) ? params.step : [params.step]).map(
874
+ (step) => typeof step === "string" ? step : step?.id
875
+ );
876
+ }
877
+ const workflowsStore = await storage?.getStore("workflows");
878
+ const snapshot = await workflowsStore?.loadWorkflowSnapshot({
879
+ workflowName: this.workflowId,
880
+ runId: this.runId
881
+ });
882
+ const suspendedStep = this.workflowSteps[steps?.[0] ?? ""];
883
+ const resumeDataToUse = await this._validateResumeData(params.resumeData, suspendedStep);
884
+ const persistedRequestContext = snapshot?.requestContext ?? {};
885
+ const newRequestContext = params.requestContext ? Object.fromEntries(params.requestContext.entries()) : {};
886
+ const mergedRequestContext = { ...persistedRequestContext, ...newRequestContext };
887
+ const eventOutput = await this.inngest.send({
888
+ name: `workflow.${this.workflowId}`,
889
+ data: {
890
+ inputData: resumeDataToUse,
891
+ initialState: snapshot?.value ?? {},
892
+ runId: this.runId,
893
+ workflowId: this.workflowId,
894
+ stepResults: snapshot?.context,
895
+ resume: {
896
+ steps,
897
+ stepResults: snapshot?.context,
898
+ resumePayload: resumeDataToUse,
899
+ resumePath: steps?.[0] ? snapshot?.suspendedPaths?.[steps?.[0]] : void 0
900
+ },
901
+ requestContext: mergedRequestContext,
902
+ perStep: params.perStep
903
+ }
904
+ });
905
+ const eventId = eventOutput.ids[0];
906
+ if (!eventId) {
907
+ throw new Error("Event ID is not set");
908
+ }
909
+ const runOutput = await this.getRunOutput(eventId);
910
+ const result = runOutput?.output?.result;
911
+ this.hydrateFailedResult(result);
912
+ return result;
913
+ }
914
+ async timeTravel(params) {
915
+ const p = this._timeTravel(params).then((result) => {
916
+ if (result.status !== "suspended") {
917
+ this.closeStreamAction?.().catch(() => {
918
+ });
919
+ }
920
+ return result;
921
+ });
922
+ this.executionResults = p;
923
+ return p;
924
+ }
925
+ async _timeTravel(params) {
926
+ if (!params.step || Array.isArray(params.step) && params.step?.length === 0) {
927
+ throw new Error("Step is required and must be a valid step or array of steps");
928
+ }
929
+ let steps = [];
930
+ if (typeof params.step === "string") {
931
+ steps = params.step.split(".");
932
+ } else {
933
+ steps = (Array.isArray(params.step) ? params.step : [params.step]).map(
934
+ (step) => typeof step === "string" ? step : step?.id
935
+ );
936
+ }
937
+ if (steps.length === 0) {
938
+ throw new Error("No steps provided to timeTravel");
939
+ }
940
+ const storage = this.#mastra?.getStorage();
941
+ const workflowsStore = await storage?.getStore("workflows");
942
+ const snapshot = await workflowsStore?.loadWorkflowSnapshot({
943
+ workflowName: this.workflowId,
944
+ runId: this.runId
945
+ });
946
+ if (!snapshot) {
947
+ await workflowsStore?.persistWorkflowSnapshot({
948
+ workflowName: this.workflowId,
949
+ runId: this.runId,
950
+ resourceId: this.resourceId,
951
+ snapshot: {
952
+ runId: this.runId,
953
+ serializedStepGraph: this.serializedStepGraph,
954
+ status: "pending",
955
+ value: {},
956
+ context: {},
957
+ activePaths: [],
958
+ suspendedPaths: {},
959
+ activeStepsPath: {},
960
+ resumeLabels: {},
961
+ waitingPaths: {},
962
+ timestamp: Date.now()
963
+ }
964
+ });
965
+ }
966
+ if (snapshot?.status === "running") {
967
+ throw new Error("This workflow run is still running, cannot time travel");
968
+ }
969
+ let inputDataToUse = params.inputData;
970
+ if (inputDataToUse && steps.length === 1) {
971
+ inputDataToUse = await this._validateTimetravelInputData(params.inputData, this.workflowSteps[steps[0]]);
972
+ }
973
+ const timeTravelData = workflows.createTimeTravelExecutionParams({
974
+ steps,
975
+ inputData: inputDataToUse,
976
+ resumeData: params.resumeData,
977
+ context: params.context,
978
+ nestedStepsContext: params.nestedStepsContext,
979
+ snapshot: snapshot ?? { context: {} },
980
+ graph: this.executionGraph,
981
+ initialState: params.initialState,
982
+ perStep: params.perStep
983
+ });
984
+ const eventOutput = await this.inngest.send({
985
+ name: `workflow.${this.workflowId}`,
986
+ data: {
987
+ initialState: timeTravelData.state,
988
+ runId: this.runId,
989
+ workflowId: this.workflowId,
990
+ stepResults: timeTravelData.stepResults,
991
+ timeTravel: timeTravelData,
992
+ tracingOptions: params.tracingOptions,
993
+ outputOptions: params.outputOptions,
994
+ requestContext: params.requestContext ? Object.fromEntries(params.requestContext.entries()) : {},
995
+ perStep: params.perStep
996
+ }
997
+ });
998
+ const eventId = eventOutput.ids[0];
999
+ if (!eventId) {
1000
+ throw new Error("Event ID is not set");
1001
+ }
1002
+ const runOutput = await this.getRunOutput(eventId);
1003
+ const result = runOutput?.output?.result;
1004
+ this.hydrateFailedResult(result);
1005
+ return result;
1006
+ }
1007
+ watch(cb) {
1008
+ let active = true;
1009
+ const streamPromise = realtime.subscribe(
1010
+ {
1011
+ channel: `workflow:${this.workflowId}:${this.runId}`,
1012
+ topics: ["watch"],
1013
+ app: this.inngest
1014
+ },
1015
+ (message) => {
1016
+ if (active) {
1017
+ cb(message.data);
1018
+ }
1019
+ }
1020
+ );
1021
+ return () => {
1022
+ active = false;
1023
+ streamPromise.then(async (stream) => {
1024
+ return stream.cancel();
1025
+ }).catch((err) => {
1026
+ console.error(err);
1027
+ });
1028
+ };
1029
+ }
1030
+ streamLegacy({ inputData, requestContext } = {}) {
1031
+ const { readable, writable } = new TransformStream();
1032
+ const writer = writable.getWriter();
1033
+ void writer.write({
1034
+ // @ts-ignore
1035
+ type: "start",
1036
+ // @ts-ignore
1037
+ payload: { runId: this.runId }
1038
+ });
1039
+ const unwatch = this.watch(async (event) => {
1040
+ try {
1041
+ const e = {
1042
+ ...event,
1043
+ type: event.type.replace("workflow-", "")
1044
+ };
1045
+ if (e.type === "step-output") {
1046
+ e.type = e.payload.output.type;
1047
+ e.payload = e.payload.output.payload;
1048
+ }
1049
+ await writer.write(e);
1050
+ } catch {
1051
+ }
1052
+ });
1053
+ this.closeStreamAction = async () => {
1054
+ await writer.write({
1055
+ type: "finish",
1056
+ // @ts-ignore
1057
+ payload: { runId: this.runId }
1058
+ });
1059
+ unwatch();
1060
+ try {
1061
+ await writer.close();
1062
+ } catch (err) {
1063
+ console.error("Error closing stream:", err);
1064
+ } finally {
1065
+ writer.releaseLock();
1066
+ }
1067
+ };
1068
+ this.executionResults = this._start({ inputData, requestContext, format: "legacy" }).then((result) => {
1069
+ if (result.status !== "suspended") {
1070
+ this.closeStreamAction?.().catch(() => {
1071
+ });
1072
+ }
1073
+ return result;
1074
+ });
1075
+ return {
1076
+ stream: readable,
1077
+ getWorkflowState: () => this.executionResults
1078
+ };
1079
+ }
1080
+ stream({
1081
+ inputData,
1082
+ requestContext,
1083
+ tracingOptions,
1084
+ closeOnSuspend = true,
1085
+ initialState,
1086
+ outputOptions,
1087
+ perStep
1088
+ } = {}) {
1089
+ if (this.closeStreamAction && this.streamOutput) {
1090
+ return this.streamOutput;
1091
+ }
1092
+ this.closeStreamAction = async () => {
1093
+ };
1094
+ const self = this;
1095
+ const stream$1 = new web.ReadableStream({
1096
+ async start(controller) {
1097
+ const unwatch = self.watch(async ({ type, from = stream.ChunkFrom.WORKFLOW, payload }) => {
1098
+ controller.enqueue({
1099
+ type,
1100
+ runId: self.runId,
1101
+ from,
1102
+ payload: {
1103
+ stepName: payload?.id,
1104
+ ...payload
1105
+ }
1106
+ });
1107
+ });
1108
+ self.closeStreamAction = async () => {
1109
+ unwatch();
1110
+ try {
1111
+ await controller.close();
1112
+ } catch (err) {
1113
+ console.error("Error closing stream:", err);
1114
+ }
1115
+ };
1116
+ const executionResultsPromise = self._start({
1117
+ inputData,
1118
+ requestContext,
1119
+ // tracingContext, // We are not able to pass a reference to a span here, what to do?
1120
+ initialState,
1121
+ tracingOptions,
1122
+ outputOptions,
1123
+ format: "vnext",
1124
+ perStep
1125
+ });
1126
+ let executionResults;
1127
+ try {
1128
+ executionResults = await executionResultsPromise;
1129
+ if (closeOnSuspend) {
1130
+ self.closeStreamAction?.().catch(() => {
1131
+ });
1132
+ } else if (executionResults.status !== "suspended") {
1133
+ self.closeStreamAction?.().catch(() => {
1134
+ });
1135
+ }
1136
+ if (self.streamOutput) {
1137
+ self.streamOutput.updateResults(
1138
+ executionResults
1139
+ );
1140
+ }
1141
+ } catch (err) {
1142
+ self.streamOutput?.rejectResults(err);
1143
+ self.closeStreamAction?.().catch(() => {
1144
+ });
1145
+ }
1146
+ }
1147
+ });
1148
+ this.streamOutput = new stream.WorkflowRunOutput({
1149
+ runId: this.runId,
1150
+ workflowId: this.workflowId,
1151
+ stream: stream$1
1152
+ });
1153
+ return this.streamOutput;
1154
+ }
1155
+ timeTravelStream({
1156
+ inputData,
1157
+ resumeData,
1158
+ initialState,
1159
+ step,
1160
+ context,
1161
+ nestedStepsContext,
1162
+ requestContext,
1163
+ // tracingContext,
1164
+ tracingOptions,
1165
+ outputOptions,
1166
+ perStep
1167
+ }) {
1168
+ this.closeStreamAction = async () => {
1169
+ };
1170
+ const self = this;
1171
+ const stream$1 = new web.ReadableStream({
1172
+ async start(controller) {
1173
+ const unwatch = self.watch(async ({ type, from = stream.ChunkFrom.WORKFLOW, payload }) => {
1174
+ controller.enqueue({
1175
+ type,
1176
+ runId: self.runId,
1177
+ from,
1178
+ payload: {
1179
+ stepName: payload?.id,
1180
+ ...payload
1181
+ }
1182
+ });
1183
+ });
1184
+ self.closeStreamAction = async () => {
1185
+ unwatch();
1186
+ try {
1187
+ controller.close();
1188
+ } catch (err) {
1189
+ console.error("Error closing stream:", err);
1190
+ }
1191
+ };
1192
+ const executionResultsPromise = self._timeTravel({
1193
+ inputData,
1194
+ step,
1195
+ context,
1196
+ nestedStepsContext,
1197
+ resumeData,
1198
+ initialState,
1199
+ requestContext,
1200
+ tracingOptions,
1201
+ outputOptions,
1202
+ perStep
1203
+ });
1204
+ self.executionResults = executionResultsPromise;
1205
+ let executionResults;
1206
+ try {
1207
+ executionResults = await executionResultsPromise;
1208
+ self.closeStreamAction?.().catch(() => {
1209
+ });
1210
+ if (self.streamOutput) {
1211
+ self.streamOutput.updateResults(executionResults);
1212
+ }
1213
+ } catch (err) {
1214
+ self.streamOutput?.rejectResults(err);
1215
+ self.closeStreamAction?.().catch(() => {
1216
+ });
1217
+ }
1218
+ }
1219
+ });
1220
+ this.streamOutput = new stream.WorkflowRunOutput({
1221
+ runId: this.runId,
1222
+ workflowId: this.workflowId,
1223
+ stream: stream$1
1224
+ });
1225
+ return this.streamOutput;
1226
+ }
1227
+ /**
1228
+ * Hydrates errors in a failed workflow result back to proper Error instances.
1229
+ * This ensures error.cause chains and custom properties are preserved.
1230
+ */
1231
+ hydrateFailedResult(result) {
1232
+ if (result.status === "failed") {
1233
+ result.error = error.getErrorFromUnknown(result.error, { serializeStack: false });
1234
+ if (result.steps) {
1235
+ workflows.hydrateSerializedStepErrors(result.steps);
1236
+ }
1237
+ }
1238
+ }
1239
+ };
1240
+
1241
+ // src/workflow.ts
1242
+ var InngestWorkflow = class _InngestWorkflow extends workflows.Workflow {
1243
+ #mastra;
1244
+ inngest;
1245
+ function;
1246
+ cronFunction;
1247
+ flowControlConfig;
1248
+ cronConfig;
1249
+ constructor(params, inngest) {
1250
+ const { concurrency, rateLimit, throttle, debounce, priority, cron, inputData, initialState, ...workflowParams } = params;
1251
+ super(workflowParams);
1252
+ this.engineType = "inngest";
1253
+ const flowControlEntries = Object.entries({ concurrency, rateLimit, throttle, debounce, priority }).filter(
1254
+ ([_, value]) => value !== void 0
1255
+ );
1256
+ this.flowControlConfig = flowControlEntries.length > 0 ? Object.fromEntries(flowControlEntries) : void 0;
1257
+ this.#mastra = params.mastra;
1258
+ this.inngest = inngest;
1259
+ if (cron) {
1260
+ this.cronConfig = { cron, inputData, initialState };
1261
+ }
1262
+ }
1263
+ async listWorkflowRuns(args) {
1264
+ const storage = this.#mastra?.getStorage();
1265
+ if (!storage) {
1266
+ this.logger.debug("Cannot get workflow runs. Mastra engine is not initialized");
1267
+ return { runs: [], total: 0 };
1268
+ }
1269
+ const workflowsStore = await storage.getStore("workflows");
1270
+ if (!workflowsStore) {
1271
+ return { runs: [], total: 0 };
1272
+ }
1273
+ return workflowsStore.listWorkflowRuns({ workflowName: this.id, ...args ?? {} });
1274
+ }
1275
+ __registerMastra(mastra) {
1276
+ super.__registerMastra(mastra);
1277
+ this.#mastra = mastra;
1278
+ this.executionEngine.__registerMastra(mastra);
1279
+ const updateNested = (step) => {
1280
+ if ((step.type === "step" || step.type === "loop" || step.type === "foreach") && step.step instanceof _InngestWorkflow) {
1281
+ step.step.__registerMastra(mastra);
1282
+ } else if (step.type === "parallel" || step.type === "conditional") {
1283
+ for (const subStep of step.steps) {
1284
+ updateNested(subStep);
1285
+ }
1286
+ }
1287
+ };
1288
+ if (this.executionGraph.steps.length) {
1289
+ for (const step of this.executionGraph.steps) {
1290
+ updateNested(step);
1291
+ }
1292
+ }
1293
+ }
1294
+ async createRun(options) {
1295
+ const runIdToUse = options?.runId || crypto$1.randomUUID();
1296
+ const existingInMemoryRun = this.runs.get(runIdToUse);
1297
+ const newRun = new InngestRun(
1298
+ {
1299
+ workflowId: this.id,
1300
+ runId: runIdToUse,
1301
+ resourceId: options?.resourceId,
1302
+ executionEngine: this.executionEngine,
1303
+ executionGraph: this.executionGraph,
1304
+ serializedStepGraph: this.serializedStepGraph,
1305
+ mastra: this.#mastra,
1306
+ retryConfig: this.retryConfig,
1307
+ cleanup: () => this.runs.delete(runIdToUse),
1308
+ workflowSteps: this.steps,
1309
+ workflowEngineType: this.engineType,
1310
+ validateInputs: this.options.validateInputs
1311
+ },
1312
+ this.inngest
1313
+ );
1314
+ const run = existingInMemoryRun ?? newRun;
1315
+ this.runs.set(runIdToUse, run);
1316
+ const shouldPersistSnapshot = this.options.shouldPersistSnapshot({
1317
+ workflowStatus: run.workflowRunStatus,
1318
+ stepResults: {}
1319
+ });
1320
+ const existingStoredRun = await this.getWorkflowRunById(runIdToUse, {
1321
+ withNestedWorkflows: false
1322
+ });
1323
+ const existsInStorage = existingStoredRun && !existingStoredRun.isFromInMemory;
1324
+ if (!existsInStorage && shouldPersistSnapshot) {
1325
+ const workflowsStore = await this.mastra?.getStorage()?.getStore("workflows");
1326
+ await workflowsStore?.persistWorkflowSnapshot({
1327
+ workflowName: this.id,
1328
+ runId: runIdToUse,
1329
+ resourceId: options?.resourceId,
1330
+ snapshot: {
1331
+ runId: runIdToUse,
1332
+ status: "pending",
1333
+ value: {},
1334
+ context: {},
1335
+ activePaths: [],
1336
+ activeStepsPath: {},
1337
+ waitingPaths: {},
1338
+ serializedStepGraph: this.serializedStepGraph,
1339
+ suspendedPaths: {},
1340
+ resumeLabels: {},
1341
+ result: void 0,
1342
+ error: void 0,
1343
+ timestamp: Date.now()
1344
+ }
1345
+ });
1346
+ }
1347
+ return run;
1348
+ }
1349
+ //createCronFunction is only called if cronConfig.cron is defined.
1350
+ createCronFunction() {
1351
+ if (this.cronFunction) {
1352
+ return this.cronFunction;
1353
+ }
1354
+ this.cronFunction = this.inngest.createFunction(
1355
+ {
1356
+ id: `workflow.${this.id}.cron`,
1357
+ retries: 0,
1358
+ cancelOn: [{ event: `cancel.workflow.${this.id}` }],
1359
+ ...this.flowControlConfig
1360
+ },
1361
+ { cron: this.cronConfig?.cron ?? "" },
1362
+ async () => {
1363
+ const run = await this.createRun();
1364
+ const result = await run.start({
1365
+ inputData: this.cronConfig?.inputData,
1366
+ initialState: this.cronConfig?.initialState
1367
+ });
1368
+ return { result, runId: run.runId };
1369
+ }
1370
+ );
1371
+ return this.cronFunction;
1372
+ }
1373
+ getFunction() {
1374
+ if (this.function) {
1375
+ return this.function;
1376
+ }
1377
+ this.function = this.inngest.createFunction(
1378
+ {
1379
+ id: `workflow.${this.id}`,
1380
+ retries: 0,
1381
+ cancelOn: [{ event: `cancel.workflow.${this.id}` }],
1382
+ // Spread flow control configuration
1383
+ ...this.flowControlConfig
1384
+ },
1385
+ { event: `workflow.${this.id}` },
1386
+ async ({ event, step, attempt, publish }) => {
1387
+ let {
1388
+ inputData,
1389
+ initialState,
1390
+ runId,
1391
+ resourceId,
1392
+ resume,
1393
+ outputOptions,
1394
+ format,
1395
+ timeTravel,
1396
+ perStep,
1397
+ tracingOptions
1398
+ } = event.data;
1399
+ if (!runId) {
1400
+ runId = await step.run(`workflow.${this.id}.runIdGen`, async () => {
1401
+ return crypto$1.randomUUID();
1402
+ });
1403
+ }
1404
+ const pubsub = new InngestPubSub(this.inngest, this.id, publish);
1405
+ const requestContext = new di.RequestContext(Object.entries(event.data.requestContext ?? {}));
1406
+ const mastra = this.#mastra;
1407
+ const tracingPolicy = this.options.tracingPolicy;
1408
+ const workflowSpanData = await step.run(`workflow.${this.id}.span.start`, async () => {
1409
+ const observability$1 = mastra?.observability?.getSelectedInstance({ requestContext });
1410
+ if (!observability$1) return void 0;
1411
+ const span = observability$1.startSpan({
1412
+ type: observability.SpanType.WORKFLOW_RUN,
1413
+ name: `workflow run: '${this.id}'`,
1414
+ entityType: observability.EntityType.WORKFLOW_RUN,
1415
+ entityId: this.id,
1416
+ input: inputData,
1417
+ metadata: {
1418
+ resourceId,
1419
+ runId
1420
+ },
1421
+ tracingPolicy,
1422
+ tracingOptions,
1423
+ requestContext
1424
+ });
1425
+ return span?.exportSpan();
1426
+ });
1427
+ const engine = new InngestExecutionEngine(this.#mastra, step, attempt, this.options);
1428
+ let result;
1429
+ try {
1430
+ result = await engine.execute({
1431
+ workflowId: this.id,
1432
+ runId,
1433
+ resourceId,
1434
+ graph: this.executionGraph,
1435
+ serializedStepGraph: this.serializedStepGraph,
1436
+ input: inputData,
1437
+ initialState,
1438
+ pubsub,
1439
+ retryConfig: this.retryConfig,
1440
+ requestContext,
1441
+ resume,
1442
+ timeTravel,
1443
+ perStep,
1444
+ format,
1445
+ abortController: new AbortController(),
1446
+ // For Inngest, we don't pass workflowSpan - step spans use tracingIds instead
1447
+ workflowSpan: void 0,
1448
+ // Pass tracing IDs for durable span operations
1449
+ tracingIds: workflowSpanData ? {
1450
+ traceId: workflowSpanData.traceId,
1451
+ workflowSpanId: workflowSpanData.id
1452
+ } : void 0,
1453
+ outputOptions,
1454
+ outputWriter: async (chunk) => {
1455
+ try {
1456
+ await pubsub.publish(`workflow.events.v2.${runId}`, {
1457
+ type: "watch",
1458
+ runId,
1459
+ data: chunk
1460
+ });
1461
+ } catch (err) {
1462
+ this.logger.debug?.("Failed to publish watch event:", err);
1463
+ }
1464
+ }
1465
+ });
1466
+ } catch (error) {
1467
+ throw error;
1468
+ }
1469
+ await step.run(`workflow.${this.id}.finalize`, async () => {
1470
+ if (result.status !== "paused") {
1471
+ await engine.invokeLifecycleCallbacksInternal({
1472
+ status: result.status,
1473
+ result: "result" in result ? result.result : void 0,
1474
+ error: "error" in result ? result.error : void 0,
1475
+ steps: result.steps,
1476
+ tripwire: "tripwire" in result ? result.tripwire : void 0,
1477
+ runId,
1478
+ workflowId: this.id,
1479
+ resourceId,
1480
+ input: inputData,
1481
+ requestContext,
1482
+ state: result.state ?? initialState ?? {}
1483
+ });
1484
+ }
1485
+ if (workflowSpanData) {
1486
+ const observability = mastra?.observability?.getSelectedInstance({ requestContext });
1487
+ if (observability) {
1488
+ const workflowSpan = observability.rebuildSpan(workflowSpanData);
1489
+ if (result.status === "failed") {
1490
+ workflowSpan.error({
1491
+ error: result.error instanceof Error ? result.error : new Error(String(result.error)),
1492
+ attributes: { status: "failed" }
1493
+ });
1494
+ } else {
1495
+ workflowSpan.end({
1496
+ output: result.status === "success" ? result.result : void 0,
1497
+ attributes: { status: result.status }
1498
+ });
1499
+ }
1500
+ }
1501
+ }
432
1502
  if (result.status === "failed") {
433
1503
  throw new inngest.NonRetriableError(`Workflow failed`, {
434
1504
  cause: result
@@ -455,1084 +1525,731 @@ var InngestWorkflow = class _InngestWorkflow extends workflows.Workflow {
455
1525
  });
456
1526
  }
457
1527
  getFunctions() {
458
- return [this.getFunction(), ...this.getNestedFunctions(this.executionGraph.steps)];
1528
+ return [
1529
+ this.getFunction(),
1530
+ ...this.cronConfig?.cron ? [this.createCronFunction()] : [],
1531
+ ...this.getNestedFunctions(this.executionGraph.steps)
1532
+ ];
459
1533
  }
460
1534
  };
461
- function isAgent(params) {
462
- return params?.component === "AGENT";
1535
+ function prepareServeOptions({ mastra, inngest, functions: userFunctions = [], registerOptions }) {
1536
+ const wfs = mastra.listWorkflows();
1537
+ const workflowFunctions = Array.from(
1538
+ new Set(
1539
+ Object.values(wfs).flatMap((wf) => {
1540
+ if (wf instanceof InngestWorkflow) {
1541
+ wf.__registerMastra(mastra);
1542
+ return wf.getFunctions();
1543
+ }
1544
+ return [];
1545
+ })
1546
+ )
1547
+ );
1548
+ return {
1549
+ ...registerOptions,
1550
+ client: inngest,
1551
+ functions: [...workflowFunctions, ...userFunctions]
1552
+ };
1553
+ }
1554
+ function createServe(adapter) {
1555
+ return (options) => {
1556
+ const serveOptions = prepareServeOptions(options);
1557
+ return adapter(serveOptions);
1558
+ };
1559
+ }
1560
+ var serve = createServe(hono.serve);
1561
+
1562
+ // src/types.ts
1563
+ var _compatibilityCheck = true;
1564
+
1565
+ // src/index.ts
1566
+ function isInngestWorkflow(input) {
1567
+ return input instanceof InngestWorkflow;
1568
+ }
1569
+ function isAgent(input) {
1570
+ return input instanceof agent.Agent;
1571
+ }
1572
+ function isToolStep(input) {
1573
+ return input instanceof tools.Tool;
1574
+ }
1575
+ function isStepParams(input) {
1576
+ return input !== null && typeof input === "object" && "id" in input && "execute" in input && !(input instanceof agent.Agent) && !(input instanceof tools.Tool) && !(input instanceof InngestWorkflow);
463
1577
  }
464
- function isTool(params) {
465
- return params instanceof tools.Tool;
1578
+ function isProcessor(obj) {
1579
+ return obj !== null && typeof obj === "object" && "id" in obj && typeof obj.id === "string" && !(obj instanceof agent.Agent) && !(obj instanceof tools.Tool) && !(obj instanceof InngestWorkflow) && (typeof obj.processInput === "function" || typeof obj.processInputStep === "function" || typeof obj.processOutputStream === "function" || typeof obj.processOutputResult === "function" || typeof obj.processOutputStep === "function");
466
1580
  }
467
- function createStep(params) {
1581
+ function createStep(params, agentOrToolOptions) {
1582
+ if (isInngestWorkflow(params)) {
1583
+ return params;
1584
+ }
468
1585
  if (isAgent(params)) {
469
- return {
470
- id: params.name,
471
- description: params.getDescription(),
472
- // @ts-ignore
473
- inputSchema: zod.z.object({
474
- prompt: zod.z.string()
475
- }),
476
- // @ts-ignore
477
- outputSchema: zod.z.object({
478
- text: zod.z.string()
479
- }),
480
- execute: async ({ inputData, [_constants.EMITTER_SYMBOL]: emitter, runtimeContext, abortSignal, abort, tracingContext }) => {
481
- let streamPromise = {};
482
- streamPromise.promise = new Promise((resolve, reject) => {
483
- streamPromise.resolve = resolve;
484
- streamPromise.reject = reject;
485
- });
486
- const toolData = {
487
- name: params.name,
488
- args: inputData
489
- };
490
- if ((await params.getLLM()).getModel().specificationVersion === `v2`) {
491
- const { fullStream } = await params.stream(inputData.prompt, {
492
- runtimeContext,
493
- tracingContext,
494
- onFinish: (result) => {
495
- streamPromise.resolve(result.text);
496
- },
497
- abortSignal
498
- });
499
- if (abortSignal.aborted) {
500
- return abort();
501
- }
502
- await emitter.emit("watch-v2", {
503
- type: "tool-call-streaming-start",
504
- ...toolData ?? {}
505
- });
506
- for await (const chunk of fullStream) {
507
- if (chunk.type === "text-delta") {
508
- await emitter.emit("watch-v2", {
509
- type: "tool-call-delta",
510
- ...toolData ?? {},
511
- argsTextDelta: chunk.payload.text
512
- });
513
- }
514
- }
515
- } else {
516
- const { fullStream } = await params.streamLegacy(inputData.prompt, {
517
- runtimeContext,
518
- tracingContext,
519
- onFinish: (result) => {
520
- streamPromise.resolve(result.text);
521
- },
522
- abortSignal
523
- });
524
- if (abortSignal.aborted) {
525
- return abort();
526
- }
527
- await emitter.emit("watch-v2", {
528
- type: "tool-call-streaming-start",
529
- ...toolData ?? {}
530
- });
531
- for await (const chunk of fullStream) {
532
- if (chunk.type === "text-delta") {
533
- await emitter.emit("watch-v2", {
534
- type: "tool-call-delta",
535
- ...toolData ?? {},
536
- argsTextDelta: chunk.textDelta
537
- });
538
- }
539
- }
540
- }
541
- await emitter.emit("watch-v2", {
542
- type: "tool-call-streaming-finish",
543
- ...toolData ?? {}
544
- });
545
- return {
546
- text: await streamPromise.promise
547
- };
548
- },
549
- component: params.component
550
- };
1586
+ return createStepFromAgent(params, agentOrToolOptions);
551
1587
  }
552
- if (isTool(params)) {
553
- if (!params.inputSchema || !params.outputSchema) {
554
- throw new Error("Tool must have input and output schemas defined");
555
- }
556
- return {
557
- // TODO: tool probably should have strong id type
558
- // @ts-ignore
559
- id: params.id,
560
- description: params.description,
561
- inputSchema: params.inputSchema,
562
- outputSchema: params.outputSchema,
563
- execute: async ({ inputData, mastra, runtimeContext, tracingContext, suspend, resumeData }) => {
564
- return params.execute({
565
- context: inputData,
566
- mastra: aiTracing.wrapMastra(mastra, tracingContext),
567
- runtimeContext,
568
- tracingContext,
569
- suspend,
570
- resumeData
571
- });
572
- },
573
- component: "TOOL"
574
- };
1588
+ if (isToolStep(params)) {
1589
+ return createStepFromTool(params, agentOrToolOptions);
1590
+ }
1591
+ if (isStepParams(params)) {
1592
+ return createStepFromParams(params);
1593
+ }
1594
+ if (isProcessor(params)) {
1595
+ return createStepFromProcessor(params);
575
1596
  }
1597
+ throw new Error("Invalid input: expected StepParams, Agent, ToolStep, Processor, or InngestWorkflow");
1598
+ }
1599
+ function createStepFromParams(params) {
576
1600
  return {
577
1601
  id: params.id,
578
1602
  description: params.description,
579
1603
  inputSchema: params.inputSchema,
1604
+ stateSchema: params.stateSchema,
580
1605
  outputSchema: params.outputSchema,
581
1606
  resumeSchema: params.resumeSchema,
582
1607
  suspendSchema: params.suspendSchema,
583
- execute: params.execute
1608
+ scorers: params.scorers,
1609
+ retries: params.retries,
1610
+ execute: params.execute.bind(params)
584
1611
  };
585
1612
  }
586
- function init(inngest) {
1613
+ function createStepFromAgent(params, agentOrToolOptions) {
1614
+ const options = agentOrToolOptions ?? {};
1615
+ const outputSchema = options?.structuredOutput?.schema ?? zod.z.object({ text: zod.z.string() });
1616
+ const { retries, scorers, ...agentOptions } = options ?? {};
587
1617
  return {
588
- createWorkflow(params) {
589
- return new InngestWorkflow(
590
- params,
591
- inngest
592
- );
593
- },
594
- createStep,
595
- cloneStep(step, opts) {
596
- return {
597
- id: opts.id,
598
- description: step.description,
599
- inputSchema: step.inputSchema,
600
- outputSchema: step.outputSchema,
601
- resumeSchema: step.resumeSchema,
602
- suspendSchema: step.suspendSchema,
603
- stateSchema: step.stateSchema,
604
- execute: step.execute,
605
- component: step.component
606
- };
607
- },
608
- cloneWorkflow(workflow, opts) {
609
- const wf = new workflows.Workflow({
610
- id: opts.id,
611
- inputSchema: workflow.inputSchema,
612
- outputSchema: workflow.outputSchema,
613
- steps: workflow.stepDefs,
614
- mastra: workflow.mastra
1618
+ id: params.name,
1619
+ description: params.getDescription(),
1620
+ inputSchema: zod.z.object({
1621
+ prompt: zod.z.string()
1622
+ }),
1623
+ outputSchema,
1624
+ retries,
1625
+ scorers,
1626
+ execute: async ({
1627
+ inputData,
1628
+ runId,
1629
+ [_constants.PUBSUB_SYMBOL]: pubsub,
1630
+ [_constants.STREAM_FORMAT_SYMBOL]: streamFormat,
1631
+ requestContext,
1632
+ tracingContext,
1633
+ abortSignal,
1634
+ abort,
1635
+ writer
1636
+ }) => {
1637
+ let streamPromise = {};
1638
+ streamPromise.promise = new Promise((resolve, reject) => {
1639
+ streamPromise.resolve = resolve;
1640
+ streamPromise.reject = reject;
615
1641
  });
616
- wf.setStepFlow(workflow.stepGraph);
617
- wf.commit();
618
- return wf;
619
- }
620
- };
621
- }
622
- var InngestExecutionEngine = class extends workflows.DefaultExecutionEngine {
623
- inngestStep;
624
- inngestAttempts;
625
- constructor(mastra, inngestStep, inngestAttempts = 0, options) {
626
- super({ mastra, options });
627
- this.inngestStep = inngestStep;
628
- this.inngestAttempts = inngestAttempts;
629
- }
630
- async execute(params) {
631
- await params.emitter.emit("watch-v2", {
632
- type: "workflow-start",
633
- payload: { runId: params.runId }
634
- });
635
- const result = await super.execute(params);
636
- await params.emitter.emit("watch-v2", {
637
- type: "workflow-finish",
638
- payload: { runId: params.runId }
639
- });
640
- return result;
641
- }
642
- async fmtReturnValue(executionSpan, emitter, stepResults, lastOutput, error) {
643
- const base = {
644
- status: lastOutput.status,
645
- steps: stepResults
646
- };
647
- if (lastOutput.status === "success") {
648
- await emitter.emit("watch", {
649
- type: "watch",
650
- payload: {
651
- workflowState: {
652
- status: lastOutput.status,
653
- steps: stepResults,
654
- result: lastOutput.output
655
- }
656
- },
657
- eventTimestamp: Date.now()
658
- });
659
- base.result = lastOutput.output;
660
- } else if (lastOutput.status === "failed") {
661
- base.error = error instanceof Error ? error?.stack ?? error.message : lastOutput?.error instanceof Error ? lastOutput.error.message : lastOutput.error ?? error ?? "Unknown error";
662
- await emitter.emit("watch", {
663
- type: "watch",
664
- payload: {
665
- workflowState: {
666
- status: lastOutput.status,
667
- steps: stepResults,
668
- result: null,
669
- error: base.error
670
- }
671
- },
672
- eventTimestamp: Date.now()
673
- });
674
- } else if (lastOutput.status === "suspended") {
675
- await emitter.emit("watch", {
676
- type: "watch",
677
- payload: {
678
- workflowState: {
679
- status: lastOutput.status,
680
- steps: stepResults,
681
- result: null,
682
- error: null
683
- }
684
- },
685
- eventTimestamp: Date.now()
686
- });
687
- const suspendedStepIds = Object.entries(stepResults).flatMap(([stepId, stepResult]) => {
688
- if (stepResult?.status === "suspended") {
689
- const nestedPath = stepResult?.suspendPayload?.__workflow_meta?.path;
690
- return nestedPath ? [[stepId, ...nestedPath]] : [[stepId]];
691
- }
692
- return [];
693
- });
694
- base.suspended = suspendedStepIds;
695
- }
696
- executionSpan?.end();
697
- return base;
698
- }
699
- // async executeSleep({ id, duration }: { id: string; duration: number }): Promise<void> {
700
- // await this.inngestStep.sleep(id, duration);
701
- // }
702
- async executeSleep({
703
- workflowId,
704
- runId,
705
- entry,
706
- prevOutput,
707
- stepResults,
708
- emitter,
709
- abortController,
710
- runtimeContext,
711
- executionContext,
712
- writableStream,
713
- tracingContext
714
- }) {
715
- let { duration, fn } = entry;
716
- const sleepSpan = tracingContext?.currentSpan?.createChildSpan({
717
- type: aiTracing.AISpanType.WORKFLOW_SLEEP,
718
- name: `sleep: ${duration ? `${duration}ms` : "dynamic"}`,
719
- attributes: {
720
- durationMs: duration,
721
- sleepType: fn ? "dynamic" : "fixed"
722
- },
723
- tracingPolicy: this.options?.tracingPolicy
724
- });
725
- if (fn) {
726
- const stepCallId = crypto.randomUUID();
727
- duration = await this.inngestStep.run(`workflow.${workflowId}.sleep.${entry.id}`, async () => {
728
- return await fn({
729
- runId,
730
- workflowId,
731
- mastra: this.mastra,
732
- runtimeContext,
733
- inputData: prevOutput,
734
- state: executionContext.state,
735
- setState: (state) => {
736
- executionContext.state = state;
737
- },
738
- runCount: -1,
739
- tracingContext: {
740
- currentSpan: sleepSpan
741
- },
742
- getInitData: () => stepResults?.input,
743
- getStepResult: workflows.getStepResult.bind(this, stepResults),
744
- // TODO: this function shouldn't have suspend probably?
745
- suspend: async (_suspendPayload) => {
746
- },
747
- bail: () => {
1642
+ let structuredResult = null;
1643
+ const toolData = {
1644
+ name: params.name,
1645
+ args: inputData
1646
+ };
1647
+ let stream;
1648
+ if ((await params.getModel()).specificationVersion === "v1") {
1649
+ const { fullStream } = await params.streamLegacy(inputData.prompt, {
1650
+ ...agentOptions ?? {},
1651
+ requestContext,
1652
+ tracingContext,
1653
+ onFinish: (result) => {
1654
+ const resultWithObject = result;
1655
+ if (agentOptions?.structuredOutput?.schema && resultWithObject.object) {
1656
+ structuredResult = resultWithObject.object;
1657
+ }
1658
+ streamPromise.resolve(result.text);
1659
+ void agentOptions?.onFinish?.(result);
748
1660
  },
749
- abort: () => {
750
- abortController?.abort();
1661
+ abortSignal
1662
+ });
1663
+ stream = fullStream;
1664
+ } else {
1665
+ const modelOutput = await params.stream(inputData.prompt, {
1666
+ ...agentOptions ?? {},
1667
+ requestContext,
1668
+ tracingContext,
1669
+ onFinish: (result) => {
1670
+ const resultWithObject = result;
1671
+ if (agentOptions?.structuredOutput?.schema && resultWithObject.object) {
1672
+ structuredResult = resultWithObject.object;
1673
+ }
1674
+ streamPromise.resolve(result.text);
1675
+ void agentOptions?.onFinish?.(result);
751
1676
  },
752
- [_constants.EMITTER_SYMBOL]: emitter,
753
- // TODO: add streamVNext support
754
- [_constants.STREAM_FORMAT_SYMBOL]: executionContext.format,
755
- engine: { step: this.inngestStep },
756
- abortSignal: abortController?.signal,
757
- writer: new tools.ToolStream(
758
- {
759
- prefix: "workflow-step",
760
- callId: stepCallId,
761
- name: "sleep",
762
- runId
763
- },
764
- writableStream
765
- )
1677
+ abortSignal
766
1678
  });
767
- });
768
- sleepSpan?.update({
769
- attributes: {
770
- durationMs: duration
1679
+ stream = modelOutput.fullStream;
1680
+ }
1681
+ if (streamFormat === "legacy") {
1682
+ await pubsub.publish(`workflow.events.v2.${runId}`, {
1683
+ type: "watch",
1684
+ runId,
1685
+ data: { type: "tool-call-streaming-start", ...toolData ?? {} }
1686
+ });
1687
+ for await (const chunk of stream) {
1688
+ if (chunk.type === "text-delta") {
1689
+ await pubsub.publish(`workflow.events.v2.${runId}`, {
1690
+ type: "watch",
1691
+ runId,
1692
+ data: { type: "tool-call-delta", ...toolData ?? {}, argsTextDelta: chunk.textDelta }
1693
+ });
1694
+ }
771
1695
  }
772
- });
773
- }
774
- try {
775
- await this.inngestStep.sleep(entry.id, !duration || duration < 0 ? 0 : duration);
776
- sleepSpan?.end();
777
- } catch (e) {
778
- sleepSpan?.error({ error: e });
779
- throw e;
780
- }
781
- }
782
- async executeSleepUntil({
783
- workflowId,
784
- runId,
785
- entry,
786
- prevOutput,
787
- stepResults,
788
- emitter,
789
- abortController,
790
- runtimeContext,
791
- executionContext,
792
- writableStream,
793
- tracingContext
794
- }) {
795
- let { date, fn } = entry;
796
- const sleepUntilSpan = tracingContext?.currentSpan?.createChildSpan({
797
- type: aiTracing.AISpanType.WORKFLOW_SLEEP,
798
- name: `sleepUntil: ${date ? date.toISOString() : "dynamic"}`,
799
- attributes: {
800
- untilDate: date,
801
- durationMs: date ? Math.max(0, date.getTime() - Date.now()) : void 0,
802
- sleepType: fn ? "dynamic" : "fixed"
803
- },
804
- tracingPolicy: this.options?.tracingPolicy
805
- });
806
- if (fn) {
807
- date = await this.inngestStep.run(`workflow.${workflowId}.sleepUntil.${entry.id}`, async () => {
808
- const stepCallId = crypto.randomUUID();
809
- return await fn({
1696
+ await pubsub.publish(`workflow.events.v2.${runId}`, {
1697
+ type: "watch",
810
1698
  runId,
811
- workflowId,
812
- mastra: this.mastra,
813
- runtimeContext,
814
- inputData: prevOutput,
815
- state: executionContext.state,
816
- setState: (state) => {
817
- executionContext.state = state;
818
- },
819
- runCount: -1,
820
- tracingContext: {
821
- currentSpan: sleepUntilSpan
822
- },
823
- getInitData: () => stepResults?.input,
824
- getStepResult: workflows.getStepResult.bind(this, stepResults),
825
- // TODO: this function shouldn't have suspend probably?
826
- suspend: async (_suspendPayload) => {
827
- },
828
- bail: () => {
829
- },
830
- abort: () => {
831
- abortController?.abort();
832
- },
833
- [_constants.EMITTER_SYMBOL]: emitter,
834
- [_constants.STREAM_FORMAT_SYMBOL]: executionContext.format,
835
- // TODO: add streamVNext support
836
- engine: { step: this.inngestStep },
837
- abortSignal: abortController?.signal,
838
- writer: new tools.ToolStream(
839
- {
840
- prefix: "workflow-step",
841
- callId: stepCallId,
842
- name: "sleep",
843
- runId
844
- },
845
- writableStream
846
- )
1699
+ data: { type: "tool-call-streaming-finish", ...toolData ?? {} }
847
1700
  });
848
- });
849
- if (date && !(date instanceof Date)) {
850
- date = new Date(date);
1701
+ } else {
1702
+ for await (const chunk of stream) {
1703
+ await writer.write(chunk);
1704
+ }
851
1705
  }
852
- const time = !date ? 0 : date.getTime() - Date.now();
853
- sleepUntilSpan?.update({
854
- attributes: {
855
- durationMs: Math.max(0, time)
1706
+ if (abortSignal.aborted) {
1707
+ return abort();
1708
+ }
1709
+ if (structuredResult !== null) {
1710
+ return structuredResult;
1711
+ }
1712
+ return {
1713
+ text: await streamPromise.promise
1714
+ };
1715
+ },
1716
+ component: params.component
1717
+ };
1718
+ }
1719
+ function createStepFromTool(params, agentOrToolOptions) {
1720
+ const toolOpts = agentOrToolOptions;
1721
+ if (!params.inputSchema || !params.outputSchema) {
1722
+ throw new Error("Tool must have input and output schemas defined");
1723
+ }
1724
+ return {
1725
+ id: params.id,
1726
+ description: params.description,
1727
+ inputSchema: params.inputSchema,
1728
+ outputSchema: params.outputSchema,
1729
+ resumeSchema: params.resumeSchema,
1730
+ suspendSchema: params.suspendSchema,
1731
+ retries: toolOpts?.retries,
1732
+ scorers: toolOpts?.scorers,
1733
+ execute: async ({
1734
+ inputData,
1735
+ mastra,
1736
+ requestContext,
1737
+ tracingContext,
1738
+ suspend,
1739
+ resumeData,
1740
+ runId,
1741
+ workflowId,
1742
+ state,
1743
+ setState
1744
+ }) => {
1745
+ const toolContext = {
1746
+ mastra,
1747
+ requestContext,
1748
+ tracingContext,
1749
+ workflow: {
1750
+ runId,
1751
+ resumeData,
1752
+ suspend,
1753
+ workflowId,
1754
+ state,
1755
+ setState
856
1756
  }
857
- });
858
- }
859
- if (!(date instanceof Date)) {
860
- sleepUntilSpan?.end();
861
- return;
1757
+ };
1758
+ return params.execute(inputData, toolContext);
1759
+ },
1760
+ component: "TOOL"
1761
+ };
1762
+ }
1763
+ function createStepFromProcessor(processor) {
1764
+ const getProcessorEntityType = (phase) => {
1765
+ switch (phase) {
1766
+ case "input":
1767
+ return observability.EntityType.INPUT_PROCESSOR;
1768
+ case "inputStep":
1769
+ return observability.EntityType.INPUT_STEP_PROCESSOR;
1770
+ case "outputStream":
1771
+ case "outputResult":
1772
+ return observability.EntityType.OUTPUT_PROCESSOR;
1773
+ case "outputStep":
1774
+ return observability.EntityType.OUTPUT_STEP_PROCESSOR;
1775
+ default:
1776
+ return observability.EntityType.OUTPUT_PROCESSOR;
862
1777
  }
863
- try {
864
- await this.inngestStep.sleepUntil(entry.id, date);
865
- sleepUntilSpan?.end();
866
- } catch (e) {
867
- sleepUntilSpan?.error({ error: e });
868
- throw e;
1778
+ };
1779
+ const getSpanNamePrefix = (phase) => {
1780
+ switch (phase) {
1781
+ case "input":
1782
+ return "input processor";
1783
+ case "inputStep":
1784
+ return "input step processor";
1785
+ case "outputStream":
1786
+ return "output stream processor";
1787
+ case "outputResult":
1788
+ return "output processor";
1789
+ case "outputStep":
1790
+ return "output step processor";
1791
+ default:
1792
+ return "processor";
869
1793
  }
870
- }
871
- async executeWaitForEvent({ event, timeout }) {
872
- const eventData = await this.inngestStep.waitForEvent(`user-event-${event}`, {
873
- event: `user-event-${event}`,
874
- timeout: timeout ?? 5e3
875
- });
876
- if (eventData === null) {
877
- throw "Timeout waiting for event";
1794
+ };
1795
+ const hasPhaseMethod = (phase) => {
1796
+ switch (phase) {
1797
+ case "input":
1798
+ return !!processor.processInput;
1799
+ case "inputStep":
1800
+ return !!processor.processInputStep;
1801
+ case "outputStream":
1802
+ return !!processor.processOutputStream;
1803
+ case "outputResult":
1804
+ return !!processor.processOutputResult;
1805
+ case "outputStep":
1806
+ return !!processor.processOutputStep;
1807
+ default:
1808
+ return false;
878
1809
  }
879
- return eventData?.data;
880
- }
881
- async executeStep({
882
- step,
883
- stepResults,
884
- executionContext,
885
- resume,
886
- prevOutput,
887
- emitter,
888
- abortController,
889
- runtimeContext,
890
- tracingContext,
891
- writableStream,
892
- disableScorers
893
- }) {
894
- const stepAISpan = tracingContext?.currentSpan?.createChildSpan({
895
- name: `workflow step: '${step.id}'`,
896
- type: aiTracing.AISpanType.WORKFLOW_STEP,
897
- input: prevOutput,
898
- attributes: {
899
- stepId: step.id
900
- },
901
- tracingPolicy: this.options?.tracingPolicy
902
- });
903
- const { inputData, validationError } = await workflows.validateStepInput({
904
- prevOutput,
905
- step,
906
- validateInputs: this.options?.validateInputs ?? false
907
- });
908
- const startedAt = await this.inngestStep.run(
909
- `workflow.${executionContext.workflowId}.run.${executionContext.runId}.step.${step.id}.running_ev`,
910
- async () => {
911
- const startedAt2 = Date.now();
912
- await emitter.emit("watch", {
913
- type: "watch",
914
- payload: {
915
- currentStep: {
916
- id: step.id,
917
- status: "running"
918
- },
919
- workflowState: {
920
- status: "running",
921
- steps: {
922
- ...stepResults,
923
- [step.id]: {
924
- status: "running"
925
- }
926
- },
927
- result: null,
928
- error: null
929
- }
930
- },
931
- eventTimestamp: Date.now()
932
- });
933
- await emitter.emit("watch-v2", {
934
- type: "workflow-step-start",
935
- payload: {
936
- id: step.id,
937
- status: "running",
938
- payload: inputData,
939
- startedAt: startedAt2
940
- }
941
- });
942
- return startedAt2;
1810
+ };
1811
+ return {
1812
+ id: `processor:${processor.id}`,
1813
+ description: processor.name ?? `Processor ${processor.id}`,
1814
+ inputSchema: processors.ProcessorStepSchema,
1815
+ outputSchema: processors.ProcessorStepOutputSchema,
1816
+ execute: async ({ inputData, requestContext, tracingContext }) => {
1817
+ const input = inputData;
1818
+ const {
1819
+ phase,
1820
+ messages,
1821
+ messageList,
1822
+ stepNumber,
1823
+ systemMessages,
1824
+ part,
1825
+ streamParts,
1826
+ state,
1827
+ finishReason,
1828
+ toolCalls,
1829
+ text,
1830
+ retryCount,
1831
+ // inputStep phase fields for model/tools configuration
1832
+ model,
1833
+ tools,
1834
+ toolChoice,
1835
+ activeTools,
1836
+ providerOptions,
1837
+ modelSettings,
1838
+ structuredOutput,
1839
+ steps
1840
+ } = input;
1841
+ const abort = (reason, options) => {
1842
+ throw new agent.TripWire(reason || `Tripwire triggered by ${processor.id}`, options, processor.id);
1843
+ };
1844
+ if (!hasPhaseMethod(phase)) {
1845
+ return input;
943
1846
  }
944
- );
945
- if (step instanceof InngestWorkflow) {
946
- const isResume = !!resume?.steps?.length;
947
- let result;
948
- let runId;
949
- try {
950
- if (isResume) {
951
- runId = stepResults[resume?.steps?.[0]]?.suspendPayload?.__workflow_meta?.runId ?? crypto.randomUUID();
952
- const snapshot = await this.mastra?.getStorage()?.loadWorkflowSnapshot({
953
- workflowName: step.id,
954
- runId
955
- });
956
- const invokeResp = await this.inngestStep.invoke(`workflow.${executionContext.workflowId}.step.${step.id}`, {
957
- function: step.getFunction(),
958
- data: {
959
- inputData,
960
- initialState: executionContext.state ?? snapshot?.value ?? {},
961
- runId,
962
- resume: {
963
- runId,
964
- steps: resume.steps.slice(1),
965
- stepResults: snapshot?.context,
966
- resumePayload: resume.resumePayload,
967
- // @ts-ignore
968
- resumePath: snapshot?.suspendedPaths?.[resume.steps?.[1]]
969
- },
970
- outputOptions: { includeState: true }
971
- }
972
- });
973
- result = invokeResp.result;
974
- runId = invokeResp.runId;
975
- executionContext.state = invokeResp.result.state;
976
- } else {
977
- const invokeResp = await this.inngestStep.invoke(`workflow.${executionContext.workflowId}.step.${step.id}`, {
978
- function: step.getFunction(),
979
- data: {
980
- inputData,
981
- initialState: executionContext.state ?? {},
982
- outputOptions: { includeState: true }
983
- }
984
- });
985
- result = invokeResp.result;
986
- runId = invokeResp.runId;
987
- executionContext.state = invokeResp.result.state;
1847
+ const currentSpan = tracingContext?.currentSpan;
1848
+ const parentSpan = phase === "inputStep" || phase === "outputStep" ? currentSpan?.findParent(observability.SpanType.MODEL_STEP) || currentSpan : currentSpan?.findParent(observability.SpanType.AGENT_RUN) || currentSpan;
1849
+ const processorSpan = phase !== "outputStream" ? parentSpan?.createChildSpan({
1850
+ type: observability.SpanType.PROCESSOR_RUN,
1851
+ name: `${getSpanNamePrefix(phase)}: ${processor.id}`,
1852
+ entityType: getProcessorEntityType(phase),
1853
+ entityId: processor.id,
1854
+ entityName: processor.name ?? processor.id,
1855
+ input: { phase, messageCount: messages?.length },
1856
+ attributes: {
1857
+ processorExecutor: "workflow",
1858
+ // Read processorIndex from processor (set in combineProcessorsIntoWorkflow)
1859
+ processorIndex: processor.processorIndex
988
1860
  }
989
- } catch (e) {
990
- const errorCause = e?.cause;
991
- if (errorCause && typeof errorCause === "object") {
992
- result = errorCause;
993
- runId = errorCause.runId || crypto.randomUUID();
994
- } else {
995
- runId = crypto.randomUUID();
996
- result = {
997
- status: "failed",
998
- error: e instanceof Error ? e : new Error(String(e)),
999
- steps: {},
1000
- input: inputData
1001
- };
1861
+ }) : void 0;
1862
+ const processorTracingContext = processorSpan ? { currentSpan: processorSpan } : tracingContext;
1863
+ const baseContext = {
1864
+ abort,
1865
+ retryCount: retryCount ?? 0,
1866
+ requestContext,
1867
+ tracingContext: processorTracingContext
1868
+ };
1869
+ const passThrough = {
1870
+ phase,
1871
+ // Auto-create MessageList from messages if not provided
1872
+ // This enables running processor workflows from the UI where messageList can't be serialized
1873
+ messageList: messageList ?? (Array.isArray(messages) ? new agent.MessageList().add(messages, "input").addSystem(systemMessages ?? []) : void 0),
1874
+ stepNumber,
1875
+ systemMessages,
1876
+ streamParts,
1877
+ state,
1878
+ finishReason,
1879
+ toolCalls,
1880
+ text,
1881
+ retryCount,
1882
+ // inputStep phase fields for model/tools configuration
1883
+ model,
1884
+ tools,
1885
+ toolChoice,
1886
+ activeTools,
1887
+ providerOptions,
1888
+ modelSettings,
1889
+ structuredOutput,
1890
+ steps
1891
+ };
1892
+ const executePhaseWithSpan = async (fn) => {
1893
+ try {
1894
+ const result = await fn();
1895
+ processorSpan?.end({ output: result });
1896
+ return result;
1897
+ } catch (error) {
1898
+ if (error instanceof agent.TripWire) {
1899
+ processorSpan?.end({ output: { tripwire: error.message } });
1900
+ } else {
1901
+ processorSpan?.error({ error, endSpan: true });
1902
+ }
1903
+ throw error;
1002
1904
  }
1003
- }
1004
- const res = await this.inngestStep.run(
1005
- `workflow.${executionContext.workflowId}.step.${step.id}.nestedwf-results`,
1006
- async () => {
1007
- if (result.status === "failed") {
1008
- await emitter.emit("watch", {
1009
- type: "watch",
1010
- payload: {
1011
- currentStep: {
1012
- id: step.id,
1013
- status: "failed",
1014
- error: result?.error
1015
- },
1016
- workflowState: {
1017
- status: "running",
1018
- steps: stepResults,
1019
- result: null,
1020
- error: null
1021
- }
1022
- },
1023
- eventTimestamp: Date.now()
1024
- });
1025
- await emitter.emit("watch-v2", {
1026
- type: "workflow-step-result",
1027
- payload: {
1028
- id: step.id,
1029
- status: "failed",
1030
- error: result?.error,
1031
- payload: prevOutput
1905
+ };
1906
+ return executePhaseWithSpan(async () => {
1907
+ switch (phase) {
1908
+ case "input": {
1909
+ if (processor.processInput) {
1910
+ if (!passThrough.messageList) {
1911
+ throw new error.MastraError({
1912
+ category: error.ErrorCategory.USER,
1913
+ domain: error.ErrorDomain.MASTRA_WORKFLOW,
1914
+ id: "PROCESSOR_MISSING_MESSAGE_LIST",
1915
+ text: `Processor ${processor.id} requires messageList or messages for processInput phase`
1916
+ });
1032
1917
  }
1033
- });
1034
- return { executionContext, result: { status: "failed", error: result?.error } };
1035
- } else if (result.status === "suspended") {
1036
- const suspendedSteps = Object.entries(result.steps).filter(([_stepName, stepResult]) => {
1037
- const stepRes2 = stepResult;
1038
- return stepRes2?.status === "suspended";
1039
- });
1040
- for (const [stepName, stepResult] of suspendedSteps) {
1041
- const suspendPath = [stepName, ...stepResult?.suspendPayload?.__workflow_meta?.path ?? []];
1042
- executionContext.suspendedPaths[step.id] = executionContext.executionPath;
1043
- await emitter.emit("watch", {
1044
- type: "watch",
1045
- payload: {
1046
- currentStep: {
1047
- id: step.id,
1048
- status: "suspended",
1049
- payload: stepResult.payload,
1050
- suspendPayload: {
1051
- ...stepResult?.suspendPayload,
1052
- __workflow_meta: { runId, path: suspendPath }
1053
- }
1054
- },
1055
- workflowState: {
1056
- status: "running",
1057
- steps: stepResults,
1058
- result: null,
1059
- error: null
1060
- }
1061
- },
1062
- eventTimestamp: Date.now()
1918
+ const idsBeforeProcessing = messages.map((m) => m.id);
1919
+ const check = passThrough.messageList.makeMessageSourceChecker();
1920
+ const result = await processor.processInput({
1921
+ ...baseContext,
1922
+ messages,
1923
+ messageList: passThrough.messageList,
1924
+ systemMessages: systemMessages ?? []
1063
1925
  });
1064
- await emitter.emit("watch-v2", {
1065
- type: "workflow-step-suspended",
1066
- payload: {
1067
- id: step.id,
1068
- status: "suspended"
1926
+ if (result instanceof agent.MessageList) {
1927
+ if (result !== passThrough.messageList) {
1928
+ throw new error.MastraError({
1929
+ category: error.ErrorCategory.USER,
1930
+ domain: error.ErrorDomain.MASTRA_WORKFLOW,
1931
+ id: "PROCESSOR_RETURNED_EXTERNAL_MESSAGE_LIST",
1932
+ text: `Processor ${processor.id} returned a MessageList instance other than the one passed in. Use the messageList argument instead.`
1933
+ });
1069
1934
  }
1935
+ return {
1936
+ ...passThrough,
1937
+ messages: result.get.all.db(),
1938
+ systemMessages: result.getAllSystemMessages()
1939
+ };
1940
+ } else if (Array.isArray(result)) {
1941
+ processors.ProcessorRunner.applyMessagesToMessageList(
1942
+ result,
1943
+ passThrough.messageList,
1944
+ idsBeforeProcessing,
1945
+ check,
1946
+ "input"
1947
+ );
1948
+ return { ...passThrough, messages: result };
1949
+ } else if (result && "messages" in result && "systemMessages" in result) {
1950
+ const typedResult = result;
1951
+ processors.ProcessorRunner.applyMessagesToMessageList(
1952
+ typedResult.messages,
1953
+ passThrough.messageList,
1954
+ idsBeforeProcessing,
1955
+ check,
1956
+ "input"
1957
+ );
1958
+ passThrough.messageList.replaceAllSystemMessages(typedResult.systemMessages);
1959
+ return {
1960
+ ...passThrough,
1961
+ messages: typedResult.messages,
1962
+ systemMessages: typedResult.systemMessages
1963
+ };
1964
+ }
1965
+ return { ...passThrough, messages };
1966
+ }
1967
+ return { ...passThrough, messages };
1968
+ }
1969
+ case "inputStep": {
1970
+ if (processor.processInputStep) {
1971
+ if (!passThrough.messageList) {
1972
+ throw new error.MastraError({
1973
+ category: error.ErrorCategory.USER,
1974
+ domain: error.ErrorDomain.MASTRA_WORKFLOW,
1975
+ id: "PROCESSOR_MISSING_MESSAGE_LIST",
1976
+ text: `Processor ${processor.id} requires messageList or messages for processInputStep phase`
1977
+ });
1978
+ }
1979
+ const idsBeforeProcessing = messages.map((m) => m.id);
1980
+ const check = passThrough.messageList.makeMessageSourceChecker();
1981
+ const result = await processor.processInputStep({
1982
+ ...baseContext,
1983
+ messages,
1984
+ messageList: passThrough.messageList,
1985
+ stepNumber: stepNumber ?? 0,
1986
+ systemMessages: systemMessages ?? [],
1987
+ // Pass model/tools configuration fields - types match ProcessInputStepArgs
1988
+ model,
1989
+ tools,
1990
+ toolChoice,
1991
+ activeTools,
1992
+ providerOptions,
1993
+ modelSettings,
1994
+ structuredOutput,
1995
+ steps: steps ?? []
1996
+ });
1997
+ const validatedResult = await processors.ProcessorRunner.validateAndFormatProcessInputStepResult(result, {
1998
+ messageList: passThrough.messageList,
1999
+ processor,
2000
+ stepNumber: stepNumber ?? 0
1070
2001
  });
1071
- return {
1072
- executionContext,
1073
- result: {
1074
- status: "suspended",
1075
- payload: stepResult.payload,
1076
- suspendPayload: {
1077
- ...stepResult?.suspendPayload,
1078
- __workflow_meta: { runId, path: suspendPath }
2002
+ if (validatedResult.messages) {
2003
+ processors.ProcessorRunner.applyMessagesToMessageList(
2004
+ validatedResult.messages,
2005
+ passThrough.messageList,
2006
+ idsBeforeProcessing,
2007
+ check
2008
+ );
2009
+ }
2010
+ if (validatedResult.systemMessages) {
2011
+ passThrough.messageList.replaceAllSystemMessages(validatedResult.systemMessages);
2012
+ }
2013
+ return { ...passThrough, messages, ...validatedResult };
2014
+ }
2015
+ return { ...passThrough, messages };
2016
+ }
2017
+ case "outputStream": {
2018
+ if (processor.processOutputStream) {
2019
+ const spanKey = `__outputStreamSpan_${processor.id}`;
2020
+ const mutableState = state ?? {};
2021
+ let processorSpan2 = mutableState[spanKey];
2022
+ if (!processorSpan2 && parentSpan) {
2023
+ processorSpan2 = parentSpan.createChildSpan({
2024
+ type: observability.SpanType.PROCESSOR_RUN,
2025
+ name: `output stream processor: ${processor.id}`,
2026
+ entityType: observability.EntityType.OUTPUT_PROCESSOR,
2027
+ entityId: processor.id,
2028
+ entityName: processor.name ?? processor.id,
2029
+ input: { phase, streamParts: [] },
2030
+ attributes: {
2031
+ processorExecutor: "workflow",
2032
+ processorIndex: processor.processorIndex
1079
2033
  }
2034
+ });
2035
+ mutableState[spanKey] = processorSpan2;
2036
+ }
2037
+ if (processorSpan2) {
2038
+ processorSpan2.input = {
2039
+ phase,
2040
+ streamParts: streamParts ?? [],
2041
+ totalChunks: (streamParts ?? []).length
2042
+ };
2043
+ }
2044
+ const processorTracingContext2 = processorSpan2 ? { currentSpan: processorSpan2 } : baseContext.tracingContext;
2045
+ let result;
2046
+ try {
2047
+ result = await processor.processOutputStream({
2048
+ ...baseContext,
2049
+ tracingContext: processorTracingContext2,
2050
+ part,
2051
+ streamParts: streamParts ?? [],
2052
+ state: mutableState,
2053
+ messageList: passThrough.messageList
2054
+ // Optional for stream processing
2055
+ });
2056
+ if (part && part.type === "finish") {
2057
+ processorSpan2?.end({ output: result });
2058
+ delete mutableState[spanKey];
1080
2059
  }
1081
- };
1082
- }
1083
- await emitter.emit("watch", {
1084
- type: "watch",
1085
- payload: {
1086
- currentStep: {
1087
- id: step.id,
1088
- status: "suspended",
1089
- payload: {}
1090
- },
1091
- workflowState: {
1092
- status: "running",
1093
- steps: stepResults,
1094
- result: null,
1095
- error: null
2060
+ } catch (error) {
2061
+ if (error instanceof agent.TripWire) {
2062
+ processorSpan2?.end({ output: { tripwire: error.message } });
2063
+ } else {
2064
+ processorSpan2?.error({ error, endSpan: true });
1096
2065
  }
1097
- },
1098
- eventTimestamp: Date.now()
1099
- });
1100
- return {
1101
- executionContext,
1102
- result: {
1103
- status: "suspended",
1104
- payload: {}
1105
- }
1106
- };
1107
- }
1108
- await emitter.emit("watch", {
1109
- type: "watch",
1110
- payload: {
1111
- currentStep: {
1112
- id: step.id,
1113
- status: "success",
1114
- output: result?.result
1115
- },
1116
- workflowState: {
1117
- status: "running",
1118
- steps: stepResults,
1119
- result: null,
1120
- error: null
2066
+ delete mutableState[spanKey];
2067
+ throw error;
1121
2068
  }
1122
- },
1123
- eventTimestamp: Date.now()
1124
- });
1125
- await emitter.emit("watch-v2", {
1126
- type: "workflow-step-result",
1127
- payload: {
1128
- id: step.id,
1129
- status: "success",
1130
- output: result?.result
1131
- }
1132
- });
1133
- await emitter.emit("watch-v2", {
1134
- type: "workflow-step-finish",
1135
- payload: {
1136
- id: step.id,
1137
- metadata: {}
2069
+ return { ...passThrough, state: mutableState, part: result };
1138
2070
  }
1139
- });
1140
- return { executionContext, result: { status: "success", output: result?.result } };
1141
- }
1142
- );
1143
- Object.assign(executionContext, res.executionContext);
1144
- return {
1145
- ...res.result,
1146
- startedAt,
1147
- endedAt: Date.now(),
1148
- payload: inputData,
1149
- resumedAt: resume?.steps[0] === step.id ? startedAt : void 0,
1150
- resumePayload: resume?.steps[0] === step.id ? resume?.resumePayload : void 0
1151
- };
1152
- }
1153
- let stepRes;
1154
- try {
1155
- stepRes = await this.inngestStep.run(`workflow.${executionContext.workflowId}.step.${step.id}`, async () => {
1156
- let execResults;
1157
- let suspended;
1158
- let bailed;
1159
- try {
1160
- if (validationError) {
1161
- throw validationError;
2071
+ return { ...passThrough, part };
1162
2072
  }
1163
- const result = await step.execute({
1164
- runId: executionContext.runId,
1165
- mastra: this.mastra,
1166
- runtimeContext,
1167
- writableStream,
1168
- state: executionContext?.state ?? {},
1169
- setState: (state) => {
1170
- executionContext.state = state;
1171
- },
1172
- inputData,
1173
- resumeData: resume?.steps[0] === step.id ? resume?.resumePayload : void 0,
1174
- tracingContext: {
1175
- currentSpan: stepAISpan
1176
- },
1177
- getInitData: () => stepResults?.input,
1178
- getStepResult: workflows.getStepResult.bind(this, stepResults),
1179
- suspend: async (suspendPayload) => {
1180
- executionContext.suspendedPaths[step.id] = executionContext.executionPath;
1181
- suspended = { payload: suspendPayload };
1182
- },
1183
- bail: (result2) => {
1184
- bailed = { payload: result2 };
1185
- },
1186
- resume: {
1187
- steps: resume?.steps?.slice(1) || [],
1188
- resumePayload: resume?.resumePayload,
1189
- // @ts-ignore
1190
- runId: stepResults[step.id]?.suspendPayload?.__workflow_meta?.runId
1191
- },
1192
- [_constants.EMITTER_SYMBOL]: emitter,
1193
- engine: {
1194
- step: this.inngestStep
1195
- },
1196
- abortSignal: abortController.signal
1197
- });
1198
- const endedAt = Date.now();
1199
- execResults = {
1200
- status: "success",
1201
- output: result,
1202
- startedAt,
1203
- endedAt,
1204
- payload: inputData,
1205
- resumedAt: resume?.steps[0] === step.id ? startedAt : void 0,
1206
- resumePayload: resume?.steps[0] === step.id ? resume?.resumePayload : void 0
1207
- };
1208
- } catch (e) {
1209
- const stepFailure = {
1210
- status: "failed",
1211
- payload: inputData,
1212
- error: e instanceof Error ? e.message : String(e),
1213
- endedAt: Date.now(),
1214
- startedAt,
1215
- resumedAt: resume?.steps[0] === step.id ? startedAt : void 0,
1216
- resumePayload: resume?.steps[0] === step.id ? resume?.resumePayload : void 0
1217
- };
1218
- execResults = stepFailure;
1219
- const fallbackErrorMessage = `Step ${step.id} failed`;
1220
- stepAISpan?.error({ error: new Error(execResults.error ?? fallbackErrorMessage) });
1221
- throw new inngest.RetryAfterError(execResults.error ?? fallbackErrorMessage, executionContext.retryConfig.delay, {
1222
- cause: execResults
1223
- });
1224
- }
1225
- if (suspended) {
1226
- execResults = {
1227
- status: "suspended",
1228
- suspendPayload: suspended.payload,
1229
- payload: inputData,
1230
- suspendedAt: Date.now(),
1231
- startedAt,
1232
- resumedAt: resume?.steps[0] === step.id ? startedAt : void 0,
1233
- resumePayload: resume?.steps[0] === step.id ? resume?.resumePayload : void 0
1234
- };
1235
- } else if (bailed) {
1236
- execResults = {
1237
- status: "bailed",
1238
- output: bailed.payload,
1239
- payload: inputData,
1240
- endedAt: Date.now(),
1241
- startedAt
1242
- };
1243
- }
1244
- await emitter.emit("watch", {
1245
- type: "watch",
1246
- payload: {
1247
- currentStep: {
1248
- id: step.id,
1249
- ...execResults
1250
- },
1251
- workflowState: {
1252
- status: "running",
1253
- steps: { ...stepResults, [step.id]: execResults },
1254
- result: null,
1255
- error: null
1256
- }
1257
- },
1258
- eventTimestamp: Date.now()
1259
- });
1260
- if (execResults.status === "suspended") {
1261
- await emitter.emit("watch-v2", {
1262
- type: "workflow-step-suspended",
1263
- payload: {
1264
- id: step.id,
1265
- ...execResults
1266
- }
1267
- });
1268
- } else {
1269
- await emitter.emit("watch-v2", {
1270
- type: "workflow-step-result",
1271
- payload: {
1272
- id: step.id,
1273
- ...execResults
1274
- }
1275
- });
1276
- await emitter.emit("watch-v2", {
1277
- type: "workflow-step-finish",
1278
- payload: {
1279
- id: step.id,
1280
- metadata: {}
2073
+ case "outputResult": {
2074
+ if (processor.processOutputResult) {
2075
+ if (!passThrough.messageList) {
2076
+ throw new error.MastraError({
2077
+ category: error.ErrorCategory.USER,
2078
+ domain: error.ErrorDomain.MASTRA_WORKFLOW,
2079
+ id: "PROCESSOR_MISSING_MESSAGE_LIST",
2080
+ text: `Processor ${processor.id} requires messageList or messages for processOutputResult phase`
2081
+ });
2082
+ }
2083
+ const idsBeforeProcessing = messages.map((m) => m.id);
2084
+ const check = passThrough.messageList.makeMessageSourceChecker();
2085
+ const result = await processor.processOutputResult({
2086
+ ...baseContext,
2087
+ messages,
2088
+ messageList: passThrough.messageList
2089
+ });
2090
+ if (result instanceof agent.MessageList) {
2091
+ if (result !== passThrough.messageList) {
2092
+ throw new error.MastraError({
2093
+ category: error.ErrorCategory.USER,
2094
+ domain: error.ErrorDomain.MASTRA_WORKFLOW,
2095
+ id: "PROCESSOR_RETURNED_EXTERNAL_MESSAGE_LIST",
2096
+ text: `Processor ${processor.id} returned a MessageList instance other than the one passed in. Use the messageList argument instead.`
2097
+ });
2098
+ }
2099
+ return {
2100
+ ...passThrough,
2101
+ messages: result.get.all.db(),
2102
+ systemMessages: result.getAllSystemMessages()
2103
+ };
2104
+ } else if (Array.isArray(result)) {
2105
+ processors.ProcessorRunner.applyMessagesToMessageList(
2106
+ result,
2107
+ passThrough.messageList,
2108
+ idsBeforeProcessing,
2109
+ check,
2110
+ "response"
2111
+ );
2112
+ return { ...passThrough, messages: result };
2113
+ } else if (result && "messages" in result && "systemMessages" in result) {
2114
+ const typedResult = result;
2115
+ processors.ProcessorRunner.applyMessagesToMessageList(
2116
+ typedResult.messages,
2117
+ passThrough.messageList,
2118
+ idsBeforeProcessing,
2119
+ check,
2120
+ "response"
2121
+ );
2122
+ passThrough.messageList.replaceAllSystemMessages(typedResult.systemMessages);
2123
+ return {
2124
+ ...passThrough,
2125
+ messages: typedResult.messages,
2126
+ systemMessages: typedResult.systemMessages
2127
+ };
2128
+ }
2129
+ return { ...passThrough, messages };
1281
2130
  }
1282
- });
1283
- }
1284
- stepAISpan?.end({ output: execResults });
1285
- return { result: execResults, executionContext, stepResults };
1286
- });
1287
- } catch (e) {
1288
- const stepFailure = e instanceof Error ? e?.cause : {
1289
- status: "failed",
1290
- error: e instanceof Error ? e.message : String(e),
1291
- payload: inputData,
1292
- startedAt,
1293
- endedAt: Date.now()
1294
- };
1295
- stepRes = {
1296
- result: stepFailure,
1297
- executionContext,
1298
- stepResults: {
1299
- ...stepResults,
1300
- [step.id]: stepFailure
1301
- }
1302
- };
1303
- }
1304
- if (disableScorers !== false && stepRes.result.status === "success") {
1305
- await this.inngestStep.run(`workflow.${executionContext.workflowId}.step.${step.id}.score`, async () => {
1306
- if (step.scorers) {
1307
- await this.runScorers({
1308
- scorers: step.scorers,
1309
- runId: executionContext.runId,
1310
- input: inputData,
1311
- output: stepRes.result,
1312
- workflowId: executionContext.workflowId,
1313
- stepId: step.id,
1314
- runtimeContext,
1315
- disableScorers,
1316
- tracingContext: { currentSpan: stepAISpan }
1317
- });
1318
- }
1319
- });
1320
- }
1321
- Object.assign(executionContext.suspendedPaths, stepRes.executionContext.suspendedPaths);
1322
- Object.assign(stepResults, stepRes.stepResults);
1323
- executionContext.state = stepRes.executionContext.state;
1324
- return stepRes.result;
1325
- }
1326
- async persistStepUpdate({
1327
- workflowId,
1328
- runId,
1329
- stepResults,
1330
- resourceId,
1331
- executionContext,
1332
- serializedStepGraph,
1333
- workflowStatus,
1334
- result,
1335
- error
1336
- }) {
1337
- await this.inngestStep.run(
1338
- `workflow.${workflowId}.run.${runId}.path.${JSON.stringify(executionContext.executionPath)}.stepUpdate`,
1339
- async () => {
1340
- const shouldPersistSnapshot = this.options.shouldPersistSnapshot({ stepResults, workflowStatus });
1341
- if (!shouldPersistSnapshot) {
1342
- return;
1343
- }
1344
- await this.mastra?.getStorage()?.persistWorkflowSnapshot({
1345
- workflowName: workflowId,
1346
- runId,
1347
- resourceId,
1348
- snapshot: {
1349
- runId,
1350
- value: executionContext.state,
1351
- context: stepResults,
1352
- activePaths: [],
1353
- suspendedPaths: executionContext.suspendedPaths,
1354
- waitingPaths: {},
1355
- serializedStepGraph,
1356
- status: workflowStatus,
1357
- result,
1358
- error,
1359
- // @ts-ignore
1360
- timestamp: Date.now()
2131
+ return { ...passThrough, messages };
1361
2132
  }
1362
- });
1363
- }
1364
- );
1365
- }
1366
- async executeConditional({
1367
- workflowId,
1368
- runId,
1369
- entry,
1370
- prevOutput,
1371
- prevStep,
1372
- stepResults,
1373
- serializedStepGraph,
1374
- resume,
1375
- executionContext,
1376
- emitter,
1377
- abortController,
1378
- runtimeContext,
1379
- writableStream,
1380
- disableScorers,
1381
- tracingContext
1382
- }) {
1383
- const conditionalSpan = tracingContext?.currentSpan?.createChildSpan({
1384
- type: aiTracing.AISpanType.WORKFLOW_CONDITIONAL,
1385
- name: `conditional: '${entry.conditions.length} conditions'`,
1386
- input: prevOutput,
1387
- attributes: {
1388
- conditionCount: entry.conditions.length
1389
- },
1390
- tracingPolicy: this.options?.tracingPolicy
1391
- });
1392
- let execResults;
1393
- const truthyIndexes = (await Promise.all(
1394
- entry.conditions.map(
1395
- (cond, index) => this.inngestStep.run(`workflow.${workflowId}.conditional.${index}`, async () => {
1396
- const evalSpan = conditionalSpan?.createChildSpan({
1397
- type: aiTracing.AISpanType.WORKFLOW_CONDITIONAL_EVAL,
1398
- name: `condition: '${index}'`,
1399
- input: prevOutput,
1400
- attributes: {
1401
- conditionIndex: index
1402
- },
1403
- tracingPolicy: this.options?.tracingPolicy
1404
- });
1405
- try {
1406
- const result = await cond({
1407
- runId,
1408
- workflowId,
1409
- mastra: this.mastra,
1410
- runtimeContext,
1411
- runCount: -1,
1412
- inputData: prevOutput,
1413
- state: executionContext.state,
1414
- setState: (state) => {
1415
- executionContext.state = state;
1416
- },
1417
- tracingContext: {
1418
- currentSpan: evalSpan
1419
- },
1420
- getInitData: () => stepResults?.input,
1421
- getStepResult: workflows.getStepResult.bind(this, stepResults),
1422
- // TODO: this function shouldn't have suspend probably?
1423
- suspend: async (_suspendPayload) => {
1424
- },
1425
- bail: () => {
1426
- },
1427
- abort: () => {
1428
- abortController.abort();
1429
- },
1430
- [_constants.EMITTER_SYMBOL]: emitter,
1431
- [_constants.STREAM_FORMAT_SYMBOL]: executionContext.format,
1432
- // TODO: add streamVNext support
1433
- engine: {
1434
- step: this.inngestStep
1435
- },
1436
- abortSignal: abortController.signal,
1437
- writer: new tools.ToolStream(
1438
- {
1439
- prefix: "workflow-step",
1440
- callId: crypto.randomUUID(),
1441
- name: "conditional",
1442
- runId
1443
- },
1444
- writableStream
1445
- )
1446
- });
1447
- evalSpan?.end({
1448
- output: result,
1449
- attributes: {
1450
- result: !!result
2133
+ case "outputStep": {
2134
+ if (processor.processOutputStep) {
2135
+ if (!passThrough.messageList) {
2136
+ throw new error.MastraError({
2137
+ category: error.ErrorCategory.USER,
2138
+ domain: error.ErrorDomain.MASTRA_WORKFLOW,
2139
+ id: "PROCESSOR_MISSING_MESSAGE_LIST",
2140
+ text: `Processor ${processor.id} requires messageList or messages for processOutputStep phase`
2141
+ });
1451
2142
  }
1452
- });
1453
- return result ? index : null;
1454
- } catch (e) {
1455
- evalSpan?.error({
1456
- error: e instanceof Error ? e : new Error(String(e)),
1457
- attributes: {
1458
- result: false
2143
+ const idsBeforeProcessing = messages.map((m) => m.id);
2144
+ const check = passThrough.messageList.makeMessageSourceChecker();
2145
+ const result = await processor.processOutputStep({
2146
+ ...baseContext,
2147
+ messages,
2148
+ messageList: passThrough.messageList,
2149
+ stepNumber: stepNumber ?? 0,
2150
+ finishReason,
2151
+ toolCalls,
2152
+ text,
2153
+ systemMessages: systemMessages ?? [],
2154
+ steps: steps ?? []
2155
+ });
2156
+ if (result instanceof agent.MessageList) {
2157
+ if (result !== passThrough.messageList) {
2158
+ throw new error.MastraError({
2159
+ category: error.ErrorCategory.USER,
2160
+ domain: error.ErrorDomain.MASTRA_WORKFLOW,
2161
+ id: "PROCESSOR_RETURNED_EXTERNAL_MESSAGE_LIST",
2162
+ text: `Processor ${processor.id} returned a MessageList instance other than the one passed in. Use the messageList argument instead.`
2163
+ });
2164
+ }
2165
+ return {
2166
+ ...passThrough,
2167
+ messages: result.get.all.db(),
2168
+ systemMessages: result.getAllSystemMessages()
2169
+ };
2170
+ } else if (Array.isArray(result)) {
2171
+ processors.ProcessorRunner.applyMessagesToMessageList(
2172
+ result,
2173
+ passThrough.messageList,
2174
+ idsBeforeProcessing,
2175
+ check,
2176
+ "response"
2177
+ );
2178
+ return { ...passThrough, messages: result };
2179
+ } else if (result && "messages" in result && "systemMessages" in result) {
2180
+ const typedResult = result;
2181
+ processors.ProcessorRunner.applyMessagesToMessageList(
2182
+ typedResult.messages,
2183
+ passThrough.messageList,
2184
+ idsBeforeProcessing,
2185
+ check,
2186
+ "response"
2187
+ );
2188
+ passThrough.messageList.replaceAllSystemMessages(typedResult.systemMessages);
2189
+ return {
2190
+ ...passThrough,
2191
+ messages: typedResult.messages,
2192
+ systemMessages: typedResult.systemMessages
2193
+ };
1459
2194
  }
1460
- });
1461
- return null;
1462
- }
1463
- })
1464
- )
1465
- )).filter((index) => index !== null);
1466
- const stepsToRun = entry.steps.filter((_, index) => truthyIndexes.includes(index));
1467
- conditionalSpan?.update({
1468
- attributes: {
1469
- truthyIndexes,
1470
- selectedSteps: stepsToRun.map((s) => s.type === "step" ? s.step.id : `control-${s.type}`)
1471
- }
1472
- });
1473
- const results = await Promise.all(
1474
- stepsToRun.map(
1475
- (step, index) => this.executeEntry({
1476
- workflowId,
1477
- runId,
1478
- entry: step,
1479
- serializedStepGraph,
1480
- prevStep,
1481
- stepResults,
1482
- resume,
1483
- executionContext: {
1484
- workflowId,
1485
- runId,
1486
- executionPath: [...executionContext.executionPath, index],
1487
- suspendedPaths: executionContext.suspendedPaths,
1488
- retryConfig: executionContext.retryConfig,
1489
- executionSpan: executionContext.executionSpan,
1490
- state: executionContext.state
1491
- },
1492
- emitter,
1493
- abortController,
1494
- runtimeContext,
1495
- writableStream,
1496
- disableScorers,
1497
- tracingContext: {
1498
- currentSpan: conditionalSpan
1499
- }
1500
- })
1501
- )
1502
- );
1503
- const hasFailed = results.find((result) => result.result.status === "failed");
1504
- const hasSuspended = results.find((result) => result.result.status === "suspended");
1505
- if (hasFailed) {
1506
- execResults = { status: "failed", error: hasFailed.result.error };
1507
- } else if (hasSuspended) {
1508
- execResults = { status: "suspended", suspendPayload: hasSuspended.result.suspendPayload };
1509
- } else {
1510
- execResults = {
1511
- status: "success",
1512
- output: results.reduce((acc, result, index) => {
1513
- if (result.result.status === "success") {
1514
- acc[stepsToRun[index].step.id] = result.output;
2195
+ return { ...passThrough, messages };
2196
+ }
2197
+ return { ...passThrough, messages };
1515
2198
  }
1516
- return acc;
1517
- }, {})
1518
- };
1519
- }
1520
- if (execResults.status === "failed") {
1521
- conditionalSpan?.error({
1522
- error: new Error(execResults.error)
2199
+ default:
2200
+ return { ...passThrough, messages };
2201
+ }
1523
2202
  });
1524
- } else {
1525
- conditionalSpan?.end({
1526
- output: execResults.output || execResults
2203
+ },
2204
+ component: "PROCESSOR"
2205
+ };
2206
+ }
2207
+ function init(inngest) {
2208
+ return {
2209
+ createWorkflow(params) {
2210
+ return new InngestWorkflow(
2211
+ params,
2212
+ inngest
2213
+ );
2214
+ },
2215
+ createStep,
2216
+ cloneStep(step, opts) {
2217
+ return {
2218
+ id: opts.id,
2219
+ description: step.description,
2220
+ inputSchema: step.inputSchema,
2221
+ outputSchema: step.outputSchema,
2222
+ resumeSchema: step.resumeSchema,
2223
+ suspendSchema: step.suspendSchema,
2224
+ stateSchema: step.stateSchema,
2225
+ execute: step.execute,
2226
+ retries: step.retries,
2227
+ scorers: step.scorers,
2228
+ component: step.component
2229
+ };
2230
+ },
2231
+ cloneWorkflow(workflow, opts) {
2232
+ const wf = new workflows.Workflow({
2233
+ id: opts.id,
2234
+ inputSchema: workflow.inputSchema,
2235
+ outputSchema: workflow.outputSchema,
2236
+ steps: workflow.stepDefs,
2237
+ mastra: workflow.mastra,
2238
+ options: workflow.options
1527
2239
  });
2240
+ wf.setStepFlow(workflow.stepGraph);
2241
+ wf.commit();
2242
+ return wf;
1528
2243
  }
1529
- return execResults;
1530
- }
1531
- };
2244
+ };
2245
+ }
1532
2246
 
1533
2247
  exports.InngestExecutionEngine = InngestExecutionEngine;
2248
+ exports.InngestPubSub = InngestPubSub;
1534
2249
  exports.InngestRun = InngestRun;
1535
2250
  exports.InngestWorkflow = InngestWorkflow;
2251
+ exports._compatibilityCheck = _compatibilityCheck;
2252
+ exports.createServe = createServe;
1536
2253
  exports.createStep = createStep;
1537
2254
  exports.init = init;
1538
2255
  exports.serve = serve;