@wahack/pi-coding-agent 15.11.0
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 +10031 -0
- package/README.md +36 -0
- package/examples/README.md +21 -0
- package/examples/custom-tools/README.md +104 -0
- package/examples/custom-tools/hello/index.ts +20 -0
- package/examples/extensions/README.md +142 -0
- package/examples/extensions/api-demo.ts +79 -0
- package/examples/extensions/chalk-logger.ts +25 -0
- package/examples/extensions/hello.ts +31 -0
- package/examples/extensions/pirate.ts +43 -0
- package/examples/extensions/plan-mode.ts +549 -0
- package/examples/extensions/reload-runtime.ts +38 -0
- package/examples/extensions/thinking-note.ts +13 -0
- package/examples/extensions/tools.ts +145 -0
- package/examples/extensions/with-deps/index.ts +36 -0
- package/examples/extensions/with-deps/package-lock.json +31 -0
- package/examples/extensions/with-deps/package.json +17 -0
- package/examples/hooks/README.md +56 -0
- package/examples/hooks/auto-commit-on-exit.ts +48 -0
- package/examples/hooks/confirm-destructive.ts +58 -0
- package/examples/hooks/custom-compaction.ts +115 -0
- package/examples/hooks/dirty-repo-guard.ts +51 -0
- package/examples/hooks/file-trigger.ts +40 -0
- package/examples/hooks/git-checkpoint.ts +52 -0
- package/examples/hooks/handoff.ts +149 -0
- package/examples/hooks/permission-gate.ts +33 -0
- package/examples/hooks/protected-paths.ts +29 -0
- package/examples/hooks/qna.ts +118 -0
- package/examples/hooks/status-line.ts +39 -0
- package/examples/sdk/01-minimal.ts +21 -0
- package/examples/sdk/02-custom-model.ts +49 -0
- package/examples/sdk/03-custom-prompt.ts +46 -0
- package/examples/sdk/04-skills.ts +43 -0
- package/examples/sdk/06-extensions.ts +82 -0
- package/examples/sdk/06-hooks.ts +61 -0
- package/examples/sdk/07-context-files.ts +35 -0
- package/examples/sdk/08-prompt-templates.ts +41 -0
- package/examples/sdk/08-slash-commands.ts +46 -0
- package/examples/sdk/09-api-keys-and-oauth.ts +54 -0
- package/examples/sdk/11-sessions.ts +47 -0
- package/examples/sdk/12-redis-sessions.ts +54 -0
- package/examples/sdk/13-sql-sessions.ts +61 -0
- package/examples/sdk/README.md +172 -0
- package/package.json +554 -0
- package/scripts/build-binary.ts +100 -0
- package/scripts/bundle-dist.ts +90 -0
- package/scripts/format-prompts.ts +68 -0
- package/scripts/generate-docs-index.ts +40 -0
- package/scripts/generate-template.ts +33 -0
- package/scripts/omp +42 -0
- package/scripts/omp.ts +19 -0
- package/src/async/index.ts +1 -0
- package/src/async/job-manager.ts +625 -0
- package/src/auto-thinking/classifier.ts +185 -0
- package/src/autoresearch/command-resume.md +14 -0
- package/src/autoresearch/dashboard.ts +436 -0
- package/src/autoresearch/git.ts +319 -0
- package/src/autoresearch/helpers.ts +218 -0
- package/src/autoresearch/index.ts +536 -0
- package/src/autoresearch/prompt-setup.md +43 -0
- package/src/autoresearch/prompt.md +103 -0
- package/src/autoresearch/resume-message.md +10 -0
- package/src/autoresearch/state.ts +273 -0
- package/src/autoresearch/storage.ts +699 -0
- package/src/autoresearch/tools/init-experiment.ts +272 -0
- package/src/autoresearch/tools/log-experiment.ts +524 -0
- package/src/autoresearch/tools/run-experiment.ts +407 -0
- package/src/autoresearch/tools/update-notes.ts +109 -0
- package/src/autoresearch/types.ts +168 -0
- package/src/bun-imports.d.ts +28 -0
- package/src/capability/context-file.ts +44 -0
- package/src/capability/extension-module.ts +34 -0
- package/src/capability/extension.ts +47 -0
- package/src/capability/fs.ts +117 -0
- package/src/capability/hook.ts +40 -0
- package/src/capability/index.ts +436 -0
- package/src/capability/instruction.ts +37 -0
- package/src/capability/mcp.ts +74 -0
- package/src/capability/prompt.ts +35 -0
- package/src/capability/rule-buckets.ts +66 -0
- package/src/capability/rule.ts +261 -0
- package/src/capability/settings.ts +34 -0
- package/src/capability/skill.ts +63 -0
- package/src/capability/slash-command.ts +40 -0
- package/src/capability/ssh.ts +41 -0
- package/src/capability/system-prompt.ts +34 -0
- package/src/capability/tool.ts +38 -0
- package/src/capability/types.ts +168 -0
- package/src/cli/agents-cli.ts +138 -0
- package/src/cli/args.ts +340 -0
- package/src/cli/auth-broker-cli.ts +895 -0
- package/src/cli/auth-gateway-cli.ts +611 -0
- package/src/cli/classify-install-target.ts +76 -0
- package/src/cli/claude-trace-cli.ts +795 -0
- package/src/cli/commands/init-xdg.ts +27 -0
- package/src/cli/completion-gen.ts +550 -0
- package/src/cli/config-cli.ts +418 -0
- package/src/cli/dry-balance-cli.ts +856 -0
- package/src/cli/extension-flags.ts +48 -0
- package/src/cli/file-processor.ts +133 -0
- package/src/cli/gallery-cli.ts +230 -0
- package/src/cli/gallery-fixtures/agentic.ts +407 -0
- package/src/cli/gallery-fixtures/codeintel.ts +187 -0
- package/src/cli/gallery-fixtures/edit.ts +194 -0
- package/src/cli/gallery-fixtures/fs.ts +220 -0
- package/src/cli/gallery-fixtures/index.ts +40 -0
- package/src/cli/gallery-fixtures/interaction.ts +49 -0
- package/src/cli/gallery-fixtures/memory.ts +81 -0
- package/src/cli/gallery-fixtures/misc.ts +250 -0
- package/src/cli/gallery-fixtures/search.ts +213 -0
- package/src/cli/gallery-fixtures/shell.ts +167 -0
- package/src/cli/gallery-fixtures/types.ts +57 -0
- package/src/cli/gallery-fixtures/web.ts +158 -0
- package/src/cli/gallery-screenshot.ts +279 -0
- package/src/cli/grep-cli.ts +160 -0
- package/src/cli/grievances-cli.ts +256 -0
- package/src/cli/initial-message.ts +58 -0
- package/src/cli/list-models.ts +194 -0
- package/src/cli/plugin-cli.ts +996 -0
- package/src/cli/read-cli.ts +57 -0
- package/src/cli/session-picker.ts +79 -0
- package/src/cli/setup-cli.ts +231 -0
- package/src/cli/shell-cli.ts +176 -0
- package/src/cli/ssh-cli.ts +179 -0
- package/src/cli/startup-cwd.ts +68 -0
- package/src/cli/stats-cli.ts +238 -0
- package/src/cli/tiny-models-cli.ts +127 -0
- package/src/cli/update-cli.ts +611 -0
- package/src/cli/usage-cli.ts +603 -0
- package/src/cli/web-search-cli.ts +132 -0
- package/src/cli/worktree-cli.ts +291 -0
- package/src/cli-commands.ts +79 -0
- package/src/cli.ts +200 -0
- package/src/commands/acp.ts +24 -0
- package/src/commands/agents.ts +57 -0
- package/src/commands/auth-broker.ts +99 -0
- package/src/commands/auth-gateway.ts +69 -0
- package/src/commands/commit.ts +46 -0
- package/src/commands/complete.ts +66 -0
- package/src/commands/completions.ts +60 -0
- package/src/commands/config.ts +51 -0
- package/src/commands/dry-balance.ts +43 -0
- package/src/commands/gallery.ts +52 -0
- package/src/commands/grep.ts +48 -0
- package/src/commands/grievances.ts +51 -0
- package/src/commands/install.ts +107 -0
- package/src/commands/launch.ts +169 -0
- package/src/commands/plugin.ts +78 -0
- package/src/commands/read.ts +38 -0
- package/src/commands/setup.ts +67 -0
- package/src/commands/shell.ts +29 -0
- package/src/commands/ssh.ts +60 -0
- package/src/commands/stats.ts +29 -0
- package/src/commands/tiny-models.ts +36 -0
- package/src/commands/update.ts +21 -0
- package/src/commands/usage.ts +35 -0
- package/src/commands/web-search.ts +42 -0
- package/src/commands/worktree.ts +56 -0
- package/src/commit/agentic/agent.ts +317 -0
- package/src/commit/agentic/fallback.ts +96 -0
- package/src/commit/agentic/index.ts +355 -0
- package/src/commit/agentic/prompts/analyze-file.md +22 -0
- package/src/commit/agentic/prompts/session-user.md +25 -0
- package/src/commit/agentic/prompts/split-confirm.md +1 -0
- package/src/commit/agentic/prompts/system.md +38 -0
- package/src/commit/agentic/state.ts +60 -0
- package/src/commit/agentic/tools/analyze-file.ts +146 -0
- package/src/commit/agentic/tools/git-file-diff.ts +191 -0
- package/src/commit/agentic/tools/git-hunk.ts +50 -0
- package/src/commit/agentic/tools/git-overview.ts +81 -0
- package/src/commit/agentic/tools/index.ts +54 -0
- package/src/commit/agentic/tools/propose-changelog.ts +144 -0
- package/src/commit/agentic/tools/propose-commit.ts +109 -0
- package/src/commit/agentic/tools/recent-commits.ts +81 -0
- package/src/commit/agentic/tools/schemas.ts +23 -0
- package/src/commit/agentic/tools/split-commit.ts +245 -0
- package/src/commit/agentic/topo-sort.ts +44 -0
- package/src/commit/agentic/trivial.ts +51 -0
- package/src/commit/agentic/validation.ts +183 -0
- package/src/commit/analysis/conventional.ts +64 -0
- package/src/commit/analysis/index.ts +4 -0
- package/src/commit/analysis/scope.ts +242 -0
- package/src/commit/analysis/summary.ts +105 -0
- package/src/commit/analysis/validation.ts +66 -0
- package/src/commit/changelog/detect.ts +40 -0
- package/src/commit/changelog/generate.ts +97 -0
- package/src/commit/changelog/index.ts +234 -0
- package/src/commit/changelog/parse.ts +44 -0
- package/src/commit/cli.ts +85 -0
- package/src/commit/git/diff.ts +148 -0
- package/src/commit/index.ts +5 -0
- package/src/commit/map-reduce/index.ts +69 -0
- package/src/commit/map-reduce/map-phase.ts +193 -0
- package/src/commit/map-reduce/reduce-phase.ts +49 -0
- package/src/commit/map-reduce/utils.ts +9 -0
- package/src/commit/message.ts +11 -0
- package/src/commit/model-selection.ts +92 -0
- package/src/commit/pipeline.ts +243 -0
- package/src/commit/prompts/analysis-system.md +148 -0
- package/src/commit/prompts/analysis-user.md +38 -0
- package/src/commit/prompts/changelog-system.md +50 -0
- package/src/commit/prompts/changelog-user.md +18 -0
- package/src/commit/prompts/file-observer-system.md +24 -0
- package/src/commit/prompts/file-observer-user.md +8 -0
- package/src/commit/prompts/reduce-system.md +50 -0
- package/src/commit/prompts/reduce-user.md +17 -0
- package/src/commit/prompts/summary-retry.md +3 -0
- package/src/commit/prompts/summary-system.md +38 -0
- package/src/commit/prompts/summary-user.md +13 -0
- package/src/commit/prompts/types-description.md +2 -0
- package/src/commit/shared-llm.ts +77 -0
- package/src/commit/types.ts +118 -0
- package/src/commit/utils/exclusions.ts +42 -0
- package/src/commit/utils.ts +58 -0
- package/src/config/api-key-resolver.ts +60 -0
- package/src/config/append-only-context-mode.ts +31 -0
- package/src/config/config-file.ts +317 -0
- package/src/config/file-lock.ts +164 -0
- package/src/config/keybindings.ts +628 -0
- package/src/config/mcp-schema.json +230 -0
- package/src/config/model-discovery.ts +554 -0
- package/src/config/model-registry.ts +2090 -0
- package/src/config/model-resolver.ts +1502 -0
- package/src/config/model-roles.ts +74 -0
- package/src/config/models-config-schema.ts +226 -0
- package/src/config/models-config.ts +129 -0
- package/src/config/prompt-templates.ts +185 -0
- package/src/config/resolve-config-value.ts +94 -0
- package/src/config/settings-schema.ts +3530 -0
- package/src/config/settings.ts +1178 -0
- package/src/config.ts +242 -0
- package/src/cursor.ts +340 -0
- package/src/dap/client.ts +760 -0
- package/src/dap/config.ts +189 -0
- package/src/dap/defaults.json +212 -0
- package/src/dap/index.ts +4 -0
- package/src/dap/session.ts +1441 -0
- package/src/dap/types.ts +610 -0
- package/src/debug/index.ts +515 -0
- package/src/debug/log-formatting.ts +58 -0
- package/src/debug/log-viewer.ts +908 -0
- package/src/debug/profiler.ts +162 -0
- package/src/debug/protocol-probe.ts +267 -0
- package/src/debug/raw-sse-buffer.ts +273 -0
- package/src/debug/raw-sse.ts +292 -0
- package/src/debug/report-bundle.ts +374 -0
- package/src/debug/system-info.ts +111 -0
- package/src/debug/terminal-info.ts +124 -0
- package/src/discovery/agents-md.ts +67 -0
- package/src/discovery/agents.ts +230 -0
- package/src/discovery/at-imports.ts +273 -0
- package/src/discovery/builtin-defaults.ts +39 -0
- package/src/discovery/builtin-rules/index.ts +54 -0
- package/src/discovery/builtin-rules/rs-box-leak.md +48 -0
- package/src/discovery/builtin-rules/rs-future-prelude.md +23 -0
- package/src/discovery/builtin-rules/rs-lazylock.md +51 -0
- package/src/discovery/builtin-rules/rs-match-ergonomics.md +67 -0
- package/src/discovery/builtin-rules/rs-parking-lot.md +44 -0
- package/src/discovery/builtin-rules/rs-result-type.md +19 -0
- package/src/discovery/builtin-rules/ts-bare-catch.md +38 -0
- package/src/discovery/builtin-rules/ts-import-type.md +42 -0
- package/src/discovery/builtin-rules/ts-no-any.md +56 -0
- package/src/discovery/builtin-rules/ts-no-deprecated-leftovers.md +44 -0
- package/src/discovery/builtin-rules/ts-no-dynamic-import.md +39 -0
- package/src/discovery/builtin-rules/ts-no-return-type.md +45 -0
- package/src/discovery/builtin-rules/ts-no-test-timers.md +55 -0
- package/src/discovery/builtin-rules/ts-no-tiny-functions.md +51 -0
- package/src/discovery/builtin-rules/ts-promise-with-resolvers.md +65 -0
- package/src/discovery/builtin-rules/ts-redundant-clear-guard.md +75 -0
- package/src/discovery/builtin-rules/ts-set-map.md +28 -0
- package/src/discovery/builtin.ts +906 -0
- package/src/discovery/claude-plugins.ts +386 -0
- package/src/discovery/claude.ts +584 -0
- package/src/discovery/cline.ts +83 -0
- package/src/discovery/codex.ts +522 -0
- package/src/discovery/cursor.ts +220 -0
- package/src/discovery/gemini.ts +383 -0
- package/src/discovery/github.ts +154 -0
- package/src/discovery/helpers.ts +1016 -0
- package/src/discovery/index.ts +81 -0
- package/src/discovery/mcp-json.ts +171 -0
- package/src/discovery/omp-extension-roots.ts +190 -0
- package/src/discovery/omp-plugins.ts +383 -0
- package/src/discovery/opencode.ts +398 -0
- package/src/discovery/plugin-dir-roots.ts +28 -0
- package/src/discovery/ssh.ts +153 -0
- package/src/discovery/substitute-plugin-root.ts +29 -0
- package/src/discovery/vscode.ts +105 -0
- package/src/discovery/windsurf.ts +147 -0
- package/src/edit/apply-patch/index.ts +87 -0
- package/src/edit/apply-patch/parser.ts +174 -0
- package/src/edit/diff.ts +999 -0
- package/src/edit/file-snapshot-store.ts +91 -0
- package/src/edit/hashline/block-resolver.ts +33 -0
- package/src/edit/hashline/diff.ts +290 -0
- package/src/edit/hashline/execute.ts +242 -0
- package/src/edit/hashline/filesystem.ts +130 -0
- package/src/edit/hashline/index.ts +5 -0
- package/src/edit/hashline/noop-loop-guard.ts +99 -0
- package/src/edit/hashline/params.ts +18 -0
- package/src/edit/index.ts +571 -0
- package/src/edit/modes/apply-patch.lark +19 -0
- package/src/edit/modes/apply-patch.ts +53 -0
- package/src/edit/modes/patch.ts +1891 -0
- package/src/edit/modes/replace.ts +1137 -0
- package/src/edit/normalize.ts +345 -0
- package/src/edit/notebook.ts +242 -0
- package/src/edit/read-file.ts +25 -0
- package/src/edit/renderer.ts +769 -0
- package/src/edit/streaming.ts +517 -0
- package/src/eval/__tests__/agent-bridge.test.ts +708 -0
- package/src/eval/__tests__/bridge-timeout.test.ts +64 -0
- package/src/eval/__tests__/budget-bridge.test.ts +69 -0
- package/src/eval/__tests__/completion-bridge.test.ts +412 -0
- package/src/eval/__tests__/helpers-local-roots.test.ts +58 -0
- package/src/eval/__tests__/idle-timeout.test.ts +80 -0
- package/src/eval/__tests__/js-context-manager.test.ts +241 -0
- package/src/eval/__tests__/kernel-spawn.test.ts +103 -0
- package/src/eval/agent-bridge.ts +319 -0
- package/src/eval/backend.ts +71 -0
- package/src/eval/bridge-timeout.ts +44 -0
- package/src/eval/budget-bridge.ts +48 -0
- package/src/eval/completion-bridge.ts +207 -0
- package/src/eval/concurrency-bridge.ts +34 -0
- package/src/eval/idle-timeout.ts +91 -0
- package/src/eval/index.ts +4 -0
- package/src/eval/js/context-manager.ts +502 -0
- package/src/eval/js/executor.ts +173 -0
- package/src/eval/js/index.ts +51 -0
- package/src/eval/js/shared/helpers.ts +283 -0
- package/src/eval/js/shared/indirect-eval.ts +30 -0
- package/src/eval/js/shared/local-module-loader.ts +342 -0
- package/src/eval/js/shared/prelude.ts +2 -0
- package/src/eval/js/shared/prelude.txt +246 -0
- package/src/eval/js/shared/rewrite-imports.ts +532 -0
- package/src/eval/js/shared/runtime.ts +352 -0
- package/src/eval/js/shared/types.ts +18 -0
- package/src/eval/js/tool-bridge.ts +162 -0
- package/src/eval/js/worker-core.ts +132 -0
- package/src/eval/js/worker-entry.ts +30 -0
- package/src/eval/js/worker-protocol.ts +47 -0
- package/src/eval/py/__tests__/prelude.test.ts +19 -0
- package/src/eval/py/display.ts +71 -0
- package/src/eval/py/executor.ts +742 -0
- package/src/eval/py/index.ts +68 -0
- package/src/eval/py/kernel.ts +748 -0
- package/src/eval/py/prelude.py +658 -0
- package/src/eval/py/prelude.ts +3 -0
- package/src/eval/py/runner.py +1133 -0
- package/src/eval/py/runtime.ts +276 -0
- package/src/eval/py/spawn-options.ts +126 -0
- package/src/eval/py/tool-bridge.ts +182 -0
- package/src/eval/session-id.ts +8 -0
- package/src/eval/types.ts +48 -0
- package/src/exa/index.ts +2 -0
- package/src/exa/mcp-client.ts +370 -0
- package/src/exa/types.ts +69 -0
- package/src/exec/bash-executor.ts +419 -0
- package/src/exec/exec.ts +53 -0
- package/src/exec/non-interactive-env.ts +48 -0
- package/src/export/custom-share.ts +65 -0
- package/src/export/html/index.ts +164 -0
- package/src/export/html/template.css +1051 -0
- package/src/export/html/template.generated.ts +2 -0
- package/src/export/html/template.html +46 -0
- package/src/export/html/template.js +2271 -0
- package/src/export/html/template.macro.ts +25 -0
- package/src/export/html/vendor/highlight.min.js +1213 -0
- package/src/export/html/vendor/marked.min.js +6 -0
- package/src/export/ttsr.ts +583 -0
- package/src/extensibility/custom-commands/bundled/ci-green/index.ts +54 -0
- package/src/extensibility/custom-commands/bundled/review/index.ts +489 -0
- package/src/extensibility/custom-commands/index.ts +2 -0
- package/src/extensibility/custom-commands/loader.ts +238 -0
- package/src/extensibility/custom-commands/types.ts +113 -0
- package/src/extensibility/custom-tools/index.ts +7 -0
- package/src/extensibility/custom-tools/loader.ts +269 -0
- package/src/extensibility/custom-tools/types.ts +270 -0
- package/src/extensibility/custom-tools/wrapper.ts +47 -0
- package/src/extensibility/extensions/compact-handler.ts +40 -0
- package/src/extensibility/extensions/get-commands-handler.ts +78 -0
- package/src/extensibility/extensions/index.ts +16 -0
- package/src/extensibility/extensions/loader.ts +572 -0
- package/src/extensibility/extensions/runner.ts +922 -0
- package/src/extensibility/extensions/types.ts +1322 -0
- package/src/extensibility/extensions/wrapper.ts +223 -0
- package/src/extensibility/hooks/index.ts +5 -0
- package/src/extensibility/hooks/loader.ts +257 -0
- package/src/extensibility/hooks/runner.ts +425 -0
- package/src/extensibility/hooks/tool-wrapper.ts +107 -0
- package/src/extensibility/hooks/types.ts +606 -0
- package/src/extensibility/legacy-pi-ai-shim.ts +24 -0
- package/src/extensibility/legacy-pi-coding-agent-shim.ts +15 -0
- package/src/extensibility/plugins/doctor.ts +65 -0
- package/src/extensibility/plugins/git-url.ts +367 -0
- package/src/extensibility/plugins/index.ts +9 -0
- package/src/extensibility/plugins/installer.ts +192 -0
- package/src/extensibility/plugins/legacy-pi-compat.ts +682 -0
- package/src/extensibility/plugins/loader.ts +313 -0
- package/src/extensibility/plugins/manager.ts +827 -0
- package/src/extensibility/plugins/marketplace/cache.ts +136 -0
- package/src/extensibility/plugins/marketplace/fetcher.ts +317 -0
- package/src/extensibility/plugins/marketplace/index.ts +6 -0
- package/src/extensibility/plugins/marketplace/manager.ts +770 -0
- package/src/extensibility/plugins/marketplace/registry.ts +196 -0
- package/src/extensibility/plugins/marketplace/source-resolver.ts +147 -0
- package/src/extensibility/plugins/marketplace/types.ts +191 -0
- package/src/extensibility/plugins/marketplace-auto-update.ts +49 -0
- package/src/extensibility/plugins/parser.ts +105 -0
- package/src/extensibility/plugins/types.ts +194 -0
- package/src/extensibility/shared-events.ts +343 -0
- package/src/extensibility/skills.ts +312 -0
- package/src/extensibility/slash-commands.ts +227 -0
- package/src/extensibility/tool-proxy.ts +25 -0
- package/src/extensibility/typebox.ts +418 -0
- package/src/extensibility/utils.ts +44 -0
- package/src/goals/index.ts +3 -0
- package/src/goals/runtime.ts +528 -0
- package/src/goals/state.ts +37 -0
- package/src/goals/tools/goal-tool.ts +251 -0
- package/src/hindsight/backend.ts +354 -0
- package/src/hindsight/bank.ts +156 -0
- package/src/hindsight/client.ts +598 -0
- package/src/hindsight/config.ts +175 -0
- package/src/hindsight/content.ts +210 -0
- package/src/hindsight/index.ts +8 -0
- package/src/hindsight/mental-models.ts +429 -0
- package/src/hindsight/seeds.json +32 -0
- package/src/hindsight/state.ts +488 -0
- package/src/hindsight/transcript.ts +71 -0
- package/src/index.ts +59 -0
- package/src/internal-urls/agent-protocol.ts +146 -0
- package/src/internal-urls/artifact-protocol.ts +107 -0
- package/src/internal-urls/docs-index.generated.ts +106 -0
- package/src/internal-urls/history-protocol.ts +113 -0
- package/src/internal-urls/index.ts +25 -0
- package/src/internal-urls/issue-pr-protocol.ts +584 -0
- package/src/internal-urls/json-query.ts +126 -0
- package/src/internal-urls/local-protocol.ts +287 -0
- package/src/internal-urls/mcp-protocol.ts +151 -0
- package/src/internal-urls/memory-protocol.ts +169 -0
- package/src/internal-urls/omp-protocol.ts +93 -0
- package/src/internal-urls/parse.ts +72 -0
- package/src/internal-urls/registry-helpers.ts +25 -0
- package/src/internal-urls/router.ts +105 -0
- package/src/internal-urls/rule-protocol.ts +45 -0
- package/src/internal-urls/skill-protocol.ts +96 -0
- package/src/internal-urls/types.ts +152 -0
- package/src/internal-urls/vault-protocol.ts +936 -0
- package/src/irc/bus.ts +292 -0
- package/src/lib/xai-http.ts +124 -0
- package/src/lsp/client.ts +1193 -0
- package/src/lsp/clients/biome-client.ts +264 -0
- package/src/lsp/clients/index.ts +50 -0
- package/src/lsp/clients/lsp-linter-client.ts +93 -0
- package/src/lsp/clients/swiftlint-client.ts +120 -0
- package/src/lsp/config.ts +502 -0
- package/src/lsp/defaults.json +493 -0
- package/src/lsp/diagnostics-ledger.ts +51 -0
- package/src/lsp/edits.ts +267 -0
- package/src/lsp/index.ts +2477 -0
- package/src/lsp/lspmux.ts +233 -0
- package/src/lsp/render.ts +694 -0
- package/src/lsp/startup-events.ts +13 -0
- package/src/lsp/types.ts +455 -0
- package/src/lsp/utils.ts +718 -0
- package/src/main.ts +1325 -0
- package/src/mcp/client.ts +484 -0
- package/src/mcp/config-writer.ts +225 -0
- package/src/mcp/config.ts +365 -0
- package/src/mcp/index.ts +29 -0
- package/src/mcp/json-rpc.ts +122 -0
- package/src/mcp/loader.ts +124 -0
- package/src/mcp/manager.ts +1275 -0
- package/src/mcp/oauth-discovery.ts +442 -0
- package/src/mcp/oauth-flow.ts +442 -0
- package/src/mcp/render.ts +132 -0
- package/src/mcp/smithery-auth.ts +104 -0
- package/src/mcp/smithery-connect.ts +145 -0
- package/src/mcp/smithery-registry.ts +477 -0
- package/src/mcp/timeout.ts +59 -0
- package/src/mcp/tool-bridge.ts +426 -0
- package/src/mcp/tool-cache.ts +117 -0
- package/src/mcp/transports/http.ts +519 -0
- package/src/mcp/transports/index.ts +6 -0
- package/src/mcp/transports/stdio.ts +528 -0
- package/src/mcp/types.ts +423 -0
- package/src/memories/index.ts +1150 -0
- package/src/memories/storage.ts +577 -0
- package/src/memory-backend/index.ts +18 -0
- package/src/memory-backend/local-backend.ts +39 -0
- package/src/memory-backend/off-backend.ts +25 -0
- package/src/memory-backend/resolve.ts +25 -0
- package/src/memory-backend/runtime.ts +66 -0
- package/src/memory-backend/types.ts +166 -0
- package/src/mnemopi/backend.ts +547 -0
- package/src/mnemopi/config.ts +160 -0
- package/src/mnemopi/index.ts +3 -0
- package/src/mnemopi/state.ts +584 -0
- package/src/modes/acp/acp-agent.ts +2407 -0
- package/src/modes/acp/acp-client-bridge.ts +154 -0
- package/src/modes/acp/acp-event-mapper.ts +929 -0
- package/src/modes/acp/acp-mode.ts +23 -0
- package/src/modes/acp/index.ts +2 -0
- package/src/modes/acp/terminal-auth.ts +37 -0
- package/src/modes/components/agent-dashboard.ts +1206 -0
- package/src/modes/components/agent-hub.ts +1071 -0
- package/src/modes/components/assistant-message.ts +307 -0
- package/src/modes/components/bash-execution.ts +220 -0
- package/src/modes/components/bordered-loader.ts +41 -0
- package/src/modes/components/branch-summary-message.ts +45 -0
- package/src/modes/components/btw-panel.ts +104 -0
- package/src/modes/components/chat-block.ts +111 -0
- package/src/modes/components/compaction-summary-message.ts +87 -0
- package/src/modes/components/copy-selector.ts +206 -0
- package/src/modes/components/countdown-timer.ts +75 -0
- package/src/modes/components/custom-editor.ts +398 -0
- package/src/modes/components/custom-message.ts +63 -0
- package/src/modes/components/diff.ts +277 -0
- package/src/modes/components/dynamic-border.ts +34 -0
- package/src/modes/components/error-banner.ts +33 -0
- package/src/modes/components/eval-execution.ts +158 -0
- package/src/modes/components/execution-shared.ts +101 -0
- package/src/modes/components/extensions/extension-dashboard.ts +399 -0
- package/src/modes/components/extensions/extension-list.ts +502 -0
- package/src/modes/components/extensions/index.ts +9 -0
- package/src/modes/components/extensions/inspector-panel.ts +317 -0
- package/src/modes/components/extensions/state-manager.ts +627 -0
- package/src/modes/components/extensions/types.ts +186 -0
- package/src/modes/components/footer.ts +274 -0
- package/src/modes/components/history-search.ts +280 -0
- package/src/modes/components/hook-editor.ts +167 -0
- package/src/modes/components/hook-input.ts +87 -0
- package/src/modes/components/hook-message.ts +66 -0
- package/src/modes/components/hook-selector.ts +660 -0
- package/src/modes/components/index.ts +38 -0
- package/src/modes/components/keybinding-hints.ts +65 -0
- package/src/modes/components/late-diagnostics-message.ts +60 -0
- package/src/modes/components/login-dialog.ts +164 -0
- package/src/modes/components/mcp-add-wizard.ts +1340 -0
- package/src/modes/components/message-frame.ts +88 -0
- package/src/modes/components/model-selector.ts +1271 -0
- package/src/modes/components/oauth-selector.ts +368 -0
- package/src/modes/components/omfg-panel.ts +141 -0
- package/src/modes/components/overlay-box.ts +108 -0
- package/src/modes/components/plan-review-overlay.ts +820 -0
- package/src/modes/components/plan-toc.ts +138 -0
- package/src/modes/components/plugin-selector.ts +95 -0
- package/src/modes/components/plugin-settings.ts +722 -0
- package/src/modes/components/queue-mode-selector.ts +56 -0
- package/src/modes/components/read-tool-group.ts +670 -0
- package/src/modes/components/segment-track.ts +52 -0
- package/src/modes/components/session-selector.ts +625 -0
- package/src/modes/components/settings-defs.ts +189 -0
- package/src/modes/components/settings-selector.ts +651 -0
- package/src/modes/components/show-images-selector.ts +45 -0
- package/src/modes/components/skill-message.ts +89 -0
- package/src/modes/components/status-line/component.ts +869 -0
- package/src/modes/components/status-line/context-thresholds.ts +79 -0
- package/src/modes/components/status-line/git-utils.ts +42 -0
- package/src/modes/components/status-line/index.ts +5 -0
- package/src/modes/components/status-line/presets.ts +106 -0
- package/src/modes/components/status-line/segments.ts +584 -0
- package/src/modes/components/status-line/separators.ts +55 -0
- package/src/modes/components/status-line/token-rate.ts +66 -0
- package/src/modes/components/status-line/types.ts +108 -0
- package/src/modes/components/theme-selector.ts +63 -0
- package/src/modes/components/thinking-selector.ts +52 -0
- package/src/modes/components/tiny-title-download-progress.ts +90 -0
- package/src/modes/components/tips.txt +19 -0
- package/src/modes/components/todo-reminder.ts +38 -0
- package/src/modes/components/tool-execution.ts +1024 -0
- package/src/modes/components/transcript-container.ts +608 -0
- package/src/modes/components/tree-selector.ts +978 -0
- package/src/modes/components/ttsr-notification.ts +122 -0
- package/src/modes/components/user-message-selector.ts +227 -0
- package/src/modes/components/user-message.ts +66 -0
- package/src/modes/components/visual-truncate.ts +63 -0
- package/src/modes/components/welcome.ts +493 -0
- package/src/modes/controllers/btw-controller.ts +105 -0
- package/src/modes/controllers/command-controller-shared.ts +109 -0
- package/src/modes/controllers/command-controller.ts +1566 -0
- package/src/modes/controllers/event-controller.ts +1054 -0
- package/src/modes/controllers/extension-ui-controller.ts +886 -0
- package/src/modes/controllers/input-controller.ts +1073 -0
- package/src/modes/controllers/mcp-command-controller.ts +2017 -0
- package/src/modes/controllers/omfg-controller.ts +283 -0
- package/src/modes/controllers/omfg-rule.ts +647 -0
- package/src/modes/controllers/selector-controller.ts +1108 -0
- package/src/modes/controllers/ssh-command-controller.ts +384 -0
- package/src/modes/controllers/streaming-reveal.ts +279 -0
- package/src/modes/controllers/tan-command-controller.ts +173 -0
- package/src/modes/controllers/todo-command-controller.ts +485 -0
- package/src/modes/data/emojis.json +1 -0
- package/src/modes/emoji-autocomplete.ts +285 -0
- package/src/modes/gradient-highlight.ts +87 -0
- package/src/modes/image-references.ts +117 -0
- package/src/modes/index.ts +17 -0
- package/src/modes/interactive-mode.ts +3370 -0
- package/src/modes/internal-url-autocomplete.ts +143 -0
- package/src/modes/loop-limit.ts +140 -0
- package/src/modes/magic-keywords.ts +20 -0
- package/src/modes/markdown-prose.ts +247 -0
- package/src/modes/oauth-manual-input.ts +69 -0
- package/src/modes/orchestrate.ts +42 -0
- package/src/modes/print-mode.ts +126 -0
- package/src/modes/prompt-action-autocomplete.ts +260 -0
- package/src/modes/rpc/host-tools.ts +186 -0
- package/src/modes/rpc/host-uris.ts +235 -0
- package/src/modes/rpc/rpc-client.ts +963 -0
- package/src/modes/rpc/rpc-mode.ts +947 -0
- package/src/modes/rpc/rpc-subagents.ts +265 -0
- package/src/modes/rpc/rpc-types.ts +458 -0
- package/src/modes/runtime-init.ts +116 -0
- package/src/modes/session-observer-registry.ts +146 -0
- package/src/modes/setup-version.ts +11 -0
- package/src/modes/setup-wizard/index.ts +99 -0
- package/src/modes/setup-wizard/lazy.ts +16 -0
- package/src/modes/setup-wizard/scenes/glyph.ts +96 -0
- package/src/modes/setup-wizard/scenes/outro.ts +35 -0
- package/src/modes/setup-wizard/scenes/providers.ts +69 -0
- package/src/modes/setup-wizard/scenes/sign-in.ts +205 -0
- package/src/modes/setup-wizard/scenes/splash.ts +201 -0
- package/src/modes/setup-wizard/scenes/theme.ts +299 -0
- package/src/modes/setup-wizard/scenes/types.ts +48 -0
- package/src/modes/setup-wizard/scenes/web-search.ts +129 -0
- package/src/modes/setup-wizard/wizard-overlay.ts +275 -0
- package/src/modes/shared.ts +47 -0
- package/src/modes/theme/dark.json +95 -0
- package/src/modes/theme/defaults/alabaster.json +93 -0
- package/src/modes/theme/defaults/amethyst.json +96 -0
- package/src/modes/theme/defaults/anthracite.json +93 -0
- package/src/modes/theme/defaults/basalt.json +91 -0
- package/src/modes/theme/defaults/birch.json +95 -0
- package/src/modes/theme/defaults/dark-abyss.json +91 -0
- package/src/modes/theme/defaults/dark-arctic.json +104 -0
- package/src/modes/theme/defaults/dark-aurora.json +95 -0
- package/src/modes/theme/defaults/dark-catppuccin.json +107 -0
- package/src/modes/theme/defaults/dark-cavern.json +91 -0
- package/src/modes/theme/defaults/dark-copper.json +95 -0
- package/src/modes/theme/defaults/dark-cosmos.json +90 -0
- package/src/modes/theme/defaults/dark-cyberpunk.json +102 -0
- package/src/modes/theme/defaults/dark-dracula.json +98 -0
- package/src/modes/theme/defaults/dark-eclipse.json +91 -0
- package/src/modes/theme/defaults/dark-ember.json +95 -0
- package/src/modes/theme/defaults/dark-equinox.json +90 -0
- package/src/modes/theme/defaults/dark-forest.json +96 -0
- package/src/modes/theme/defaults/dark-github.json +105 -0
- package/src/modes/theme/defaults/dark-gruvbox.json +112 -0
- package/src/modes/theme/defaults/dark-lavender.json +95 -0
- package/src/modes/theme/defaults/dark-lunar.json +89 -0
- package/src/modes/theme/defaults/dark-midnight.json +95 -0
- package/src/modes/theme/defaults/dark-monochrome.json +94 -0
- package/src/modes/theme/defaults/dark-monokai.json +98 -0
- package/src/modes/theme/defaults/dark-nebula.json +90 -0
- package/src/modes/theme/defaults/dark-nord.json +97 -0
- package/src/modes/theme/defaults/dark-ocean.json +101 -0
- package/src/modes/theme/defaults/dark-one.json +100 -0
- package/src/modes/theme/defaults/dark-poimandres.json +142 -0
- package/src/modes/theme/defaults/dark-rainforest.json +91 -0
- package/src/modes/theme/defaults/dark-reef.json +91 -0
- package/src/modes/theme/defaults/dark-retro.json +92 -0
- package/src/modes/theme/defaults/dark-rose-pine.json +96 -0
- package/src/modes/theme/defaults/dark-sakura.json +95 -0
- package/src/modes/theme/defaults/dark-slate.json +95 -0
- package/src/modes/theme/defaults/dark-solarized.json +97 -0
- package/src/modes/theme/defaults/dark-solstice.json +90 -0
- package/src/modes/theme/defaults/dark-starfall.json +91 -0
- package/src/modes/theme/defaults/dark-sunset.json +99 -0
- package/src/modes/theme/defaults/dark-swamp.json +90 -0
- package/src/modes/theme/defaults/dark-synthwave.json +103 -0
- package/src/modes/theme/defaults/dark-taiga.json +91 -0
- package/src/modes/theme/defaults/dark-terminal.json +95 -0
- package/src/modes/theme/defaults/dark-tokyo-night.json +101 -0
- package/src/modes/theme/defaults/dark-tundra.json +91 -0
- package/src/modes/theme/defaults/dark-twilight.json +91 -0
- package/src/modes/theme/defaults/dark-volcanic.json +91 -0
- package/src/modes/theme/defaults/graphite.json +92 -0
- package/src/modes/theme/defaults/index.ts +199 -0
- package/src/modes/theme/defaults/light-arctic.json +107 -0
- package/src/modes/theme/defaults/light-aurora-day.json +91 -0
- package/src/modes/theme/defaults/light-canyon.json +91 -0
- package/src/modes/theme/defaults/light-catppuccin.json +106 -0
- package/src/modes/theme/defaults/light-cirrus.json +90 -0
- package/src/modes/theme/defaults/light-coral.json +95 -0
- package/src/modes/theme/defaults/light-cyberpunk.json +96 -0
- package/src/modes/theme/defaults/light-dawn.json +90 -0
- package/src/modes/theme/defaults/light-dunes.json +91 -0
- package/src/modes/theme/defaults/light-eucalyptus.json +95 -0
- package/src/modes/theme/defaults/light-forest.json +100 -0
- package/src/modes/theme/defaults/light-frost.json +95 -0
- package/src/modes/theme/defaults/light-github.json +115 -0
- package/src/modes/theme/defaults/light-glacier.json +91 -0
- package/src/modes/theme/defaults/light-gruvbox.json +108 -0
- package/src/modes/theme/defaults/light-haze.json +90 -0
- package/src/modes/theme/defaults/light-honeycomb.json +95 -0
- package/src/modes/theme/defaults/light-lagoon.json +91 -0
- package/src/modes/theme/defaults/light-lavender.json +95 -0
- package/src/modes/theme/defaults/light-meadow.json +91 -0
- package/src/modes/theme/defaults/light-mint.json +95 -0
- package/src/modes/theme/defaults/light-monochrome.json +101 -0
- package/src/modes/theme/defaults/light-ocean.json +99 -0
- package/src/modes/theme/defaults/light-one.json +99 -0
- package/src/modes/theme/defaults/light-opal.json +91 -0
- package/src/modes/theme/defaults/light-orchard.json +91 -0
- package/src/modes/theme/defaults/light-paper.json +95 -0
- package/src/modes/theme/defaults/light-poimandres.json +142 -0
- package/src/modes/theme/defaults/light-prism.json +90 -0
- package/src/modes/theme/defaults/light-retro.json +98 -0
- package/src/modes/theme/defaults/light-sand.json +95 -0
- package/src/modes/theme/defaults/light-savanna.json +91 -0
- package/src/modes/theme/defaults/light-solarized.json +102 -0
- package/src/modes/theme/defaults/light-soleil.json +90 -0
- package/src/modes/theme/defaults/light-sunset.json +99 -0
- package/src/modes/theme/defaults/light-synthwave.json +98 -0
- package/src/modes/theme/defaults/light-tokyo-night.json +111 -0
- package/src/modes/theme/defaults/light-wetland.json +91 -0
- package/src/modes/theme/defaults/light-zenith.json +89 -0
- package/src/modes/theme/defaults/limestone.json +94 -0
- package/src/modes/theme/defaults/mahogany.json +97 -0
- package/src/modes/theme/defaults/marble.json +93 -0
- package/src/modes/theme/defaults/obsidian.json +91 -0
- package/src/modes/theme/defaults/onyx.json +91 -0
- package/src/modes/theme/defaults/pearl.json +93 -0
- package/src/modes/theme/defaults/porcelain.json +91 -0
- package/src/modes/theme/defaults/quartz.json +96 -0
- package/src/modes/theme/defaults/sandstone.json +95 -0
- package/src/modes/theme/defaults/titanium.json +90 -0
- package/src/modes/theme/light.json +93 -0
- package/src/modes/theme/mermaid-cache.ts +29 -0
- package/src/modes/theme/shimmer.ts +235 -0
- package/src/modes/theme/theme-schema.json +459 -0
- package/src/modes/theme/theme.ts +2676 -0
- package/src/modes/turn-budget.ts +31 -0
- package/src/modes/types.ts +359 -0
- package/src/modes/ultrathink.ts +41 -0
- package/src/modes/utils/context-usage.ts +339 -0
- package/src/modes/utils/copy-targets.ts +360 -0
- package/src/modes/utils/hotkeys-markdown.ts +61 -0
- package/src/modes/utils/keybinding-matchers.ts +51 -0
- package/src/modes/utils/tools-markdown.ts +27 -0
- package/src/modes/utils/ui-helpers.ts +801 -0
- package/src/modes/workflow.ts +42 -0
- package/src/plan-mode/approved-plan.ts +186 -0
- package/src/plan-mode/plan-handoff.ts +37 -0
- package/src/plan-mode/plan-protection.ts +31 -0
- package/src/plan-mode/state.ts +6 -0
- package/src/priority.json +41 -0
- package/src/prompts/agents/designer.md +66 -0
- package/src/prompts/agents/explore.md +58 -0
- package/src/prompts/agents/frontmatter.md +11 -0
- package/src/prompts/agents/init.md +33 -0
- package/src/prompts/agents/librarian.md +119 -0
- package/src/prompts/agents/oracle.md +55 -0
- package/src/prompts/agents/plan.md +48 -0
- package/src/prompts/agents/reviewer.md +140 -0
- package/src/prompts/agents/task.md +16 -0
- package/src/prompts/ci-green-request.md +36 -0
- package/src/prompts/dry-balance-bench.md +8 -0
- package/src/prompts/goals/goal-budget-limit.md +16 -0
- package/src/prompts/goals/goal-continuation.md +28 -0
- package/src/prompts/goals/goal-mode-active.md +23 -0
- package/src/prompts/memories/consolidation.md +30 -0
- package/src/prompts/memories/read-path.md +11 -0
- package/src/prompts/memories/stage_one_input.md +6 -0
- package/src/prompts/memories/stage_one_system.md +21 -0
- package/src/prompts/review-custom-request.md +22 -0
- package/src/prompts/review-headless-request.md +16 -0
- package/src/prompts/review-request.md +69 -0
- package/src/prompts/steering/user-interjection.md +10 -0
- package/src/prompts/system/agent-creation-architect.md +50 -0
- package/src/prompts/system/agent-creation-user.md +6 -0
- package/src/prompts/system/auto-continue.md +1 -0
- package/src/prompts/system/auto-thinking-difficulty-local.md +14 -0
- package/src/prompts/system/auto-thinking-difficulty.md +12 -0
- package/src/prompts/system/background-tan-dispatch.md +8 -0
- package/src/prompts/system/btw-user.md +8 -0
- package/src/prompts/system/commit-message-system.md +14 -0
- package/src/prompts/system/custom-system-prompt.md +64 -0
- package/src/prompts/system/eager-todo.md +13 -0
- package/src/prompts/system/empty-stop-retry.md +6 -0
- package/src/prompts/system/irc-incoming.md +7 -0
- package/src/prompts/system/manual-continue.md +7 -0
- package/src/prompts/system/memory-consolidation-system.md +8 -0
- package/src/prompts/system/memory-extraction-system.md +26 -0
- package/src/prompts/system/omfg-user.md +50 -0
- package/src/prompts/system/orchestrate-notice.md +40 -0
- package/src/prompts/system/plan-mode-active.md +109 -0
- package/src/prompts/system/plan-mode-approved.md +25 -0
- package/src/prompts/system/plan-mode-compact-instructions.md +16 -0
- package/src/prompts/system/plan-mode-reference.md +11 -0
- package/src/prompts/system/plan-mode-subagent.md +33 -0
- package/src/prompts/system/plan-mode-tool-decision-reminder.md +9 -0
- package/src/prompts/system/project-prompt.md +52 -0
- package/src/prompts/system/subagent-system-prompt.md +64 -0
- package/src/prompts/system/subagent-user-prompt.md +3 -0
- package/src/prompts/system/subagent-yield-reminder.md +12 -0
- package/src/prompts/system/system-prompt.md +258 -0
- package/src/prompts/system/tiny-title-system.md +8 -0
- package/src/prompts/system/title-system.md +16 -0
- package/src/prompts/system/ttsr-interrupt.md +7 -0
- package/src/prompts/system/ttsr-tool-reminder.md +5 -0
- package/src/prompts/system/ultrathink-notice.md +3 -0
- package/src/prompts/system/web-search.md +25 -0
- package/src/prompts/system/workflow-notice.md +70 -0
- package/src/prompts/tools/apply-patch.md +65 -0
- package/src/prompts/tools/ask.md +30 -0
- package/src/prompts/tools/ast-edit.md +39 -0
- package/src/prompts/tools/ast-grep.md +42 -0
- package/src/prompts/tools/async-result.md +8 -0
- package/src/prompts/tools/bash.md +46 -0
- package/src/prompts/tools/browser.md +73 -0
- package/src/prompts/tools/checkpoint.md +16 -0
- package/src/prompts/tools/debug.md +34 -0
- package/src/prompts/tools/eval.md +92 -0
- package/src/prompts/tools/find.md +36 -0
- package/src/prompts/tools/github.md +21 -0
- package/src/prompts/tools/goal.md +18 -0
- package/src/prompts/tools/image-gen.md +7 -0
- package/src/prompts/tools/inspect-image-system.md +20 -0
- package/src/prompts/tools/inspect-image.md +32 -0
- package/src/prompts/tools/irc.md +59 -0
- package/src/prompts/tools/job.md +19 -0
- package/src/prompts/tools/lsp-late-diagnostic.md +8 -0
- package/src/prompts/tools/lsp.md +42 -0
- package/src/prompts/tools/memory-edit.md +8 -0
- package/src/prompts/tools/patch.md +70 -0
- package/src/prompts/tools/read.md +84 -0
- package/src/prompts/tools/recall.md +5 -0
- package/src/prompts/tools/reflect.md +5 -0
- package/src/prompts/tools/render-mermaid.md +9 -0
- package/src/prompts/tools/replace.md +30 -0
- package/src/prompts/tools/resolve.md +9 -0
- package/src/prompts/tools/retain.md +6 -0
- package/src/prompts/tools/rewind.md +13 -0
- package/src/prompts/tools/search-tool-bm25.md +32 -0
- package/src/prompts/tools/search.md +24 -0
- package/src/prompts/tools/ssh.md +31 -0
- package/src/prompts/tools/task-summary.md +17 -0
- package/src/prompts/tools/task.md +88 -0
- package/src/prompts/tools/todo.md +62 -0
- package/src/prompts/tools/web-search.md +10 -0
- package/src/prompts/tools/write.md +14 -0
- package/src/registry/agent-lifecycle.ts +218 -0
- package/src/registry/agent-registry.ts +151 -0
- package/src/sdk.ts +2558 -0
- package/src/secrets/index.ts +123 -0
- package/src/secrets/obfuscator.ts +298 -0
- package/src/secrets/regex.ts +21 -0
- package/src/session/agent-session.ts +10121 -0
- package/src/session/agent-storage.ts +455 -0
- package/src/session/artifacts.ts +135 -0
- package/src/session/auth-broker-config.ts +131 -0
- package/src/session/auth-storage.ts +29 -0
- package/src/session/blob-store.ts +255 -0
- package/src/session/client-bridge.ts +85 -0
- package/src/session/history-storage.ts +348 -0
- package/src/session/indexed-session-storage.ts +430 -0
- package/src/session/messages.ts +541 -0
- package/src/session/redis-session-storage.ts +170 -0
- package/src/session/session-dump-format.ts +209 -0
- package/src/session/session-history-format.ts +246 -0
- package/src/session/session-manager.ts +3676 -0
- package/src/session/session-storage.ts +529 -0
- package/src/session/shake-types.ts +43 -0
- package/src/session/sql-session-storage.ts +314 -0
- package/src/session/streaming-output.ts +1330 -0
- package/src/session/tool-choice-queue.ts +213 -0
- package/src/session/yield-queue.ts +173 -0
- package/src/slash-commands/acp-builtins.ts +70 -0
- package/src/slash-commands/builtin-registry.ts +1798 -0
- package/src/slash-commands/helpers/context-report.ts +39 -0
- package/src/slash-commands/helpers/format.ts +46 -0
- package/src/slash-commands/helpers/marketplace-manager.ts +25 -0
- package/src/slash-commands/helpers/mcp.ts +532 -0
- package/src/slash-commands/helpers/parse.ts +85 -0
- package/src/slash-commands/helpers/ssh.ts +195 -0
- package/src/slash-commands/helpers/stats-dashboard.ts +85 -0
- package/src/slash-commands/helpers/todo.ts +279 -0
- package/src/slash-commands/helpers/usage-report.ts +95 -0
- package/src/slash-commands/marketplace-install-parser.ts +99 -0
- package/src/slash-commands/types.ts +135 -0
- package/src/ssh/config-writer.ts +183 -0
- package/src/ssh/connection-manager.ts +509 -0
- package/src/ssh/ssh-executor.ts +189 -0
- package/src/ssh/sshfs-mount.ts +140 -0
- package/src/ssh/utils.ts +8 -0
- package/src/stt/downloader.ts +71 -0
- package/src/stt/index.ts +3 -0
- package/src/stt/recorder.ts +351 -0
- package/src/stt/setup.ts +52 -0
- package/src/stt/stt-controller.ts +160 -0
- package/src/stt/transcribe.py +70 -0
- package/src/stt/transcriber.ts +91 -0
- package/src/stubs/natives/index.ts +814 -0
- package/src/stubs/natives/package.json +7 -0
- package/src/stubs/tui/index.ts +282 -0
- package/src/stubs/tui/package.json +7 -0
- package/src/system-prompt.ts +611 -0
- package/src/task/agents.ts +167 -0
- package/src/task/commands.ts +132 -0
- package/src/task/discovery.ts +122 -0
- package/src/task/executor.ts +2133 -0
- package/src/task/index.ts +1419 -0
- package/src/task/name-generator.ts +1577 -0
- package/src/task/omp-command.ts +26 -0
- package/src/task/output-manager.ts +88 -0
- package/src/task/parallel.ts +116 -0
- package/src/task/render.ts +1381 -0
- package/src/task/repair-args.ts +129 -0
- package/src/task/subprocess-tool-registry.ts +88 -0
- package/src/task/types.ts +336 -0
- package/src/task/worktree.ts +514 -0
- package/src/telemetry-export.ts +144 -0
- package/src/thinking.ts +167 -0
- package/src/tiny/compiled-runtime.ts +179 -0
- package/src/tiny/device.ts +111 -0
- package/src/tiny/dtype.ts +101 -0
- package/src/tiny/models.ts +242 -0
- package/src/tiny/text.ts +165 -0
- package/src/tiny/title-client.ts +543 -0
- package/src/tiny/title-protocol.ts +56 -0
- package/src/tiny/worker.ts +568 -0
- package/src/tool-discovery/mode.ts +24 -0
- package/src/tool-discovery/tool-index.ts +256 -0
- package/src/tools/approval.ts +189 -0
- package/src/tools/archive-reader.ts +721 -0
- package/src/tools/ask.ts +928 -0
- package/src/tools/ast-edit.ts +642 -0
- package/src/tools/ast-grep.ts +452 -0
- package/src/tools/auto-generated-guard.ts +322 -0
- package/src/tools/bash-command-fixup.ts +37 -0
- package/src/tools/bash-interactive.ts +408 -0
- package/src/tools/bash-interceptor.ts +67 -0
- package/src/tools/bash-pty-selection.ts +14 -0
- package/src/tools/bash-skill-urls.ts +248 -0
- package/src/tools/bash.ts +1386 -0
- package/src/tools/browser/attach.ts +175 -0
- package/src/tools/browser/launch.ts +660 -0
- package/src/tools/browser/readable.ts +112 -0
- package/src/tools/browser/registry.ts +197 -0
- package/src/tools/browser/render.ts +216 -0
- package/src/tools/browser/tab-protocol.ts +105 -0
- package/src/tools/browser/tab-supervisor.ts +628 -0
- package/src/tools/browser/tab-worker-entry.ts +21 -0
- package/src/tools/browser/tab-worker.ts +1226 -0
- package/src/tools/browser.ts +343 -0
- package/src/tools/checkpoint.ts +136 -0
- package/src/tools/conflict-detect.ts +718 -0
- package/src/tools/context.ts +39 -0
- package/src/tools/debug.ts +1067 -0
- package/src/tools/eval-backends.ts +27 -0
- package/src/tools/eval-render.ts +752 -0
- package/src/tools/eval.ts +577 -0
- package/src/tools/fetch.ts +1926 -0
- package/src/tools/file-recorder.ts +35 -0
- package/src/tools/find.ts +609 -0
- package/src/tools/fs-cache-invalidation.ts +28 -0
- package/src/tools/gh-cache-invalidation.ts +255 -0
- package/src/tools/gh-format.ts +12 -0
- package/src/tools/gh-renderer.ts +481 -0
- package/src/tools/gh.ts +3720 -0
- package/src/tools/github-cache.ts +637 -0
- package/src/tools/grouped-file-output.ts +210 -0
- package/src/tools/image-gen.ts +1517 -0
- package/src/tools/index.ts +599 -0
- package/src/tools/inspect-image-renderer.ts +132 -0
- package/src/tools/inspect-image.ts +174 -0
- package/src/tools/irc.ts +723 -0
- package/src/tools/job.ts +557 -0
- package/src/tools/json-tree.ts +243 -0
- package/src/tools/jtd-to-json-schema.ts +219 -0
- package/src/tools/jtd-to-typescript.ts +136 -0
- package/src/tools/jtd-utils.ts +102 -0
- package/src/tools/list-limit.ts +40 -0
- package/src/tools/match-line-format.ts +20 -0
- package/src/tools/memory-edit.ts +59 -0
- package/src/tools/memory-recall.ts +100 -0
- package/src/tools/memory-reflect.ts +88 -0
- package/src/tools/memory-render.ts +202 -0
- package/src/tools/memory-retain.ts +91 -0
- package/src/tools/output-meta.ts +754 -0
- package/src/tools/output-schema-validator.ts +132 -0
- package/src/tools/path-utils.ts +1054 -0
- package/src/tools/plan-mode-guard.ts +108 -0
- package/src/tools/puppeteer/00_stealth_tampering.txt +63 -0
- package/src/tools/puppeteer/01_stealth_activity.txt +20 -0
- package/src/tools/puppeteer/02_stealth_hairline.txt +11 -0
- package/src/tools/puppeteer/03_stealth_botd.txt +384 -0
- package/src/tools/puppeteer/04_stealth_iframe.txt +81 -0
- package/src/tools/puppeteer/05_stealth_webgl.txt +75 -0
- package/src/tools/puppeteer/06_stealth_screen.txt +72 -0
- package/src/tools/puppeteer/07_stealth_fonts.txt +97 -0
- package/src/tools/puppeteer/08_stealth_audio.txt +51 -0
- package/src/tools/puppeteer/09_stealth_locale.txt +46 -0
- package/src/tools/puppeteer/10_stealth_plugins.txt +206 -0
- package/src/tools/puppeteer/11_stealth_hardware.txt +8 -0
- package/src/tools/puppeteer/12_stealth_codecs.txt +40 -0
- package/src/tools/puppeteer/13_stealth_worker.txt +74 -0
- package/src/tools/read.ts +2929 -0
- package/src/tools/render-mermaid.ts +69 -0
- package/src/tools/render-utils.ts +838 -0
- package/src/tools/renderers.ts +77 -0
- package/src/tools/report-tool-issue.ts +534 -0
- package/src/tools/resolve.ts +276 -0
- package/src/tools/review.ts +253 -0
- package/src/tools/search-tool-bm25.ts +351 -0
- package/src/tools/search.ts +1580 -0
- package/src/tools/sqlite-reader.ts +828 -0
- package/src/tools/ssh.ts +349 -0
- package/src/tools/todo.ts +982 -0
- package/src/tools/tool-errors.ts +62 -0
- package/src/tools/tool-result.ts +94 -0
- package/src/tools/tool-timeouts.ts +30 -0
- package/src/tools/tts.ts +133 -0
- package/src/tools/write.ts +1217 -0
- package/src/tools/yield.ts +269 -0
- package/src/tui/code-cell.ts +216 -0
- package/src/tui/file-list.ts +55 -0
- package/src/tui/hyperlink.ts +175 -0
- package/src/tui/index.ts +12 -0
- package/src/tui/output-block.ts +240 -0
- package/src/tui/status-line.ts +54 -0
- package/src/tui/tree-list.ts +84 -0
- package/src/tui/types.ts +15 -0
- package/src/tui/utils.ts +103 -0
- package/src/utils/block-context.ts +312 -0
- package/src/utils/changelog.ts +132 -0
- package/src/utils/clipboard.ts +193 -0
- package/src/utils/command-args.ts +76 -0
- package/src/utils/commit-message-generator.ts +151 -0
- package/src/utils/edit-mode.ts +41 -0
- package/src/utils/enhanced-paste.ts +230 -0
- package/src/utils/event-bus.ts +33 -0
- package/src/utils/external-editor.ts +65 -0
- package/src/utils/file-display-mode.ts +45 -0
- package/src/utils/file-mentions.ts +281 -0
- package/src/utils/git.ts +1833 -0
- package/src/utils/image-loading.ts +132 -0
- package/src/utils/image-resize.ts +309 -0
- package/src/utils/jj.ts +248 -0
- package/src/utils/lang-from-path.ts +239 -0
- package/src/utils/markit.ts +89 -0
- package/src/utils/open.ts +55 -0
- package/src/utils/session-color.ts +68 -0
- package/src/utils/shell-snapshot.ts +187 -0
- package/src/utils/sixel.ts +69 -0
- package/src/utils/title-generator.ts +373 -0
- package/src/utils/tool-choice.ts +33 -0
- package/src/utils/tools-manager.ts +363 -0
- package/src/web/kagi.ts +305 -0
- package/src/web/parallel.ts +353 -0
- package/src/web/scrapers/artifacthub.ts +207 -0
- package/src/web/scrapers/arxiv.ts +83 -0
- package/src/web/scrapers/aur.ts +162 -0
- package/src/web/scrapers/biorxiv.ts +133 -0
- package/src/web/scrapers/bluesky.ts +262 -0
- package/src/web/scrapers/brew.ts +172 -0
- package/src/web/scrapers/cheatsh.ts +68 -0
- package/src/web/scrapers/chocolatey.ts +196 -0
- package/src/web/scrapers/choosealicense.ts +95 -0
- package/src/web/scrapers/cisa-kev.ts +87 -0
- package/src/web/scrapers/clojars.ts +154 -0
- package/src/web/scrapers/coingecko.ts +177 -0
- package/src/web/scrapers/crates-io.ts +97 -0
- package/src/web/scrapers/crossref.ts +136 -0
- package/src/web/scrapers/devto.ts +147 -0
- package/src/web/scrapers/discogs.ts +306 -0
- package/src/web/scrapers/discourse.ts +197 -0
- package/src/web/scrapers/dockerhub.ts +138 -0
- package/src/web/scrapers/docs-rs.ts +653 -0
- package/src/web/scrapers/fdroid.ts +134 -0
- package/src/web/scrapers/firefox-addons.ts +191 -0
- package/src/web/scrapers/flathub.ts +223 -0
- package/src/web/scrapers/github-gist.ts +58 -0
- package/src/web/scrapers/github.ts +704 -0
- package/src/web/scrapers/gitlab.ts +401 -0
- package/src/web/scrapers/go-pkg.ts +266 -0
- package/src/web/scrapers/hackage.ts +140 -0
- package/src/web/scrapers/hackernews.ts +189 -0
- package/src/web/scrapers/hex.ts +105 -0
- package/src/web/scrapers/huggingface.ts +321 -0
- package/src/web/scrapers/iacr.ts +89 -0
- package/src/web/scrapers/index.ts +252 -0
- package/src/web/scrapers/jetbrains-marketplace.ts +159 -0
- package/src/web/scrapers/lemmy.ts +203 -0
- package/src/web/scrapers/lobsters.ts +175 -0
- package/src/web/scrapers/mastodon.ts +292 -0
- package/src/web/scrapers/maven.ts +138 -0
- package/src/web/scrapers/mdn.ts +173 -0
- package/src/web/scrapers/metacpan.ts +222 -0
- package/src/web/scrapers/musicbrainz.ts +250 -0
- package/src/web/scrapers/npm.ts +98 -0
- package/src/web/scrapers/nuget.ts +183 -0
- package/src/web/scrapers/nvd.ts +222 -0
- package/src/web/scrapers/ollama.ts +239 -0
- package/src/web/scrapers/open-vsx.ts +106 -0
- package/src/web/scrapers/opencorporates.ts +292 -0
- package/src/web/scrapers/openlibrary.ts +336 -0
- package/src/web/scrapers/orcid.ts +286 -0
- package/src/web/scrapers/osv.ts +176 -0
- package/src/web/scrapers/packagist.ts +160 -0
- package/src/web/scrapers/pub-dev.ts +143 -0
- package/src/web/scrapers/pubmed.ts +211 -0
- package/src/web/scrapers/pypi.ts +112 -0
- package/src/web/scrapers/rawg.ts +110 -0
- package/src/web/scrapers/readthedocs.ts +120 -0
- package/src/web/scrapers/reddit.ts +95 -0
- package/src/web/scrapers/repology.ts +251 -0
- package/src/web/scrapers/rfc.ts +201 -0
- package/src/web/scrapers/rubygems.ts +103 -0
- package/src/web/scrapers/searchcode.ts +189 -0
- package/src/web/scrapers/sec-edgar.ts +261 -0
- package/src/web/scrapers/semantic-scholar.ts +171 -0
- package/src/web/scrapers/snapcraft.ts +187 -0
- package/src/web/scrapers/sourcegraph.ts +336 -0
- package/src/web/scrapers/spdx.ts +108 -0
- package/src/web/scrapers/spotify.ts +198 -0
- package/src/web/scrapers/stackoverflow.ts +120 -0
- package/src/web/scrapers/terraform.ts +277 -0
- package/src/web/scrapers/tldr.ts +47 -0
- package/src/web/scrapers/twitter.ts +94 -0
- package/src/web/scrapers/types.ts +397 -0
- package/src/web/scrapers/utils.ts +109 -0
- package/src/web/scrapers/vimeo.ts +133 -0
- package/src/web/scrapers/vscode-marketplace.ts +187 -0
- package/src/web/scrapers/w3c.ts +156 -0
- package/src/web/scrapers/wikidata.ts +344 -0
- package/src/web/scrapers/wikipedia.ts +84 -0
- package/src/web/scrapers/youtube.ts +325 -0
- package/src/web/search/index.ts +292 -0
- package/src/web/search/provider.ts +157 -0
- package/src/web/search/providers/anthropic.ts +318 -0
- package/src/web/search/providers/base.ts +89 -0
- package/src/web/search/providers/brave.ts +152 -0
- package/src/web/search/providers/codex.ts +591 -0
- package/src/web/search/providers/exa.ts +400 -0
- package/src/web/search/providers/gemini.ts +460 -0
- package/src/web/search/providers/jina.ts +111 -0
- package/src/web/search/providers/kagi.ts +86 -0
- package/src/web/search/providers/kimi.ts +196 -0
- package/src/web/search/providers/parallel.ts +225 -0
- package/src/web/search/providers/perplexity.ts +730 -0
- package/src/web/search/providers/searxng.ts +313 -0
- package/src/web/search/providers/synthetic.ts +114 -0
- package/src/web/search/providers/tavily.ts +176 -0
- package/src/web/search/providers/utils.ts +128 -0
- package/src/web/search/providers/zai.ts +333 -0
- package/src/web/search/render.ts +262 -0
- package/src/web/search/types.ts +482 -0
- package/src/web/search/utils.ts +17 -0
- package/src/workspace-tree.ts +286 -0
package/src/sdk.ts
ADDED
|
@@ -0,0 +1,2558 @@
|
|
|
1
|
+
import {
|
|
2
|
+
Agent,
|
|
3
|
+
type AgentEvent,
|
|
4
|
+
type AgentMessage,
|
|
5
|
+
type AgentTelemetryConfig,
|
|
6
|
+
type AgentTool,
|
|
7
|
+
AppendOnlyContextManager,
|
|
8
|
+
INTENT_FIELD,
|
|
9
|
+
type ThinkingLevel,
|
|
10
|
+
} from "@oh-my-pi/pi-agent-core";
|
|
11
|
+
import {
|
|
12
|
+
type CredentialDisabledEvent,
|
|
13
|
+
type Message,
|
|
14
|
+
type Model,
|
|
15
|
+
type SimpleStreamOptions,
|
|
16
|
+
streamSimple,
|
|
17
|
+
} from "@oh-my-pi/pi-ai";
|
|
18
|
+
import {
|
|
19
|
+
getOpenAICodexTransportDetails,
|
|
20
|
+
prewarmOpenAICodexResponses,
|
|
21
|
+
} from "@oh-my-pi/pi-ai/providers/openai-codex-responses";
|
|
22
|
+
import { DEFAULT_MODEL_PER_PROVIDER } from "@oh-my-pi/pi-catalog/provider-models";
|
|
23
|
+
import type { Component } from './stubs/tui/index.ts';
|
|
24
|
+
import {
|
|
25
|
+
$env,
|
|
26
|
+
$flag,
|
|
27
|
+
getAgentDbPath,
|
|
28
|
+
getAgentDir,
|
|
29
|
+
getAuthBrokerSnapshotCachePath,
|
|
30
|
+
getProjectDir,
|
|
31
|
+
logger,
|
|
32
|
+
postmortem,
|
|
33
|
+
prompt,
|
|
34
|
+
Snowflake,
|
|
35
|
+
} from "@oh-my-pi/pi-utils";
|
|
36
|
+
import chalk from "chalk";
|
|
37
|
+
import { type AsyncJob, AsyncJobManager } from "./async";
|
|
38
|
+
import { loadCapability } from "./capability";
|
|
39
|
+
import { type Rule, ruleCapability, setActiveRules } from "./capability/rule";
|
|
40
|
+
import { bucketRules } from "./capability/rule-buckets";
|
|
41
|
+
import { createApiKeyResolver } from "./config/api-key-resolver";
|
|
42
|
+
import { shouldEnableAppendOnlyContext } from "./config/append-only-context-mode";
|
|
43
|
+
import { ModelRegistry } from "./config/model-registry";
|
|
44
|
+
import {
|
|
45
|
+
formatModelString,
|
|
46
|
+
getModelMatchPreferences,
|
|
47
|
+
parseModelPattern,
|
|
48
|
+
parseModelString,
|
|
49
|
+
resolveAllowedModels,
|
|
50
|
+
resolveModelRoleValue,
|
|
51
|
+
} from "./config/model-resolver";
|
|
52
|
+
import { loadPromptTemplates as loadPromptTemplatesInternal, type PromptTemplate } from "./config/prompt-templates";
|
|
53
|
+
import { Settings, type SkillsSettings } from "./config/settings";
|
|
54
|
+
import { CursorExecHandlers } from "./cursor";
|
|
55
|
+
import "./discovery";
|
|
56
|
+
import { resolveConfigValue } from "./config/resolve-config-value";
|
|
57
|
+
import { initializeWithSettings } from "./discovery";
|
|
58
|
+
import { disposeAllKernelSessions, disposeKernelSessionsByOwner } from "./eval/py/executor";
|
|
59
|
+
import { defaultEvalSessionId } from "./eval/session-id";
|
|
60
|
+
import {
|
|
61
|
+
type CustomCommandsLoadResult,
|
|
62
|
+
type LoadedCustomCommand,
|
|
63
|
+
loadCustomCommands as loadCustomCommandsInternal,
|
|
64
|
+
} from "./extensibility/custom-commands";
|
|
65
|
+
import { discoverCustomToolPaths, loadCustomTools, type ToolPathWithSource } from "./extensibility/custom-tools";
|
|
66
|
+
import type { CustomTool, CustomToolContext, CustomToolSessionEvent } from "./extensibility/custom-tools/types";
|
|
67
|
+
import {
|
|
68
|
+
discoverAndLoadExtensions,
|
|
69
|
+
discoverExtensionPaths,
|
|
70
|
+
type ExtensionContext,
|
|
71
|
+
type ExtensionFactory,
|
|
72
|
+
ExtensionRunner,
|
|
73
|
+
ExtensionToolWrapper,
|
|
74
|
+
type ExtensionUIContext,
|
|
75
|
+
type LoadExtensionsResult,
|
|
76
|
+
loadExtensionFromFactory,
|
|
77
|
+
loadExtensions,
|
|
78
|
+
type ToolDefinition,
|
|
79
|
+
wrapRegisteredTools,
|
|
80
|
+
} from "./extensibility/extensions";
|
|
81
|
+
import {
|
|
82
|
+
loadSkills as loadSkillsInternal,
|
|
83
|
+
type Skill,
|
|
84
|
+
type SkillWarning,
|
|
85
|
+
setActiveSkills,
|
|
86
|
+
} from "./extensibility/skills";
|
|
87
|
+
import { type FileSlashCommand, loadSlashCommands as loadSlashCommandsInternal } from "./extensibility/slash-commands";
|
|
88
|
+
import type { HindsightSessionState } from "./hindsight/state";
|
|
89
|
+
import { LocalProtocolHandler, type LocalProtocolOptions } from "./internal-urls";
|
|
90
|
+
import { LSP_STARTUP_EVENT_CHANNEL, type LspStartupEvent } from "./lsp/startup-events";
|
|
91
|
+
import { discoverAndLoadMCPTools, MCPManager, type MCPToolsLoadResult } from "./mcp";
|
|
92
|
+
import { createSessionMemoryRuntimeContext, resolveMemoryBackend } from "./memory-backend";
|
|
93
|
+
import type { MnemopiSessionState } from "./mnemopi/state";
|
|
94
|
+
import asyncResultTemplate from "./prompts/tools/async-result.md" with { type: "text" };
|
|
95
|
+
import lateDiagnosticTemplate from "./prompts/tools/lsp-late-diagnostic.md" with { type: "text" };
|
|
96
|
+
import { AgentLifecycleManager } from "./registry/agent-lifecycle";
|
|
97
|
+
import { AgentRegistry, MAIN_AGENT_ID } from "./registry/agent-registry";
|
|
98
|
+
import {
|
|
99
|
+
collectEnvSecrets,
|
|
100
|
+
deobfuscateSessionContext,
|
|
101
|
+
loadSecrets,
|
|
102
|
+
obfuscateMessages,
|
|
103
|
+
obfuscateProviderContext,
|
|
104
|
+
SecretObfuscator,
|
|
105
|
+
} from "./secrets";
|
|
106
|
+
import { AgentSession } from "./session/agent-session";
|
|
107
|
+
import { resolveAuthBrokerConfig } from "./session/auth-broker-config";
|
|
108
|
+
import {
|
|
109
|
+
AuthBrokerClient,
|
|
110
|
+
AuthStorage,
|
|
111
|
+
DEFAULT_SNAPSHOT_CACHE_TTL_MS,
|
|
112
|
+
RemoteAuthCredentialStore,
|
|
113
|
+
readAuthBrokerSnapshotCache,
|
|
114
|
+
type SnapshotResponse,
|
|
115
|
+
writeAuthBrokerSnapshotCache,
|
|
116
|
+
} from "./session/auth-storage";
|
|
117
|
+
import {
|
|
118
|
+
type CustomMessage,
|
|
119
|
+
convertToLlm,
|
|
120
|
+
LSP_LATE_DIAGNOSTIC_MESSAGE_TYPE,
|
|
121
|
+
wrapSteeringForModel,
|
|
122
|
+
} from "./session/messages";
|
|
123
|
+
import { getRestorableSessionModels, SessionManager } from "./session/session-manager";
|
|
124
|
+
import { closeAllConnections } from "./ssh/connection-manager";
|
|
125
|
+
import { unmountAll } from "./ssh/sshfs-mount";
|
|
126
|
+
import {
|
|
127
|
+
type BuildSystemPromptResult,
|
|
128
|
+
buildSystemPrompt as buildSystemPromptInternal,
|
|
129
|
+
buildSystemPromptToolMetadata,
|
|
130
|
+
loadProjectContextFiles as loadContextFilesInternal,
|
|
131
|
+
} from "./system-prompt";
|
|
132
|
+
import { AgentOutputManager } from "./task/output-manager";
|
|
133
|
+
import {
|
|
134
|
+
AUTO_THINKING,
|
|
135
|
+
type ConfiguredThinkingLevel,
|
|
136
|
+
parseThinkingLevel,
|
|
137
|
+
resolveProvisionalAutoLevel,
|
|
138
|
+
resolveThinkingLevelForModel,
|
|
139
|
+
shouldDisableReasoning,
|
|
140
|
+
toReasoningEffort,
|
|
141
|
+
} from "./thinking";
|
|
142
|
+
import { countToolsForAutoDiscovery, resolveEffectiveToolDiscoveryMode } from "./tool-discovery/mode";
|
|
143
|
+
import {
|
|
144
|
+
collectDiscoverableTools,
|
|
145
|
+
type DiscoverableTool,
|
|
146
|
+
filterBySource,
|
|
147
|
+
formatDiscoverableToolServerSummary,
|
|
148
|
+
selectDiscoverableToolNamesByServer,
|
|
149
|
+
summarizeDiscoverableTools,
|
|
150
|
+
} from "./tool-discovery/tool-index";
|
|
151
|
+
import {
|
|
152
|
+
BashTool,
|
|
153
|
+
BUILTIN_TOOLS,
|
|
154
|
+
computeEssentialBuiltinNames,
|
|
155
|
+
createTools,
|
|
156
|
+
type DeferredDiagnosticsEntry,
|
|
157
|
+
discoverStartupLspServers,
|
|
158
|
+
EditTool,
|
|
159
|
+
EvalTool,
|
|
160
|
+
FindTool,
|
|
161
|
+
filterInitialToolsForDiscoveryAll,
|
|
162
|
+
getSearchTools,
|
|
163
|
+
HIDDEN_TOOLS,
|
|
164
|
+
isImageProviderPreference,
|
|
165
|
+
isSearchProviderPreference,
|
|
166
|
+
type LspStartupServerInfo,
|
|
167
|
+
loadSshTool,
|
|
168
|
+
ReadTool,
|
|
169
|
+
ResolveTool,
|
|
170
|
+
renderSearchToolBm25Description,
|
|
171
|
+
SearchTool,
|
|
172
|
+
SearchToolBm25Tool,
|
|
173
|
+
setPreferredImageProvider,
|
|
174
|
+
setPreferredSearchProvider,
|
|
175
|
+
type Tool,
|
|
176
|
+
type ToolSession,
|
|
177
|
+
WebSearchTool,
|
|
178
|
+
WriteTool,
|
|
179
|
+
warmupLspServers,
|
|
180
|
+
} from "./tools";
|
|
181
|
+
import { ToolContextStore } from "./tools/context";
|
|
182
|
+
import { getImageGenTools } from "./tools/image-gen";
|
|
183
|
+
import { wrapToolWithMetaNotice } from "./tools/output-meta";
|
|
184
|
+
import { queueResolveHandler } from "./tools/resolve";
|
|
185
|
+
import { ttsTool } from "./tools/tts";
|
|
186
|
+
import { EventBus } from "./utils/event-bus";
|
|
187
|
+
import { buildNamedToolChoice } from "./utils/tool-choice";
|
|
188
|
+
import { buildWorkspaceTree, type WorkspaceTree } from "./workspace-tree";
|
|
189
|
+
|
|
190
|
+
type AsyncResultEntry = {
|
|
191
|
+
jobId: string;
|
|
192
|
+
result: string;
|
|
193
|
+
job: AsyncJob | undefined;
|
|
194
|
+
durationMs: number | undefined;
|
|
195
|
+
};
|
|
196
|
+
|
|
197
|
+
type AsyncResultJobDetails = {
|
|
198
|
+
jobId: string;
|
|
199
|
+
type?: "bash" | "task";
|
|
200
|
+
label?: string;
|
|
201
|
+
durationMs?: number;
|
|
202
|
+
};
|
|
203
|
+
|
|
204
|
+
type AsyncResultDetails = {
|
|
205
|
+
jobs: AsyncResultJobDetails[];
|
|
206
|
+
};
|
|
207
|
+
|
|
208
|
+
type McpNotificationEntry = {
|
|
209
|
+
serverName: string;
|
|
210
|
+
uri: string;
|
|
211
|
+
};
|
|
212
|
+
|
|
213
|
+
function buildAsyncResultBatchMessage(entries: AsyncResultEntry[]): CustomMessage<AsyncResultDetails> | null {
|
|
214
|
+
if (entries.length === 0) return null;
|
|
215
|
+
const jobs = entries.map(entry => ({
|
|
216
|
+
jobId: entry.jobId,
|
|
217
|
+
result: entry.result,
|
|
218
|
+
type: entry.job?.type,
|
|
219
|
+
label: entry.job?.label,
|
|
220
|
+
durationMs: entry.durationMs,
|
|
221
|
+
}));
|
|
222
|
+
const details: AsyncResultDetails = {
|
|
223
|
+
jobs: jobs.map(job => ({
|
|
224
|
+
jobId: job.jobId,
|
|
225
|
+
type: job.type,
|
|
226
|
+
label: job.label,
|
|
227
|
+
durationMs: job.durationMs,
|
|
228
|
+
})),
|
|
229
|
+
};
|
|
230
|
+
return {
|
|
231
|
+
role: "custom",
|
|
232
|
+
customType: "async-result",
|
|
233
|
+
content: prompt.render(asyncResultTemplate, {
|
|
234
|
+
multiple: jobs.length > 1,
|
|
235
|
+
jobs,
|
|
236
|
+
}),
|
|
237
|
+
display: true,
|
|
238
|
+
attribution: "agent",
|
|
239
|
+
details,
|
|
240
|
+
timestamp: Date.now(),
|
|
241
|
+
};
|
|
242
|
+
}
|
|
243
|
+
|
|
244
|
+
type LateDiagnosticsDetails = {
|
|
245
|
+
files: Array<{ path: string; summary: string; errored: boolean; messages: string[] }>;
|
|
246
|
+
};
|
|
247
|
+
|
|
248
|
+
function buildLateDiagnosticsBatchMessage(
|
|
249
|
+
entries: DeferredDiagnosticsEntry[],
|
|
250
|
+
): CustomMessage<LateDiagnosticsDetails> | null {
|
|
251
|
+
if (entries.length === 0) return null;
|
|
252
|
+
const files = entries.map(entry => ({
|
|
253
|
+
path: entry.path,
|
|
254
|
+
summary: entry.summary,
|
|
255
|
+
messages: entry.messages,
|
|
256
|
+
errored: entry.errored,
|
|
257
|
+
}));
|
|
258
|
+
const details: LateDiagnosticsDetails = {
|
|
259
|
+
files: files.map(file => ({
|
|
260
|
+
path: file.path,
|
|
261
|
+
summary: file.summary,
|
|
262
|
+
errored: file.errored,
|
|
263
|
+
messages: file.messages,
|
|
264
|
+
})),
|
|
265
|
+
};
|
|
266
|
+
return {
|
|
267
|
+
role: "custom",
|
|
268
|
+
customType: LSP_LATE_DIAGNOSTIC_MESSAGE_TYPE,
|
|
269
|
+
content: prompt.render(lateDiagnosticTemplate, {
|
|
270
|
+
multiple: files.length > 1,
|
|
271
|
+
files,
|
|
272
|
+
}),
|
|
273
|
+
display: true,
|
|
274
|
+
attribution: "agent",
|
|
275
|
+
details,
|
|
276
|
+
timestamp: Date.now(),
|
|
277
|
+
};
|
|
278
|
+
}
|
|
279
|
+
|
|
280
|
+
function buildMcpNotificationBatchMessage(entries: McpNotificationEntry[]): AgentMessage | null {
|
|
281
|
+
const resources: McpNotificationEntry[] = [];
|
|
282
|
+
const seen = new Set<string>();
|
|
283
|
+
for (const entry of entries) {
|
|
284
|
+
const key = `${entry.serverName}\0${entry.uri}`;
|
|
285
|
+
if (seen.has(key)) continue;
|
|
286
|
+
seen.add(key);
|
|
287
|
+
resources.push(entry);
|
|
288
|
+
}
|
|
289
|
+
if (resources.length === 0) return null;
|
|
290
|
+
const lines = [`[MCP notification] ${resources.length} resource(s) updated:`];
|
|
291
|
+
for (const resource of resources) {
|
|
292
|
+
lines.push(`- server="${resource.serverName}" uri=${resource.uri}`);
|
|
293
|
+
}
|
|
294
|
+
lines.push('Use read(path="mcp://<uri>") to inspect if relevant.');
|
|
295
|
+
return {
|
|
296
|
+
role: "user",
|
|
297
|
+
content: [{ type: "text", text: lines.join("\n") }],
|
|
298
|
+
attribution: "agent",
|
|
299
|
+
timestamp: Date.now(),
|
|
300
|
+
};
|
|
301
|
+
}
|
|
302
|
+
|
|
303
|
+
// Types
|
|
304
|
+
export interface CreateAgentSessionOptions {
|
|
305
|
+
/** Working directory for project-local discovery. Default: getProjectDir() */
|
|
306
|
+
cwd?: string;
|
|
307
|
+
/** Global config directory. Default: ~/.omp/agent */
|
|
308
|
+
agentDir?: string;
|
|
309
|
+
/** Spawns to allow. Default: "*" */
|
|
310
|
+
spawns?: string;
|
|
311
|
+
|
|
312
|
+
/** Auth storage for credentials. Default: discoverAuthStorage(agentDir) */
|
|
313
|
+
authStorage?: AuthStorage;
|
|
314
|
+
/** Model registry. Default: discoverModels(authStorage, agentDir) */
|
|
315
|
+
modelRegistry?: ModelRegistry;
|
|
316
|
+
|
|
317
|
+
/** Model to use. Default: from settings, else first available */
|
|
318
|
+
model?: Model;
|
|
319
|
+
/** Raw model pattern string (e.g. from --model CLI flag) to resolve after extensions load.
|
|
320
|
+
* Used when model lookup is deferred because extension-provided models aren't registered yet. */
|
|
321
|
+
modelPattern?: string;
|
|
322
|
+
/** Thinking selector. Default: from settings, else unset */
|
|
323
|
+
thinkingLevel?: ConfiguredThinkingLevel;
|
|
324
|
+
/** Models available for cycling (Ctrl+P in interactive mode) */
|
|
325
|
+
scopedModels?: Array<{ model: Model; thinkingLevel?: ThinkingLevel }>;
|
|
326
|
+
|
|
327
|
+
/** System prompt blocks. Array replaces default, function receives default blocks and returns final blocks. */
|
|
328
|
+
systemPrompt?: string[] | ((defaultPrompt: string[]) => string[]);
|
|
329
|
+
/** Optional provider-facing session identifier for prompt caches and sticky auth selection.
|
|
330
|
+
* Keeps persisted session files isolated while reusing provider-side caches. */
|
|
331
|
+
providerSessionId?: string;
|
|
332
|
+
/** Optional provider-facing prompt cache key, distinct from request lineage. */
|
|
333
|
+
providerPromptCacheKey?: string;
|
|
334
|
+
|
|
335
|
+
/** Custom tools to register (in addition to built-in tools). Accepts both CustomTool and ToolDefinition. */
|
|
336
|
+
customTools?: (CustomTool | ToolDefinition)[];
|
|
337
|
+
/** Inline extensions (merged with discovery). */
|
|
338
|
+
extensions?: ExtensionFactory[];
|
|
339
|
+
/** Additional extension paths to load (merged with discovery). */
|
|
340
|
+
additionalExtensionPaths?: string[];
|
|
341
|
+
/** Disable extension discovery (explicit paths still load). */
|
|
342
|
+
disableExtensionDiscovery?: boolean;
|
|
343
|
+
/**
|
|
344
|
+
* Pre-loaded extensions (skips file discovery and the per-session factory
|
|
345
|
+
* call). Used by the CLI when extensions are loaded early to parse custom
|
|
346
|
+
* flags — the same process owns the returned instances, so reusing them is
|
|
347
|
+
* safe.
|
|
348
|
+
*
|
|
349
|
+
* NEVER pass this across session boundaries (e.g. parent → subagent).
|
|
350
|
+
* `Extension` instances close over a parent-bound `ExtensionAPI` (cwd,
|
|
351
|
+
* eventBus, runtime), and reusing them would route tools/handlers/commands
|
|
352
|
+
* back through the parent. For subagents, forward
|
|
353
|
+
* {@link preloadedExtensionPaths} instead.
|
|
354
|
+
*
|
|
355
|
+
* @internal
|
|
356
|
+
*/
|
|
357
|
+
preloadedExtensions?: LoadExtensionsResult;
|
|
358
|
+
/**
|
|
359
|
+
* Pre-discovered extension source paths. When provided, the filesystem-scan
|
|
360
|
+
* inside `discoverExtensionPaths()` is skipped — the session still calls
|
|
361
|
+
* `loadExtensions()` itself so each `Extension` is bound to THIS session's
|
|
362
|
+
* `ExtensionAPI` (cwd, eventBus, runtime).
|
|
363
|
+
*
|
|
364
|
+
* This is the safe pass-through for parent → subagent forwarding.
|
|
365
|
+
*/
|
|
366
|
+
preloadedExtensionPaths?: string[];
|
|
367
|
+
/**
|
|
368
|
+
* Pre-discovered custom-tool source paths from `.omp/tools/`, `.claude/tools/`,
|
|
369
|
+
* plugins, etc. When provided, the filesystem-scan inside
|
|
370
|
+
* `discoverCustomToolPaths()` is skipped — subagents inherit the parent's
|
|
371
|
+
* scan result and call `loadCustomTools()` themselves so each session binds
|
|
372
|
+
* tools to its OWN `CustomToolAPI` (cwd, exec, pushPendingAction, UI).
|
|
373
|
+
*
|
|
374
|
+
* Forwarding the loaded `LoadedCustomTool[]` instances directly would reuse
|
|
375
|
+
* the parent's session-bound API and route tool execution back through the
|
|
376
|
+
* parent — wrong for isolated tasks and for pending-action routing.
|
|
377
|
+
*/
|
|
378
|
+
preloadedCustomToolPaths?: ToolPathWithSource[];
|
|
379
|
+
|
|
380
|
+
/** Shared event bus for tool/extension communication. Default: creates new bus. */
|
|
381
|
+
eventBus?: EventBus;
|
|
382
|
+
|
|
383
|
+
/** Skills. Default: discovered from multiple locations */
|
|
384
|
+
skills?: Skill[];
|
|
385
|
+
/** Rules. Default: discovered from multiple locations */
|
|
386
|
+
rules?: Rule[];
|
|
387
|
+
/** Context files (AGENTS.md content). Default: discovered walking up from cwd */
|
|
388
|
+
contextFiles?: Array<{ path: string; content: string }>;
|
|
389
|
+
/** Pre-built workspace tree (skips re-scanning; passed by parents to subagents). */
|
|
390
|
+
workspaceTree?: WorkspaceTree;
|
|
391
|
+
/** Prompt templates. Default: discovered from cwd/.omp/prompts/ + agentDir/prompts/ */
|
|
392
|
+
promptTemplates?: PromptTemplate[];
|
|
393
|
+
/** File-based slash commands. Default: discovered from commands/ directories */
|
|
394
|
+
slashCommands?: FileSlashCommand[];
|
|
395
|
+
|
|
396
|
+
/** Enable MCP server discovery from .mcp.json files. Default: true */
|
|
397
|
+
enableMCP?: boolean;
|
|
398
|
+
/** Existing MCP manager to reuse (skips discovery, propagates to toolSession). */
|
|
399
|
+
mcpManager?: MCPManager;
|
|
400
|
+
|
|
401
|
+
/** Enable LSP integration (tool, formatting, diagnostics, warmup). Default: true */
|
|
402
|
+
enableLsp?: boolean;
|
|
403
|
+
/** Skip Python kernel availability check and prelude warmup */
|
|
404
|
+
skipPythonPreflight?: boolean;
|
|
405
|
+
/** Tool names explicitly requested (enables disabled-by-default tools) */
|
|
406
|
+
toolNames?: string[];
|
|
407
|
+
|
|
408
|
+
/** Output schema for structured completion (subagents) */
|
|
409
|
+
outputSchema?: unknown;
|
|
410
|
+
/** Whether to include the yield tool by default */
|
|
411
|
+
requireYieldTool?: boolean;
|
|
412
|
+
/** Task recursion depth (for subagent sessions). Default: 0 */
|
|
413
|
+
taskDepth?: number;
|
|
414
|
+
/** Parent Hindsight state to alias for subagent memory tools. */
|
|
415
|
+
parentHindsightSessionState?: HindsightSessionState;
|
|
416
|
+
/** Parent Mnemopi state to alias for subagent memory tools. */
|
|
417
|
+
parentMnemopiSessionState?: MnemopiSessionState;
|
|
418
|
+
/** Pre-allocated agent identity for IRC routing. Default: "Main" for top-level, parentTaskPrefix-derived for sub. */
|
|
419
|
+
agentId?: string;
|
|
420
|
+
/** Display name for the agent in IRC. Default: "main" or "sub". */
|
|
421
|
+
agentDisplayName?: string;
|
|
422
|
+
/** Optional shared agent registry for IRC routing. Default: AgentRegistry.global(). */
|
|
423
|
+
agentRegistry?: AgentRegistry;
|
|
424
|
+
/** Parent task ID prefix for nested artifact naming (e.g., "Extensions") */
|
|
425
|
+
parentTaskPrefix?: string;
|
|
426
|
+
/** Inherited eval executor session id for subagents sharing parent eval state. */
|
|
427
|
+
parentEvalSessionId?: string;
|
|
428
|
+
|
|
429
|
+
/** Session manager. Default: session stored under the configured agentDir sessions root */
|
|
430
|
+
sessionManager?: SessionManager;
|
|
431
|
+
|
|
432
|
+
/** Override local:// protocol options for subagent local:// sharing. Default: uses the session's own artifacts dir and session ID. */
|
|
433
|
+
localProtocolOptions?: LocalProtocolOptions;
|
|
434
|
+
|
|
435
|
+
/** Settings instance. Default: Settings.init({ cwd, agentDir }) */
|
|
436
|
+
settings?: Settings;
|
|
437
|
+
|
|
438
|
+
/** Whether UI is available (enables interactive tools like ask). Default: false */
|
|
439
|
+
hasUI?: boolean;
|
|
440
|
+
|
|
441
|
+
/**
|
|
442
|
+
* Opt-in OpenTelemetry instrumentation forwarded to the underlying Agent.
|
|
443
|
+
* Passing `{}` enables the loop's GenAI-semantic-convention spans. See
|
|
444
|
+
* {@link AgentTelemetryConfig} for the full surface (hooks, content capture,
|
|
445
|
+
* cost estimator, agent identity).
|
|
446
|
+
*
|
|
447
|
+
* Safe to enable without an OTEL SDK registered in the host: the
|
|
448
|
+
* `@opentelemetry/api` package returns a no-op tracer in that case.
|
|
449
|
+
*/
|
|
450
|
+
telemetry?: AgentTelemetryConfig;
|
|
451
|
+
|
|
452
|
+
/** Whether to auto-approve all tool calls (--auto-approve CLI flag). Default: false */
|
|
453
|
+
autoApprove?: boolean;
|
|
454
|
+
}
|
|
455
|
+
|
|
456
|
+
/** Result from createAgentSession */
|
|
457
|
+
export interface CreateAgentSessionResult {
|
|
458
|
+
/** The created session */
|
|
459
|
+
session: AgentSession;
|
|
460
|
+
/** Extensions result (loaded extensions + runtime) */
|
|
461
|
+
extensionsResult: LoadExtensionsResult;
|
|
462
|
+
/** Update tool UI context (interactive mode) */
|
|
463
|
+
setToolUIContext: (uiContext: ExtensionUIContext, hasUI: boolean) => void;
|
|
464
|
+
/** MCP manager for server lifecycle management (undefined if MCP disabled) */
|
|
465
|
+
mcpManager?: MCPManager;
|
|
466
|
+
/** Warning if session was restored with a different model than saved */
|
|
467
|
+
modelFallbackMessage?: string;
|
|
468
|
+
/** LSP servers detected for startup; warmup may continue in the background */
|
|
469
|
+
lspServers?: LspStartupServerInfo[];
|
|
470
|
+
/** Shared event bus for tool/extension communication */
|
|
471
|
+
eventBus: EventBus;
|
|
472
|
+
}
|
|
473
|
+
|
|
474
|
+
// Re-exports
|
|
475
|
+
|
|
476
|
+
export type { PromptTemplate } from "./config/prompt-templates";
|
|
477
|
+
export { Settings, type SkillsSettings } from "./config/settings";
|
|
478
|
+
export type { CustomCommand, CustomCommandFactory } from "./extensibility/custom-commands/types";
|
|
479
|
+
export type { CustomTool, CustomToolFactory } from "./extensibility/custom-tools/types";
|
|
480
|
+
export type * from "./extensibility/extensions";
|
|
481
|
+
export type { Skill } from "./extensibility/skills";
|
|
482
|
+
export type { FileSlashCommand } from "./extensibility/slash-commands";
|
|
483
|
+
export type { MCPManager, MCPServerConfig, MCPServerConnection, MCPToolsLoadResult } from "./mcp";
|
|
484
|
+
export type { Tool } from "./tools";
|
|
485
|
+
export { buildDirectoryTree, buildWorkspaceTree, type DirectoryTree, type WorkspaceTree } from "./workspace-tree";
|
|
486
|
+
|
|
487
|
+
export {
|
|
488
|
+
// Individual tool classes (for custom usage)
|
|
489
|
+
BashTool,
|
|
490
|
+
// Tool classes and factories
|
|
491
|
+
BUILTIN_TOOLS,
|
|
492
|
+
createTools,
|
|
493
|
+
EditTool,
|
|
494
|
+
EvalTool,
|
|
495
|
+
FindTool,
|
|
496
|
+
HIDDEN_TOOLS,
|
|
497
|
+
loadSshTool,
|
|
498
|
+
ReadTool,
|
|
499
|
+
ResolveTool,
|
|
500
|
+
SearchTool,
|
|
501
|
+
type ToolSession,
|
|
502
|
+
WebSearchTool,
|
|
503
|
+
WriteTool,
|
|
504
|
+
};
|
|
505
|
+
|
|
506
|
+
// Helper Functions
|
|
507
|
+
|
|
508
|
+
function getDefaultAgentDir(): string {
|
|
509
|
+
return getAgentDir();
|
|
510
|
+
}
|
|
511
|
+
|
|
512
|
+
function resolveSnapshotTtlMs(): number {
|
|
513
|
+
const raw = process.env.OMP_AUTH_BROKER_SNAPSHOT_TTL_MS;
|
|
514
|
+
if (raw === undefined) return DEFAULT_SNAPSHOT_CACHE_TTL_MS;
|
|
515
|
+
const value = raw.trim();
|
|
516
|
+
if (value === "") return DEFAULT_SNAPSHOT_CACHE_TTL_MS;
|
|
517
|
+
const ttlMs = Number(value);
|
|
518
|
+
if (Number.isFinite(ttlMs) && ttlMs >= 0) return ttlMs;
|
|
519
|
+
logger.warn("Invalid OMP_AUTH_BROKER_SNAPSHOT_TTL_MS; using default", { value: raw });
|
|
520
|
+
return DEFAULT_SNAPSHOT_CACHE_TTL_MS;
|
|
521
|
+
}
|
|
522
|
+
|
|
523
|
+
// Discovery Functions
|
|
524
|
+
|
|
525
|
+
/**
|
|
526
|
+
* Create an AuthStorage instance.
|
|
527
|
+
*
|
|
528
|
+
* Default: local SQLite store at `<agentDir>/agent.db`.
|
|
529
|
+
*
|
|
530
|
+
* Broker mode: when `OMP_AUTH_BROKER_URL` is set, credentials are pulled from
|
|
531
|
+
* a remote auth-broker over the wire. Refresh tokens never leave the broker;
|
|
532
|
+
* the client receives access tokens with `refresh = "__remote__"` and calls
|
|
533
|
+
* back into the broker through the {@link AuthStorageOptions.refreshOAuthCredential}
|
|
534
|
+
* override to re-mint access tokens when needed.
|
|
535
|
+
*/
|
|
536
|
+
export async function discoverAuthStorage(agentDir: string = getDefaultAgentDir()): Promise<AuthStorage> {
|
|
537
|
+
const brokerConfigPromise = resolveAuthBrokerConfig();
|
|
538
|
+
const cachePath = getAuthBrokerSnapshotCachePath();
|
|
539
|
+
// Warm the encrypted snapshot cache into the page cache while the broker
|
|
540
|
+
// config resolves (it may shell out for a `!command` token). Decryption
|
|
541
|
+
// needs the resolved token, so the real cache read cannot start earlier.
|
|
542
|
+
void Bun.file(cachePath)
|
|
543
|
+
.arrayBuffer()
|
|
544
|
+
.catch(() => undefined);
|
|
545
|
+
const brokerConfig = await brokerConfigPromise;
|
|
546
|
+
if (brokerConfig) {
|
|
547
|
+
const client = new AuthBrokerClient({ url: brokerConfig.url, token: brokerConfig.token });
|
|
548
|
+
const ttlMs = resolveSnapshotTtlMs();
|
|
549
|
+
const persist =
|
|
550
|
+
ttlMs > 0
|
|
551
|
+
? (snapshot: SnapshotResponse): void => {
|
|
552
|
+
void writeAuthBrokerSnapshotCache({
|
|
553
|
+
path: cachePath,
|
|
554
|
+
token: brokerConfig.token,
|
|
555
|
+
url: brokerConfig.url,
|
|
556
|
+
snapshot,
|
|
557
|
+
}).catch(error => {
|
|
558
|
+
logger.debug("auth-broker snapshot cache write failed", { error: String(error) });
|
|
559
|
+
});
|
|
560
|
+
}
|
|
561
|
+
: undefined;
|
|
562
|
+
|
|
563
|
+
let initialSnapshot: SnapshotResponse | undefined;
|
|
564
|
+
if (ttlMs > 0) {
|
|
565
|
+
initialSnapshot =
|
|
566
|
+
(await readAuthBrokerSnapshotCache({
|
|
567
|
+
path: cachePath,
|
|
568
|
+
token: brokerConfig.token,
|
|
569
|
+
url: brokerConfig.url,
|
|
570
|
+
ttlMs,
|
|
571
|
+
}).catch(error => {
|
|
572
|
+
logger.debug("auth-broker snapshot cache read failed", { error: String(error) });
|
|
573
|
+
return null;
|
|
574
|
+
})) ?? undefined;
|
|
575
|
+
}
|
|
576
|
+
if (!initialSnapshot) {
|
|
577
|
+
const initialResult = await client.fetchSnapshot();
|
|
578
|
+
if (initialResult.status !== 200) throw new Error("Auth broker returned no initial snapshot");
|
|
579
|
+
initialSnapshot = initialResult.snapshot;
|
|
580
|
+
persist?.(initialSnapshot);
|
|
581
|
+
}
|
|
582
|
+
const store = new RemoteAuthCredentialStore({ client, initialSnapshot, onSnapshot: persist });
|
|
583
|
+
// Refresh + usage hooks live on RemoteAuthCredentialStore; AuthStorage
|
|
584
|
+
// discovers them automatically when no explicit option overrides them.
|
|
585
|
+
const storage = new AuthStorage(store, {
|
|
586
|
+
configValueResolver: resolveConfigValue,
|
|
587
|
+
sourceLabel: `broker ${brokerConfig.url}`,
|
|
588
|
+
});
|
|
589
|
+
await storage.reload();
|
|
590
|
+
return storage;
|
|
591
|
+
}
|
|
592
|
+
const dbPath = getAgentDbPath(agentDir);
|
|
593
|
+
const storage = await AuthStorage.create(dbPath, {
|
|
594
|
+
configValueResolver: resolveConfigValue,
|
|
595
|
+
sourceLabel: `local ${dbPath}`,
|
|
596
|
+
});
|
|
597
|
+
await storage.reload();
|
|
598
|
+
return storage;
|
|
599
|
+
}
|
|
600
|
+
|
|
601
|
+
/**
|
|
602
|
+
* Discover extensions from cwd.
|
|
603
|
+
*/
|
|
604
|
+
export async function discoverExtensions(cwd?: string): Promise<LoadExtensionsResult> {
|
|
605
|
+
const resolvedCwd = cwd ?? getProjectDir();
|
|
606
|
+
|
|
607
|
+
return discoverAndLoadExtensions([], resolvedCwd);
|
|
608
|
+
}
|
|
609
|
+
|
|
610
|
+
/**
|
|
611
|
+
* Path-only counterpart of {@link loadSessionExtensions}: the FS-heavy scan
|
|
612
|
+
* without the per-session module load. Subagents reuse the parent's path list
|
|
613
|
+
* (cached on {@link ToolSession.extensionPaths}) and rebuild Extension
|
|
614
|
+
* instances themselves so each session's `ExtensionAPI` (cwd, eventBus,
|
|
615
|
+
* runtime) is its own.
|
|
616
|
+
*/
|
|
617
|
+
export async function discoverSessionExtensionPaths(
|
|
618
|
+
options: Pick<CreateAgentSessionOptions, "disableExtensionDiscovery" | "additionalExtensionPaths">,
|
|
619
|
+
cwd: string,
|
|
620
|
+
settings: Settings,
|
|
621
|
+
): Promise<string[]> {
|
|
622
|
+
if (options.disableExtensionDiscovery) {
|
|
623
|
+
return options.additionalExtensionPaths ?? [];
|
|
624
|
+
}
|
|
625
|
+
const configuredPaths = [...(options.additionalExtensionPaths ?? []), ...(settings.get("extensions") ?? [])];
|
|
626
|
+
const disabledExtensionIds = settings.get("disabledExtensions") ?? [];
|
|
627
|
+
return discoverExtensionPaths(configuredPaths, cwd, disabledExtensionIds);
|
|
628
|
+
}
|
|
629
|
+
|
|
630
|
+
/**
|
|
631
|
+
* Load the discovered/configured extensions for a session — everything {@link
|
|
632
|
+
* createAgentSession} would load except the inline factory extensions it appends
|
|
633
|
+
* itself. Extracted so the CLI can resolve extension-registered flags (and thus
|
|
634
|
+
* classify `@file` arguments extension-aware) *before* a session — and its
|
|
635
|
+
* terminal breadcrumb — is created, then hand the result back through
|
|
636
|
+
* {@link CreateAgentSessionOptions.preloadedExtensions} so the work is not
|
|
637
|
+
* repeated. Keep this the single source of the discovery branch logic.
|
|
638
|
+
*/
|
|
639
|
+
export async function loadSessionExtensions(
|
|
640
|
+
options: Pick<CreateAgentSessionOptions, "disableExtensionDiscovery" | "additionalExtensionPaths">,
|
|
641
|
+
cwd: string,
|
|
642
|
+
settings: Settings,
|
|
643
|
+
eventBus: EventBus,
|
|
644
|
+
): Promise<LoadExtensionsResult> {
|
|
645
|
+
const paths = await discoverSessionExtensionPaths(options, cwd, settings);
|
|
646
|
+
const result = await logger.time("loadExtensions", loadExtensions, paths, cwd, eventBus);
|
|
647
|
+
for (const { path, error } of result.errors) {
|
|
648
|
+
logger.error("Failed to load extension", { path, error });
|
|
649
|
+
}
|
|
650
|
+
return result;
|
|
651
|
+
}
|
|
652
|
+
|
|
653
|
+
/**
|
|
654
|
+
* Discover skills from cwd and agentDir.
|
|
655
|
+
*/
|
|
656
|
+
export async function discoverSkills(
|
|
657
|
+
cwd?: string,
|
|
658
|
+
_agentDir?: string,
|
|
659
|
+
settings?: SkillsSettings,
|
|
660
|
+
): Promise<{ skills: Skill[]; warnings: SkillWarning[] }> {
|
|
661
|
+
return await loadSkillsInternal({
|
|
662
|
+
...settings,
|
|
663
|
+
cwd: cwd ?? getProjectDir(),
|
|
664
|
+
});
|
|
665
|
+
}
|
|
666
|
+
|
|
667
|
+
/**
|
|
668
|
+
* Discover context files (AGENTS.md) walking up from cwd.
|
|
669
|
+
* Returns files sorted by depth (farther from cwd first, so closer files appear last/more prominent).
|
|
670
|
+
*/
|
|
671
|
+
export async function discoverContextFiles(
|
|
672
|
+
cwd?: string,
|
|
673
|
+
_agentDir?: string,
|
|
674
|
+
): Promise<Array<{ path: string; content: string; depth?: number }>> {
|
|
675
|
+
return await loadContextFilesInternal({
|
|
676
|
+
cwd: cwd ?? getProjectDir(),
|
|
677
|
+
});
|
|
678
|
+
}
|
|
679
|
+
|
|
680
|
+
/**
|
|
681
|
+
* Discover prompt templates from cwd and agentDir.
|
|
682
|
+
*/
|
|
683
|
+
export async function discoverPromptTemplates(cwd?: string, agentDir?: string): Promise<PromptTemplate[]> {
|
|
684
|
+
return await loadPromptTemplatesInternal({
|
|
685
|
+
cwd: cwd ?? getProjectDir(),
|
|
686
|
+
agentDir: agentDir ?? getDefaultAgentDir(),
|
|
687
|
+
});
|
|
688
|
+
}
|
|
689
|
+
|
|
690
|
+
/**
|
|
691
|
+
* Discover file-based slash commands from commands/ directories.
|
|
692
|
+
*/
|
|
693
|
+
export async function discoverSlashCommands(cwd?: string): Promise<FileSlashCommand[]> {
|
|
694
|
+
return loadSlashCommandsInternal({ cwd: cwd ?? getProjectDir() });
|
|
695
|
+
}
|
|
696
|
+
|
|
697
|
+
/**
|
|
698
|
+
* Discover custom commands (TypeScript slash commands) from cwd and agentDir.
|
|
699
|
+
*/
|
|
700
|
+
export async function discoverCustomTSCommands(cwd?: string, agentDir?: string): Promise<CustomCommandsLoadResult> {
|
|
701
|
+
const resolvedCwd = cwd ?? getProjectDir();
|
|
702
|
+
const resolvedAgentDir = agentDir ?? getDefaultAgentDir();
|
|
703
|
+
|
|
704
|
+
return loadCustomCommandsInternal({
|
|
705
|
+
cwd: resolvedCwd,
|
|
706
|
+
agentDir: resolvedAgentDir,
|
|
707
|
+
});
|
|
708
|
+
}
|
|
709
|
+
|
|
710
|
+
/**
|
|
711
|
+
* Discover MCP servers from .mcp.json files.
|
|
712
|
+
* Returns the manager and loaded tools.
|
|
713
|
+
*/
|
|
714
|
+
export async function discoverMCPServers(cwd?: string): Promise<MCPToolsLoadResult> {
|
|
715
|
+
const resolvedCwd = cwd ?? getProjectDir();
|
|
716
|
+
return discoverAndLoadMCPTools(resolvedCwd);
|
|
717
|
+
}
|
|
718
|
+
|
|
719
|
+
// API Key Helpers
|
|
720
|
+
|
|
721
|
+
// System Prompt
|
|
722
|
+
|
|
723
|
+
export interface BuildSystemPromptOptions {
|
|
724
|
+
tools?: Tool[];
|
|
725
|
+
skills?: Skill[];
|
|
726
|
+
contextFiles?: Array<{ path: string; content: string }>;
|
|
727
|
+
cwd?: string;
|
|
728
|
+
appendPrompt?: string;
|
|
729
|
+
repeatToolDescriptions?: boolean;
|
|
730
|
+
}
|
|
731
|
+
|
|
732
|
+
/**
|
|
733
|
+
* Build the default provider-facing system prompt blocks.
|
|
734
|
+
*
|
|
735
|
+
* The returned `systemPrompt` preserves the stable harness prompt and dynamic project context
|
|
736
|
+
* as separate entries so providers can cache prompt prefixes without concatenating blocks.
|
|
737
|
+
*/
|
|
738
|
+
export async function buildSystemPrompt(options: BuildSystemPromptOptions = {}): Promise<BuildSystemPromptResult> {
|
|
739
|
+
return await buildSystemPromptInternal({
|
|
740
|
+
cwd: options.cwd,
|
|
741
|
+
skills: options.skills,
|
|
742
|
+
contextFiles: options.contextFiles,
|
|
743
|
+
appendSystemPrompt: options.appendPrompt,
|
|
744
|
+
repeatToolDescriptions: options.repeatToolDescriptions,
|
|
745
|
+
});
|
|
746
|
+
}
|
|
747
|
+
|
|
748
|
+
// Internal Helpers
|
|
749
|
+
|
|
750
|
+
function createCustomToolContext(ctx: ExtensionContext): CustomToolContext {
|
|
751
|
+
return {
|
|
752
|
+
sessionManager: ctx.sessionManager,
|
|
753
|
+
modelRegistry: ctx.modelRegistry,
|
|
754
|
+
model: ctx.model,
|
|
755
|
+
isIdle: ctx.isIdle,
|
|
756
|
+
hasQueuedMessages: ctx.hasPendingMessages,
|
|
757
|
+
abort: ctx.abort,
|
|
758
|
+
};
|
|
759
|
+
}
|
|
760
|
+
|
|
761
|
+
function isCustomTool(tool: CustomTool | ToolDefinition): tool is CustomTool {
|
|
762
|
+
// To distinguish, we mark converted tools with a hidden symbol property.
|
|
763
|
+
// If the tool doesn't have this marker, it's a CustomTool that needs conversion.
|
|
764
|
+
return !(tool as any).__isToolDefinition;
|
|
765
|
+
}
|
|
766
|
+
|
|
767
|
+
const TOOL_DEFINITION_MARKER = Symbol("__isToolDefinition");
|
|
768
|
+
|
|
769
|
+
/** Matches the truncation applied to per-server instructions inside `rebuildSystemPrompt`. */
|
|
770
|
+
const MAX_MCP_INSTRUCTIONS_LENGTH = 4000;
|
|
771
|
+
|
|
772
|
+
let sshCleanupRegistered = false;
|
|
773
|
+
|
|
774
|
+
async function cleanupSshResources(): Promise<void> {
|
|
775
|
+
const results = await Promise.allSettled([closeAllConnections(), unmountAll()]);
|
|
776
|
+
for (const result of results) {
|
|
777
|
+
if (result.status === "rejected") {
|
|
778
|
+
logger.warn("SSH cleanup failed", { error: String(result.reason) });
|
|
779
|
+
}
|
|
780
|
+
}
|
|
781
|
+
}
|
|
782
|
+
|
|
783
|
+
function registerSshCleanup(): void {
|
|
784
|
+
if (sshCleanupRegistered) return;
|
|
785
|
+
sshCleanupRegistered = true;
|
|
786
|
+
postmortem.register("ssh-cleanup", cleanupSshResources);
|
|
787
|
+
}
|
|
788
|
+
|
|
789
|
+
let pythonCleanupRegistered = false;
|
|
790
|
+
|
|
791
|
+
function registerPythonCleanup(): void {
|
|
792
|
+
if (pythonCleanupRegistered) return;
|
|
793
|
+
pythonCleanupRegistered = true;
|
|
794
|
+
postmortem.register("python-cleanup", disposeAllKernelSessions);
|
|
795
|
+
}
|
|
796
|
+
|
|
797
|
+
function customToolToDefinition(tool: CustomTool): ToolDefinition {
|
|
798
|
+
const definition: ToolDefinition & { [TOOL_DEFINITION_MARKER]: true } = {
|
|
799
|
+
name: tool.name,
|
|
800
|
+
label: tool.label,
|
|
801
|
+
description: tool.description,
|
|
802
|
+
parameters: tool.parameters,
|
|
803
|
+
hidden: tool.hidden,
|
|
804
|
+
deferrable: tool.deferrable,
|
|
805
|
+
approval: typeof tool.approval === "function" ? tool.approval.bind(tool) : tool.approval,
|
|
806
|
+
mcpServerName: tool.mcpServerName,
|
|
807
|
+
mcpToolName: tool.mcpToolName,
|
|
808
|
+
execute: (toolCallId, params, signal, onUpdate, ctx) =>
|
|
809
|
+
tool.execute(toolCallId, params, onUpdate, createCustomToolContext(ctx), signal),
|
|
810
|
+
onSession: tool.onSession ? (event, ctx) => tool.onSession?.(event, createCustomToolContext(ctx)) : undefined,
|
|
811
|
+
renderCall: tool.renderCall,
|
|
812
|
+
renderResult: tool.renderResult
|
|
813
|
+
? (result, options, theme): Component => {
|
|
814
|
+
const component = tool.renderResult?.(
|
|
815
|
+
result,
|
|
816
|
+
{ expanded: options.expanded, isPartial: options.isPartial, spinnerFrame: options.spinnerFrame },
|
|
817
|
+
theme,
|
|
818
|
+
);
|
|
819
|
+
// Return empty component if undefined to match Component type requirement
|
|
820
|
+
return component ?? ({ render: () => [] } as unknown as Component);
|
|
821
|
+
}
|
|
822
|
+
: undefined,
|
|
823
|
+
[TOOL_DEFINITION_MARKER]: true,
|
|
824
|
+
};
|
|
825
|
+
return definition;
|
|
826
|
+
}
|
|
827
|
+
|
|
828
|
+
function createCustomToolsExtension(tools: CustomTool[]): ExtensionFactory {
|
|
829
|
+
return api => {
|
|
830
|
+
for (const tool of tools) {
|
|
831
|
+
api.registerTool(customToolToDefinition(tool));
|
|
832
|
+
}
|
|
833
|
+
|
|
834
|
+
const runOnSession = async (event: CustomToolSessionEvent, ctx: ExtensionContext) => {
|
|
835
|
+
for (const tool of tools) {
|
|
836
|
+
if (!tool.onSession) continue;
|
|
837
|
+
try {
|
|
838
|
+
await tool.onSession(event, createCustomToolContext(ctx));
|
|
839
|
+
} catch (err) {
|
|
840
|
+
logger.warn("Custom tool onSession error", { tool: tool.name, error: String(err) });
|
|
841
|
+
}
|
|
842
|
+
}
|
|
843
|
+
};
|
|
844
|
+
|
|
845
|
+
api.on("session_start", async (_event, ctx) =>
|
|
846
|
+
runOnSession({ reason: "start", previousSessionFile: undefined }, ctx),
|
|
847
|
+
);
|
|
848
|
+
api.on("session_switch", async (event, ctx) =>
|
|
849
|
+
runOnSession({ reason: "switch", previousSessionFile: event.previousSessionFile }, ctx),
|
|
850
|
+
);
|
|
851
|
+
api.on("session_branch", async (event, ctx) =>
|
|
852
|
+
runOnSession({ reason: "branch", previousSessionFile: event.previousSessionFile }, ctx),
|
|
853
|
+
);
|
|
854
|
+
api.on("session_tree", async (_event, ctx) =>
|
|
855
|
+
runOnSession({ reason: "tree", previousSessionFile: undefined }, ctx),
|
|
856
|
+
);
|
|
857
|
+
api.on("session_shutdown", async (_event, ctx) =>
|
|
858
|
+
runOnSession({ reason: "shutdown", previousSessionFile: undefined }, ctx),
|
|
859
|
+
);
|
|
860
|
+
api.on("auto_compaction_start", async (event, ctx) =>
|
|
861
|
+
runOnSession({ reason: "auto_compaction_start", trigger: event.reason, action: event.action }, ctx),
|
|
862
|
+
);
|
|
863
|
+
api.on("auto_compaction_end", async (event, ctx) =>
|
|
864
|
+
runOnSession(
|
|
865
|
+
{
|
|
866
|
+
reason: "auto_compaction_end",
|
|
867
|
+
action: event.action,
|
|
868
|
+
result: event.result,
|
|
869
|
+
aborted: event.aborted,
|
|
870
|
+
willRetry: event.willRetry,
|
|
871
|
+
errorMessage: event.errorMessage,
|
|
872
|
+
},
|
|
873
|
+
ctx,
|
|
874
|
+
),
|
|
875
|
+
);
|
|
876
|
+
api.on("auto_retry_start", async (event, ctx) =>
|
|
877
|
+
runOnSession(
|
|
878
|
+
{
|
|
879
|
+
reason: "auto_retry_start",
|
|
880
|
+
attempt: event.attempt,
|
|
881
|
+
maxAttempts: event.maxAttempts,
|
|
882
|
+
delayMs: event.delayMs,
|
|
883
|
+
errorMessage: event.errorMessage,
|
|
884
|
+
},
|
|
885
|
+
ctx,
|
|
886
|
+
),
|
|
887
|
+
);
|
|
888
|
+
api.on("auto_retry_end", async (event, ctx) =>
|
|
889
|
+
runOnSession(
|
|
890
|
+
{
|
|
891
|
+
reason: "auto_retry_end",
|
|
892
|
+
success: event.success,
|
|
893
|
+
attempt: event.attempt,
|
|
894
|
+
finalError: event.finalError,
|
|
895
|
+
},
|
|
896
|
+
ctx,
|
|
897
|
+
),
|
|
898
|
+
);
|
|
899
|
+
api.on("ttsr_triggered", async (event, ctx) =>
|
|
900
|
+
runOnSession({ reason: "ttsr_triggered", rules: event.rules }, ctx),
|
|
901
|
+
);
|
|
902
|
+
api.on("todo_reminder", async (event, ctx) =>
|
|
903
|
+
runOnSession(
|
|
904
|
+
{
|
|
905
|
+
reason: "todo_reminder",
|
|
906
|
+
todos: event.todos,
|
|
907
|
+
attempt: event.attempt,
|
|
908
|
+
maxAttempts: event.maxAttempts,
|
|
909
|
+
},
|
|
910
|
+
ctx,
|
|
911
|
+
),
|
|
912
|
+
);
|
|
913
|
+
};
|
|
914
|
+
}
|
|
915
|
+
|
|
916
|
+
// Factory
|
|
917
|
+
|
|
918
|
+
/**
|
|
919
|
+
* Build LoadedCustomCommand entries for all MCP prompts across connected servers.
|
|
920
|
+
* These are re-created whenever prompts change (setOnPromptsChanged callback).
|
|
921
|
+
*/
|
|
922
|
+
function buildMCPPromptCommands(manager: MCPManager): LoadedCustomCommand[] {
|
|
923
|
+
const commands: LoadedCustomCommand[] = [];
|
|
924
|
+
for (const serverName of manager.getConnectedServers()) {
|
|
925
|
+
const prompts = manager.getServerPrompts(serverName);
|
|
926
|
+
if (!prompts?.length) continue;
|
|
927
|
+
for (const prompt of prompts) {
|
|
928
|
+
const commandName = `${serverName}:${prompt.name}`;
|
|
929
|
+
commands.push({
|
|
930
|
+
path: `mcp:${commandName}`,
|
|
931
|
+
resolvedPath: `mcp:${commandName}`,
|
|
932
|
+
source: "bundled",
|
|
933
|
+
command: {
|
|
934
|
+
name: commandName,
|
|
935
|
+
description: prompt.description ?? `MCP prompt from ${serverName}`,
|
|
936
|
+
async execute(args: string[]) {
|
|
937
|
+
const promptArgs: Record<string, string> = {};
|
|
938
|
+
for (const arg of args) {
|
|
939
|
+
const eqIdx = arg.indexOf("=");
|
|
940
|
+
if (eqIdx > 0) {
|
|
941
|
+
promptArgs[arg.slice(0, eqIdx)] = arg.slice(eqIdx + 1);
|
|
942
|
+
}
|
|
943
|
+
}
|
|
944
|
+
const result = await manager.executePrompt(serverName, prompt.name, promptArgs);
|
|
945
|
+
if (!result) return "";
|
|
946
|
+
const parts: string[] = [];
|
|
947
|
+
for (const msg of result.messages) {
|
|
948
|
+
const contentItems = Array.isArray(msg.content) ? msg.content : [msg.content];
|
|
949
|
+
for (const item of contentItems) {
|
|
950
|
+
if (item.type === "text") {
|
|
951
|
+
parts.push(item.text);
|
|
952
|
+
} else if (item.type === "resource") {
|
|
953
|
+
const resource = item.resource;
|
|
954
|
+
if (resource.text) parts.push(resource.text);
|
|
955
|
+
}
|
|
956
|
+
}
|
|
957
|
+
}
|
|
958
|
+
return parts.join("\n\n");
|
|
959
|
+
},
|
|
960
|
+
},
|
|
961
|
+
});
|
|
962
|
+
}
|
|
963
|
+
}
|
|
964
|
+
return commands;
|
|
965
|
+
}
|
|
966
|
+
/**
|
|
967
|
+
* Create an AgentSession with the specified options.
|
|
968
|
+
*
|
|
969
|
+
* @example
|
|
970
|
+
* ```typescript
|
|
971
|
+
* // Minimal - uses defaults
|
|
972
|
+
* const { session } = await createAgentSession();
|
|
973
|
+
*
|
|
974
|
+
* // With explicit model
|
|
975
|
+
* import { getModel } from '@oh-my-pi/pi-ai';
|
|
976
|
+
* const { session } = await createAgentSession({
|
|
977
|
+
* model: getModel('anthropic', 'claude-opus-4-5'),
|
|
978
|
+
* thinkingLevel: 'high',
|
|
979
|
+
* });
|
|
980
|
+
*
|
|
981
|
+
* // Continue previous session
|
|
982
|
+
* const { session, modelFallbackMessage } = await createAgentSession({
|
|
983
|
+
* continueSession: true,
|
|
984
|
+
* });
|
|
985
|
+
*
|
|
986
|
+
* // Full control
|
|
987
|
+
* const { session } = await createAgentSession({
|
|
988
|
+
* model: myModel,
|
|
989
|
+
* getApiKey: async () => Bun.env.MY_KEY,
|
|
990
|
+
* systemPrompt: ['You are helpful.'],
|
|
991
|
+
* tools: codingTools({ cwd: getProjectDir() }),
|
|
992
|
+
* skills: [],
|
|
993
|
+
* sessionManager: SessionManager.inMemory(),
|
|
994
|
+
* });
|
|
995
|
+
* ```
|
|
996
|
+
*/
|
|
997
|
+
export async function createAgentSession(options: CreateAgentSessionOptions = {}): Promise<CreateAgentSessionResult> {
|
|
998
|
+
const cwd = options.cwd ?? getProjectDir();
|
|
999
|
+
const agentDir = options.agentDir ?? getDefaultAgentDir();
|
|
1000
|
+
const eventBus = options.eventBus ?? new EventBus();
|
|
1001
|
+
|
|
1002
|
+
registerSshCleanup();
|
|
1003
|
+
registerPythonCleanup();
|
|
1004
|
+
|
|
1005
|
+
// Pin authStorage to modelRegistry.authStorage: ModelRegistry.getApiKey() routes refresh
|
|
1006
|
+
// failures through that instance, so any divergent storage handed to the bridge / mcpManager
|
|
1007
|
+
// / session would silently miss credential_disabled events.
|
|
1008
|
+
const modelRegistry =
|
|
1009
|
+
options.modelRegistry ??
|
|
1010
|
+
new ModelRegistry(options.authStorage ?? (await logger.time("discoverModels", discoverAuthStorage, agentDir)));
|
|
1011
|
+
const authStorage = modelRegistry.authStorage;
|
|
1012
|
+
if (options.authStorage && options.authStorage !== authStorage) {
|
|
1013
|
+
throw new Error(
|
|
1014
|
+
"options.authStorage and options.modelRegistry.authStorage must be the same instance when both are provided",
|
|
1015
|
+
);
|
|
1016
|
+
}
|
|
1017
|
+
// Subscribe before any getApiKey() call so startup model probes can't fire a
|
|
1018
|
+
// credential_disabled event past us. An embedder's constructor handler makes the
|
|
1019
|
+
// listener set non-empty from construction, which defeats AuthStorage's no-listener
|
|
1020
|
+
// buffer — so we can't rely on it to catch startup events for the extension runner.
|
|
1021
|
+
const startupCredentialDisabledEvents: CredentialDisabledEvent[] = [];
|
|
1022
|
+
let credentialDisabledTarget: ExtensionRunner | undefined;
|
|
1023
|
+
const unsubscribeCredentialDisabled: (() => void) | undefined = authStorage.onCredentialDisabled(event => {
|
|
1024
|
+
if (credentialDisabledTarget) {
|
|
1025
|
+
// Discard return: any handler error is routed through runner.onError listeners.
|
|
1026
|
+
void credentialDisabledTarget.emitCredentialDisabled(event);
|
|
1027
|
+
} else {
|
|
1028
|
+
startupCredentialDisabledEvents.push(event);
|
|
1029
|
+
}
|
|
1030
|
+
});
|
|
1031
|
+
const settings = options.settings ?? (await logger.time("settings", Settings.init, { cwd, agentDir }));
|
|
1032
|
+
logger.time("initializeWithSettings", initializeWithSettings, settings);
|
|
1033
|
+
if (!options.modelRegistry) {
|
|
1034
|
+
modelRegistry.refreshInBackground();
|
|
1035
|
+
}
|
|
1036
|
+
// Kick off workspace tree discovery early. The native workspace scan returns
|
|
1037
|
+
// both the rendered-tree input and the AGENTS.md directory-context index, so
|
|
1038
|
+
// startup does not perform a second recursive filesystem search. Subagents
|
|
1039
|
+
// inherit the parent's resolved values via options.
|
|
1040
|
+
const STARTUP_SCAN_DEADLINE_MS = 5000;
|
|
1041
|
+
const workspaceTreePromise: Promise<WorkspaceTree> = options.workspaceTree
|
|
1042
|
+
? Promise.resolve(options.workspaceTree)
|
|
1043
|
+
: logger.time("buildWorkspaceTree", () => buildWorkspaceTree(cwd, { timeoutMs: STARTUP_SCAN_DEADLINE_MS }));
|
|
1044
|
+
workspaceTreePromise.catch(() => {});
|
|
1045
|
+
|
|
1046
|
+
// Independent discoveries that depend only on cwd/agentDir — kicked off in parallel and awaited
|
|
1047
|
+
// at their respective consumer sites. Their work can overlap with model resolution, secret loading,
|
|
1048
|
+
// session-context build, tool creation, MCP discovery, and extension discovery.
|
|
1049
|
+
const contextFilesPromise = options.contextFiles
|
|
1050
|
+
? Promise.resolve(options.contextFiles)
|
|
1051
|
+
: logger.time("discoverContextFiles", discoverContextFiles, cwd, agentDir);
|
|
1052
|
+
contextFilesPromise.catch(() => {});
|
|
1053
|
+
const promptTemplatesPromise = options.promptTemplates
|
|
1054
|
+
? Promise.resolve(options.promptTemplates)
|
|
1055
|
+
: logger.time("discoverPromptTemplates", discoverPromptTemplates, cwd, agentDir);
|
|
1056
|
+
promptTemplatesPromise.catch(() => {});
|
|
1057
|
+
const slashCommandsPromise = options.slashCommands
|
|
1058
|
+
? Promise.resolve(options.slashCommands)
|
|
1059
|
+
: logger.time("discoverSlashCommands", discoverSlashCommands, cwd);
|
|
1060
|
+
slashCommandsPromise.catch(() => {});
|
|
1061
|
+
const skillsSettings = settings.getGroup("skills");
|
|
1062
|
+
const disabledExtensionIds = settings.get("disabledExtensions") ?? [];
|
|
1063
|
+
const discoveredSkillsPromise =
|
|
1064
|
+
options.skills === undefined
|
|
1065
|
+
? logger.time("discoverSkills", discoverSkills, cwd, agentDir, {
|
|
1066
|
+
...skillsSettings,
|
|
1067
|
+
disabledExtensions: disabledExtensionIds,
|
|
1068
|
+
})
|
|
1069
|
+
: undefined;
|
|
1070
|
+
discoveredSkillsPromise?.catch(() => {});
|
|
1071
|
+
|
|
1072
|
+
// Initialize provider preferences from settings
|
|
1073
|
+
const webSearchProvider = settings.get("providers.webSearch");
|
|
1074
|
+
if (typeof webSearchProvider === "string" && isSearchProviderPreference(webSearchProvider)) {
|
|
1075
|
+
setPreferredSearchProvider(webSearchProvider);
|
|
1076
|
+
}
|
|
1077
|
+
|
|
1078
|
+
const imageProvider = settings.get("providers.image");
|
|
1079
|
+
if (isImageProviderPreference(imageProvider)) {
|
|
1080
|
+
setPreferredImageProvider(imageProvider);
|
|
1081
|
+
}
|
|
1082
|
+
|
|
1083
|
+
const sessionManager =
|
|
1084
|
+
options.sessionManager ??
|
|
1085
|
+
logger.time("sessionManager", () =>
|
|
1086
|
+
SessionManager.create(cwd, SessionManager.getDefaultSessionDir(cwd, agentDir)),
|
|
1087
|
+
);
|
|
1088
|
+
const providerSessionId = options.providerSessionId ?? sessionManager.getSessionId();
|
|
1089
|
+
const modelApiKeyAvailability = new Map<string, boolean>();
|
|
1090
|
+
const getModelAvailabilityKey = (candidate: Model): string =>
|
|
1091
|
+
`${candidate.provider}\u0000${candidate.baseUrl ?? ""}`;
|
|
1092
|
+
const hasModelApiKey = async (candidate: Model): Promise<boolean> => {
|
|
1093
|
+
const availabilityKey = getModelAvailabilityKey(candidate);
|
|
1094
|
+
const cached = modelApiKeyAvailability.get(availabilityKey);
|
|
1095
|
+
if (cached !== undefined) {
|
|
1096
|
+
return cached;
|
|
1097
|
+
}
|
|
1098
|
+
|
|
1099
|
+
const hasKey = !!(await modelRegistry.getApiKey(candidate, providerSessionId));
|
|
1100
|
+
modelApiKeyAvailability.set(availabilityKey, hasKey);
|
|
1101
|
+
return hasKey;
|
|
1102
|
+
};
|
|
1103
|
+
|
|
1104
|
+
// Load and create secret obfuscator early so resumed session state and prompt warnings
|
|
1105
|
+
// reflect actual loaded secrets, not just the setting toggle.
|
|
1106
|
+
let obfuscator: SecretObfuscator | undefined;
|
|
1107
|
+
if (settings.get("secrets.enabled")) {
|
|
1108
|
+
const fileEntries = await logger.time("loadSecrets", loadSecrets, cwd, agentDir);
|
|
1109
|
+
const envEntries = collectEnvSecrets();
|
|
1110
|
+
const allEntries = [...envEntries, ...fileEntries];
|
|
1111
|
+
if (allEntries.length > 0) {
|
|
1112
|
+
obfuscator = new SecretObfuscator(allEntries);
|
|
1113
|
+
}
|
|
1114
|
+
}
|
|
1115
|
+
const secretsEnabled = obfuscator?.hasSecrets() === true;
|
|
1116
|
+
|
|
1117
|
+
// Check if session has existing data to restore
|
|
1118
|
+
const existingSession = logger.time("loadSessionContext", () =>
|
|
1119
|
+
deobfuscateSessionContext(sessionManager.buildSessionContext(), obfuscator),
|
|
1120
|
+
);
|
|
1121
|
+
const existingBranch = logger.time("getSessionBranch", () => sessionManager.getBranch());
|
|
1122
|
+
const hasExistingSession = existingBranch.length > 0;
|
|
1123
|
+
const hasThinkingEntry = existingBranch.some(entry => entry.type === "thinking_level_change");
|
|
1124
|
+
const hasServiceTierEntry = existingBranch.some(entry => entry.type === "service_tier_change");
|
|
1125
|
+
|
|
1126
|
+
const hasExplicitModel = options.model !== undefined || options.modelPattern !== undefined;
|
|
1127
|
+
const modelMatchPreferences = getModelMatchPreferences(settings);
|
|
1128
|
+
const allowedModels = await logger.time("resolveAllowedModels", () =>
|
|
1129
|
+
resolveAllowedModels(modelRegistry, settings, modelMatchPreferences),
|
|
1130
|
+
);
|
|
1131
|
+
const defaultRoleSpec = logger.time("resolveDefaultModelRole", () =>
|
|
1132
|
+
resolveModelRoleValue(settings.getModelRole("default"), allowedModels, {
|
|
1133
|
+
settings,
|
|
1134
|
+
matchPreferences: modelMatchPreferences,
|
|
1135
|
+
modelRegistry,
|
|
1136
|
+
}),
|
|
1137
|
+
);
|
|
1138
|
+
let model = options.model;
|
|
1139
|
+
let modelFallbackMessage: string | undefined;
|
|
1140
|
+
// Identify session model strings to restore in fallback order. We do an
|
|
1141
|
+
// initial pass here so model-dependent setup (thinking-level resolution,
|
|
1142
|
+
// host preconnect) can use the restored model; extension-registered
|
|
1143
|
+
// providers aren't visible yet, so we retry the preferred candidates once
|
|
1144
|
+
// extensions register below.
|
|
1145
|
+
const sessionModelStrings =
|
|
1146
|
+
!hasExplicitModel && hasExistingSession
|
|
1147
|
+
? getRestorableSessionModels(existingSession.models, sessionManager.getLastModelChangeRole())
|
|
1148
|
+
: [];
|
|
1149
|
+
let restoredSessionModelIndex = -1;
|
|
1150
|
+
if (!hasExplicitModel && !model && sessionModelStrings.length > 0) {
|
|
1151
|
+
await logger.time("restoreSessionModel", async () => {
|
|
1152
|
+
let failedSessionModel: string | undefined;
|
|
1153
|
+
for (let i = 0; i < sessionModelStrings.length; i++) {
|
|
1154
|
+
const sessionModelStr = sessionModelStrings[i];
|
|
1155
|
+
const parsedModel = parseModelString(sessionModelStr);
|
|
1156
|
+
if (!parsedModel) {
|
|
1157
|
+
failedSessionModel ??= sessionModelStr;
|
|
1158
|
+
continue;
|
|
1159
|
+
}
|
|
1160
|
+
|
|
1161
|
+
const restoredModel = modelRegistry.find(parsedModel.provider, parsedModel.id);
|
|
1162
|
+
if (restoredModel && (await hasModelApiKey(restoredModel))) {
|
|
1163
|
+
model = restoredModel;
|
|
1164
|
+
restoredSessionModelIndex = i;
|
|
1165
|
+
break;
|
|
1166
|
+
}
|
|
1167
|
+
failedSessionModel ??= sessionModelStr;
|
|
1168
|
+
}
|
|
1169
|
+
if (failedSessionModel) {
|
|
1170
|
+
modelFallbackMessage = `Could not restore model ${failedSessionModel}`;
|
|
1171
|
+
}
|
|
1172
|
+
});
|
|
1173
|
+
}
|
|
1174
|
+
|
|
1175
|
+
// If still no model, try settings default.
|
|
1176
|
+
// Skip settings fallback when an explicit model was requested.
|
|
1177
|
+
if (!hasExplicitModel && !model && defaultRoleSpec.model) {
|
|
1178
|
+
const settingsDefaultModel = defaultRoleSpec.model;
|
|
1179
|
+
logger.time("resolveSettingsDefaultModel", () => {
|
|
1180
|
+
// defaultRoleSpec.model already comes from modelRegistry.getAvailable(),
|
|
1181
|
+
// so re-validating auth here just repeats the expensive lookup path.
|
|
1182
|
+
model = settingsDefaultModel;
|
|
1183
|
+
});
|
|
1184
|
+
}
|
|
1185
|
+
|
|
1186
|
+
const taskDepth = options.taskDepth ?? 0;
|
|
1187
|
+
|
|
1188
|
+
// Resolves the session/agent thinking level using the same precedence we
|
|
1189
|
+
// apply at startup: explicit option → persisted session entry → default
|
|
1190
|
+
// role's explicit selector → selected model's defaultLevel → global
|
|
1191
|
+
// settings default. Run again after extension role reclaim so the final
|
|
1192
|
+
// model's own defaults aren't masked by an earlier fallback model's.
|
|
1193
|
+
const pickInitialThinkingLevel = (selectedModel: Model | undefined): ConfiguredThinkingLevel | undefined => {
|
|
1194
|
+
let level = options.thinkingLevel;
|
|
1195
|
+
if (level === undefined && hasExistingSession && hasThinkingEntry) {
|
|
1196
|
+
level = parseThinkingLevel(existingSession.thinkingLevel);
|
|
1197
|
+
}
|
|
1198
|
+
if (level === undefined && !hasExplicitModel && !hasThinkingEntry && defaultRoleSpec.explicitThinkingLevel) {
|
|
1199
|
+
level = defaultRoleSpec.thinkingLevel;
|
|
1200
|
+
}
|
|
1201
|
+
if (level === undefined && selectedModel?.thinking?.defaultLevel !== undefined) {
|
|
1202
|
+
level = selectedModel.thinking.defaultLevel;
|
|
1203
|
+
}
|
|
1204
|
+
if (level === undefined) {
|
|
1205
|
+
level = settings.get("defaultThinkingLevel");
|
|
1206
|
+
}
|
|
1207
|
+
return level;
|
|
1208
|
+
};
|
|
1209
|
+
let thinkingLevel = pickInitialThinkingLevel(model);
|
|
1210
|
+
let autoThinking = thinkingLevel === AUTO_THINKING;
|
|
1211
|
+
// Concrete level the agent/session start with. With `auto` this is the
|
|
1212
|
+
// provisional level shown until the first per-turn classification resolves;
|
|
1213
|
+
// `auto` itself stays a session-only concept handled by AgentSession.
|
|
1214
|
+
let effectiveThinkingLevel: ThinkingLevel | undefined = thinkingLevel === AUTO_THINKING ? undefined : thinkingLevel;
|
|
1215
|
+
if (model) {
|
|
1216
|
+
const resolvedModel = model;
|
|
1217
|
+
effectiveThinkingLevel = logger.time("resolveThinkingLevelForModel", () =>
|
|
1218
|
+
autoThinking
|
|
1219
|
+
? resolveProvisionalAutoLevel(resolvedModel)
|
|
1220
|
+
: resolveThinkingLevelForModel(resolvedModel, effectiveThinkingLevel),
|
|
1221
|
+
);
|
|
1222
|
+
// Fire-and-forget TLS+H2 handshake to the model's host so it overlaps
|
|
1223
|
+
// with the rest of session setup (extension/skill load, tool registry,
|
|
1224
|
+
// system prompt build). Without this, the first `fetch(...)` pays the
|
|
1225
|
+
// full handshake serially — 100–300 ms transcontinental for
|
|
1226
|
+
// api.anthropic.com from a residential IP. Every mode benefits
|
|
1227
|
+
// (interactive, print, rpc, acp).
|
|
1228
|
+
preconnectModelHost(model.baseUrl);
|
|
1229
|
+
}
|
|
1230
|
+
|
|
1231
|
+
let skills: Skill[];
|
|
1232
|
+
let skillWarnings: SkillWarning[];
|
|
1233
|
+
if (options.skills !== undefined) {
|
|
1234
|
+
skills = options.skills;
|
|
1235
|
+
skillWarnings = [];
|
|
1236
|
+
} else {
|
|
1237
|
+
const discovered = await (discoveredSkillsPromise ?? Promise.resolve({ skills: [], warnings: [] }));
|
|
1238
|
+
skills = discovered.skills;
|
|
1239
|
+
skillWarnings = discovered.warnings;
|
|
1240
|
+
}
|
|
1241
|
+
|
|
1242
|
+
// Discover rules and bucket them in one pass to avoid repeated scans over large rule sets.
|
|
1243
|
+
const { ttsrManager, rulebookRules, alwaysApplyRules, allRules } = await logger.time(
|
|
1244
|
+
"discoverTtsrRules",
|
|
1245
|
+
async () => {
|
|
1246
|
+
const { TtsrManager } = await import("./export/ttsr");
|
|
1247
|
+
const ttsrSettings = settings.getGroup("ttsr");
|
|
1248
|
+
const ttsrManager = new TtsrManager(ttsrSettings);
|
|
1249
|
+
const rulesResult =
|
|
1250
|
+
options.rules !== undefined
|
|
1251
|
+
? { items: options.rules, warnings: undefined }
|
|
1252
|
+
: await loadCapability<Rule>(ruleCapability.id, { cwd });
|
|
1253
|
+
const { rulebookRules, alwaysApplyRules } = bucketRules(rulesResult.items, ttsrManager, {
|
|
1254
|
+
builtinRules: ttsrSettings.builtinRules,
|
|
1255
|
+
disabledRules: ttsrSettings.disabledRules,
|
|
1256
|
+
});
|
|
1257
|
+
if (existingSession.injectedTtsrRules.length > 0) {
|
|
1258
|
+
ttsrManager.restoreInjected(existingSession.injectedTtsrRules);
|
|
1259
|
+
}
|
|
1260
|
+
return { ttsrManager, rulebookRules, alwaysApplyRules, allRules: rulesResult.items };
|
|
1261
|
+
},
|
|
1262
|
+
);
|
|
1263
|
+
|
|
1264
|
+
// Resolve contextFiles up-front (it's needed before tool creation). The
|
|
1265
|
+
// workspace tree scan is slow on large repos and we MUST NOT block startup on
|
|
1266
|
+
// it. On timeout we forward `undefined` to ToolSession; buildSystemPromptInternal
|
|
1267
|
+
// will re-race the same promise through its own withDeadline path. Background
|
|
1268
|
+
// work continues so caches still warm.
|
|
1269
|
+
const raceWithDeadline = async <T>(name: string, work: Promise<T>): Promise<T | undefined> => {
|
|
1270
|
+
let timedOut = false;
|
|
1271
|
+
const result = await Promise.race([
|
|
1272
|
+
work,
|
|
1273
|
+
Bun.sleep(STARTUP_SCAN_DEADLINE_MS).then(() => {
|
|
1274
|
+
timedOut = true;
|
|
1275
|
+
return undefined;
|
|
1276
|
+
}),
|
|
1277
|
+
]);
|
|
1278
|
+
if (timedOut) {
|
|
1279
|
+
logger.warn("Startup scan exceeded deadline; deferring to system prompt fallback", {
|
|
1280
|
+
name,
|
|
1281
|
+
timeoutMs: STARTUP_SCAN_DEADLINE_MS,
|
|
1282
|
+
cwd,
|
|
1283
|
+
});
|
|
1284
|
+
}
|
|
1285
|
+
return result;
|
|
1286
|
+
};
|
|
1287
|
+
const [contextFiles, resolvedWorkspaceTree] = await Promise.all([
|
|
1288
|
+
contextFilesPromise,
|
|
1289
|
+
raceWithDeadline("buildWorkspaceTree", workspaceTreePromise),
|
|
1290
|
+
]);
|
|
1291
|
+
|
|
1292
|
+
let agent: Agent;
|
|
1293
|
+
let session!: AgentSession;
|
|
1294
|
+
let hasSession = false;
|
|
1295
|
+
let hasRegistered = false;
|
|
1296
|
+
const enableLsp = options.enableLsp ?? true;
|
|
1297
|
+
const asyncMaxJobs = Math.min(100, Math.max(1, settings.get("async.maxJobs") ?? 100));
|
|
1298
|
+
const ASYNC_INLINE_RESULT_MAX_CHARS = 12_000;
|
|
1299
|
+
const ASYNC_PREVIEW_MAX_CHARS = 4_000;
|
|
1300
|
+
const formatAsyncResultForFollowUp = async (result: string): Promise<string> => {
|
|
1301
|
+
if (result.length <= ASYNC_INLINE_RESULT_MAX_CHARS) {
|
|
1302
|
+
return result;
|
|
1303
|
+
}
|
|
1304
|
+
|
|
1305
|
+
const preview = `${result.slice(0, ASYNC_PREVIEW_MAX_CHARS)}\n\n[Output truncated. Showing first ${ASYNC_PREVIEW_MAX_CHARS.toLocaleString()} characters.]`;
|
|
1306
|
+
try {
|
|
1307
|
+
const { path: artifactPath, id: artifactId } = await sessionManager.allocateArtifactPath("async");
|
|
1308
|
+
if (artifactPath && artifactId) {
|
|
1309
|
+
await Bun.write(artifactPath, result);
|
|
1310
|
+
return `${preview}\nFull output: artifact://${artifactId}`;
|
|
1311
|
+
}
|
|
1312
|
+
} catch (error) {
|
|
1313
|
+
logger.warn("Failed to persist async follow-up artifact", {
|
|
1314
|
+
error: error instanceof Error ? error.message : String(error),
|
|
1315
|
+
});
|
|
1316
|
+
}
|
|
1317
|
+
|
|
1318
|
+
return preview;
|
|
1319
|
+
};
|
|
1320
|
+
// Only the first top-level session in a process owns an AsyncJobManager.
|
|
1321
|
+
// Subagents inherit the parent's manager via `AsyncJobManager.instance()`
|
|
1322
|
+
// (set below), and any additional top-level session spun up in-process
|
|
1323
|
+
// (e.g. the agent-creation architect in `agent-dashboard.ts`) must share
|
|
1324
|
+
// the live singleton — otherwise its dispose path would clobber the
|
|
1325
|
+
// owning session's manager and break the `task`/`bash` async paths
|
|
1326
|
+
// (issue #1923). The `instance()` guard means later sessions also skip
|
|
1327
|
+
// constructing an orphaned manager that nothing would ever route to.
|
|
1328
|
+
const asyncJobManager =
|
|
1329
|
+
!options.parentTaskPrefix && !AsyncJobManager.instance()
|
|
1330
|
+
? new AsyncJobManager({
|
|
1331
|
+
maxRunningJobs: asyncMaxJobs,
|
|
1332
|
+
onJobComplete: async (jobId, result, job) => {
|
|
1333
|
+
if (!session || asyncJobManager!.isDeliverySuppressed(jobId)) return;
|
|
1334
|
+
const formattedResult = await formatAsyncResultForFollowUp(result);
|
|
1335
|
+
if (asyncJobManager!.isDeliverySuppressed(jobId)) return;
|
|
1336
|
+
|
|
1337
|
+
const durationMs = job ? Math.max(0, Date.now() - job.startTime) : undefined;
|
|
1338
|
+
session.yieldQueue.enqueue<AsyncResultEntry>("async-result", {
|
|
1339
|
+
jobId,
|
|
1340
|
+
result: formattedResult,
|
|
1341
|
+
job,
|
|
1342
|
+
durationMs,
|
|
1343
|
+
});
|
|
1344
|
+
},
|
|
1345
|
+
})
|
|
1346
|
+
: undefined;
|
|
1347
|
+
|
|
1348
|
+
const scopedAsyncJobManager = asyncJobManager ?? (options.parentTaskPrefix ? AsyncJobManager.instance() : undefined);
|
|
1349
|
+
|
|
1350
|
+
const agentRegistry = options.agentRegistry ?? AgentRegistry.global();
|
|
1351
|
+
const resolvedAgentId = options.agentId ?? options.parentTaskPrefix ?? MAIN_AGENT_ID;
|
|
1352
|
+
const resolvedAgentDisplayName =
|
|
1353
|
+
options.agentDisplayName ?? ((options.taskDepth ?? 0) > 0 || options.parentTaskPrefix ? "sub" : "main");
|
|
1354
|
+
const agentKind = (options.taskDepth ?? 0) > 0 || options.parentTaskPrefix ? ("sub" as const) : ("main" as const);
|
|
1355
|
+
/**
|
|
1356
|
+
* Forget the agent ref on teardown — unless the agent is being parked (or is
|
|
1357
|
+
* already parked). Parking disposes the session but keeps the ref addressable
|
|
1358
|
+
* (history://, revive); only process teardown / explicit kill unregisters.
|
|
1359
|
+
*/
|
|
1360
|
+
const unregisterUnlessParked = (): void => {
|
|
1361
|
+
if (agentRegistry.get(resolvedAgentId)?.status === "parked") return;
|
|
1362
|
+
if (AgentLifecycleManager.global().isParking(resolvedAgentId)) return;
|
|
1363
|
+
agentRegistry.unregister(resolvedAgentId);
|
|
1364
|
+
};
|
|
1365
|
+
const evalKernelOwnerId = `agent-session:${Snowflake.next()}`;
|
|
1366
|
+
|
|
1367
|
+
try {
|
|
1368
|
+
const getActiveModelString = (): string | undefined => {
|
|
1369
|
+
const activeModel = agent?.state.model;
|
|
1370
|
+
if (activeModel) return formatModelString(activeModel);
|
|
1371
|
+
if (model) return formatModelString(model);
|
|
1372
|
+
return undefined;
|
|
1373
|
+
};
|
|
1374
|
+
// Per-path mutation counter shared across edit/write tools. Late-diagnostics
|
|
1375
|
+
// entries capture it at fetch time and are dropped at injection if a newer
|
|
1376
|
+
// mutation (any tool) bumped it in the meantime.
|
|
1377
|
+
const fileMutationVersions = new Map<string, number>();
|
|
1378
|
+
const toolSession: ToolSession = {
|
|
1379
|
+
get cwd() {
|
|
1380
|
+
return sessionManager.getCwd();
|
|
1381
|
+
},
|
|
1382
|
+
hasUI: options.hasUI ?? false,
|
|
1383
|
+
enableLsp,
|
|
1384
|
+
get hasEditTool() {
|
|
1385
|
+
const requestedToolNames = options.toolNames
|
|
1386
|
+
? [...new Set(options.toolNames.map(name => name.toLowerCase()))]
|
|
1387
|
+
: undefined;
|
|
1388
|
+
return !requestedToolNames || requestedToolNames.includes("edit");
|
|
1389
|
+
},
|
|
1390
|
+
skipPythonPreflight: options.skipPythonPreflight,
|
|
1391
|
+
contextFiles,
|
|
1392
|
+
workspaceTree: resolvedWorkspaceTree,
|
|
1393
|
+
skills,
|
|
1394
|
+
rules: allRules,
|
|
1395
|
+
eventBus,
|
|
1396
|
+
outputSchema: options.outputSchema,
|
|
1397
|
+
requireYieldTool: options.requireYieldTool,
|
|
1398
|
+
taskDepth: options.taskDepth ?? 0,
|
|
1399
|
+
getSessionFile: () => sessionManager.getSessionFile() ?? null,
|
|
1400
|
+
getEvalKernelOwnerId: () => evalKernelOwnerId,
|
|
1401
|
+
getEvalSessionId: () =>
|
|
1402
|
+
session?.getEvalSessionId() ?? options.parentEvalSessionId ?? defaultEvalSessionId(toolSession),
|
|
1403
|
+
assertEvalExecutionAllowed: () => session?.assertEvalExecutionAllowed(),
|
|
1404
|
+
trackEvalExecution: (execution, abortController) =>
|
|
1405
|
+
session ? session.trackEvalExecution(execution, abortController) : execution,
|
|
1406
|
+
getSessionId: () => sessionManager.getSessionId?.() ?? null,
|
|
1407
|
+
getHindsightSessionState: () => session?.getHindsightSessionState(),
|
|
1408
|
+
getMnemopiSessionState: () => session?.getMnemopiSessionState(),
|
|
1409
|
+
getAgentId: () => resolvedAgentId,
|
|
1410
|
+
getToolByName: name => session?.getToolByName(name),
|
|
1411
|
+
agentRegistry,
|
|
1412
|
+
getSessionSpawns: () => options.spawns ?? "*",
|
|
1413
|
+
getModelString: () => (hasExplicitModel && model ? formatModelString(model) : undefined),
|
|
1414
|
+
getActiveModelString,
|
|
1415
|
+
getPlanModeState: () => session?.getPlanModeState(),
|
|
1416
|
+
getPlanReferencePath: () => session?.getPlanReferencePath() ?? "local://PLAN.md",
|
|
1417
|
+
getGoalModeState: () => session?.getGoalModeState(),
|
|
1418
|
+
getGoalRuntime: () => session?.goalRuntime,
|
|
1419
|
+
getUsageStatistics: () => sessionManager.getUsageStatistics(),
|
|
1420
|
+
getTurnBudget: () => sessionManager.getTurnBudget(),
|
|
1421
|
+
recordEvalSubagentUsage: output => sessionManager.recordEvalSubagentOutput(output),
|
|
1422
|
+
getClientBridge: () => session?.clientBridge,
|
|
1423
|
+
queueDeferredDiagnostics: entry => session?.yieldQueue.enqueue(LSP_LATE_DIAGNOSTIC_MESSAGE_TYPE, entry),
|
|
1424
|
+
bumpFileMutationVersion: path => {
|
|
1425
|
+
const next = (fileMutationVersions.get(path) ?? 0) + 1;
|
|
1426
|
+
fileMutationVersions.set(path, next);
|
|
1427
|
+
return next;
|
|
1428
|
+
},
|
|
1429
|
+
getFileMutationVersion: path => fileMutationVersions.get(path) ?? 0,
|
|
1430
|
+
getTodoPhases: () => session.getTodoPhases(),
|
|
1431
|
+
setTodoPhases: phases => session.setTodoPhases(phases),
|
|
1432
|
+
isMCPDiscoveryEnabled: () => session.isMCPDiscoveryEnabled(),
|
|
1433
|
+
getSelectedMCPToolNames: () => session.getSelectedMCPToolNames(),
|
|
1434
|
+
activateDiscoveredMCPTools: toolNames => session.activateDiscoveredMCPTools(toolNames),
|
|
1435
|
+
// Generic tool discovery (unified — covers built-in + MCP + extension)
|
|
1436
|
+
isToolDiscoveryEnabled: () => session.isToolDiscoveryEnabled(),
|
|
1437
|
+
getDiscoverableTools: filter => session.getDiscoverableTools(filter),
|
|
1438
|
+
getDiscoverableToolSearchIndex: () => session.getDiscoverableToolSearchIndex(),
|
|
1439
|
+
getSelectedDiscoveredToolNames: () => session.getSelectedDiscoveredToolNames(),
|
|
1440
|
+
activateDiscoveredTools: toolNames => session.activateDiscoveredTools(toolNames),
|
|
1441
|
+
getCheckpointState: () => session.getCheckpointState(),
|
|
1442
|
+
setCheckpointState: state => session.setCheckpointState(state ?? undefined),
|
|
1443
|
+
getToolChoiceQueue: () => session.toolChoiceQueue,
|
|
1444
|
+
buildToolChoice: name => {
|
|
1445
|
+
const m = session.model;
|
|
1446
|
+
return m ? buildNamedToolChoice(name, m) : undefined;
|
|
1447
|
+
},
|
|
1448
|
+
steer: msg =>
|
|
1449
|
+
session.agent.steer({
|
|
1450
|
+
role: "custom",
|
|
1451
|
+
customType: msg.customType,
|
|
1452
|
+
content: msg.content,
|
|
1453
|
+
display: false,
|
|
1454
|
+
details: msg.details,
|
|
1455
|
+
attribution: "agent",
|
|
1456
|
+
timestamp: Date.now(),
|
|
1457
|
+
}),
|
|
1458
|
+
peekQueueInvoker: () => session.peekQueueInvoker(),
|
|
1459
|
+
peekStandingResolveHandler: () => session.peekStandingResolveHandler(),
|
|
1460
|
+
setStandingResolveHandler: handler => session.setStandingResolveHandler(handler),
|
|
1461
|
+
allocateOutputArtifact: async toolType => {
|
|
1462
|
+
try {
|
|
1463
|
+
return await sessionManager.allocateArtifactPath(toolType);
|
|
1464
|
+
} catch {
|
|
1465
|
+
return {};
|
|
1466
|
+
}
|
|
1467
|
+
},
|
|
1468
|
+
getArtifactManager: () => sessionManager.getArtifactManager(),
|
|
1469
|
+
settings,
|
|
1470
|
+
authStorage,
|
|
1471
|
+
modelRegistry,
|
|
1472
|
+
getTelemetry: () => agent?.telemetry,
|
|
1473
|
+
// Subagents inherit the singleton (the parent's manager) so their bash/task
|
|
1474
|
+
// completions still flow into the spawning conversation's yieldQueue.
|
|
1475
|
+
// Secondary in-process top-level sessions (no parentTaskPrefix, no
|
|
1476
|
+
// constructed manager because the singleton was already installed) leave
|
|
1477
|
+
// this undefined so tools and session job snapshots refuse async work
|
|
1478
|
+
// instead of silently routing into the owning session (issue #1923).
|
|
1479
|
+
asyncJobManager: scopedAsyncJobManager,
|
|
1480
|
+
};
|
|
1481
|
+
|
|
1482
|
+
// Wire process-wide internal URL singletons owned by their real classes.
|
|
1483
|
+
// Top-level sessions install the active snapshots; subagents inherit them.
|
|
1484
|
+
// Artifact and agent-output URLs resolve via `AgentRegistry.global()` —
|
|
1485
|
+
// the protocol handlers walk each ref's `sessionManager.getArtifactsDir()`,
|
|
1486
|
+
// which collapses to the parent's dir for subagents (they adopt the
|
|
1487
|
+
// parent's ArtifactManager) so one lookup hits everything.
|
|
1488
|
+
const getArtifactsDir = () => sessionManager.getArtifactsDir();
|
|
1489
|
+
if (!options.parentTaskPrefix) {
|
|
1490
|
+
setActiveSkills(skills);
|
|
1491
|
+
// Include TTSR rules so `rule://<name>` can resolve them too. They are
|
|
1492
|
+
// registered with the manager and bucketed out before rulebook/always,
|
|
1493
|
+
// so without this a TTSR-only rule (e.g. a triggered builtin) is not
|
|
1494
|
+
// addressable and `rule://` reports "Available: none".
|
|
1495
|
+
setActiveRules([...rulebookRules, ...alwaysApplyRules, ...ttsrManager.getRules()]);
|
|
1496
|
+
if (asyncJobManager) AsyncJobManager.setInstance(asyncJobManager);
|
|
1497
|
+
}
|
|
1498
|
+
const localProtocolOptions = options.localProtocolOptions ?? {
|
|
1499
|
+
getArtifactsDir,
|
|
1500
|
+
getSessionId: () => sessionManager.getSessionId?.() ?? null,
|
|
1501
|
+
};
|
|
1502
|
+
if (options.localProtocolOptions) {
|
|
1503
|
+
LocalProtocolHandler.setOverride(options.localProtocolOptions);
|
|
1504
|
+
}
|
|
1505
|
+
toolSession.getArtifactsDir = getArtifactsDir;
|
|
1506
|
+
toolSession.localProtocolOptions = localProtocolOptions;
|
|
1507
|
+
toolSession.agentOutputManager = new AgentOutputManager(
|
|
1508
|
+
getArtifactsDir,
|
|
1509
|
+
options.parentTaskPrefix ? { parentPrefix: options.parentTaskPrefix } : undefined,
|
|
1510
|
+
);
|
|
1511
|
+
|
|
1512
|
+
// Create built-in tools (already wrapped with meta notice formatting)
|
|
1513
|
+
const builtinTools = await logger.time("createAllTools", createTools, toolSession, options.toolNames);
|
|
1514
|
+
|
|
1515
|
+
// Discover MCP tools from .mcp.json files
|
|
1516
|
+
let mcpManager: MCPManager | undefined = options.mcpManager;
|
|
1517
|
+
toolSession.mcpManager = mcpManager;
|
|
1518
|
+
const enableMCP = options.enableMCP ?? true;
|
|
1519
|
+
const customTools: CustomTool[] = [];
|
|
1520
|
+
if (enableMCP && !mcpManager) {
|
|
1521
|
+
const mcpResult = await logger.time("discoverAndLoadMCPTools", discoverAndLoadMCPTools, cwd, {
|
|
1522
|
+
onConnecting: serverNames => {
|
|
1523
|
+
if (options.hasUI && serverNames.length > 0) {
|
|
1524
|
+
process.stderr.write(`${chalk.gray(`Connecting to MCP servers: ${serverNames.join(", ")}…`)}\n`);
|
|
1525
|
+
}
|
|
1526
|
+
},
|
|
1527
|
+
enableProjectConfig: settings.get("mcp.enableProjectConfig") ?? true,
|
|
1528
|
+
// Always filter Exa - we have native integration
|
|
1529
|
+
filterExa: true,
|
|
1530
|
+
// Filter browser MCP servers when builtin browser tool is active
|
|
1531
|
+
filterBrowser: settings.get("browser.enabled") ?? false,
|
|
1532
|
+
cacheStorage: settings.getStorage(),
|
|
1533
|
+
authStorage,
|
|
1534
|
+
});
|
|
1535
|
+
mcpManager = mcpResult.manager;
|
|
1536
|
+
toolSession.mcpManager = mcpManager;
|
|
1537
|
+
|
|
1538
|
+
if (settings.get("mcp.notifications")) {
|
|
1539
|
+
mcpManager.setNotificationsEnabled(true);
|
|
1540
|
+
}
|
|
1541
|
+
// If we extracted Exa API keys from MCP configs and EXA_API_KEY isn't set, use the first one
|
|
1542
|
+
if (mcpResult.exaApiKeys.length > 0 && !$env.EXA_API_KEY) {
|
|
1543
|
+
Bun.env.EXA_API_KEY = mcpResult.exaApiKeys[0];
|
|
1544
|
+
}
|
|
1545
|
+
|
|
1546
|
+
// Log MCP errors
|
|
1547
|
+
for (const { path, error } of mcpResult.errors) {
|
|
1548
|
+
logger.error("MCP tool load failed", { path, error });
|
|
1549
|
+
}
|
|
1550
|
+
|
|
1551
|
+
if (mcpResult.tools.length > 0) {
|
|
1552
|
+
// MCP tools are LoadedCustomTool, extract the tool property
|
|
1553
|
+
customTools.push(...mcpResult.tools.map(loaded => loaded.tool));
|
|
1554
|
+
}
|
|
1555
|
+
}
|
|
1556
|
+
// Only top-level sessions own the global MCPManager. Subagents already
|
|
1557
|
+
// receive the parent's manager via `options.mcpManager`, and reassigning
|
|
1558
|
+
// the singleton to the same value is a no-op \u2014 keep the gate explicit
|
|
1559
|
+
// to mirror the AsyncJobManager ownership rule.
|
|
1560
|
+
if (mcpManager && !options.parentTaskPrefix) MCPManager.setInstance(mcpManager);
|
|
1561
|
+
|
|
1562
|
+
// Add image tools when the active model or configured image providers can generate images.
|
|
1563
|
+
const imageGenTools = await logger.time("getImageGenTools", () => getImageGenTools(modelRegistry, model));
|
|
1564
|
+
if (imageGenTools.length > 0) {
|
|
1565
|
+
customTools.push(...(imageGenTools as unknown as CustomTool[]));
|
|
1566
|
+
}
|
|
1567
|
+
|
|
1568
|
+
if (settings.get("tts.enabled")) {
|
|
1569
|
+
customTools.push(ttsTool as unknown as CustomTool);
|
|
1570
|
+
}
|
|
1571
|
+
|
|
1572
|
+
// Add web search tools
|
|
1573
|
+
if (options.toolNames?.includes("web_search")) {
|
|
1574
|
+
customTools.push(...getSearchTools());
|
|
1575
|
+
}
|
|
1576
|
+
|
|
1577
|
+
// Discover custom tools from `.omp/tools/`, `.claude/tools/`, plugins, etc.
|
|
1578
|
+
// Subagents reuse the parent's scan via `preloadedCustomToolPaths` to skip
|
|
1579
|
+
// the FS walk, but ALWAYS re-call `loadCustomTools` here so factories bind
|
|
1580
|
+
// to THIS session's `CustomToolAPI` (cwd, exec, pushPendingAction, UI).
|
|
1581
|
+
// Forwarding the parent's `LoadedCustomTool[]` directly would route tool
|
|
1582
|
+
// execution back through the parent — wrong for isolated tasks and for
|
|
1583
|
+
// pending-action queueing.
|
|
1584
|
+
const builtInToolNames = builtinTools.map(t => t.name);
|
|
1585
|
+
const customToolPaths: ToolPathWithSource[] =
|
|
1586
|
+
options.preloadedCustomToolPaths ??
|
|
1587
|
+
(await logger.time("discoverCustomToolPaths", () => discoverCustomToolPaths([], cwd)));
|
|
1588
|
+
const customToolsLoadResult = await logger.time("loadCustomTools", () =>
|
|
1589
|
+
loadCustomTools(customToolPaths, cwd, builtInToolNames, action => queueResolveHandler(toolSession, action)),
|
|
1590
|
+
);
|
|
1591
|
+
for (const { path, error } of customToolsLoadResult.errors) {
|
|
1592
|
+
logger.error("Custom tool load failed", { path, error });
|
|
1593
|
+
}
|
|
1594
|
+
if (customToolsLoadResult.tools.length > 0) {
|
|
1595
|
+
customTools.push(...customToolsLoadResult.tools.map(loaded => loaded.tool));
|
|
1596
|
+
}
|
|
1597
|
+
// Forward the path list (NOT the loaded tools) to subagents so they
|
|
1598
|
+
// re-bind under their own `CustomToolAPI` while skipping the FS scan.
|
|
1599
|
+
toolSession.customToolPaths = customToolPaths;
|
|
1600
|
+
|
|
1601
|
+
const inlineExtensions: ExtensionFactory[] = options.extensions ? [...options.extensions] : [];
|
|
1602
|
+
inlineExtensions.push((await import("./autoresearch")).createAutoresearchExtension);
|
|
1603
|
+
if (customTools.length > 0) {
|
|
1604
|
+
inlineExtensions.push(createCustomToolsExtension(customTools));
|
|
1605
|
+
}
|
|
1606
|
+
|
|
1607
|
+
// Load extensions. Three paths:
|
|
1608
|
+
// 1. `preloadedExtensions` (CLI): caller already loaded — reuse the
|
|
1609
|
+
// Extension instances. Shallow-clone `extensions` so the inline
|
|
1610
|
+
// push below cannot mutate the caller's array. `runtime` is shared
|
|
1611
|
+
// so flag values set pre-creation flow into the live session.
|
|
1612
|
+
// 2. `preloadedExtensionPaths` (subagent): caller resolved paths;
|
|
1613
|
+
// skip the FS scan but always re-call `loadExtensions` here so
|
|
1614
|
+
// each `Extension` binds to THIS session's `ExtensionAPI`
|
|
1615
|
+
// (cwd, eventBus, runtime).
|
|
1616
|
+
// 3. No preload: run the full session discovery.
|
|
1617
|
+
// `disableExtensionDiscovery` is honored implicitly: a caller that set
|
|
1618
|
+
// the flag and pre-resolved the result already reflects that choice.
|
|
1619
|
+
let extensionPaths: string[];
|
|
1620
|
+
let extensionsResult: LoadExtensionsResult;
|
|
1621
|
+
if (options.preloadedExtensions) {
|
|
1622
|
+
extensionsResult = {
|
|
1623
|
+
...options.preloadedExtensions,
|
|
1624
|
+
extensions: [...options.preloadedExtensions.extensions],
|
|
1625
|
+
};
|
|
1626
|
+
// Capture paths for downstream forwarding; filter inline-factory
|
|
1627
|
+
// entries (`<inline-N>`) — those are per-session, not source paths.
|
|
1628
|
+
extensionPaths = extensionsResult.extensions
|
|
1629
|
+
.map(ext => ext.resolvedPath)
|
|
1630
|
+
.filter(p => !p.startsWith("<inline"));
|
|
1631
|
+
} else if (options.preloadedExtensionPaths) {
|
|
1632
|
+
extensionPaths = options.preloadedExtensionPaths;
|
|
1633
|
+
extensionsResult = await logger.time("loadExtensions", loadExtensions, extensionPaths, cwd, eventBus);
|
|
1634
|
+
for (const { path, error } of extensionsResult.errors) {
|
|
1635
|
+
logger.error("Failed to load extension", { path, error });
|
|
1636
|
+
}
|
|
1637
|
+
} else {
|
|
1638
|
+
extensionPaths = await logger.time("discoverSessionExtensionPaths", () =>
|
|
1639
|
+
discoverSessionExtensionPaths(options, cwd, settings),
|
|
1640
|
+
);
|
|
1641
|
+
extensionsResult = await logger.time("loadExtensions", loadExtensions, extensionPaths, cwd, eventBus);
|
|
1642
|
+
for (const { path, error } of extensionsResult.errors) {
|
|
1643
|
+
logger.error("Failed to load extension", { path, error });
|
|
1644
|
+
}
|
|
1645
|
+
}
|
|
1646
|
+
// Forward the source-path list (NOT the loaded instances) so subagents
|
|
1647
|
+
// rebuild their own session-scoped extensions.
|
|
1648
|
+
toolSession.extensionPaths = extensionPaths;
|
|
1649
|
+
|
|
1650
|
+
// Load inline extensions from factories
|
|
1651
|
+
if (inlineExtensions.length > 0) {
|
|
1652
|
+
for (let i = 0; i < inlineExtensions.length; i++) {
|
|
1653
|
+
const factory = inlineExtensions[i];
|
|
1654
|
+
const loaded = await loadExtensionFromFactory(
|
|
1655
|
+
factory,
|
|
1656
|
+
cwd,
|
|
1657
|
+
eventBus,
|
|
1658
|
+
extensionsResult.runtime,
|
|
1659
|
+
`<inline-${i}>`,
|
|
1660
|
+
);
|
|
1661
|
+
extensionsResult.extensions.push(loaded);
|
|
1662
|
+
}
|
|
1663
|
+
}
|
|
1664
|
+
|
|
1665
|
+
// Process provider registrations queued during extension loading.
|
|
1666
|
+
// This must happen before the runner is created so that models registered by
|
|
1667
|
+
// extensions are available for model selection on session resume / fallback.
|
|
1668
|
+
const activeExtensionSources = extensionsResult.extensions.map(extension => extension.path);
|
|
1669
|
+
modelRegistry.syncExtensionSources(activeExtensionSources);
|
|
1670
|
+
for (const sourceId of new Set(activeExtensionSources)) {
|
|
1671
|
+
modelRegistry.clearSourceRegistrations(sourceId);
|
|
1672
|
+
}
|
|
1673
|
+
if (extensionsResult.runtime.pendingProviderRegistrations.length > 0) {
|
|
1674
|
+
for (const { name, config, sourceId } of extensionsResult.runtime.pendingProviderRegistrations) {
|
|
1675
|
+
modelRegistry.registerProvider(name, config, sourceId);
|
|
1676
|
+
}
|
|
1677
|
+
extensionsResult.runtime.pendingProviderRegistrations = [];
|
|
1678
|
+
}
|
|
1679
|
+
// Discover runtime (extension) provider catalogs now that they are
|
|
1680
|
+
// registered. The startup refreshInBackground() ran before extensions
|
|
1681
|
+
// loaded, so dynamic extension providers are only discovered here. Runs in
|
|
1682
|
+
// the background (cache-aware) so startup is never blocked on the fetch; the
|
|
1683
|
+
// model list re-renders when the catalog arrives, like other dynamic providers.
|
|
1684
|
+
void modelRegistry.refreshRuntimeProviders().catch(error => {
|
|
1685
|
+
logger.warn("runtime provider discovery failed", {
|
|
1686
|
+
error: error instanceof Error ? error.message : String(error),
|
|
1687
|
+
});
|
|
1688
|
+
});
|
|
1689
|
+
|
|
1690
|
+
// Retry session-model candidates now that extension providers are
|
|
1691
|
+
// registered. The initial restore runs before extensions load, so a role
|
|
1692
|
+
// model supplied by an extension would have either fallen back to the
|
|
1693
|
+
// saved default (`restoredSessionModelIndex > 0`) or failed entirely
|
|
1694
|
+
// (`restoredSessionModelIndex === -1`, with the settings default or
|
|
1695
|
+
// downstream fallback filling `model`). Reclaim it here so resume
|
|
1696
|
+
// honors the last active role in either case.
|
|
1697
|
+
const sessionRetryLimit = restoredSessionModelIndex >= 0 ? restoredSessionModelIndex : sessionModelStrings.length;
|
|
1698
|
+
if (!hasExplicitModel && sessionRetryLimit > 0) {
|
|
1699
|
+
for (let i = 0; i < sessionRetryLimit; i++) {
|
|
1700
|
+
const sessionModelStr = sessionModelStrings[i];
|
|
1701
|
+
const parsedModel = parseModelString(sessionModelStr);
|
|
1702
|
+
if (!parsedModel) continue;
|
|
1703
|
+
const restoredModel = modelRegistry.find(parsedModel.provider, parsedModel.id);
|
|
1704
|
+
if (restoredModel && (await hasModelApiKey(restoredModel))) {
|
|
1705
|
+
model = restoredModel;
|
|
1706
|
+
modelFallbackMessage = undefined;
|
|
1707
|
+
restoredSessionModelIndex = i;
|
|
1708
|
+
// Recompute thinking-level from scratch against the reclaimed
|
|
1709
|
+
// model: any value derived from the earlier fallback model's
|
|
1710
|
+
// `thinking.defaultLevel` must not become sticky.
|
|
1711
|
+
thinkingLevel = pickInitialThinkingLevel(restoredModel);
|
|
1712
|
+
autoThinking = thinkingLevel === AUTO_THINKING;
|
|
1713
|
+
effectiveThinkingLevel = thinkingLevel === AUTO_THINKING ? undefined : thinkingLevel;
|
|
1714
|
+
effectiveThinkingLevel = logger.time("resolveThinkingLevelForModel", () =>
|
|
1715
|
+
autoThinking
|
|
1716
|
+
? resolveProvisionalAutoLevel(restoredModel)
|
|
1717
|
+
: resolveThinkingLevelForModel(restoredModel, effectiveThinkingLevel),
|
|
1718
|
+
);
|
|
1719
|
+
preconnectModelHost(restoredModel.baseUrl);
|
|
1720
|
+
break;
|
|
1721
|
+
}
|
|
1722
|
+
}
|
|
1723
|
+
}
|
|
1724
|
+
// Resolve deferred --model pattern now that extension models are registered.
|
|
1725
|
+
if (!model && options.modelPattern) {
|
|
1726
|
+
const availableModels = modelRegistry.getAll();
|
|
1727
|
+
const matchPreferences = getModelMatchPreferences(settings);
|
|
1728
|
+
const { model: resolved } = parseModelPattern(options.modelPattern, availableModels, matchPreferences, {
|
|
1729
|
+
modelRegistry,
|
|
1730
|
+
});
|
|
1731
|
+
if (resolved) {
|
|
1732
|
+
model = resolved;
|
|
1733
|
+
modelFallbackMessage = undefined;
|
|
1734
|
+
} else {
|
|
1735
|
+
modelFallbackMessage = `Model "${options.modelPattern}" not found`;
|
|
1736
|
+
}
|
|
1737
|
+
}
|
|
1738
|
+
|
|
1739
|
+
// Fall back to first available model with a valid API key, honoring the
|
|
1740
|
+
// path-scoped `enabledModels` allow-list when configured. Skip when the
|
|
1741
|
+
// user explicitly requested a model via --model that wasn't found.
|
|
1742
|
+
if (!model && !options.modelPattern) {
|
|
1743
|
+
// Re-resolve the allowed set: extension factories above may have
|
|
1744
|
+
// registered providers/models that weren't visible at startup.
|
|
1745
|
+
const fallbackCandidates = await resolveAllowedModels(modelRegistry, settings, modelMatchPreferences);
|
|
1746
|
+
// Prefer each provider's configured default model
|
|
1747
|
+
// (DEFAULT_MODEL_PER_PROVIDER) over raw catalog order. Without this the
|
|
1748
|
+
// first-run fallback picks whatever model sorts first in models.json for
|
|
1749
|
+
// the winning provider (e.g. anthropic's claude-3-5-sonnet-20240620)
|
|
1750
|
+
// instead of the intended provider default (claude-sonnet-4-6). Mirrors
|
|
1751
|
+
// findInitialModel's precedence.
|
|
1752
|
+
for (const [provider, defaultId] of Object.entries(DEFAULT_MODEL_PER_PROVIDER)) {
|
|
1753
|
+
const preferred = fallbackCandidates.find(
|
|
1754
|
+
candidate => candidate.provider === provider && candidate.id === defaultId,
|
|
1755
|
+
);
|
|
1756
|
+
if (preferred && (await hasModelApiKey(preferred))) {
|
|
1757
|
+
model = preferred;
|
|
1758
|
+
break;
|
|
1759
|
+
}
|
|
1760
|
+
}
|
|
1761
|
+
// Otherwise, first available model with a valid API key.
|
|
1762
|
+
if (!model) {
|
|
1763
|
+
for (const candidate of fallbackCandidates) {
|
|
1764
|
+
if (await hasModelApiKey(candidate)) {
|
|
1765
|
+
model = candidate;
|
|
1766
|
+
break;
|
|
1767
|
+
}
|
|
1768
|
+
}
|
|
1769
|
+
}
|
|
1770
|
+
if (model) {
|
|
1771
|
+
if (modelFallbackMessage) {
|
|
1772
|
+
modelFallbackMessage += `. Using ${model.provider}/${model.id}`;
|
|
1773
|
+
}
|
|
1774
|
+
} else {
|
|
1775
|
+
const patterns = settings.get("enabledModels");
|
|
1776
|
+
modelFallbackMessage =
|
|
1777
|
+
patterns && patterns.length > 0
|
|
1778
|
+
? `No model available matching enabledModels (${patterns.join(", ")}) with usable credentials. Configure auth for an allowed provider or adjust enabledModels.`
|
|
1779
|
+
: "No models available. Use /login or set an API key environment variable. Then use /model to select a model.";
|
|
1780
|
+
}
|
|
1781
|
+
}
|
|
1782
|
+
|
|
1783
|
+
// Discover custom commands (TypeScript slash commands)
|
|
1784
|
+
const customCommandsResult: CustomCommandsLoadResult = options.disableExtensionDiscovery
|
|
1785
|
+
? { commands: [], errors: [] }
|
|
1786
|
+
: await logger.time("discoverCustomCommands", loadCustomCommandsInternal, { cwd, agentDir });
|
|
1787
|
+
if (!options.disableExtensionDiscovery) {
|
|
1788
|
+
for (const { path, error } of customCommandsResult.errors) {
|
|
1789
|
+
logger.error("Failed to load custom command", { path, error });
|
|
1790
|
+
}
|
|
1791
|
+
}
|
|
1792
|
+
|
|
1793
|
+
// The runner is created unconditionally — even with zero extensions loaded — because the
|
|
1794
|
+
// `ExtensionToolWrapper` installed below is the only place the per-tool approval gate runs.
|
|
1795
|
+
// A conditional runner means the approval system silently disappears for users with no
|
|
1796
|
+
// extensions, contradicting non-yolo `tools.approvalMode` settings without feedback.
|
|
1797
|
+
// (The builtin autoresearch extension is unconditionally loaded above, so this scenario
|
|
1798
|
+
// is unreachable; unconditional runner construction keeps that invariant explicit and
|
|
1799
|
+
// prevents future optional extensions from silently re-opening the hole.)
|
|
1800
|
+
const extensionRunner: ExtensionRunner = new ExtensionRunner(
|
|
1801
|
+
extensionsResult.extensions,
|
|
1802
|
+
extensionsResult.runtime,
|
|
1803
|
+
cwd,
|
|
1804
|
+
sessionManager,
|
|
1805
|
+
modelRegistry,
|
|
1806
|
+
() => (hasSession ? createSessionMemoryRuntimeContext(session, agentDir, cwd) : undefined),
|
|
1807
|
+
);
|
|
1808
|
+
|
|
1809
|
+
credentialDisabledTarget = extensionRunner;
|
|
1810
|
+
for (const event of startupCredentialDisabledEvents.splice(0)) {
|
|
1811
|
+
// Discard return: any handler error is routed through runner.onError listeners.
|
|
1812
|
+
void extensionRunner.emitCredentialDisabled(event);
|
|
1813
|
+
}
|
|
1814
|
+
|
|
1815
|
+
const getSessionContext = () => ({
|
|
1816
|
+
sessionManager,
|
|
1817
|
+
modelRegistry,
|
|
1818
|
+
model: agent.state.model,
|
|
1819
|
+
isIdle: () => !session.isStreaming,
|
|
1820
|
+
hasQueuedMessages: () => session.queuedMessageCount > 0,
|
|
1821
|
+
abort: () => {
|
|
1822
|
+
session.abort();
|
|
1823
|
+
},
|
|
1824
|
+
settings,
|
|
1825
|
+
autoApprove: options.autoApprove ?? false,
|
|
1826
|
+
});
|
|
1827
|
+
const toolContextStore = new ToolContextStore(getSessionContext);
|
|
1828
|
+
|
|
1829
|
+
const registeredTools = extensionRunner.getAllRegisteredTools();
|
|
1830
|
+
const allCustomTools = [
|
|
1831
|
+
...registeredTools,
|
|
1832
|
+
...(options.customTools?.map(tool => {
|
|
1833
|
+
const definition = isCustomTool(tool) ? customToolToDefinition(tool) : tool;
|
|
1834
|
+
return { definition, extensionPath: "<sdk>" };
|
|
1835
|
+
}) ?? []),
|
|
1836
|
+
];
|
|
1837
|
+
const wrappedExtensionTools: Tool[] = wrapRegisteredTools(allCustomTools, extensionRunner);
|
|
1838
|
+
|
|
1839
|
+
// All built-in tools are active (conditional tools like git/ask return null from factory if disabled)
|
|
1840
|
+
const toolRegistry = new Map<string, Tool>();
|
|
1841
|
+
for (const tool of builtinTools) {
|
|
1842
|
+
toolRegistry.set(tool.name, tool);
|
|
1843
|
+
}
|
|
1844
|
+
if (!toolRegistry.has("goal") && settings.get("goal.enabled")) {
|
|
1845
|
+
const goalTool = await logger.time("createTools:goal:session", HIDDEN_TOOLS.goal, toolSession);
|
|
1846
|
+
if (goalTool) {
|
|
1847
|
+
toolRegistry.set(goalTool.name, wrapToolWithMetaNotice(goalTool));
|
|
1848
|
+
}
|
|
1849
|
+
}
|
|
1850
|
+
for (const tool of wrappedExtensionTools) {
|
|
1851
|
+
toolRegistry.set(tool.name, tool);
|
|
1852
|
+
}
|
|
1853
|
+
// Wrap every tool with `ExtensionToolWrapper` so the per-tool approval gate runs on every
|
|
1854
|
+
// call site, regardless of whether any user extensions are loaded. See the runner-construction
|
|
1855
|
+
// comment above for the safety invariant this enforces.
|
|
1856
|
+
for (const tool of toolRegistry.values()) {
|
|
1857
|
+
toolRegistry.set(tool.name, new ExtensionToolWrapper(tool, extensionRunner));
|
|
1858
|
+
}
|
|
1859
|
+
if (model?.provider === "cursor") {
|
|
1860
|
+
toolRegistry.delete("edit");
|
|
1861
|
+
}
|
|
1862
|
+
|
|
1863
|
+
// `resolve` is hidden but must stay in the registry whenever any code path can invoke it:
|
|
1864
|
+
// either a deferrable tool stages a preview action, or plan mode installs a standing handler
|
|
1865
|
+
// that consumes `resolve { action: "apply" }` to submit the plan for approval (issue #1428).
|
|
1866
|
+
// Dropping it on read-only sessions (e.g. plan-mode toolset `read`, `search`, `find`,
|
|
1867
|
+
// `web_search`) leaves plan mode unable to exit through the intended path.
|
|
1868
|
+
const hasDeferrableTools = Array.from(toolRegistry.values()).some(tool => tool.deferrable === true);
|
|
1869
|
+
const planModeAvailable = settings.get("plan.enabled");
|
|
1870
|
+
const needsResolveTool = hasDeferrableTools || planModeAvailable;
|
|
1871
|
+
if (!needsResolveTool) {
|
|
1872
|
+
toolRegistry.delete("resolve");
|
|
1873
|
+
} else if (!toolRegistry.has("resolve")) {
|
|
1874
|
+
const resolveTool = await logger.time("createTools:resolve:session", HIDDEN_TOOLS.resolve, toolSession);
|
|
1875
|
+
if (resolveTool) {
|
|
1876
|
+
toolRegistry.set(resolveTool.name, wrapToolWithMetaNotice(resolveTool));
|
|
1877
|
+
}
|
|
1878
|
+
}
|
|
1879
|
+
|
|
1880
|
+
const effectiveDiscoveryMode = resolveEffectiveToolDiscoveryMode(
|
|
1881
|
+
settings,
|
|
1882
|
+
countToolsForAutoDiscovery(toolRegistry.keys()),
|
|
1883
|
+
);
|
|
1884
|
+
if (effectiveDiscoveryMode !== "off" && !toolRegistry.has("search_tool_bm25")) {
|
|
1885
|
+
const searchTool: Tool = new SearchToolBm25Tool(toolSession);
|
|
1886
|
+
toolRegistry.set(
|
|
1887
|
+
searchTool.name,
|
|
1888
|
+
new ExtensionToolWrapper(wrapToolWithMetaNotice(searchTool), extensionRunner) as Tool,
|
|
1889
|
+
);
|
|
1890
|
+
}
|
|
1891
|
+
const mcpDiscoveryEnabled = effectiveDiscoveryMode !== "off"; // back-compat: true when any discovery active
|
|
1892
|
+
|
|
1893
|
+
const reloadSshTool = async (): Promise<AgentTool | null> => {
|
|
1894
|
+
if (!requestedToolNameSet.has("ssh")) return null;
|
|
1895
|
+
const sshTool = (await loadSshTool({
|
|
1896
|
+
...toolSession,
|
|
1897
|
+
cwd: sessionManager.getCwd(),
|
|
1898
|
+
})) as unknown as AgentTool | null;
|
|
1899
|
+
if (!sshTool) return null;
|
|
1900
|
+
const wrapped = wrapToolWithMetaNotice(sshTool);
|
|
1901
|
+
return new ExtensionToolWrapper(wrapped, extensionRunner) as AgentTool;
|
|
1902
|
+
};
|
|
1903
|
+
|
|
1904
|
+
let cursorEventEmitter: ((event: AgentEvent) => void) | undefined;
|
|
1905
|
+
const cursorExecHandlers = new CursorExecHandlers({
|
|
1906
|
+
cwd,
|
|
1907
|
+
tools: toolRegistry,
|
|
1908
|
+
getToolContext: () => toolContextStore.getContext(),
|
|
1909
|
+
emitEvent: event => cursorEventEmitter?.(event),
|
|
1910
|
+
});
|
|
1911
|
+
|
|
1912
|
+
const repeatToolDescriptions = settings.get("repeatToolDescriptions");
|
|
1913
|
+
const eagerTasks = settings.get("task.eager");
|
|
1914
|
+
const intentField = $flag("PI_INTENT_TRACING", settings.get("tools.intentTracing")) ? INTENT_FIELD : undefined;
|
|
1915
|
+
const rebuildSystemPrompt = async (
|
|
1916
|
+
toolNames: string[],
|
|
1917
|
+
tools: Map<string, AgentTool>,
|
|
1918
|
+
): Promise<BuildSystemPromptResult> => {
|
|
1919
|
+
toolContextStore.setToolNames(toolNames);
|
|
1920
|
+
const discoverableMCPTools: DiscoverableTool[] = mcpDiscoveryEnabled
|
|
1921
|
+
? filterBySource(collectDiscoverableTools(tools.values()), "mcp")
|
|
1922
|
+
: [];
|
|
1923
|
+
const activeToolNames = new Set(toolNames);
|
|
1924
|
+
const discoverableBuiltinTools: DiscoverableTool[] =
|
|
1925
|
+
effectiveDiscoveryMode === "all"
|
|
1926
|
+
? collectDiscoverableTools(
|
|
1927
|
+
Array.from(tools.values()).filter(
|
|
1928
|
+
tool => tool.loadMode === "discoverable" && !activeToolNames.has(tool.name),
|
|
1929
|
+
),
|
|
1930
|
+
{ source: "builtin" },
|
|
1931
|
+
)
|
|
1932
|
+
: [];
|
|
1933
|
+
const discoverableToolsForDesc: DiscoverableTool[] = [...discoverableBuiltinTools, ...discoverableMCPTools];
|
|
1934
|
+
const discoverableToolSummary = summarizeDiscoverableTools(discoverableToolsForDesc);
|
|
1935
|
+
const hasDiscoverableTools =
|
|
1936
|
+
mcpDiscoveryEnabled && toolNames.includes("search_tool_bm25") && discoverableToolsForDesc.length > 0;
|
|
1937
|
+
const promptTools = buildSystemPromptToolMetadata(tools, {
|
|
1938
|
+
search_tool_bm25: { description: renderSearchToolBm25Description(discoverableToolsForDesc) },
|
|
1939
|
+
});
|
|
1940
|
+
const memoryBackend = await resolveMemoryBackend(settings);
|
|
1941
|
+
const memoryInstructions = await memoryBackend.buildDeveloperInstructions(agentDir, settings, session);
|
|
1942
|
+
|
|
1943
|
+
// Build combined append prompt: memory instructions + MCP server instructions
|
|
1944
|
+
const serverInstructions = mcpManager?.getServerInstructions();
|
|
1945
|
+
let appendPrompt: string | undefined = memoryInstructions ?? undefined;
|
|
1946
|
+
if (serverInstructions && serverInstructions.size > 0) {
|
|
1947
|
+
const parts: string[] = [];
|
|
1948
|
+
if (appendPrompt) parts.push(appendPrompt);
|
|
1949
|
+
parts.push(
|
|
1950
|
+
"## MCP Server Instructions\n\nThe following instructions are provided by connected MCP servers. They are server-controlled and may not be verified.",
|
|
1951
|
+
);
|
|
1952
|
+
for (const [srvName, srvInstructions] of serverInstructions) {
|
|
1953
|
+
const truncated =
|
|
1954
|
+
srvInstructions.length > MAX_MCP_INSTRUCTIONS_LENGTH
|
|
1955
|
+
? `${srvInstructions.slice(0, MAX_MCP_INSTRUCTIONS_LENGTH)}\n[truncated]`
|
|
1956
|
+
: srvInstructions;
|
|
1957
|
+
parts.push(`### ${srvName}\n${truncated}`);
|
|
1958
|
+
}
|
|
1959
|
+
appendPrompt = parts.join("\n\n");
|
|
1960
|
+
}
|
|
1961
|
+
const defaultPrompt = await buildSystemPromptInternal({
|
|
1962
|
+
cwd,
|
|
1963
|
+
skills,
|
|
1964
|
+
contextFiles,
|
|
1965
|
+
tools: promptTools,
|
|
1966
|
+
toolNames,
|
|
1967
|
+
rules: rulebookRules,
|
|
1968
|
+
alwaysApplyRules,
|
|
1969
|
+
skillsSettings: settings.getGroup("skills"),
|
|
1970
|
+
appendSystemPrompt: appendPrompt,
|
|
1971
|
+
repeatToolDescriptions,
|
|
1972
|
+
intentField,
|
|
1973
|
+
mcpDiscoveryMode: hasDiscoverableTools,
|
|
1974
|
+
mcpDiscoveryServerSummaries: discoverableToolSummary.servers.map(formatDiscoverableToolServerSummary),
|
|
1975
|
+
eagerTasks,
|
|
1976
|
+
secretsEnabled,
|
|
1977
|
+
workspaceTree: workspaceTreePromise,
|
|
1978
|
+
memoryRootEnabled: memoryBackend.id === "local",
|
|
1979
|
+
model: settings.get("includeModelInPrompt") ? getActiveModelString() : undefined,
|
|
1980
|
+
});
|
|
1981
|
+
|
|
1982
|
+
if (options.systemPrompt === undefined) {
|
|
1983
|
+
return defaultPrompt;
|
|
1984
|
+
}
|
|
1985
|
+
if (Array.isArray(options.systemPrompt)) {
|
|
1986
|
+
return { systemPrompt: options.systemPrompt };
|
|
1987
|
+
}
|
|
1988
|
+
return {
|
|
1989
|
+
systemPrompt: options.systemPrompt(defaultPrompt.systemPrompt),
|
|
1990
|
+
};
|
|
1991
|
+
};
|
|
1992
|
+
|
|
1993
|
+
const toolNamesFromRegistry = Array.from(toolRegistry.keys());
|
|
1994
|
+
const explicitlyRequestedToolNames = options.toolNames
|
|
1995
|
+
? [...new Set(options.toolNames.map(name => name.toLowerCase()))]
|
|
1996
|
+
: undefined;
|
|
1997
|
+
// When `requireYieldTool` is set, the subagent's prompts and idle-reminders demand a
|
|
1998
|
+
// `yield` call to terminate. The tool registry already includes `yield` (see
|
|
1999
|
+
// `createTools`), but an explicit `toolNames` list would otherwise drop it from the
|
|
2000
|
+
// active set — leaving the model unable to satisfy the contract. Mirror the same
|
|
2001
|
+
// invariant `parseAgentFields` enforces on frontmatter `tools`.
|
|
2002
|
+
if (
|
|
2003
|
+
options.requireYieldTool === true &&
|
|
2004
|
+
explicitlyRequestedToolNames &&
|
|
2005
|
+
!explicitlyRequestedToolNames.includes("yield")
|
|
2006
|
+
) {
|
|
2007
|
+
explicitlyRequestedToolNames.push("yield");
|
|
2008
|
+
}
|
|
2009
|
+
const requestedToolNames = explicitlyRequestedToolNames ?? toolNamesFromRegistry;
|
|
2010
|
+
const normalizedRequested = requestedToolNames.filter(name => toolRegistry.has(name));
|
|
2011
|
+
const requestedToolNameSet = new Set(normalizedRequested);
|
|
2012
|
+
// Effective discovery mode is resolved after the full registry exists so auto mode can count MCP/extension tools.
|
|
2013
|
+
const defaultInactiveToolNames = new Set(
|
|
2014
|
+
registeredTools.filter(tool => tool.definition.defaultInactive).map(tool => tool.definition.name),
|
|
2015
|
+
);
|
|
2016
|
+
const requestedActiveToolNames = normalizedRequested.filter(name => name !== "goal");
|
|
2017
|
+
const initialRequestedActiveToolNames = options.toolNames
|
|
2018
|
+
? requestedActiveToolNames
|
|
2019
|
+
: requestedActiveToolNames.filter(name => !defaultInactiveToolNames.has(name));
|
|
2020
|
+
const explicitlyRequestedMCPToolNames = options.toolNames
|
|
2021
|
+
? requestedActiveToolNames.filter(name => name.startsWith("mcp__"))
|
|
2022
|
+
: [];
|
|
2023
|
+
const discoveryDefaultServers = new Set(
|
|
2024
|
+
(settings.get("mcp.discoveryDefaultServers") ?? []).map(serverName => serverName.trim()).filter(Boolean),
|
|
2025
|
+
);
|
|
2026
|
+
const discoveryDefaultServerToolNames = mcpDiscoveryEnabled
|
|
2027
|
+
? selectDiscoverableToolNamesByServer(
|
|
2028
|
+
filterBySource(collectDiscoverableTools(toolRegistry.values()), "mcp"),
|
|
2029
|
+
discoveryDefaultServers,
|
|
2030
|
+
)
|
|
2031
|
+
: [];
|
|
2032
|
+
let initialSelectedMCPToolNames: string[] = [];
|
|
2033
|
+
let defaultSelectedMCPToolNames: string[] = [];
|
|
2034
|
+
let initialToolNames = [...initialRequestedActiveToolNames];
|
|
2035
|
+
if (mcpDiscoveryEnabled) {
|
|
2036
|
+
const restoredSelectedMCPToolNames = existingSession.selectedMCPToolNames.filter(name =>
|
|
2037
|
+
toolRegistry.has(name),
|
|
2038
|
+
);
|
|
2039
|
+
defaultSelectedMCPToolNames = [
|
|
2040
|
+
...new Set([...discoveryDefaultServerToolNames, ...explicitlyRequestedMCPToolNames]),
|
|
2041
|
+
];
|
|
2042
|
+
initialSelectedMCPToolNames = existingSession.hasPersistedMCPToolSelection
|
|
2043
|
+
? restoredSelectedMCPToolNames
|
|
2044
|
+
: [...new Set([...restoredSelectedMCPToolNames, ...defaultSelectedMCPToolNames])];
|
|
2045
|
+
initialToolNames = [
|
|
2046
|
+
...new Set([
|
|
2047
|
+
...initialRequestedActiveToolNames.filter(name => !name.startsWith("mcp__")),
|
|
2048
|
+
...initialSelectedMCPToolNames,
|
|
2049
|
+
]),
|
|
2050
|
+
];
|
|
2051
|
+
}
|
|
2052
|
+
|
|
2053
|
+
// Custom tools and extension-registered tools are always included regardless of toolNames filter
|
|
2054
|
+
const alwaysInclude: string[] = [
|
|
2055
|
+
...(options.customTools?.map(t => (isCustomTool(t) ? t.name : t.name)) ?? []),
|
|
2056
|
+
...registeredTools.filter(t => !t.definition.defaultInactive).map(t => t.definition.name),
|
|
2057
|
+
];
|
|
2058
|
+
for (const name of alwaysInclude) {
|
|
2059
|
+
if (mcpDiscoveryEnabled && name.startsWith("mcp__")) {
|
|
2060
|
+
continue;
|
|
2061
|
+
}
|
|
2062
|
+
if (toolRegistry.has(name) && !initialToolNames.includes(name)) {
|
|
2063
|
+
initialToolNames.push(name);
|
|
2064
|
+
}
|
|
2065
|
+
}
|
|
2066
|
+
|
|
2067
|
+
// When tools.discoveryMode === "all", hide non-essential built-in discoverable tools
|
|
2068
|
+
// from the initial set unless they were explicitly requested or restored from persistence.
|
|
2069
|
+
// The model finds them via search_tool_bm25 and activates them on demand.
|
|
2070
|
+
if (effectiveDiscoveryMode === "all") {
|
|
2071
|
+
// Tools a forced tool_choice will target must stay active, or the named
|
|
2072
|
+
// choice references a tool absent from the request (provider 400). Eager
|
|
2073
|
+
// todos force a named `todo` choice on the first turn.
|
|
2074
|
+
const forceActive = new Set<string>();
|
|
2075
|
+
if (settings.get("todo.eager") && settings.get("todo.enabled") && toolRegistry.has("todo")) {
|
|
2076
|
+
forceActive.add("todo");
|
|
2077
|
+
}
|
|
2078
|
+
initialToolNames = filterInitialToolsForDiscoveryAll(initialToolNames, {
|
|
2079
|
+
loadModeOf: name => toolRegistry.get(name)?.loadMode,
|
|
2080
|
+
essentialNames: new Set(computeEssentialBuiltinNames(settings)),
|
|
2081
|
+
explicitlyRequested: new Set(options.toolNames?.map(name => name.toLowerCase()) ?? []),
|
|
2082
|
+
// Back-compat: persisted activations live under selectedMCPToolNames today (built-in
|
|
2083
|
+
// activation persistence is a follow-up). MCP names won't collide with built-in names.
|
|
2084
|
+
restored: new Set(existingSession.selectedMCPToolNames),
|
|
2085
|
+
forceActive,
|
|
2086
|
+
});
|
|
2087
|
+
}
|
|
2088
|
+
|
|
2089
|
+
// Pre-register in the global agent registry BEFORE building the system prompt,
|
|
2090
|
+
// so that subagents launched in the same parallel batch can see each other in
|
|
2091
|
+
// their initial `# IRC Peers` block (rendered inside `rebuildSystemPrompt`).
|
|
2092
|
+
// The session reference is attached after construction below.
|
|
2093
|
+
agentRegistry.register({
|
|
2094
|
+
id: resolvedAgentId,
|
|
2095
|
+
displayName: resolvedAgentDisplayName,
|
|
2096
|
+
kind: agentKind,
|
|
2097
|
+
parentId: options.parentTaskPrefix,
|
|
2098
|
+
session: null,
|
|
2099
|
+
sessionFile: sessionManager.getSessionFile() ?? null,
|
|
2100
|
+
status: "running",
|
|
2101
|
+
});
|
|
2102
|
+
hasRegistered = true;
|
|
2103
|
+
|
|
2104
|
+
const { systemPrompt } = await logger.time(
|
|
2105
|
+
"buildSystemPrompt",
|
|
2106
|
+
rebuildSystemPrompt,
|
|
2107
|
+
initialToolNames,
|
|
2108
|
+
toolRegistry,
|
|
2109
|
+
);
|
|
2110
|
+
|
|
2111
|
+
const promptTemplates = await promptTemplatesPromise;
|
|
2112
|
+
toolSession.promptTemplates = promptTemplates;
|
|
2113
|
+
|
|
2114
|
+
const slashCommands = await slashCommandsPromise;
|
|
2115
|
+
|
|
2116
|
+
// Create convertToLlm wrapper that filters images if blockImages is enabled (defense-in-depth)
|
|
2117
|
+
const convertToLlmWithBlockImages = (messages: AgentMessage[]): Message[] => {
|
|
2118
|
+
const converted = convertToLlm(messages);
|
|
2119
|
+
// Check setting dynamically so mid-session changes take effect
|
|
2120
|
+
if (!settings.get("images.blockImages")) {
|
|
2121
|
+
return converted;
|
|
2122
|
+
}
|
|
2123
|
+
// Filter out ImageContent from all messages, replacing with text placeholder
|
|
2124
|
+
return converted.map(msg => {
|
|
2125
|
+
if (msg.role === "user" || msg.role === "toolResult") {
|
|
2126
|
+
const content = msg.content;
|
|
2127
|
+
if (Array.isArray(content)) {
|
|
2128
|
+
const hasImages = content.some(c => c.type === "image");
|
|
2129
|
+
if (hasImages) {
|
|
2130
|
+
const filteredContent = content
|
|
2131
|
+
.map(c =>
|
|
2132
|
+
c.type === "image" ? { type: "text" as const, text: "Image reading is disabled." } : c,
|
|
2133
|
+
)
|
|
2134
|
+
.filter((c, i, arr) => {
|
|
2135
|
+
// Dedupe consecutive "Image reading is disabled." texts
|
|
2136
|
+
if (!(c.type === "text" && c.text === "Image reading is disabled." && i > 0)) return true;
|
|
2137
|
+
const prev = arr[i - 1];
|
|
2138
|
+
return !(prev.type === "text" && prev.text === "Image reading is disabled.");
|
|
2139
|
+
});
|
|
2140
|
+
return { ...msg, content: filteredContent };
|
|
2141
|
+
}
|
|
2142
|
+
}
|
|
2143
|
+
}
|
|
2144
|
+
return msg;
|
|
2145
|
+
});
|
|
2146
|
+
};
|
|
2147
|
+
|
|
2148
|
+
// Final convertToLlm: chain block-images filter with secret obfuscation
|
|
2149
|
+
const convertToLlmFinal = (messages: AgentMessage[]): Message[] => {
|
|
2150
|
+
const converted = convertToLlmWithBlockImages(messages);
|
|
2151
|
+
if (!obfuscator?.hasSecrets()) return converted;
|
|
2152
|
+
return obfuscateMessages(obfuscator, converted);
|
|
2153
|
+
};
|
|
2154
|
+
|
|
2155
|
+
const transformContext = async (messages: AgentMessage[], _signal?: AbortSignal) => {
|
|
2156
|
+
const withContext = await extensionRunner.emitContext(messages);
|
|
2157
|
+
return wrapSteeringForModel(withContext);
|
|
2158
|
+
};
|
|
2159
|
+
const onPayload = async (payload: unknown, _model?: Model) => {
|
|
2160
|
+
return await extensionRunner.emitBeforeProviderRequest(payload);
|
|
2161
|
+
};
|
|
2162
|
+
const onResponse: SimpleStreamOptions["onResponse"] = async (response, model) => {
|
|
2163
|
+
await extensionRunner.emitAfterProviderResponse(response, model);
|
|
2164
|
+
};
|
|
2165
|
+
|
|
2166
|
+
const setToolUIContext = (uiContext: ExtensionUIContext, hasUI: boolean) => {
|
|
2167
|
+
toolContextStore.setUIContext(uiContext, hasUI);
|
|
2168
|
+
};
|
|
2169
|
+
|
|
2170
|
+
const initialTools = initialToolNames
|
|
2171
|
+
.map(name => toolRegistry.get(name))
|
|
2172
|
+
.filter((tool): tool is AgentTool => tool !== undefined);
|
|
2173
|
+
|
|
2174
|
+
const openaiWebsocketSetting = settings.get("providers.openaiWebsockets") ?? "off";
|
|
2175
|
+
const preferOpenAICodexWebsockets =
|
|
2176
|
+
openaiWebsocketSetting === "on" ? true : openaiWebsocketSetting === "off" ? false : undefined;
|
|
2177
|
+
const serviceTierSetting = settings.get("serviceTier");
|
|
2178
|
+
|
|
2179
|
+
const initialServiceTier = hasServiceTierEntry
|
|
2180
|
+
? existingSession.serviceTier
|
|
2181
|
+
: serviceTierSetting === "none"
|
|
2182
|
+
? undefined
|
|
2183
|
+
: serviceTierSetting;
|
|
2184
|
+
|
|
2185
|
+
agent = new Agent({
|
|
2186
|
+
initialState: {
|
|
2187
|
+
systemPrompt,
|
|
2188
|
+
model,
|
|
2189
|
+
thinkingLevel: toReasoningEffort(effectiveThinkingLevel),
|
|
2190
|
+
disableReasoning: shouldDisableReasoning(effectiveThinkingLevel),
|
|
2191
|
+
tools: initialTools,
|
|
2192
|
+
},
|
|
2193
|
+
convertToLlm: convertToLlmFinal,
|
|
2194
|
+
onPayload,
|
|
2195
|
+
onResponse,
|
|
2196
|
+
sessionId: providerSessionId,
|
|
2197
|
+
promptCacheKey: options.providerPromptCacheKey,
|
|
2198
|
+
transformContext,
|
|
2199
|
+
transformProviderContext: obfuscator ? context => obfuscateProviderContext(obfuscator, context) : undefined,
|
|
2200
|
+
steeringMode: settings.get("steeringMode") ?? "one-at-a-time",
|
|
2201
|
+
followUpMode: settings.get("followUpMode") ?? "one-at-a-time",
|
|
2202
|
+
interruptMode: settings.get("interruptMode") ?? "immediate",
|
|
2203
|
+
thinkingBudgets: settings.getGroup("thinkingBudgets"),
|
|
2204
|
+
temperature: settings.get("temperature") >= 0 ? settings.get("temperature") : undefined,
|
|
2205
|
+
topP: settings.get("topP") >= 0 ? settings.get("topP") : undefined,
|
|
2206
|
+
topK: settings.get("topK") >= 0 ? settings.get("topK") : undefined,
|
|
2207
|
+
minP: settings.get("minP") >= 0 ? settings.get("minP") : undefined,
|
|
2208
|
+
presencePenalty: settings.get("presencePenalty") >= 0 ? settings.get("presencePenalty") : undefined,
|
|
2209
|
+
repetitionPenalty: settings.get("repetitionPenalty") >= 0 ? settings.get("repetitionPenalty") : undefined,
|
|
2210
|
+
serviceTier: initialServiceTier,
|
|
2211
|
+
hideThinkingSummary: settings.get("hideThinkingBlock"),
|
|
2212
|
+
kimiApiFormat: settings.get("providers.kimiApiFormat") ?? "anthropic",
|
|
2213
|
+
preferWebsockets: preferOpenAICodexWebsockets,
|
|
2214
|
+
getToolContext: tc => toolContextStore.getContext(tc),
|
|
2215
|
+
getApiKey: async (provider, ctx) => {
|
|
2216
|
+
// Read agent.sessionId at call time so credential selection stays aligned
|
|
2217
|
+
// with metadataResolver after /new, fork, resume, or branch switches.
|
|
2218
|
+
// Retry steps (ctx carries an auth error) drive the central a/b/c
|
|
2219
|
+
// policy — force-refresh the same account, then rotate to a sibling —
|
|
2220
|
+
// and may legitimately yield no key when every account is exhausted.
|
|
2221
|
+
if (ctx?.error !== undefined) {
|
|
2222
|
+
return createApiKeyResolver(modelRegistry, provider, { sessionId: agent.sessionId })(ctx);
|
|
2223
|
+
}
|
|
2224
|
+
const key = await modelRegistry.getApiKeyForProvider(provider, agent.sessionId);
|
|
2225
|
+
if (!key) {
|
|
2226
|
+
throw new Error(`No API key found for provider "${provider}"`);
|
|
2227
|
+
}
|
|
2228
|
+
return key;
|
|
2229
|
+
},
|
|
2230
|
+
streamFn: (streamModel, context, streamOptions) => {
|
|
2231
|
+
const openrouterRoutingPreset = settings.get("providers.openrouterVariant");
|
|
2232
|
+
const openrouterVariant =
|
|
2233
|
+
openrouterRoutingPreset && openrouterRoutingPreset !== "default" ? openrouterRoutingPreset : undefined;
|
|
2234
|
+
return streamSimple(streamModel, context, {
|
|
2235
|
+
...streamOptions,
|
|
2236
|
+
openrouterVariant: streamOptions?.openrouterVariant ?? openrouterVariant,
|
|
2237
|
+
});
|
|
2238
|
+
},
|
|
2239
|
+
cursorExecHandlers,
|
|
2240
|
+
transformToolCallArguments: (args, _toolName) => {
|
|
2241
|
+
let result = args;
|
|
2242
|
+
const maxTimeout = settings.get("tools.maxTimeout");
|
|
2243
|
+
if (maxTimeout > 0 && typeof result.timeout === "number") {
|
|
2244
|
+
result = { ...result, timeout: Math.min(result.timeout, maxTimeout) };
|
|
2245
|
+
}
|
|
2246
|
+
if (obfuscator?.hasSecrets()) {
|
|
2247
|
+
result = obfuscator.deobfuscateObject(result);
|
|
2248
|
+
}
|
|
2249
|
+
return result;
|
|
2250
|
+
},
|
|
2251
|
+
intentTracing: !!intentField,
|
|
2252
|
+
getToolChoice: () => session?.nextToolChoice(),
|
|
2253
|
+
telemetry: options.telemetry,
|
|
2254
|
+
appendOnlyContext: model
|
|
2255
|
+
? shouldEnableAppendOnlyContext(settings.get("provider.appendOnlyContext"), model)
|
|
2256
|
+
? new AppendOnlyContextManager()
|
|
2257
|
+
: undefined
|
|
2258
|
+
: undefined,
|
|
2259
|
+
});
|
|
2260
|
+
|
|
2261
|
+
cursorEventEmitter = event => agent.emitExternalEvent(event);
|
|
2262
|
+
|
|
2263
|
+
// Restore messages if session has existing data
|
|
2264
|
+
if (hasExistingSession) {
|
|
2265
|
+
agent.replaceMessages(existingSession.messages);
|
|
2266
|
+
} else {
|
|
2267
|
+
// Save initial model, thinking level, and service tier for new sessions so they can be restored on resume.
|
|
2268
|
+
if (model) {
|
|
2269
|
+
sessionManager.appendModelChange(`${model.provider}/${model.id}`);
|
|
2270
|
+
}
|
|
2271
|
+
if (!autoThinking) {
|
|
2272
|
+
// Do not write the `auto` selector before the first turn resolves; auto
|
|
2273
|
+
// classification persists its concrete effort once a real user turn runs.
|
|
2274
|
+
sessionManager.appendThinkingLevelChange(effectiveThinkingLevel);
|
|
2275
|
+
}
|
|
2276
|
+
if (initialServiceTier) {
|
|
2277
|
+
sessionManager.appendServiceTierChange(initialServiceTier);
|
|
2278
|
+
}
|
|
2279
|
+
}
|
|
2280
|
+
|
|
2281
|
+
session = new AgentSession({
|
|
2282
|
+
agent,
|
|
2283
|
+
thinkingLevel: autoThinking ? AUTO_THINKING : effectiveThinkingLevel,
|
|
2284
|
+
sessionManager,
|
|
2285
|
+
settings,
|
|
2286
|
+
autoApprove: options.autoApprove,
|
|
2287
|
+
evalKernelOwnerId,
|
|
2288
|
+
// Defined only for top-level sessions (creation is gated above).
|
|
2289
|
+
// AgentSession uses this to decide whether it may dispose the global
|
|
2290
|
+
// AsyncJobManager on teardown; subagents inherit the parent's and
|
|
2291
|
+
// **MUST NOT** tear it down.
|
|
2292
|
+
ownedAsyncJobManager: asyncJobManager,
|
|
2293
|
+
asyncJobManager: scopedAsyncJobManager,
|
|
2294
|
+
scopedModels: options.scopedModels,
|
|
2295
|
+
promptTemplates,
|
|
2296
|
+
slashCommands,
|
|
2297
|
+
extensionRunner,
|
|
2298
|
+
customCommands: customCommandsResult.commands,
|
|
2299
|
+
skills,
|
|
2300
|
+
skillWarnings,
|
|
2301
|
+
skillsSettings: settings.getGroup("skills"),
|
|
2302
|
+
modelRegistry,
|
|
2303
|
+
toolRegistry,
|
|
2304
|
+
transformContext,
|
|
2305
|
+
onPayload,
|
|
2306
|
+
onResponse,
|
|
2307
|
+
convertToLlm: convertToLlmFinal,
|
|
2308
|
+
rebuildSystemPrompt,
|
|
2309
|
+
reloadSshTool,
|
|
2310
|
+
requestedToolNames: requestedToolNameSet,
|
|
2311
|
+
getMcpServerInstructions: mcpManager
|
|
2312
|
+
? () => {
|
|
2313
|
+
const raw = mcpManager.getServerInstructions();
|
|
2314
|
+
if (!raw || raw.size === 0) return raw;
|
|
2315
|
+
const out = new Map<string, string>();
|
|
2316
|
+
for (const [name, text] of raw) {
|
|
2317
|
+
out.set(
|
|
2318
|
+
name,
|
|
2319
|
+
text.length > MAX_MCP_INSTRUCTIONS_LENGTH ? text.slice(0, MAX_MCP_INSTRUCTIONS_LENGTH) : text,
|
|
2320
|
+
);
|
|
2321
|
+
}
|
|
2322
|
+
return out;
|
|
2323
|
+
}
|
|
2324
|
+
: undefined,
|
|
2325
|
+
mcpDiscoveryEnabled,
|
|
2326
|
+
initialSelectedMCPToolNames,
|
|
2327
|
+
defaultSelectedMCPToolNames,
|
|
2328
|
+
persistInitialMCPToolSelection: !hasExistingSession,
|
|
2329
|
+
defaultSelectedMCPServerNames: [...discoveryDefaultServers],
|
|
2330
|
+
ttsrManager,
|
|
2331
|
+
obfuscator,
|
|
2332
|
+
agentId: resolvedAgentId,
|
|
2333
|
+
providerSessionId: options.providerSessionId,
|
|
2334
|
+
parentEvalSessionId: options.parentEvalSessionId,
|
|
2335
|
+
});
|
|
2336
|
+
hasSession = true;
|
|
2337
|
+
if (asyncJobManager) {
|
|
2338
|
+
session.yieldQueue.register<AsyncResultEntry>("async-result", {
|
|
2339
|
+
isStale: entry => asyncJobManager.isDeliverySuppressed(entry.jobId),
|
|
2340
|
+
build: buildAsyncResultBatchMessage,
|
|
2341
|
+
});
|
|
2342
|
+
}
|
|
2343
|
+
session.yieldQueue.register<McpNotificationEntry>("mcp-notification", {
|
|
2344
|
+
build: buildMcpNotificationBatchMessage,
|
|
2345
|
+
});
|
|
2346
|
+
session.yieldQueue.register<DeferredDiagnosticsEntry>(LSP_LATE_DIAGNOSTIC_MESSAGE_TYPE, {
|
|
2347
|
+
isStale: entry => entry.isStale(),
|
|
2348
|
+
build: buildLateDiagnosticsBatchMessage,
|
|
2349
|
+
});
|
|
2350
|
+
|
|
2351
|
+
// Attach the live session to the pre-registered ref so peers can route IRC
|
|
2352
|
+
// messages here. Refresh sessionFile in case it was unavailable at pre-register
|
|
2353
|
+
// time. The dispose wrapper below unregisters on teardown (unless parked).
|
|
2354
|
+
agentRegistry.attachSession(resolvedAgentId, session, sessionManager.getSessionFile() ?? null);
|
|
2355
|
+
{
|
|
2356
|
+
const originalDispose = session.dispose.bind(session);
|
|
2357
|
+
session.dispose = async () => {
|
|
2358
|
+
try {
|
|
2359
|
+
// Reject new session work (Python/eval starts) the moment disposal
|
|
2360
|
+
// begins — the lifecycle await below opens an async gap before
|
|
2361
|
+
// AgentSession.dispose() would otherwise set its guards.
|
|
2362
|
+
session.beginDispose();
|
|
2363
|
+
if (agentKind === "main") {
|
|
2364
|
+
// Top-level teardown owns the global agent lifecycle: park timers,
|
|
2365
|
+
// adopted subagent sessions, revivers. Tear it down while shared
|
|
2366
|
+
// resources (kernels, MCP, LSP) are still live. Subagent disposal
|
|
2367
|
+
// must NOT touch the global lifecycle.
|
|
2368
|
+
await AgentLifecycleManager.global().dispose();
|
|
2369
|
+
}
|
|
2370
|
+
await originalDispose();
|
|
2371
|
+
} finally {
|
|
2372
|
+
unregisterUnlessParked();
|
|
2373
|
+
unsubscribeCredentialDisabled?.();
|
|
2374
|
+
}
|
|
2375
|
+
};
|
|
2376
|
+
}
|
|
2377
|
+
|
|
2378
|
+
if (model?.api === "openai-codex-responses") {
|
|
2379
|
+
// `.api` equality doesn't narrow the generic; the guard makes this cast sound.
|
|
2380
|
+
const codexModel = model as Model<"openai-codex-responses">;
|
|
2381
|
+
const codexTransport = getOpenAICodexTransportDetails(codexModel, {
|
|
2382
|
+
sessionId: providerSessionId,
|
|
2383
|
+
baseUrl: codexModel.baseUrl,
|
|
2384
|
+
preferWebsockets: preferOpenAICodexWebsockets,
|
|
2385
|
+
providerSessionState: session.providerSessionState,
|
|
2386
|
+
});
|
|
2387
|
+
if (codexTransport.websocketPreferred) {
|
|
2388
|
+
void (async () => {
|
|
2389
|
+
try {
|
|
2390
|
+
const codexPrewarmApiKey = await modelRegistry.getApiKey(codexModel, providerSessionId);
|
|
2391
|
+
if (!codexPrewarmApiKey) return;
|
|
2392
|
+
await logger.time("prewarmOpenAICodexResponses", prewarmOpenAICodexResponses, codexModel, {
|
|
2393
|
+
apiKey: codexPrewarmApiKey,
|
|
2394
|
+
sessionId: providerSessionId,
|
|
2395
|
+
preferWebsockets: preferOpenAICodexWebsockets,
|
|
2396
|
+
providerSessionState: session.providerSessionState,
|
|
2397
|
+
});
|
|
2398
|
+
} catch (error) {
|
|
2399
|
+
const errorMessage = error instanceof Error ? error.message : String(error);
|
|
2400
|
+
logger.debug("Codex websocket prewarm failed", {
|
|
2401
|
+
error: errorMessage,
|
|
2402
|
+
provider: codexModel.provider,
|
|
2403
|
+
model: codexModel.id,
|
|
2404
|
+
});
|
|
2405
|
+
}
|
|
2406
|
+
})();
|
|
2407
|
+
}
|
|
2408
|
+
}
|
|
2409
|
+
|
|
2410
|
+
// Start LSP warmup in the background so startup does not block on language server initialization.
|
|
2411
|
+
// With `lsp.lazy` (the default) the warmup is skipped: recognized servers are still discovered and
|
|
2412
|
+
// surfaced in the UI as "available", but cold-start on first use — the lsp tool or an edit/write
|
|
2413
|
+
// touching a matching file type — through `getOrCreateClient`.
|
|
2414
|
+
// Print/script invocations (`hasUI=false`) skip it regardless: they don't render the warmup status
|
|
2415
|
+
// indicator AND typically finish before LSP servers would have stabilized — warming them just spends
|
|
2416
|
+
// CPU parsing big `initialize` responses concurrently with the LLM stream consumer, jittering
|
|
2417
|
+
// perceived latency.
|
|
2418
|
+
let lspServers: CreateAgentSessionResult["lspServers"];
|
|
2419
|
+
if (enableLsp && options.hasUI && settings.get("lsp.lazy")) {
|
|
2420
|
+
lspServers = discoverStartupLspServers(cwd, "available");
|
|
2421
|
+
} else if (enableLsp && options.hasUI) {
|
|
2422
|
+
lspServers = discoverStartupLspServers(cwd);
|
|
2423
|
+
if (lspServers.length > 0) {
|
|
2424
|
+
void (async () => {
|
|
2425
|
+
try {
|
|
2426
|
+
const result = await logger.time("warmupLspServers", warmupLspServers, cwd);
|
|
2427
|
+
const serversByName = new Map(result.servers.map(server => [server.name, server] as const));
|
|
2428
|
+
for (const server of lspServers ?? []) {
|
|
2429
|
+
const next = serversByName.get(server.name);
|
|
2430
|
+
if (!next) continue;
|
|
2431
|
+
server.status = next.status;
|
|
2432
|
+
server.fileTypes = next.fileTypes;
|
|
2433
|
+
server.error = next.error;
|
|
2434
|
+
}
|
|
2435
|
+
const event: LspStartupEvent = {
|
|
2436
|
+
type: "completed",
|
|
2437
|
+
servers: result.servers,
|
|
2438
|
+
};
|
|
2439
|
+
eventBus.emit(LSP_STARTUP_EVENT_CHANNEL, event);
|
|
2440
|
+
} catch (error) {
|
|
2441
|
+
const errorMessage = error instanceof Error ? error.message : String(error);
|
|
2442
|
+
logger.warn("LSP server warmup failed", { cwd, error: errorMessage });
|
|
2443
|
+
for (const server of lspServers ?? []) {
|
|
2444
|
+
server.status = "error";
|
|
2445
|
+
server.error = errorMessage;
|
|
2446
|
+
}
|
|
2447
|
+
const event: LspStartupEvent = {
|
|
2448
|
+
type: "failed",
|
|
2449
|
+
error: errorMessage,
|
|
2450
|
+
};
|
|
2451
|
+
eventBus.emit(LSP_STARTUP_EVENT_CHANNEL, event);
|
|
2452
|
+
}
|
|
2453
|
+
})();
|
|
2454
|
+
}
|
|
2455
|
+
}
|
|
2456
|
+
|
|
2457
|
+
logger.time("startMemoryStartupTask", async () => {
|
|
2458
|
+
const memoryBackend = await resolveMemoryBackend(settings);
|
|
2459
|
+
await memoryBackend.start({
|
|
2460
|
+
session,
|
|
2461
|
+
settings,
|
|
2462
|
+
modelRegistry,
|
|
2463
|
+
agentDir,
|
|
2464
|
+
taskDepth,
|
|
2465
|
+
parentHindsightSessionState: options.parentHindsightSessionState,
|
|
2466
|
+
parentMnemopiSessionState: options.parentMnemopiSessionState,
|
|
2467
|
+
});
|
|
2468
|
+
});
|
|
2469
|
+
|
|
2470
|
+
// Wire MCP manager callbacks to session for reactive tool updates.
|
|
2471
|
+
// Skip when reusing a parent's manager — the parent owns the callbacks.
|
|
2472
|
+
if (mcpManager && !options.mcpManager) {
|
|
2473
|
+
mcpManager.setOnToolsChanged(tools => {
|
|
2474
|
+
void session.refreshMCPTools(tools);
|
|
2475
|
+
});
|
|
2476
|
+
// Wire prompt refresh → rebuild MCP prompt slash commands
|
|
2477
|
+
mcpManager.setOnPromptsChanged(serverName => {
|
|
2478
|
+
const promptCommands = buildMCPPromptCommands(mcpManager);
|
|
2479
|
+
session.setMCPPromptCommands(promptCommands);
|
|
2480
|
+
logger.debug("MCP prompt commands refreshed", { path: `mcp:${serverName}` });
|
|
2481
|
+
});
|
|
2482
|
+
const notificationDebounceTimers = new Map<string, Timer>();
|
|
2483
|
+
const clearDebounceTimers = () => {
|
|
2484
|
+
for (const timer of notificationDebounceTimers.values()) clearTimeout(timer);
|
|
2485
|
+
notificationDebounceTimers.clear();
|
|
2486
|
+
};
|
|
2487
|
+
postmortem.register("mcp-notification-cleanup", clearDebounceTimers);
|
|
2488
|
+
mcpManager.setOnResourcesChanged((serverName, uri) => {
|
|
2489
|
+
logger.debug("MCP resources changed", { path: `mcp:${serverName}`, uri });
|
|
2490
|
+
if (!settings.get("mcp.notifications")) return;
|
|
2491
|
+
const debounceMs = settings.get("mcp.notificationDebounceMs");
|
|
2492
|
+
const key = `${serverName}:${uri}`;
|
|
2493
|
+
const existing = notificationDebounceTimers.get(key);
|
|
2494
|
+
if (existing) clearTimeout(existing);
|
|
2495
|
+
notificationDebounceTimers.set(
|
|
2496
|
+
key,
|
|
2497
|
+
setTimeout(() => {
|
|
2498
|
+
notificationDebounceTimers.delete(key);
|
|
2499
|
+
// Re-check: user may have disabled notifications during the debounce window
|
|
2500
|
+
if (!settings.get("mcp.notifications")) return;
|
|
2501
|
+
session.yieldQueue.enqueue<McpNotificationEntry>("mcp-notification", { serverName, uri });
|
|
2502
|
+
}, debounceMs),
|
|
2503
|
+
);
|
|
2504
|
+
});
|
|
2505
|
+
}
|
|
2506
|
+
|
|
2507
|
+
return {
|
|
2508
|
+
session,
|
|
2509
|
+
extensionsResult,
|
|
2510
|
+
setToolUIContext,
|
|
2511
|
+
mcpManager,
|
|
2512
|
+
modelFallbackMessage,
|
|
2513
|
+
lspServers,
|
|
2514
|
+
eventBus,
|
|
2515
|
+
};
|
|
2516
|
+
} catch (error) {
|
|
2517
|
+
// Release the subscription if the throw happened after install but before the
|
|
2518
|
+
// dispose-wrap took ownership. Idempotent with dispose() — Set.delete is a no-op
|
|
2519
|
+
// for already-removed listeners.
|
|
2520
|
+
unsubscribeCredentialDisabled?.();
|
|
2521
|
+
try {
|
|
2522
|
+
if (hasSession) {
|
|
2523
|
+
await session.dispose();
|
|
2524
|
+
} else {
|
|
2525
|
+
if (hasRegistered) unregisterUnlessParked();
|
|
2526
|
+
if (asyncJobManager) {
|
|
2527
|
+
if (AsyncJobManager.instance() === asyncJobManager) {
|
|
2528
|
+
AsyncJobManager.setInstance(undefined);
|
|
2529
|
+
}
|
|
2530
|
+
await asyncJobManager.dispose({ timeoutMs: 3_000 });
|
|
2531
|
+
}
|
|
2532
|
+
await disposeKernelSessionsByOwner(evalKernelOwnerId);
|
|
2533
|
+
}
|
|
2534
|
+
} catch (cleanupError) {
|
|
2535
|
+
logger.warn("Failed to clean up createAgentSession resources after startup error", {
|
|
2536
|
+
error: cleanupError instanceof Error ? cleanupError.message : String(cleanupError),
|
|
2537
|
+
});
|
|
2538
|
+
}
|
|
2539
|
+
throw error;
|
|
2540
|
+
}
|
|
2541
|
+
}
|
|
2542
|
+
|
|
2543
|
+
/**
|
|
2544
|
+
* Best-effort preconnect to the model's API host. Bun's `fetch.preconnect`
|
|
2545
|
+
* primes DNS + TCP + TLS + H2 so the first real request reuses the warm
|
|
2546
|
+
* connection. Errors are swallowed: preconnect is an optimization, never a
|
|
2547
|
+
* hard dependency.
|
|
2548
|
+
*/
|
|
2549
|
+
function preconnectModelHost(baseUrl: string | undefined): void {
|
|
2550
|
+
if (!baseUrl) return;
|
|
2551
|
+
const preconnect = (globalThis.fetch as typeof fetch & { preconnect?: (url: string) => void }).preconnect;
|
|
2552
|
+
if (typeof preconnect !== "function") return;
|
|
2553
|
+
try {
|
|
2554
|
+
preconnect(baseUrl);
|
|
2555
|
+
} catch {
|
|
2556
|
+
// Best effort.
|
|
2557
|
+
}
|
|
2558
|
+
}
|