@mastra/inngest 0.0.0-unified-sidebar-20251010130811 → 0.0.0-unified-workspace-snapshot-20260128233410

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