@wrongstack/core 0.1.10 → 0.3.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/{agent-bridge-6KPqsFx6.d.ts → agent-bridge-C3DUGjSb.d.ts} +1 -1
- package/dist/{compactor-B4mQZXf2.d.ts → compactor-BUU6Zm_3.d.ts} +1 -1
- package/dist/{config-BU9f_5yH.d.ts → config-CKLYPkCi.d.ts} +1 -1
- package/dist/{context-BmM2xGUZ.d.ts → context-IovtuTf8.d.ts} +10 -0
- package/dist/coordination/index.d.ts +211 -13
- package/dist/coordination/index.js +964 -67
- package/dist/coordination/index.js.map +1 -1
- package/dist/defaults/index.d.ts +33 -18
- package/dist/defaults/index.js +1273 -42
- package/dist/defaults/index.js.map +1 -1
- package/dist/{events-BMNaEFZl.d.ts → events-CNB9PALO.d.ts} +99 -1
- package/dist/execution/index.d.ts +12 -12
- package/dist/extension/index.d.ts +9 -0
- package/dist/extension/index.js +234 -0
- package/dist/extension/index.js.map +1 -0
- package/dist/index-BDb0cAMP.d.ts +806 -0
- package/dist/index.d.ts +112 -29
- package/dist/index.js +2036 -490
- package/dist/index.js.map +1 -1
- package/dist/infrastructure/index.d.ts +6 -6
- package/dist/kernel/index.d.ts +12 -9
- package/dist/kernel/index.js +73 -7
- package/dist/kernel/index.js.map +1 -1
- package/dist/{mcp-servers-Dzgg4x1w.d.ts → mcp-servers-DR35ojJZ.d.ts} +3 -3
- package/dist/models/index.d.ts +2 -2
- package/dist/models/index.js +24 -1
- package/dist/models/index.js.map +1 -1
- package/dist/{multi-agent-fmkRHtof.d.ts → multi-agent-B9a6sflH.d.ts} +71 -3
- package/dist/observability/index.d.ts +2 -2
- package/dist/{path-resolver-DBjaoXFq.d.ts → path-resolver-Cl_q0u-R.d.ts} +2 -2
- package/dist/provider-runner-BXuADQqQ.d.ts +36 -0
- package/dist/sdd/index.d.ts +3 -3
- package/dist/{secret-scrubber-CicHLN4G.d.ts → secret-scrubber-CgG2tV2B.d.ts} +1 -1
- package/dist/{secret-scrubber-DF88luOe.d.ts → secret-scrubber-Cuy5afaQ.d.ts} +1 -1
- package/dist/security/index.d.ts +20 -4
- package/dist/security/index.js +37 -2
- package/dist/security/index.js.map +1 -1
- package/dist/{selector-BbJqiEP4.d.ts → selector-wT2fv9Fg.d.ts} +1 -1
- package/dist/{session-reader-Drq8RvJu.d.ts → session-reader-CcPi4BQ8.d.ts} +1 -1
- package/dist/{skill-DhfSizKv.d.ts → skill-C_7znCIC.d.ts} +2 -2
- package/dist/storage/index.d.ts +164 -6
- package/dist/storage/index.js +297 -2
- package/dist/storage/index.js.map +1 -1
- package/dist/{renderer-rk_1Swwc.d.ts → system-prompt-Dk1qm8ey.d.ts} +30 -2
- package/dist/{tool-executor-CpuJPYm9.d.ts → tool-executor-DKu4A6nB.d.ts} +5 -5
- package/dist/types/index.d.ts +16 -16
- package/dist/types/index.js +24 -1
- package/dist/types/index.js.map +1 -1
- package/dist/utils/index.d.ts +1 -1
- package/dist/utils/index.js +24 -1
- package/dist/utils/index.js.map +1 -1
- package/package.json +5 -1
- package/dist/plugin-DJk6LL8B.d.ts +0 -434
- package/dist/system-prompt-BC_8ypCG.d.ts +0 -23
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import { U as Usage, a0 as Context, y as ToolProgressEvent, u as Tool } from './context-
|
|
1
|
+
import { U as Usage, a0 as Context, y as ToolProgressEvent, u as Tool } from './context-IovtuTf8.js';
|
|
2
2
|
|
|
3
3
|
/**
|
|
4
4
|
* EventBus — observe-only typed event bus.
|
|
@@ -173,6 +173,78 @@ interface EventMap {
|
|
|
173
173
|
load: number;
|
|
174
174
|
fatal: boolean;
|
|
175
175
|
};
|
|
176
|
+
/**
|
|
177
|
+
* Subagent lifecycle events. Emitted by `MultiAgentHost` so the TUI can
|
|
178
|
+
* surface what's happening in the fleet without needing director-mode
|
|
179
|
+
* (which renders the live FleetPanel). These complement the FleetBus
|
|
180
|
+
* (director-only) by giving the TUI a uniform feed for both `/spawn`
|
|
181
|
+
* and director-orchestrated work.
|
|
182
|
+
*/
|
|
183
|
+
'subagent.spawned': {
|
|
184
|
+
subagentId: string;
|
|
185
|
+
taskId: string;
|
|
186
|
+
name?: string;
|
|
187
|
+
provider?: string;
|
|
188
|
+
model?: string;
|
|
189
|
+
description?: string;
|
|
190
|
+
/**
|
|
191
|
+
* Absolute path to the per-subagent JSONL transcript on disk, when
|
|
192
|
+
* one was created. Undefined when the subagent shares the parent
|
|
193
|
+
* session writer (in-memory or single-file configurations).
|
|
194
|
+
* Surfaced so the TUI (FleetPanel) and `/fleet log` can show the
|
|
195
|
+
* user *where* to look without computing it from the run id.
|
|
196
|
+
*/
|
|
197
|
+
transcriptPath?: string;
|
|
198
|
+
};
|
|
199
|
+
'subagent.task_started': {
|
|
200
|
+
subagentId: string;
|
|
201
|
+
taskId: string;
|
|
202
|
+
description?: string;
|
|
203
|
+
};
|
|
204
|
+
/**
|
|
205
|
+
* Per-tool-call event re-emitted from a subagent's own EventBus
|
|
206
|
+
* onto the host EventBus, so the TUI / non-director surfaces can
|
|
207
|
+
* render "AGENT#1 ● bash 250ms" without having to subscribe to
|
|
208
|
+
* the director-only FleetBus. Fired AFTER the tool completes
|
|
209
|
+
* (paired with `tool.executed`). Includes the subagent id so
|
|
210
|
+
* multiple parallel subagents are distinguishable.
|
|
211
|
+
*/
|
|
212
|
+
'subagent.tool_executed': {
|
|
213
|
+
subagentId: string;
|
|
214
|
+
taskId?: string;
|
|
215
|
+
name: string;
|
|
216
|
+
durationMs: number;
|
|
217
|
+
ok: boolean;
|
|
218
|
+
input?: unknown;
|
|
219
|
+
outputBytes?: number;
|
|
220
|
+
};
|
|
221
|
+
'subagent.task_completed': {
|
|
222
|
+
subagentId: string;
|
|
223
|
+
taskId: string;
|
|
224
|
+
status: 'success' | 'failed' | 'timeout' | 'stopped';
|
|
225
|
+
iterations: number;
|
|
226
|
+
toolCalls: number;
|
|
227
|
+
durationMs: number;
|
|
228
|
+
/**
|
|
229
|
+
* Structured failure envelope when `status !== 'success'`. Carries
|
|
230
|
+
* `kind` (one of `SubagentErrorKind`), `message`, `retryable`, and
|
|
231
|
+
* optional `backoffMs`. UIs branch on `kind` to render the right
|
|
232
|
+
* chip (rate_limit vs auth vs tool_failed). The type is imported
|
|
233
|
+
* lazily as a structural object to avoid a coordination → kernel
|
|
234
|
+
* cycle in the dependency graph.
|
|
235
|
+
*/
|
|
236
|
+
error?: {
|
|
237
|
+
kind: string;
|
|
238
|
+
message: string;
|
|
239
|
+
retryable: boolean;
|
|
240
|
+
backoffMs?: number;
|
|
241
|
+
cause?: {
|
|
242
|
+
name: string;
|
|
243
|
+
message: string;
|
|
244
|
+
stack?: string;
|
|
245
|
+
};
|
|
246
|
+
};
|
|
247
|
+
};
|
|
176
248
|
'mcp.server.connected': {
|
|
177
249
|
name: string;
|
|
178
250
|
toolCount: number;
|
|
@@ -200,19 +272,45 @@ interface EventLogger {
|
|
|
200
272
|
}
|
|
201
273
|
declare class EventBus {
|
|
202
274
|
private readonly listeners;
|
|
275
|
+
private readonly wildcards;
|
|
203
276
|
private logger?;
|
|
204
277
|
setLogger(logger: EventLogger): void;
|
|
205
278
|
on<E extends EventName>(event: E, fn: Listener<E>): () => void;
|
|
206
279
|
off<E extends EventName>(event: E, fn: Listener<E>): void;
|
|
207
280
|
once<E extends EventName>(event: E, fn: Listener<E>): () => void;
|
|
281
|
+
/**
|
|
282
|
+
* Subscribe to all events whose name matches a glob-style prefix.
|
|
283
|
+
* `'tool.*'` matches `tool.started`, `tool.executed`, `tool.progress`, etc.
|
|
284
|
+
* `'*'` matches every event.
|
|
285
|
+
*
|
|
286
|
+
* The handler receives `(eventName, payload)` with the event name as a
|
|
287
|
+
* string and the payload as `unknown`. Use for logging, debugging, or
|
|
288
|
+
* metrics collection across a family of events.
|
|
289
|
+
*
|
|
290
|
+
* Returns an unsubscribe function.
|
|
291
|
+
*/
|
|
292
|
+
onPattern(pattern: string, fn: (event: string, payload: unknown) => void): () => void;
|
|
293
|
+
/**
|
|
294
|
+
* Subscribe to all events whose name matches a RegExp.
|
|
295
|
+
* More flexible than `onPattern` — use when you need regex features
|
|
296
|
+
* (alternation, character classes, capture groups).
|
|
297
|
+
*
|
|
298
|
+
* Returns an unsubscribe function.
|
|
299
|
+
*/
|
|
300
|
+
onRegex(regex: RegExp, fn: (event: string, payload: unknown) => void): () => void;
|
|
208
301
|
emit<E extends EventName>(event: E, payload: EventMap[E]): void;
|
|
209
302
|
clear(): void;
|
|
210
303
|
/**
|
|
211
304
|
* V2-D: introspection helper. Pass an `event` to count handlers for a
|
|
212
305
|
* single key, or omit to get the total across every event. Used by the
|
|
213
306
|
* leak-detection smoke test to flag handler accumulation across runs.
|
|
307
|
+
* Does NOT include wildcard listeners.
|
|
214
308
|
*/
|
|
215
309
|
listenerCount(event?: EventName): number;
|
|
310
|
+
/**
|
|
311
|
+
* Number of wildcard listeners currently registered.
|
|
312
|
+
*/
|
|
313
|
+
wildcardCount(): number;
|
|
216
314
|
}
|
|
217
315
|
|
|
218
316
|
export { EventBus as E, type Listener as L, type EventName as a, type EventLogger as b, type EventMap as c };
|
|
@@ -1,18 +1,18 @@
|
|
|
1
|
-
export { C as CompactorOptions,
|
|
2
|
-
import { g as Provider, a0 as Context } from '../context-
|
|
3
|
-
import { C as Compactor, a as CompactReport } from '../compactor-
|
|
4
|
-
import { M as MessageSelector } from '../selector-
|
|
5
|
-
import { E as EventBus } from '../events-
|
|
6
|
-
import {
|
|
7
|
-
import {
|
|
8
|
-
import { D as DoneCondition } from '../multi-agent-
|
|
9
|
-
import { a as SkillLoader, b as SkillManifest, S as SkillEntry } from '../skill-
|
|
1
|
+
export { C as CompactorOptions, a as DefaultErrorHandler, b as DefaultRetryPolicy, H as HybridCompactor, T as ToolExecutor } from '../tool-executor-DKu4A6nB.js';
|
|
2
|
+
import { g as Provider, a0 as Context } from '../context-IovtuTf8.js';
|
|
3
|
+
import { C as Compactor, a as CompactReport } from '../compactor-BUU6Zm_3.js';
|
|
4
|
+
import { M as MessageSelector } from '../selector-wT2fv9Fg.js';
|
|
5
|
+
import { E as EventBus } from '../events-CNB9PALO.js';
|
|
6
|
+
import { c as MiddlewareHandler } from '../system-prompt-Dk1qm8ey.js';
|
|
7
|
+
import { r as Agent, R as RunResult } from '../index-BDb0cAMP.js';
|
|
8
|
+
import { D as DoneCondition } from '../multi-agent-B9a6sflH.js';
|
|
9
|
+
import { a as SkillLoader, b as SkillManifest, S as SkillEntry } from '../skill-C_7znCIC.js';
|
|
10
10
|
import { a as WstackPaths } from '../wstack-paths-BGu2INTm.js';
|
|
11
11
|
import '../models-registry-Y2xbog0E.js';
|
|
12
|
-
import '../observability-BhnVLBLS.js';
|
|
13
|
-
import '../secret-scrubber-CicHLN4G.js';
|
|
14
|
-
import '../config-BU9f_5yH.js';
|
|
15
12
|
import '../logger-BMQgxvdy.js';
|
|
13
|
+
import '../observability-BhnVLBLS.js';
|
|
14
|
+
import '../secret-scrubber-CgG2tV2B.js';
|
|
15
|
+
import '../config-CKLYPkCi.js';
|
|
16
16
|
|
|
17
17
|
interface SkillLoaderOptions {
|
|
18
18
|
paths: WstackPaths;
|
|
@@ -0,0 +1,9 @@
|
|
|
1
|
+
export { A as AfterIterationHook, p as AfterRunHook, q as AfterToolExecutionHook, s as AgentExtension, B as BeforeIterationHook, w as BeforeRunHook, x as BeforeToolExecutionHook, E as ExtensionRegistry, O as OnErrorHook, G as ProviderRunnerFn, z as ProviderRunnerWrapper } from '../index-BDb0cAMP.js';
|
|
2
|
+
import '../context-IovtuTf8.js';
|
|
3
|
+
import '../logger-BMQgxvdy.js';
|
|
4
|
+
import '../system-prompt-Dk1qm8ey.js';
|
|
5
|
+
import '../observability-BhnVLBLS.js';
|
|
6
|
+
import '../events-CNB9PALO.js';
|
|
7
|
+
import '../secret-scrubber-CgG2tV2B.js';
|
|
8
|
+
import '../config-CKLYPkCi.js';
|
|
9
|
+
import '../models-registry-Y2xbog0E.js';
|
|
@@ -0,0 +1,234 @@
|
|
|
1
|
+
// src/types/errors.ts
|
|
2
|
+
var WrongStackError = class extends Error {
|
|
3
|
+
code;
|
|
4
|
+
subsystem;
|
|
5
|
+
severity;
|
|
6
|
+
recoverable;
|
|
7
|
+
context;
|
|
8
|
+
constructor(opts) {
|
|
9
|
+
super(opts.message, { cause: opts.cause });
|
|
10
|
+
this.name = "WrongStackError";
|
|
11
|
+
this.code = opts.code;
|
|
12
|
+
this.subsystem = opts.subsystem;
|
|
13
|
+
this.severity = opts.severity ?? "error";
|
|
14
|
+
this.recoverable = opts.recoverable ?? false;
|
|
15
|
+
this.context = opts.context;
|
|
16
|
+
}
|
|
17
|
+
/**
|
|
18
|
+
* Render a one-line user-facing description.
|
|
19
|
+
* Subclasses should override for domain-specific formatting.
|
|
20
|
+
*/
|
|
21
|
+
describe() {
|
|
22
|
+
const ctx = this.context ? ` ${formatContext(this.context)}` : "";
|
|
23
|
+
return `${this.code}: ${this.message}${ctx}`;
|
|
24
|
+
}
|
|
25
|
+
};
|
|
26
|
+
function formatContext(ctx) {
|
|
27
|
+
const parts = Object.entries(ctx).filter(([, v]) => v !== void 0).slice(0, 3).map(([k, v]) => `${k}=${String(v)}`);
|
|
28
|
+
return parts.length > 0 ? `[${parts.join(" ")}]` : "";
|
|
29
|
+
}
|
|
30
|
+
|
|
31
|
+
// src/extension/registry.ts
|
|
32
|
+
var ExtensionRegistry = class {
|
|
33
|
+
extensions = [];
|
|
34
|
+
promptContributors = [];
|
|
35
|
+
log;
|
|
36
|
+
setLogger(log) {
|
|
37
|
+
this.log = log;
|
|
38
|
+
}
|
|
39
|
+
/**
|
|
40
|
+
* Register a system prompt contributor. Returns an unregister function.
|
|
41
|
+
* Contributors are called on every system prompt build in registration
|
|
42
|
+
* order. Their output blocks are inserted after the core environment
|
|
43
|
+
* block, before the mode and plan blocks.
|
|
44
|
+
*/
|
|
45
|
+
registerSystemPromptContributor(c) {
|
|
46
|
+
this.promptContributors.push(c);
|
|
47
|
+
return () => {
|
|
48
|
+
const idx = this.promptContributors.indexOf(c);
|
|
49
|
+
if (idx >= 0) this.promptContributors.splice(idx, 1);
|
|
50
|
+
};
|
|
51
|
+
}
|
|
52
|
+
/**
|
|
53
|
+
* Build all registered system prompt contributions.
|
|
54
|
+
* Failures are caught and logged — one bad contributor doesn't
|
|
55
|
+
* break the prompt assembly.
|
|
56
|
+
*/
|
|
57
|
+
async buildSystemPromptContributions(ctx) {
|
|
58
|
+
const blocks = [];
|
|
59
|
+
for (const c of this.promptContributors) {
|
|
60
|
+
try {
|
|
61
|
+
const contributed = await c(ctx);
|
|
62
|
+
blocks.push(...contributed);
|
|
63
|
+
} catch (err) {
|
|
64
|
+
this.log?.error("SystemPromptContributor failed", err);
|
|
65
|
+
}
|
|
66
|
+
}
|
|
67
|
+
return blocks;
|
|
68
|
+
}
|
|
69
|
+
/**
|
|
70
|
+
* Returns the live array of contributors (readonly snapshot for
|
|
71
|
+
* passing to DefaultSystemPromptBuilder at build time).
|
|
72
|
+
*/
|
|
73
|
+
listSystemPromptContributors() {
|
|
74
|
+
return this.promptContributors;
|
|
75
|
+
}
|
|
76
|
+
/**
|
|
77
|
+
* Register an extension. Duplicate names are rejected.
|
|
78
|
+
* Returns an unregister function.
|
|
79
|
+
*/
|
|
80
|
+
register(ext) {
|
|
81
|
+
if (this.extensions.some((e) => e.name === ext.name)) {
|
|
82
|
+
throw new WrongStackError({
|
|
83
|
+
message: `Extension "${ext.name}" already registered`,
|
|
84
|
+
code: "REGISTRY_DUPLICATE",
|
|
85
|
+
subsystem: "container",
|
|
86
|
+
context: { extension: ext.name }
|
|
87
|
+
});
|
|
88
|
+
}
|
|
89
|
+
this.extensions.push(ext);
|
|
90
|
+
return () => this.unregister(ext.name);
|
|
91
|
+
}
|
|
92
|
+
/**
|
|
93
|
+
* Register an extension, silently replacing any previous registration
|
|
94
|
+
* with the same name. Use this when overriding a default extension.
|
|
95
|
+
*/
|
|
96
|
+
registerOrReplace(ext) {
|
|
97
|
+
const idx = this.extensions.findIndex((e) => e.name === ext.name);
|
|
98
|
+
if (idx >= 0) this.extensions.splice(idx, 1);
|
|
99
|
+
return this.register(ext);
|
|
100
|
+
}
|
|
101
|
+
/**
|
|
102
|
+
* Unregister an extension by name. Returns true if found.
|
|
103
|
+
*/
|
|
104
|
+
unregister(name) {
|
|
105
|
+
const idx = this.extensions.findIndex((e) => e.name === name);
|
|
106
|
+
if (idx === -1) return false;
|
|
107
|
+
this.extensions.splice(idx, 1);
|
|
108
|
+
return true;
|
|
109
|
+
}
|
|
110
|
+
/**
|
|
111
|
+
* List registered extension names in order.
|
|
112
|
+
*/
|
|
113
|
+
list() {
|
|
114
|
+
return this.extensions.map((e) => e.name);
|
|
115
|
+
}
|
|
116
|
+
/**
|
|
117
|
+
* Check if an extension with the given name is registered.
|
|
118
|
+
*/
|
|
119
|
+
has(name) {
|
|
120
|
+
return this.extensions.some((e) => e.name === name);
|
|
121
|
+
}
|
|
122
|
+
/**
|
|
123
|
+
* Remove all registered extensions and contributors.
|
|
124
|
+
*/
|
|
125
|
+
clear() {
|
|
126
|
+
this.extensions.length = 0;
|
|
127
|
+
this.promptContributors.length = 0;
|
|
128
|
+
}
|
|
129
|
+
// ── Hook runners ─────────────────────────────────────────────────
|
|
130
|
+
async runBeforeRun(...args) {
|
|
131
|
+
for (const ext of this.extensions) {
|
|
132
|
+
if (!ext.beforeRun) continue;
|
|
133
|
+
try {
|
|
134
|
+
await ext.beforeRun(...args);
|
|
135
|
+
} catch (err) {
|
|
136
|
+
this.log?.error(`Extension "${ext.name}" beforeRun hook failed`, err);
|
|
137
|
+
}
|
|
138
|
+
}
|
|
139
|
+
}
|
|
140
|
+
async runAfterRun(...args) {
|
|
141
|
+
for (const ext of this.extensions) {
|
|
142
|
+
if (!ext.afterRun) continue;
|
|
143
|
+
try {
|
|
144
|
+
await ext.afterRun(...args);
|
|
145
|
+
} catch (err) {
|
|
146
|
+
this.log?.error(`Extension "${ext.name}" afterRun hook failed`, err);
|
|
147
|
+
}
|
|
148
|
+
}
|
|
149
|
+
}
|
|
150
|
+
async runBeforeIteration(...args) {
|
|
151
|
+
for (const ext of this.extensions) {
|
|
152
|
+
if (!ext.beforeIteration) continue;
|
|
153
|
+
try {
|
|
154
|
+
await ext.beforeIteration(...args);
|
|
155
|
+
} catch (err) {
|
|
156
|
+
this.log?.error(`Extension "${ext.name}" beforeIteration hook failed`, err);
|
|
157
|
+
}
|
|
158
|
+
}
|
|
159
|
+
}
|
|
160
|
+
async runAfterIteration(...args) {
|
|
161
|
+
for (const ext of this.extensions) {
|
|
162
|
+
if (!ext.afterIteration) continue;
|
|
163
|
+
try {
|
|
164
|
+
await ext.afterIteration(...args);
|
|
165
|
+
} catch (err) {
|
|
166
|
+
this.log?.error(`Extension "${ext.name}" afterIteration hook failed`, err);
|
|
167
|
+
}
|
|
168
|
+
}
|
|
169
|
+
}
|
|
170
|
+
/**
|
|
171
|
+
* Run onError hooks in order. The first hook that returns a non-void
|
|
172
|
+
* result wins; subsequent hooks are skipped.
|
|
173
|
+
*/
|
|
174
|
+
async runOnError(...args) {
|
|
175
|
+
for (const ext of this.extensions) {
|
|
176
|
+
if (!ext.onError) continue;
|
|
177
|
+
try {
|
|
178
|
+
const result = await ext.onError(...args);
|
|
179
|
+
if (result) return result;
|
|
180
|
+
} catch (err) {
|
|
181
|
+
this.log?.error(`Extension "${ext.name}" onError hook failed`, err);
|
|
182
|
+
}
|
|
183
|
+
}
|
|
184
|
+
}
|
|
185
|
+
/**
|
|
186
|
+
* Build a composed provider runner. Extensions with `wrapProviderRunner`
|
|
187
|
+
* form a middleware-style chain: the innermost extension wraps the
|
|
188
|
+
* default runner, each subsequent wrapper wraps the previous.
|
|
189
|
+
*/
|
|
190
|
+
wrapProviderRunner(inner) {
|
|
191
|
+
const wrappers = this.extensions.filter((e) => e.wrapProviderRunner).map((e) => ({ name: e.name, wrap: e.wrapProviderRunner }));
|
|
192
|
+
if (wrappers.length === 0) return inner;
|
|
193
|
+
let composed = inner;
|
|
194
|
+
for (let i = wrappers.length - 1; i >= 0; i--) {
|
|
195
|
+
const wrapper = wrappers[i];
|
|
196
|
+
const next = composed;
|
|
197
|
+
composed = async (ctx, req) => {
|
|
198
|
+
try {
|
|
199
|
+
return await wrapper.wrap(ctx, req, next);
|
|
200
|
+
} catch (err) {
|
|
201
|
+
this.log?.error(`Extension "${wrapper.name}" wrapProviderRunner failed`, err);
|
|
202
|
+
throw err;
|
|
203
|
+
}
|
|
204
|
+
};
|
|
205
|
+
}
|
|
206
|
+
return composed;
|
|
207
|
+
}
|
|
208
|
+
async runBeforeToolExecution(...args) {
|
|
209
|
+
let toolUses = args[1];
|
|
210
|
+
for (const ext of this.extensions) {
|
|
211
|
+
if (!ext.beforeToolExecution) continue;
|
|
212
|
+
try {
|
|
213
|
+
toolUses = await ext.beforeToolExecution(args[0], toolUses);
|
|
214
|
+
} catch (err) {
|
|
215
|
+
this.log?.error(`Extension "${ext.name}" beforeToolExecution hook failed`, err);
|
|
216
|
+
}
|
|
217
|
+
}
|
|
218
|
+
return toolUses;
|
|
219
|
+
}
|
|
220
|
+
async runAfterToolExecution(...args) {
|
|
221
|
+
for (const ext of this.extensions) {
|
|
222
|
+
if (!ext.afterToolExecution) continue;
|
|
223
|
+
try {
|
|
224
|
+
await ext.afterToolExecution(...args);
|
|
225
|
+
} catch (err) {
|
|
226
|
+
this.log?.error(`Extension "${ext.name}" afterToolExecution hook failed`, err);
|
|
227
|
+
}
|
|
228
|
+
}
|
|
229
|
+
}
|
|
230
|
+
};
|
|
231
|
+
|
|
232
|
+
export { ExtensionRegistry };
|
|
233
|
+
//# sourceMappingURL=index.js.map
|
|
234
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["../../src/types/errors.ts","../../src/extension/registry.ts"],"names":[],"mappings":";AAiEO,IAAM,eAAA,GAAN,cAA8B,KAAA,CAAM;AAAA,EAChC,IAAA;AAAA,EACA,SAAA;AAAA,EACA,QAAA;AAAA,EACA,WAAA;AAAA,EACA,OAAA;AAAA,EAET,YAAY,IAAA,EAQT;AACD,IAAA,KAAA,CAAM,KAAK,OAAA,EAAS,EAAE,KAAA,EAAO,IAAA,CAAK,OAAO,CAAA;AACzC,IAAA,IAAA,CAAK,IAAA,GAAO,iBAAA;AACZ,IAAA,IAAA,CAAK,OAAO,IAAA,CAAK,IAAA;AACjB,IAAA,IAAA,CAAK,YAAY,IAAA,CAAK,SAAA;AACtB,IAAA,IAAA,CAAK,QAAA,GAAW,KAAK,QAAA,IAAY,OAAA;AACjC,IAAA,IAAA,CAAK,WAAA,GAAc,KAAK,WAAA,IAAe,KAAA;AACvC,IAAA,IAAA,CAAK,UAAU,IAAA,CAAK,OAAA;AAAA,EACtB;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,QAAA,GAAmB;AACjB,IAAA,MAAM,GAAA,GAAM,KAAK,OAAA,GAAU,CAAA,CAAA,EAAI,cAAc,IAAA,CAAK,OAAO,CAAC,CAAA,CAAA,GAAK,EAAA;AAC/D,IAAA,OAAO,GAAG,IAAA,CAAK,IAAI,KAAK,IAAA,CAAK,OAAO,GAAG,GAAG,CAAA,CAAA;AAAA,EAC5C;AACF,CAAA;AAEA,SAAS,cAAc,GAAA,EAAsC;AAC3D,EAAA,MAAM,KAAA,GAAQ,MAAA,CAAO,OAAA,CAAQ,GAAG,CAAA,CAC7B,MAAA,CAAO,CAAC,GAAG,CAAC,CAAA,KAAM,CAAA,KAAM,MAAS,CAAA,CACjC,KAAA,CAAM,CAAA,EAAG,CAAC,CAAA,CACV,GAAA,CAAI,CAAC,CAAC,CAAA,EAAG,CAAC,CAAA,KAAM,CAAA,EAAG,CAAC,CAAA,CAAA,EAAI,MAAA,CAAO,CAAC,CAAC,CAAA,CAAE,CAAA;AACtC,EAAA,OAAO,KAAA,CAAM,SAAS,CAAA,GAAI,CAAA,CAAA,EAAI,MAAM,IAAA,CAAK,GAAG,CAAC,CAAA,CAAA,CAAA,GAAM,EAAA;AACrD;;;ACjFO,IAAM,oBAAN,MAAwB;AAAA,EACZ,aAA+B,EAAC;AAAA,EAChC,qBAAgD,EAAC;AAAA,EAC1D,GAAA;AAAA,EAER,UAAU,GAAA,EAAmB;AAC3B,IAAA,IAAA,CAAK,GAAA,GAAM,GAAA;AAAA,EACb;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,gCAAgC,CAAA,EAAwC;AACtE,IAAA,IAAA,CAAK,kBAAA,CAAmB,KAAK,CAAC,CAAA;AAC9B,IAAA,OAAO,MAAM;AACX,MAAA,MAAM,GAAA,GAAM,IAAA,CAAK,kBAAA,CAAmB,OAAA,CAAQ,CAAC,CAAA;AAC7C,MAAA,IAAI,OAAO,CAAA,EAAG,IAAA,CAAK,kBAAA,CAAmB,MAAA,CAAO,KAAK,CAAC,CAAA;AAAA,IACrD,CAAA;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,MAAM,+BACJ,GAAA,EACsB;AACtB,IAAA,MAAM,SAAsB,EAAC;AAC7B,IAAA,KAAA,MAAW,CAAA,IAAK,KAAK,kBAAA,EAAoB;AACvC,MAAA,IAAI;AACF,QAAA,MAAM,WAAA,GAAc,MAAM,CAAA,CAAE,GAAG,CAAA;AAC/B,QAAA,MAAA,CAAO,IAAA,CAAK,GAAG,WAAW,CAAA;AAAA,MAC5B,SAAS,GAAA,EAAK;AACZ,QAAA,IAAA,CAAK,GAAA,EAAK,KAAA,CAAM,gCAAA,EAAkC,GAAG,CAAA;AAAA,MACvD;AAAA,IACF;AACA,IAAA,OAAO,MAAA;AAAA,EACT;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,4BAAA,GAAmE;AACjE,IAAA,OAAO,IAAA,CAAK,kBAAA;AAAA,EACd;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,SAAS,GAAA,EAAiC;AACxC,IAAA,IAAI,IAAA,CAAK,WAAW,IAAA,CAAK,CAAC,MAAM,CAAA,CAAE,IAAA,KAAS,GAAA,CAAI,IAAI,CAAA,EAAG;AACpD,MAAA,MAAM,IAAI,eAAA,CAAgB;AAAA,QACxB,OAAA,EAAS,CAAA,WAAA,EAAc,GAAA,CAAI,IAAI,CAAA,oBAAA,CAAA;AAAA,QAC/B,IAAA,EAAM,oBAAA;AAAA,QACN,SAAA,EAAW,WAAA;AAAA,QACX,OAAA,EAAS,EAAE,SAAA,EAAW,GAAA,CAAI,IAAA;AAAK,OAChC,CAAA;AAAA,IACH;AACA,IAAA,IAAA,CAAK,UAAA,CAAW,KAAK,GAAG,CAAA;AACxB,IAAA,OAAO,MAAM,IAAA,CAAK,UAAA,CAAW,GAAA,CAAI,IAAI,CAAA;AAAA,EACvC;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,kBAAkB,GAAA,EAAiC;AACjD,IAAA,MAAM,GAAA,GAAM,KAAK,UAAA,CAAW,SAAA,CAAU,CAAC,CAAA,KAAM,CAAA,CAAE,IAAA,KAAS,GAAA,CAAI,IAAI,CAAA;AAChE,IAAA,IAAI,OAAO,CAAA,EAAG,IAAA,CAAK,UAAA,CAAW,MAAA,CAAO,KAAK,CAAC,CAAA;AAC3C,IAAA,OAAO,IAAA,CAAK,SAAS,GAAG,CAAA;AAAA,EAC1B;AAAA;AAAA;AAAA;AAAA,EAKA,WAAW,IAAA,EAAuB;AAChC,IAAA,MAAM,GAAA,GAAM,KAAK,UAAA,CAAW,SAAA,CAAU,CAAC,CAAA,KAAM,CAAA,CAAE,SAAS,IAAI,CAAA;AAC5D,IAAA,IAAI,GAAA,KAAQ,IAAI,OAAO,KAAA;AACvB,IAAA,IAAA,CAAK,UAAA,CAAW,MAAA,CAAO,GAAA,EAAK,CAAC,CAAA;AAC7B,IAAA,OAAO,IAAA;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKA,IAAA,GAA0B;AACxB,IAAA,OAAO,KAAK,UAAA,CAAW,GAAA,CAAI,CAAC,CAAA,KAAM,EAAE,IAAI,CAAA;AAAA,EAC1C;AAAA;AAAA;AAAA;AAAA,EAKA,IAAI,IAAA,EAAuB;AACzB,IAAA,OAAO,KAAK,UAAA,CAAW,IAAA,CAAK,CAAC,CAAA,KAAM,CAAA,CAAE,SAAS,IAAI,CAAA;AAAA,EACpD;AAAA;AAAA;AAAA;AAAA,EAKA,KAAA,GAAc;AACZ,IAAA,IAAA,CAAK,WAAW,MAAA,GAAS,CAAA;AACzB,IAAA,IAAA,CAAK,mBAAmB,MAAA,GAAS,CAAA;AAAA,EACnC;AAAA;AAAA,EAIA,MAAM,gBAAgB,IAAA,EAAgD;AACpE,IAAA,KAAA,MAAW,GAAA,IAAO,KAAK,UAAA,EAAY;AACjC,MAAA,IAAI,CAAC,IAAI,SAAA,EAAW;AACpB,MAAA,IAAI;AACF,QAAA,MAAM,GAAA,CAAI,SAAA,CAAU,GAAG,IAAI,CAAA;AAAA,MAC7B,SAAS,GAAA,EAAK;AACZ,QAAA,IAAA,CAAK,KAAK,KAAA,CAAM,CAAA,WAAA,EAAc,GAAA,CAAI,IAAI,2BAA2B,GAAG,CAAA;AAAA,MACtE;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAM,eAAe,IAAA,EAA+C;AAClE,IAAA,KAAA,MAAW,GAAA,IAAO,KAAK,UAAA,EAAY;AACjC,MAAA,IAAI,CAAC,IAAI,QAAA,EAAU;AACnB,MAAA,IAAI;AACF,QAAA,MAAM,GAAA,CAAI,QAAA,CAAS,GAAG,IAAI,CAAA;AAAA,MAC5B,SAAS,GAAA,EAAK;AACZ,QAAA,IAAA,CAAK,KAAK,KAAA,CAAM,CAAA,WAAA,EAAc,GAAA,CAAI,IAAI,0BAA0B,GAAG,CAAA;AAAA,MACrE;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAM,sBAAsB,IAAA,EAAsD;AAChF,IAAA,KAAA,MAAW,GAAA,IAAO,KAAK,UAAA,EAAY;AACjC,MAAA,IAAI,CAAC,IAAI,eAAA,EAAiB;AAC1B,MAAA,IAAI;AACF,QAAA,MAAM,GAAA,CAAI,eAAA,CAAgB,GAAG,IAAI,CAAA;AAAA,MACnC,SAAS,GAAA,EAAK;AACZ,QAAA,IAAA,CAAK,KAAK,KAAA,CAAM,CAAA,WAAA,EAAc,GAAA,CAAI,IAAI,iCAAiC,GAAG,CAAA;AAAA,MAC5E;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAM,qBAAqB,IAAA,EAAqD;AAC9E,IAAA,KAAA,MAAW,GAAA,IAAO,KAAK,UAAA,EAAY;AACjC,MAAA,IAAI,CAAC,IAAI,cAAA,EAAgB;AACzB,MAAA,IAAI;AACF,QAAA,MAAM,GAAA,CAAI,cAAA,CAAe,GAAG,IAAI,CAAA;AAAA,MAClC,SAAS,GAAA,EAAK;AACZ,QAAA,IAAA,CAAK,KAAK,KAAA,CAAM,CAAA,WAAA,EAAc,GAAA,CAAI,IAAI,gCAAgC,GAAG,CAAA;AAAA,MAC3E;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAM,cACD,IAAA,EAGH;AACA,IAAA,KAAA,MAAW,GAAA,IAAO,KAAK,UAAA,EAAY;AACjC,MAAA,IAAI,CAAC,IAAI,OAAA,EAAS;AAClB,MAAA,IAAI;AACF,QAAA,MAAM,MAAA,GAAS,MAAM,GAAA,CAAI,OAAA,CAAQ,GAAG,IAAI,CAAA;AACxC,QAAA,IAAI,QAAQ,OAAO,MAAA;AAAA,MACrB,SAAS,GAAA,EAAK;AACZ,QAAA,IAAA,CAAK,KAAK,KAAA,CAAM,CAAA,WAAA,EAAc,GAAA,CAAI,IAAI,yBAAyB,GAAG,CAAA;AAAA,MACpE;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,mBAAmB,KAAA,EAA2C;AAC5D,IAAA,MAAM,WAAW,IAAA,CAAK,UAAA,CACnB,OAAO,CAAC,CAAA,KAAM,EAAE,kBAAkB,CAAA,CAClC,IAAI,CAAC,CAAA,MAAO,EAAE,IAAA,EAAM,CAAA,CAAE,MAAM,IAAA,EAAM,CAAA,CAAE,oBAAoB,CAAE,CAAA;AAE7D,IAAA,IAAI,QAAA,CAAS,MAAA,KAAW,CAAA,EAAG,OAAO,KAAA;AAGlC,IAAA,IAAI,QAAA,GAA6B,KAAA;AACjC,IAAA,KAAA,IAAS,IAAI,QAAA,CAAS,MAAA,GAAS,CAAA,EAAG,CAAA,IAAK,GAAG,CAAA,EAAA,EAAK;AAC7C,MAAA,MAAM,OAAA,GAAU,SAAS,CAAC,CAAA;AAC1B,MAAA,MAAM,IAAA,GAAO,QAAA;AACb,MAAA,QAAA,GAAW,OAAO,KAAK,GAAA,KAAQ;AAC7B,QAAA,IAAI;AACF,UAAA,OAAO,MAAM,OAAA,CAAQ,IAAA,CAAK,GAAA,EAAK,KAAK,IAAI,CAAA;AAAA,QAC1C,SAAS,GAAA,EAAK;AACZ,UAAA,IAAA,CAAK,KAAK,KAAA,CAAM,CAAA,WAAA,EAAc,OAAA,CAAQ,IAAI,+BAA+B,GAAG,CAAA;AAC5E,UAAA,MAAM,GAAA;AAAA,QACR;AAAA,MACF,CAAA;AAAA,IACF;AACA,IAAA,OAAO,QAAA;AAAA,EACT;AAAA,EAEA,MAAM,0BACD,IAAA,EAC8C;AACjD,IAAA,IAAI,QAAA,GAAW,KAAK,CAAC,CAAA;AACrB,IAAA,KAAA,MAAW,GAAA,IAAO,KAAK,UAAA,EAAY;AACjC,MAAA,IAAI,CAAC,IAAI,mBAAA,EAAqB;AAC9B,MAAA,IAAI;AACF,QAAA,QAAA,GAAW,MAAM,GAAA,CAAI,mBAAA,CAAoB,IAAA,CAAK,CAAC,GAAG,QAAQ,CAAA;AAAA,MAC5D,SAAS,GAAA,EAAK;AACZ,QAAA,IAAA,CAAK,KAAK,KAAA,CAAM,CAAA,WAAA,EAAc,GAAA,CAAI,IAAI,qCAAqC,GAAG,CAAA;AAAA,MAChF;AAAA,IACF;AACA,IAAA,OAAO,QAAA;AAAA,EACT;AAAA,EAEA,MAAM,yBAAyB,IAAA,EAAyD;AACtF,IAAA,KAAA,MAAW,GAAA,IAAO,KAAK,UAAA,EAAY;AACjC,MAAA,IAAI,CAAC,IAAI,kBAAA,EAAoB;AAC7B,MAAA,IAAI;AACF,QAAA,MAAM,GAAA,CAAI,kBAAA,CAAmB,GAAG,IAAI,CAAA;AAAA,MACtC,SAAS,GAAA,EAAK;AACZ,QAAA,IAAA,CAAK,KAAK,KAAA,CAAM,CAAA,WAAA,EAAc,GAAA,CAAI,IAAI,oCAAoC,GAAG,CAAA;AAAA,MAC/E;AAAA,IACF;AAAA,EACF;AACF","file":"index.js","sourcesContent":["/**\n * WrongStack error hierarchy.\n *\n * Every error thrown by the framework is a `WrongStackError` with a\n * machine-readable `code`, a `subsystem` tag, and a `severity` level.\n * This lets consumers (CLI, TUI, plugins, tests) branch on structured\n * data instead of parsing error messages.\n */\n\n// ── Error codes ──────────────────────────────────────────────────────\n\nexport type ErrorCode =\n // Provider\n | 'PROVIDER_RATE_LIMITED'\n | 'PROVIDER_AUTH_FAILED'\n | 'PROVIDER_OVERLOADED'\n | 'PROVIDER_INVALID_REQUEST'\n | 'PROVIDER_SERVER_ERROR'\n | 'PROVIDER_NETWORK_ERROR'\n | 'PROVIDER_CONTEXT_OVERFLOW'\n // Tool\n | 'TOOL_NOT_FOUND'\n | 'TOOL_PERMISSION_DENIED'\n | 'TOOL_EXECUTION_FAILED'\n | 'TOOL_TIMEOUT'\n | 'TOOL_INPUT_INVALID'\n // Config\n | 'CONFIG_INVALID'\n | 'CONFIG_NOT_FOUND'\n | 'CONFIG_PARSE_FAILED'\n | 'CONFIG_MIGRATION_NEEDED'\n // Plugin\n | 'PLUGIN_LOAD_FAILED'\n | 'PLUGIN_API_MISMATCH'\n | 'PLUGIN_MISSING_DEPENDENCY'\n // Agent\n | 'AGENT_ITERATION_LIMIT'\n | 'AGENT_CONTEXT_OVERFLOW'\n | 'AGENT_ABORTED'\n | 'AGENT_RUN_FAILED'\n // Session\n | 'SESSION_NOT_FOUND'\n | 'SESSION_CORRUPTED'\n | 'SESSION_WRITE_FAILED'\n // Container / Registry\n | 'CONTAINER_TOKEN_ALREADY_BOUND'\n | 'CONTAINER_TOKEN_NOT_BOUND'\n | 'REGISTRY_DUPLICATE'\n | 'REGISTRY_NOT_FOUND'\n // General\n | 'UNKNOWN';\n\nexport type ErrorSubsystem =\n | 'provider'\n | 'tool'\n | 'config'\n | 'plugin'\n | 'agent'\n | 'session'\n | 'container'\n | 'general';\nexport type ErrorSeverity = 'fatal' | 'error' | 'warning';\n\n// ── Base error class ─────────────────────────────────────────────────\n\nexport class WrongStackError extends Error {\n readonly code: ErrorCode;\n readonly subsystem: ErrorSubsystem;\n readonly severity: ErrorSeverity;\n readonly recoverable: boolean;\n readonly context?: Record<string, unknown>;\n\n constructor(opts: {\n message: string;\n code: ErrorCode;\n subsystem: ErrorSubsystem;\n severity?: ErrorSeverity;\n recoverable?: boolean;\n context?: Record<string, unknown>;\n cause?: unknown;\n }) {\n super(opts.message, { cause: opts.cause });\n this.name = 'WrongStackError';\n this.code = opts.code;\n this.subsystem = opts.subsystem;\n this.severity = opts.severity ?? 'error';\n this.recoverable = opts.recoverable ?? false;\n this.context = opts.context;\n }\n\n /**\n * Render a one-line user-facing description.\n * Subclasses should override for domain-specific formatting.\n */\n describe(): string {\n const ctx = this.context ? ` ${formatContext(this.context)}` : '';\n return `${this.code}: ${this.message}${ctx}`;\n }\n}\n\nfunction formatContext(ctx: Record<string, unknown>): string {\n const parts = Object.entries(ctx)\n .filter(([, v]) => v !== undefined)\n .slice(0, 3)\n .map(([k, v]) => `${k}=${String(v)}`);\n return parts.length > 0 ? `[${parts.join(' ')}]` : '';\n}\n\n// ── Specific error classes ───────────────────────────────────────────\n\n/**\n * Tool execution errors — thrown by ToolExecutor and individual tools.\n */\nexport class ToolError extends WrongStackError {\n readonly toolName: string;\n\n constructor(opts: {\n message: string;\n code: Extract<\n ErrorCode,\n | 'TOOL_NOT_FOUND'\n | 'TOOL_PERMISSION_DENIED'\n | 'TOOL_EXECUTION_FAILED'\n | 'TOOL_TIMEOUT'\n | 'TOOL_INPUT_INVALID'\n >;\n toolName: string;\n recoverable?: boolean;\n context?: Record<string, unknown>;\n cause?: unknown;\n }) {\n super({\n message: opts.message,\n code: opts.code,\n subsystem: 'tool',\n recoverable: opts.recoverable,\n context: { tool: opts.toolName, ...opts.context },\n cause: opts.cause,\n });\n this.name = 'ToolError';\n this.toolName = opts.toolName;\n }\n}\n\n/**\n * Config loading / validation errors.\n */\nexport class ConfigError extends WrongStackError {\n constructor(opts: {\n message: string;\n code: Extract<\n ErrorCode,\n 'CONFIG_INVALID' | 'CONFIG_NOT_FOUND' | 'CONFIG_PARSE_FAILED' | 'CONFIG_MIGRATION_NEEDED'\n >;\n context?: Record<string, unknown>;\n cause?: unknown;\n }) {\n super({\n message: opts.message,\n code: opts.code,\n subsystem: 'config',\n severity: 'fatal',\n recoverable: false,\n context: opts.context,\n cause: opts.cause,\n });\n this.name = 'ConfigError';\n }\n}\n\n/**\n * Plugin loading / lifecycle errors.\n */\nexport class PluginError extends WrongStackError {\n readonly pluginName: string;\n\n constructor(opts: {\n message: string;\n code: Extract<\n ErrorCode,\n 'PLUGIN_LOAD_FAILED' | 'PLUGIN_API_MISMATCH' | 'PLUGIN_MISSING_DEPENDENCY'\n >;\n pluginName: string;\n context?: Record<string, unknown>;\n cause?: unknown;\n }) {\n super({\n message: opts.message,\n code: opts.code,\n subsystem: 'plugin',\n severity: 'error',\n recoverable: opts.code === 'PLUGIN_MISSING_DEPENDENCY',\n context: { plugin: opts.pluginName, ...opts.context },\n cause: opts.cause,\n });\n this.name = 'PluginError';\n this.pluginName = opts.pluginName;\n }\n}\n\n/**\n * Agent runtime errors — thrown by Agent.run when a non-WrongStackError\n * escapes the inner loop, so callers always see a structured error.\n */\nexport class AgentError extends WrongStackError {\n constructor(opts: {\n message: string;\n code: Extract<\n ErrorCode,\n 'AGENT_ITERATION_LIMIT' | 'AGENT_CONTEXT_OVERFLOW' | 'AGENT_ABORTED' | 'AGENT_RUN_FAILED'\n >;\n recoverable?: boolean;\n context?: Record<string, unknown>;\n cause?: unknown;\n }) {\n super({\n message: opts.message,\n code: opts.code,\n subsystem: 'agent',\n severity: opts.code === 'AGENT_ABORTED' ? 'warning' : 'error',\n recoverable: opts.recoverable ?? opts.code === 'AGENT_ITERATION_LIMIT',\n context: opts.context,\n cause: opts.cause,\n });\n this.name = 'AgentError';\n }\n}\n\n/**\n * Wrap an arbitrary thrown value into a `WrongStackError` so the caller\n * always gets a structured error. Pass-throughs WrongStackError instances\n * unchanged; raw `Error`s and primitives get an `AGENT_RUN_FAILED` wrapper\n * with the original preserved as `cause`.\n */\nexport function toWrongStackError(\n err: unknown,\n code: Extract<ErrorCode, 'AGENT_RUN_FAILED' | 'AGENT_ABORTED' | 'UNKNOWN'> = 'AGENT_RUN_FAILED',\n): WrongStackError {\n if (err instanceof WrongStackError) return err;\n const message = err instanceof Error ? err.message : String(err);\n return new AgentError({\n message,\n code: code === 'UNKNOWN' ? 'AGENT_RUN_FAILED' : code,\n cause: err,\n });\n}\n\n/**\n * Session storage errors.\n */\nexport class SessionError extends WrongStackError {\n readonly sessionId?: string;\n\n constructor(opts: {\n message: string;\n code: Extract<ErrorCode, 'SESSION_NOT_FOUND' | 'SESSION_CORRUPTED' | 'SESSION_WRITE_FAILED'>;\n sessionId?: string;\n context?: Record<string, unknown>;\n cause?: unknown;\n }) {\n super({\n message: opts.message,\n code: opts.code,\n subsystem: 'session',\n severity: opts.code === 'SESSION_WRITE_FAILED' ? 'error' : 'warning',\n recoverable: opts.code !== 'SESSION_CORRUPTED',\n context: { sessionId: opts.sessionId, ...opts.context },\n cause: opts.cause,\n });\n this.name = 'SessionError';\n this.sessionId = opts.sessionId;\n }\n}\n\n// ── Type guards ──────────────────────────────────────────────────────\n\nexport function isWrongStackError(err: unknown): err is WrongStackError {\n return err instanceof WrongStackError;\n}\n\nexport function isToolError(err: unknown): err is ToolError {\n return err instanceof ToolError;\n}\n\nexport function isConfigError(err: unknown): err is ConfigError {\n return err instanceof ConfigError;\n}\n\nexport function isPluginError(err: unknown): err is PluginError {\n return err instanceof PluginError;\n}\n\nexport function isSessionError(err: unknown): err is SessionError {\n return err instanceof SessionError;\n}\n\nexport function isAgentError(err: unknown): err is AgentError {\n return err instanceof AgentError;\n}\n","/**\n * ExtensionRegistry — manages AgentExtension registrations.\n *\n * Extensions are called in registration order at each lifecycle phase.\n * Each extension hook failure is caught and logged independently so\n * one bad extension can't take down the agent.\n */\n\nimport type { TextBlock } from '../types/blocks.js';\nimport { WrongStackError } from '../types/errors.js';\nimport type { Logger } from '../types/logger.js';\nimport type { SystemPromptContributor } from '../types/system-prompt-contributor.js';\nimport type {\n AfterIterationHook,\n AfterRunHook,\n AfterToolExecutionHook,\n AgentExtension,\n BeforeIterationHook,\n BeforeRunHook,\n BeforeToolExecutionHook,\n OnErrorHook,\n ProviderRunnerFn,\n ProviderRunnerWrapper,\n} from './extension-points.js';\n\nexport class ExtensionRegistry {\n private readonly extensions: AgentExtension[] = [];\n private readonly promptContributors: SystemPromptContributor[] = [];\n private log: Logger | undefined;\n\n setLogger(log: Logger): void {\n this.log = log;\n }\n\n /**\n * Register a system prompt contributor. Returns an unregister function.\n * Contributors are called on every system prompt build in registration\n * order. Their output blocks are inserted after the core environment\n * block, before the mode and plan blocks.\n */\n registerSystemPromptContributor(c: SystemPromptContributor): () => void {\n this.promptContributors.push(c);\n return () => {\n const idx = this.promptContributors.indexOf(c);\n if (idx >= 0) this.promptContributors.splice(idx, 1);\n };\n }\n\n /**\n * Build all registered system prompt contributions.\n * Failures are caught and logged — one bad contributor doesn't\n * break the prompt assembly.\n */\n async buildSystemPromptContributions(\n ctx: Parameters<SystemPromptContributor>[0],\n ): Promise<TextBlock[]> {\n const blocks: TextBlock[] = [];\n for (const c of this.promptContributors) {\n try {\n const contributed = await c(ctx);\n blocks.push(...contributed);\n } catch (err) {\n this.log?.error('SystemPromptContributor failed', err);\n }\n }\n return blocks;\n }\n\n /**\n * Returns the live array of contributors (readonly snapshot for\n * passing to DefaultSystemPromptBuilder at build time).\n */\n listSystemPromptContributors(): readonly SystemPromptContributor[] {\n return this.promptContributors;\n }\n\n /**\n * Register an extension. Duplicate names are rejected.\n * Returns an unregister function.\n */\n register(ext: AgentExtension): () => void {\n if (this.extensions.some((e) => e.name === ext.name)) {\n throw new WrongStackError({\n message: `Extension \"${ext.name}\" already registered`,\n code: 'REGISTRY_DUPLICATE',\n subsystem: 'container',\n context: { extension: ext.name },\n });\n }\n this.extensions.push(ext);\n return () => this.unregister(ext.name);\n }\n\n /**\n * Register an extension, silently replacing any previous registration\n * with the same name. Use this when overriding a default extension.\n */\n registerOrReplace(ext: AgentExtension): () => void {\n const idx = this.extensions.findIndex((e) => e.name === ext.name);\n if (idx >= 0) this.extensions.splice(idx, 1);\n return this.register(ext);\n }\n\n /**\n * Unregister an extension by name. Returns true if found.\n */\n unregister(name: string): boolean {\n const idx = this.extensions.findIndex((e) => e.name === name);\n if (idx === -1) return false;\n this.extensions.splice(idx, 1);\n return true;\n }\n\n /**\n * List registered extension names in order.\n */\n list(): readonly string[] {\n return this.extensions.map((e) => e.name);\n }\n\n /**\n * Check if an extension with the given name is registered.\n */\n has(name: string): boolean {\n return this.extensions.some((e) => e.name === name);\n }\n\n /**\n * Remove all registered extensions and contributors.\n */\n clear(): void {\n this.extensions.length = 0;\n this.promptContributors.length = 0;\n }\n\n // ── Hook runners ─────────────────────────────────────────────────\n\n async runBeforeRun(...args: Parameters<BeforeRunHook>): Promise<void> {\n for (const ext of this.extensions) {\n if (!ext.beforeRun) continue;\n try {\n await ext.beforeRun(...args);\n } catch (err) {\n this.log?.error(`Extension \"${ext.name}\" beforeRun hook failed`, err);\n }\n }\n }\n\n async runAfterRun(...args: Parameters<AfterRunHook>): Promise<void> {\n for (const ext of this.extensions) {\n if (!ext.afterRun) continue;\n try {\n await ext.afterRun(...args);\n } catch (err) {\n this.log?.error(`Extension \"${ext.name}\" afterRun hook failed`, err);\n }\n }\n }\n\n async runBeforeIteration(...args: Parameters<BeforeIterationHook>): Promise<void> {\n for (const ext of this.extensions) {\n if (!ext.beforeIteration) continue;\n try {\n await ext.beforeIteration(...args);\n } catch (err) {\n this.log?.error(`Extension \"${ext.name}\" beforeIteration hook failed`, err);\n }\n }\n }\n\n async runAfterIteration(...args: Parameters<AfterIterationHook>): Promise<void> {\n for (const ext of this.extensions) {\n if (!ext.afterIteration) continue;\n try {\n await ext.afterIteration(...args);\n } catch (err) {\n this.log?.error(`Extension \"${ext.name}\" afterIteration hook failed`, err);\n }\n }\n }\n\n /**\n * Run onError hooks in order. The first hook that returns a non-void\n * result wins; subsequent hooks are skipped.\n */\n async runOnError(\n ...args: Parameters<OnErrorHook>\n ): Promise<\n { action: 'retry'; model?: string } | { action: 'fail' } | { action: 'continue' } | void\n > {\n for (const ext of this.extensions) {\n if (!ext.onError) continue;\n try {\n const result = await ext.onError(...args);\n if (result) return result;\n } catch (err) {\n this.log?.error(`Extension \"${ext.name}\" onError hook failed`, err);\n }\n }\n }\n\n /**\n * Build a composed provider runner. Extensions with `wrapProviderRunner`\n * form a middleware-style chain: the innermost extension wraps the\n * default runner, each subsequent wrapper wraps the previous.\n */\n wrapProviderRunner(inner: ProviderRunnerFn): ProviderRunnerFn {\n const wrappers = this.extensions\n .filter((e) => e.wrapProviderRunner)\n .map((e) => ({ name: e.name, wrap: e.wrapProviderRunner! }));\n\n if (wrappers.length === 0) return inner;\n\n // Build chain from innermost to outermost\n let composed: ProviderRunnerFn = inner;\n for (let i = wrappers.length - 1; i >= 0; i--) {\n const wrapper = wrappers[i]!;\n const next = composed;\n composed = async (ctx, req) => {\n try {\n return await wrapper.wrap(ctx, req, next);\n } catch (err) {\n this.log?.error(`Extension \"${wrapper.name}\" wrapProviderRunner failed`, err);\n throw err;\n }\n };\n }\n return composed;\n }\n\n async runBeforeToolExecution(\n ...args: Parameters<BeforeToolExecutionHook>\n ): Promise<Parameters<BeforeToolExecutionHook>[1]> {\n let toolUses = args[1];\n for (const ext of this.extensions) {\n if (!ext.beforeToolExecution) continue;\n try {\n toolUses = await ext.beforeToolExecution(args[0], toolUses);\n } catch (err) {\n this.log?.error(`Extension \"${ext.name}\" beforeToolExecution hook failed`, err);\n }\n }\n return toolUses;\n }\n\n async runAfterToolExecution(...args: Parameters<AfterToolExecutionHook>): Promise<void> {\n for (const ext of this.extensions) {\n if (!ext.afterToolExecution) continue;\n try {\n await ext.afterToolExecution(...args);\n } catch (err) {\n this.log?.error(`Extension \"${ext.name}\" afterToolExecution hook failed`, err);\n }\n }\n }\n}\n"]}
|