@pugi/cli 0.1.0-beta.10 → 0.1.0-beta.100
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 +132 -0
- package/LICENSE +1 -1
- package/README.md +53 -11
- 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/retro.js +210 -0
- 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 +3 -3
- 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/db.js +506 -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/codegraph/parser.js +71 -0
- package/dist/core/codegraph/types.js +34 -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 +13 -13
- 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 +72 -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 +333 -7
- package/dist/core/edits/format-detector.js +260 -0
- package/dist/core/edits/format-matrix.js +26 -0
- package/dist/core/edits/fuzzy-ladder.js +650 -0
- package/dist/core/edits/index.js +5 -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 +29 -29
- package/dist/core/engine/anvil-client.js +214 -26
- package/dist/core/engine/auto-compact.js +247 -0
- package/dist/core/engine/budgets.js +220 -0
- package/dist/core/engine/compact-llm-summarizer.js +124 -0
- package/dist/core/engine/context-prefix.js +155 -0
- package/dist/core/engine/index.js +1 -1
- package/dist/core/engine/intensity.js +163 -0
- package/dist/core/engine/intent.js +260 -0
- package/dist/core/engine/native-pugi.js +1559 -227
- package/dist/core/engine/prompts.js +187 -19
- package/dist/core/engine/strip-internal-fields.js +124 -0
- package/dist/core/engine/tool-bridge.js +1887 -59
- package/dist/core/engine/verification-patterns.js +195 -0
- 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 +46 -0
- package/dist/core/hooks/index.js +15 -0
- package/dist/core/hooks/registry.js +216 -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/hooks/worktree-events.js +158 -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 +551 -41
- package/dist/core/lsp/language-detect.js +66 -0
- package/dist/core/lsp/post-edit-diagnostics.js +171 -0
- package/dist/core/lsp/server-detect.js +173 -0
- package/dist/core/lsp/symbol-cache.js +162 -0
- package/dist/core/lsp/symbol-tools.js +664 -0
- package/dist/core/mcp/client.js +97 -28
- package/dist/core/mcp/http-server.js +553 -0
- package/dist/core/mcp/orchestrator-config.js +192 -0
- package/dist/core/mcp/orchestrator-tools.js +806 -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/notes/notes-paths.js +113 -0
- package/dist/core/notes/notes-recorder.js +140 -0
- package/dist/core/notes/notes-writer.js +53 -0
- package/dist/core/notes/renderers.js +0 -0
- package/dist/core/notes/slug.js +105 -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 +107 -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-gitignore.js +52 -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/engine-bridge.js +303 -0
- 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 +2690 -229
- package/dist/core/repl/slash-commands.js +540 -41
- 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/tool-route.js +382 -0
- 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/retro/git-collector.js +251 -0
- package/dist/core/retro/health-card.js +25 -0
- package/dist/core/retro/metrics.js +342 -0
- package/dist/core/retro/narrative.js +249 -0
- package/dist/core/retro/plane-collector.js +274 -0
- package/dist/core/retro/pr-issue-link.js +65 -0
- package/dist/core/retro/types.js +16 -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/sandboxing/adapter.js +29 -0
- package/dist/core/sandboxing/index.js +49 -0
- package/dist/core/sandboxing/none.js +19 -0
- package/dist/core/sandboxing/seatbelt.js +183 -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 +119 -0
- package/dist/core/settings.js +378 -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 +30 -30
- 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 +146 -52
- 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/include-parser.js +249 -0
- package/dist/core/worktree-manager/cleanup.js +123 -0
- package/dist/core/worktree-manager/manager.js +303 -0
- package/dist/index.js +36 -0
- package/dist/runtime/bootstrap.js +190 -0
- package/dist/runtime/cli.js +4345 -561
- package/dist/runtime/commands/agents.js +31 -31
- 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 +74 -40
- package/dist/runtime/commands/cost.js +199 -0
- package/dist/runtime/commands/delegate.js +27 -4
- 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 +187 -0
- package/dist/runtime/commands/index-cmd.js +353 -0
- package/dist/runtime/commands/init.js +254 -0
- package/dist/runtime/commands/lsp.js +200 -38
- package/dist/runtime/commands/mcp.js +935 -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 +12 -12
- 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/servers.js +236 -0
- 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 +8 -8
- package/dist/runtime/commands/worktrees.js +155 -0
- package/dist/runtime/deprecation-warning.js +69 -0
- package/dist/runtime/engine-exit-code.js +50 -0
- package/dist/runtime/headless-repl.js +195 -0
- package/dist/runtime/headless.js +548 -0
- package/dist/runtime/load-hooks-or-exit.js +71 -0
- package/dist/runtime/plan-decompose.js +22 -22
- package/dist/runtime/sigint-guard.js +272 -0
- package/dist/runtime/stream-renderer.js +195 -0
- package/dist/runtime/update-check.js +28 -28
- package/dist/runtime/version.js +65 -0
- package/dist/runtime/worktree-bootstrap.js +579 -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 +89 -28
- package/dist/tools/ask-user-question.js +337 -0
- package/dist/tools/ask-user.js +115 -0
- package/dist/tools/bash.js +624 -46
- package/dist/tools/brief.js +224 -0
- package/dist/tools/cron.js +433 -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/http-request.js +336 -0
- package/dist/tools/lsp-tools.js +377 -1
- 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 +120 -5
- package/dist/tools/server-tools.js +892 -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 +22 -1
- package/dist/tui/ask-modal.js +14 -14
- package/dist/tui/ask-user-question-chips.js +315 -0
- 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/multi-file-diff-approval.js +375 -0
- package/dist/tui/onboarding-wizard.js +240 -0
- package/dist/tui/permissions-picker.js +86 -0
- package/dist/tui/render.js +36 -1
- package/dist/tui/repl-render.js +239 -25
- 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 +125 -45
- 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/package.json +21 -5
- package/test/scenarios/codegen-create-file.scenario.txt +13 -0
- package/test/scenarios/compact-force.scenario.txt +12 -0
- package/test/scenarios/identity.scenario.txt +11 -0
- package/test/scenarios/persona-handoff.scenario.txt +12 -0
- package/test/scenarios/walkback.scenario.txt +12 -0
- package/dist/core/engine/compaction-hook.js +0 -154
|
@@ -0,0 +1,330 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Adaptive router — primitive #4 of the 9-layer RAG architecture gap
|
|
3
|
+
* analysis (docs/research/2026-05-31-9-layer-rag-architecture-.md).
|
|
4
|
+
*
|
|
5
|
+
* Picks the best source (RAG workspace / web fetch / code-intel / cache
|
|
6
|
+
* tier / agent persona / tool) for a given query by asking an LLM to
|
|
7
|
+
* rank candidates from a caller-supplied catalog. The router is pure
|
|
8
|
+
* logic — it never opens a network connection itself. The caller
|
|
9
|
+
* injects a `transport` callback that maps a prompt string to a model
|
|
10
|
+
* response string, which keeps Anvil-engine wiring (Hiroshi persona,
|
|
11
|
+
* auth, retry, telemetry) out of the primitive and makes tests
|
|
12
|
+
* deterministic.
|
|
13
|
+
*
|
|
14
|
+
* Failure model is fail-closed to a heuristic pick. Every malformed
|
|
15
|
+
* response, transport throw, schema violation, empty catalog, or
|
|
16
|
+
* unknown source id collapses to a deterministic ranking based on
|
|
17
|
+
* `kind` priority (cache > agent > everything-else, in catalog order).
|
|
18
|
+
* The caller inspects `usedFallback` plus `reason` and decides whether
|
|
19
|
+
* the degradation is acceptable for the active route. Routing is never
|
|
20
|
+
* load-bearing for correctness — the caller can always fall back to a
|
|
21
|
+
* monolithic default source.
|
|
22
|
+
*
|
|
23
|
+
* Composition with sibling primitives:
|
|
24
|
+
* - #3 query-decomposer feeds an ordered SubQuery list. The engine
|
|
25
|
+
* consumer will call `chooseRoute` once per SubQuery to pick the
|
|
26
|
+
* per-step source. That binding lives in the engine wire-up PR, not
|
|
27
|
+
* here.
|
|
28
|
+
* - #5 pre-flight token estimator may pre-set `tierBudget` to gate
|
|
29
|
+
* out 'expensive' sources before the LLM call is made.
|
|
30
|
+
*
|
|
31
|
+
* Out of scope (separate PRs):
|
|
32
|
+
* - Engine wire-up that binds Hiroshi persona to `transport`.
|
|
33
|
+
* - Source-catalog registry — caller supplies the catalog per call.
|
|
34
|
+
* - Per-source cost telemetry (separate follow-up).
|
|
35
|
+
* - Streaming partial-ranks during LLM emission.
|
|
36
|
+
*/
|
|
37
|
+
import { estimateTokens } from '../compact/token-counter.js';
|
|
38
|
+
/** Whitelisted source kinds. New kinds require a corresponding catalog migration. */
|
|
39
|
+
const SOURCE_KIND_VALUES = ['rag', 'web', 'code', 'cache', 'agent', 'tool'];
|
|
40
|
+
const DEFAULT_MAX_FALLBACKS = 3;
|
|
41
|
+
const DEFAULT_MAX_INPUT_TOKENS = 4000;
|
|
42
|
+
/** Hard upper bound on pickIds we accept from the LLM. 1 primary + 3 fallbacks. */
|
|
43
|
+
const HARD_MAX_PICKS = 4;
|
|
44
|
+
/** Trim the user query before substitution. Keeps router prompt within sane bounds. */
|
|
45
|
+
const PROMPT_QUERY_HARD_CAP_CHARS = 2000;
|
|
46
|
+
/**
|
|
47
|
+
* Prompt template embedded as a constant so the primitive has zero
|
|
48
|
+
* filesystem dependencies. The `{contextHint}`, `{query}`, and
|
|
49
|
+
* `{candidates}` markers are substituted at call time. The instruction
|
|
50
|
+
* explicitly forbids markdown fences and prose around the JSON, which
|
|
51
|
+
* trims the most common parse failures observed with mid-tier models.
|
|
52
|
+
*/
|
|
53
|
+
export const ADAPTIVE_ROUTER_PROMPT_TEMPLATE = [
|
|
54
|
+
'You are an adaptive router for a software-engineering AI assistant.',
|
|
55
|
+
"Pick the best source(s) from the candidate list to answer the user's query.",
|
|
56
|
+
'',
|
|
57
|
+
'Return JSON only matching this schema:',
|
|
58
|
+
'{ "pickIds": ["source-id-1", "source-id-2", ...], "confidence": 0.85, "reasoning": "..." }',
|
|
59
|
+
'',
|
|
60
|
+
'Rules:',
|
|
61
|
+
'- Order pickIds best to worst (first = primary).',
|
|
62
|
+
'- Max 4 ids total (1 primary + up to 3 fallbacks).',
|
|
63
|
+
'- confidence in [0, 1].',
|
|
64
|
+
'- reasoning is one sentence under 200 chars.',
|
|
65
|
+
'- No prose around JSON. No markdown fences.',
|
|
66
|
+
'',
|
|
67
|
+
'CONTEXT: {contextHint}',
|
|
68
|
+
'QUERY: {query}',
|
|
69
|
+
'',
|
|
70
|
+
'CANDIDATES:',
|
|
71
|
+
'{candidates}',
|
|
72
|
+
].join('\n');
|
|
73
|
+
/**
|
|
74
|
+
* Pick the best source(s) for a query. Never throws.
|
|
75
|
+
*
|
|
76
|
+
* Every internal failure mode collapses to a heuristic fallback that
|
|
77
|
+
* still produces a usable `RouteDecision`. Caller inspects
|
|
78
|
+
* `usedFallback` and `reason` to decide whether to surface a warning or
|
|
79
|
+
* treat the routing as opaque.
|
|
80
|
+
*/
|
|
81
|
+
export async function chooseRoute(opts) {
|
|
82
|
+
const filtered = filterCatalog(opts.catalog, opts.tierBudget);
|
|
83
|
+
const maxFallbacks = clampMaxFallbacks(opts.maxFallbacks);
|
|
84
|
+
const maxInputTokens = clampMaxInputTokens(opts.maxInputTokens);
|
|
85
|
+
if (filtered.length === 0) {
|
|
86
|
+
return emptyCatalogFallback();
|
|
87
|
+
}
|
|
88
|
+
if (filtered.length === 1) {
|
|
89
|
+
// Single available source — no need to round-trip through the LLM.
|
|
90
|
+
const only = filtered[0];
|
|
91
|
+
if (!only)
|
|
92
|
+
return emptyCatalogFallback();
|
|
93
|
+
return {
|
|
94
|
+
primary: only,
|
|
95
|
+
fallbacks: [],
|
|
96
|
+
confidence: 1,
|
|
97
|
+
usedFallback: false,
|
|
98
|
+
};
|
|
99
|
+
}
|
|
100
|
+
const safeQuery = truncateToTokenBudget(opts.query.trim(), maxInputTokens);
|
|
101
|
+
const prompt = buildPrompt(safeQuery, opts.contextHint ?? 'unknown', filtered);
|
|
102
|
+
let raw;
|
|
103
|
+
try {
|
|
104
|
+
raw = await opts.transport(prompt);
|
|
105
|
+
}
|
|
106
|
+
catch {
|
|
107
|
+
return heuristicFallback(filtered, maxFallbacks, 'transport-error');
|
|
108
|
+
}
|
|
109
|
+
const parsed = parseAndValidate(raw, filtered);
|
|
110
|
+
if (!parsed.ok) {
|
|
111
|
+
return heuristicFallback(filtered, maxFallbacks, parsed.reason);
|
|
112
|
+
}
|
|
113
|
+
const ranked = parsed.sources;
|
|
114
|
+
const first = ranked[0];
|
|
115
|
+
if (!first) {
|
|
116
|
+
return heuristicFallback(filtered, maxFallbacks, 'empty-pick-ids');
|
|
117
|
+
}
|
|
118
|
+
const fallbacks = ranked.slice(1, 1 + maxFallbacks);
|
|
119
|
+
return {
|
|
120
|
+
primary: first,
|
|
121
|
+
fallbacks,
|
|
122
|
+
confidence: clampConfidence(parsed.confidence),
|
|
123
|
+
reasoning: parsed.reasoning,
|
|
124
|
+
usedFallback: false,
|
|
125
|
+
};
|
|
126
|
+
}
|
|
127
|
+
// -----------------------------------------------------------------------
|
|
128
|
+
// Internals
|
|
129
|
+
// -----------------------------------------------------------------------
|
|
130
|
+
function filterCatalog(catalog, tierBudget) {
|
|
131
|
+
const out = [];
|
|
132
|
+
for (const source of catalog) {
|
|
133
|
+
if (!source.available)
|
|
134
|
+
continue;
|
|
135
|
+
if (tierBudget === 'cheap' && source.costHint === 'expensive')
|
|
136
|
+
continue;
|
|
137
|
+
out.push(source);
|
|
138
|
+
}
|
|
139
|
+
return out;
|
|
140
|
+
}
|
|
141
|
+
function buildPrompt(query, contextHint, sources) {
|
|
142
|
+
const trimmedQuery = query.length > PROMPT_QUERY_HARD_CAP_CHARS
|
|
143
|
+
? query.slice(0, PROMPT_QUERY_HARD_CAP_CHARS)
|
|
144
|
+
: query;
|
|
145
|
+
const candidateLines = sources
|
|
146
|
+
.map((s) => `- ${s.id} (${s.kind}, ${s.costHint}, ${s.latencyHint}): ${s.description}`)
|
|
147
|
+
.join('\n');
|
|
148
|
+
return ADAPTIVE_ROUTER_PROMPT_TEMPLATE
|
|
149
|
+
.replace('{contextHint}', contextHint)
|
|
150
|
+
.replace('{query}', trimmedQuery)
|
|
151
|
+
.replace('{candidates}', candidateLines);
|
|
152
|
+
}
|
|
153
|
+
function parseAndValidate(raw, catalog) {
|
|
154
|
+
let parsed;
|
|
155
|
+
try {
|
|
156
|
+
parsed = JSON.parse(stripFences(raw));
|
|
157
|
+
}
|
|
158
|
+
catch {
|
|
159
|
+
return { ok: false, reason: 'parse-failed' };
|
|
160
|
+
}
|
|
161
|
+
if (!isRecord(parsed)) {
|
|
162
|
+
return { ok: false, reason: 'schema-violation' };
|
|
163
|
+
}
|
|
164
|
+
const pickIdsRaw = parsed.pickIds;
|
|
165
|
+
if (!Array.isArray(pickIdsRaw)) {
|
|
166
|
+
return { ok: false, reason: 'schema-violation' };
|
|
167
|
+
}
|
|
168
|
+
if (pickIdsRaw.length === 0) {
|
|
169
|
+
return { ok: false, reason: 'empty-pick-ids' };
|
|
170
|
+
}
|
|
171
|
+
const byId = new Map(catalog.map((s) => [s.id, s]));
|
|
172
|
+
const seen = new Set();
|
|
173
|
+
const resolved = [];
|
|
174
|
+
for (const candidate of pickIdsRaw) {
|
|
175
|
+
if (typeof candidate !== 'string') {
|
|
176
|
+
return { ok: false, reason: 'schema-violation' };
|
|
177
|
+
}
|
|
178
|
+
if (seen.has(candidate)) {
|
|
179
|
+
return { ok: false, reason: 'duplicate-pick-id' };
|
|
180
|
+
}
|
|
181
|
+
const source = byId.get(candidate);
|
|
182
|
+
if (!source) {
|
|
183
|
+
return { ok: false, reason: 'unknown-source-id' };
|
|
184
|
+
}
|
|
185
|
+
seen.add(candidate);
|
|
186
|
+
resolved.push(source);
|
|
187
|
+
// Cap at the hard ceiling so a malicious / runaway LLM cannot make us allocate unbounded fallbacks.
|
|
188
|
+
if (resolved.length >= HARD_MAX_PICKS)
|
|
189
|
+
break;
|
|
190
|
+
}
|
|
191
|
+
const confidenceRaw = parsed.confidence;
|
|
192
|
+
// Confidence is intentionally lenient — we coerce non-numbers to 0.5
|
|
193
|
+
// and clamp the rest. Out-of-range is NOT a fallback trigger per the
|
|
194
|
+
// spec: pickIds is the load-bearing field.
|
|
195
|
+
const confidence = typeof confidenceRaw === 'number' && Number.isFinite(confidenceRaw)
|
|
196
|
+
? confidenceRaw
|
|
197
|
+
: 0.5;
|
|
198
|
+
const reasoningRaw = parsed.reasoning;
|
|
199
|
+
const reasoning = typeof reasoningRaw === 'string' && reasoningRaw.trim().length > 0
|
|
200
|
+
? reasoningRaw.trim()
|
|
201
|
+
: undefined;
|
|
202
|
+
return {
|
|
203
|
+
ok: true,
|
|
204
|
+
sources: resolved,
|
|
205
|
+
confidence,
|
|
206
|
+
...(reasoning !== undefined ? { reasoning } : {}),
|
|
207
|
+
};
|
|
208
|
+
}
|
|
209
|
+
/**
|
|
210
|
+
* Heuristic ranking applied when the LLM call fails. Priority order:
|
|
211
|
+
*
|
|
212
|
+
* 1. `kind === 'cache'` first (always cheapest and safe to try).
|
|
213
|
+
* 2. `kind === 'agent'` next (persona has its own routing intelligence).
|
|
214
|
+
* 3. Everything else in the catalog's natural order.
|
|
215
|
+
*
|
|
216
|
+
* Within each bucket the catalog's natural order acts as a stable
|
|
217
|
+
* tiebreaker — the caller controls the priority by ordering the
|
|
218
|
+
* catalog deliberately.
|
|
219
|
+
*/
|
|
220
|
+
function heuristicFallback(catalog, maxFallbacks, reason) {
|
|
221
|
+
const ranked = heuristicRank(catalog);
|
|
222
|
+
const first = ranked[0];
|
|
223
|
+
// Defensive: caller guarantees catalog.length >= 1 by the time we
|
|
224
|
+
// get here (the `filtered.length === 0` short-circuit runs upstream),
|
|
225
|
+
// but the type checker needs the guard.
|
|
226
|
+
if (!first)
|
|
227
|
+
return emptyCatalogFallback();
|
|
228
|
+
const fallbacks = ranked.slice(1, 1 + maxFallbacks);
|
|
229
|
+
return {
|
|
230
|
+
primary: first,
|
|
231
|
+
fallbacks,
|
|
232
|
+
confidence: 0,
|
|
233
|
+
usedFallback: true,
|
|
234
|
+
reason,
|
|
235
|
+
};
|
|
236
|
+
}
|
|
237
|
+
function heuristicRank(catalog) {
|
|
238
|
+
const cache = [];
|
|
239
|
+
const agents = [];
|
|
240
|
+
const rest = [];
|
|
241
|
+
for (const source of catalog) {
|
|
242
|
+
if (source.kind === 'cache')
|
|
243
|
+
cache.push(source);
|
|
244
|
+
else if (source.kind === 'agent')
|
|
245
|
+
agents.push(source);
|
|
246
|
+
else
|
|
247
|
+
rest.push(source);
|
|
248
|
+
}
|
|
249
|
+
return [...cache, ...agents, ...rest];
|
|
250
|
+
}
|
|
251
|
+
/**
|
|
252
|
+
* Synthetic fallback used only when the filtered catalog is empty.
|
|
253
|
+
* The caller MUST treat `usedFallback === true` as a signal to skip
|
|
254
|
+
* the routed step — the `primary` here is a placeholder so the return
|
|
255
|
+
* type stays non-nullable.
|
|
256
|
+
*/
|
|
257
|
+
function emptyCatalogFallback() {
|
|
258
|
+
return {
|
|
259
|
+
primary: {
|
|
260
|
+
id: 'unknown',
|
|
261
|
+
kind: 'tool',
|
|
262
|
+
description: 'No source available for this query.',
|
|
263
|
+
costHint: 'cheap',
|
|
264
|
+
latencyHint: 'sub-second',
|
|
265
|
+
available: false,
|
|
266
|
+
},
|
|
267
|
+
fallbacks: [],
|
|
268
|
+
confidence: 0,
|
|
269
|
+
usedFallback: true,
|
|
270
|
+
reason: 'empty-catalog',
|
|
271
|
+
};
|
|
272
|
+
}
|
|
273
|
+
function clampMaxFallbacks(value) {
|
|
274
|
+
if (typeof value !== 'number' || !Number.isFinite(value) || value < 0) {
|
|
275
|
+
return DEFAULT_MAX_FALLBACKS;
|
|
276
|
+
}
|
|
277
|
+
return Math.floor(value);
|
|
278
|
+
}
|
|
279
|
+
function clampMaxInputTokens(value) {
|
|
280
|
+
if (typeof value !== 'number' || !Number.isFinite(value) || value < 1) {
|
|
281
|
+
return DEFAULT_MAX_INPUT_TOKENS;
|
|
282
|
+
}
|
|
283
|
+
return Math.floor(value);
|
|
284
|
+
}
|
|
285
|
+
function clampConfidence(value) {
|
|
286
|
+
if (!Number.isFinite(value))
|
|
287
|
+
return 0.5;
|
|
288
|
+
if (value < 0)
|
|
289
|
+
return 0;
|
|
290
|
+
if (value > 1)
|
|
291
|
+
return 1;
|
|
292
|
+
return value;
|
|
293
|
+
}
|
|
294
|
+
/**
|
|
295
|
+
* Truncate input to the token budget using the existing chars/4
|
|
296
|
+
* heuristic. We slice by characters proportional to the budget rather
|
|
297
|
+
* than binary-searching `estimateTokens`, because the heuristic is
|
|
298
|
+
* monotonic in length and an over-shoot is acceptable — the prompt also
|
|
299
|
+
* carries the static template overhead.
|
|
300
|
+
*/
|
|
301
|
+
function truncateToTokenBudget(text, maxTokens) {
|
|
302
|
+
const current = estimateTokens(text);
|
|
303
|
+
if (current <= maxTokens)
|
|
304
|
+
return text;
|
|
305
|
+
const ratio = maxTokens / current;
|
|
306
|
+
const targetLength = Math.max(1, Math.floor(text.length * ratio));
|
|
307
|
+
return text.slice(0, targetLength);
|
|
308
|
+
}
|
|
309
|
+
/**
|
|
310
|
+
* Tolerate the most common mid-tier-model deviation: wrapping JSON in
|
|
311
|
+
* a fenced code block. We strip a single leading and trailing fence
|
|
312
|
+
* marker when both are present; otherwise the raw text is returned
|
|
313
|
+
* unchanged and `JSON.parse` will reject it through the normal path.
|
|
314
|
+
*/
|
|
315
|
+
function stripFences(raw) {
|
|
316
|
+
const trimmed = raw.trim();
|
|
317
|
+
if (!trimmed.startsWith('```'))
|
|
318
|
+
return trimmed;
|
|
319
|
+
const firstNewline = trimmed.indexOf('\n');
|
|
320
|
+
if (firstNewline < 0)
|
|
321
|
+
return trimmed;
|
|
322
|
+
const closing = trimmed.lastIndexOf('```');
|
|
323
|
+
if (closing <= firstNewline)
|
|
324
|
+
return trimmed;
|
|
325
|
+
return trimmed.slice(firstNewline + 1, closing).trim();
|
|
326
|
+
}
|
|
327
|
+
function isRecord(value) {
|
|
328
|
+
return typeof value === 'object' && value !== null && !Array.isArray(value);
|
|
329
|
+
}
|
|
330
|
+
//# sourceMappingURL=adaptive-router.js.map
|
|
@@ -0,0 +1,297 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Query decomposer — primitive #3 of the 9-layer RAG architecture gap
|
|
3
|
+
* analysis (docs/research/2026-05-31-9-layer-rag-architecture-.md).
|
|
4
|
+
*
|
|
5
|
+
* Splits a user request into ordered atomic sub-queries the engine can
|
|
6
|
+
* route, plan, and execute step-by-step. This file is pure logic — it
|
|
7
|
+
* never opens a network connection itself. The caller injects a
|
|
8
|
+
* `transport` callback that maps a prompt string to a model response
|
|
9
|
+
* string, which keeps Anvil-engine wiring (Hiroshi persona, auth,
|
|
10
|
+
* retry, telemetry) out of the primitive and makes tests deterministic.
|
|
11
|
+
*
|
|
12
|
+
* Failure model is fail-closed to a single-step pass-through. Every
|
|
13
|
+
* malformed response, transport throw, schema violation, dependency
|
|
14
|
+
* cycle, or oversized step list collapses to:
|
|
15
|
+
*
|
|
16
|
+
* { steps: [{ id: 'step-1', query, dependsOn: [], intent: 'unknown' }],
|
|
17
|
+
* usedFallback: true, reason: '<why>' }
|
|
18
|
+
*
|
|
19
|
+
* The caller inspects `usedFallback` plus `reason` and decides whether
|
|
20
|
+
* the degradation is acceptable for the active route. Decomposition is
|
|
21
|
+
* never load-bearing for correctness — the engine can always treat the
|
|
22
|
+
* pass-through as the original query.
|
|
23
|
+
*
|
|
24
|
+
* Out of scope (separate PRs):
|
|
25
|
+
* - Engine wire-up that binds Hiroshi persona to `transport`.
|
|
26
|
+
* - Adaptive router (primitive #4) that consumes the step list.
|
|
27
|
+
* - Cost telemetry for the decomposer call itself.
|
|
28
|
+
*/
|
|
29
|
+
import { estimateTokens } from '../compact/token-counter.js';
|
|
30
|
+
/** Whitelisted intent categories. Anything outside collapses to 'unknown'. */
|
|
31
|
+
const INTENT_VALUES = ['read', 'write', 'verify', 'plan', 'explain', 'unknown'];
|
|
32
|
+
const DEFAULT_MAX_STEPS = 8;
|
|
33
|
+
const DEFAULT_MAX_INPUT_TOKENS = 4000;
|
|
34
|
+
const STEP_ID_PATTERN = /^step-\d+$/;
|
|
35
|
+
/**
|
|
36
|
+
* Prompt template embedded as a constant so the primitive has zero
|
|
37
|
+
* filesystem dependencies. The `{query}` marker is substituted at call
|
|
38
|
+
* time. The instruction explicitly forbids markdown fences and prose
|
|
39
|
+
* around the JSON, which trims the most common parse failures observed
|
|
40
|
+
* with mid-tier models.
|
|
41
|
+
*/
|
|
42
|
+
export const DECOMPOSER_PROMPT_TEMPLATE = [
|
|
43
|
+
'You are a query decomposer for a software-engineering AI assistant.',
|
|
44
|
+
"Split the user's request into atomic ordered steps.",
|
|
45
|
+
'',
|
|
46
|
+
'Return JSON only, matching this exact schema:',
|
|
47
|
+
'{ "steps": [{ "id": "step-N", "query": "...", "dependsOn": ["step-M", ...], "intent": "read|write|verify|plan|explain|unknown" }] }',
|
|
48
|
+
'',
|
|
49
|
+
'Rules:',
|
|
50
|
+
'- Max 8 steps.',
|
|
51
|
+
'- Single-step queries return one step.',
|
|
52
|
+
'- "dependsOn" lists earlier step IDs this step needs. Empty if independent.',
|
|
53
|
+
'- "intent" is the top action verb category.',
|
|
54
|
+
'- No prose around the JSON. No markdown fences.',
|
|
55
|
+
'',
|
|
56
|
+
'USER QUERY:',
|
|
57
|
+
'{query}',
|
|
58
|
+
].join('\n');
|
|
59
|
+
/**
|
|
60
|
+
* Decompose a query into ordered atomic sub-queries.
|
|
61
|
+
*
|
|
62
|
+
* The function never throws. Every internal failure mode collapses to
|
|
63
|
+
* a single-step fallback result. Caller inspects `usedFallback` and
|
|
64
|
+
* `reason` to decide whether to surface a warning or treat the
|
|
65
|
+
* decomposition as opaque.
|
|
66
|
+
*/
|
|
67
|
+
export async function decomposeQuery(query, opts) {
|
|
68
|
+
const trimmed = query.trim();
|
|
69
|
+
if (trimmed.length === 0) {
|
|
70
|
+
return fallback(query, 'empty-query');
|
|
71
|
+
}
|
|
72
|
+
const maxSteps = clampMaxSteps(opts.maxSteps);
|
|
73
|
+
const maxInputTokens = clampMaxInputTokens(opts.maxInputTokens);
|
|
74
|
+
const safeQuery = truncateToTokenBudget(trimmed, maxInputTokens);
|
|
75
|
+
const prompt = DECOMPOSER_PROMPT_TEMPLATE.replace('{query}', safeQuery);
|
|
76
|
+
let raw;
|
|
77
|
+
try {
|
|
78
|
+
raw = await opts.transport(prompt);
|
|
79
|
+
}
|
|
80
|
+
catch {
|
|
81
|
+
return fallback(safeQuery, 'transport-error');
|
|
82
|
+
}
|
|
83
|
+
const parseResult = parseAndValidate(raw, maxSteps);
|
|
84
|
+
if (parseResult.ok) {
|
|
85
|
+
return { steps: parseResult.steps, usedFallback: false };
|
|
86
|
+
}
|
|
87
|
+
return fallback(safeQuery, parseResult.reason);
|
|
88
|
+
}
|
|
89
|
+
function parseAndValidate(raw, maxSteps) {
|
|
90
|
+
let parsed;
|
|
91
|
+
try {
|
|
92
|
+
parsed = JSON.parse(stripFences(raw));
|
|
93
|
+
}
|
|
94
|
+
catch {
|
|
95
|
+
return { ok: false, reason: 'parse-failed' };
|
|
96
|
+
}
|
|
97
|
+
if (!isRecord(parsed)) {
|
|
98
|
+
return { ok: false, reason: 'schema-violation' };
|
|
99
|
+
}
|
|
100
|
+
const stepsRaw = parsed.steps;
|
|
101
|
+
if (!Array.isArray(stepsRaw)) {
|
|
102
|
+
return { ok: false, reason: 'schema-violation' };
|
|
103
|
+
}
|
|
104
|
+
if (stepsRaw.length === 0) {
|
|
105
|
+
return { ok: false, reason: 'empty-step-list' };
|
|
106
|
+
}
|
|
107
|
+
if (stepsRaw.length > maxSteps) {
|
|
108
|
+
return { ok: false, reason: 'exceeds-max-steps' };
|
|
109
|
+
}
|
|
110
|
+
const steps = [];
|
|
111
|
+
const seenIds = new Set();
|
|
112
|
+
for (const candidate of stepsRaw) {
|
|
113
|
+
const built = buildStep(candidate);
|
|
114
|
+
if (!built.ok) {
|
|
115
|
+
return { ok: false, reason: built.reason };
|
|
116
|
+
}
|
|
117
|
+
if (seenIds.has(built.step.id)) {
|
|
118
|
+
return { ok: false, reason: 'duplicate-id' };
|
|
119
|
+
}
|
|
120
|
+
seenIds.add(built.step.id);
|
|
121
|
+
steps.push(built.step);
|
|
122
|
+
}
|
|
123
|
+
for (const step of steps) {
|
|
124
|
+
for (const dep of step.dependsOn) {
|
|
125
|
+
if (!seenIds.has(dep)) {
|
|
126
|
+
return { ok: false, reason: 'unknown-dependency' };
|
|
127
|
+
}
|
|
128
|
+
}
|
|
129
|
+
}
|
|
130
|
+
if (hasCycle(steps)) {
|
|
131
|
+
return { ok: false, reason: 'dependency-cycle' };
|
|
132
|
+
}
|
|
133
|
+
return { ok: true, steps };
|
|
134
|
+
}
|
|
135
|
+
function buildStep(candidate) {
|
|
136
|
+
if (!isRecord(candidate)) {
|
|
137
|
+
return { ok: false, reason: 'schema-violation' };
|
|
138
|
+
}
|
|
139
|
+
const id = candidate.id;
|
|
140
|
+
const query = candidate.query;
|
|
141
|
+
const dependsOn = candidate.dependsOn;
|
|
142
|
+
const intent = candidate.intent;
|
|
143
|
+
if (typeof id !== 'string' || !STEP_ID_PATTERN.test(id)) {
|
|
144
|
+
return { ok: false, reason: 'schema-violation' };
|
|
145
|
+
}
|
|
146
|
+
if (typeof query !== 'string' || query.trim().length === 0) {
|
|
147
|
+
return { ok: false, reason: 'schema-violation' };
|
|
148
|
+
}
|
|
149
|
+
const depsList = [];
|
|
150
|
+
if (dependsOn !== undefined) {
|
|
151
|
+
if (!Array.isArray(dependsOn)) {
|
|
152
|
+
return { ok: false, reason: 'schema-violation' };
|
|
153
|
+
}
|
|
154
|
+
for (const dep of dependsOn) {
|
|
155
|
+
if (typeof dep !== 'string' || !STEP_ID_PATTERN.test(dep)) {
|
|
156
|
+
return { ok: false, reason: 'schema-violation' };
|
|
157
|
+
}
|
|
158
|
+
if (dep === id) {
|
|
159
|
+
// self-dependency is a degenerate cycle, surface explicitly
|
|
160
|
+
return { ok: false, reason: 'dependency-cycle' };
|
|
161
|
+
}
|
|
162
|
+
depsList.push(dep);
|
|
163
|
+
}
|
|
164
|
+
}
|
|
165
|
+
let intentValue;
|
|
166
|
+
if (intent === undefined) {
|
|
167
|
+
intentValue = 'unknown';
|
|
168
|
+
}
|
|
169
|
+
else if (typeof intent === 'string' && isIntent(intent)) {
|
|
170
|
+
intentValue = intent;
|
|
171
|
+
}
|
|
172
|
+
else {
|
|
173
|
+
return { ok: false, reason: 'schema-violation' };
|
|
174
|
+
}
|
|
175
|
+
return {
|
|
176
|
+
ok: true,
|
|
177
|
+
step: {
|
|
178
|
+
id,
|
|
179
|
+
query: query.trim(),
|
|
180
|
+
dependsOn: depsList,
|
|
181
|
+
intent: intentValue,
|
|
182
|
+
},
|
|
183
|
+
};
|
|
184
|
+
}
|
|
185
|
+
/**
|
|
186
|
+
* Cycle detection via iterative depth-first search with coloring.
|
|
187
|
+
* `WHITE` = unvisited, `GRAY` = on current stack, `BLACK` = fully
|
|
188
|
+
* explored. A back edge to a GRAY node is a cycle.
|
|
189
|
+
*/
|
|
190
|
+
function hasCycle(steps) {
|
|
191
|
+
const WHITE = 0;
|
|
192
|
+
const GRAY = 1;
|
|
193
|
+
const BLACK = 2;
|
|
194
|
+
const colors = new Map();
|
|
195
|
+
for (const step of steps)
|
|
196
|
+
colors.set(step.id, WHITE);
|
|
197
|
+
const byId = new Map(steps.map((s) => [s.id, s]));
|
|
198
|
+
for (const root of steps) {
|
|
199
|
+
if (colors.get(root.id) !== WHITE)
|
|
200
|
+
continue;
|
|
201
|
+
const stack = [{ id: root.id, depIdx: 0 }];
|
|
202
|
+
colors.set(root.id, GRAY);
|
|
203
|
+
while (stack.length > 0) {
|
|
204
|
+
const frame = stack[stack.length - 1];
|
|
205
|
+
if (!frame)
|
|
206
|
+
break;
|
|
207
|
+
const node = byId.get(frame.id);
|
|
208
|
+
if (!node) {
|
|
209
|
+
stack.pop();
|
|
210
|
+
continue;
|
|
211
|
+
}
|
|
212
|
+
if (frame.depIdx >= node.dependsOn.length) {
|
|
213
|
+
colors.set(node.id, BLACK);
|
|
214
|
+
stack.pop();
|
|
215
|
+
continue;
|
|
216
|
+
}
|
|
217
|
+
const next = node.dependsOn[frame.depIdx];
|
|
218
|
+
frame.depIdx += 1;
|
|
219
|
+
if (next === undefined)
|
|
220
|
+
continue;
|
|
221
|
+
const c = colors.get(next);
|
|
222
|
+
if (c === GRAY)
|
|
223
|
+
return true;
|
|
224
|
+
if (c === WHITE) {
|
|
225
|
+
colors.set(next, GRAY);
|
|
226
|
+
stack.push({ id: next, depIdx: 0 });
|
|
227
|
+
}
|
|
228
|
+
}
|
|
229
|
+
}
|
|
230
|
+
return false;
|
|
231
|
+
}
|
|
232
|
+
function fallback(query, reason) {
|
|
233
|
+
return {
|
|
234
|
+
steps: [
|
|
235
|
+
{
|
|
236
|
+
id: 'step-1',
|
|
237
|
+
query: query.trim(),
|
|
238
|
+
dependsOn: [],
|
|
239
|
+
intent: 'unknown',
|
|
240
|
+
},
|
|
241
|
+
],
|
|
242
|
+
usedFallback: true,
|
|
243
|
+
reason,
|
|
244
|
+
};
|
|
245
|
+
}
|
|
246
|
+
function clampMaxSteps(value) {
|
|
247
|
+
if (typeof value !== 'number' || !Number.isFinite(value) || value < 1) {
|
|
248
|
+
return DEFAULT_MAX_STEPS;
|
|
249
|
+
}
|
|
250
|
+
return Math.floor(value);
|
|
251
|
+
}
|
|
252
|
+
function clampMaxInputTokens(value) {
|
|
253
|
+
if (typeof value !== 'number' || !Number.isFinite(value) || value < 1) {
|
|
254
|
+
return DEFAULT_MAX_INPUT_TOKENS;
|
|
255
|
+
}
|
|
256
|
+
return Math.floor(value);
|
|
257
|
+
}
|
|
258
|
+
/**
|
|
259
|
+
* Truncate input to the token budget using the existing chars/4
|
|
260
|
+
* heuristic. We slice by characters proportional to the budget rather
|
|
261
|
+
* than by binary-searching `estimateTokens`, because the heuristic is
|
|
262
|
+
* monotonic in length and an over-shoot is acceptable — the prompt
|
|
263
|
+
* also carries the static template overhead.
|
|
264
|
+
*/
|
|
265
|
+
function truncateToTokenBudget(text, maxTokens) {
|
|
266
|
+
const current = estimateTokens(text);
|
|
267
|
+
if (current <= maxTokens)
|
|
268
|
+
return text;
|
|
269
|
+
const ratio = maxTokens / current;
|
|
270
|
+
const targetLength = Math.max(1, Math.floor(text.length * ratio));
|
|
271
|
+
return text.slice(0, targetLength);
|
|
272
|
+
}
|
|
273
|
+
/**
|
|
274
|
+
* Tolerate the most common mid-tier-model deviation: wrapping JSON in
|
|
275
|
+
* a fenced code block. We strip a single leading and trailing fence
|
|
276
|
+
* marker when both are present; otherwise the raw text is returned
|
|
277
|
+
* unchanged and `JSON.parse` will reject it through the normal path.
|
|
278
|
+
*/
|
|
279
|
+
function stripFences(raw) {
|
|
280
|
+
const trimmed = raw.trim();
|
|
281
|
+
if (!trimmed.startsWith('```'))
|
|
282
|
+
return trimmed;
|
|
283
|
+
const firstNewline = trimmed.indexOf('\n');
|
|
284
|
+
if (firstNewline < 0)
|
|
285
|
+
return trimmed;
|
|
286
|
+
const closing = trimmed.lastIndexOf('```');
|
|
287
|
+
if (closing <= firstNewline)
|
|
288
|
+
return trimmed;
|
|
289
|
+
return trimmed.slice(firstNewline + 1, closing).trim();
|
|
290
|
+
}
|
|
291
|
+
function isRecord(value) {
|
|
292
|
+
return typeof value === 'object' && value !== null && !Array.isArray(value);
|
|
293
|
+
}
|
|
294
|
+
function isIntent(value) {
|
|
295
|
+
return INTENT_VALUES.includes(value);
|
|
296
|
+
}
|
|
297
|
+
//# sourceMappingURL=query-decomposer.js.map
|
|
@@ -7,10 +7,10 @@
|
|
|
7
7
|
* that mapping lives — keep it tight and explicit so persona drift never
|
|
8
8
|
* leaks back into the dispatch surface.
|
|
9
9
|
*
|
|
10
|
-
* M1 closed set
|
|
10
|
+
* M1 closed set : 9 roles, all mapped to Tier 1
|
|
11
11
|
* Engineering Core or Tier 2 Specialist personas from THE_TEN. Tier 1
|
|
12
12
|
* Missing Functions (Growth/Legal/Security/Sales/Support) are deferred
|
|
13
|
-
* to
|
|
13
|
+
* to ; Sigma is intentionally absent because it is an OES Enterprise
|
|
14
14
|
* persona via Anvil triple-review proxy, not a Cyber-Zoo brand persona.
|
|
15
15
|
*/
|
|
16
16
|
import { THE_TEN, getPersona } from '@pugi/personas';
|
|
@@ -30,7 +30,7 @@ function requirePersona(slug) {
|
|
|
30
30
|
/**
|
|
31
31
|
* CLI-only role-to-persona mapping. Roles are dispatcher-facing strings;
|
|
32
32
|
* personas come from the brand-canonical THE_TEN. Vera (qa) intentionally
|
|
33
|
-
* dual-roles as verifier + reviewer per
|
|
33
|
+
* dual-roles as verifier + reviewer per — the cabinet's review
|
|
34
34
|
* pipeline already merges the two surfaces.
|
|
35
35
|
*/
|
|
36
36
|
export const SUBAGENT_REGISTRY = [
|