@pugi/cli 0.1.0-beta.7 → 0.1.0-beta.87
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/CHANGELOG.md +96 -0
- package/THIRD_PARTY_NOTICES.md +40 -0
- package/assets/pugi-prozr2-mascot.ansi +9 -0
- package/bin/run.js +33 -1
- package/dist/commands/deploy.js +40 -40
- package/dist/commands/flatten.js +191 -0
- package/dist/commands/jobs-watch.js +201 -0
- package/dist/commands/jobs.js +42 -27
- package/dist/commands/smoke.js +133 -0
- package/dist/core/agent-progress/cleanup.js +134 -0
- package/dist/core/agent-progress/schema.js +144 -0
- package/dist/core/agent-progress/writer.js +101 -0
- package/dist/core/agents/adaptive-router.js +330 -0
- package/dist/core/agents/query-decomposer.js +297 -0
- package/dist/core/agents/registry.js +2 -2
- package/dist/core/approvals/shortcut-resolver.js +98 -0
- package/dist/core/artifact-chain/dispatcher.js +148 -0
- package/dist/core/artifact-chain/exporter.js +164 -0
- package/dist/core/artifact-chain/state.js +243 -0
- package/dist/core/artifact-chain/steps.js +169 -0
- package/dist/core/ask-user/question.js +92 -0
- package/dist/core/audit/audit-trail.js +275 -0
- package/dist/core/auth/ensure-authenticated.js +129 -0
- package/dist/core/auth/env-provider.js +238 -0
- package/dist/core/auto-open-browser.js +4 -4
- package/dist/core/auto-update/channels.js +122 -0
- package/dist/core/auto-update/checker.js +241 -0
- package/dist/core/auto-update/state.js +235 -0
- package/dist/core/bare-mode/index.js +107 -0
- package/dist/core/bash/redirect.js +281 -0
- package/dist/core/bash-classifier.js +436 -40
- package/dist/core/checkpoint/resumer.js +149 -0
- package/dist/core/checkpoint/rewinder.js +291 -0
- package/dist/core/checkpoints/shadow-git.js +670 -0
- package/dist/core/citations/parser.js +109 -0
- package/dist/core/classifier/yolo-classifier.js +88 -0
- package/dist/core/codegraph/decision-store.js +248 -0
- package/dist/core/codegraph/detect-repo.js +459 -0
- package/dist/core/codegraph/install.js +134 -0
- package/dist/core/codegraph/offer-hook.js +220 -0
- package/dist/core/compact/auto-trigger.js +96 -0
- package/dist/core/compact/buffer-rewriter.js +115 -0
- package/dist/core/compact/summarizer.js +208 -0
- package/dist/core/compact/token-counter.js +108 -0
- package/dist/core/consensus/anvil-fanout.js +25 -25
- package/dist/core/consensus/diff-capture.js +121 -12
- package/dist/core/consensus/rubric.js +21 -21
- package/dist/core/context/builder.js +6 -6
- package/dist/core/context/compaction-events.js +8 -8
- package/dist/core/context/compaction.js +31 -31
- package/dist/core/context/index.js +15 -8
- package/dist/core/context/invariants.js +51 -51
- package/dist/core/context/markdown-loader.js +28 -10
- package/dist/core/context/markdown-traverse.js +255 -0
- package/dist/core/context/pugiignore.js +41 -41
- package/dist/core/context/repo-skeleton.js +37 -37
- package/dist/core/context/tool-eviction.js +55 -0
- package/dist/core/context/watcher.js +32 -32
- package/dist/core/context/working-set.js +23 -23
- package/dist/core/coordinator/agent-tools.js +77 -0
- package/dist/core/coordinator/agent-toolset.js +65 -0
- package/dist/core/coordinator/fsm.js +73 -0
- package/dist/core/coordinator/mode-fsm.js +70 -0
- package/dist/core/cost/rate-card.js +129 -0
- package/dist/core/cost/tracker.js +221 -0
- package/dist/core/credentials.js +12 -12
- package/dist/core/cron/scheduler.js +138 -0
- package/dist/core/denial-tracking/index.js +8 -0
- package/dist/core/denial-tracking/state.js +264 -0
- package/dist/core/diagnostics/probe-runner.js +93 -0
- package/dist/core/diagnostics/probes/api.js +46 -0
- package/dist/core/diagnostics/probes/auth.js +93 -0
- package/dist/core/diagnostics/probes/bare-mode.js +42 -0
- package/dist/core/diagnostics/probes/cli-version.js +127 -0
- package/dist/core/diagnostics/probes/config.js +72 -0
- package/dist/core/diagnostics/probes/denial-tracking.js +57 -0
- package/dist/core/diagnostics/probes/disk.js +81 -0
- package/dist/core/diagnostics/probes/engine-live.js +46 -0
- package/dist/core/diagnostics/probes/git.js +65 -0
- package/dist/core/diagnostics/probes/hooks.js +118 -0
- package/dist/core/diagnostics/probes/mcp.js +75 -0
- package/dist/core/diagnostics/probes/node.js +59 -0
- package/dist/core/diagnostics/probes/pnpm.js +36 -0
- package/dist/core/diagnostics/probes/pugi-md.js +89 -0
- package/dist/core/diagnostics/probes/sandbox.js +40 -0
- package/dist/core/diagnostics/probes/session.js +74 -0
- package/dist/core/diagnostics/probes/status-snapshot.js +488 -0
- package/dist/core/diagnostics/probes/workspace.js +63 -0
- package/dist/core/diagnostics/types.js +70 -0
- package/dist/core/dispatch/cache-cleanup.js +197 -0
- package/dist/core/dispatch/cache-handoff.js +295 -0
- package/dist/core/edits/apply-patch-layer-e.js +189 -0
- package/dist/core/edits/dispatch.js +293 -7
- package/dist/core/edits/format-matrix.js +26 -0
- package/dist/core/edits/fuzzy-ladder.js +650 -0
- package/dist/core/edits/index.js +3 -1
- package/dist/core/edits/journal.js +199 -0
- package/dist/core/edits/layer-a-apply.js +15 -15
- package/dist/core/edits/layer-a-fuzzy-apply.js +198 -0
- package/dist/core/edits/layer-b-apply.js +9 -9
- package/dist/core/edits/layer-c-apply.js +6 -6
- package/dist/core/edits/layer-d-ast.js +557 -14
- package/dist/core/edits/marker-parser.js +12 -12
- package/dist/core/edits/security-gate.js +27 -27
- package/dist/core/edits/verify-hook.js +273 -0
- package/dist/core/edits/worktree.js +322 -0
- package/dist/core/engine/anvil-client.js +140 -26
- package/dist/core/engine/auto-compact.js +179 -0
- package/dist/core/engine/budgets.js +186 -0
- package/dist/core/engine/context-prefix.js +155 -0
- package/dist/core/engine/index.js +1 -1
- package/dist/core/engine/intensity.js +158 -0
- package/dist/core/engine/intent.js +260 -0
- package/dist/core/engine/native-pugi.js +1295 -227
- package/dist/core/engine/prompts.js +134 -16
- package/dist/core/engine/strip-internal-fields.js +124 -0
- package/dist/core/engine/tool-bridge.js +1295 -59
- package/dist/core/evaluation/golden-dataset.js +293 -0
- package/dist/core/feedback/queue.js +177 -0
- package/dist/core/feedback/submitter.js +145 -0
- package/dist/core/file-cache.js +113 -1
- package/dist/core/flatten/flatten-repo.js +439 -0
- package/dist/core/format/osc8-link.js +28 -0
- package/dist/core/hook-chains.js +392 -0
- package/dist/core/hooks/citation-verify-hook.js +138 -0
- package/dist/core/hooks/citation-verify.js +112 -0
- package/dist/core/hooks/events.js +44 -0
- package/dist/core/hooks/index.js +15 -0
- package/dist/core/hooks/registry.js +213 -0
- package/dist/core/hooks/runner.js +236 -0
- package/dist/core/hooks/v2/event-emitter.js +115 -0
- package/dist/core/hooks/v2/executor.js +282 -0
- package/dist/core/hooks/v2/index.js +25 -0
- package/dist/core/hooks/v2/lifecycle.js +104 -0
- package/dist/core/hooks/v2/loader.js +216 -0
- package/dist/core/hooks/v2/matcher.js +125 -0
- package/dist/core/hooks/v2/trust.js +143 -0
- package/dist/core/hooks/v2/types.js +86 -0
- package/dist/core/image/renderer.js +71 -0
- package/dist/core/init/detector.js +582 -0
- package/dist/core/init/template-renderer.js +242 -0
- package/dist/core/jobs/registry.js +18 -18
- package/dist/core/ledger/results-tsv.js +142 -0
- package/dist/core/log-discipline/stdout-redirect.js +51 -0
- package/dist/core/lsp/cache.js +105 -0
- package/dist/core/lsp/client.js +776 -0
- package/dist/core/lsp/language-detect.js +66 -0
- package/dist/core/lsp/post-edit-diagnostics.js +171 -0
- package/dist/core/lsp/symbol-tools.js +372 -0
- package/dist/core/mcp/client.js +97 -28
- package/dist/core/mcp/http-server.js +553 -0
- package/dist/core/mcp/orchestrator-tools.js +662 -0
- package/dist/core/mcp/permission.js +190 -0
- package/dist/core/mcp/registry.js +39 -17
- package/dist/core/mcp/server-tools.js +219 -0
- package/dist/core/mcp/server.js +397 -0
- package/dist/core/mcp/trust.js +10 -10
- package/dist/core/memory/dual-write.js +416 -0
- package/dist/core/memory/passive-extract.js +130 -0
- package/dist/core/memory/phase1-kinds.js +20 -0
- package/dist/core/memory/secret-scanner.js +304 -0
- package/dist/core/memory-sync/queue.js +170 -0
- package/dist/core/metrics/extract.js +113 -0
- package/dist/core/modes/roo-modes.js +68 -0
- package/dist/core/onboarding/ensure-initialized.js +133 -0
- package/dist/core/onboarding/marker.js +111 -0
- package/dist/core/onboarding/telemetry-state.js +108 -0
- package/dist/core/output-style/presets.js +176 -0
- package/dist/core/output-style/state.js +185 -0
- package/dist/core/path-security.js +287 -5
- package/dist/core/permission.js +82 -22
- package/dist/core/permissions/auto-classifier.js +124 -0
- package/dist/core/permissions/bash-parser.js +371 -0
- package/dist/core/permissions/circuit-breaker.js +83 -0
- package/dist/core/permissions/constrained-edit.js +91 -0
- package/dist/core/permissions/gate.js +278 -0
- package/dist/core/permissions/index.js +20 -0
- package/dist/core/permissions/mode.js +174 -0
- package/dist/core/permissions/network-egress.js +137 -0
- package/dist/core/permissions/state.js +241 -0
- package/dist/core/permissions/tool-class.js +93 -0
- package/dist/core/plan-mode/ui-state.js +51 -0
- package/dist/core/plans/plan-artifact.js +721 -0
- package/dist/core/policy-limits/etag-store.js +122 -0
- package/dist/core/prd-check/parser.js +215 -0
- package/dist/core/prd-check/reporter.js +127 -0
- package/dist/core/prd-check/session-review.js +557 -0
- package/dist/core/prd-check/verifiers.js +223 -0
- package/dist/core/prompt-cache/client-cache.js +99 -0
- package/dist/core/prompts/assembly.js +29 -0
- package/dist/core/prompts/registry.js +364 -0
- package/dist/core/pugi-md/cc-compat-rules.js +735 -0
- package/dist/core/pugi-md/context-injector.js +76 -0
- package/dist/core/pugi-md/walk-up.js +207 -0
- package/dist/core/python/uv-installer.js +270 -0
- package/dist/core/python/uv-resolver.js +83 -0
- package/dist/core/rate-limit/narrator.js +146 -0
- package/dist/core/recipes/cli-types.js +20 -0
- package/dist/core/recipes/loader.js +103 -0
- package/dist/core/recipes/runner.js +345 -0
- package/dist/core/recipes/schema.js +587 -0
- package/dist/core/release-notes/parser.js +241 -0
- package/dist/core/release-notes/state.js +116 -0
- package/dist/core/repl/ask.js +37 -37
- package/dist/core/repl/cancellation.js +26 -26
- package/dist/core/repl/cap-warning.js +4 -4
- package/dist/core/repl/clipboard-read.js +11 -11
- package/dist/core/repl/dispatch-fsm.js +12 -12
- package/dist/core/repl/history-search.js +15 -15
- package/dist/core/repl/history.js +28 -18
- package/dist/core/repl/kill-ring.js +5 -5
- package/dist/core/repl/model-pricing.js +135 -0
- package/dist/core/repl/privacy-banner.js +22 -22
- package/dist/core/repl/session.js +2157 -214
- package/dist/core/repl/slash-commands.js +533 -40
- package/dist/core/repl/store/index.js +1 -1
- package/dist/core/repl/store/jsonl-log.js +22 -22
- package/dist/core/repl/store/lockfile.js +10 -10
- package/dist/core/repl/store/session-store.js +136 -107
- package/dist/core/repl/store/types.js +15 -15
- package/dist/core/repl/store/uuid-v7.js +12 -12
- package/dist/core/repl/workspace-context.js +43 -21
- package/dist/core/repo-map/build.js +125 -0
- package/dist/core/repo-map/cache.js +185 -0
- package/dist/core/repo-map/extractor.js +254 -0
- package/dist/core/repo-map/formatter.js +145 -0
- package/dist/core/repo-map/page-rank.js +105 -0
- package/dist/core/repo-map/scanner.js +211 -0
- package/dist/core/retry-budget/budget.js +284 -0
- package/dist/core/retry-budget/index.js +5 -0
- package/dist/core/retry-budget/retry-cap.js +74 -0
- package/dist/core/routing/lead-worker.js +43 -0
- package/dist/core/routing/pre-flight-estimator.js +108 -0
- package/dist/core/runs/run-tree.js +103 -0
- package/dist/core/security/injection-scanner.js +367 -0
- package/dist/core/security/output-filter.js +418 -0
- package/dist/core/session/env-file.js +105 -0
- package/dist/core/session/section-budgets.js +140 -0
- package/dist/core/session.js +92 -0
- package/dist/core/settings.js +286 -5
- package/dist/core/share/formatter.js +271 -0
- package/dist/core/share/redactor.js +221 -0
- package/dist/core/share/uploader.js +267 -0
- package/dist/core/skills/defaults.js +457 -0
- package/dist/core/skills/loader.js +22 -22
- package/dist/core/skills/sources.js +27 -27
- package/dist/core/smoke/headless-driver.js +174 -0
- package/dist/core/smoke/orchestrator.js +194 -0
- package/dist/core/smoke/runner.js +238 -0
- package/dist/core/smoke/scenario-parser.js +316 -0
- package/dist/core/statusline.js +99 -0
- package/dist/core/subagents/dispatcher-real.js +600 -0
- package/dist/core/subagents/dispatcher.js +132 -43
- package/dist/core/subagents/index.js +19 -6
- package/dist/core/subagents/isolation-matrix.js +213 -0
- package/dist/core/subagents/spawn.js +19 -4
- package/dist/core/telemetry/emitter.js +229 -0
- package/dist/core/telemetry/queue.js +251 -0
- package/dist/core/theme/context.js +91 -0
- package/dist/core/theme/presets.js +228 -0
- package/dist/core/theme/state.js +181 -0
- package/dist/core/todos/invariant.js +10 -0
- package/dist/core/todos/state.js +177 -0
- package/dist/core/tool-schema/compressor.js +89 -0
- package/dist/core/transport/version-interceptor.js +166 -0
- package/dist/core/trust.js +2 -2
- package/dist/core/tui/thinking-block.js +64 -0
- package/dist/core/vim/keymap.js +288 -0
- package/dist/core/vim/state.js +92 -0
- package/dist/core/watch-markers/marker-watcher.js +133 -0
- package/dist/core/worktree-manager/cleanup.js +123 -0
- package/dist/core/worktree-manager/manager.js +303 -0
- package/dist/index.js +28 -0
- package/dist/runtime/bootstrap.js +190 -0
- package/dist/runtime/cli.js +4162 -488
- package/dist/runtime/commands/agents.js +30 -30
- package/dist/runtime/commands/budget.js +5 -5
- package/dist/runtime/commands/cancel.js +231 -0
- package/dist/runtime/commands/chain.js +489 -0
- package/dist/runtime/commands/codegraph-status.js +227 -0
- package/dist/runtime/commands/compact.js +297 -0
- package/dist/runtime/commands/config.js +32 -32
- package/dist/runtime/commands/cost.js +199 -0
- package/dist/runtime/commands/delegate.js +244 -13
- package/dist/runtime/commands/dispatch.js +126 -0
- package/dist/runtime/commands/doctor.js +579 -0
- package/dist/runtime/commands/feedback.js +184 -0
- package/dist/runtime/commands/hooks.js +184 -0
- package/dist/runtime/commands/init.js +254 -0
- package/dist/runtime/commands/lsp.js +368 -0
- package/dist/runtime/commands/mcp.js +879 -0
- package/dist/runtime/commands/memory.js +582 -0
- package/dist/runtime/commands/model.js +237 -0
- package/dist/runtime/commands/onboarding.js +275 -0
- package/dist/runtime/commands/patch.js +128 -0
- package/dist/runtime/commands/permissions.js +112 -0
- package/dist/runtime/commands/plan.js +143 -0
- package/dist/runtime/commands/prd-check.js +285 -0
- package/dist/runtime/commands/privacy.js +17 -17
- package/dist/runtime/commands/recipe.js +325 -0
- package/dist/runtime/commands/redo-blob-store.js +92 -0
- package/dist/runtime/commands/redo.js +361 -0
- package/dist/runtime/commands/release-notes.js +229 -0
- package/dist/runtime/commands/repo-map.js +95 -0
- package/dist/runtime/commands/report.js +299 -0
- package/dist/runtime/commands/resume.js +118 -0
- package/dist/runtime/commands/review-consensus.js +68 -53
- package/dist/runtime/commands/rewind.js +333 -0
- package/dist/runtime/commands/roster.js +14 -14
- package/dist/runtime/commands/sessions.js +163 -0
- package/dist/runtime/commands/share.js +316 -0
- package/dist/runtime/commands/skills.js +31 -31
- package/dist/runtime/commands/status.js +186 -0
- package/dist/runtime/commands/stickers.js +82 -0
- package/dist/runtime/commands/style.js +194 -0
- package/dist/runtime/commands/theme.js +196 -0
- package/dist/runtime/commands/undo.js +54 -22
- package/dist/runtime/commands/update.js +289 -0
- package/dist/runtime/commands/vim.js +140 -0
- package/dist/runtime/commands/worktree.js +177 -0
- package/dist/runtime/commands/worktrees.js +155 -0
- package/dist/runtime/headless-repl.js +195 -0
- package/dist/runtime/headless.js +543 -0
- package/dist/runtime/load-hooks-or-exit.js +71 -0
- package/dist/runtime/plan-decompose.js +531 -0
- package/dist/runtime/update-check.js +28 -28
- package/dist/runtime/version.js +65 -0
- package/dist/skills/bundled/batch.js +617 -0
- package/dist/skills/bundled/index.js +45 -0
- package/dist/skills/bundled/loop.js +358 -0
- package/dist/skills/bundled/remember.js +383 -0
- package/dist/skills/bundled/simplify.js +289 -0
- package/dist/skills/bundled/skillify.js +373 -0
- package/dist/skills/bundled/stuck.js +558 -0
- package/dist/skills/bundled/verify.js +439 -0
- package/dist/testing/vcr.js +486 -0
- package/dist/tools/agent-tool.js +229 -0
- package/dist/tools/apply-patch.js +556 -0
- package/dist/tools/ask-user-question.js +222 -0
- package/dist/tools/ask-user.js +115 -0
- package/dist/tools/bash.js +623 -45
- package/dist/tools/brief.js +224 -0
- package/dist/tools/enter-worktree.js +250 -0
- package/dist/tools/exit-worktree.js +147 -0
- package/dist/tools/file-tools.js +161 -44
- package/dist/tools/lsp-tools.js +189 -0
- package/dist/tools/mcp-tool.js +260 -0
- package/dist/tools/multi-edit.js +361 -0
- package/dist/tools/powershell.js +268 -0
- package/dist/tools/registry.js +85 -0
- package/dist/tools/skill-tool.js +96 -0
- package/dist/tools/sleep.js +99 -0
- package/dist/tools/synthetic-output.js +133 -0
- package/dist/tools/tasks.js +208 -0
- package/dist/tools/todo-write.js +184 -0
- package/dist/tools/verify-plan-execution.js +295 -0
- package/dist/tools/web-fetch-injection-scanner.js +207 -0
- package/dist/tools/web-fetch.js +195 -10
- package/dist/tools/web-search.js +458 -0
- package/dist/tui/agent-progress-card.js +111 -0
- package/dist/tui/agent-tree.js +11 -1
- package/dist/tui/ask-modal.js +14 -14
- package/dist/tui/ask-user-question-prompt.js +203 -0
- package/dist/tui/compact-banner.js +81 -0
- package/dist/tui/conversation-pane.js +85 -11
- package/dist/tui/cost-table.js +111 -0
- package/dist/tui/device-flow.js +2 -2
- package/dist/tui/doctor-table.js +46 -0
- package/dist/tui/feedback-prompt.js +156 -0
- package/dist/tui/input-box.js +247 -32
- package/dist/tui/login-picker.js +3 -3
- package/dist/tui/markdown-render.js +6 -6
- package/dist/tui/onboarding-wizard.js +240 -0
- package/dist/tui/permissions-picker.js +86 -0
- package/dist/tui/render.js +35 -0
- package/dist/tui/repl-render.js +332 -54
- package/dist/tui/repl-splash-art.js +16 -16
- package/dist/tui/repl-splash-mascot.js +48 -24
- package/dist/tui/repl-splash.js +22 -22
- package/dist/tui/repl.js +124 -44
- package/dist/tui/slash-palette.js +6 -6
- package/dist/tui/splash.js +2 -2
- package/dist/tui/status-bar.js +109 -31
- package/dist/tui/status-table.js +7 -0
- package/dist/tui/stickers-art.js +136 -0
- package/dist/tui/style-table.js +28 -0
- package/dist/tui/theme-table.js +29 -0
- package/dist/tui/thinking-spinner.js +123 -0
- package/dist/tui/tool-stream-pane.js +53 -4
- package/dist/tui/update-banner.js +27 -2
- package/dist/tui/vim-input.js +267 -0
- package/dist/tui/welcome-banner.js +107 -0
- package/dist/tui/welcome-data.js +293 -0
- package/dist/tui/workspace-context.js +2 -2
- package/docs/examples/codegraph.mcp.json +10 -0
- package/package.json +23 -6
- package/test/scenarios/codegen-create-file.scenario.txt +13 -0
- package/test/scenarios/compact-force.scenario.txt +11 -0
- package/test/scenarios/identity.scenario.txt +11 -0
- package/test/scenarios/persona-handoff.scenario.txt +11 -0
- package/test/scenarios/walkback.scenario.txt +12 -0
- package/dist/core/engine/compaction-hook.js +0 -154
|
@@ -0,0 +1,284 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* — Per-command tool retry budget (the upstream tool parity).
|
|
3
|
+
*
|
|
4
|
+
* the upstream tool limits the number of times the model may retry the SAME
|
|
5
|
+
* tool with the SAME arguments inside a single operator-input cycle.
|
|
6
|
+
* Once the cap is hit, the dispatcher hard-refuses and surfaces a
|
|
7
|
+
* sentinel string telling the model that this exact call has exhausted
|
|
8
|
+
* its retry budget. The model is expected (via system-prompt rule) to
|
|
9
|
+
* either change approach or ask the operator for guidance instead of
|
|
10
|
+
* looping forever on a transient failure.
|
|
11
|
+
*
|
|
12
|
+
* Why per-cycle, not per-session: a retry budget that persists across
|
|
13
|
+
* operator turns would surprise the operator. After the operator says
|
|
14
|
+
* "try again" the model rightly retries; the budget must reset when a
|
|
15
|
+
* fresh brief arrives. The simplest reset boundary is the executor
|
|
16
|
+
* lifetime — `buildExecutor` is called once per `runEngineLoop` and
|
|
17
|
+
* the loop drives exactly one operator-input cycle. Constructing the
|
|
18
|
+
* budget inside `buildExecutor` therefore gives us per-cycle scoping
|
|
19
|
+
* "for free" via closure lifetime; no external clear() call is needed
|
|
20
|
+
* from production callsites. The exported `clear()` exists so tests
|
|
21
|
+
* and a future hook surface (PreToolUse) can introspect the state.
|
|
22
|
+
*
|
|
23
|
+
* Hash design: same tool + same canonical args = same bucket. We
|
|
24
|
+
* canonicalise the args record by sorting object keys (stable across
|
|
25
|
+
* model output ordering) and then sha256 the JSON. The model emits
|
|
26
|
+
* `arguments` as a raw JSON string; we parse, canonicalise, hash. If
|
|
27
|
+
* parse fails we hash the raw string verbatim — that way an
|
|
28
|
+
* unparseable repeat still counts toward the cap (otherwise the model
|
|
29
|
+
* could loop on syntactic noise variants forever).
|
|
30
|
+
*
|
|
31
|
+
* Env overrides:
|
|
32
|
+
* PUGI_RETRY_BUDGET_<TOOLNAME>=<N> — override a single tool's cap.
|
|
33
|
+
* Toolname matches DEFAULT_CAPS
|
|
34
|
+
* keys verbatim, uppercased
|
|
35
|
+
* (PUGI_RETRY_BUDGET_BASH=8).
|
|
36
|
+
* PUGI_RETRY_BUDGET_DEFAULT=<N> — override the fallback cap for
|
|
37
|
+
* any tool not in DEFAULT_CAPS.
|
|
38
|
+
* PUGI_RETRY_BUDGET_DISABLED=1 — warn-only mode. `shouldAllow`
|
|
39
|
+
* still records but always
|
|
40
|
+
* returns `allowed: true`. The
|
|
41
|
+
* count is preserved so
|
|
42
|
+
* diagnostics can still surface
|
|
43
|
+
* the pattern.
|
|
44
|
+
*/
|
|
45
|
+
import { createHash } from 'node:crypto';
|
|
46
|
+
/**
|
|
47
|
+
* Default per-tool retry caps. Tuned per the upstream behavior :
|
|
48
|
+
*
|
|
49
|
+
* bash — 5 (most volatile; transient flakes common)
|
|
50
|
+
* edit — 3 (deterministic; repeat = real bug)
|
|
51
|
+
* write — 3 (same)
|
|
52
|
+
* read — 10 (cheap; legitimate re-reads after edits)
|
|
53
|
+
* search/grep/glob — 10 (cheap; exploration loop)
|
|
54
|
+
* web_fetch — 5 (transient network; not infinite)
|
|
55
|
+
* default — 5 (any tool not in the table)
|
|
56
|
+
*
|
|
57
|
+
* Operators override per-tool via `PUGI_RETRY_BUDGET_<NAME>` env vars.
|
|
58
|
+
* Caps are bounded `[1, 1000]` after override to defend against typo
|
|
59
|
+
* runaway (e.g. `PUGI_RETRY_BUDGET_BASH=5000000`).
|
|
60
|
+
*/
|
|
61
|
+
export const DEFAULT_CAPS = Object.freeze({
|
|
62
|
+
bash: 5,
|
|
63
|
+
edit: 3,
|
|
64
|
+
write: 3,
|
|
65
|
+
read: 10,
|
|
66
|
+
search: 10,
|
|
67
|
+
grep: 10,
|
|
68
|
+
glob: 10,
|
|
69
|
+
web_fetch: 5,
|
|
70
|
+
default: 5,
|
|
71
|
+
});
|
|
72
|
+
/**
|
|
73
|
+
* Lower / upper bound for any resolved cap. Defends against:
|
|
74
|
+
* - PUGI_RETRY_BUDGET_BASH=0 -> first call instantly denied
|
|
75
|
+
* - PUGI_RETRY_BUDGET_BASH=99999 -> effectively unbounded loop
|
|
76
|
+
*/
|
|
77
|
+
export const MIN_CAP = 1;
|
|
78
|
+
export const MAX_CAP = 1000;
|
|
79
|
+
/**
|
|
80
|
+
* Per-cycle retry budget. One instance per `buildExecutor` call.
|
|
81
|
+
*
|
|
82
|
+
* Not thread-safe: the executor is single-threaded by construction
|
|
83
|
+
* (Node event loop + sequential await in dispatcher). If a future
|
|
84
|
+
* executor parallelises tool dispatch it must serialise the budget
|
|
85
|
+
* mutation explicitly.
|
|
86
|
+
*/
|
|
87
|
+
export class RetryBudget {
|
|
88
|
+
counts = new Map();
|
|
89
|
+
capCache = new Map();
|
|
90
|
+
env;
|
|
91
|
+
programmaticCaps;
|
|
92
|
+
constructor(options = {}) {
|
|
93
|
+
this.env = options.env ?? process.env;
|
|
94
|
+
this.programmaticCaps = options.caps ?? {};
|
|
95
|
+
}
|
|
96
|
+
/**
|
|
97
|
+
* Returns true when PUGI_RETRY_BUDGET_DISABLED=1. In disabled mode
|
|
98
|
+
* `shouldAllow` still records attempts but always allows the
|
|
99
|
+
* dispatch — useful for operators triaging a false-positive without
|
|
100
|
+
* a code change.
|
|
101
|
+
*/
|
|
102
|
+
isDisabled() {
|
|
103
|
+
return this.env.PUGI_RETRY_BUDGET_DISABLED === '1';
|
|
104
|
+
}
|
|
105
|
+
/**
|
|
106
|
+
* Record one dispatch attempt. Idempotent on the bucket key (tool
|
|
107
|
+
* + argHash). Call this BEFORE the dispatch (or after `shouldAllow`
|
|
108
|
+
* but before `dispatch()` resolves) so a thrown dispatch counts.
|
|
109
|
+
*/
|
|
110
|
+
recordAttempt(toolName, argHash) {
|
|
111
|
+
const key = `${toolName}::${argHash}`;
|
|
112
|
+
const next = (this.counts.get(key) ?? 0) + 1;
|
|
113
|
+
this.counts.set(key, next);
|
|
114
|
+
return next;
|
|
115
|
+
}
|
|
116
|
+
/**
|
|
117
|
+
* Returns the current count for (tool, argHash) WITHOUT mutating.
|
|
118
|
+
*/
|
|
119
|
+
peek(toolName, argHash) {
|
|
120
|
+
return this.counts.get(`${toolName}::${argHash}`) ?? 0;
|
|
121
|
+
}
|
|
122
|
+
/**
|
|
123
|
+
* Resolve the effective cap for a tool.
|
|
124
|
+
*
|
|
125
|
+
* Precedence:
|
|
126
|
+
* 1. PUGI_RETRY_BUDGET_<TOOL_UPPER>=<N> (env)
|
|
127
|
+
* 2. programmaticCaps[toolName] (constructor)
|
|
128
|
+
* 3. DEFAULT_CAPS[toolName] (this module)
|
|
129
|
+
* 4. PUGI_RETRY_BUDGET_DEFAULT=<N> (env fallback)
|
|
130
|
+
* 5. DEFAULT_CAPS.default (final fallback)
|
|
131
|
+
*
|
|
132
|
+
* Bounded by [MIN_CAP, MAX_CAP] post-resolution. Invalid (NaN, ≤0,
|
|
133
|
+
* non-integer) env values are ignored and the next layer wins.
|
|
134
|
+
*/
|
|
135
|
+
capFor(toolName) {
|
|
136
|
+
const cached = this.capCache.get(toolName);
|
|
137
|
+
if (cached !== undefined)
|
|
138
|
+
return cached;
|
|
139
|
+
const envKey = `PUGI_RETRY_BUDGET_${toolName.toUpperCase()}`;
|
|
140
|
+
const envCap = parseCapEnv(this.env[envKey]);
|
|
141
|
+
const programmaticCap = this.programmaticCaps[toolName];
|
|
142
|
+
const defaultCap = DEFAULT_CAPS[toolName];
|
|
143
|
+
const fallbackEnvCap = parseCapEnv(this.env.PUGI_RETRY_BUDGET_DEFAULT);
|
|
144
|
+
// DEFAULT_CAPS.default is hard-coded above; cast keeps the type-
|
|
145
|
+
// narrower happy without leaking `| undefined` through the index
|
|
146
|
+
// access (tsc cannot prove the literal key exists).
|
|
147
|
+
const finalFallback = DEFAULT_CAPS.default;
|
|
148
|
+
let resolved;
|
|
149
|
+
if (envCap !== undefined) {
|
|
150
|
+
resolved = envCap;
|
|
151
|
+
}
|
|
152
|
+
else if (programmaticCap !== undefined) {
|
|
153
|
+
resolved = programmaticCap;
|
|
154
|
+
}
|
|
155
|
+
else if (defaultCap !== undefined) {
|
|
156
|
+
resolved = defaultCap;
|
|
157
|
+
}
|
|
158
|
+
else {
|
|
159
|
+
resolved = fallbackEnvCap ?? finalFallback;
|
|
160
|
+
}
|
|
161
|
+
const bounded = Math.min(MAX_CAP, Math.max(MIN_CAP, resolved));
|
|
162
|
+
this.capCache.set(toolName, bounded);
|
|
163
|
+
return bounded;
|
|
164
|
+
}
|
|
165
|
+
/**
|
|
166
|
+
* Should this dispatch be allowed? Caller passes the current count
|
|
167
|
+
* BEFORE recording — i.e. shouldAllow returns true when count < cap,
|
|
168
|
+
* then recordAttempt fires, bringing count up to cap. The next
|
|
169
|
+
* identical call sees count === cap and is refused.
|
|
170
|
+
*
|
|
171
|
+
* In disabled mode `allowed` is forced to true; `count` and `cap`
|
|
172
|
+
* still reflect reality so logs / diagnostics can spot the pattern.
|
|
173
|
+
*/
|
|
174
|
+
shouldAllow(toolName, argHash) {
|
|
175
|
+
const cap = this.capFor(toolName);
|
|
176
|
+
const count = this.peek(toolName, argHash);
|
|
177
|
+
const disabled = this.isDisabled();
|
|
178
|
+
const allowed = disabled ? true : count < cap;
|
|
179
|
+
return { allowed, count, cap, argHash, disabled };
|
|
180
|
+
}
|
|
181
|
+
/** Reset all state. Used between operator-input cycles when the
|
|
182
|
+
* budget instance is reused (most callers throw the instance away
|
|
183
|
+
* per cycle, so clear() is mostly for tests and hook surfaces). */
|
|
184
|
+
clear() {
|
|
185
|
+
this.counts.clear();
|
|
186
|
+
this.capCache.clear();
|
|
187
|
+
}
|
|
188
|
+
/**
|
|
189
|
+
* Snapshot the current state for diagnostics. Returns a plain
|
|
190
|
+
* object so it round-trips through JSON.stringify cleanly.
|
|
191
|
+
*/
|
|
192
|
+
snapshot() {
|
|
193
|
+
const out = [];
|
|
194
|
+
for (const [key, count] of this.counts) {
|
|
195
|
+
const sep = key.indexOf('::');
|
|
196
|
+
if (sep < 0)
|
|
197
|
+
continue;
|
|
198
|
+
out.push({ tool: key.slice(0, sep), argHash: key.slice(sep + 2), count });
|
|
199
|
+
}
|
|
200
|
+
return out;
|
|
201
|
+
}
|
|
202
|
+
}
|
|
203
|
+
/**
|
|
204
|
+
* Hash the model's tool-call arguments into a stable key. Same
|
|
205
|
+
* canonical args = same hash regardless of JSON whitespace / key
|
|
206
|
+
* order. Unparseable JSON is hashed verbatim so the budget still
|
|
207
|
+
* catches syntactically degenerate retry loops.
|
|
208
|
+
*/
|
|
209
|
+
export function hashArgs(argsRaw) {
|
|
210
|
+
const canonical = canonicalise(argsRaw);
|
|
211
|
+
return createHash('sha256').update(canonical).digest('hex');
|
|
212
|
+
}
|
|
213
|
+
/**
|
|
214
|
+
* Canonicalise a raw JSON arg string. Object keys are sorted
|
|
215
|
+
* recursively. Arrays preserve order (semantic). Primitives untouched.
|
|
216
|
+
* On parse failure, returns the original string prefixed with `raw:`
|
|
217
|
+
* so a malformed-args repeat still hashes to the same bucket.
|
|
218
|
+
*/
|
|
219
|
+
function canonicalise(argsRaw) {
|
|
220
|
+
try {
|
|
221
|
+
const parsed = JSON.parse(argsRaw);
|
|
222
|
+
return JSON.stringify(sortKeys(parsed));
|
|
223
|
+
}
|
|
224
|
+
catch {
|
|
225
|
+
return `raw:${argsRaw}`;
|
|
226
|
+
}
|
|
227
|
+
}
|
|
228
|
+
function sortKeys(value) {
|
|
229
|
+
if (value === null || typeof value !== 'object')
|
|
230
|
+
return value;
|
|
231
|
+
if (Array.isArray(value))
|
|
232
|
+
return value.map(sortKeys);
|
|
233
|
+
const obj = value;
|
|
234
|
+
const sorted = {};
|
|
235
|
+
for (const k of Object.keys(obj).sort()) {
|
|
236
|
+
sorted[k] = sortKeys(obj[k]);
|
|
237
|
+
}
|
|
238
|
+
return sorted;
|
|
239
|
+
}
|
|
240
|
+
/**
|
|
241
|
+
* Parse and bound a `PUGI_RETRY_BUDGET_*` env var. Returns `undefined`
|
|
242
|
+
* for any non-positive-integer string so the resolver can fall
|
|
243
|
+
* through to the next precedence layer. Bounded by [MIN_CAP, MAX_CAP]
|
|
244
|
+
* is NOT applied here — `capFor` clamps after the final layer wins,
|
|
245
|
+
* matching the "operator typo defends against runaway" requirement
|
|
246
|
+
* without silently swallowing a meaningful low value (e.g.
|
|
247
|
+
* `PUGI_RETRY_BUDGET_BASH=1` should clamp to MIN_CAP=1, which it
|
|
248
|
+
* does naturally since 1 >= MIN_CAP).
|
|
249
|
+
*/
|
|
250
|
+
function parseCapEnv(raw) {
|
|
251
|
+
if (raw === undefined || raw === '')
|
|
252
|
+
return undefined;
|
|
253
|
+
const n = Number(raw);
|
|
254
|
+
if (!Number.isInteger(n) || n <= 0)
|
|
255
|
+
return undefined;
|
|
256
|
+
return n;
|
|
257
|
+
}
|
|
258
|
+
/**
|
|
259
|
+
* Sentinel emitted to the model when the budget is exhausted. The
|
|
260
|
+
* format is stable so the engine adapter, spec layer, and operator
|
|
261
|
+
* dashboards can pattern-match on it.
|
|
262
|
+
*/
|
|
263
|
+
export function retryBudgetExhaustedSentinel(toolName, cap) {
|
|
264
|
+
return `RETRY_BUDGET_EXHAUSTED: ${toolName} exceeded ${cap} attempts with these args. Operator must intervene.`;
|
|
265
|
+
}
|
|
266
|
+
/**
|
|
267
|
+
* Typed error thrown by the tool-bridge when the cap is hit. Carries
|
|
268
|
+
* the sentinel string so the engine loop can pattern-match without
|
|
269
|
+
* re-parsing. `instanceof RetryBudgetExhausted` is the canonical
|
|
270
|
+
* downstream test.
|
|
271
|
+
*/
|
|
272
|
+
export class RetryBudgetExhausted extends Error {
|
|
273
|
+
toolName;
|
|
274
|
+
cap;
|
|
275
|
+
argHash;
|
|
276
|
+
constructor(toolName, cap, argHash) {
|
|
277
|
+
super(retryBudgetExhaustedSentinel(toolName, cap));
|
|
278
|
+
this.name = 'RetryBudgetExhausted';
|
|
279
|
+
this.toolName = toolName;
|
|
280
|
+
this.cap = cap;
|
|
281
|
+
this.argHash = argHash;
|
|
282
|
+
}
|
|
283
|
+
}
|
|
284
|
+
//# sourceMappingURL=budget.js.map
|
|
@@ -0,0 +1,74 @@
|
|
|
1
|
+
const DEFAULT_CAP = 3;
|
|
2
|
+
const DEFAULT_RESET_AFTER_MS = 300_000;
|
|
3
|
+
export function createRetryBudget(options = {}) {
|
|
4
|
+
const cap = normalizePositiveInteger(options.cap, DEFAULT_CAP);
|
|
5
|
+
const resetAfterMs = normalizeNonNegativeInteger(options.resetAfterMs, DEFAULT_RESET_AFTER_MS);
|
|
6
|
+
const states = new Map();
|
|
7
|
+
function clone(state) {
|
|
8
|
+
return { ...state };
|
|
9
|
+
}
|
|
10
|
+
function getFreshState(operationKey, now = Date.now()) {
|
|
11
|
+
const state = states.get(operationKey);
|
|
12
|
+
if (!state)
|
|
13
|
+
return null;
|
|
14
|
+
if (now - state.lastAttemptAt >= resetAfterMs) {
|
|
15
|
+
states.delete(operationKey);
|
|
16
|
+
return null;
|
|
17
|
+
}
|
|
18
|
+
return state;
|
|
19
|
+
}
|
|
20
|
+
return {
|
|
21
|
+
record(operationKey) {
|
|
22
|
+
const now = Date.now();
|
|
23
|
+
const existing = getFreshState(operationKey, now);
|
|
24
|
+
const attempts = (existing?.attempts ?? 0) + 1;
|
|
25
|
+
const next = {
|
|
26
|
+
operationKey,
|
|
27
|
+
attempts,
|
|
28
|
+
firstAttemptAt: existing?.firstAttemptAt ?? now,
|
|
29
|
+
lastAttemptAt: now,
|
|
30
|
+
exhausted: attempts >= cap,
|
|
31
|
+
};
|
|
32
|
+
states.set(operationKey, next);
|
|
33
|
+
return clone(next);
|
|
34
|
+
},
|
|
35
|
+
reset(operationKey) {
|
|
36
|
+
states.delete(operationKey);
|
|
37
|
+
},
|
|
38
|
+
isExhausted(operationKey) {
|
|
39
|
+
return getFreshState(operationKey)?.exhausted ?? false;
|
|
40
|
+
},
|
|
41
|
+
getState(operationKey) {
|
|
42
|
+
const state = getFreshState(operationKey);
|
|
43
|
+
return state ? clone(state) : null;
|
|
44
|
+
},
|
|
45
|
+
};
|
|
46
|
+
}
|
|
47
|
+
export function validatePromptWordCount(text, opts) {
|
|
48
|
+
const words = countWords(text);
|
|
49
|
+
const chars = text.length;
|
|
50
|
+
if (opts.min !== undefined && words < opts.min) {
|
|
51
|
+
return { valid: false, words, chars, reason: 'too-short' };
|
|
52
|
+
}
|
|
53
|
+
if (opts.max !== undefined && words > opts.max) {
|
|
54
|
+
return { valid: false, words, chars, reason: 'too-long' };
|
|
55
|
+
}
|
|
56
|
+
return { valid: true, words, chars };
|
|
57
|
+
}
|
|
58
|
+
function countWords(text) {
|
|
59
|
+
const trimmed = text.trim();
|
|
60
|
+
if (trimmed.length === 0)
|
|
61
|
+
return 0;
|
|
62
|
+
return trimmed.split(/\s+/).length;
|
|
63
|
+
}
|
|
64
|
+
function normalizePositiveInteger(value, fallback) {
|
|
65
|
+
if (value === undefined || !Number.isFinite(value))
|
|
66
|
+
return fallback;
|
|
67
|
+
return Math.max(1, Math.floor(value));
|
|
68
|
+
}
|
|
69
|
+
function normalizeNonNegativeInteger(value, fallback) {
|
|
70
|
+
if (value === undefined || !Number.isFinite(value))
|
|
71
|
+
return fallback;
|
|
72
|
+
return Math.max(0, Math.floor(value));
|
|
73
|
+
}
|
|
74
|
+
//# sourceMappingURL=retry-cap.js.map
|
|
@@ -0,0 +1,43 @@
|
|
|
1
|
+
export function assignRoles(opts) {
|
|
2
|
+
let hasAssignedWriteLead = false;
|
|
3
|
+
return opts.steps.map((step) => {
|
|
4
|
+
const role = assignRole(step, hasAssignedWriteLead);
|
|
5
|
+
if (step.intent === 'write' && role.role === 'lead') {
|
|
6
|
+
hasAssignedWriteLead = true;
|
|
7
|
+
}
|
|
8
|
+
return {
|
|
9
|
+
step: step.id,
|
|
10
|
+
role: role.role,
|
|
11
|
+
model: role.role === 'lead' ? opts.leadModel : opts.workerModel,
|
|
12
|
+
reason: role.reason,
|
|
13
|
+
};
|
|
14
|
+
});
|
|
15
|
+
}
|
|
16
|
+
function assignRole(step, hasAssignedWriteLead) {
|
|
17
|
+
if (step.intent === 'plan') {
|
|
18
|
+
return { role: 'lead', reason: 'planning step requires lead orchestration' };
|
|
19
|
+
}
|
|
20
|
+
if (hasLeadHeuristic(step.id)) {
|
|
21
|
+
return { role: 'lead', reason: 'step id indicates planning or orchestration' };
|
|
22
|
+
}
|
|
23
|
+
if (step.intent === 'write' && !hasAssignedWriteLead) {
|
|
24
|
+
return { role: 'lead', reason: 'first write step needs lead architecture' };
|
|
25
|
+
}
|
|
26
|
+
if (step.intent === 'write') {
|
|
27
|
+
return { role: 'worker', reason: 'subsequent write step is bulk execution' };
|
|
28
|
+
}
|
|
29
|
+
if (step.intent === 'read' || step.intent === 'verify') {
|
|
30
|
+
return { role: 'worker', reason: 'read and verify steps are worker execution' };
|
|
31
|
+
}
|
|
32
|
+
if (step.intent === 'explain') {
|
|
33
|
+
return { role: 'worker', reason: 'explanation step can run on worker model' };
|
|
34
|
+
}
|
|
35
|
+
return { role: 'lead', reason: 'unknown intent defaults to lead defensively' };
|
|
36
|
+
}
|
|
37
|
+
function hasLeadHeuristic(stepId) {
|
|
38
|
+
const normalized = stepId.toLowerCase();
|
|
39
|
+
return (normalized.includes('plan') ||
|
|
40
|
+
normalized.includes('design') ||
|
|
41
|
+
normalized.includes('orchestrate'));
|
|
42
|
+
}
|
|
43
|
+
//# sourceMappingURL=lead-worker.js.map
|
|
@@ -0,0 +1,108 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Pre-flight token estimator — Nuekkis tokenEstimation port,
|
|
3
|
+
* adapted for Anvil's 3-tier routing.
|
|
4
|
+
*
|
|
5
|
+
* The auto-compact gate counts tokens AFTER a turn lands. This module
|
|
6
|
+
* runs BEFORE the request leaves the CLI so the router can pick the
|
|
7
|
+
* cheapest pool that still fits. Three pools exist:
|
|
8
|
+
*
|
|
9
|
+
* cheap → DeepSeek V4-Pro / Cerebras Qwen3-Coder (128k-256k window)
|
|
10
|
+
* mid → Anthropic Sonnet 4.6 / GPT-5 (200k window, 2-3× cheap-pool cost)
|
|
11
|
+
* long → Kimi K2.6 / Gemini 2.5 Pro (1M window, 5-10× cheap-pool cost)
|
|
12
|
+
*
|
|
13
|
+
* The estimator is intentionally synchronous, pure, and free of I/O so
|
|
14
|
+
* the call site can run it inside a render loop without yielding to
|
|
15
|
+
* the event loop. The token approximation reuses the existing
|
|
16
|
+
* char-per-token heuristic from `core/compact/token-counter.ts` (4 chars
|
|
17
|
+
* ≈ 1 token, biased high). No tiktoken dependency added.
|
|
18
|
+
*/
|
|
19
|
+
import { estimateTokens } from '../compact/token-counter.js';
|
|
20
|
+
/**
|
|
21
|
+
* Default tier ceilings tuned для Anvil 2026-06 routing matrix.
|
|
22
|
+
* cheap-pool models (DeepSeek/Cerebras) hard-cap at 100k effective input.
|
|
23
|
+
* mid-pool (Sonnet/GPT-5) safe through 180k.
|
|
24
|
+
* long-pool (Kimi/Gemini-Pro) accepts к 900k.
|
|
25
|
+
*
|
|
26
|
+
* The numbers stay below the nominal context window к leave room for
|
|
27
|
+
* output tokens, тек streaming overhead, and tokenizer skew.
|
|
28
|
+
*/
|
|
29
|
+
const DEFAULT_CHEAP_MAX = 100_000;
|
|
30
|
+
const DEFAULT_MID_MAX = 180_000;
|
|
31
|
+
const DEFAULT_LONG_MAX = 900_000;
|
|
32
|
+
const DEFAULT_OUTPUT_BUFFER = 4_096;
|
|
33
|
+
export function estimatePreFlight(input, options = {}) {
|
|
34
|
+
const cheapMax = options.cheapTierMaxInput ?? DEFAULT_CHEAP_MAX;
|
|
35
|
+
const midMax = options.midTierMaxInput ?? DEFAULT_MID_MAX;
|
|
36
|
+
const longMax = options.longTierMaxInput ?? DEFAULT_LONG_MAX;
|
|
37
|
+
const outputBuffer = input.expectedOutputTokens
|
|
38
|
+
?? options.outputBuffer
|
|
39
|
+
?? DEFAULT_OUTPUT_BUFFER;
|
|
40
|
+
if (cheapMax <= 0 || midMax <= 0 || longMax <= 0) {
|
|
41
|
+
throw new RangeError('tier max values must be positive');
|
|
42
|
+
}
|
|
43
|
+
if (cheapMax > midMax || midMax > longMax) {
|
|
44
|
+
throw new RangeError('tier ceilings must be monotonic: cheap <= mid <= long');
|
|
45
|
+
}
|
|
46
|
+
if (outputBuffer < 0) {
|
|
47
|
+
throw new RangeError('outputBuffer must be >= 0');
|
|
48
|
+
}
|
|
49
|
+
const systemTokens = sumStrings(input.systemPrompt ? [input.systemPrompt] : []);
|
|
50
|
+
const dialogTokens = sumStrings(input.dialogHistory ?? []);
|
|
51
|
+
const ragTokens = sumStrings(input.ragContext ?? []);
|
|
52
|
+
const toolTokens = sumStrings(input.toolResults ?? []);
|
|
53
|
+
const userTokens = sumStrings(input.userMessage ? [input.userMessage] : []);
|
|
54
|
+
const inputTokens = systemTokens + dialogTokens + ragTokens + toolTokens + userTokens;
|
|
55
|
+
const totalTokens = inputTokens + outputBuffer;
|
|
56
|
+
const tier = pickTier(inputTokens, cheapMax, midMax);
|
|
57
|
+
const overLongTier = inputTokens > longMax;
|
|
58
|
+
return {
|
|
59
|
+
inputTokens,
|
|
60
|
+
outputBuffer,
|
|
61
|
+
totalTokens,
|
|
62
|
+
tier,
|
|
63
|
+
breakdown: {
|
|
64
|
+
systemPrompt: systemTokens,
|
|
65
|
+
dialogHistory: dialogTokens,
|
|
66
|
+
ragContext: ragTokens,
|
|
67
|
+
toolResults: toolTokens,
|
|
68
|
+
userMessage: userTokens,
|
|
69
|
+
},
|
|
70
|
+
overLongTier,
|
|
71
|
+
};
|
|
72
|
+
}
|
|
73
|
+
function sumStrings(parts) {
|
|
74
|
+
let total = 0;
|
|
75
|
+
for (const part of parts) {
|
|
76
|
+
total += estimateTokens(part);
|
|
77
|
+
}
|
|
78
|
+
return total;
|
|
79
|
+
}
|
|
80
|
+
function pickTier(inputTokens, cheapMax, midMax) {
|
|
81
|
+
if (inputTokens <= cheapMax)
|
|
82
|
+
return 'cheap';
|
|
83
|
+
if (inputTokens <= midMax)
|
|
84
|
+
return 'mid';
|
|
85
|
+
return 'long';
|
|
86
|
+
}
|
|
87
|
+
/**
|
|
88
|
+
* Human-readable explanation для CLI / TUI surfacing.
|
|
89
|
+
* The format is stable and may be parsed by the doctor command.
|
|
90
|
+
*/
|
|
91
|
+
export function explainEstimate(estimate) {
|
|
92
|
+
const lines = [];
|
|
93
|
+
lines.push(`Input tokens: ${estimate.inputTokens.toLocaleString('en-US')}`);
|
|
94
|
+
lines.push(`Output buffer: ${estimate.outputBuffer.toLocaleString('en-US')}`);
|
|
95
|
+
lines.push(`Total: ${estimate.totalTokens.toLocaleString('en-US')}`);
|
|
96
|
+
lines.push(`Routing tier: ${estimate.tier}`);
|
|
97
|
+
if (estimate.overLongTier) {
|
|
98
|
+
lines.push('WARNING: input exceeds long-tier ceiling — request will likely fail');
|
|
99
|
+
}
|
|
100
|
+
lines.push('Breakdown:');
|
|
101
|
+
lines.push(` system prompt: ${estimate.breakdown.systemPrompt.toLocaleString('en-US')}`);
|
|
102
|
+
lines.push(` dialog: ${estimate.breakdown.dialogHistory.toLocaleString('en-US')}`);
|
|
103
|
+
lines.push(` rag: ${estimate.breakdown.ragContext.toLocaleString('en-US')}`);
|
|
104
|
+
lines.push(` tool results: ${estimate.breakdown.toolResults.toLocaleString('en-US')}`);
|
|
105
|
+
lines.push(` user message: ${estimate.breakdown.userMessage.toLocaleString('en-US')}`);
|
|
106
|
+
return lines.join('\n');
|
|
107
|
+
}
|
|
108
|
+
//# sourceMappingURL=pre-flight-estimator.js.map
|
|
@@ -0,0 +1,103 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Canonical `.pugi/runs/<id>/` artifact tree .
|
|
3
|
+
*
|
|
4
|
+
* Karpathy hn-time-capsule pattern: every Pugi execution produces a
|
|
5
|
+
* deterministic directory structure where downstream tooling (eval
|
|
6
|
+
* harness, leaderboard, replay, debugging) can find
|
|
7
|
+
* artifacts at predictable paths.
|
|
8
|
+
*
|
|
9
|
+
* Layout per run:
|
|
10
|
+
* .pugi/runs/<id>/
|
|
11
|
+
* meta.json — metadata: id, startedAt, finishedAt?, command, tier
|
|
12
|
+
* stdout.log — captured stdout (the engine writes it directly)
|
|
13
|
+
* stderr.log — captured stderr
|
|
14
|
+
* events.jsonl — structured event stream (NDJSON)
|
|
15
|
+
* metrics.json — final metrics summary (written at run end)
|
|
16
|
+
* artifacts/ — арbitrary file outputs (plans, diffs, exports)
|
|
17
|
+
*
|
|
18
|
+
* The `<id>` is `<ISO-timestamp>-<short-rand>` so runs sort
|
|
19
|
+
* chronologically when listed by directory order.
|
|
20
|
+
*
|
|
21
|
+
* This module only handles the directory + metadata primitive. Actual
|
|
22
|
+
* stream writing (stdout.log, events.jsonl) is the engine's job —
|
|
23
|
+
* we return paths so the engine knows where к write.
|
|
24
|
+
*/
|
|
25
|
+
import { mkdir, writeFile, readFile, stat } from 'node:fs/promises';
|
|
26
|
+
import { randomBytes } from 'node:crypto';
|
|
27
|
+
import path from 'node:path';
|
|
28
|
+
export function generateRunId(now = new Date()) {
|
|
29
|
+
const iso = now.toISOString().replace(/[:.]/g, '-');
|
|
30
|
+
const rand = randomBytes(3).toString('hex');
|
|
31
|
+
return `${iso}-${rand}`;
|
|
32
|
+
}
|
|
33
|
+
export function resolveRunPaths(workspaceRoot, id) {
|
|
34
|
+
const root = path.join(workspaceRoot, '.pugi', 'runs', id);
|
|
35
|
+
return {
|
|
36
|
+
root,
|
|
37
|
+
meta: path.join(root, 'meta.json'),
|
|
38
|
+
stdout: path.join(root, 'stdout.log'),
|
|
39
|
+
stderr: path.join(root, 'stderr.log'),
|
|
40
|
+
events: path.join(root, 'events.jsonl'),
|
|
41
|
+
metrics: path.join(root, 'metrics.json'),
|
|
42
|
+
artifacts: path.join(root, 'artifacts'),
|
|
43
|
+
};
|
|
44
|
+
}
|
|
45
|
+
export async function createRun(options) {
|
|
46
|
+
if (!options.workspaceRoot) {
|
|
47
|
+
throw new TypeError('workspaceRoot is required');
|
|
48
|
+
}
|
|
49
|
+
const id = options.id ?? generateRunId();
|
|
50
|
+
if (!/^[A-Za-z0-9_.\-T:Z]+$/.test(id)) {
|
|
51
|
+
throw new RangeError(`invalid run id: ${id} (forbidden characters)`);
|
|
52
|
+
}
|
|
53
|
+
const paths = resolveRunPaths(options.workspaceRoot, id);
|
|
54
|
+
await mkdir(paths.artifacts, { recursive: true });
|
|
55
|
+
const meta = {
|
|
56
|
+
id,
|
|
57
|
+
startedAt: new Date().toISOString(),
|
|
58
|
+
};
|
|
59
|
+
if (options.command !== undefined)
|
|
60
|
+
meta.command = options.command;
|
|
61
|
+
if (options.tier !== undefined)
|
|
62
|
+
meta.tier = options.tier;
|
|
63
|
+
if (options.extra !== undefined)
|
|
64
|
+
meta.extra = options.extra;
|
|
65
|
+
await writeFile(paths.meta, JSON.stringify(meta, null, 2) + '\n');
|
|
66
|
+
return paths;
|
|
67
|
+
}
|
|
68
|
+
export async function readRunMetadata(paths) {
|
|
69
|
+
try {
|
|
70
|
+
const buf = await readFile(paths.meta, 'utf8');
|
|
71
|
+
return JSON.parse(buf);
|
|
72
|
+
}
|
|
73
|
+
catch {
|
|
74
|
+
return null;
|
|
75
|
+
}
|
|
76
|
+
}
|
|
77
|
+
export async function finalizeRun(paths, options = {}) {
|
|
78
|
+
const existing = await readRunMetadata(paths);
|
|
79
|
+
if (!existing) {
|
|
80
|
+
throw new Error(`run metadata missing at ${paths.meta}`);
|
|
81
|
+
}
|
|
82
|
+
const finished = {
|
|
83
|
+
...existing,
|
|
84
|
+
finishedAt: new Date().toISOString(),
|
|
85
|
+
};
|
|
86
|
+
if (options.exitCode !== undefined) {
|
|
87
|
+
finished.exitCode = options.exitCode;
|
|
88
|
+
}
|
|
89
|
+
await writeFile(paths.meta, JSON.stringify(finished, null, 2) + '\n');
|
|
90
|
+
if (options.metrics !== undefined) {
|
|
91
|
+
await writeFile(paths.metrics, JSON.stringify(options.metrics, null, 2) + '\n');
|
|
92
|
+
}
|
|
93
|
+
}
|
|
94
|
+
export async function runExists(paths) {
|
|
95
|
+
try {
|
|
96
|
+
const stats = await stat(paths.root);
|
|
97
|
+
return stats.isDirectory();
|
|
98
|
+
}
|
|
99
|
+
catch {
|
|
100
|
+
return false;
|
|
101
|
+
}
|
|
102
|
+
}
|
|
103
|
+
//# sourceMappingURL=run-tree.js.map
|