agent-relay 4.0.35 → 4.0.37
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 +3 -6
- package/dist/index.cjs +776 -41
- package/dist/src/cli/bootstrap.js.map +1 -1
- package/dist/src/cli/commands/cloud.js.map +1 -1
- package/dist/src/cli/commands/core.d.ts.map +1 -1
- package/dist/src/cli/commands/core.js.map +1 -1
- package/dist/src/cli/commands/messaging.d.ts.map +1 -1
- package/dist/src/cli/commands/messaging.js.map +1 -1
- package/dist/src/cli/commands/monitoring.d.ts.map +1 -1
- package/dist/src/cli/commands/monitoring.js.map +1 -1
- package/node_modules/@agent-relay/cloud/dist/workflows.d.ts +12 -0
- package/node_modules/@agent-relay/cloud/dist/workflows.d.ts.map +1 -1
- package/node_modules/@agent-relay/cloud/dist/workflows.js +38 -0
- package/node_modules/@agent-relay/cloud/dist/workflows.js.map +1 -1
- package/node_modules/@agent-relay/cloud/package.json +2 -2
- package/node_modules/@agent-relay/config/package.json +2 -2
- package/node_modules/@agent-relay/hooks/package.json +4 -4
- package/node_modules/@agent-relay/sdk/dist/relay.d.ts.map +1 -1
- package/node_modules/@agent-relay/sdk/dist/relay.js.map +1 -1
- package/node_modules/@agent-relay/sdk/dist/workflows/__tests__/budget-enforcement.test.d.ts +2 -0
- package/node_modules/@agent-relay/sdk/dist/workflows/__tests__/budget-enforcement.test.d.ts.map +1 -0
- package/node_modules/@agent-relay/sdk/dist/workflows/__tests__/budget-enforcement.test.js +437 -0
- package/node_modules/@agent-relay/sdk/dist/workflows/__tests__/budget-enforcement.test.js.map +1 -0
- package/node_modules/@agent-relay/sdk/dist/workflows/__tests__/budget-tracker.test.d.ts +2 -0
- package/node_modules/@agent-relay/sdk/dist/workflows/__tests__/budget-tracker.test.d.ts.map +1 -0
- package/node_modules/@agent-relay/sdk/dist/workflows/__tests__/budget-tracker.test.js +99 -0
- package/node_modules/@agent-relay/sdk/dist/workflows/__tests__/budget-tracker.test.js.map +1 -0
- package/node_modules/@agent-relay/sdk/dist/workflows/__tests__/proxy-env.test.d.ts +2 -0
- package/node_modules/@agent-relay/sdk/dist/workflows/__tests__/proxy-env.test.d.ts.map +1 -0
- package/node_modules/@agent-relay/sdk/dist/workflows/__tests__/proxy-env.test.js +135 -0
- package/node_modules/@agent-relay/sdk/dist/workflows/__tests__/proxy-env.test.js.map +1 -0
- package/node_modules/@agent-relay/sdk/dist/workflows/__tests__/verification-custom.test.d.ts +2 -0
- package/node_modules/@agent-relay/sdk/dist/workflows/__tests__/verification-custom.test.d.ts.map +1 -0
- package/node_modules/@agent-relay/sdk/dist/workflows/__tests__/verification-custom.test.js +236 -0
- package/node_modules/@agent-relay/sdk/dist/workflows/__tests__/verification-custom.test.js.map +1 -0
- package/node_modules/@agent-relay/sdk/dist/workflows/__tests__/verification-traceback.test.d.ts +2 -0
- package/node_modules/@agent-relay/sdk/dist/workflows/__tests__/verification-traceback.test.d.ts.map +1 -0
- package/node_modules/@agent-relay/sdk/dist/workflows/__tests__/verification-traceback.test.js +448 -0
- package/node_modules/@agent-relay/sdk/dist/workflows/__tests__/verification-traceback.test.js.map +1 -0
- package/node_modules/@agent-relay/sdk/dist/workflows/__tests__/verification.test.js +71 -4
- package/node_modules/@agent-relay/sdk/dist/workflows/__tests__/verification.test.js.map +1 -1
- package/node_modules/@agent-relay/sdk/dist/workflows/budget-tracker.d.ts +75 -0
- package/node_modules/@agent-relay/sdk/dist/workflows/budget-tracker.d.ts.map +1 -0
- package/node_modules/@agent-relay/sdk/dist/workflows/budget-tracker.js +180 -0
- package/node_modules/@agent-relay/sdk/dist/workflows/budget-tracker.js.map +1 -0
- package/node_modules/@agent-relay/sdk/dist/workflows/builder.d.ts +1 -0
- package/node_modules/@agent-relay/sdk/dist/workflows/builder.d.ts.map +1 -1
- package/node_modules/@agent-relay/sdk/dist/workflows/builder.js +17 -2
- package/node_modules/@agent-relay/sdk/dist/workflows/builder.js.map +1 -1
- package/node_modules/@agent-relay/sdk/dist/workflows/index.d.ts +2 -0
- package/node_modules/@agent-relay/sdk/dist/workflows/index.d.ts.map +1 -1
- package/node_modules/@agent-relay/sdk/dist/workflows/index.js +2 -0
- package/node_modules/@agent-relay/sdk/dist/workflows/index.js.map +1 -1
- package/node_modules/@agent-relay/sdk/dist/workflows/proxy-env.d.ts +52 -0
- package/node_modules/@agent-relay/sdk/dist/workflows/proxy-env.d.ts.map +1 -0
- package/node_modules/@agent-relay/sdk/dist/workflows/proxy-env.js +92 -0
- package/node_modules/@agent-relay/sdk/dist/workflows/proxy-env.js.map +1 -0
- package/node_modules/@agent-relay/sdk/dist/workflows/run-summary-table.d.ts +2 -1
- package/node_modules/@agent-relay/sdk/dist/workflows/run-summary-table.d.ts.map +1 -1
- package/node_modules/@agent-relay/sdk/dist/workflows/run-summary-table.js +41 -9
- package/node_modules/@agent-relay/sdk/dist/workflows/run-summary-table.js.map +1 -1
- package/node_modules/@agent-relay/sdk/dist/workflows/runner.d.ts +17 -0
- package/node_modules/@agent-relay/sdk/dist/workflows/runner.d.ts.map +1 -1
- package/node_modules/@agent-relay/sdk/dist/workflows/runner.js +390 -16
- package/node_modules/@agent-relay/sdk/dist/workflows/runner.js.map +1 -1
- package/node_modules/@agent-relay/sdk/dist/workflows/types.d.ts +47 -1
- package/node_modules/@agent-relay/sdk/dist/workflows/types.d.ts.map +1 -1
- package/node_modules/@agent-relay/sdk/dist/workflows/types.js.map +1 -1
- package/node_modules/@agent-relay/sdk/dist/workflows/verification.d.ts +9 -0
- package/node_modules/@agent-relay/sdk/dist/workflows/verification.d.ts.map +1 -1
- package/node_modules/@agent-relay/sdk/dist/workflows/verification.js +78 -2
- package/node_modules/@agent-relay/sdk/dist/workflows/verification.js.map +1 -1
- package/node_modules/@agent-relay/sdk/package.json +7 -3
- package/node_modules/@agent-relay/telemetry/dist/config.d.ts.map +1 -1
- package/node_modules/@agent-relay/telemetry/dist/config.js +2 -4
- package/node_modules/@agent-relay/telemetry/dist/config.js.map +1 -1
- package/node_modules/@agent-relay/telemetry/dist/events.d.ts +1 -2
- package/node_modules/@agent-relay/telemetry/dist/events.d.ts.map +1 -1
- package/node_modules/@agent-relay/telemetry/dist/index.d.ts +1 -1
- package/node_modules/@agent-relay/telemetry/dist/index.d.ts.map +1 -1
- package/node_modules/@agent-relay/telemetry/dist/index.js +1 -1
- package/node_modules/@agent-relay/telemetry/dist/index.js.map +1 -1
- package/node_modules/@agent-relay/telemetry/package.json +1 -1
- package/node_modules/@agent-relay/trajectory/package.json +2 -2
- package/node_modules/@agent-relay/user-directory/package.json +2 -2
- package/node_modules/@agent-relay/utils/package.json +2 -2
- package/node_modules/@aws-sdk/core/dist-cjs/index.js +2 -0
- package/node_modules/@aws-sdk/core/dist-cjs/submodules/client/index.js +3 -0
- package/node_modules/@aws-sdk/core/dist-es/submodules/client/setFeature.js +2 -0
- package/node_modules/@aws-sdk/core/dist-types/submodules/client/setFeature.d.ts +1 -1
- package/node_modules/@aws-sdk/core/package.json +6 -4
- package/node_modules/@aws-sdk/credential-provider-env/package.json +2 -2
- package/node_modules/@aws-sdk/credential-provider-http/package.json +5 -5
- package/node_modules/@aws-sdk/credential-provider-ini/package.json +9 -9
- package/node_modules/@aws-sdk/credential-provider-login/package.json +3 -3
- package/node_modules/@aws-sdk/credential-provider-node/package.json +7 -7
- package/node_modules/@aws-sdk/credential-provider-process/package.json +2 -2
- package/node_modules/@aws-sdk/credential-provider-sso/package.json +4 -4
- package/node_modules/@aws-sdk/credential-provider-web-identity/package.json +3 -3
- package/node_modules/@aws-sdk/middleware-flexible-checksums/package.json +4 -4
- package/node_modules/@aws-sdk/middleware-sdk-s3/package.json +5 -5
- package/node_modules/@aws-sdk/middleware-user-agent/package.json +5 -5
- package/node_modules/@aws-sdk/nested-clients/package.json +18 -18
- package/node_modules/@aws-sdk/region-config-resolver/package.json +2 -2
- package/node_modules/@aws-sdk/signature-v4-multi-region/package.json +2 -2
- package/node_modules/@aws-sdk/token-providers/package.json +3 -3
- package/node_modules/@aws-sdk/util-endpoints/package.json +2 -2
- package/node_modules/@aws-sdk/util-user-agent-node/package.json +2 -2
- package/node_modules/@relaycast/sdk/dist/version.d.ts +1 -1
- package/node_modules/@relaycast/sdk/dist/version.js +1 -1
- package/node_modules/@relaycast/sdk/node_modules/@relaycast/types/package.json +1 -1
- package/node_modules/@relaycast/sdk/package.json +2 -2
- package/node_modules/axios/CHANGELOG.md +112 -44
- package/node_modules/axios/README.md +30 -0
- package/node_modules/axios/dist/axios.js +34 -11
- package/node_modules/axios/dist/axios.js.map +1 -1
- package/node_modules/axios/dist/axios.min.js +2 -2
- package/node_modules/axios/dist/axios.min.js.map +1 -1
- package/node_modules/axios/dist/browser/axios.cjs +32 -6
- package/node_modules/axios/dist/browser/axios.cjs.map +1 -1
- package/node_modules/axios/dist/esm/axios.js +32 -6
- package/node_modules/axios/dist/esm/axios.js.map +1 -1
- package/node_modules/axios/dist/esm/axios.min.js +2 -2
- package/node_modules/axios/dist/esm/axios.min.js.map +1 -1
- package/node_modules/axios/dist/node/axios.cjs +91 -37
- package/node_modules/axios/dist/node/axios.cjs.map +1 -1
- package/node_modules/axios/index.d.cts +1 -0
- package/node_modules/axios/index.d.ts +1 -0
- package/node_modules/axios/lib/adapters/http.js +69 -22
- package/node_modules/axios/lib/core/mergeConfig.js +13 -1
- package/node_modules/axios/lib/env/data.js +1 -1
- package/node_modules/axios/lib/helpers/resolveConfig.js +14 -2
- package/node_modules/axios/lib/helpers/validator.js +3 -1
- package/node_modules/axios/package.json +1 -1
- package/package.json +9 -9
- package/packages/cloud/dist/workflows.d.ts +12 -0
- package/packages/cloud/dist/workflows.d.ts.map +1 -1
- package/packages/cloud/dist/workflows.js +38 -0
- package/packages/cloud/dist/workflows.js.map +1 -1
- package/packages/cloud/package.json +2 -2
- package/packages/config/package.json +2 -2
- package/packages/hooks/package.json +4 -4
- package/packages/sdk/dist/relay.d.ts.map +1 -1
- package/packages/sdk/dist/relay.js.map +1 -1
- package/packages/sdk/dist/workflows/__tests__/budget-enforcement.test.d.ts +2 -0
- package/packages/sdk/dist/workflows/__tests__/budget-enforcement.test.d.ts.map +1 -0
- package/packages/sdk/dist/workflows/__tests__/budget-enforcement.test.js +437 -0
- package/packages/sdk/dist/workflows/__tests__/budget-enforcement.test.js.map +1 -0
- package/packages/sdk/dist/workflows/__tests__/budget-tracker.test.d.ts +2 -0
- package/packages/sdk/dist/workflows/__tests__/budget-tracker.test.d.ts.map +1 -0
- package/packages/sdk/dist/workflows/__tests__/budget-tracker.test.js +99 -0
- package/packages/sdk/dist/workflows/__tests__/budget-tracker.test.js.map +1 -0
- package/packages/sdk/dist/workflows/__tests__/proxy-env.test.d.ts +2 -0
- package/packages/sdk/dist/workflows/__tests__/proxy-env.test.d.ts.map +1 -0
- package/packages/sdk/dist/workflows/__tests__/proxy-env.test.js +135 -0
- package/packages/sdk/dist/workflows/__tests__/proxy-env.test.js.map +1 -0
- package/packages/sdk/dist/workflows/__tests__/verification-custom.test.d.ts +2 -0
- package/packages/sdk/dist/workflows/__tests__/verification-custom.test.d.ts.map +1 -0
- package/packages/sdk/dist/workflows/__tests__/verification-custom.test.js +236 -0
- package/packages/sdk/dist/workflows/__tests__/verification-custom.test.js.map +1 -0
- package/packages/sdk/dist/workflows/__tests__/verification-traceback.test.d.ts +2 -0
- package/packages/sdk/dist/workflows/__tests__/verification-traceback.test.d.ts.map +1 -0
- package/packages/sdk/dist/workflows/__tests__/verification-traceback.test.js +448 -0
- package/packages/sdk/dist/workflows/__tests__/verification-traceback.test.js.map +1 -0
- package/packages/sdk/dist/workflows/__tests__/verification.test.js +71 -4
- package/packages/sdk/dist/workflows/__tests__/verification.test.js.map +1 -1
- package/packages/sdk/dist/workflows/budget-tracker.d.ts +75 -0
- package/packages/sdk/dist/workflows/budget-tracker.d.ts.map +1 -0
- package/packages/sdk/dist/workflows/budget-tracker.js +180 -0
- package/packages/sdk/dist/workflows/budget-tracker.js.map +1 -0
- package/packages/sdk/dist/workflows/builder.d.ts +1 -0
- package/packages/sdk/dist/workflows/builder.d.ts.map +1 -1
- package/packages/sdk/dist/workflows/builder.js +17 -2
- package/packages/sdk/dist/workflows/builder.js.map +1 -1
- package/packages/sdk/dist/workflows/index.d.ts +2 -0
- package/packages/sdk/dist/workflows/index.d.ts.map +1 -1
- package/packages/sdk/dist/workflows/index.js +2 -0
- package/packages/sdk/dist/workflows/index.js.map +1 -1
- package/packages/sdk/dist/workflows/proxy-env.d.ts +52 -0
- package/packages/sdk/dist/workflows/proxy-env.d.ts.map +1 -0
- package/packages/sdk/dist/workflows/proxy-env.js +92 -0
- package/packages/sdk/dist/workflows/proxy-env.js.map +1 -0
- package/packages/sdk/dist/workflows/run-summary-table.d.ts +2 -1
- package/packages/sdk/dist/workflows/run-summary-table.d.ts.map +1 -1
- package/packages/sdk/dist/workflows/run-summary-table.js +41 -9
- package/packages/sdk/dist/workflows/run-summary-table.js.map +1 -1
- package/packages/sdk/dist/workflows/runner.d.ts +17 -0
- package/packages/sdk/dist/workflows/runner.d.ts.map +1 -1
- package/packages/sdk/dist/workflows/runner.js +390 -16
- package/packages/sdk/dist/workflows/runner.js.map +1 -1
- package/packages/sdk/dist/workflows/types.d.ts +47 -1
- package/packages/sdk/dist/workflows/types.d.ts.map +1 -1
- package/packages/sdk/dist/workflows/types.js.map +1 -1
- package/packages/sdk/dist/workflows/verification.d.ts +9 -0
- package/packages/sdk/dist/workflows/verification.d.ts.map +1 -1
- package/packages/sdk/dist/workflows/verification.js +78 -2
- package/packages/sdk/dist/workflows/verification.js.map +1 -1
- package/packages/sdk/package.json +7 -3
- package/packages/telemetry/dist/config.d.ts.map +1 -1
- package/packages/telemetry/dist/config.js +2 -4
- package/packages/telemetry/dist/config.js.map +1 -1
- package/packages/telemetry/dist/events.d.ts +1 -2
- package/packages/telemetry/dist/events.d.ts.map +1 -1
- package/packages/telemetry/dist/index.d.ts +1 -1
- package/packages/telemetry/dist/index.d.ts.map +1 -1
- package/packages/telemetry/dist/index.js +1 -1
- package/packages/telemetry/dist/index.js.map +1 -1
- package/packages/telemetry/package.json +1 -1
- package/packages/trajectory/package.json +2 -2
- package/packages/user-directory/package.json +2 -2
- package/packages/utils/package.json +2 -2
|
@@ -0,0 +1,437 @@
|
|
|
1
|
+
import { EventEmitter } from 'node:events';
|
|
2
|
+
import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
|
|
3
|
+
import { mkdtempSync, mkdirSync, rmSync } from 'node:fs';
|
|
4
|
+
import os from 'node:os';
|
|
5
|
+
import path from 'node:path';
|
|
6
|
+
const tempDirs = [];
|
|
7
|
+
let queuedSubprocessResults = [];
|
|
8
|
+
let queuedCollectorResults = [];
|
|
9
|
+
let collectorResultsByCwd = new Map();
|
|
10
|
+
let activeRunner;
|
|
11
|
+
const mockCollectCliSession = vi.fn(async (query) => {
|
|
12
|
+
const next = queuedCollectorResults.length > 0
|
|
13
|
+
? queuedCollectorResults.shift()
|
|
14
|
+
: collectorResultsByCwd.get(query.cwd);
|
|
15
|
+
if (typeof next === 'function') {
|
|
16
|
+
return next(query);
|
|
17
|
+
}
|
|
18
|
+
return next ?? null;
|
|
19
|
+
});
|
|
20
|
+
vi.mock('../cli-session-collector.js', () => ({
|
|
21
|
+
collectCliSession: mockCollectCliSession,
|
|
22
|
+
}));
|
|
23
|
+
const mockSubprocessSpawn = vi.fn().mockImplementation((_cmd, _args, _options) => {
|
|
24
|
+
const result = queuedSubprocessResults.shift() ?? {
|
|
25
|
+
stdout: 'completed\n',
|
|
26
|
+
code: 0,
|
|
27
|
+
};
|
|
28
|
+
const child = new EventEmitter();
|
|
29
|
+
child.stdout = new EventEmitter();
|
|
30
|
+
child.stderr = new EventEmitter();
|
|
31
|
+
child.pid = 4321;
|
|
32
|
+
child.kill = vi.fn();
|
|
33
|
+
result.onSpawn?.();
|
|
34
|
+
const emitResult = () => {
|
|
35
|
+
if (result.error) {
|
|
36
|
+
child.emit('error', result.error);
|
|
37
|
+
return;
|
|
38
|
+
}
|
|
39
|
+
if (result.stdout) {
|
|
40
|
+
child.stdout.emit('data', Buffer.from(result.stdout));
|
|
41
|
+
}
|
|
42
|
+
if (result.stderr) {
|
|
43
|
+
child.stderr.emit('data', Buffer.from(result.stderr));
|
|
44
|
+
}
|
|
45
|
+
child.emit('close', result.code ?? 0, result.signal ?? null);
|
|
46
|
+
};
|
|
47
|
+
if (result.delayMs && result.delayMs > 0) {
|
|
48
|
+
setTimeout(emitResult, result.delayMs);
|
|
49
|
+
}
|
|
50
|
+
else {
|
|
51
|
+
queueMicrotask(emitResult);
|
|
52
|
+
}
|
|
53
|
+
return child;
|
|
54
|
+
});
|
|
55
|
+
vi.mock('node:child_process', async () => {
|
|
56
|
+
const actual = await vi.importActual('node:child_process');
|
|
57
|
+
return {
|
|
58
|
+
...actual,
|
|
59
|
+
spawn: mockSubprocessSpawn,
|
|
60
|
+
};
|
|
61
|
+
});
|
|
62
|
+
const mockRelayInstance = {
|
|
63
|
+
spawnPty: vi.fn(),
|
|
64
|
+
human: vi.fn().mockReturnValue({ sendMessage: vi.fn().mockResolvedValue(undefined) }),
|
|
65
|
+
shutdown: vi.fn().mockResolvedValue(undefined),
|
|
66
|
+
onBrokerStderr: vi.fn().mockReturnValue(() => { }),
|
|
67
|
+
listAgentsRaw: vi.fn().mockResolvedValue([]),
|
|
68
|
+
listAgents: vi.fn().mockResolvedValue([]),
|
|
69
|
+
onWorkerOutput: null,
|
|
70
|
+
onMessageReceived: null,
|
|
71
|
+
onAgentSpawned: null,
|
|
72
|
+
onAgentReleased: null,
|
|
73
|
+
onAgentExited: null,
|
|
74
|
+
onAgentIdle: null,
|
|
75
|
+
onDeliveryUpdate: null,
|
|
76
|
+
};
|
|
77
|
+
vi.mock('@relaycast/sdk', () => ({
|
|
78
|
+
RelayCast: vi.fn(),
|
|
79
|
+
RelayError: class RelayError extends Error {
|
|
80
|
+
},
|
|
81
|
+
}));
|
|
82
|
+
vi.mock('../../relay.js', () => ({
|
|
83
|
+
AgentRelay: vi.fn().mockImplementation(() => mockRelayInstance),
|
|
84
|
+
}));
|
|
85
|
+
const { WorkflowRunner } = await import('../runner.js');
|
|
86
|
+
function makeDbHarness() {
|
|
87
|
+
const runs = new Map();
|
|
88
|
+
const steps = new Map();
|
|
89
|
+
return {
|
|
90
|
+
db: {
|
|
91
|
+
insertRun: vi.fn(async (run) => {
|
|
92
|
+
runs.set(run.id, { ...run });
|
|
93
|
+
}),
|
|
94
|
+
updateRun: vi.fn(async (id, patch) => {
|
|
95
|
+
const existing = runs.get(id);
|
|
96
|
+
if (existing) {
|
|
97
|
+
runs.set(id, { ...existing, ...patch });
|
|
98
|
+
}
|
|
99
|
+
}),
|
|
100
|
+
getRun: vi.fn(async (id) => {
|
|
101
|
+
const run = runs.get(id);
|
|
102
|
+
return run ? { ...run } : null;
|
|
103
|
+
}),
|
|
104
|
+
insertStep: vi.fn(async (step) => {
|
|
105
|
+
steps.set(step.id, { ...step });
|
|
106
|
+
}),
|
|
107
|
+
updateStep: vi.fn(async (id, patch) => {
|
|
108
|
+
const existing = steps.get(id);
|
|
109
|
+
if (existing) {
|
|
110
|
+
steps.set(id, { ...existing, ...patch });
|
|
111
|
+
}
|
|
112
|
+
}),
|
|
113
|
+
getStepsByRunId: vi.fn(async (runId) => {
|
|
114
|
+
return [...steps.values()].filter((step) => step.runId === runId).map((step) => ({ ...step }));
|
|
115
|
+
}),
|
|
116
|
+
},
|
|
117
|
+
getRun(id) {
|
|
118
|
+
const run = runs.get(id);
|
|
119
|
+
return run ? { ...run } : null;
|
|
120
|
+
},
|
|
121
|
+
getSteps(runId) {
|
|
122
|
+
return [...steps.values()].filter((step) => step.runId === runId).map((step) => ({ ...step }));
|
|
123
|
+
},
|
|
124
|
+
};
|
|
125
|
+
}
|
|
126
|
+
function createWorkspace(subdirs = []) {
|
|
127
|
+
const dir = mkdtempSync(path.join(os.tmpdir(), 'relay-budget-enforcement-'));
|
|
128
|
+
tempDirs.push(dir);
|
|
129
|
+
for (const subdir of subdirs) {
|
|
130
|
+
mkdirSync(path.join(dir, subdir), { recursive: true });
|
|
131
|
+
}
|
|
132
|
+
return dir;
|
|
133
|
+
}
|
|
134
|
+
function makeRunner(cwd, db) {
|
|
135
|
+
return new WorkflowRunner({
|
|
136
|
+
cwd,
|
|
137
|
+
db,
|
|
138
|
+
workspaceId: 'ws-test',
|
|
139
|
+
relay: {
|
|
140
|
+
env: {
|
|
141
|
+
AGENT_RELAY_WORKFLOW_DISABLE_RELAYCAST: '1',
|
|
142
|
+
},
|
|
143
|
+
},
|
|
144
|
+
});
|
|
145
|
+
}
|
|
146
|
+
function makeAgent(name, overrides = {}) {
|
|
147
|
+
return {
|
|
148
|
+
name,
|
|
149
|
+
cli: 'claude',
|
|
150
|
+
interactive: false,
|
|
151
|
+
...overrides,
|
|
152
|
+
};
|
|
153
|
+
}
|
|
154
|
+
function makeStep(name, agent, overrides = {}) {
|
|
155
|
+
return {
|
|
156
|
+
name,
|
|
157
|
+
agent,
|
|
158
|
+
task: `Complete ${name}`,
|
|
159
|
+
...overrides,
|
|
160
|
+
};
|
|
161
|
+
}
|
|
162
|
+
function makeConfig(input) {
|
|
163
|
+
return {
|
|
164
|
+
version: '1',
|
|
165
|
+
name: 'budget-enforcement',
|
|
166
|
+
swarm: {
|
|
167
|
+
pattern: 'dag',
|
|
168
|
+
...input.swarm,
|
|
169
|
+
},
|
|
170
|
+
agents: input.agents,
|
|
171
|
+
workflows: [
|
|
172
|
+
{
|
|
173
|
+
name: 'default',
|
|
174
|
+
steps: input.steps,
|
|
175
|
+
},
|
|
176
|
+
],
|
|
177
|
+
trajectories: false,
|
|
178
|
+
};
|
|
179
|
+
}
|
|
180
|
+
function makeReport(tokens, overrides = {}) {
|
|
181
|
+
return {
|
|
182
|
+
cli: 'claude',
|
|
183
|
+
sessionId: 'session-1',
|
|
184
|
+
model: 'claude-sonnet-4',
|
|
185
|
+
provider: 'anthropic',
|
|
186
|
+
durationMs: 1_000,
|
|
187
|
+
cost: null,
|
|
188
|
+
tokens: {
|
|
189
|
+
input: tokens.input ?? 0,
|
|
190
|
+
output: tokens.output ?? 0,
|
|
191
|
+
cacheRead: tokens.cacheRead ?? 0,
|
|
192
|
+
},
|
|
193
|
+
turns: 1,
|
|
194
|
+
toolCalls: [],
|
|
195
|
+
errors: [],
|
|
196
|
+
finalStatus: 'completed',
|
|
197
|
+
summary: 'done',
|
|
198
|
+
...overrides,
|
|
199
|
+
};
|
|
200
|
+
}
|
|
201
|
+
function getBudgetTracker(runner) {
|
|
202
|
+
return runner.budgetTracker;
|
|
203
|
+
}
|
|
204
|
+
beforeEach(() => {
|
|
205
|
+
vi.clearAllMocks();
|
|
206
|
+
queuedSubprocessResults = [];
|
|
207
|
+
queuedCollectorResults = [];
|
|
208
|
+
collectorResultsByCwd = new Map();
|
|
209
|
+
activeRunner = undefined;
|
|
210
|
+
mockRelayInstance.shutdown.mockResolvedValue(undefined);
|
|
211
|
+
mockRelayInstance.onBrokerStderr.mockReturnValue(() => { });
|
|
212
|
+
mockRelayInstance.listAgents.mockResolvedValue([]);
|
|
213
|
+
});
|
|
214
|
+
afterEach(() => {
|
|
215
|
+
activeRunner = undefined;
|
|
216
|
+
while (tempDirs.length > 0) {
|
|
217
|
+
rmSync(tempDirs.pop(), { recursive: true, force: true });
|
|
218
|
+
}
|
|
219
|
+
});
|
|
220
|
+
describe('WorkflowRunner budget enforcement integration', () => {
|
|
221
|
+
it('workflow with no budget config runs normally', async () => {
|
|
222
|
+
const workspace = createWorkspace(['step-1', 'step-2']);
|
|
223
|
+
const dbHarness = makeDbHarness();
|
|
224
|
+
const runner = makeRunner(workspace, dbHarness.db);
|
|
225
|
+
activeRunner = runner;
|
|
226
|
+
queuedSubprocessResults = [
|
|
227
|
+
{ stdout: 'step 1 complete\n', code: 0 },
|
|
228
|
+
{ stdout: 'step 2 complete\n', code: 0 },
|
|
229
|
+
];
|
|
230
|
+
queuedCollectorResults = [null, null];
|
|
231
|
+
const run = await runner.execute(makeConfig({
|
|
232
|
+
agents: [
|
|
233
|
+
makeAgent('worker-1', { cwd: 'step-1' }),
|
|
234
|
+
makeAgent('worker-2', { cwd: 'step-2' }),
|
|
235
|
+
],
|
|
236
|
+
steps: [
|
|
237
|
+
makeStep('step-1', 'worker-1'),
|
|
238
|
+
makeStep('step-2', 'worker-2', { dependsOn: ['step-1'] }),
|
|
239
|
+
],
|
|
240
|
+
}), 'default');
|
|
241
|
+
expect(run.status).toBe('completed');
|
|
242
|
+
expect(getBudgetTracker(runner)).toBeUndefined();
|
|
243
|
+
expect(dbHarness.getSteps(run.id).map((step) => step.status)).toEqual(['completed', 'completed']);
|
|
244
|
+
expect(mockSubprocessSpawn).toHaveBeenCalledTimes(2);
|
|
245
|
+
});
|
|
246
|
+
it('per-agent maxTokens recorded in budget tracker', async () => {
|
|
247
|
+
const workspace = createWorkspace(['writer']);
|
|
248
|
+
const dbHarness = makeDbHarness();
|
|
249
|
+
const runner = makeRunner(workspace, dbHarness.db);
|
|
250
|
+
activeRunner = runner;
|
|
251
|
+
queuedSubprocessResults = [{ stdout: 'draft complete\n', code: 0 }];
|
|
252
|
+
collectorResultsByCwd.set(path.join(workspace, 'writer'), makeReport({ input: 800, output: 150 }));
|
|
253
|
+
const run = await runner.execute(makeConfig({
|
|
254
|
+
agents: [
|
|
255
|
+
makeAgent('writer', {
|
|
256
|
+
cwd: 'writer',
|
|
257
|
+
constraints: { maxTokens: 1_000 },
|
|
258
|
+
}),
|
|
259
|
+
],
|
|
260
|
+
steps: [makeStep('draft', 'writer')],
|
|
261
|
+
}), 'default');
|
|
262
|
+
const tracker = getBudgetTracker(runner);
|
|
263
|
+
expect(run.status).toBe('completed');
|
|
264
|
+
expect(tracker).toBeDefined();
|
|
265
|
+
expect(tracker?.getStepUsage('draft')).toEqual({
|
|
266
|
+
input: 800,
|
|
267
|
+
output: 150,
|
|
268
|
+
cacheRead: 0,
|
|
269
|
+
total: 950,
|
|
270
|
+
});
|
|
271
|
+
expect(tracker?.getStepBudgetStatus('draft')).toEqual({
|
|
272
|
+
used: 950,
|
|
273
|
+
limit: 1_000,
|
|
274
|
+
over: false,
|
|
275
|
+
});
|
|
276
|
+
});
|
|
277
|
+
it('per-workflow tokenBudget prevents spawning when exhausted', async () => {
|
|
278
|
+
const workspace = createWorkspace(['planner', 'writer']);
|
|
279
|
+
const dbHarness = makeDbHarness();
|
|
280
|
+
const runner = makeRunner(workspace, dbHarness.db);
|
|
281
|
+
activeRunner = runner;
|
|
282
|
+
queuedSubprocessResults = [{ stdout: 'plan complete\n', code: 0 }];
|
|
283
|
+
collectorResultsByCwd.set(path.join(workspace, 'planner'), makeReport({ input: 1_500, output: 300 }));
|
|
284
|
+
const run = await runner.execute(makeConfig({
|
|
285
|
+
agents: [
|
|
286
|
+
makeAgent('planner', { cwd: 'planner' }),
|
|
287
|
+
makeAgent('writer', { cwd: 'writer' }),
|
|
288
|
+
],
|
|
289
|
+
steps: [
|
|
290
|
+
makeStep('step-1', 'planner'),
|
|
291
|
+
makeStep('step-2', 'writer', { dependsOn: ['step-1'] }),
|
|
292
|
+
],
|
|
293
|
+
swarm: {
|
|
294
|
+
tokenBudget: 2_000,
|
|
295
|
+
},
|
|
296
|
+
}), 'default');
|
|
297
|
+
const tracker = getBudgetTracker(runner);
|
|
298
|
+
const failedStep = dbHarness.getSteps(run.id).find((step) => step.stepName === 'step-2');
|
|
299
|
+
expect(run.status).toBe('failed');
|
|
300
|
+
expect(mockSubprocessSpawn).toHaveBeenCalledTimes(1);
|
|
301
|
+
expect(tracker?.getTotalUsage().total).toBe(1_800);
|
|
302
|
+
expect(failedStep?.status).toBe('failed');
|
|
303
|
+
expect(failedStep?.error).toContain('workflow budget exhausted');
|
|
304
|
+
expect(failedStep?.error).toContain('1800/2000');
|
|
305
|
+
});
|
|
306
|
+
it('pre-spawn check allows step when budget has headroom', async () => {
|
|
307
|
+
const workspace = createWorkspace(['first', 'second']);
|
|
308
|
+
const dbHarness = makeDbHarness();
|
|
309
|
+
const runner = makeRunner(workspace, dbHarness.db);
|
|
310
|
+
activeRunner = runner;
|
|
311
|
+
let checkCanSpawnAllowed;
|
|
312
|
+
queuedSubprocessResults = [
|
|
313
|
+
{ stdout: 'first complete\n', code: 0 },
|
|
314
|
+
{
|
|
315
|
+
stdout: 'second complete\n',
|
|
316
|
+
code: 0,
|
|
317
|
+
onSpawn: () => {
|
|
318
|
+
checkCanSpawnAllowed = getBudgetTracker(runner)?.checkCanSpawn('step-2').allowed;
|
|
319
|
+
},
|
|
320
|
+
},
|
|
321
|
+
];
|
|
322
|
+
collectorResultsByCwd.set(path.join(workspace, 'first'), makeReport({ input: 900, output: 100 }));
|
|
323
|
+
collectorResultsByCwd.set(path.join(workspace, 'second'), makeReport({ input: 150, output: 50 }));
|
|
324
|
+
const run = await runner.execute(makeConfig({
|
|
325
|
+
agents: [
|
|
326
|
+
makeAgent('first-agent', { cwd: 'first' }),
|
|
327
|
+
makeAgent('second-agent', { cwd: 'second' }),
|
|
328
|
+
],
|
|
329
|
+
steps: [
|
|
330
|
+
makeStep('step-1', 'first-agent'),
|
|
331
|
+
makeStep('step-2', 'second-agent', { dependsOn: ['step-1'] }),
|
|
332
|
+
],
|
|
333
|
+
swarm: {
|
|
334
|
+
tokenBudget: 5_000,
|
|
335
|
+
},
|
|
336
|
+
}), 'default');
|
|
337
|
+
expect(run.status).toBe('completed');
|
|
338
|
+
expect(checkCanSpawnAllowed).toBe(true);
|
|
339
|
+
expect(mockSubprocessSpawn).toHaveBeenCalledTimes(2);
|
|
340
|
+
});
|
|
341
|
+
it('retry attempts consume from same budget', async () => {
|
|
342
|
+
const workspace = createWorkspace(['retry-agent']);
|
|
343
|
+
const dbHarness = makeDbHarness();
|
|
344
|
+
const runner = makeRunner(workspace, dbHarness.db);
|
|
345
|
+
activeRunner = runner;
|
|
346
|
+
let usageBeforeRetry;
|
|
347
|
+
queuedSubprocessResults = [
|
|
348
|
+
{ stdout: 'first attempt failed\n', code: 1 },
|
|
349
|
+
{
|
|
350
|
+
stdout: 'retry succeeded\n',
|
|
351
|
+
code: 0,
|
|
352
|
+
onSpawn: () => {
|
|
353
|
+
usageBeforeRetry = getBudgetTracker(runner)?.getStepUsage('retry-step').total;
|
|
354
|
+
},
|
|
355
|
+
},
|
|
356
|
+
];
|
|
357
|
+
queuedCollectorResults = [
|
|
358
|
+
makeReport({ input: 500, output: 100 }, { finalStatus: 'failed' }),
|
|
359
|
+
makeReport({ input: 250, output: 100 }),
|
|
360
|
+
];
|
|
361
|
+
const run = await runner.execute(makeConfig({
|
|
362
|
+
agents: [
|
|
363
|
+
makeAgent('retry-agent', {
|
|
364
|
+
cwd: 'retry-agent',
|
|
365
|
+
constraints: { maxTokens: 1_000 },
|
|
366
|
+
}),
|
|
367
|
+
],
|
|
368
|
+
steps: [makeStep('retry-step', 'retry-agent', { retries: 1 })],
|
|
369
|
+
}), 'default');
|
|
370
|
+
const tracker = getBudgetTracker(runner);
|
|
371
|
+
expect(run.status).toBe('completed');
|
|
372
|
+
expect(usageBeforeRetry).toBe(600);
|
|
373
|
+
expect(tracker?.getStepUsage('retry-step')).toEqual({
|
|
374
|
+
input: 750,
|
|
375
|
+
output: 200,
|
|
376
|
+
cacheRead: 0,
|
|
377
|
+
total: 950,
|
|
378
|
+
});
|
|
379
|
+
expect(tracker?.getStepBudgetStatus('retry-step')).toEqual({
|
|
380
|
+
used: 950,
|
|
381
|
+
limit: 1_000,
|
|
382
|
+
over: false,
|
|
383
|
+
});
|
|
384
|
+
expect(mockCollectCliSession).toHaveBeenCalledTimes(2);
|
|
385
|
+
});
|
|
386
|
+
it('parallel steps track budget correctly', async () => {
|
|
387
|
+
const workspace = createWorkspace(['parallel-a', 'parallel-b']);
|
|
388
|
+
const dbHarness = makeDbHarness();
|
|
389
|
+
const runner = makeRunner(workspace, dbHarness.db);
|
|
390
|
+
activeRunner = runner;
|
|
391
|
+
queuedSubprocessResults = [
|
|
392
|
+
{ stdout: 'parallel a\n', code: 0, delayMs: 10 },
|
|
393
|
+
{ stdout: 'parallel b\n', code: 0, delayMs: 1 },
|
|
394
|
+
];
|
|
395
|
+
collectorResultsByCwd.set(path.join(workspace, 'parallel-a'), makeReport({ input: 500, output: 200 }));
|
|
396
|
+
collectorResultsByCwd.set(path.join(workspace, 'parallel-b'), makeReport({ input: 700, output: 200 }));
|
|
397
|
+
const run = await runner.execute(makeConfig({
|
|
398
|
+
agents: [
|
|
399
|
+
makeAgent('parallel-a', { cwd: 'parallel-a' }),
|
|
400
|
+
makeAgent('parallel-b', { cwd: 'parallel-b' }),
|
|
401
|
+
],
|
|
402
|
+
steps: [
|
|
403
|
+
makeStep('parallel-a', 'parallel-a'),
|
|
404
|
+
makeStep('parallel-b', 'parallel-b'),
|
|
405
|
+
],
|
|
406
|
+
swarm: {
|
|
407
|
+
tokenBudget: 5_000,
|
|
408
|
+
},
|
|
409
|
+
}), 'default');
|
|
410
|
+
const tracker = getBudgetTracker(runner);
|
|
411
|
+
expect(run.status).toBe('completed');
|
|
412
|
+
expect(tracker?.getStepUsage('parallel-a')).toEqual({
|
|
413
|
+
input: 500,
|
|
414
|
+
output: 200,
|
|
415
|
+
cacheRead: 0,
|
|
416
|
+
total: 700,
|
|
417
|
+
});
|
|
418
|
+
expect(tracker?.getStepUsage('parallel-b')).toEqual({
|
|
419
|
+
input: 700,
|
|
420
|
+
output: 200,
|
|
421
|
+
cacheRead: 0,
|
|
422
|
+
total: 900,
|
|
423
|
+
});
|
|
424
|
+
expect(tracker?.getTotalUsage()).toEqual({
|
|
425
|
+
input: 1_200,
|
|
426
|
+
output: 400,
|
|
427
|
+
cacheRead: 0,
|
|
428
|
+
total: 1_600,
|
|
429
|
+
});
|
|
430
|
+
expect(tracker?.getRunSummaryBudgetData()?.workflow).toEqual({
|
|
431
|
+
used: 1_600,
|
|
432
|
+
limit: 5_000,
|
|
433
|
+
exhausted: false,
|
|
434
|
+
});
|
|
435
|
+
});
|
|
436
|
+
});
|
|
437
|
+
//# sourceMappingURL=budget-enforcement.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"budget-enforcement.test.js","sourceRoot":"","sources":["../../../src/workflows/__tests__/budget-enforcement.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAC3C,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AACzD,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAO7B,MAAM,QAAQ,GAAa,EAAE,CAAC;AAiB9B,IAAI,uBAAuB,GAA6B,EAAE,CAAC;AAC3D,IAAI,sBAAsB,GAAsB,EAAE,CAAC;AACnD,IAAI,qBAAqB,GAAG,IAAI,GAAG,EAA2B,CAAC;AAC/D,IAAI,YAA6D,CAAC;AAElE,MAAM,qBAAqB,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,EAAE,KAAsB,EAAoC,EAAE;IACrG,MAAM,IAAI,GACR,sBAAsB,CAAC,MAAM,GAAG,CAAC;QAC/B,CAAC,CAAC,sBAAsB,CAAC,KAAK,EAAE;QAChC,CAAC,CAAC,qBAAqB,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IAE3C,IAAI,OAAO,IAAI,KAAK,UAAU,EAAE,CAAC;QAC/B,OAAO,IAAI,CAAC,KAAK,CAAC,CAAC;IACrB,CAAC;IAED,OAAO,IAAI,IAAI,IAAI,CAAC;AACtB,CAAC,CAAC,CAAC;AAEH,EAAE,CAAC,IAAI,CAAC,6BAA6B,EAAE,GAAG,EAAE,CAAC,CAAC;IAC5C,iBAAiB,EAAE,qBAAqB;CACzC,CAAC,CAAC,CAAC;AAEJ,MAAM,mBAAmB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE;IAC/E,MAAM,MAAM,GAAG,uBAAuB,CAAC,KAAK,EAAE,IAAI;QAChD,MAAM,EAAE,aAAa;QACrB,IAAI,EAAE,CAAC;KACR,CAAC;IAEF,MAAM,KAAK,GAAG,IAAI,YAAY,EAK7B,CAAC;IAEF,KAAK,CAAC,MAAM,GAAG,IAAI,YAAY,EAAE,CAAC;IAClC,KAAK,CAAC,MAAM,GAAG,IAAI,YAAY,EAAE,CAAC;IAClC,KAAK,CAAC,GAAG,GAAG,IAAI,CAAC;IACjB,KAAK,CAAC,IAAI,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IAErB,MAAM,CAAC,OAAO,EAAE,EAAE,CAAC;IAEnB,MAAM,UAAU,GAAG,GAAG,EAAE;QACtB,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;YACjB,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;YAClC,OAAO;QACT,CAAC;QACD,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;YAClB,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;QACxD,CAAC;QACD,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;YAClB,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;QACxD,CAAC;QACD,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,IAAI,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,IAAI,IAAI,CAAC,CAAC;IAC/D,CAAC,CAAC;IAEF,IAAI,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC,OAAO,GAAG,CAAC,EAAE,CAAC;QACzC,UAAU,CAAC,UAAU,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC;IACzC,CAAC;SAAM,CAAC;QACN,cAAc,CAAC,UAAU,CAAC,CAAC;IAC7B,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC,CAAC,CAAC;AAEH,EAAE,CAAC,IAAI,CAAC,oBAAoB,EAAE,KAAK,IAAI,EAAE;IACvC,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC,YAAY,CAAsC,oBAAoB,CAAC,CAAC;IAChG,OAAO;QACL,GAAG,MAAM;QACT,KAAK,EAAE,mBAAmB;KAC3B,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,MAAM,iBAAiB,GAAG;IACxB,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;IACjB,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC,EAAE,CAAC;IACrF,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;IAC9C,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC;IACjD,aAAa,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;IAC5C,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;IACzC,cAAc,EAAE,IAAiE;IACjF,iBAAiB,EAAE,IAAW;IAC9B,cAAc,EAAE,IAAW;IAC3B,eAAe,EAAE,IAAW;IAC5B,aAAa,EAAE,IAAW;IAC1B,WAAW,EAAE,IAAW;IACxB,gBAAgB,EAAE,IAAW;CAC9B,CAAC;AAEF,EAAE,CAAC,IAAI,CAAC,gBAAgB,EAAE,GAAG,EAAE,CAAC,CAAC;IAC/B,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE;IAClB,UAAU,EAAE,MAAM,UAAW,SAAQ,KAAK;KAAG;CAC9C,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,gBAAgB,EAAE,GAAG,EAAE,CAAC,CAAC;IAC/B,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,iBAAiB,CAAC;CAChE,CAAC,CAAC,CAAC;AAEJ,MAAM,EAAE,cAAc,EAAE,GAAG,MAAM,MAAM,CAAC,cAAc,CAAC,CAAC;AAQxD,SAAS,aAAa;IACpB,MAAM,IAAI,GAAG,IAAI,GAAG,EAA0B,CAAC;IAC/C,MAAM,KAAK,GAAG,IAAI,GAAG,EAA2B,CAAC;IAEjD,OAAO;QACL,EAAE,EAAE;YACF,SAAS,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,EAAE,GAAmB,EAAE,EAAE;gBAC7C,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,EAAE,EAAE,GAAG,GAAG,EAAE,CAAC,CAAC;YAC/B,CAAC,CAAC;YACF,SAAS,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,EAAE,EAAU,EAAE,KAA8B,EAAE,EAAE;gBACpE,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBAC9B,IAAI,QAAQ,EAAE,CAAC;oBACb,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,EAAE,GAAG,QAAQ,EAAE,GAAG,KAAK,EAAE,CAAC,CAAC;gBAC1C,CAAC;YACH,CAAC,CAAC;YACF,MAAM,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,EAAE,EAAU,EAAE,EAAE;gBACjC,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBACzB,OAAO,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,GAAG,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;YACjC,CAAC,CAAC;YACF,UAAU,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,EAAE,IAAqB,EAAE,EAAE;gBAChD,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,GAAG,IAAI,EAAE,CAAC,CAAC;YAClC,CAAC,CAAC;YACF,UAAU,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,EAAE,EAAU,EAAE,KAA+B,EAAE,EAAE;gBACtE,MAAM,QAAQ,GAAG,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBAC/B,IAAI,QAAQ,EAAE,CAAC;oBACb,KAAK,CAAC,GAAG,CAAC,EAAE,EAAE,EAAE,GAAG,QAAQ,EAAE,GAAG,KAAK,EAAE,CAAC,CAAC;gBAC3C,CAAC;YACH,CAAC,CAAC;YACF,eAAe,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,EAAE,KAAa,EAAE,EAAE;gBAC7C,OAAO,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,KAAK,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,IAAI,EAAE,CAAC,CAAC,CAAC;YACjG,CAAC,CAAC;SACH;QACD,MAAM,CAAC,EAAU;YACf,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YACzB,OAAO,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,GAAG,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;QACjC,CAAC;QACD,QAAQ,CAAC,KAAa;YACpB,OAAO,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,KAAK,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,IAAI,EAAE,CAAC,CAAC,CAAC;QACjG,CAAC;KACF,CAAC;AACJ,CAAC;AAED,SAAS,eAAe,CAAC,UAAoB,EAAE;IAC7C,MAAM,GAAG,GAAG,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,2BAA2B,CAAC,CAAC,CAAC;IAC7E,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAEnB,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;QAC7B,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,MAAM,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IACzD,CAAC;IAED,OAAO,GAAG,CAAC;AACb,CAAC;AAED,SAAS,UAAU,CAAC,GAAW,EAAE,EAAc;IAC7C,OAAO,IAAI,cAAc,CAAC;QACxB,GAAG;QACH,EAAE;QACF,WAAW,EAAE,SAAS;QACtB,KAAK,EAAE;YACL,GAAG,EAAE;gBACH,sCAAsC,EAAE,GAAG;aAC5C;SACF;KACF,CAAC,CAAC;AACL,CAAC;AAED,SAAS,SAAS,CAChB,IAAY,EACZ,YAAwD,EAAE;IAE1D,OAAO;QACL,IAAI;QACJ,GAAG,EAAE,QAAQ;QACb,WAAW,EAAE,KAAK;QAClB,GAAG,SAAS;KACb,CAAC;AACJ,CAAC;AAED,SAAS,QAAQ,CACf,IAAY,EACZ,KAAa,EACb,YAAyF,EAAE;IAE3F,OAAO;QACL,IAAI;QACJ,KAAK;QACL,IAAI,EAAE,YAAY,IAAI,EAAE;QACxB,GAAG,SAAS;KACb,CAAC;AACJ,CAAC;AAED,SAAS,UAAU,CAAC,KAInB;IACC,OAAO;QACL,OAAO,EAAE,GAAG;QACZ,IAAI,EAAE,oBAAoB;QAC1B,KAAK,EAAE;YACL,OAAO,EAAE,KAAK;YACd,GAAG,KAAK,CAAC,KAAK;SACf;QACD,MAAM,EAAE,KAAK,CAAC,MAAM;QACpB,SAAS,EAAE;YACT;gBACE,IAAI,EAAE,SAAS;gBACf,KAAK,EAAE,KAAK,CAAC,KAAK;aACnB;SACF;QACD,YAAY,EAAE,KAAK;KACpB,CAAC;AACJ,CAAC;AAED,SAAS,UAAU,CACjB,MAAwD,EACxD,YAAuC,EAAE;IAEzC,OAAO;QACL,GAAG,EAAE,QAAQ;QACb,SAAS,EAAE,WAAW;QACtB,KAAK,EAAE,iBAAiB;QACxB,QAAQ,EAAE,WAAW;QACrB,UAAU,EAAE,KAAK;QACjB,IAAI,EAAE,IAAI;QACV,MAAM,EAAE;YACN,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,CAAC;YACxB,MAAM,EAAE,MAAM,CAAC,MAAM,IAAI,CAAC;YAC1B,SAAS,EAAE,MAAM,CAAC,SAAS,IAAI,CAAC;SACjC;QACD,KAAK,EAAE,CAAC;QACR,SAAS,EAAE,EAAE;QACb,MAAM,EAAE,EAAE;QACV,WAAW,EAAE,WAAW;QACxB,OAAO,EAAE,MAAM;QACf,GAAG,SAAS;KACb,CAAC;AACJ,CAAC;AAED,SAAS,gBAAgB,CAAC,MAA2C;IACnE,OAAQ,MAAc,CAAC,aAA0C,CAAC;AACpE,CAAC;AAED,UAAU,CAAC,GAAG,EAAE;IACd,EAAE,CAAC,aAAa,EAAE,CAAC;IACnB,uBAAuB,GAAG,EAAE,CAAC;IAC7B,sBAAsB,GAAG,EAAE,CAAC;IAC5B,qBAAqB,GAAG,IAAI,GAAG,EAAE,CAAC;IAClC,YAAY,GAAG,SAAS,CAAC;IACzB,iBAAiB,CAAC,QAAQ,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;IACxD,iBAAiB,CAAC,cAAc,CAAC,eAAe,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;IAC3D,iBAAiB,CAAC,UAAU,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;AACrD,CAAC,CAAC,CAAC;AAEH,SAAS,CAAC,GAAG,EAAE;IACb,YAAY,GAAG,SAAS,CAAC;IACzB,OAAO,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC3B,MAAM,CAAC,QAAQ,CAAC,GAAG,EAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAC5D,CAAC;AACH,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,+CAA+C,EAAE,GAAG,EAAE;IAC7D,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;QAC5D,MAAM,SAAS,GAAG,eAAe,CAAC,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC;QACxD,MAAM,SAAS,GAAG,aAAa,EAAE,CAAC;QAClC,MAAM,MAAM,GAAG,UAAU,CAAC,SAAS,EAAE,SAAS,CAAC,EAAE,CAAC,CAAC;QACnD,YAAY,GAAG,MAAM,CAAC;QAEtB,uBAAuB,GAAG;YACxB,EAAE,MAAM,EAAE,mBAAmB,EAAE,IAAI,EAAE,CAAC,EAAE;YACxC,EAAE,MAAM,EAAE,mBAAmB,EAAE,IAAI,EAAE,CAAC,EAAE;SACzC,CAAC;QACF,sBAAsB,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;QAEtC,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,OAAO,CAC9B,UAAU,CAAC;YACT,MAAM,EAAE;gBACN,SAAS,CAAC,UAAU,EAAE,EAAE,GAAG,EAAE,QAAQ,EAAE,CAAC;gBACxC,SAAS,CAAC,UAAU,EAAE,EAAE,GAAG,EAAE,QAAQ,EAAE,CAAC;aACzC;YACD,KAAK,EAAE;gBACL,QAAQ,CAAC,QAAQ,EAAE,UAAU,CAAC;gBAC9B,QAAQ,CAAC,QAAQ,EAAE,UAAU,EAAE,EAAE,SAAS,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC;aAC1D;SACF,CAAC,EACF,SAAS,CACV,CAAC;QAEF,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACrC,MAAM,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;QACjD,MAAM,CAAC,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,WAAW,CAAC,CAAC,CAAC;QAClG,MAAM,CAAC,mBAAmB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;QAC9D,MAAM,SAAS,GAAG,eAAe,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC;QAC9C,MAAM,SAAS,GAAG,aAAa,EAAE,CAAC;QAClC,MAAM,MAAM,GAAG,UAAU,CAAC,SAAS,EAAE,SAAS,CAAC,EAAE,CAAC,CAAC;QACnD,YAAY,GAAG,MAAM,CAAC;QAEtB,uBAAuB,GAAG,CAAC,EAAE,MAAM,EAAE,kBAAkB,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;QACpE,qBAAqB,CAAC,GAAG,CACvB,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,QAAQ,CAAC,EAC9B,UAAU,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CACxC,CAAC;QAEF,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,OAAO,CAC9B,UAAU,CAAC;YACT,MAAM,EAAE;gBACN,SAAS,CAAC,QAAQ,EAAE;oBAClB,GAAG,EAAE,QAAQ;oBACb,WAAW,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE;iBAClC,CAAC;aACH;YACD,KAAK,EAAE,CAAC,QAAQ,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;SACrC,CAAC,EACF,SAAS,CACV,CAAC;QAEF,MAAM,OAAO,GAAG,gBAAgB,CAAC,MAAM,CAAC,CAAC;QAEzC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACrC,MAAM,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,CAAC;QAC9B,MAAM,CAAC,OAAO,EAAE,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC;YAC7C,KAAK,EAAE,GAAG;YACV,MAAM,EAAE,GAAG;YACX,SAAS,EAAE,CAAC;YACZ,KAAK,EAAE,GAAG;SACX,CAAC,CAAC;QACH,MAAM,CAAC,OAAO,EAAE,mBAAmB,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC;YACpD,IAAI,EAAE,GAAG;YACT,KAAK,EAAE,KAAK;YACZ,IAAI,EAAE,KAAK;SACZ,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;QACzE,MAAM,SAAS,GAAG,eAAe,CAAC,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC,CAAC;QACzD,MAAM,SAAS,GAAG,aAAa,EAAE,CAAC;QAClC,MAAM,MAAM,GAAG,UAAU,CAAC,SAAS,EAAE,SAAS,CAAC,EAAE,CAAC,CAAC;QACnD,YAAY,GAAG,MAAM,CAAC;QAEtB,uBAAuB,GAAG,CAAC,EAAE,MAAM,EAAE,iBAAiB,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;QACnE,qBAAqB,CAAC,GAAG,CACvB,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,SAAS,CAAC,EAC/B,UAAU,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAC1C,CAAC;QAEF,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,OAAO,CAC9B,UAAU,CAAC;YACT,MAAM,EAAE;gBACN,SAAS,CAAC,SAAS,EAAE,EAAE,GAAG,EAAE,SAAS,EAAE,CAAC;gBACxC,SAAS,CAAC,QAAQ,EAAE,EAAE,GAAG,EAAE,QAAQ,EAAE,CAAC;aACvC;YACD,KAAK,EAAE;gBACL,QAAQ,CAAC,QAAQ,EAAE,SAAS,CAAC;gBAC7B,QAAQ,CAAC,QAAQ,EAAE,QAAQ,EAAE,EAAE,SAAS,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC;aACxD;YACD,KAAK,EAAE;gBACL,WAAW,EAAE,KAAK;aACnB;SACF,CAAC,EACF,SAAS,CACV,CAAC;QAEF,MAAM,OAAO,GAAG,gBAAgB,CAAC,MAAM,CAAC,CAAC;QACzC,MAAM,UAAU,GAAG,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,KAAK,QAAQ,CAAC,CAAC;QAEzF,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAClC,MAAM,CAAC,mBAAmB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACrD,MAAM,CAAC,OAAO,EAAE,aAAa,EAAE,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnD,MAAM,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC1C,MAAM,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC,SAAS,CAAC,2BAA2B,CAAC,CAAC;QACjE,MAAM,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;QACpE,MAAM,SAAS,GAAG,eAAe,CAAC,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC,CAAC;QACvD,MAAM,SAAS,GAAG,aAAa,EAAE,CAAC;QAClC,MAAM,MAAM,GAAG,UAAU,CAAC,SAAS,EAAE,SAAS,CAAC,EAAE,CAAC,CAAC;QACnD,YAAY,GAAG,MAAM,CAAC;QAEtB,IAAI,oBAAyC,CAAC;QAE9C,uBAAuB,GAAG;YACxB,EAAE,MAAM,EAAE,kBAAkB,EAAE,IAAI,EAAE,CAAC,EAAE;YACvC;gBACE,MAAM,EAAE,mBAAmB;gBAC3B,IAAI,EAAE,CAAC;gBACP,OAAO,EAAE,GAAG,EAAE;oBACZ,oBAAoB,GAAG,gBAAgB,CAAC,MAAM,CAAC,EAAE,aAAa,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;gBACnF,CAAC;aACF;SACF,CAAC;QACF,qBAAqB,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,OAAO,CAAC,EAAE,UAAU,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC;QAClG,qBAAqB,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,QAAQ,CAAC,EAAE,UAAU,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;QAElG,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,OAAO,CAC9B,UAAU,CAAC;YACT,MAAM,EAAE;gBACN,SAAS,CAAC,aAAa,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,CAAC;gBAC1C,SAAS,CAAC,cAAc,EAAE,EAAE,GAAG,EAAE,QAAQ,EAAE,CAAC;aAC7C;YACD,KAAK,EAAE;gBACL,QAAQ,CAAC,QAAQ,EAAE,aAAa,CAAC;gBACjC,QAAQ,CAAC,QAAQ,EAAE,cAAc,EAAE,EAAE,SAAS,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC;aAC9D;YACD,KAAK,EAAE;gBACL,WAAW,EAAE,KAAK;aACnB;SACF,CAAC,EACF,SAAS,CACV,CAAC;QAEF,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACrC,MAAM,CAAC,oBAAoB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACxC,MAAM,CAAC,mBAAmB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;QACvD,MAAM,SAAS,GAAG,eAAe,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC;QACnD,MAAM,SAAS,GAAG,aAAa,EAAE,CAAC;QAClC,MAAM,MAAM,GAAG,UAAU,CAAC,SAAS,EAAE,SAAS,CAAC,EAAE,CAAC,CAAC;QACnD,YAAY,GAAG,MAAM,CAAC;QAEtB,IAAI,gBAAoC,CAAC;QAEzC,uBAAuB,GAAG;YACxB,EAAE,MAAM,EAAE,wBAAwB,EAAE,IAAI,EAAE,CAAC,EAAE;YAC7C;gBACE,MAAM,EAAE,mBAAmB;gBAC3B,IAAI,EAAE,CAAC;gBACP,OAAO,EAAE,GAAG,EAAE;oBACZ,gBAAgB,GAAG,gBAAgB,CAAC,MAAM,CAAC,EAAE,YAAY,CAAC,YAAY,CAAC,CAAC,KAAK,CAAC;gBAChF,CAAC;aACF;SACF,CAAC;QACF,sBAAsB,GAAG;YACvB,UAAU,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE,EAAE,WAAW,EAAE,QAAQ,EAAE,CAAC;YAClE,UAAU,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC;SACxC,CAAC;QAEF,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,OAAO,CAC9B,UAAU,CAAC;YACT,MAAM,EAAE;gBACN,SAAS,CAAC,aAAa,EAAE;oBACvB,GAAG,EAAE,aAAa;oBAClB,WAAW,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE;iBAClC,CAAC;aACH;YACD,KAAK,EAAE,CAAC,QAAQ,CAAC,YAAY,EAAE,aAAa,EAAE,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;SAC/D,CAAC,EACF,SAAS,CACV,CAAC;QAEF,MAAM,OAAO,GAAG,gBAAgB,CAAC,MAAM,CAAC,CAAC;QAEzC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACrC,MAAM,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACnC,MAAM,CAAC,OAAO,EAAE,YAAY,CAAC,YAAY,CAAC,CAAC,CAAC,OAAO,CAAC;YAClD,KAAK,EAAE,GAAG;YACV,MAAM,EAAE,GAAG;YACX,SAAS,EAAE,CAAC;YACZ,KAAK,EAAE,GAAG;SACX,CAAC,CAAC;QACH,MAAM,CAAC,OAAO,EAAE,mBAAmB,CAAC,YAAY,CAAC,CAAC,CAAC,OAAO,CAAC;YACzD,IAAI,EAAE,GAAG;YACT,KAAK,EAAE,KAAK;YACZ,IAAI,EAAE,KAAK;SACZ,CAAC,CAAC;QACH,MAAM,CAAC,qBAAqB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IACzD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;QACrD,MAAM,SAAS,GAAG,eAAe,CAAC,CAAC,YAAY,EAAE,YAAY,CAAC,CAAC,CAAC;QAChE,MAAM,SAAS,GAAG,aAAa,EAAE,CAAC;QAClC,MAAM,MAAM,GAAG,UAAU,CAAC,SAAS,EAAE,SAAS,CAAC,EAAE,CAAC,CAAC;QACnD,YAAY,GAAG,MAAM,CAAC;QAEtB,uBAAuB,GAAG;YACxB,EAAE,MAAM,EAAE,cAAc,EAAE,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE;YAChD,EAAE,MAAM,EAAE,cAAc,EAAE,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE;SAChD,CAAC;QACF,qBAAqB,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,YAAY,CAAC,EAAE,UAAU,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC;QACvG,qBAAqB,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,YAAY,CAAC,EAAE,UAAU,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC;QAEvG,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,OAAO,CAC9B,UAAU,CAAC;YACT,MAAM,EAAE;gBACN,SAAS,CAAC,YAAY,EAAE,EAAE,GAAG,EAAE,YAAY,EAAE,CAAC;gBAC9C,SAAS,CAAC,YAAY,EAAE,EAAE,GAAG,EAAE,YAAY,EAAE,CAAC;aAC/C;YACD,KAAK,EAAE;gBACL,QAAQ,CAAC,YAAY,EAAE,YAAY,CAAC;gBACpC,QAAQ,CAAC,YAAY,EAAE,YAAY,CAAC;aACrC;YACD,KAAK,EAAE;gBACL,WAAW,EAAE,KAAK;aACnB;SACF,CAAC,EACF,SAAS,CACV,CAAC;QAEF,MAAM,OAAO,GAAG,gBAAgB,CAAC,MAAM,CAAC,CAAC;QAEzC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACrC,MAAM,CAAC,OAAO,EAAE,YAAY,CAAC,YAAY,CAAC,CAAC,CAAC,OAAO,CAAC;YAClD,KAAK,EAAE,GAAG;YACV,MAAM,EAAE,GAAG;YACX,SAAS,EAAE,CAAC;YACZ,KAAK,EAAE,GAAG;SACX,CAAC,CAAC;QACH,MAAM,CAAC,OAAO,EAAE,YAAY,CAAC,YAAY,CAAC,CAAC,CAAC,OAAO,CAAC;YAClD,KAAK,EAAE,GAAG;YACV,MAAM,EAAE,GAAG;YACX,SAAS,EAAE,CAAC;YACZ,KAAK,EAAE,GAAG;SACX,CAAC,CAAC;QACH,MAAM,CAAC,OAAO,EAAE,aAAa,EAAE,CAAC,CAAC,OAAO,CAAC;YACvC,KAAK,EAAE,KAAK;YACZ,MAAM,EAAE,GAAG;YACX,SAAS,EAAE,CAAC;YACZ,KAAK,EAAE,KAAK;SACb,CAAC,CAAC;QACH,MAAM,CAAC,OAAO,EAAE,uBAAuB,EAAE,EAAE,QAAQ,CAAC,CAAC,OAAO,CAAC;YAC3D,IAAI,EAAE,KAAK;YACX,KAAK,EAAE,KAAK;YACZ,SAAS,EAAE,KAAK;SACjB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"budget-tracker.test.d.ts","sourceRoot":"","sources":["../../../src/workflows/__tests__/budget-tracker.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,99 @@
|
|
|
1
|
+
import { describe, expect, it } from 'vitest';
|
|
2
|
+
import { BudgetExceededError, BudgetTracker, } from '../budget-tracker.js';
|
|
3
|
+
function expectUsage(actual, expected) {
|
|
4
|
+
expect(actual).toEqual(expected);
|
|
5
|
+
}
|
|
6
|
+
describe('BudgetTracker', () => {
|
|
7
|
+
it('tracks usage across multiple steps', () => {
|
|
8
|
+
const tracker = new BudgetTracker({ perAgent: 100, perWorkflow: 500 });
|
|
9
|
+
tracker.recordUsage('planner', { input: 10, output: 5, cacheRead: 3 });
|
|
10
|
+
tracker.recordUsage('writer', { input: 20, output: 4 });
|
|
11
|
+
tracker.recordUsage('planner', { input: 1, output: 1, cacheRead: 2 });
|
|
12
|
+
expectUsage(tracker.getStepUsage('planner'), {
|
|
13
|
+
input: 11,
|
|
14
|
+
output: 6,
|
|
15
|
+
cacheRead: 5,
|
|
16
|
+
total: 17,
|
|
17
|
+
});
|
|
18
|
+
expectUsage(tracker.getStepUsage('writer'), {
|
|
19
|
+
input: 20,
|
|
20
|
+
output: 4,
|
|
21
|
+
cacheRead: 0,
|
|
22
|
+
total: 24,
|
|
23
|
+
});
|
|
24
|
+
expectUsage(tracker.getTotalUsage(), {
|
|
25
|
+
input: 31,
|
|
26
|
+
output: 10,
|
|
27
|
+
cacheRead: 5,
|
|
28
|
+
total: 41,
|
|
29
|
+
});
|
|
30
|
+
// total = input + output (cacheRead excluded from budget)
|
|
31
|
+
expect(tracker.getRemainingBudget()).toEqual({
|
|
32
|
+
agent: 59,
|
|
33
|
+
workflow: 459,
|
|
34
|
+
});
|
|
35
|
+
});
|
|
36
|
+
it('detects when a step exceeds the per-agent budget', () => {
|
|
37
|
+
const tracker = new BudgetTracker({ perAgent: 25, perWorkflow: 100 });
|
|
38
|
+
tracker.recordUsage('specialist', { input: 18, output: 9 });
|
|
39
|
+
expect(tracker.isOverBudget('specialist')).toEqual({
|
|
40
|
+
over: true,
|
|
41
|
+
reason: 'Step "specialist" exceeded per-agent budget (27/25)',
|
|
42
|
+
});
|
|
43
|
+
});
|
|
44
|
+
it('detects when total usage exceeds the per-workflow budget', () => {
|
|
45
|
+
const tracker = new BudgetTracker({ perAgent: 100, perWorkflow: 40 });
|
|
46
|
+
tracker.recordUsage('step-a', { input: 10, output: 10 });
|
|
47
|
+
tracker.recordUsage('step-b', { input: 15, output: 10 });
|
|
48
|
+
expect(tracker.isOverBudget('step-b')).toEqual({
|
|
49
|
+
over: true,
|
|
50
|
+
reason: 'Workflow exceeded total budget (45/40)',
|
|
51
|
+
});
|
|
52
|
+
expect(tracker.checkCanSpawn('step-c')).toEqual({
|
|
53
|
+
allowed: false,
|
|
54
|
+
reason: 'Cannot spawn step-c: workflow budget exceeded (45/40)',
|
|
55
|
+
});
|
|
56
|
+
});
|
|
57
|
+
it('refuses to spawn when the remaining workflow budget is nearly exhausted', () => {
|
|
58
|
+
const tracker = new BudgetTracker({ perAgent: 100, perWorkflow: 250 });
|
|
59
|
+
tracker.recordUsage('lead', { input: 120, output: 121 });
|
|
60
|
+
expect(tracker.checkCanSpawn('reviewer')).toEqual({
|
|
61
|
+
allowed: false,
|
|
62
|
+
reason: 'Cannot spawn reviewer: remaining workflow budget 9 is below step budget 100',
|
|
63
|
+
});
|
|
64
|
+
});
|
|
65
|
+
it('maintains correct totals when parallel steps record usage concurrently', async () => {
|
|
66
|
+
const tracker = new BudgetTracker({ perAgent: 1_000, perWorkflow: 10_000 });
|
|
67
|
+
await Promise.all(Array.from({ length: 40 }, async (_, index) => {
|
|
68
|
+
await new Promise((resolve) => setTimeout(resolve, index % 5));
|
|
69
|
+
tracker.recordUsage(`step-${index % 4}`, {
|
|
70
|
+
input: 2,
|
|
71
|
+
output: 3,
|
|
72
|
+
cacheRead: 1,
|
|
73
|
+
});
|
|
74
|
+
}));
|
|
75
|
+
expectUsage(tracker.getTotalUsage(), {
|
|
76
|
+
input: 80,
|
|
77
|
+
output: 120,
|
|
78
|
+
cacheRead: 40,
|
|
79
|
+
total: 200,
|
|
80
|
+
});
|
|
81
|
+
for (const stepName of ['step-0', 'step-1', 'step-2', 'step-3']) {
|
|
82
|
+
expectUsage(tracker.getStepUsage(stepName), {
|
|
83
|
+
input: 20,
|
|
84
|
+
output: 30,
|
|
85
|
+
cacheRead: 10,
|
|
86
|
+
total: 50,
|
|
87
|
+
});
|
|
88
|
+
}
|
|
89
|
+
});
|
|
90
|
+
it('exposes budget metadata on BudgetExceededError', () => {
|
|
91
|
+
const error = new BudgetExceededError('planner', 'workflow', 100, 125);
|
|
92
|
+
expect(error.name).toBe('BudgetExceededError');
|
|
93
|
+
expect(error.stepName).toBe('planner');
|
|
94
|
+
expect(error.budgetType).toBe('workflow');
|
|
95
|
+
expect(error.limit).toBe(100);
|
|
96
|
+
expect(error.actual).toBe(125);
|
|
97
|
+
});
|
|
98
|
+
});
|
|
99
|
+
//# sourceMappingURL=budget-tracker.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"budget-tracker.test.js","sourceRoot":"","sources":["../../../src/workflows/__tests__/budget-tracker.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAE9C,OAAO,EACL,mBAAmB,EACnB,aAAa,GAEd,MAAM,sBAAsB,CAAC;AAE9B,SAAS,WAAW,CAAC,MAAkB,EAAE,QAAoB;IAC3D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;AACnC,CAAC;AAED,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;QAC5C,MAAM,OAAO,GAAG,IAAI,aAAa,CAAC,EAAE,QAAQ,EAAE,GAAG,EAAE,WAAW,EAAE,GAAG,EAAE,CAAC,CAAC;QAEvE,OAAO,CAAC,WAAW,CAAC,SAAS,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC,CAAC;QACvE,OAAO,CAAC,WAAW,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC;QACxD,OAAO,CAAC,WAAW,CAAC,SAAS,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC,CAAC;QAEtE,WAAW,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS,CAAC,EAAE;YAC3C,KAAK,EAAE,EAAE;YACT,MAAM,EAAE,CAAC;YACT,SAAS,EAAE,CAAC;YACZ,KAAK,EAAE,EAAE;SACV,CAAC,CAAC;QACH,WAAW,CAAC,OAAO,CAAC,YAAY,CAAC,QAAQ,CAAC,EAAE;YAC1C,KAAK,EAAE,EAAE;YACT,MAAM,EAAE,CAAC;YACT,SAAS,EAAE,CAAC;YACZ,KAAK,EAAE,EAAE;SACV,CAAC,CAAC;QACH,WAAW,CAAC,OAAO,CAAC,aAAa,EAAE,EAAE;YACnC,KAAK,EAAE,EAAE;YACT,MAAM,EAAE,EAAE;YACV,SAAS,EAAE,CAAC;YACZ,KAAK,EAAE,EAAE;SACV,CAAC,CAAC;QACH,0DAA0D;QAC1D,MAAM,CAAC,OAAO,CAAC,kBAAkB,EAAE,CAAC,CAAC,OAAO,CAAC;YAC3C,KAAK,EAAE,EAAE;YACT,QAAQ,EAAE,GAAG;SACd,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,MAAM,OAAO,GAAG,IAAI,aAAa,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,WAAW,EAAE,GAAG,EAAE,CAAC,CAAC;QAEtE,OAAO,CAAC,WAAW,CAAC,YAAY,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC;QAE5D,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,YAAY,CAAC,CAAC,CAAC,OAAO,CAAC;YACjD,IAAI,EAAE,IAAI;YACV,MAAM,EAAE,qDAAqD;SAC9D,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;QAClE,MAAM,OAAO,GAAG,IAAI,aAAa,CAAC,EAAE,QAAQ,EAAE,GAAG,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC,CAAC;QAEtE,OAAO,CAAC,WAAW,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;QACzD,OAAO,CAAC,WAAW,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;QAEzD,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC;YAC7C,IAAI,EAAE,IAAI;YACV,MAAM,EAAE,wCAAwC;SACjD,CAAC,CAAC;QACH,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC;YAC9C,OAAO,EAAE,KAAK;YACd,MAAM,EAAE,uDAAuD;SAChE,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yEAAyE,EAAE,GAAG,EAAE;QACjF,MAAM,OAAO,GAAG,IAAI,aAAa,CAAC,EAAE,QAAQ,EAAE,GAAG,EAAE,WAAW,EAAE,GAAG,EAAE,CAAC,CAAC;QAEvE,OAAO,CAAC,WAAW,CAAC,MAAM,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC;QAEzD,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC,CAAC,OAAO,CAAC;YAChD,OAAO,EAAE,KAAK;YACd,MAAM,EAAE,6EAA6E;SACtF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wEAAwE,EAAE,KAAK,IAAI,EAAE;QACtF,MAAM,OAAO,GAAG,IAAI,aAAa,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,WAAW,EAAE,MAAM,EAAE,CAAC,CAAC;QAE5E,MAAM,OAAO,CAAC,GAAG,CACf,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE;YAC5C,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC;YAC/D,OAAO,CAAC,WAAW,CAAC,QAAQ,KAAK,GAAG,CAAC,EAAE,EAAE;gBACvC,KAAK,EAAE,CAAC;gBACR,MAAM,EAAE,CAAC;gBACT,SAAS,EAAE,CAAC;aACb,CAAC,CAAC;QACL,CAAC,CAAC,CACH,CAAC;QAEF,WAAW,CAAC,OAAO,CAAC,aAAa,EAAE,EAAE;YACnC,KAAK,EAAE,EAAE;YACT,MAAM,EAAE,GAAG;YACX,SAAS,EAAE,EAAE;YACb,KAAK,EAAE,GAAG;SACX,CAAC,CAAC;QAEH,KAAK,MAAM,QAAQ,IAAI,CAAC,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAE,QAAQ,CAAC,EAAE,CAAC;YAChE,WAAW,CAAC,OAAO,CAAC,YAAY,CAAC,QAAQ,CAAC,EAAE;gBAC1C,KAAK,EAAE,EAAE;gBACT,MAAM,EAAE,EAAE;gBACV,SAAS,EAAE,EAAE;gBACb,KAAK,EAAE,EAAE;aACV,CAAC,CAAC;QACL,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;QACxD,MAAM,KAAK,GAAG,IAAI,mBAAmB,CAAC,SAAS,EAAE,UAAU,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;QAEvE,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QAC/C,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACvC,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC1C,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC9B,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IACjC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"proxy-env.test.d.ts","sourceRoot":"","sources":["../../../src/workflows/__tests__/proxy-env.test.ts"],"names":[],"mappings":""}
|