pantheon-opencode 1.5.0-beta.2 → 1.5.0-beta.20

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 (107) hide show
  1. package/.pantheon/code-mode/manifest.json +21 -0
  2. package/AGENTS.md +14 -63
  3. package/CHANGELOG.md +1260 -0
  4. package/README.md +162 -23
  5. package/README.pt-BR.md +104 -19
  6. package/bin/pantheon-init.mjs +178 -21
  7. package/commands/pantheon.md +1 -35
  8. package/package.json +23 -15
  9. package/plugin.json +1 -1
  10. package/pyproject.toml +62 -0
  11. package/scripts/code_mode_server.py +431 -55
  12. package/scripts/doctor.mjs +207 -9
  13. package/scripts/install/cli-ui.mjs +17 -11
  14. package/scripts/install/config-migration.mjs +4 -6
  15. package/scripts/install/opencode.mjs +137 -35
  16. package/scripts/install/plugin.mjs +12 -45
  17. package/scripts/install/shared.mjs +43 -2
  18. package/scripts/install/strings.mjs +129 -0
  19. package/scripts/install-mcp.mjs +1 -0
  20. package/scripts/manifest-inventory.mjs +410 -0
  21. package/scripts/mcp_persistence_server.py +536 -164
  22. package/scripts/package-evidence.mjs +242 -0
  23. package/scripts/probe-pantheon-cost.mjs +10 -1
  24. package/scripts/release-beta-version.mjs +56 -60
  25. package/scripts/sync-artifacts.mjs +169 -0
  26. package/scripts/sync-tui.mjs +129 -64
  27. package/scripts/test-opencode-v1-v2-sandbox.sh +79 -161
  28. package/scripts/validation-policy.mjs +7 -7
  29. package/scripts/version-check.mjs +113 -108
  30. package/scripts/versioning.mjs +230 -86
  31. package/src/agents/apollo.md +1 -0
  32. package/src/agents/gaia.md +1 -0
  33. package/src/agents/mnemosyne.md +1 -1
  34. package/src/agents/zeus.md +11 -54
  35. package/src/instructions/memory-protocol.instructions.md +6 -3
  36. package/src/instructions/zeus-anti-stall.instructions.md +2 -4
  37. package/src/instructions/zeus-council-synthesis.instructions.md +3 -28
  38. package/src/instructions/zeus-timeout-retry.instructions.md +3 -28
  39. package/src/mcp/code_mode_server.py +431 -55
  40. package/src/mcp/mcp_persistence_server.py +536 -164
  41. package/src/mcp/toon_codec.py +2 -4
  42. package/src/pantheon/background-job-board.ts +47 -5
  43. package/src/pantheon/{delegation-compaction.ts → compaction-context.ts} +10 -100
  44. package/src/pantheon/cost-command.ts +309 -44
  45. package/src/pantheon/delegation-enforce.ts +59 -14
  46. package/src/pantheon/file-persistence-salvage.ts +64 -0
  47. package/src/pantheon/file-persistence.ts +194 -45
  48. package/src/pantheon/goal-loop.ts +1 -1
  49. package/src/pantheon/goal-store.ts +1 -1
  50. package/src/pantheon/idle-continuation.ts +4 -6
  51. package/src/pantheon/model-command.ts +1 -2
  52. package/src/pantheon/native-task-status.ts +139 -0
  53. package/src/pantheon/presets.d.mts +0 -20
  54. package/src/pantheon/presets.mjs +0 -64
  55. package/src/pantheon/session-guard.ts +2 -2
  56. package/src/pantheon/session-id.ts +30 -0
  57. package/src/pantheon/shared-board.ts +37 -0
  58. package/src/pantheon/task-result-guard.ts +106 -0
  59. package/src/pantheon/todo-enforcer.ts +3 -3
  60. package/src/pantheon/tool-ceiling.ts +159 -0
  61. package/src/pantheon/tool-context.ts +23 -0
  62. package/src/pantheon/v2-bridge.ts +1 -4
  63. package/src/plugin-v2.ts +8 -21
  64. package/src/plugin.ts +83 -251
  65. package/src/plugins/pantheon-hooks.ts +51 -0
  66. package/src/plugins/tui/dist/server.d.ts +1 -1
  67. package/src/plugins/tui/dist/server.d.ts.map +1 -1
  68. package/src/plugins/tui/dist/server.js +1 -1
  69. package/src/plugins/tui/dist/server.js.map +1 -1
  70. package/src/plugins/tui/dist/tui.d.ts +261 -77
  71. package/src/plugins/tui/dist/tui.d.ts.map +1 -1
  72. package/src/plugins/tui/dist/tui.js +640 -234
  73. package/src/plugins/tui/dist/tui.js.map +1 -1
  74. package/src/plugins/tui/package-lock.json +2 -2
  75. package/src/plugins/tui/package.json +2 -2
  76. package/src/plugins/tui/src/index.tsx +872 -226
  77. package/src/routing.yml +0 -16
  78. package/src/skills/context-compression/SKILL.md +1 -1
  79. package/scripts/generate-preset-docs.mjs +0 -300
  80. package/scripts/generate-prompts.sh +0 -222
  81. package/scripts/hash_verify.py +0 -192
  82. package/scripts/install/component.mjs +0 -24
  83. package/scripts/install/migrate-config.mjs +0 -302
  84. package/scripts/test-context-compression.sh +0 -169
  85. package/src/pantheon/auto-wake.ts +0 -89
  86. package/src/pantheon/chat-reminders.ts +0 -9
  87. package/src/pantheon/comment-checker.ts +0 -48
  88. package/src/pantheon/compaction-assert.ts +0 -160
  89. package/src/pantheon/cost-tracker.ts +0 -108
  90. package/src/pantheon/delegate-manager.ts +0 -669
  91. package/src/pantheon/delegation-classifier.ts +0 -307
  92. package/src/pantheon/delegation-finalize.ts +0 -374
  93. package/src/pantheon/delegation-notify.ts +0 -212
  94. package/src/pantheon/delegation-toast.ts +0 -73
  95. package/src/pantheon/delegation.ts +0 -1120
  96. package/src/pantheon/dispatch-guard.ts +0 -109
  97. package/src/pantheon/permission-eval.ts +0 -284
  98. package/src/pantheon/preemptive-compact.ts +0 -76
  99. package/src/pantheon/retry-policy.ts +0 -212
  100. package/src/pantheon/step-cap.ts +0 -110
  101. package/src/pantheon/token-opt.ts +0 -781
  102. package/src/pantheon/toon-codec.ts +0 -481
  103. package/src/pantheon/v2-events.ts +0 -131
  104. package/src/pantheon/v2-hooks.ts +0 -287
  105. package/src/pantheon/v2-tool-definitions.ts +0 -331
  106. package/src/pantheon/zeus-delegate-with-retry.ts +0 -406
  107. package/src/plugins/tui/dist/tui.tsx +0 -2605
