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.
Files changed (211) hide show
  1. package/README.md +3 -6
  2. package/dist/index.cjs +776 -41
  3. package/dist/src/cli/bootstrap.js.map +1 -1
  4. package/dist/src/cli/commands/cloud.js.map +1 -1
  5. package/dist/src/cli/commands/core.d.ts.map +1 -1
  6. package/dist/src/cli/commands/core.js.map +1 -1
  7. package/dist/src/cli/commands/messaging.d.ts.map +1 -1
  8. package/dist/src/cli/commands/messaging.js.map +1 -1
  9. package/dist/src/cli/commands/monitoring.d.ts.map +1 -1
  10. package/dist/src/cli/commands/monitoring.js.map +1 -1
  11. package/node_modules/@agent-relay/cloud/dist/workflows.d.ts +12 -0
  12. package/node_modules/@agent-relay/cloud/dist/workflows.d.ts.map +1 -1
  13. package/node_modules/@agent-relay/cloud/dist/workflows.js +38 -0
  14. package/node_modules/@agent-relay/cloud/dist/workflows.js.map +1 -1
  15. package/node_modules/@agent-relay/cloud/package.json +2 -2
  16. package/node_modules/@agent-relay/config/package.json +2 -2
  17. package/node_modules/@agent-relay/hooks/package.json +4 -4
  18. package/node_modules/@agent-relay/sdk/dist/relay.d.ts.map +1 -1
  19. package/node_modules/@agent-relay/sdk/dist/relay.js.map +1 -1
  20. package/node_modules/@agent-relay/sdk/dist/workflows/__tests__/budget-enforcement.test.d.ts +2 -0
  21. package/node_modules/@agent-relay/sdk/dist/workflows/__tests__/budget-enforcement.test.d.ts.map +1 -0
  22. package/node_modules/@agent-relay/sdk/dist/workflows/__tests__/budget-enforcement.test.js +437 -0
  23. package/node_modules/@agent-relay/sdk/dist/workflows/__tests__/budget-enforcement.test.js.map +1 -0
  24. package/node_modules/@agent-relay/sdk/dist/workflows/__tests__/budget-tracker.test.d.ts +2 -0
  25. package/node_modules/@agent-relay/sdk/dist/workflows/__tests__/budget-tracker.test.d.ts.map +1 -0
  26. package/node_modules/@agent-relay/sdk/dist/workflows/__tests__/budget-tracker.test.js +99 -0
  27. package/node_modules/@agent-relay/sdk/dist/workflows/__tests__/budget-tracker.test.js.map +1 -0
  28. package/node_modules/@agent-relay/sdk/dist/workflows/__tests__/proxy-env.test.d.ts +2 -0
  29. package/node_modules/@agent-relay/sdk/dist/workflows/__tests__/proxy-env.test.d.ts.map +1 -0
  30. package/node_modules/@agent-relay/sdk/dist/workflows/__tests__/proxy-env.test.js +135 -0
  31. package/node_modules/@agent-relay/sdk/dist/workflows/__tests__/proxy-env.test.js.map +1 -0
  32. package/node_modules/@agent-relay/sdk/dist/workflows/__tests__/verification-custom.test.d.ts +2 -0
  33. package/node_modules/@agent-relay/sdk/dist/workflows/__tests__/verification-custom.test.d.ts.map +1 -0
  34. package/node_modules/@agent-relay/sdk/dist/workflows/__tests__/verification-custom.test.js +236 -0
  35. package/node_modules/@agent-relay/sdk/dist/workflows/__tests__/verification-custom.test.js.map +1 -0
  36. package/node_modules/@agent-relay/sdk/dist/workflows/__tests__/verification-traceback.test.d.ts +2 -0
  37. package/node_modules/@agent-relay/sdk/dist/workflows/__tests__/verification-traceback.test.d.ts.map +1 -0
  38. package/node_modules/@agent-relay/sdk/dist/workflows/__tests__/verification-traceback.test.js +448 -0
  39. package/node_modules/@agent-relay/sdk/dist/workflows/__tests__/verification-traceback.test.js.map +1 -0
  40. package/node_modules/@agent-relay/sdk/dist/workflows/__tests__/verification.test.js +71 -4
  41. package/node_modules/@agent-relay/sdk/dist/workflows/__tests__/verification.test.js.map +1 -1
  42. package/node_modules/@agent-relay/sdk/dist/workflows/budget-tracker.d.ts +75 -0
  43. package/node_modules/@agent-relay/sdk/dist/workflows/budget-tracker.d.ts.map +1 -0
  44. package/node_modules/@agent-relay/sdk/dist/workflows/budget-tracker.js +180 -0
  45. package/node_modules/@agent-relay/sdk/dist/workflows/budget-tracker.js.map +1 -0
  46. package/node_modules/@agent-relay/sdk/dist/workflows/builder.d.ts +1 -0
  47. package/node_modules/@agent-relay/sdk/dist/workflows/builder.d.ts.map +1 -1
  48. package/node_modules/@agent-relay/sdk/dist/workflows/builder.js +17 -2
  49. package/node_modules/@agent-relay/sdk/dist/workflows/builder.js.map +1 -1
  50. package/node_modules/@agent-relay/sdk/dist/workflows/index.d.ts +2 -0
  51. package/node_modules/@agent-relay/sdk/dist/workflows/index.d.ts.map +1 -1
  52. package/node_modules/@agent-relay/sdk/dist/workflows/index.js +2 -0
  53. package/node_modules/@agent-relay/sdk/dist/workflows/index.js.map +1 -1
  54. package/node_modules/@agent-relay/sdk/dist/workflows/proxy-env.d.ts +52 -0
  55. package/node_modules/@agent-relay/sdk/dist/workflows/proxy-env.d.ts.map +1 -0
  56. package/node_modules/@agent-relay/sdk/dist/workflows/proxy-env.js +92 -0
  57. package/node_modules/@agent-relay/sdk/dist/workflows/proxy-env.js.map +1 -0
  58. package/node_modules/@agent-relay/sdk/dist/workflows/run-summary-table.d.ts +2 -1
  59. package/node_modules/@agent-relay/sdk/dist/workflows/run-summary-table.d.ts.map +1 -1
  60. package/node_modules/@agent-relay/sdk/dist/workflows/run-summary-table.js +41 -9
  61. package/node_modules/@agent-relay/sdk/dist/workflows/run-summary-table.js.map +1 -1
  62. package/node_modules/@agent-relay/sdk/dist/workflows/runner.d.ts +17 -0
  63. package/node_modules/@agent-relay/sdk/dist/workflows/runner.d.ts.map +1 -1
  64. package/node_modules/@agent-relay/sdk/dist/workflows/runner.js +390 -16
  65. package/node_modules/@agent-relay/sdk/dist/workflows/runner.js.map +1 -1
  66. package/node_modules/@agent-relay/sdk/dist/workflows/types.d.ts +47 -1
  67. package/node_modules/@agent-relay/sdk/dist/workflows/types.d.ts.map +1 -1
  68. package/node_modules/@agent-relay/sdk/dist/workflows/types.js.map +1 -1
  69. package/node_modules/@agent-relay/sdk/dist/workflows/verification.d.ts +9 -0
  70. package/node_modules/@agent-relay/sdk/dist/workflows/verification.d.ts.map +1 -1
  71. package/node_modules/@agent-relay/sdk/dist/workflows/verification.js +78 -2
  72. package/node_modules/@agent-relay/sdk/dist/workflows/verification.js.map +1 -1
  73. package/node_modules/@agent-relay/sdk/package.json +7 -3
  74. package/node_modules/@agent-relay/telemetry/dist/config.d.ts.map +1 -1
  75. package/node_modules/@agent-relay/telemetry/dist/config.js +2 -4
  76. package/node_modules/@agent-relay/telemetry/dist/config.js.map +1 -1
  77. package/node_modules/@agent-relay/telemetry/dist/events.d.ts +1 -2
  78. package/node_modules/@agent-relay/telemetry/dist/events.d.ts.map +1 -1
  79. package/node_modules/@agent-relay/telemetry/dist/index.d.ts +1 -1
  80. package/node_modules/@agent-relay/telemetry/dist/index.d.ts.map +1 -1
  81. package/node_modules/@agent-relay/telemetry/dist/index.js +1 -1
  82. package/node_modules/@agent-relay/telemetry/dist/index.js.map +1 -1
  83. package/node_modules/@agent-relay/telemetry/package.json +1 -1
  84. package/node_modules/@agent-relay/trajectory/package.json +2 -2
  85. package/node_modules/@agent-relay/user-directory/package.json +2 -2
  86. package/node_modules/@agent-relay/utils/package.json +2 -2
  87. package/node_modules/@aws-sdk/core/dist-cjs/index.js +2 -0
  88. package/node_modules/@aws-sdk/core/dist-cjs/submodules/client/index.js +3 -0
  89. package/node_modules/@aws-sdk/core/dist-es/submodules/client/setFeature.js +2 -0
  90. package/node_modules/@aws-sdk/core/dist-types/submodules/client/setFeature.d.ts +1 -1
  91. package/node_modules/@aws-sdk/core/package.json +6 -4
  92. package/node_modules/@aws-sdk/credential-provider-env/package.json +2 -2
  93. package/node_modules/@aws-sdk/credential-provider-http/package.json +5 -5
  94. package/node_modules/@aws-sdk/credential-provider-ini/package.json +9 -9
  95. package/node_modules/@aws-sdk/credential-provider-login/package.json +3 -3
  96. package/node_modules/@aws-sdk/credential-provider-node/package.json +7 -7
  97. package/node_modules/@aws-sdk/credential-provider-process/package.json +2 -2
  98. package/node_modules/@aws-sdk/credential-provider-sso/package.json +4 -4
  99. package/node_modules/@aws-sdk/credential-provider-web-identity/package.json +3 -3
  100. package/node_modules/@aws-sdk/middleware-flexible-checksums/package.json +4 -4
  101. package/node_modules/@aws-sdk/middleware-sdk-s3/package.json +5 -5
  102. package/node_modules/@aws-sdk/middleware-user-agent/package.json +5 -5
  103. package/node_modules/@aws-sdk/nested-clients/package.json +18 -18
  104. package/node_modules/@aws-sdk/region-config-resolver/package.json +2 -2
  105. package/node_modules/@aws-sdk/signature-v4-multi-region/package.json +2 -2
  106. package/node_modules/@aws-sdk/token-providers/package.json +3 -3
  107. package/node_modules/@aws-sdk/util-endpoints/package.json +2 -2
  108. package/node_modules/@aws-sdk/util-user-agent-node/package.json +2 -2
  109. package/node_modules/@relaycast/sdk/dist/version.d.ts +1 -1
  110. package/node_modules/@relaycast/sdk/dist/version.js +1 -1
  111. package/node_modules/@relaycast/sdk/node_modules/@relaycast/types/package.json +1 -1
  112. package/node_modules/@relaycast/sdk/package.json +2 -2
  113. package/node_modules/axios/CHANGELOG.md +112 -44
  114. package/node_modules/axios/README.md +30 -0
  115. package/node_modules/axios/dist/axios.js +34 -11
  116. package/node_modules/axios/dist/axios.js.map +1 -1
  117. package/node_modules/axios/dist/axios.min.js +2 -2
  118. package/node_modules/axios/dist/axios.min.js.map +1 -1
  119. package/node_modules/axios/dist/browser/axios.cjs +32 -6
  120. package/node_modules/axios/dist/browser/axios.cjs.map +1 -1
  121. package/node_modules/axios/dist/esm/axios.js +32 -6
  122. package/node_modules/axios/dist/esm/axios.js.map +1 -1
  123. package/node_modules/axios/dist/esm/axios.min.js +2 -2
  124. package/node_modules/axios/dist/esm/axios.min.js.map +1 -1
  125. package/node_modules/axios/dist/node/axios.cjs +91 -37
  126. package/node_modules/axios/dist/node/axios.cjs.map +1 -1
  127. package/node_modules/axios/index.d.cts +1 -0
  128. package/node_modules/axios/index.d.ts +1 -0
  129. package/node_modules/axios/lib/adapters/http.js +69 -22
  130. package/node_modules/axios/lib/core/mergeConfig.js +13 -1
  131. package/node_modules/axios/lib/env/data.js +1 -1
  132. package/node_modules/axios/lib/helpers/resolveConfig.js +14 -2
  133. package/node_modules/axios/lib/helpers/validator.js +3 -1
  134. package/node_modules/axios/package.json +1 -1
  135. package/package.json +9 -9
  136. package/packages/cloud/dist/workflows.d.ts +12 -0
  137. package/packages/cloud/dist/workflows.d.ts.map +1 -1
  138. package/packages/cloud/dist/workflows.js +38 -0
  139. package/packages/cloud/dist/workflows.js.map +1 -1
  140. package/packages/cloud/package.json +2 -2
  141. package/packages/config/package.json +2 -2
  142. package/packages/hooks/package.json +4 -4
  143. package/packages/sdk/dist/relay.d.ts.map +1 -1
  144. package/packages/sdk/dist/relay.js.map +1 -1
  145. package/packages/sdk/dist/workflows/__tests__/budget-enforcement.test.d.ts +2 -0
  146. package/packages/sdk/dist/workflows/__tests__/budget-enforcement.test.d.ts.map +1 -0
  147. package/packages/sdk/dist/workflows/__tests__/budget-enforcement.test.js +437 -0
  148. package/packages/sdk/dist/workflows/__tests__/budget-enforcement.test.js.map +1 -0
  149. package/packages/sdk/dist/workflows/__tests__/budget-tracker.test.d.ts +2 -0
  150. package/packages/sdk/dist/workflows/__tests__/budget-tracker.test.d.ts.map +1 -0
  151. package/packages/sdk/dist/workflows/__tests__/budget-tracker.test.js +99 -0
  152. package/packages/sdk/dist/workflows/__tests__/budget-tracker.test.js.map +1 -0
  153. package/packages/sdk/dist/workflows/__tests__/proxy-env.test.d.ts +2 -0
  154. package/packages/sdk/dist/workflows/__tests__/proxy-env.test.d.ts.map +1 -0
  155. package/packages/sdk/dist/workflows/__tests__/proxy-env.test.js +135 -0
  156. package/packages/sdk/dist/workflows/__tests__/proxy-env.test.js.map +1 -0
  157. package/packages/sdk/dist/workflows/__tests__/verification-custom.test.d.ts +2 -0
  158. package/packages/sdk/dist/workflows/__tests__/verification-custom.test.d.ts.map +1 -0
  159. package/packages/sdk/dist/workflows/__tests__/verification-custom.test.js +236 -0
  160. package/packages/sdk/dist/workflows/__tests__/verification-custom.test.js.map +1 -0
  161. package/packages/sdk/dist/workflows/__tests__/verification-traceback.test.d.ts +2 -0
  162. package/packages/sdk/dist/workflows/__tests__/verification-traceback.test.d.ts.map +1 -0
  163. package/packages/sdk/dist/workflows/__tests__/verification-traceback.test.js +448 -0
  164. package/packages/sdk/dist/workflows/__tests__/verification-traceback.test.js.map +1 -0
  165. package/packages/sdk/dist/workflows/__tests__/verification.test.js +71 -4
  166. package/packages/sdk/dist/workflows/__tests__/verification.test.js.map +1 -1
  167. package/packages/sdk/dist/workflows/budget-tracker.d.ts +75 -0
  168. package/packages/sdk/dist/workflows/budget-tracker.d.ts.map +1 -0
  169. package/packages/sdk/dist/workflows/budget-tracker.js +180 -0
  170. package/packages/sdk/dist/workflows/budget-tracker.js.map +1 -0
  171. package/packages/sdk/dist/workflows/builder.d.ts +1 -0
  172. package/packages/sdk/dist/workflows/builder.d.ts.map +1 -1
  173. package/packages/sdk/dist/workflows/builder.js +17 -2
  174. package/packages/sdk/dist/workflows/builder.js.map +1 -1
  175. package/packages/sdk/dist/workflows/index.d.ts +2 -0
  176. package/packages/sdk/dist/workflows/index.d.ts.map +1 -1
  177. package/packages/sdk/dist/workflows/index.js +2 -0
  178. package/packages/sdk/dist/workflows/index.js.map +1 -1
  179. package/packages/sdk/dist/workflows/proxy-env.d.ts +52 -0
  180. package/packages/sdk/dist/workflows/proxy-env.d.ts.map +1 -0
  181. package/packages/sdk/dist/workflows/proxy-env.js +92 -0
  182. package/packages/sdk/dist/workflows/proxy-env.js.map +1 -0
  183. package/packages/sdk/dist/workflows/run-summary-table.d.ts +2 -1
  184. package/packages/sdk/dist/workflows/run-summary-table.d.ts.map +1 -1
  185. package/packages/sdk/dist/workflows/run-summary-table.js +41 -9
  186. package/packages/sdk/dist/workflows/run-summary-table.js.map +1 -1
  187. package/packages/sdk/dist/workflows/runner.d.ts +17 -0
  188. package/packages/sdk/dist/workflows/runner.d.ts.map +1 -1
  189. package/packages/sdk/dist/workflows/runner.js +390 -16
  190. package/packages/sdk/dist/workflows/runner.js.map +1 -1
  191. package/packages/sdk/dist/workflows/types.d.ts +47 -1
  192. package/packages/sdk/dist/workflows/types.d.ts.map +1 -1
  193. package/packages/sdk/dist/workflows/types.js.map +1 -1
  194. package/packages/sdk/dist/workflows/verification.d.ts +9 -0
  195. package/packages/sdk/dist/workflows/verification.d.ts.map +1 -1
  196. package/packages/sdk/dist/workflows/verification.js +78 -2
  197. package/packages/sdk/dist/workflows/verification.js.map +1 -1
  198. package/packages/sdk/package.json +7 -3
  199. package/packages/telemetry/dist/config.d.ts.map +1 -1
  200. package/packages/telemetry/dist/config.js +2 -4
  201. package/packages/telemetry/dist/config.js.map +1 -1
  202. package/packages/telemetry/dist/events.d.ts +1 -2
  203. package/packages/telemetry/dist/events.d.ts.map +1 -1
  204. package/packages/telemetry/dist/index.d.ts +1 -1
  205. package/packages/telemetry/dist/index.d.ts.map +1 -1
  206. package/packages/telemetry/dist/index.js +1 -1
  207. package/packages/telemetry/dist/index.js.map +1 -1
  208. package/packages/telemetry/package.json +1 -1
  209. package/packages/trajectory/package.json +2 -2
  210. package/packages/user-directory/package.json +2 -2
  211. 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,2 @@
1
+ export {};
2
+ //# sourceMappingURL=budget-tracker.test.d.ts.map
@@ -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,2 @@
1
+ export {};
2
+ //# sourceMappingURL=proxy-env.test.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"proxy-env.test.d.ts","sourceRoot":"","sources":["../../../src/workflows/__tests__/proxy-env.test.ts"],"names":[],"mappings":""}