@mastra/inngest 0.0.0-transpile-packages-20250731152758 → 0.0.0-type-testing-20260120105120

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