@@ -1,109 +0,0 @@
1
- /**
2
- * Dispatch Guard (Wave 4, PR #46) — auto-retry-on-empty for background
3
- * delegations. Classifies a dispatch result and redispatch once when it
4
- * came back empty.
5
- *
6
- * Failure signature classification:
7
- * - 'content' — text content present, nothing to do.
8
- * - 'empty-mode1' — NO content AND NO tokens: the dispatch returned
9
- * nothing at all (no reply / early abort).
10
- * - 'empty-mode2' — no text part but tokens present: reasoning happened
11
- * but the text part was lost — the themis Wave-2
12
- * failure signature that triggered this guard.
13
- *
14
- * Retry cap is HARD at 1 (never 2x). The guard is stateful per instance:
15
- * one retry budget total. It is a PURE library — it does NOT intercept
16
- * task() (opencode 1.18.x cannot intercept task completion via hooks, so
17
- * integration is manual-orchestration; see zeus.md "Empty-Result Retry").
18
- *
19
- * @module dispatch-guard
20
- */
21
-
22
- import { createPantheonLogger } from './logger.ts'
23
-
24
- // Silence-by-default TUI policy (pantheon-hooks L42-58): the default warn
25
- // fallback logs to .pantheon/logs/hooks.log; console echo is opt-in via
26
- // PANTHEON_HOOKS_LOG=1. `logger` injection stays for tests.
27
- const log = createPantheonLogger({ module: 'pantheon-dispatch-guard' })
28
-
29
- /** A dispatch result — the minimal shape needed to classify emptiness. */
30
- export interface DispatchResultLike {
31
- /** Text content of the result (task_result / report markdown). */
32
- content?: string | null
33
- /** Input tokens (present when the model actually reasoned). */
34
- tokensInput?: number
35
- /** Output tokens (present when the model actually reasoned). */
36
- tokensOutput?: number
37
- /** R1: the failure that produced this result (classified by retry-policy). */
38
- error?: unknown
39
- }
40
-
41
- /** Classification of a dispatch result. */
42
- export type DispatchClassification = 'content' | 'empty-mode1' | 'empty-mode2'
43
-
44
- export interface DispatchGuardOptions {
45
- /** Enable auto-retry on empty results. Default: true. */
46
- retryOnEmpty?: boolean
47
- /** Injectable logger (defaults to the Pantheon file log, env-gated echo). */
48
- logger?: { warn: (message: string) => void }
49
- }
50
-
51
- export interface DispatchGuard {
52
- /** Classify a result: content, empty-mode1 (nothing), empty-mode2 (reasoning, no text). */
53
- classifyResult(result: DispatchResultLike): DispatchClassification
54
- /**
55
- * Redispatch ONCE when the result is empty and retryOnEmpty is enabled.
56
- * A retried result that is STILL empty is returned as-is — the cap of 1
57
- * is never exceeded. Never throws.
58
- */
59
- maybeRetry(
60
- result: DispatchResultLike,
61
- redispatch: () => Promise<DispatchResultLike>,
62
- ): Promise<{
63
- classification: DispatchClassification
64
- retried: boolean
65
- result: DispatchResultLike
66
- }>
67
- }
68
-
69
- /** 1-based retry budget — the hard cap. Never 2x. */
70
- const MAX_RETRIES = 1
71
-
72
- /** Whether a result carries any text content (whitespace-only counts as empty). */
73
- function hasContent(result: DispatchResultLike): boolean {
74
- return typeof result.content === 'string' && result.content.trim() !== ''
75
- }
76
-
77
- /** Whether the result carries any token counts (i.e. the model actually ran). */
78
- function hasTokens(result: DispatchResultLike): boolean {
79
- return (result.tokensInput ?? 0) > 0 || (result.tokensOutput ?? 0) > 0
80
- }
81
-
82
- export function createDispatchGuard(options?: DispatchGuardOptions): DispatchGuard {
83
- const retryOnEmpty = options?.retryOnEmpty ?? true
84
- const warn = options?.logger?.warn ?? ((message: string) => log.warn(message))
85
- // Per-instance retry budget — the "cap 1" enforcement point.
86
- let retriesUsed = 0
87
-
88
- return {
89
- classifyResult(result: DispatchResultLike): DispatchClassification {
90
- if (hasContent(result)) return 'content'
91
- return hasTokens(result) ? 'empty-mode2' : 'empty-mode1'
92
- },
93
-
94
- async maybeRetry(result, redispatch) {
95
- const classification = this.classifyResult(result)
96
- if (!retryOnEmpty || classification === 'content' || retriesUsed >= MAX_RETRIES) {
97
- return { classification, retried: false, result }
98
- }
99
- retriesUsed += 1
100
- warn(`dispatch returned ${classification}; retrying once (cap ${MAX_RETRIES})`)
101
- const retriedResult = await redispatch()
102
- return {
103
- classification: this.classifyResult(retriedResult),
104
- retried: true,
105
- result: retriedResult,
106
- }
107
- },
108
- }
109
- }
@@ -1,284 +0,0 @@
1
- /**
2
- * S1 — Deny-first permission evaluation + PreToolUse-style hooks.
3
- *
4
- * Deny rules checked BEFORE allow rules. User-level denies cannot be
5
- * overridden by project config. Hook exit-code semantics:
6
- * exit 0 → allow, exit 2 → hard deny, missing/throw → fail-open.
7
- *
8
- * @module permission-eval
9
- */
10
-
11
- // ─── Types ─────────────────────────────────────────────────────────────
12
-
13
- /** Result of a permission evaluation. */
14
- export interface PermissionResult {
15
- allowed: boolean
16
- reason: 'allow' | 'deny' | 'hook_deny' | 'no_match'
17
- message?: string
18
- }
19
-
20
- /** Hook script result (mirrors hook-runner.ts HookResult). */
21
- export interface HookResult {
22
- code: number
23
- stdout: string
24
- stderr: string
25
- /** When true, the hook script was not found — treat as fail-open. */
26
- missing?: boolean
27
- }
28
-
29
- /**
30
- * PreToolUse hook function signature.
31
- * exit 0 → allow, exit 2 → hard deny, exit 1 → soft deny.
32
- * Throw = fail-open.
33
- */
34
- export type PreToolUseHook = (tool: string, args: unknown) => Promise<HookResult> | HookResult
35
-
36
- /** Permission configuration. */
37
- export interface PermissionConfig {
38
- /** Deny rules — evaluated first. User-level denies override project allows. */
39
- deny: readonly string[]
40
- /** Allow rules — evaluated after deny. */
41
- allow: readonly string[]
42
- /** Optional hooks configuration. */
43
- hooks?: {
44
- /** PreToolUse hook function (in-process, not script path). */
45
- preToolUse?: PreToolUseHook
46
- }
47
- }
48
-
49
- // ─── Pattern matching ──────────────────────────────────────────────────
50
-
51
- /**
52
- * Parse a permission rule pattern into { tool, argPattern }.
53
- *
54
- * Supported formats:
55
- * - "tool" → tool="tool", argPattern="*"
56
- * - "tool:*" → tool="tool", argPattern="*"
57
- * - "tool:subcommand" → tool="tool", argPattern="subcommand"
58
- * - "tool(pattern)" → tool="tool", argPattern="pattern" (full match)
59
- * - "*:*" → tool="*", argPattern="*"
60
- * - "git push:*" → tool="git push", argPattern="*" (subcommand prefix)
61
- */
62
- function parsePattern(pattern: string): { tool: string; argPattern: string } {
63
- // Handle "tool(pattern)" format — the content inside parens is the full
64
- // arg pattern. Colons inside parens are literal, not separators.
65
- const parenMatch = /^([^(]+)\(([^)]+)\)$/.exec(pattern)
66
- if (parenMatch) {
67
- const tool = parenMatch[1] ?? ''
68
- const argPattern = parenMatch[2] ?? ''
69
- return { tool, argPattern }
70
- }
71
-
72
- // Handle "tool:argPattern" or just "tool"
73
- const colonIdx = pattern.indexOf(':')
74
- if (colonIdx === -1) {
75
- return { tool: pattern, argPattern: '*' }
76
- }
77
- const argPattern = pattern.slice(colonIdx + 1)
78
- return {
79
- tool: pattern.slice(0, colonIdx),
80
- // Empty argPattern after colon = wildcard (e.g. "*:" matches any args)
81
- argPattern: argPattern === '' ? '*' : argPattern,
82
- }
83
- }
84
-
85
- /**
86
- * Convert a glob-like pattern to a regex. `*` matches any sequence.
87
- * Anchored to full match.
88
- */
89
- function globToRegex(pattern: string): RegExp {
90
- const escaped = pattern
91
- .replace(/[.+^${}()|[\]\\]/g, '\\$&')
92
- .replace(/\*/g, '.*')
93
- .replace(/\?/g, '.')
94
- return new RegExp(`^${escaped}$`)
95
- }
96
-
97
- /** Extract a string representation from tool args for pattern matching. */
98
- function argsToString(args: unknown): string {
99
- if (args === null || args === undefined) return ''
100
- if (typeof args === 'string') return args
101
- if (typeof args === 'object') {
102
- const record = args as Record<string, unknown>
103
- // Try common field names
104
- for (const key of ['command', 'cmd', 'query', 'pattern', 'filePath', 'path', 'url']) {
105
- if (typeof record[key] === 'string') return record[key]
106
- }
107
- return JSON.stringify(args)
108
- }
109
- return String(args)
110
- }
111
-
112
- /**
113
- * Build a combined string for matching: "toolName argString".
114
- * This allows patterns like "git push:*" to match tool="git" with args
115
- * containing "push".
116
- */
117
- function combinedString(tool: string, args: unknown): string {
118
- const argsStr = argsToString(args)
119
- return argsStr !== '' ? `${tool} ${argsStr}` : tool
120
- }
121
-
122
- /**
123
- * Check if a tool+args matches a permission rule pattern.
124
- * Returns true when the pattern matches.
125
- *
126
- * Matching strategy:
127
- * 1. Wildcard tool ("*") → match any tool with argPattern
128
- * 2. Direct tool match → check argPattern against args
129
- * 3. Combined match (for "git push:*" style patterns) → prefix match
130
- */
131
- function matchesRule(pattern: string, tool: string, args: unknown): boolean {
132
- const { tool: patternTool, argPattern } = parsePattern(pattern)
133
-
134
- // Wildcard tool: match any tool
135
- if (patternTool === '*') {
136
- if (argPattern === '*') return true
137
- const argsStr = argsToString(args)
138
- return globToRegex(argPattern).test(argsStr)
139
- }
140
-
141
- // Check direct tool match
142
- if (globToRegex(patternTool).test(tool)) {
143
- if (argPattern === '*') return true
144
- const argsStr = argsToString(args)
145
- return globToRegex(argPattern).test(argsStr)
146
- }
147
-
148
- // Combined match for patterns with spaces (e.g. "git push:*")
149
- // The pattern tool includes a subcommand that appears in args.
150
- // Use prefix matching: "git push" matches "git push origin main"
151
- if (patternTool.includes(' ')) {
152
- const combined = combinedString(tool, args)
153
- // Prefix match: combined starts with pattern tool
154
- if (combined.startsWith(patternTool)) {
155
- return true
156
- }
157
- }
158
-
159
- return false
160
- }
161
-
162
- // ─── Evaluation ────────────────────────────────────────────────────────
163
-
164
- /**
165
- * Evaluate a permission config against a tool+args invocation.
166
- *
167
- * Evaluation order:
168
- * 1. Check deny rules (user-level first, then project-level)
169
- * 2. If no deny matched, run the PreToolUse hook
170
- * 3. If hook exits 0 (or missing), check allow rules
171
- * 4. If no allow matched → deny (implicit deny)
172
- *
173
- * User-level denies override project-level allows: a deny in the user config
174
- * cannot be cancelled by an allow in the project config.
175
- */
176
- export function evaluatePermission(
177
- config: PermissionConfig,
178
- invocation: { tool: string; args?: unknown },
179
- ): PermissionResult {
180
- const { tool, args } = invocation
181
-
182
- // ── Step 1: Deny-first check ────────────────────────────────────────
183
- for (const pattern of config.deny) {
184
- if (matchesRule(pattern, tool, args)) {
185
- return { allowed: false, reason: 'deny', message: `denied by rule: ${pattern}` }
186
- }
187
- }
188
-
189
- // ── Step 2: PreToolUse hook ─────────────────────────────────────────
190
- if (config.hooks?.preToolUse !== undefined) {
191
- try {
192
- const hookResult = config.hooks.preToolUse(tool, args)
193
- // Support both sync and async hooks
194
- // For sync hooks (non-Promise), check immediately
195
- if (!(hookResult instanceof Promise)) {
196
- // missing flag: hook script not found → fail-open
197
- if (hookResult.missing === true) {
198
- // Fall through to config evaluation (fail-open)
199
- } else if (hookResult.code === 2) {
200
- return {
201
- allowed: false,
202
- reason: 'hook_deny',
203
- message: `hook hard deny (exit 2): ${hookResult.stderr}`,
204
- }
205
- } else if (hookResult.code !== 0) {
206
- return {
207
- allowed: false,
208
- reason: 'hook_deny',
209
- message: `hook deny (exit ${hookResult.code}): ${hookResult.stderr}`,
210
- }
211
- }
212
- // Hook exited 0 (or missing) → fall through to allow check
213
- }
214
- // For async hooks, we cannot await in a sync function.
215
- // Use evaluatePermissionAsync for async hooks.
216
- } catch {
217
- // Hook throw → fail-open: fall through to config evaluation
218
- }
219
- }
220
-
221
- // ── Step 3: Allow check ─────────────────────────────────────────────
222
- for (const pattern of config.allow) {
223
- if (matchesRule(pattern, tool, args)) {
224
- return { allowed: true, reason: 'allow', message: `allowed by rule: ${pattern}` }
225
- }
226
- }
227
-
228
- // ── Step 4: No match → implicit deny ────────────────────────────────
229
- return { allowed: false, reason: 'no_match', message: 'no matching allow rule' }
230
- }
231
-
232
- /**
233
- * Async version of evaluatePermission that properly awaits hooks.
234
- * Use this when hooks may be async (e.g., shell scripts).
235
- */
236
- export async function evaluatePermissionAsync(
237
- config: PermissionConfig,
238
- invocation: { tool: string; args?: unknown },
239
- ): Promise<PermissionResult> {
240
- const { tool, args } = invocation
241
-
242
- // ── Step 1: Deny-first check ────────────────────────────────────────
243
- for (const pattern of config.deny) {
244
- if (matchesRule(pattern, tool, args)) {
245
- return { allowed: false, reason: 'deny', message: `denied by rule: ${pattern}` }
246
- }
247
- }
248
-
249
- // ── Step 2: PreToolUse hook ─────────────────────────────────────────
250
- if (config.hooks?.preToolUse !== undefined) {
251
- try {
252
- const hookResult = await config.hooks.preToolUse(tool, args)
253
- // missing flag: hook script not found → fail-open
254
- if (hookResult.missing === true) {
255
- // Fall through to config evaluation (fail-open)
256
- } else if (hookResult.code === 2) {
257
- return {
258
- allowed: false,
259
- reason: 'hook_deny',
260
- message: `hook hard deny (exit 2): ${hookResult.stderr}`,
261
- }
262
- } else if (hookResult.code !== 0) {
263
- return {
264
- allowed: false,
265
- reason: 'hook_deny',
266
- message: `hook deny (exit ${hookResult.code}): ${hookResult.stderr}`,
267
- }
268
- }
269
- // Hook exited 0 (or missing) → fall through to allow check
270
- } catch {
271
- // Hook throw → fail-open: fall through to config evaluation
272
- }
273
- }
274
-
275
- // ── Step 3: Allow check ─────────────────────────────────────────────
276
- for (const pattern of config.allow) {
277
- if (matchesRule(pattern, tool, args)) {
278
- return { allowed: true, reason: 'allow', message: `allowed by rule: ${pattern}` }
279
- }
280
- }
281
-
282
- // ── Step 4: No match → implicit deny ────────────────────────────────
283
- return { allowed: false, reason: 'no_match', message: 'no matching allow rule' }
284
- }
@@ -1,76 +0,0 @@
1
- /**
2
- * Preemptive Compaction Check (release-134 Phase 5) — dormant, ready-to-wire.
3
- *
4
- * WHY: the SDK 1.18.x exposes NO runtime source of context-usage percentage,
5
- * so there is nothing to observe at runtime yet. This module is the pure
6
- * decision core that warns the model BEFORE the context fills up — ready to
7
- * be wired to a future usage source, but deliberately NOT wired today.
8
- *
9
- * HOW: `preemptiveCompactCheck` is a pure function: given the current usage
10
- * fraction and the last fraction at which a warning was emitted, it decides
11
- * whether to warn again (threshold + re-warn step) and returns the effective
12
- * `lastWarnedPct`. The `enqueue` callback is INJECTED by the caller — this
13
- * module stays standalone and does NOT import the shared chat-reminders
14
- * buffer.
15
- *
16
- * YAGNI: no background timers, no polling, no plugin wiring. When (if) the
17
- * SDK exposes a usage source, the caller wires it into this check.
18
- *
19
- * FAIL-OPEN: the whole check is wrapped — an enqueue failure returns the
20
- * previous state and never throws.
21
- *
22
- * @module preemptive-compact
23
- */
24
-
25
- // ─── Constants ─────────────────────────────────────────────────────────
26
-
27
- /** Warn when context usage reaches this fraction (78%). */
28
- export const PREEMPTIVE_COMPACTION_THRESHOLD = 0.78
29
-
30
- /** Re-warn only when usage rose at least this much since the last warning (5pp). */
31
- export const PREEMPTIVE_REWARN_STEP = 0.05
32
-
33
- // ─── Types ─────────────────────────────────────────────────────────────
34
-
35
- /** Dependencies threaded to preemptiveCompactCheck (wired from plugin.ts). */
36
- export interface PreemptiveCompactDeps {
37
- /** The session whose context usage is being checked. */
38
- sessionID: string
39
- /** Current context usage as a fraction (0..1). */
40
- usagePct: number
41
- /** Last fraction at which a warning was emitted (undefined = never). */
42
- lastWarnedPct: number | undefined
43
- /** Injectable enqueue (testable). The module stays standalone. */
44
- enqueue: (text: string) => void
45
- }
46
-
47
- // ─── Decision ──────────────────────────────────────────────────────────
48
-
49
- /**
50
- * Pure decision: should a warning be emitted at the given usage?
51
- * True when usage is at/above the threshold AND either no warning was
52
- * emitted yet or usage rose by at least the re-warn step since the last one.
53
- */
54
- export function shouldWarn(usagePct: number, lastWarnedPct: number | undefined): boolean {
55
- if (usagePct < PREEMPTIVE_COMPACTION_THRESHOLD) return false
56
- if (lastWarnedPct === undefined) return true
57
- return usagePct >= lastWarnedPct + PREEMPTIVE_REWARN_STEP
58
- }
59
-
60
- /**
61
- * Preemptive compaction check: when a warning should fire, enqueue the
62
- * reminder and return the usage fraction as the new `lastWarnedPct`.
63
- * Otherwise (or on any failure) return the previous `lastWarnedPct`.
64
- * NEVER throws — the hook must never break the session.
65
- */
66
- export function preemptiveCompactCheck(deps: PreemptiveCompactDeps): number | undefined {
67
- try {
68
- if (!shouldWarn(deps.usagePct, deps.lastWarnedPct)) return deps.lastWarnedPct
69
- deps.enqueue(
70
- `context usage at ${Math.round(deps.usagePct * 100)}% — consider compacting before continuing`,
71
- )
72
- return deps.usagePct
73
- } catch {
74
- return deps.lastWarnedPct
75
- }
76
- }
@@ -1,212 +0,0 @@
1
- /**
2
- * R1 — Per-error-type retry + provider cooldown (LiteLLM pattern).
3
- *
4
- * Pure library (no SDK, no I/O). When a delegation fails, the error is
5
- * classified (auth | rate_limit | timeout | other), the per-error-type
6
- * retry policy from routing.yml `retry_policy` is applied with exponential
7
- * backoff, and provider cooldown state (routing.yml `cooldown`) is tracked
8
- * in-memory: after `allowed_fails` consecutive failures a provider is
9
- * skipped for `cooldown_time_seconds`; a success resets the counter.
10
- *
11
- * YAGNI: no usage-based routing, no Redis — a simple in-memory cooldown
12
- * map plus config.
13
- *
14
- * @module retry-policy
15
- */
16
-
17
- // ─── Types ─────────────────────────────────────────────────────────────
18
-
19
- /** Error classes a delegation failure can be classified into. */
20
- export type ErrorType = 'auth' | 'rate_limit' | 'timeout' | 'other'
21
-
22
- /** Max retries per error class (routing.yml `retry_policy`). */
23
- export interface RetryPolicy {
24
- auth: number
25
- rate_limit: number
26
- timeout: number
27
- other: number
28
- }
29
-
30
- /** Provider cooldown config (routing.yml `cooldown`). */
31
- export interface CooldownConfig {
32
- /** Consecutive failures before the provider is skipped. */
33
- allowedFails: number
34
- /** How long the provider stays skipped, in seconds. */
35
- cooldownTimeSeconds: number
36
- }
37
-
38
- /** Defaults matching routing.yml `retry_policy`. */
39
- export const DEFAULT_RETRY_POLICY: RetryPolicy = {
40
- auth: 0,
41
- rate_limit: 3,
42
- timeout: 2,
43
- other: 1,
44
- }
45
-
46
- /** Defaults matching routing.yml `cooldown`. */
47
- export const DEFAULT_COOLDOWN: CooldownConfig = {
48
- allowedFails: 3,
49
- cooldownTimeSeconds: 60,
50
- }
51
-
52
- // ─── Error classification ──────────────────────────────────────────────
53
-
54
- const AUTH_RE = /(auth|unauthorized|401|403|api[ _-]?key|invalid credential|permission denied)/i
55
- const RATE_LIMIT_RE = /(rate[ _-]?limit|429|too many requests|quota|throttl)/i
56
- const TIMEOUT_RE = /(timeout|timed out|deadline|504|408|etimedout)/i
57
-
58
- /**
59
- * Classify a thrown error into one of the four retry-policy buckets.
60
- * Message-based matching (LiteLLM pattern); unknown errors → 'other'.
61
- */
62
- export function classifyError(err: unknown): ErrorType {
63
- const message = err instanceof Error ? err.message : String(err ?? '')
64
- if (AUTH_RE.test(message)) return 'auth'
65
- if (RATE_LIMIT_RE.test(message)) return 'rate_limit'
66
- if (TIMEOUT_RE.test(message)) return 'timeout'
67
- return 'other'
68
- }
69
-
70
- // ─── Exponential backoff ───────────────────────────────────────────────
71
-
72
- /**
73
- * Exponential backoff delay for a retry attempt (0-based): base * 2^attempt,
74
- * capped at maxMs. attempt 0 → base, attempt 1 → 2x base, etc.
75
- */
76
- export function computeBackoffMs(attempt: number, baseMs = 1000, maxMs = 30_000): number {
77
- const delay = baseMs * 2 ** attempt
78
- return Math.min(delay, maxMs)
79
- }
80
-
81
- // ─── Provider cooldown tracker ─────────────────────────────────────────
82
-
83
- interface CooldownEntry {
84
- fails: number
85
- cooldownUntil: number
86
- }
87
-
88
- /**
89
- * In-memory provider cooldown state. After `allowedFails` consecutive
90
- * failures a provider is skipped until `cooldownUntil`; a success resets
91
- * the counter. The clock is injectable for tests.
92
- */
93
- export class ProviderCooldownTracker {
94
- private readonly state = new Map<string, CooldownEntry>()
95
- readonly config: CooldownConfig
96
-
97
- constructor(config: CooldownConfig = DEFAULT_COOLDOWN) {
98
- this.config = config
99
- }
100
-
101
- /** Record one failure for a provider; trips cooldown at allowedFails. */
102
- recordFailure(provider: string): void {
103
- const entry = this.state.get(provider) ?? { fails: 0, cooldownUntil: 0 }
104
- entry.fails += 1
105
- if (entry.fails >= this.config.allowedFails) {
106
- entry.cooldownUntil = Date.now() + this.config.cooldownTimeSeconds * 1000
107
- }
108
- this.state.set(provider, entry)
109
- }
110
-
111
- /** Record a success — resets the failure counter (no spurious cooldown). */
112
- recordSuccess(provider: string): void {
113
- this.state.delete(provider)
114
- }
115
-
116
- /**
117
- * Whether the provider is currently skipped. Expired cooldowns are
118
- * cleared; entries that never tripped (cooldownUntil 0) are kept so the
119
- * consecutive-failure counter survives the check.
120
- */
121
- isInCooldown(provider: string, now = Date.now()): boolean {
122
- const entry = this.state.get(provider)
123
- if (entry === undefined) return false
124
- if (entry.cooldownUntil > now) return true
125
- if (entry.cooldownUntil > 0) this.state.delete(provider)
126
- return false
127
- }
128
-
129
- /** Remaining cooldown time in ms (0 when not in cooldown). */
130
- remainingCooldownMs(provider: string, now = Date.now()): number {
131
- const entry = this.state.get(provider)
132
- if (entry === undefined) return 0
133
- return Math.max(0, entry.cooldownUntil - now)
134
- }
135
-
136
- /** Current consecutive failure count for a provider. */
137
- failCount(provider: string): number {
138
- return this.state.get(provider)?.fails ?? 0
139
- }
140
- }
141
-
142
- // ─── Retry policy engine ───────────────────────────────────────────────
143
-
144
- export type RetryDecisionReason = 'policy' | 'cooldown'
145
-
146
- export interface RetryDecision {
147
- shouldRetry: boolean
148
- /** Retries still available for this error class (0 when exhausted). */
149
- retriesRemaining: number
150
- /** Backoff delay to wait before the next attempt (ms). */
151
- delayMs: number
152
- /** Why the decision was made: policy budget or provider cooldown. */
153
- reason: RetryDecisionReason
154
- }
155
-
156
- /**
157
- * Combines error classification, the per-error-type retry policy, and the
158
- * provider cooldown tracker into one decision point. Enforcement: when a
159
- * delegation fails, call `decide(provider, err, attemptsUsed)`; when the
160
- * result is content, call `recordSuccess`; when the budget is exhausted,
161
- * call `recordFailure` so the cooldown can trip.
162
- */
163
- export class RetryPolicyEngine {
164
- private readonly cooldown: ProviderCooldownTracker
165
- readonly policy: RetryPolicy
166
- readonly backoffBaseMs: number
167
-
168
- constructor(
169
- policy: RetryPolicy = DEFAULT_RETRY_POLICY,
170
- cooldown?: ProviderCooldownTracker,
171
- backoffBaseMs = 1000,
172
- ) {
173
- this.policy = policy
174
- this.backoffBaseMs = backoffBaseMs
175
- this.cooldown = cooldown ?? new ProviderCooldownTracker()
176
- }
177
-
178
- /**
179
- * Decide whether a failed dispatch should be retried.
180
- *
181
- * @param provider provider name (cooldown key, e.g. "opencode")
182
- * @param err the failure (classified into auth/rate_limit/timeout/other)
183
- * @param attemptsUsed retries already consumed for this dispatch
184
- * @param now injectable clock (tests)
185
- */
186
- decide(provider: string, err: unknown, attemptsUsed: number, now = Date.now()): RetryDecision {
187
- if (this.cooldown.isInCooldown(provider, now)) {
188
- return { shouldRetry: false, retriesRemaining: 0, delayMs: 0, reason: 'cooldown' }
189
- }
190
- const type = classifyError(err)
191
- const retriesRemaining = Math.max(0, this.policy[type] - attemptsUsed)
192
- if (retriesRemaining <= 0) {
193
- return { shouldRetry: false, retriesRemaining: 0, delayMs: 0, reason: 'policy' }
194
- }
195
- return {
196
- shouldRetry: true,
197
- retriesRemaining,
198
- delayMs: computeBackoffMs(attemptsUsed, this.backoffBaseMs),
199
- reason: 'policy',
200
- }
201
- }
202
-
203
- /** Record a failure (feeds the cooldown tracker). */
204
- recordFailure(provider: string, _err: unknown): void {
205
- this.cooldown.recordFailure(provider)
206
- }
207
-
208
- /** Record a success (resets the provider's failure counter). */
209
- recordSuccess(provider: string): void {
210
- this.cooldown.recordSuccess(provider)
211
- }
212
- }