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