@usestratus/sdk 1.6.0 → 1.9.0
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/README.md +109 -2
- package/dist/ai-sdk/index.d.ts +294 -0
- package/dist/ai-sdk/index.d.ts.map +1 -0
- package/dist/ai-sdk/index.js +741 -0
- package/dist/ai-sdk/index.js.map +1 -0
- package/dist/azure/index.d.ts +4 -4
- package/dist/azure/index.d.ts.map +1 -1
- package/dist/azure/index.js +1 -1
- package/dist/azure/index.js.map +1 -1
- package/dist/azure/responses-model.js +15 -5
- package/dist/azure/responses-model.js.map +1 -1
- package/dist/core/codemode/index.d.ts +1 -2
- package/dist/core/codemode/index.d.ts.map +1 -1
- package/dist/core/codemode/index.js.map +1 -1
- package/dist/core/index.d.ts +41 -34
- package/dist/core/index.d.ts.map +1 -1
- package/dist/core/index.js +18 -15
- package/dist/core/index.js.map +1 -1
- package/dist/core/run.d.ts.map +1 -1
- package/dist/core/run.js +1 -1
- package/dist/core/run.js.map +1 -1
- package/dist/core/session.d.ts.map +1 -1
- package/dist/core/session.js.map +1 -1
- package/dist/core/subagent.js +1 -1
- package/dist/core/subagent.js.map +1 -1
- package/dist/core/types.d.ts +4 -1
- package/dist/core/types.d.ts.map +1 -1
- package/dist/core/utils/zod.d.ts.map +1 -1
- package/dist/core/utils/zod.js.map +1 -1
- package/dist/core/validate-agent.d.ts.map +1 -1
- package/dist/core/validate-agent.js.map +1 -1
- package/dist/core/workflow-draft.d.ts +32 -0
- package/dist/core/workflow-draft.d.ts.map +1 -0
- package/dist/core/workflow-draft.js +84 -0
- package/dist/core/workflow-draft.js.map +1 -0
- package/dist/core/workflow-manager.d.ts +25 -0
- package/dist/core/workflow-manager.d.ts.map +1 -0
- package/dist/core/workflow-manager.js +178 -0
- package/dist/core/workflow-manager.js.map +1 -0
- package/dist/core/workflow-saved.d.ts +21 -0
- package/dist/core/workflow-saved.d.ts.map +1 -0
- package/dist/core/workflow-saved.js +79 -0
- package/dist/core/workflow-saved.js.map +1 -0
- package/dist/core/workflow.d.ts +262 -0
- package/dist/core/workflow.d.ts.map +1 -0
- package/dist/core/workflow.js +560 -0
- package/dist/core/workflow.js.map +1 -0
- package/dist/effect/index.d.ts +39 -0
- package/dist/effect/index.d.ts.map +1 -0
- package/dist/effect/index.js +90 -0
- package/dist/effect/index.js.map +1 -0
- package/dist/index.d.ts +1 -1
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +1 -1
- package/dist/index.js.map +1 -1
- package/dist/testing/index.d.ts +1 -1
- package/dist/testing/index.d.ts.map +1 -1
- package/dist/testing/index.js.map +1 -1
- package/package.json +19 -5
|
@@ -0,0 +1,560 @@
|
|
|
1
|
+
import { RunAbortedError, StratusError } from "./errors";
|
|
2
|
+
import { run } from "./run";
|
|
3
|
+
const DEFAULT_WORKFLOW_CONCURRENCY = 4;
|
|
4
|
+
const DEFAULT_MAX_CONCURRENCY = 16;
|
|
5
|
+
const DEFAULT_MAX_TASKS = 1000;
|
|
6
|
+
const DEFAULT_MAX_WORKFLOW_ITERATIONS = 25;
|
|
7
|
+
export function workflow(config) {
|
|
8
|
+
return config;
|
|
9
|
+
}
|
|
10
|
+
export function workflowTask(task) {
|
|
11
|
+
return task;
|
|
12
|
+
}
|
|
13
|
+
export async function runWorkflow(wf, args, options) {
|
|
14
|
+
return executeWorkflow(wf, args, options, () => { });
|
|
15
|
+
}
|
|
16
|
+
export function streamWorkflow(wf, args, options) {
|
|
17
|
+
const channel = createWorkflowEventChannel();
|
|
18
|
+
const result = executeWorkflow(wf, args, options, (event) => channel.push(event)).finally(() => channel.done());
|
|
19
|
+
return { stream: channel[Symbol.asyncIterator](), result };
|
|
20
|
+
}
|
|
21
|
+
async function executeWorkflow(wf, args, options, emit) {
|
|
22
|
+
const normalized = normalizeOptions(options);
|
|
23
|
+
const startedAt = new Date();
|
|
24
|
+
const state = {
|
|
25
|
+
id: normalized.id ?? createRunId(),
|
|
26
|
+
name: wf.name,
|
|
27
|
+
args,
|
|
28
|
+
options: normalized,
|
|
29
|
+
startedAt,
|
|
30
|
+
phases: [],
|
|
31
|
+
totalTasks: 0,
|
|
32
|
+
usage: emptyUsage(),
|
|
33
|
+
numTurns: 0,
|
|
34
|
+
totalCostUsd: 0,
|
|
35
|
+
emit: emit,
|
|
36
|
+
resumeCache: buildResumeCache(normalized.resumeFrom),
|
|
37
|
+
};
|
|
38
|
+
const ctx = new DefaultWorkflowRuntimeContext(state);
|
|
39
|
+
emit({ type: "workflow_started", runId: state.id, name: wf.name });
|
|
40
|
+
try {
|
|
41
|
+
checkWorkflowAborted(normalized.signal);
|
|
42
|
+
checkWorkflowBudget(state);
|
|
43
|
+
const output = await wf.run(ctx, args);
|
|
44
|
+
checkWorkflowBudget(state);
|
|
45
|
+
const completedAt = new Date();
|
|
46
|
+
const result = {
|
|
47
|
+
id: state.id,
|
|
48
|
+
name: wf.name,
|
|
49
|
+
status: "completed",
|
|
50
|
+
output,
|
|
51
|
+
phases: [...state.phases],
|
|
52
|
+
usage: state.usage,
|
|
53
|
+
numTurns: state.numTurns,
|
|
54
|
+
totalCostUsd: state.totalCostUsd,
|
|
55
|
+
startedAt: startedAt.toISOString(),
|
|
56
|
+
completedAt: completedAt.toISOString(),
|
|
57
|
+
durationMs: completedAt.getTime() - startedAt.getTime(),
|
|
58
|
+
};
|
|
59
|
+
emit({ type: "workflow_completed", runId: state.id, result });
|
|
60
|
+
return result;
|
|
61
|
+
}
|
|
62
|
+
catch (error) {
|
|
63
|
+
const message = getErrorMessage(error);
|
|
64
|
+
emit({ type: "workflow_failed", runId: state.id, error: message });
|
|
65
|
+
throw error;
|
|
66
|
+
}
|
|
67
|
+
}
|
|
68
|
+
class DefaultWorkflowRuntimeContext {
|
|
69
|
+
state;
|
|
70
|
+
runId;
|
|
71
|
+
name;
|
|
72
|
+
args;
|
|
73
|
+
constructor(state) {
|
|
74
|
+
this.state = state;
|
|
75
|
+
this.runId = state.id;
|
|
76
|
+
this.name = state.name;
|
|
77
|
+
this.args = state.args;
|
|
78
|
+
}
|
|
79
|
+
get snapshot() {
|
|
80
|
+
const now = new Date();
|
|
81
|
+
return {
|
|
82
|
+
id: this.state.id,
|
|
83
|
+
name: this.state.name,
|
|
84
|
+
status: "running",
|
|
85
|
+
output: "",
|
|
86
|
+
phases: [...this.state.phases],
|
|
87
|
+
usage: this.state.usage,
|
|
88
|
+
numTurns: this.state.numTurns,
|
|
89
|
+
totalCostUsd: this.state.totalCostUsd,
|
|
90
|
+
startedAt: this.state.startedAt.toISOString(),
|
|
91
|
+
completedAt: now.toISOString(),
|
|
92
|
+
durationMs: now.getTime() - this.state.startedAt.getTime(),
|
|
93
|
+
};
|
|
94
|
+
}
|
|
95
|
+
async phase(name, tasks, options) {
|
|
96
|
+
checkWorkflowAborted(this.state.options.signal);
|
|
97
|
+
if (tasks.length === 0) {
|
|
98
|
+
const now = new Date();
|
|
99
|
+
const emptyPhase = {
|
|
100
|
+
id: slugify(name),
|
|
101
|
+
name,
|
|
102
|
+
status: "completed",
|
|
103
|
+
tasks: [],
|
|
104
|
+
startedAt: now.toISOString(),
|
|
105
|
+
completedAt: now.toISOString(),
|
|
106
|
+
durationMs: 0,
|
|
107
|
+
};
|
|
108
|
+
this.state.phases.push(emptyPhase);
|
|
109
|
+
this.state.emit({ type: "workflow_phase_started", runId: this.runId, phase: emptyPhase });
|
|
110
|
+
this.state.emit({ type: "workflow_phase_completed", runId: this.runId, phase: emptyPhase });
|
|
111
|
+
return [];
|
|
112
|
+
}
|
|
113
|
+
this.state.totalTasks += tasks.length;
|
|
114
|
+
if (this.state.totalTasks > this.state.options.maxTasks) {
|
|
115
|
+
throw new StratusError(`Workflow exceeded maxTasks (${this.state.options.maxTasks}); received ${this.state.totalTasks}`);
|
|
116
|
+
}
|
|
117
|
+
const startedAt = new Date();
|
|
118
|
+
const phaseId = uniquePhaseId(name, this.state.phases.length);
|
|
119
|
+
const phaseShell = { id: phaseId, name };
|
|
120
|
+
this.state.emit({ type: "workflow_phase_started", runId: this.runId, phase: phaseShell });
|
|
121
|
+
const concurrency = clampConcurrency(options?.concurrency ?? this.state.options.concurrency, this.state.options.maxConcurrency);
|
|
122
|
+
const { results, failure } = await runTaskPool({
|
|
123
|
+
ctx: this,
|
|
124
|
+
phaseId,
|
|
125
|
+
phaseName: name,
|
|
126
|
+
tasks,
|
|
127
|
+
concurrency,
|
|
128
|
+
failFast: options?.failFast ?? true,
|
|
129
|
+
});
|
|
130
|
+
const completedAt = new Date();
|
|
131
|
+
const failed = results.some((result) => result.status === "failed");
|
|
132
|
+
const phase = {
|
|
133
|
+
id: phaseId,
|
|
134
|
+
name,
|
|
135
|
+
status: failed || failure ? "failed" : "completed",
|
|
136
|
+
tasks: results,
|
|
137
|
+
startedAt: startedAt.toISOString(),
|
|
138
|
+
completedAt: completedAt.toISOString(),
|
|
139
|
+
durationMs: completedAt.getTime() - startedAt.getTime(),
|
|
140
|
+
};
|
|
141
|
+
this.state.phases.push(phase);
|
|
142
|
+
this.state.emit({ type: "workflow_phase_completed", runId: this.runId, phase });
|
|
143
|
+
checkWorkflowBudget(this.state);
|
|
144
|
+
if (failure) {
|
|
145
|
+
throw new StratusError(`Workflow task "${failure.name}" failed: ${failure.error}`);
|
|
146
|
+
}
|
|
147
|
+
return results;
|
|
148
|
+
}
|
|
149
|
+
async runTask(task, options) {
|
|
150
|
+
const phaseId = options?.phaseId ?? "adhoc";
|
|
151
|
+
const phaseName = options?.phaseName ?? "Ad hoc";
|
|
152
|
+
const taskId = task.id ?? `${phaseId}:${task.name ?? "task"}`;
|
|
153
|
+
const cacheKey = `${phaseId}:${taskId}`;
|
|
154
|
+
const cached = this.state.resumeCache.get(cacheKey);
|
|
155
|
+
if (cached?.status === "completed" || cached?.status === "interrupted") {
|
|
156
|
+
const skipped = { ...cached, status: "skipped" };
|
|
157
|
+
this.state.emit({ type: "workflow_task_skipped", runId: this.runId, phaseId, task: skipped });
|
|
158
|
+
return skipped;
|
|
159
|
+
}
|
|
160
|
+
checkWorkflowAborted(this.state.options.signal);
|
|
161
|
+
const name = task.name ?? taskId;
|
|
162
|
+
this.state.emit({
|
|
163
|
+
type: "workflow_task_started",
|
|
164
|
+
runId: this.runId,
|
|
165
|
+
phaseId,
|
|
166
|
+
task: { id: taskId, name },
|
|
167
|
+
});
|
|
168
|
+
const startedAt = new Date();
|
|
169
|
+
let result;
|
|
170
|
+
try {
|
|
171
|
+
result = await executeTask(task, {
|
|
172
|
+
task,
|
|
173
|
+
context: task.context,
|
|
174
|
+
signal: this.state.options.signal,
|
|
175
|
+
workflow: this,
|
|
176
|
+
});
|
|
177
|
+
}
|
|
178
|
+
catch (error) {
|
|
179
|
+
const completedAt = new Date();
|
|
180
|
+
const taskResult = {
|
|
181
|
+
id: taskId,
|
|
182
|
+
name,
|
|
183
|
+
phaseId,
|
|
184
|
+
phaseName,
|
|
185
|
+
status: "failed",
|
|
186
|
+
output: "",
|
|
187
|
+
error: getErrorMessage(error),
|
|
188
|
+
usage: emptyUsage(),
|
|
189
|
+
numTurns: 0,
|
|
190
|
+
totalCostUsd: 0,
|
|
191
|
+
startedAt: startedAt.toISOString(),
|
|
192
|
+
completedAt: completedAt.toISOString(),
|
|
193
|
+
durationMs: completedAt.getTime() - startedAt.getTime(),
|
|
194
|
+
metadata: task.metadata,
|
|
195
|
+
};
|
|
196
|
+
this.state.emit({
|
|
197
|
+
type: "workflow_task_failed",
|
|
198
|
+
runId: this.runId,
|
|
199
|
+
phaseId,
|
|
200
|
+
task: taskResult,
|
|
201
|
+
});
|
|
202
|
+
return taskResult;
|
|
203
|
+
}
|
|
204
|
+
const completedAt = new Date();
|
|
205
|
+
const taskResult = {
|
|
206
|
+
id: taskId,
|
|
207
|
+
name,
|
|
208
|
+
phaseId,
|
|
209
|
+
phaseName,
|
|
210
|
+
status: result.status,
|
|
211
|
+
output: result.output,
|
|
212
|
+
value: result.value,
|
|
213
|
+
interrupted: result.interrupted,
|
|
214
|
+
runResult: result.runResult,
|
|
215
|
+
usage: result.usage,
|
|
216
|
+
numTurns: result.numTurns,
|
|
217
|
+
totalCostUsd: result.totalCostUsd,
|
|
218
|
+
startedAt: startedAt.toISOString(),
|
|
219
|
+
completedAt: completedAt.toISOString(),
|
|
220
|
+
durationMs: completedAt.getTime() - startedAt.getTime(),
|
|
221
|
+
metadata: task.metadata,
|
|
222
|
+
};
|
|
223
|
+
this.addTaskTotals(taskResult);
|
|
224
|
+
this.state.emit({
|
|
225
|
+
type: "workflow_task_completed",
|
|
226
|
+
runId: this.runId,
|
|
227
|
+
phaseId,
|
|
228
|
+
task: taskResult,
|
|
229
|
+
});
|
|
230
|
+
checkWorkflowBudget(this.state);
|
|
231
|
+
return taskResult;
|
|
232
|
+
}
|
|
233
|
+
async synthesize(agent, input, options) {
|
|
234
|
+
this.state.emit({
|
|
235
|
+
type: "workflow_synthesis_started",
|
|
236
|
+
runId: this.runId,
|
|
237
|
+
agentName: agent.name,
|
|
238
|
+
});
|
|
239
|
+
const result = await run(agent, input, {
|
|
240
|
+
...options,
|
|
241
|
+
signal: options?.signal ?? this.state.options.signal,
|
|
242
|
+
});
|
|
243
|
+
addUsage(this.state.usage, result.usage);
|
|
244
|
+
this.state.numTurns += result.numTurns;
|
|
245
|
+
this.state.totalCostUsd += result.totalCostUsd;
|
|
246
|
+
this.state.emit({
|
|
247
|
+
type: "workflow_synthesis_completed",
|
|
248
|
+
runId: this.runId,
|
|
249
|
+
output: result.output,
|
|
250
|
+
});
|
|
251
|
+
checkWorkflowBudget(this.state);
|
|
252
|
+
return result;
|
|
253
|
+
}
|
|
254
|
+
async fanOutAndSynthesize(config) {
|
|
255
|
+
const tasks = await this.phase(config.name, config.tasks, config.phaseOptions);
|
|
256
|
+
const synthesis = await this.synthesize(config.synthesizer, config.prompt(tasks), config.synthesisOptions);
|
|
257
|
+
return { tasks, synthesis, output: synthesis.output };
|
|
258
|
+
}
|
|
259
|
+
async adversarialVerify(config) {
|
|
260
|
+
const tasks = config.findings.map((finding, index) => workflowTask({
|
|
261
|
+
id: finding.id ?? `verify-${index + 1}`,
|
|
262
|
+
name: finding.name ? `verify ${finding.name}` : `verify ${index + 1}`,
|
|
263
|
+
agent: config.verifier,
|
|
264
|
+
input: config.prompt?.(finding) ??
|
|
265
|
+
`Adversarially verify this finding. Start with ACCEPT or REJECT, then explain briefly.\n\n${finding.output}`,
|
|
266
|
+
runOptions: config.runOptions,
|
|
267
|
+
metadata: finding.metadata,
|
|
268
|
+
}));
|
|
269
|
+
const verifications = await this.phase(config.name, tasks, config.phaseOptions);
|
|
270
|
+
const accepted = verifications.filter((result) => verdictAccepts(result.output));
|
|
271
|
+
return {
|
|
272
|
+
verifications,
|
|
273
|
+
accepted,
|
|
274
|
+
rejected: verifications.filter((result) => !accepted.includes(result)),
|
|
275
|
+
};
|
|
276
|
+
}
|
|
277
|
+
async generateAndFilter(config) {
|
|
278
|
+
const candidates = await this.phase(config.name, config.generators, config.phaseOptions);
|
|
279
|
+
const filterTasks = candidates.map((candidate, index) => workflowTask({
|
|
280
|
+
id: `filter-${candidate.id}`,
|
|
281
|
+
name: `filter ${candidate.name}`,
|
|
282
|
+
agent: config.filter,
|
|
283
|
+
input: config.prompt?.(candidate, index) ??
|
|
284
|
+
`Evaluate this candidate. Start with ACCEPT or REJECT, then explain briefly.\n\n${candidate.output}`,
|
|
285
|
+
runOptions: config.runOptions,
|
|
286
|
+
metadata: candidate.metadata,
|
|
287
|
+
}));
|
|
288
|
+
const filtered = await this.phase(`${config.name} filter`, filterTasks, config.phaseOptions);
|
|
289
|
+
return {
|
|
290
|
+
candidates,
|
|
291
|
+
filtered,
|
|
292
|
+
accepted: candidates.filter((_, index) => verdictAccepts(filtered[index]?.output ?? "")),
|
|
293
|
+
};
|
|
294
|
+
}
|
|
295
|
+
async tournament(config) {
|
|
296
|
+
const contestants = await this.phase(config.name, config.contestants, config.phaseOptions);
|
|
297
|
+
let round = contestants.filter((result) => result.status === "completed");
|
|
298
|
+
const matches = [];
|
|
299
|
+
let roundIndex = 1;
|
|
300
|
+
while (round.length > 1) {
|
|
301
|
+
const matchTasks = [];
|
|
302
|
+
for (let index = 0; index < round.length; index += 2) {
|
|
303
|
+
const left = round[index];
|
|
304
|
+
const right = round[index + 1];
|
|
305
|
+
if (!left || !right)
|
|
306
|
+
continue;
|
|
307
|
+
matchTasks.push(workflowTask({
|
|
308
|
+
id: `round-${roundIndex}-match-${Math.floor(index / 2) + 1}`,
|
|
309
|
+
name: `${left.name} vs ${right.name}`,
|
|
310
|
+
agent: config.judge,
|
|
311
|
+
input: config.prompt?.(left, right) ??
|
|
312
|
+
`Pick the better output. Start with LEFT or RIGHT, then explain briefly.\n\nLEFT:\n${left.output}\n\nRIGHT:\n${right.output}`,
|
|
313
|
+
runOptions: config.runOptions,
|
|
314
|
+
metadata: { leftId: left.id, rightId: right.id },
|
|
315
|
+
}));
|
|
316
|
+
}
|
|
317
|
+
const roundMatches = await this.phase(`${config.name} round ${roundIndex}`, matchTasks, {
|
|
318
|
+
...config.phaseOptions,
|
|
319
|
+
failFast: config.phaseOptions?.failFast ?? false,
|
|
320
|
+
});
|
|
321
|
+
matches.push(...roundMatches);
|
|
322
|
+
const winners = [];
|
|
323
|
+
for (let index = 0; index < round.length; index += 2) {
|
|
324
|
+
const left = round[index];
|
|
325
|
+
const right = round[index + 1];
|
|
326
|
+
if (!left)
|
|
327
|
+
continue;
|
|
328
|
+
if (!right) {
|
|
329
|
+
winners.push(left);
|
|
330
|
+
continue;
|
|
331
|
+
}
|
|
332
|
+
const match = roundMatches[Math.floor(index / 2)];
|
|
333
|
+
winners.push(verdictPrefersRight(match?.output ?? "") ? right : left);
|
|
334
|
+
}
|
|
335
|
+
round = winners;
|
|
336
|
+
roundIndex++;
|
|
337
|
+
}
|
|
338
|
+
return { contestants, matches, winner: round[0] };
|
|
339
|
+
}
|
|
340
|
+
async loopUntilDone(config) {
|
|
341
|
+
const maxIterations = config.maxIterations ?? DEFAULT_MAX_WORKFLOW_ITERATIONS;
|
|
342
|
+
const iterations = [];
|
|
343
|
+
for (let iteration = 1; iteration <= maxIterations; iteration++) {
|
|
344
|
+
checkWorkflowAborted(this.state.options.signal);
|
|
345
|
+
checkWorkflowBudget(this.state);
|
|
346
|
+
const previous = iterations.at(-1);
|
|
347
|
+
const step = await config.step({ ctx: this, iteration, previous });
|
|
348
|
+
iterations.push(step);
|
|
349
|
+
if (step.done) {
|
|
350
|
+
return { iterations, done: true, output: step.output ?? stringifyTaskValue(step.value) };
|
|
351
|
+
}
|
|
352
|
+
}
|
|
353
|
+
return {
|
|
354
|
+
iterations,
|
|
355
|
+
done: false,
|
|
356
|
+
output: `Stopped after ${maxIterations} iterations without meeting the stop condition.`,
|
|
357
|
+
};
|
|
358
|
+
}
|
|
359
|
+
addTaskTotals(result) {
|
|
360
|
+
addUsage(this.state.usage, result.usage);
|
|
361
|
+
this.state.numTurns += result.numTurns;
|
|
362
|
+
this.state.totalCostUsd += result.totalCostUsd;
|
|
363
|
+
}
|
|
364
|
+
}
|
|
365
|
+
async function executeTask(task, params) {
|
|
366
|
+
if ("execute" in task && task.execute) {
|
|
367
|
+
const value = await task.execute(params);
|
|
368
|
+
return {
|
|
369
|
+
status: "completed",
|
|
370
|
+
output: stringifyTaskValue(value),
|
|
371
|
+
value: value,
|
|
372
|
+
usage: emptyUsage(),
|
|
373
|
+
numTurns: 0,
|
|
374
|
+
totalCostUsd: 0,
|
|
375
|
+
};
|
|
376
|
+
}
|
|
377
|
+
const result = await run(task.agent, task.input, {
|
|
378
|
+
...task.runOptions,
|
|
379
|
+
context: task.context,
|
|
380
|
+
model: task.model,
|
|
381
|
+
maxTurns: task.maxTurns,
|
|
382
|
+
signal: params.signal,
|
|
383
|
+
});
|
|
384
|
+
return {
|
|
385
|
+
status: result.interrupted ? "interrupted" : "completed",
|
|
386
|
+
output: result.output,
|
|
387
|
+
value: result.output,
|
|
388
|
+
interrupted: result.interrupted,
|
|
389
|
+
runResult: result,
|
|
390
|
+
usage: result.usage,
|
|
391
|
+
numTurns: result.numTurns,
|
|
392
|
+
totalCostUsd: result.totalCostUsd,
|
|
393
|
+
};
|
|
394
|
+
}
|
|
395
|
+
async function runTaskPool(params) {
|
|
396
|
+
const results = new Array(params.tasks.length);
|
|
397
|
+
let next = 0;
|
|
398
|
+
let failure;
|
|
399
|
+
async function worker() {
|
|
400
|
+
while (next < params.tasks.length && !failure) {
|
|
401
|
+
const index = next++;
|
|
402
|
+
const task = params.tasks[index];
|
|
403
|
+
if (!task)
|
|
404
|
+
return;
|
|
405
|
+
const runnableTask = task.id === undefined
|
|
406
|
+
? { ...task, id: `${slugify(task.name ?? "task")}-${index + 1}` }
|
|
407
|
+
: task;
|
|
408
|
+
const result = await params.ctx.runTask(runnableTask, {
|
|
409
|
+
phaseId: params.phaseId,
|
|
410
|
+
phaseName: params.phaseName,
|
|
411
|
+
});
|
|
412
|
+
results[index] = result;
|
|
413
|
+
if (params.failFast && result.status === "failed") {
|
|
414
|
+
failure = result;
|
|
415
|
+
}
|
|
416
|
+
}
|
|
417
|
+
}
|
|
418
|
+
const workerCount = Math.min(params.concurrency, params.tasks.length);
|
|
419
|
+
await Promise.all(Array.from({ length: workerCount }, () => worker()));
|
|
420
|
+
const completedResults = results.filter(Boolean);
|
|
421
|
+
return { results: completedResults, failure };
|
|
422
|
+
}
|
|
423
|
+
function normalizeOptions(options) {
|
|
424
|
+
const maxConcurrency = options?.maxConcurrency ?? DEFAULT_MAX_CONCURRENCY;
|
|
425
|
+
return {
|
|
426
|
+
...options,
|
|
427
|
+
concurrency: clampConcurrency(options?.concurrency ?? DEFAULT_WORKFLOW_CONCURRENCY, maxConcurrency),
|
|
428
|
+
maxConcurrency,
|
|
429
|
+
maxTasks: options?.maxTasks ?? DEFAULT_MAX_TASKS,
|
|
430
|
+
};
|
|
431
|
+
}
|
|
432
|
+
function checkWorkflowBudget(state) {
|
|
433
|
+
const budget = state.options.budget;
|
|
434
|
+
if (!budget)
|
|
435
|
+
return;
|
|
436
|
+
const elapsedMs = Date.now() - state.startedAt.getTime();
|
|
437
|
+
if (budget.maxDurationMs !== undefined && elapsedMs > budget.maxDurationMs) {
|
|
438
|
+
throw new StratusError(`Workflow exceeded maxDurationMs (${budget.maxDurationMs})`);
|
|
439
|
+
}
|
|
440
|
+
if (budget.maxPromptTokens !== undefined &&
|
|
441
|
+
state.usage.promptTokens > budget.maxPromptTokens) {
|
|
442
|
+
throw new StratusError(`Workflow exceeded maxPromptTokens (${budget.maxPromptTokens})`);
|
|
443
|
+
}
|
|
444
|
+
if (budget.maxCompletionTokens !== undefined &&
|
|
445
|
+
state.usage.completionTokens > budget.maxCompletionTokens) {
|
|
446
|
+
throw new StratusError(`Workflow exceeded maxCompletionTokens (${budget.maxCompletionTokens})`);
|
|
447
|
+
}
|
|
448
|
+
if (budget.maxTotalTokens !== undefined && state.usage.totalTokens > budget.maxTotalTokens) {
|
|
449
|
+
throw new StratusError(`Workflow exceeded maxTotalTokens (${budget.maxTotalTokens})`);
|
|
450
|
+
}
|
|
451
|
+
if (budget.maxCostUsd !== undefined && state.totalCostUsd > budget.maxCostUsd) {
|
|
452
|
+
throw new StratusError(`Workflow exceeded maxCostUsd (${budget.maxCostUsd})`);
|
|
453
|
+
}
|
|
454
|
+
}
|
|
455
|
+
function clampConcurrency(concurrency, maxConcurrency) {
|
|
456
|
+
if (!Number.isFinite(concurrency) || concurrency < 1) {
|
|
457
|
+
throw new StratusError("Workflow concurrency must be a positive number");
|
|
458
|
+
}
|
|
459
|
+
return Math.min(Math.floor(concurrency), maxConcurrency);
|
|
460
|
+
}
|
|
461
|
+
function buildResumeCache(snapshot) {
|
|
462
|
+
const cache = new Map();
|
|
463
|
+
if (!snapshot)
|
|
464
|
+
return cache;
|
|
465
|
+
for (const phase of snapshot.phases) {
|
|
466
|
+
for (const task of phase.tasks) {
|
|
467
|
+
cache.set(`${phase.id}:${task.id}`, task);
|
|
468
|
+
}
|
|
469
|
+
}
|
|
470
|
+
return cache;
|
|
471
|
+
}
|
|
472
|
+
function emptyUsage() {
|
|
473
|
+
return { promptTokens: 0, completionTokens: 0, totalTokens: 0 };
|
|
474
|
+
}
|
|
475
|
+
function addUsage(target, usage) {
|
|
476
|
+
if (!usage)
|
|
477
|
+
return;
|
|
478
|
+
target.promptTokens += usage.promptTokens;
|
|
479
|
+
target.completionTokens += usage.completionTokens;
|
|
480
|
+
target.totalTokens += usage.totalTokens;
|
|
481
|
+
target.cacheReadTokens = addOptional(target.cacheReadTokens, usage.cacheReadTokens);
|
|
482
|
+
target.cacheCreationTokens = addOptional(target.cacheCreationTokens, usage.cacheCreationTokens);
|
|
483
|
+
target.reasoningTokens = addOptional(target.reasoningTokens, usage.reasoningTokens);
|
|
484
|
+
}
|
|
485
|
+
function addOptional(current, next) {
|
|
486
|
+
if (next === undefined)
|
|
487
|
+
return current;
|
|
488
|
+
return (current ?? 0) + next;
|
|
489
|
+
}
|
|
490
|
+
function checkWorkflowAborted(signal) {
|
|
491
|
+
if (signal?.aborted)
|
|
492
|
+
throw new RunAbortedError();
|
|
493
|
+
}
|
|
494
|
+
function uniquePhaseId(name, index) {
|
|
495
|
+
return `${slugify(name)}-${index + 1}`;
|
|
496
|
+
}
|
|
497
|
+
function slugify(value) {
|
|
498
|
+
const slug = value
|
|
499
|
+
.toLowerCase()
|
|
500
|
+
.replace(/[^a-z0-9]+/g, "-")
|
|
501
|
+
.replace(/^-|-$/g, "");
|
|
502
|
+
return slug || "phase";
|
|
503
|
+
}
|
|
504
|
+
function createRunId() {
|
|
505
|
+
return `wf_${Date.now().toString(36)}_${Math.random().toString(36).slice(2, 8)}`;
|
|
506
|
+
}
|
|
507
|
+
function stringifyTaskValue(value) {
|
|
508
|
+
if (typeof value === "string")
|
|
509
|
+
return value;
|
|
510
|
+
if (value === undefined)
|
|
511
|
+
return "";
|
|
512
|
+
try {
|
|
513
|
+
return JSON.stringify(value);
|
|
514
|
+
}
|
|
515
|
+
catch {
|
|
516
|
+
return String(value);
|
|
517
|
+
}
|
|
518
|
+
}
|
|
519
|
+
function getErrorMessage(error) {
|
|
520
|
+
if (error instanceof Error)
|
|
521
|
+
return error.message;
|
|
522
|
+
return String(error);
|
|
523
|
+
}
|
|
524
|
+
function verdictAccepts(output) {
|
|
525
|
+
return /^\s*(accept|accepted|pass|passed|valid|yes)\b/i.test(output);
|
|
526
|
+
}
|
|
527
|
+
function verdictPrefersRight(output) {
|
|
528
|
+
return /^\s*(right|b|second|winner:\s*right)\b/i.test(output);
|
|
529
|
+
}
|
|
530
|
+
function createWorkflowEventChannel() {
|
|
531
|
+
const queue = [];
|
|
532
|
+
let resolve;
|
|
533
|
+
let finished = false;
|
|
534
|
+
return {
|
|
535
|
+
push(event) {
|
|
536
|
+
queue.push(event);
|
|
537
|
+
resolve?.();
|
|
538
|
+
resolve = undefined;
|
|
539
|
+
},
|
|
540
|
+
done() {
|
|
541
|
+
finished = true;
|
|
542
|
+
resolve?.();
|
|
543
|
+
resolve = undefined;
|
|
544
|
+
},
|
|
545
|
+
async *[Symbol.asyncIterator]() {
|
|
546
|
+
while (!finished || queue.length > 0) {
|
|
547
|
+
if (queue.length === 0) {
|
|
548
|
+
await new Promise((r) => {
|
|
549
|
+
resolve = r;
|
|
550
|
+
});
|
|
551
|
+
continue;
|
|
552
|
+
}
|
|
553
|
+
const event = queue.shift();
|
|
554
|
+
if (event)
|
|
555
|
+
yield event;
|
|
556
|
+
}
|
|
557
|
+
},
|
|
558
|
+
};
|
|
559
|
+
}
|
|
560
|
+
//# sourceMappingURL=workflow.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"workflow.js","sourceRoot":"","sources":["../../src/core/workflow.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,eAAe,EAAE,YAAY,EAAE,MAAM,UAAU,CAAC;AAGzD,OAAO,EAAmB,GAAG,EAAE,MAAM,OAAO,CAAC;AAG7C,MAAM,4BAA4B,GAAG,CAAC,CAAC;AACvC,MAAM,uBAAuB,GAAG,EAAE,CAAC;AACnC,MAAM,iBAAiB,GAAG,IAAI,CAAC;AAC/B,MAAM,+BAA+B,GAAG,EAAE,CAAC;AAuS3C,MAAM,UAAU,QAAQ,CACvB,MAAsC;IAEtC,OAAO,MAAM,CAAC;AACf,CAAC;AAED,MAAM,UAAU,YAAY,CAC3B,IAAoC;IAEpC,OAAO,IAAI,CAAC;AACb,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,WAAW,CAChC,EAA4B,EAC5B,IAAW,EACX,OAAyB;IAEzB,OAAO,eAAe,CAAC,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;AACrD,CAAC;AAED,MAAM,UAAU,cAAc,CAC7B,EAA4B,EAC5B,IAAW,EACX,OAAyB;IAEzB,MAAM,OAAO,GAAG,0BAA0B,EAAW,CAAC;IACtD,MAAM,MAAM,GAAG,eAAe,CAAC,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,CAC9F,OAAO,CAAC,IAAI,EAAE,CACd,CAAC;IAEF,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,EAAE,MAAM,EAAE,CAAC;AAC5D,CAAC;AAED,KAAK,UAAU,eAAe,CAC7B,EAA4B,EAC5B,IAAW,EACX,OAAoC,EACpC,IAA6C;IAE7C,MAAM,UAAU,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;IAC7C,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC;IAC7B,MAAM,KAAK,GAAyB;QACnC,EAAE,EAAE,UAAU,CAAC,EAAE,IAAI,WAAW,EAAE;QAClC,IAAI,EAAE,EAAE,CAAC,IAAI;QACb,IAAI;QACJ,OAAO,EAAE,UAAU;QACnB,SAAS;QACT,MAAM,EAAE,EAAE;QACV,UAAU,EAAE,CAAC;QACb,KAAK,EAAE,UAAU,EAAE;QACnB,QAAQ,EAAE,CAAC;QACX,YAAY,EAAE,CAAC;QACf,IAAI,EAAE,IAA+C;QACrD,WAAW,EAAE,gBAAgB,CAAC,UAAU,CAAC,UAAU,CAAC;KACpD,CAAC;IAEF,MAAM,GAAG,GAAG,IAAI,6BAA6B,CAAC,KAAK,CAAC,CAAC;IACrD,IAAI,CAAC,EAAE,IAAI,EAAE,kBAAkB,EAAE,KAAK,EAAE,KAAK,CAAC,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC;IAEnE,IAAI,CAAC;QACJ,oBAAoB,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;QACxC,mBAAmB,CAAC,KAAK,CAAC,CAAC;QAC3B,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;QACvC,mBAAmB,CAAC,KAAK,CAAC,CAAC;QAC3B,MAAM,WAAW,GAAG,IAAI,IAAI,EAAE,CAAC;QAC/B,MAAM,MAAM,GAA+B;YAC1C,EAAE,EAAE,KAAK,CAAC,EAAE;YACZ,IAAI,EAAE,EAAE,CAAC,IAAI;YACb,MAAM,EAAE,WAAW;YACnB,MAAM;YACN,MAAM,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,CAAC;YACzB,KAAK,EAAE,KAAK,CAAC,KAAK;YAClB,QAAQ,EAAE,KAAK,CAAC,QAAQ;YACxB,YAAY,EAAE,KAAK,CAAC,YAAY;YAChC,SAAS,EAAE,SAAS,CAAC,WAAW,EAAE;YAClC,WAAW,EAAE,WAAW,CAAC,WAAW,EAAE;YACtC,UAAU,EAAE,WAAW,CAAC,OAAO,EAAE,GAAG,SAAS,CAAC,OAAO,EAAE;SACvD,CAAC;QACF,IAAI,CAAC,EAAE,IAAI,EAAE,oBAAoB,EAAE,KAAK,EAAE,KAAK,CAAC,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;QAC9D,OAAO,MAAM,CAAC;IACf,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAChB,MAAM,OAAO,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC;QACvC,IAAI,CAAC,EAAE,IAAI,EAAE,iBAAiB,EAAE,KAAK,EAAE,KAAK,CAAC,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC;QACnE,MAAM,KAAK,CAAC;IACb,CAAC;AACF,CAAC;AAED,MAAM,6BAA6B;IAKL;IAJpB,KAAK,CAAS;IACd,IAAI,CAAS;IACb,IAAI,CAAQ;IAErB,YAA6B,KAA2B;QAA3B,UAAK,GAAL,KAAK,CAAsB;QACvD,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC,EAAE,CAAC;QACtB,IAAI,CAAC,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC;QACvB,IAAI,CAAC,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC;IACxB,CAAC;IAED,IAAI,QAAQ;QACX,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,OAAO;YACN,EAAE,EAAE,IAAI,CAAC,KAAK,CAAC,EAAE;YACjB,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI;YACrB,MAAM,EAAE,SAAS;YACjB,MAAM,EAAE,EAAE;YACV,MAAM,EAAE,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC;YAC9B,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK;YACvB,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,QAAQ;YAC7B,YAAY,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY;YACrC,SAAS,EAAE,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,WAAW,EAAE;YAC7C,WAAW,EAAE,GAAG,CAAC,WAAW,EAAE;YAC9B,UAAU,EAAE,GAAG,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,OAAO,EAAE;SAC1D,CAAC;IACH,CAAC;IAED,KAAK,CAAC,KAAK,CACV,IAAY,EACZ,KAAuC,EACvC,OAA8B;QAE9B,oBAAoB,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAChD,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACxB,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;YACvB,MAAM,UAAU,GAAwB;gBACvC,EAAE,EAAE,OAAO,CAAC,IAAI,CAAC;gBACjB,IAAI;gBACJ,MAAM,EAAE,WAAW;gBACnB,KAAK,EAAE,EAAE;gBACT,SAAS,EAAE,GAAG,CAAC,WAAW,EAAE;gBAC5B,WAAW,EAAE,GAAG,CAAC,WAAW,EAAE;gBAC9B,UAAU,EAAE,CAAC;aACb,CAAC;YACF,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YACnC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,wBAAwB,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;YAC1F,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,0BAA0B,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;YAC5F,OAAO,EAAE,CAAC;QACX,CAAC;QAED,IAAI,CAAC,KAAK,CAAC,UAAU,IAAI,KAAK,CAAC,MAAM,CAAC;QACtC,IAAI,IAAI,CAAC,KAAK,CAAC,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC;YACzD,MAAM,IAAI,YAAY,CACrB,+BAA+B,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,eAAe,IAAI,CAAC,KAAK,CAAC,UAAU,EAAE,CAChG,CAAC;QACH,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC;QAC7B,MAAM,OAAO,GAAG,aAAa,CAAC,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAC9D,MAAM,UAAU,GAAG,EAAE,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;QACzC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,wBAAwB,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;QAE1F,MAAM,WAAW,GAAG,gBAAgB,CACnC,OAAO,EAAE,WAAW,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,WAAW,EACtD,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,cAAc,CACjC,CAAC;QACF,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,MAAM,WAAW,CAAmB;YAChE,GAAG,EAAE,IAAI;YACT,OAAO;YACP,SAAS,EAAE,IAAI;YACf,KAAK;YACL,WAAW;YACX,QAAQ,EAAE,OAAO,EAAE,QAAQ,IAAI,IAAI;SACnC,CAAC,CAAC;QACH,MAAM,WAAW,GAAG,IAAI,IAAI,EAAE,CAAC;QAC/B,MAAM,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,KAAK,QAAQ,CAAC,CAAC;QACpE,MAAM,KAAK,GAAwB;YAClC,EAAE,EAAE,OAAO;YACX,IAAI;YACJ,MAAM,EAAE,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,WAAW;YAClD,KAAK,EAAE,OAAO;YACd,SAAS,EAAE,SAAS,CAAC,WAAW,EAAE;YAClC,WAAW,EAAE,WAAW,CAAC,WAAW,EAAE;YACtC,UAAU,EAAE,WAAW,CAAC,OAAO,EAAE,GAAG,SAAS,CAAC,OAAO,EAAE;SACvD,CAAC;QACF,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC9B,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,0BAA0B,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;QAChF,mBAAmB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAChC,IAAI,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,YAAY,CAAC,kBAAkB,OAAO,CAAC,IAAI,aAAa,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC;QACpF,CAAC;QACD,OAAO,OAAO,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,OAAO,CACZ,IAAoC,EACpC,OAAkD;QAElD,MAAM,OAAO,GAAG,OAAO,EAAE,OAAO,IAAI,OAAO,CAAC;QAC5C,MAAM,SAAS,GAAG,OAAO,EAAE,SAAS,IAAI,QAAQ,CAAC;QACjD,MAAM,MAAM,GAAG,IAAI,CAAC,EAAE,IAAI,GAAG,OAAO,IAAI,IAAI,CAAC,IAAI,IAAI,MAAM,EAAE,CAAC;QAC9D,MAAM,QAAQ,GAAG,GAAG,OAAO,IAAI,MAAM,EAAE,CAAC;QACxC,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACpD,IAAI,MAAM,EAAE,MAAM,KAAK,WAAW,IAAI,MAAM,EAAE,MAAM,KAAK,aAAa,EAAE,CAAC;YACxE,MAAM,OAAO,GAAG,EAAE,GAAG,MAAM,EAAE,MAAM,EAAE,SAAkB,EAAE,CAAC;YAC1D,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,uBAAuB,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;YAC9F,OAAO,OAAqC,CAAC;QAC9C,CAAC;QAED,oBAAoB,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAChD,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,IAAI,MAAM,CAAC;QACjC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC;YACf,IAAI,EAAE,uBAAuB;YAC7B,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,OAAO;YACP,IAAI,EAAE,EAAE,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE;SAC1B,CAAC,CAAC;QAEH,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC;QAC7B,IAAI,MAAiE,CAAC;QACtE,IAAI,CAAC;YACJ,MAAM,GAAG,MAAM,WAAW,CAAC,IAAI,EAAE;gBAChC,IAAI;gBACJ,OAAO,EAAE,IAAI,CAAC,OAAO;gBACrB,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM;gBACjC,QAAQ,EAAE,IAAI;aACd,CAAC,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,MAAM,WAAW,GAAG,IAAI,IAAI,EAAE,CAAC;YAC/B,MAAM,UAAU,GAA+B;gBAC9C,EAAE,EAAE,MAAM;gBACV,IAAI;gBACJ,OAAO;gBACP,SAAS;gBACT,MAAM,EAAE,QAAQ;gBAChB,MAAM,EAAE,EAAE;gBACV,KAAK,EAAE,eAAe,CAAC,KAAK,CAAC;gBAC7B,KAAK,EAAE,UAAU,EAAE;gBACnB,QAAQ,EAAE,CAAC;gBACX,YAAY,EAAE,CAAC;gBACf,SAAS,EAAE,SAAS,CAAC,WAAW,EAAE;gBAClC,WAAW,EAAE,WAAW,CAAC,WAAW,EAAE;gBACtC,UAAU,EAAE,WAAW,CAAC,OAAO,EAAE,GAAG,SAAS,CAAC,OAAO,EAAE;gBACvD,QAAQ,EAAE,IAAI,CAAC,QAAQ;aACvB,CAAC;YACF,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC;gBACf,IAAI,EAAE,sBAAsB;gBAC5B,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,OAAO;gBACP,IAAI,EAAE,UAAU;aAChB,CAAC,CAAC;YACH,OAAO,UAAU,CAAC;QACnB,CAAC;QAED,MAAM,WAAW,GAAG,IAAI,IAAI,EAAE,CAAC;QAC/B,MAAM,UAAU,GAA+B;YAC9C,EAAE,EAAE,MAAM;YACV,IAAI;YACJ,OAAO;YACP,SAAS;YACT,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,KAAK,EAAE,MAAM,CAAC,KAAe;YAC7B,WAAW,EAAE,MAAM,CAAC,WAAW;YAC/B,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,YAAY,EAAE,MAAM,CAAC,YAAY;YACjC,SAAS,EAAE,SAAS,CAAC,WAAW,EAAE;YAClC,WAAW,EAAE,WAAW,CAAC,WAAW,EAAE;YACtC,UAAU,EAAE,WAAW,CAAC,OAAO,EAAE,GAAG,SAAS,CAAC,OAAO,EAAE;YACvD,QAAQ,EAAE,IAAI,CAAC,QAAQ;SACvB,CAAC;QACF,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;QAC/B,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC;YACf,IAAI,EAAE,yBAAyB;YAC/B,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,OAAO;YACP,IAAI,EAAE,UAAU;SAChB,CAAC,CAAC;QACH,mBAAmB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAChC,OAAO,UAAU,CAAC;IACnB,CAAC;IAED,KAAK,CAAC,UAAU,CACf,KAA+B,EAC/B,KAA6B,EAC7B,OAAuC;QAEvC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC;YACf,IAAI,EAAE,4BAA4B;YAClC,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,SAAS,EAAE,KAAK,CAAC,IAAI;SACrB,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,KAAK,EAAE,KAAK,EAAE;YACtC,GAAG,OAAO;YACV,MAAM,EAAE,OAAO,EAAE,MAAM,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM;SACpD,CAAC,CAAC;QACH,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;QACzC,IAAI,CAAC,KAAK,CAAC,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC;QACvC,IAAI,CAAC,KAAK,CAAC,YAAY,IAAI,MAAM,CAAC,YAAY,CAAC;QAC/C,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC;YACf,IAAI,EAAE,8BAA8B;YACpC,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,MAAM,EAAE,MAAM,CAAC,MAAM;SACrB,CAAC,CAAC;QACH,mBAAmB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAChC,OAAO,MAAM,CAAC;IACf,CAAC;IAED,KAAK,CAAC,mBAAmB,CACxB,MAAoE;QAEpE,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,YAAY,CAAC,CAAC;QAC/E,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,UAAU,CACtC,MAAM,CAAC,WAAW,EAClB,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,EACpB,MAAM,CAAC,gBAAgB,CACvB,CAAC;QACF,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,MAAM,EAAE,SAAS,CAAC,MAAM,EAAE,CAAC;IACvD,CAAC;IAED,KAAK,CAAC,iBAAiB,CACtB,MAAuD;QAEvD,MAAM,KAAK,GAAG,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE,CACpD,YAAY,CAAW;YACtB,EAAE,EAAE,OAAO,CAAC,EAAE,IAAI,UAAU,KAAK,GAAG,CAAC,EAAE;YACvC,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,UAAU,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,UAAU,KAAK,GAAG,CAAC,EAAE;YACrE,KAAK,EAAE,MAAM,CAAC,QAAQ;YACtB,KAAK,EACJ,MAAM,CAAC,MAAM,EAAE,CAAC,OAAO,CAAC;gBACxB,4FAA4F,OAAO,CAAC,MAAM,EAAE;YAC7G,UAAU,EAAE,MAAM,CAAC,UAAU;YAC7B,QAAQ,EAAE,OAAO,CAAC,QAAQ;SAC1B,CAAC,CACF,CAAC;QACF,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE,KAAK,EAAE,MAAM,CAAC,YAAY,CAAC,CAAC;QAChF,MAAM,QAAQ,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,cAAc,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;QACjF,OAAO;YACN,aAAa;YACb,QAAQ;YACR,QAAQ,EAAE,aAAa,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;SACtE,CAAC;IACH,CAAC;IAED,KAAK,CAAC,iBAAiB,CACtB,MAAiD;QAEjD,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,UAAU,EAAE,MAAM,CAAC,YAAY,CAAC,CAAC;QACzF,MAAM,WAAW,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,KAAK,EAAE,EAAE,CACvD,YAAY,CAAW;YACtB,EAAE,EAAE,UAAU,SAAS,CAAC,EAAE,EAAE;YAC5B,IAAI,EAAE,UAAU,SAAS,CAAC,IAAI,EAAE;YAChC,KAAK,EAAE,MAAM,CAAC,MAAM;YACpB,KAAK,EACJ,MAAM,CAAC,MAAM,EAAE,CAAC,SAAS,EAAE,KAAK,CAAC;gBACjC,kFAAkF,SAAS,CAAC,MAAM,EAAE;YACrG,UAAU,EAAE,MAAM,CAAC,UAAU;YAC7B,QAAQ,EAAE,SAAS,CAAC,QAAQ;SAC5B,CAAC,CACF,CAAC;QACF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,GAAG,MAAM,CAAC,IAAI,SAAS,EAAE,WAAW,EAAE,MAAM,CAAC,YAAY,CAAC,CAAC;QAC7F,OAAO;YACN,UAAU;YACV,QAAQ;YACR,QAAQ,EAAE,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC,cAAc,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,MAAM,IAAI,EAAE,CAAC,CAAC;SACxF,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU,CACf,MAA0C;QAE1C,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,WAAW,EAAE,MAAM,CAAC,YAAY,CAAC,CAAC;QAC3F,IAAI,KAAK,GAAG,WAAW,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,KAAK,WAAW,CAAC,CAAC;QAC1E,MAAM,OAAO,GAAyB,EAAE,CAAC;QACzC,IAAI,UAAU,GAAG,CAAC,CAAC;QAEnB,OAAO,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACzB,MAAM,UAAU,GAA6B,EAAE,CAAC;YAChD,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,KAAK,CAAC,MAAM,EAAE,KAAK,IAAI,CAAC,EAAE,CAAC;gBACtD,MAAM,IAAI,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC;gBAC1B,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC;gBAC/B,IAAI,CAAC,IAAI,IAAI,CAAC,KAAK;oBAAE,SAAS;gBAC9B,UAAU,CAAC,IAAI,CACd,YAAY,CAAW;oBACtB,EAAE,EAAE,SAAS,UAAU,UAAU,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,CAAC,CAAC,GAAG,CAAC,EAAE;oBAC5D,IAAI,EAAE,GAAG,IAAI,CAAC,IAAI,OAAO,KAAK,CAAC,IAAI,EAAE;oBACrC,KAAK,EAAE,MAAM,CAAC,KAAK;oBACnB,KAAK,EACJ,MAAM,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,KAAK,CAAC;wBAC5B,qFAAqF,IAAI,CAAC,MAAM,eAAe,KAAK,CAAC,MAAM,EAAE;oBAC9H,UAAU,EAAE,MAAM,CAAC,UAAU;oBAC7B,QAAQ,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,OAAO,EAAE,KAAK,CAAC,EAAE,EAAE;iBAChD,CAAC,CACF,CAAC;YACH,CAAC;YAED,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,GAAG,MAAM,CAAC,IAAI,UAAU,UAAU,EAAE,EAAE,UAAU,EAAE;gBACvF,GAAG,MAAM,CAAC,YAAY;gBACtB,QAAQ,EAAE,MAAM,CAAC,YAAY,EAAE,QAAQ,IAAI,KAAK;aAChD,CAAC,CAAC;YACH,OAAO,CAAC,IAAI,CAAC,GAAG,YAAY,CAAC,CAAC;YAC9B,MAAM,OAAO,GAAyB,EAAE,CAAC;YACzC,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,KAAK,CAAC,MAAM,EAAE,KAAK,IAAI,CAAC,EAAE,CAAC;gBACtD,MAAM,IAAI,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC;gBAC1B,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC;gBAC/B,IAAI,CAAC,IAAI;oBAAE,SAAS;gBACpB,IAAI,CAAC,KAAK,EAAE,CAAC;oBACZ,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;oBACnB,SAAS;gBACV,CAAC;gBACD,MAAM,KAAK,GAAG,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC;gBAClD,OAAO,CAAC,IAAI,CAAC,mBAAmB,CAAC,KAAK,EAAE,MAAM,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;YACvE,CAAC;YACD,KAAK,GAAG,OAAO,CAAC;YAChB,UAAU,EAAE,CAAC;QACd,CAAC;QAED,OAAO,EAAE,WAAW,EAAE,OAAO,EAAE,MAAM,EAAE,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;IACnD,CAAC;IAED,KAAK,CAAC,aAAa,CAClB,MAAkD;QAElD,MAAM,aAAa,GAAG,MAAM,CAAC,aAAa,IAAI,+BAA+B,CAAC;QAC9E,MAAM,UAAU,GAAqC,EAAE,CAAC;QACxD,KAAK,IAAI,SAAS,GAAG,CAAC,EAAE,SAAS,IAAI,aAAa,EAAE,SAAS,EAAE,EAAE,CAAC;YACjE,oBAAoB,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YAChD,mBAAmB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAChC,MAAM,QAAQ,GAAG,UAAU,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;YACnC,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE,IAAI,EAAE,SAAS,EAAE,QAAQ,EAAE,CAAC,CAAC;YACnE,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACtB,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;gBACf,OAAO,EAAE,UAAU,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,IAAI,kBAAkB,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;YAC1F,CAAC;QACF,CAAC;QACD,OAAO;YACN,UAAU;YACV,IAAI,EAAE,KAAK;YACX,MAAM,EAAE,iBAAiB,aAAa,iDAAiD;SACvF,CAAC;IACH,CAAC;IAEO,aAAa,CAAC,MAA0B;QAC/C,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;QACzC,IAAI,CAAC,KAAK,CAAC,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC;QACvC,IAAI,CAAC,KAAK,CAAC,YAAY,IAAI,MAAM,CAAC,YAAY,CAAC;IAChD,CAAC;CACD;AAED,KAAK,UAAU,WAAW,CACzB,IAAoC,EACpC,MAA2C;IAW3C,IAAI,SAAS,IAAI,IAAI,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;QACvC,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QACzC,OAAO;YACN,MAAM,EAAE,WAAW;YACnB,MAAM,EAAE,kBAAkB,CAAC,KAAK,CAAC;YACjC,KAAK,EAAE,KAAe;YACtB,KAAK,EAAE,UAAU,EAAE;YACnB,QAAQ,EAAE,CAAC;YACX,YAAY,EAAE,CAAC;SACf,CAAC;IACH,CAAC;IAED,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE;QAChD,GAAG,IAAI,CAAC,UAAU;QAClB,OAAO,EAAE,IAAI,CAAC,OAAO;QACrB,KAAK,EAAE,IAAI,CAAC,KAAK;QACjB,QAAQ,EAAE,IAAI,CAAC,QAAQ;QACvB,MAAM,EAAE,MAAM,CAAC,MAAM;KACrB,CAAC,CAAC;IAEH,OAAO;QACN,MAAM,EAAE,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,WAAW;QACxD,MAAM,EAAE,MAAM,CAAC,MAAM;QACrB,KAAK,EAAE,MAAM,CAAC,MAAgB;QAC9B,WAAW,EAAE,MAAM,CAAC,WAAW;QAC/B,SAAS,EAAE,MAAM;QACjB,KAAK,EAAE,MAAM,CAAC,KAAK;QACnB,QAAQ,EAAE,MAAM,CAAC,QAAQ;QACzB,YAAY,EAAE,MAAM,CAAC,YAAY;KACjC,CAAC;AACH,CAAC;AAED,KAAK,UAAU,WAAW,CAAmB,MAO5C;IAIA,MAAM,OAAO,GAAiC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;IAC7E,IAAI,IAAI,GAAG,CAAC,CAAC;IACb,IAAI,OAA+C,CAAC;IAEpD,KAAK,UAAU,MAAM;QACpB,OAAO,IAAI,GAAG,MAAM,CAAC,KAAK,CAAC,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;YAC/C,MAAM,KAAK,GAAG,IAAI,EAAE,CAAC;YACrB,MAAM,IAAI,GAAG,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YACjC,IAAI,CAAC,IAAI;gBAAE,OAAO;YAClB,MAAM,YAAY,GACjB,IAAI,CAAC,EAAE,KAAK,SAAS;gBACpB,CAAC,CAAC,EAAE,GAAG,IAAI,EAAE,EAAE,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,IAAI,IAAI,MAAM,CAAC,IAAI,KAAK,GAAG,CAAC,EAAE,EAAE;gBACjE,CAAC,CAAC,IAAI,CAAC;YACT,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,YAAY,EAAE;gBACrD,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,SAAS,EAAE,MAAM,CAAC,SAAS;aAC3B,CAAC,CAAC;YACH,OAAO,CAAC,KAAK,CAAC,GAAG,MAAM,CAAC;YACxB,IAAI,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;gBACnD,OAAO,GAAG,MAAM,CAAC;YAClB,CAAC;QACF,CAAC;IACF,CAAC;IAED,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,WAAW,EAAE,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;IACtE,MAAM,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,WAAW,EAAE,EAAE,GAAG,EAAE,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;IAEvE,MAAM,gBAAgB,GAAG,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IACjD,OAAO,EAAE,OAAO,EAAE,gBAAgB,EAAE,OAAO,EAAE,CAAC;AAC/C,CAAC;AAED,SAAS,gBAAgB,CAAC,OAAyB;IAClD,MAAM,cAAc,GAAG,OAAO,EAAE,cAAc,IAAI,uBAAuB,CAAC;IAC1E,OAAO;QACN,GAAG,OAAO;QACV,WAAW,EAAE,gBAAgB,CAC5B,OAAO,EAAE,WAAW,IAAI,4BAA4B,EACpD,cAAc,CACd;QACD,cAAc;QACd,QAAQ,EAAE,OAAO,EAAE,QAAQ,IAAI,iBAAiB;KAChD,CAAC;AACH,CAAC;AAED,SAAS,mBAAmB,CAAC,KAA6B;IACzD,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC;IACpC,IAAI,CAAC,MAAM;QAAE,OAAO;IACpB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC;IACzD,IAAI,MAAM,CAAC,aAAa,KAAK,SAAS,IAAI,SAAS,GAAG,MAAM,CAAC,aAAa,EAAE,CAAC;QAC5E,MAAM,IAAI,YAAY,CAAC,oCAAoC,MAAM,CAAC,aAAa,GAAG,CAAC,CAAC;IACrF,CAAC;IACD,IACC,MAAM,CAAC,eAAe,KAAK,SAAS;QACpC,KAAK,CAAC,KAAK,CAAC,YAAY,GAAG,MAAM,CAAC,eAAe,EAChD,CAAC;QACF,MAAM,IAAI,YAAY,CAAC,sCAAsC,MAAM,CAAC,eAAe,GAAG,CAAC,CAAC;IACzF,CAAC;IACD,IACC,MAAM,CAAC,mBAAmB,KAAK,SAAS;QACxC,KAAK,CAAC,KAAK,CAAC,gBAAgB,GAAG,MAAM,CAAC,mBAAmB,EACxD,CAAC;QACF,MAAM,IAAI,YAAY,CACrB,0CAA0C,MAAM,CAAC,mBAAmB,GAAG,CACvE,CAAC;IACH,CAAC;IACD,IAAI,MAAM,CAAC,cAAc,KAAK,SAAS,IAAI,KAAK,CAAC,KAAK,CAAC,WAAW,GAAG,MAAM,CAAC,cAAc,EAAE,CAAC;QAC5F,MAAM,IAAI,YAAY,CAAC,qCAAqC,MAAM,CAAC,cAAc,GAAG,CAAC,CAAC;IACvF,CAAC;IACD,IAAI,MAAM,CAAC,UAAU,KAAK,SAAS,IAAI,KAAK,CAAC,YAAY,GAAG,MAAM,CAAC,UAAU,EAAE,CAAC;QAC/E,MAAM,IAAI,YAAY,CAAC,iCAAiC,MAAM,CAAC,UAAU,GAAG,CAAC,CAAC;IAC/E,CAAC;AACF,CAAC;AAED,SAAS,gBAAgB,CAAC,WAAmB,EAAE,cAAsB;IACpE,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,WAAW,GAAG,CAAC,EAAE,CAAC;QACtD,MAAM,IAAI,YAAY,CAAC,gDAAgD,CAAC,CAAC;IAC1E,CAAC;IACD,OAAO,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,EAAE,cAAc,CAAC,CAAC;AAC1D,CAAC;AAED,SAAS,gBAAgB,CACxB,QAAuC;IAEvC,MAAM,KAAK,GAAG,IAAI,GAAG,EAA8B,CAAC;IACpD,IAAI,CAAC,QAAQ;QAAE,OAAO,KAAK,CAAC;IAC5B,KAAK,MAAM,KAAK,IAAI,QAAQ,CAAC,MAAM,EAAE,CAAC;QACrC,KAAK,MAAM,IAAI,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;YAChC,KAAK,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC,EAAE,IAAI,IAAI,CAAC,EAAE,EAAE,EAAE,IAAI,CAAC,CAAC;QAC3C,CAAC;IACF,CAAC;IACD,OAAO,KAAK,CAAC;AACd,CAAC;AAED,SAAS,UAAU;IAClB,OAAO,EAAE,YAAY,EAAE,CAAC,EAAE,gBAAgB,EAAE,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE,CAAC;AACjE,CAAC;AAED,SAAS,QAAQ,CAAC,MAAiB,EAAE,KAA4B;IAChE,IAAI,CAAC,KAAK;QAAE,OAAO;IACnB,MAAM,CAAC,YAAY,IAAI,KAAK,CAAC,YAAY,CAAC;IAC1C,MAAM,CAAC,gBAAgB,IAAI,KAAK,CAAC,gBAAgB,CAAC;IAClD,MAAM,CAAC,WAAW,IAAI,KAAK,CAAC,WAAW,CAAC;IACxC,MAAM,CAAC,eAAe,GAAG,WAAW,CAAC,MAAM,CAAC,eAAe,EAAE,KAAK,CAAC,eAAe,CAAC,CAAC;IACpF,MAAM,CAAC,mBAAmB,GAAG,WAAW,CAAC,MAAM,CAAC,mBAAmB,EAAE,KAAK,CAAC,mBAAmB,CAAC,CAAC;IAChG,MAAM,CAAC,eAAe,GAAG,WAAW,CAAC,MAAM,CAAC,eAAe,EAAE,KAAK,CAAC,eAAe,CAAC,CAAC;AACrF,CAAC;AAED,SAAS,WAAW,CAAC,OAA2B,EAAE,IAAwB;IACzE,IAAI,IAAI,KAAK,SAAS;QAAE,OAAO,OAAO,CAAC;IACvC,OAAO,CAAC,OAAO,IAAI,CAAC,CAAC,GAAG,IAAI,CAAC;AAC9B,CAAC;AAED,SAAS,oBAAoB,CAAC,MAAoB;IACjD,IAAI,MAAM,EAAE,OAAO;QAAE,MAAM,IAAI,eAAe,EAAE,CAAC;AAClD,CAAC;AAED,SAAS,aAAa,CAAC,IAAY,EAAE,KAAa;IACjD,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,IAAI,KAAK,GAAG,CAAC,EAAE,CAAC;AACxC,CAAC;AAED,SAAS,OAAO,CAAC,KAAa;IAC7B,MAAM,IAAI,GAAG,KAAK;SAChB,WAAW,EAAE;SACb,OAAO,CAAC,aAAa,EAAE,GAAG,CAAC;SAC3B,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;IACxB,OAAO,IAAI,IAAI,OAAO,CAAC;AACxB,CAAC;AAED,SAAS,WAAW;IACnB,OAAO,MAAM,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;AAClF,CAAC;AAED,SAAS,kBAAkB,CAAC,KAAc;IACzC,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,KAAK,CAAC;IAC5C,IAAI,KAAK,KAAK,SAAS;QAAE,OAAO,EAAE,CAAC;IACnC,IAAI,CAAC;QACJ,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;IAC9B,CAAC;IAAC,MAAM,CAAC;QACR,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC;IACtB,CAAC;AACF,CAAC;AAED,SAAS,eAAe,CAAC,KAAc;IACtC,IAAI,KAAK,YAAY,KAAK;QAAE,OAAO,KAAK,CAAC,OAAO,CAAC;IACjD,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC;AACtB,CAAC;AAED,SAAS,cAAc,CAAC,MAAc;IACrC,OAAO,gDAAgD,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;AACtE,CAAC;AAED,SAAS,mBAAmB,CAAC,MAAc;IAC1C,OAAO,yCAAyC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;AAC/D,CAAC;AAED,SAAS,0BAA0B;IAKlC,MAAM,KAAK,GAA6B,EAAE,CAAC;IAC3C,IAAI,OAAiC,CAAC;IACtC,IAAI,QAAQ,GAAG,KAAK,CAAC;IAErB,OAAO;QACN,IAAI,CAAC,KAAK;YACT,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAClB,OAAO,EAAE,EAAE,CAAC;YACZ,OAAO,GAAG,SAAS,CAAC;QACrB,CAAC;QACD,IAAI;YACH,QAAQ,GAAG,IAAI,CAAC;YAChB,OAAO,EAAE,EAAE,CAAC;YACZ,OAAO,GAAG,SAAS,CAAC;QACrB,CAAC;QACD,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,aAAa,CAAC;YAC5B,OAAO,CAAC,QAAQ,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACtC,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;oBACxB,MAAM,IAAI,OAAO,CAAO,CAAC,CAAC,EAAE,EAAE;wBAC7B,OAAO,GAAG,CAAC,CAAC;oBACb,CAAC,CAAC,CAAC;oBACH,SAAS;gBACV,CAAC;gBACD,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,EAAE,CAAC;gBAC5B,IAAI,KAAK;oBAAE,MAAM,KAAK,CAAC;YACxB,CAAC;QACF,CAAC;KACD,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1,39 @@
|
|
|
1
|
+
import { Effect, type Layer } from "effect";
|
|
2
|
+
import type { z } from "zod";
|
|
3
|
+
import type { Agent } from "../core/agent";
|
|
4
|
+
import type { Model, ModelRequest, ModelRequestOptions, ModelResponse, StreamEvent } from "../core/model";
|
|
5
|
+
import type { InterruptedRunResult, RunResult } from "../core/result";
|
|
6
|
+
import { type RunOptions, type StreamedRunResult, type StreamOptions, type ToolApproval } from "../core/run";
|
|
7
|
+
import { type FunctionTool, type ToolExecuteOptions } from "../core/tool";
|
|
8
|
+
import type { ChatMessage } from "../core/types";
|
|
9
|
+
declare const StratusEffectError_base: new <A extends Record<string, any> = {}>(args: import("effect/Types").VoidIfEmpty<{ readonly [P in keyof A as P extends "_tag" ? never : P]: A[P]; }>) => import("effect/Cause").YieldableError & {
|
|
10
|
+
readonly _tag: "StratusEffectError";
|
|
11
|
+
} & Readonly<A>;
|
|
12
|
+
export declare class StratusEffectError extends StratusEffectError_base<{
|
|
13
|
+
readonly message: string;
|
|
14
|
+
readonly cause: unknown;
|
|
15
|
+
}> {
|
|
16
|
+
}
|
|
17
|
+
export interface EffectToolConfig<TParams, TContext = unknown, Requirements = never, Error = unknown> {
|
|
18
|
+
name: string;
|
|
19
|
+
description: string;
|
|
20
|
+
parameters: z.ZodType<TParams>;
|
|
21
|
+
execute: (context: TContext, params: TParams, options?: ToolExecuteOptions) => Effect.Effect<string, Error, Requirements>;
|
|
22
|
+
layer?: Layer.Layer<Requirements, unknown, never>;
|
|
23
|
+
timeout?: number;
|
|
24
|
+
isEnabled?: boolean | ((context: TContext) => boolean | Promise<boolean>);
|
|
25
|
+
needsApproval?: boolean | ((params: TParams, context: TContext) => boolean | Promise<boolean>);
|
|
26
|
+
retries?: FunctionTool<TParams, TContext>["retries"];
|
|
27
|
+
}
|
|
28
|
+
export interface EffectModelConfig<Requirements = never, Error = unknown> {
|
|
29
|
+
getResponse: (request: ModelRequest, options?: ModelRequestOptions) => Effect.Effect<ModelResponse, Error, Requirements>;
|
|
30
|
+
getStreamedResponse?: (request: ModelRequest, options?: ModelRequestOptions) => Effect.Effect<AsyncIterable<StreamEvent>, Error, Requirements>;
|
|
31
|
+
layer?: Layer.Layer<Requirements, unknown, never>;
|
|
32
|
+
}
|
|
33
|
+
export declare function effectTool<TParams, TContext = unknown, Requirements = never, Error = unknown>(config: EffectToolConfig<TParams, TContext, Requirements, Error>): FunctionTool<TParams, TContext>;
|
|
34
|
+
export declare function effectModel<Requirements = never, Error = unknown>(config: EffectModelConfig<Requirements, Error>): Model;
|
|
35
|
+
export declare function runEffect<TContext, TOutput = undefined>(agent: Agent<TContext, TOutput>, input: string | ChatMessage[], options?: RunOptions<TContext, TOutput>): Effect.Effect<RunResult<TOutput> | InterruptedRunResult<TOutput>, StratusEffectError, never>;
|
|
36
|
+
export declare function resumeRunEffect<TContext, TOutput = undefined>(interrupted: InterruptedRunResult<TOutput>, approvals: ToolApproval[], options?: RunOptions<TContext, TOutput>): Effect.Effect<RunResult<TOutput> | InterruptedRunResult<TOutput>, StratusEffectError, never>;
|
|
37
|
+
export declare function streamEffect<TContext, TOutput = undefined>(agent: Agent<TContext, TOutput>, input: string | ChatMessage[], options?: StreamOptions<TContext, TOutput>): Effect.Effect<StreamedRunResult<TOutput>, StratusEffectError, never>;
|
|
38
|
+
export {};
|
|
39
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/effect/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAQ,MAAM,EAAE,KAAK,KAAK,EAAE,MAAM,QAAQ,CAAC;AAClD,OAAO,KAAK,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAC7B,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,eAAe,CAAC;AAC3C,OAAO,KAAK,EACX,KAAK,EACL,YAAY,EACZ,mBAAmB,EACnB,aAAa,EACb,WAAW,EACX,MAAM,eAAe,CAAC;AACvB,OAAO,KAAK,EAAE,oBAAoB,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AACtE,OAAO,EACN,KAAK,UAAU,EAGf,KAAK,iBAAiB,EACtB,KAAK,aAAa,EAElB,KAAK,YAAY,EACjB,MAAM,aAAa,CAAC;AACrB,OAAO,EAAE,KAAK,YAAY,EAAE,KAAK,kBAAkB,EAAQ,MAAM,cAAc,CAAC;AAChF,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;;;;AAEjD,qBAAa,kBAAmB,SAAQ,wBAAuC;IAC9E,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC;IACzB,QAAQ,CAAC,KAAK,EAAE,OAAO,CAAC;CACxB,CAAC;CAAG;AAEL,MAAM,WAAW,gBAAgB,CAChC,OAAO,EACP,QAAQ,GAAG,OAAO,EAClB,YAAY,GAAG,KAAK,EACpB,KAAK,GAAG,OAAO;IAEf,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,UAAU,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;IAC/B,OAAO,EAAE,CACR,OAAO,EAAE,QAAQ,EACjB,MAAM,EAAE,OAAO,EACf,OAAO,CAAC,EAAE,kBAAkB,KACxB,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,KAAK,EAAE,YAAY,CAAC,CAAC;IAChD,KAAK,CAAC,EAAE,KAAK,CAAC,KAAK,CAAC,YAAY,EAAE,OAAO,EAAE,KAAK,CAAC,CAAC;IAClD,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,SAAS,CAAC,EAAE,OAAO,GAAG,CAAC,CAAC,OAAO,EAAE,QAAQ,KAAK,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC;IAC1E,aAAa,CAAC,EAAE,OAAO,GAAG,CAAC,CAAC,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,QAAQ,KAAK,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC;IAC/F,OAAO,CAAC,EAAE,YAAY,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC,SAAS,CAAC,CAAC;CACrD;AAED,MAAM,WAAW,iBAAiB,CAAC,YAAY,GAAG,KAAK,EAAE,KAAK,GAAG,OAAO;IACvE,WAAW,EAAE,CACZ,OAAO,EAAE,YAAY,EACrB,OAAO,CAAC,EAAE,mBAAmB,KACzB,MAAM,CAAC,MAAM,CAAC,aAAa,EAAE,KAAK,EAAE,YAAY,CAAC,CAAC;IACvD,mBAAmB,CAAC,EAAE,CACrB,OAAO,EAAE,YAAY,EACrB,OAAO,CAAC,EAAE,mBAAmB,KACzB,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,WAAW,CAAC,EAAE,KAAK,EAAE,YAAY,CAAC,CAAC;IACpE,KAAK,CAAC,EAAE,KAAK,CAAC,KAAK,CAAC,YAAY,EAAE,OAAO,EAAE,KAAK,CAAC,CAAC;CAClD;AA4DD,wBAAgB,UAAU,CAAC,OAAO,EAAE,QAAQ,GAAG,OAAO,EAAE,YAAY,GAAG,KAAK,EAAE,KAAK,GAAG,OAAO,EAC5F,MAAM,EAAE,gBAAgB,CAAC,OAAO,EAAE,QAAQ,EAAE,YAAY,EAAE,KAAK,CAAC,GAC9D,YAAY,CAAC,OAAO,EAAE,QAAQ,CAAC,CAYjC;AAED,wBAAgB,WAAW,CAAC,YAAY,GAAG,KAAK,EAAE,KAAK,GAAG,OAAO,EAChE,MAAM,EAAE,iBAAiB,CAAC,YAAY,EAAE,KAAK,CAAC,GAC5C,KAAK,CAiBP;AAED,wBAAgB,SAAS,CAAC,QAAQ,EAAE,OAAO,GAAG,SAAS,EACtD,KAAK,EAAE,KAAK,CAAC,QAAQ,EAAE,OAAO,CAAC,EAC/B,KAAK,EAAE,MAAM,GAAG,WAAW,EAAE,EAC7B,OAAO,CAAC,EAAE,UAAU,CAAC,QAAQ,EAAE,OAAO,CAAC,GACrC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,OAAO,CAAC,GAAG,oBAAoB,CAAC,OAAO,CAAC,EAAE,kBAAkB,EAAE,KAAK,CAAC,CAK9F;AAED,wBAAgB,eAAe,CAAC,QAAQ,EAAE,OAAO,GAAG,SAAS,EAC5D,WAAW,EAAE,oBAAoB,CAAC,OAAO,CAAC,EAC1C,SAAS,EAAE,YAAY,EAAE,EACzB,OAAO,CAAC,EAAE,UAAU,CAAC,QAAQ,EAAE,OAAO,CAAC,GACrC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,OAAO,CAAC,GAAG,oBAAoB,CAAC,OAAO,CAAC,EAAE,kBAAkB,EAAE,KAAK,CAAC,CAM9F;AAED,wBAAgB,YAAY,CAAC,QAAQ,EAAE,OAAO,GAAG,SAAS,EACzD,KAAK,EAAE,KAAK,CAAC,QAAQ,EAAE,OAAO,CAAC,EAC/B,KAAK,EAAE,MAAM,GAAG,WAAW,EAAE,EAC7B,OAAO,CAAC,EAAE,aAAa,CAAC,QAAQ,EAAE,OAAO,CAAC,GACxC,MAAM,CAAC,MAAM,CAAC,iBAAiB,CAAC,OAAO,CAAC,EAAE,kBAAkB,EAAE,KAAK,CAAC,CAKtE"}
|