@mastra/inngest 0.0.0-allow-to-pass-a-mastra-url-instance-20251105224938 → 0.0.0-auth-and-authz-20260130180831

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