@epoch-ai/cli 2.2.5 → 2.2.7
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/.artifacts/unit/junit.xml +2823 -0
- package/.project-map/backups/20260530_223453/.project-map.json +90101 -0
- package/.project-map/backups/20260530_223507/.project-map.json +90101 -0
- package/.project-map/backups/20260530_223512/.project-map.json +90101 -0
- package/.project-map/backups/20260530_223512/map.toon +666 -0
- package/.project-map/backups/20260530_223516/.project-map.json +90101 -0
- package/.project-map/backups/20260530_223516/map.toon +666 -0
- package/.project-map/backups/20260530_223520/.project-map.json +90101 -0
- package/.project-map/backups/20260530_223520/map.toon +666 -0
- package/AGENTS.md +47 -0
- package/BUN_SHELL_MIGRATION_PLAN.md +136 -0
- package/Dockerfile +18 -0
- package/README.md +15 -0
- package/bunfig.toml +7 -0
- package/drizzle.config.ts +10 -0
- package/git +0 -0
- package/migration/20260127222353_familiar_lady_ursula/migration.sql +90 -0
- package/migration/20260127222353_familiar_lady_ursula/snapshot.json +796 -0
- package/migration/20260211171708_add_project_commands/migration.sql +1 -0
- package/migration/20260211171708_add_project_commands/snapshot.json +806 -0
- package/migration/20260213144116_wakeful_the_professor/migration.sql +11 -0
- package/migration/20260213144116_wakeful_the_professor/snapshot.json +897 -0
- package/migration/20260225215848_workspace/migration.sql +7 -0
- package/migration/20260225215848_workspace/snapshot.json +959 -0
- package/migration/20260227213759_add_session_workspace_id/migration.sql +2 -0
- package/migration/20260227213759_add_session_workspace_id/snapshot.json +983 -0
- package/migration/20260228203230_blue_harpoon/migration.sql +17 -0
- package/migration/20260228203230_blue_harpoon/snapshot.json +1102 -0
- package/migration/20260303231226_add_workspace_fields/migration.sql +5 -0
- package/migration/20260303231226_add_workspace_fields/snapshot.json +1013 -0
- package/migration/20260309230000_move_org_to_state/migration.sql +3 -0
- package/migration/20260309230000_move_org_to_state/snapshot.json +1156 -0
- package/migration/20260312043431_session_message_cursor/migration.sql +4 -0
- package/migration/20260312043431_session_message_cursor/snapshot.json +1168 -0
- package/migration/20260323234822_events/migration.sql +13 -0
- package/migration/20260323234822_events/snapshot.json +1271 -0
- package/migration/20260418092949_add_yolo_to_session/migration.sql +2 -0
- package/migration/20260418092949_add_yolo_to_session/snapshot.json +1199 -0
- package/migration/20260419120000_add_intervention_to_session/migration.sql +2 -0
- package/package.json +179 -18
- package/parsers-config.ts +290 -0
- package/script/build-node.ts +71 -0
- package/script/build.ts +255 -0
- package/script/check-migrations.ts +16 -0
- package/script/fix-node-pty.ts +28 -0
- package/script/publish.ts +182 -0
- package/script/schema.ts +63 -0
- package/script/seed-e2e.ts +60 -0
- package/script/upgrade-opentui.ts +64 -0
- package/specs/effect-migration.md +310 -0
- package/specs/tui-plugins.md +436 -0
- package/specs/v2.md +14 -0
- package/src/account/account.sql.ts +39 -0
- package/src/account/index.ts +465 -0
- package/src/account/repo.ts +163 -0
- package/src/account/schema.ts +91 -0
- package/src/acp/README.md +174 -0
- package/src/acp/agent.ts +1847 -0
- package/src/acp/session.ts +116 -0
- package/src/acp/types.ts +24 -0
- package/src/agent/agent.ts +445 -0
- package/src/agent/generate.txt +75 -0
- package/src/agent/prompt/compaction.txt +15 -0
- package/src/agent/prompt/explore.txt +9 -0
- package/src/agent/prompt/summary.txt +11 -0
- package/src/agent/prompt/title.txt +44 -0
- package/src/auth/index.ts +110 -0
- package/src/bus/bus-event.ts +40 -0
- package/src/bus/global.ts +10 -0
- package/src/bus/index.ts +232 -0
- package/src/cli/bootstrap.ts +17 -0
- package/src/cli/cmd/account.ts +257 -0
- package/src/cli/cmd/acp.ts +70 -0
- package/src/cli/cmd/agent.ts +245 -0
- package/src/cli/cmd/cmd.ts +7 -0
- package/src/cli/cmd/db.ts +119 -0
- package/src/cli/cmd/debug/agent.ts +167 -0
- package/src/cli/cmd/debug/config.ts +16 -0
- package/src/cli/cmd/debug/file.ts +97 -0
- package/src/cli/cmd/debug/index.ts +48 -0
- package/src/cli/cmd/debug/lsp.ts +53 -0
- package/src/cli/cmd/debug/ripgrep.ts +87 -0
- package/src/cli/cmd/debug/scrap.ts +16 -0
- package/src/cli/cmd/debug/skill.ts +16 -0
- package/src/cli/cmd/debug/snapshot.ts +52 -0
- package/src/cli/cmd/export.ts +89 -0
- package/src/cli/cmd/generate.ts +38 -0
- package/src/cli/cmd/github.ts +1639 -0
- package/src/cli/cmd/import.ts +169 -0
- package/src/cli/cmd/mcp.ts +754 -0
- package/src/cli/cmd/models.ts +78 -0
- package/src/cli/cmd/plug.ts +233 -0
- package/src/cli/cmd/pr.ts +127 -0
- package/src/cli/cmd/providers.ts +478 -0
- package/src/cli/cmd/run.ts +681 -0
- package/src/cli/cmd/serve.ts +24 -0
- package/src/cli/cmd/session.ts +159 -0
- package/src/cli/cmd/stats.ts +410 -0
- package/src/cli/cmd/tui/app.tsx +945 -0
- package/src/cli/cmd/tui/attach.ts +88 -0
- package/src/cli/cmd/tui/component/border.tsx +21 -0
- package/src/cli/cmd/tui/component/dialog-agent.tsx +31 -0
- package/src/cli/cmd/tui/component/dialog-command.tsx +171 -0
- package/src/cli/cmd/tui/component/dialog-console-org.tsx +103 -0
- package/src/cli/cmd/tui/component/dialog-mcp.tsx +86 -0
- package/src/cli/cmd/tui/component/dialog-model.tsx +190 -0
- package/src/cli/cmd/tui/component/dialog-provider.tsx +364 -0
- package/src/cli/cmd/tui/component/dialog-session-list.tsx +108 -0
- package/src/cli/cmd/tui/component/dialog-session-rename.tsx +31 -0
- package/src/cli/cmd/tui/component/dialog-skill.tsx +36 -0
- package/src/cli/cmd/tui/component/dialog-stash.tsx +87 -0
- package/src/cli/cmd/tui/component/dialog-status.tsx +168 -0
- package/src/cli/cmd/tui/component/dialog-tag.tsx +44 -0
- package/src/cli/cmd/tui/component/dialog-theme-list.tsx +50 -0
- package/src/cli/cmd/tui/component/dialog-variant.tsx +39 -0
- package/src/cli/cmd/tui/component/dialog-workspace-list.tsx +320 -0
- package/src/cli/cmd/tui/component/error-component.tsx +92 -0
- package/src/cli/cmd/tui/component/logo.tsx +85 -0
- package/src/cli/cmd/tui/component/plugin-route-missing.tsx +14 -0
- package/src/cli/cmd/tui/component/prompt/autocomplete.tsx +672 -0
- package/src/cli/cmd/tui/component/prompt/frecency.tsx +90 -0
- package/src/cli/cmd/tui/component/prompt/history.tsx +109 -0
- package/src/cli/cmd/tui/component/prompt/index.tsx +1348 -0
- package/src/cli/cmd/tui/component/prompt/part.ts +16 -0
- package/src/cli/cmd/tui/component/prompt/stash.tsx +101 -0
- package/src/cli/cmd/tui/component/spinner.tsx +24 -0
- package/src/cli/cmd/tui/component/startup-loading.tsx +63 -0
- package/src/cli/cmd/tui/component/textarea-keybindings.ts +73 -0
- package/src/cli/cmd/tui/component/todo-item.tsx +32 -0
- package/src/cli/cmd/tui/component/workspace/dialog-session-list.tsx +151 -0
- package/src/cli/cmd/tui/context/args.tsx +15 -0
- package/src/cli/cmd/tui/context/directory.ts +13 -0
- package/src/cli/cmd/tui/context/exit.tsx +60 -0
- package/src/cli/cmd/tui/context/helper.tsx +25 -0
- package/src/cli/cmd/tui/context/keybind.tsx +105 -0
- package/src/cli/cmd/tui/context/kv.tsx +52 -0
- package/src/cli/cmd/tui/context/local.tsx +456 -0
- package/src/cli/cmd/tui/context/plugin-keybinds.ts +41 -0
- package/src/cli/cmd/tui/context/prompt.tsx +18 -0
- package/src/cli/cmd/tui/context/route.tsx +52 -0
- package/src/cli/cmd/tui/context/sdk.tsx +115 -0
- package/src/cli/cmd/tui/context/sync.tsx +516 -0
- package/src/cli/cmd/tui/context/theme/aura.json +69 -0
- package/src/cli/cmd/tui/context/theme/ayu.json +80 -0
- package/src/cli/cmd/tui/context/theme/carbonfox.json +248 -0
- package/src/cli/cmd/tui/context/theme/catppuccin-frappe.json +233 -0
- package/src/cli/cmd/tui/context/theme/catppuccin-macchiato.json +233 -0
- package/src/cli/cmd/tui/context/theme/catppuccin.json +112 -0
- package/src/cli/cmd/tui/context/theme/cobalt2.json +228 -0
- package/src/cli/cmd/tui/context/theme/cursor.json +249 -0
- package/src/cli/cmd/tui/context/theme/dracula.json +219 -0
- package/src/cli/cmd/tui/context/theme/epochcli.json +245 -0
- package/src/cli/cmd/tui/context/theme/everforest.json +241 -0
- package/src/cli/cmd/tui/context/theme/flexoki.json +237 -0
- package/src/cli/cmd/tui/context/theme/github.json +233 -0
- package/src/cli/cmd/tui/context/theme/gruvbox.json +242 -0
- package/src/cli/cmd/tui/context/theme/kanagawa.json +77 -0
- package/src/cli/cmd/tui/context/theme/lucent-orng.json +237 -0
- package/src/cli/cmd/tui/context/theme/material.json +235 -0
- package/src/cli/cmd/tui/context/theme/matrix.json +77 -0
- package/src/cli/cmd/tui/context/theme/mercury.json +252 -0
- package/src/cli/cmd/tui/context/theme/monokai.json +221 -0
- package/src/cli/cmd/tui/context/theme/nightowl.json +221 -0
- package/src/cli/cmd/tui/context/theme/nord.json +223 -0
- package/src/cli/cmd/tui/context/theme/one-dark.json +84 -0
- package/src/cli/cmd/tui/context/theme/orng.json +249 -0
- package/src/cli/cmd/tui/context/theme/osaka-jade.json +93 -0
- package/src/cli/cmd/tui/context/theme/palenight.json +222 -0
- package/src/cli/cmd/tui/context/theme/rosepine.json +234 -0
- package/src/cli/cmd/tui/context/theme/solarized.json +223 -0
- package/src/cli/cmd/tui/context/theme/synthwave84.json +226 -0
- package/src/cli/cmd/tui/context/theme/tokyonight.json +243 -0
- package/src/cli/cmd/tui/context/theme/vercel.json +245 -0
- package/src/cli/cmd/tui/context/theme/vesper.json +218 -0
- package/src/cli/cmd/tui/context/theme/zenburn.json +223 -0
- package/src/cli/cmd/tui/context/theme.tsx +1236 -0
- package/src/cli/cmd/tui/context/tui-config.tsx +9 -0
- package/src/cli/cmd/tui/event.ts +48 -0
- package/src/cli/cmd/tui/feature-plugins/home/footer.tsx +93 -0
- package/src/cli/cmd/tui/feature-plugins/home/tips-view.tsx +145 -0
- package/src/cli/cmd/tui/feature-plugins/home/tips.tsx +50 -0
- package/src/cli/cmd/tui/feature-plugins/sidebar/context.tsx +82 -0
- package/src/cli/cmd/tui/feature-plugins/sidebar/files.tsx +62 -0
- package/src/cli/cmd/tui/feature-plugins/sidebar/footer.tsx +93 -0
- package/src/cli/cmd/tui/feature-plugins/sidebar/lsp.tsx +66 -0
- package/src/cli/cmd/tui/feature-plugins/sidebar/mcp.tsx +96 -0
- package/src/cli/cmd/tui/feature-plugins/sidebar/todo.tsx +48 -0
- package/src/cli/cmd/tui/feature-plugins/system/plugins.tsx +270 -0
- package/src/cli/cmd/tui/plugin/api.tsx +397 -0
- package/src/cli/cmd/tui/plugin/index.ts +3 -0
- package/src/cli/cmd/tui/plugin/internal.ts +27 -0
- package/src/cli/cmd/tui/plugin/runtime.ts +1031 -0
- package/src/cli/cmd/tui/plugin/slots.tsx +60 -0
- package/src/cli/cmd/tui/routes/home.tsx +94 -0
- package/src/cli/cmd/tui/routes/session/dialog-fork-from-timeline.tsx +65 -0
- package/src/cli/cmd/tui/routes/session/dialog-message.tsx +110 -0
- package/src/cli/cmd/tui/routes/session/dialog-subagent.tsx +26 -0
- package/src/cli/cmd/tui/routes/session/dialog-timeline.tsx +47 -0
- package/src/cli/cmd/tui/routes/session/footer.tsx +91 -0
- package/src/cli/cmd/tui/routes/session/index.tsx +2161 -0
- package/src/cli/cmd/tui/routes/session/permission.tsx +691 -0
- package/src/cli/cmd/tui/routes/session/question.tsx +468 -0
- package/src/cli/cmd/tui/routes/session/sidebar.tsx +70 -0
- package/src/cli/cmd/tui/routes/session/subagent-footer.tsx +131 -0
- package/src/cli/cmd/tui/thread.ts +241 -0
- package/src/cli/cmd/tui/ui/dialog-alert.tsx +59 -0
- package/src/cli/cmd/tui/ui/dialog-confirm.tsx +89 -0
- package/src/cli/cmd/tui/ui/dialog-export-options.tsx +211 -0
- package/src/cli/cmd/tui/ui/dialog-help.tsx +40 -0
- package/src/cli/cmd/tui/ui/dialog-prompt.tsx +115 -0
- package/src/cli/cmd/tui/ui/dialog-select.tsx +417 -0
- package/src/cli/cmd/tui/ui/dialog.tsx +192 -0
- package/src/cli/cmd/tui/ui/link.tsx +28 -0
- package/src/cli/cmd/tui/ui/spinner.ts +368 -0
- package/src/cli/cmd/tui/ui/toast.tsx +100 -0
- package/src/cli/cmd/tui/util/clipboard.ts +192 -0
- package/src/cli/cmd/tui/util/editor.ts +37 -0
- package/src/cli/cmd/tui/util/model.ts +23 -0
- package/src/cli/cmd/tui/util/provider-origin.ts +20 -0
- package/src/cli/cmd/tui/util/scroll.ts +23 -0
- package/src/cli/cmd/tui/util/selection.ts +25 -0
- package/src/cli/cmd/tui/util/signal.ts +7 -0
- package/src/cli/cmd/tui/util/terminal.ts +114 -0
- package/src/cli/cmd/tui/util/transcript.ts +112 -0
- package/src/cli/cmd/tui/win32.ts +129 -0
- package/src/cli/cmd/tui/worker.ts +195 -0
- package/src/cli/cmd/uninstall.ts +353 -0
- package/src/cli/cmd/upgrade.ts +73 -0
- package/src/cli/cmd/web.ts +81 -0
- package/src/cli/effect/prompt.ts +25 -0
- package/src/cli/error.ts +46 -0
- package/src/cli/heap.ts +59 -0
- package/src/cli/logo.ts +6 -0
- package/src/cli/network.ts +60 -0
- package/src/cli/ui.ts +133 -0
- package/src/cli/upgrade.ts +31 -0
- package/src/command/index.ts +197 -0
- package/src/command/template/initialize.txt +66 -0
- package/src/command/template/review.txt +101 -0
- package/src/config/config.ts +1610 -0
- package/src/config/console-state.ts +15 -0
- package/src/config/markdown.ts +99 -0
- package/src/config/paths.ts +167 -0
- package/src/config/tui-migrate.ts +155 -0
- package/src/config/tui-schema.ts +37 -0
- package/src/config/tui.ts +179 -0
- package/src/config/validator.ts +52 -0
- package/src/control-plane/adaptors/index.ts +20 -0
- package/src/control-plane/adaptors/worktree.ts +42 -0
- package/src/control-plane/schema.ts +17 -0
- package/src/control-plane/sse.ts +66 -0
- package/src/control-plane/types.ts +32 -0
- package/src/control-plane/workspace.sql.ts +17 -0
- package/src/control-plane/workspace.ts +168 -0
- package/src/effect/cross-spawn-spawner.ts +502 -0
- package/src/effect/instance-ref.ts +6 -0
- package/src/effect/instance-registry.ts +12 -0
- package/src/effect/instance-state.ts +82 -0
- package/src/effect/run-service.ts +33 -0
- package/src/effect/runner.ts +216 -0
- package/src/env/index.ts +28 -0
- package/src/file/ignore.ts +82 -0
- package/src/file/index.ts +686 -0
- package/src/file/protected.ts +59 -0
- package/src/file/ripgrep.ts +376 -0
- package/src/file/time.ts +133 -0
- package/src/file/watcher.ts +172 -0
- package/src/filesystem/index.ts +236 -0
- package/src/flag/flag.ts +157 -0
- package/src/format/formatter.ts +413 -0
- package/src/format/index.ts +203 -0
- package/src/git/index.ts +303 -0
- package/src/global/index.ts +54 -0
- package/src/id/id.ts +85 -0
- package/src/ide/index.ts +74 -0
- package/src/index.ts +253 -0
- package/src/installation/index.ts +355 -0
- package/src/installation/meta.ts +7 -0
- package/src/lsp/client.ts +256 -0
- package/src/lsp/index.ts +558 -0
- package/src/lsp/language.ts +120 -0
- package/src/lsp/launch.ts +21 -0
- package/src/lsp/server.ts +1968 -0
- package/src/mcp/auth.ts +173 -0
- package/src/mcp/index.ts +1250 -0
- package/src/mcp/oauth-callback.ts +216 -0
- package/src/mcp/oauth-provider.ts +185 -0
- package/src/mcp/schema-loader.ts +82 -0
- package/src/node.ts +1 -0
- package/src/npm/index.ts +188 -0
- package/src/patch/index.ts +680 -0
- package/src/permission/arity.ts +163 -0
- package/src/permission/evaluate.ts +15 -0
- package/src/permission/index.ts +323 -0
- package/src/permission/schema.ts +17 -0
- package/src/plugin/cloudflare.ts +67 -0
- package/src/plugin/codex.ts +608 -0
- package/src/plugin/github-copilot/copilot.ts +361 -0
- package/src/plugin/github-copilot/models.ts +144 -0
- package/src/plugin/index.ts +288 -0
- package/src/plugin/install.ts +439 -0
- package/src/plugin/loader.ts +174 -0
- package/src/plugin/meta.ts +188 -0
- package/src/plugin/shared.ts +323 -0
- package/src/project/bootstrap.ts +35 -0
- package/src/project/init-files.ts +328 -0
- package/src/project/instance.ts +175 -0
- package/src/project/project.sql.ts +16 -0
- package/src/project/project.ts +519 -0
- package/src/project/schema.ts +16 -0
- package/src/project/state.ts +70 -0
- package/src/project/vcs.ts +240 -0
- package/src/provider/auth.ts +253 -0
- package/src/provider/error.ts +297 -0
- package/src/provider/models.ts +162 -0
- package/src/provider/provider.ts +1776 -0
- package/src/provider/schema.ts +38 -0
- package/src/provider/sdk/copilot/README.md +5 -0
- package/src/provider/sdk/copilot/chat/convert-to-openai-compatible-chat-messages.ts +170 -0
- package/src/provider/sdk/copilot/chat/get-response-metadata.ts +15 -0
- package/src/provider/sdk/copilot/chat/map-openai-compatible-finish-reason.ts +19 -0
- package/src/provider/sdk/copilot/chat/openai-compatible-api-types.ts +64 -0
- package/src/provider/sdk/copilot/chat/openai-compatible-chat-language-model.ts +814 -0
- package/src/provider/sdk/copilot/chat/openai-compatible-chat-options.ts +28 -0
- package/src/provider/sdk/copilot/chat/openai-compatible-metadata-extractor.ts +44 -0
- package/src/provider/sdk/copilot/chat/openai-compatible-prepare-tools.ts +83 -0
- package/src/provider/sdk/copilot/copilot-provider.ts +100 -0
- package/src/provider/sdk/copilot/index.ts +2 -0
- package/src/provider/sdk/copilot/openai-compatible-error.ts +27 -0
- package/src/provider/sdk/copilot/responses/convert-to-openai-responses-input.ts +335 -0
- package/src/provider/sdk/copilot/responses/map-openai-responses-finish-reason.ts +22 -0
- package/src/provider/sdk/copilot/responses/openai-config.ts +18 -0
- package/src/provider/sdk/copilot/responses/openai-error.ts +22 -0
- package/src/provider/sdk/copilot/responses/openai-responses-api-types.ts +214 -0
- package/src/provider/sdk/copilot/responses/openai-responses-language-model.ts +1769 -0
- package/src/provider/sdk/copilot/responses/openai-responses-prepare-tools.ts +173 -0
- package/src/provider/sdk/copilot/responses/openai-responses-settings.ts +1 -0
- package/src/provider/sdk/copilot/responses/tool/code-interpreter.ts +87 -0
- package/src/provider/sdk/copilot/responses/tool/file-search.ts +127 -0
- package/src/provider/sdk/copilot/responses/tool/image-generation.ts +114 -0
- package/src/provider/sdk/copilot/responses/tool/local-shell.ts +64 -0
- package/src/provider/sdk/copilot/responses/tool/web-search-preview.ts +103 -0
- package/src/provider/sdk/copilot/responses/tool/web-search.ts +102 -0
- package/src/provider/transform.ts +1124 -0
- package/src/pty/index.ts +397 -0
- package/src/pty/pty.bun.ts +26 -0
- package/src/pty/pty.node.ts +27 -0
- package/src/pty/pty.ts +25 -0
- package/src/pty/schema.ts +17 -0
- package/src/question/index.ts +224 -0
- package/src/question/schema.ts +17 -0
- package/src/server/error.ts +36 -0
- package/src/server/event.ts +7 -0
- package/src/server/instance.ts +315 -0
- package/src/server/mdns.ts +60 -0
- package/src/server/middleware.ts +33 -0
- package/src/server/projectors.ts +28 -0
- package/src/server/proxy.ts +130 -0
- package/src/server/router.ts +105 -0
- package/src/server/routes/config.ts +92 -0
- package/src/server/routes/event.ts +83 -0
- package/src/server/routes/experimental.ts +374 -0
- package/src/server/routes/file.ts +197 -0
- package/src/server/routes/global.ts +312 -0
- package/src/server/routes/mcp.ts +225 -0
- package/src/server/routes/permission.ts +69 -0
- package/src/server/routes/project.ts +118 -0
- package/src/server/routes/provider.ts +171 -0
- package/src/server/routes/pty.ts +210 -0
- package/src/server/routes/question.ts +99 -0
- package/src/server/routes/session.ts +984 -0
- package/src/server/routes/tui.ts +378 -0
- package/src/server/routes/workspace.ts +94 -0
- package/src/server/server.ts +353 -0
- package/src/session/compaction.ts +86 -0
- package/src/session/index.ts +904 -0
- package/src/session/instruction.ts +261 -0
- package/src/session/llm/monitor.ts +87 -0
- package/src/session/llm.ts +1676 -0
- package/src/session/message-v2.ts +1082 -0
- package/src/session/message.ts +191 -0
- package/src/session/overflow.ts +34 -0
- package/src/session/processor.ts +635 -0
- package/src/session/projectors.ts +136 -0
- package/src/session/prompt/build-switch.txt +5 -0
- package/src/session/prompt/builder.ts +135 -0
- package/src/session/prompt/default.txt +11 -0
- package/src/session/prompt/engine.ts +1072 -0
- package/src/session/prompt/gemma4.txt +1 -0
- package/src/session/prompt/max-steps.txt +16 -0
- package/src/session/prompt/orchestrator.ts +426 -0
- package/src/session/prompt/plan.txt +28 -0
- package/src/session/prompt/qwen.txt +19 -0
- package/src/session/prompt/resolver.ts +670 -0
- package/src/session/prompt/router.ts +197 -0
- package/src/session/prompt/state.ts +96 -0
- package/src/session/prompt/types.ts +115 -0
- package/src/session/prompt/utils.ts +15 -0
- package/src/session/prompt.ts +362 -0
- package/src/session/retry.ts +106 -0
- package/src/session/revert.ts +176 -0
- package/src/session/sanitizer.ts +125 -0
- package/src/session/schema.ts +38 -0
- package/src/session/session.sql.ts +106 -0
- package/src/session/status.ts +102 -0
- package/src/session/summary.ts +183 -0
- package/src/session/system.ts +79 -0
- package/src/session/todo.ts +166 -0
- package/src/session/worker.ts +382 -0
- package/src/shell/shell.ts +110 -0
- package/src/skill/discovery.ts +116 -0
- package/src/skill/index.ts +287 -0
- package/src/snapshot/index.ts +726 -0
- package/src/sql.d.ts +4 -0
- package/src/storage/db.bun.ts +8 -0
- package/src/storage/db.node.ts +8 -0
- package/src/storage/db.ts +174 -0
- package/src/storage/json-migration.ts +387 -0
- package/src/storage/schema.sql.ts +10 -0
- package/src/storage/schema.ts +4 -0
- package/src/storage/storage.ts +353 -0
- package/src/sync/README.md +179 -0
- package/src/sync/event.sql.ts +16 -0
- package/src/sync/index.ts +263 -0
- package/src/sync/schema.ts +14 -0
- package/src/tool/apply_patch.ts +281 -0
- package/src/tool/apply_patch.txt +1 -0
- package/src/tool/arbitration.txt +5 -0
- package/src/tool/bash.ts +494 -0
- package/src/tool/bash.txt +2 -0
- package/src/tool/batch.ts +183 -0
- package/src/tool/batch.txt +1 -0
- package/src/tool/codesearch.ts +132 -0
- package/src/tool/codesearch.txt +1 -0
- package/src/tool/edit.ts +734 -0
- package/src/tool/edit.txt +1 -0
- package/src/tool/external-directory.ts +46 -0
- package/src/tool/glob.ts +73 -0
- package/src/tool/glob.txt +2 -0
- package/src/tool/grep.ts +156 -0
- package/src/tool/grep.txt +2 -0
- package/src/tool/invalid.ts +20 -0
- package/src/tool/ls.ts +121 -0
- package/src/tool/ls.txt +1 -0
- package/src/tool/lsp.ts +97 -0
- package/src/tool/lsp.txt +1 -0
- package/src/tool/multiedit.ts +46 -0
- package/src/tool/multiedit.txt +1 -0
- package/src/tool/plan-enter.txt +14 -0
- package/src/tool/plan-exit.txt +13 -0
- package/src/tool/plan.ts +131 -0
- package/src/tool/question.ts +46 -0
- package/src/tool/question.txt +10 -0
- package/src/tool/read.ts +332 -0
- package/src/tool/read.txt +1 -0
- package/src/tool/registry.ts +288 -0
- package/src/tool/revert.ts +37 -0
- package/src/tool/schema.ts +17 -0
- package/src/tool/skill.ts +105 -0
- package/src/tool/task.ts +150 -0
- package/src/tool/task.txt +3 -0
- package/src/tool/task_complete.ts +21 -0
- package/src/tool/tool.ts +112 -0
- package/src/tool/truncate.ts +144 -0
- package/src/tool/truncation-dir.ts +4 -0
- package/src/tool/webfetch.ts +206 -0
- package/src/tool/webfetch.txt +1 -0
- package/src/tool/websearch.ts +150 -0
- package/src/tool/websearch.txt +1 -0
- package/src/tool/write.ts +101 -0
- package/src/tool/write.txt +1 -0
- package/src/util/abort.ts +35 -0
- package/src/util/ai-sdk.ts +59 -0
- package/src/util/archive.ts +17 -0
- package/src/util/color.ts +19 -0
- package/src/util/context.ts +25 -0
- package/src/util/data-url.ts +9 -0
- package/src/util/defer.ts +12 -0
- package/src/util/effect-http-client.ts +11 -0
- package/src/util/effect-zod.ts +98 -0
- package/src/util/error.ts +77 -0
- package/src/util/filesystem.ts +245 -0
- package/src/util/flock.ts +333 -0
- package/src/util/fn.ts +21 -0
- package/src/util/format.ts +20 -0
- package/src/util/glob.ts +34 -0
- package/src/util/hash.ts +7 -0
- package/src/util/iife.ts +3 -0
- package/src/util/keybind.ts +103 -0
- package/src/util/lazy.ts +23 -0
- package/src/util/locale.ts +81 -0
- package/src/util/lock.ts +98 -0
- package/src/util/log-parser.ts +114 -0
- package/src/util/log.ts +250 -0
- package/src/util/network.ts +23 -0
- package/src/util/process.ts +176 -0
- package/src/util/queue.ts +32 -0
- package/src/util/record.ts +3 -0
- package/src/util/rpc.ts +66 -0
- package/src/util/schema.ts +53 -0
- package/src/util/scrap.ts +10 -0
- package/src/util/session-analyzer.ts +331 -0
- package/src/util/session-telemetry.ts +91 -0
- package/src/util/signal.ts +12 -0
- package/src/util/timeout.ts +14 -0
- package/src/util/token.ts +7 -0
- package/src/util/tokenizer.ts +50 -0
- package/src/util/toon.ts +45 -0
- package/src/util/update-schema.ts +13 -0
- package/src/util/which.ts +14 -0
- package/src/util/wildcard.ts +59 -0
- package/src/worktree/index.ts +612 -0
- package/sst-env.d.ts +10 -0
- package/test/AGENTS.md +81 -0
- package/test/account/repo.test.ts +326 -0
- package/test/account/service.test.ts +393 -0
- package/test/acp/agent-interface.test.ts +51 -0
- package/test/acp/event-subscription.test.ts +685 -0
- package/test/agent/agent.test.ts +716 -0
- package/test/auth/auth.test.ts +58 -0
- package/test/bus/bus-effect.test.ts +164 -0
- package/test/bus/bus-integration.test.ts +87 -0
- package/test/bus/bus.test.ts +219 -0
- package/test/cli/account.test.ts +26 -0
- package/test/cli/cmd/tui/prompt-part.test.ts +47 -0
- package/test/cli/github-action.test.ts +198 -0
- package/test/cli/github-remote.test.ts +80 -0
- package/test/cli/plugin-auth-picker.test.ts +120 -0
- package/test/cli/tui/keybind-plugin.test.ts +90 -0
- package/test/cli/tui/plugin-add.test.ts +107 -0
- package/test/cli/tui/plugin-install.test.ts +89 -0
- package/test/cli/tui/plugin-lifecycle.test.ts +225 -0
- package/test/cli/tui/plugin-loader-entrypoint.test.ts +492 -0
- package/test/cli/tui/plugin-loader-pure.test.ts +72 -0
- package/test/cli/tui/plugin-loader.test.ts +752 -0
- package/test/cli/tui/plugin-toggle.test.ts +159 -0
- package/test/cli/tui/slot-replace.test.tsx +47 -0
- package/test/cli/tui/theme-store.test.ts +51 -0
- package/test/cli/tui/thread.test.ts +128 -0
- package/test/cli/tui/transcript.test.ts +426 -0
- package/test/config/agent-color.test.ts +71 -0
- package/test/config/config.test.ts +2337 -0
- package/test/config/fixtures/empty-frontmatter.md +4 -0
- package/test/config/fixtures/frontmatter.md +28 -0
- package/test/config/fixtures/markdown-header.md +11 -0
- package/test/config/fixtures/no-frontmatter.md +1 -0
- package/test/config/fixtures/weird-model-id.md +13 -0
- package/test/config/markdown.test.ts +228 -0
- package/test/config/tui.test.ts +800 -0
- package/test/control-plane/sse.test.ts +56 -0
- package/test/effect/cross-spawn-spawner.test.ts +412 -0
- package/test/effect/instance-state.test.ts +482 -0
- package/test/effect/run-service.test.ts +46 -0
- package/test/effect/runner.test.ts +523 -0
- package/test/fake/provider.ts +82 -0
- package/test/file/fsmonitor.test.ts +62 -0
- package/test/file/ignore.test.ts +10 -0
- package/test/file/index.test.ts +946 -0
- package/test/file/path-traversal.test.ts +198 -0
- package/test/file/ripgrep.test.ts +54 -0
- package/test/file/time.test.ts +445 -0
- package/test/file/watcher.test.ts +247 -0
- package/test/filesystem/filesystem.test.ts +319 -0
- package/test/fixture/db.ts +11 -0
- package/test/fixture/fixture.test.ts +26 -0
- package/test/fixture/fixture.ts +172 -0
- package/test/fixture/flock-worker.ts +72 -0
- package/test/fixture/lsp/fake-lsp-server.js +77 -0
- package/test/fixture/plug-worker.ts +93 -0
- package/test/fixture/plugin-meta-worker.ts +26 -0
- package/test/fixture/skills/agents-sdk/SKILL.md +152 -0
- package/test/fixture/skills/agents-sdk/references/callable.md +92 -0
- package/test/fixture/skills/cloudflare/SKILL.md +211 -0
- package/test/fixture/skills/index.json +6 -0
- package/test/fixture/tui-plugin.ts +328 -0
- package/test/fixture/tui-runtime.ts +27 -0
- package/test/format/format.test.ts +171 -0
- package/test/git/git.test.ts +128 -0
- package/test/ide/ide.test.ts +82 -0
- package/test/installation/installation.test.ts +152 -0
- package/test/keybind.test.ts +421 -0
- package/test/lib/effect.ts +53 -0
- package/test/lib/filesystem.ts +10 -0
- package/test/lib/llm-server.ts +794 -0
- package/test/lsp/client.test.ts +95 -0
- package/test/lsp/index.test.ts +133 -0
- package/test/lsp/launch.test.ts +22 -0
- package/test/lsp/lifecycle.test.ts +147 -0
- package/test/mcp/headers.test.ts +153 -0
- package/test/mcp/lifecycle.test.ts +750 -0
- package/test/mcp/oauth-auto-connect.test.ts +199 -0
- package/test/mcp/oauth-browser.test.ts +249 -0
- package/test/mcp/sc-approve-validator.test.ts +431 -0
- package/test/memory/abort-leak.test.ts +137 -0
- package/test/npm.test.ts +18 -0
- package/test/patch/patch.test.ts +348 -0
- package/test/permission/arity.test.ts +33 -0
- package/test/permission/next.test.ts +1123 -0
- package/test/permission-task.test.ts +323 -0
- package/test/plugin/auth-override.test.ts +74 -0
- package/test/plugin/codex.test.ts +123 -0
- package/test/plugin/github-copilot-models.test.ts +117 -0
- package/test/plugin/install-concurrency.test.ts +140 -0
- package/test/plugin/install.test.ts +570 -0
- package/test/plugin/loader-shared.test.ts +1136 -0
- package/test/plugin/meta.test.ts +137 -0
- package/test/plugin/shared.test.ts +88 -0
- package/test/plugin/trigger.test.ts +111 -0
- package/test/preload.ts +90 -0
- package/test/project/migrate-global.test.ts +140 -0
- package/test/project/project.test.ts +459 -0
- package/test/project/state.test.ts +115 -0
- package/test/project/vcs.test.ts +228 -0
- package/test/project/worktree-remove.test.ts +96 -0
- package/test/project/worktree.test.ts +173 -0
- package/test/provider/amazon-bedrock.test.ts +447 -0
- package/test/provider/copilot/convert-to-copilot-messages.test.ts +523 -0
- package/test/provider/copilot/copilot-chat-model.test.ts +592 -0
- package/test/provider/error.test.ts +49 -0
- package/test/provider/gitlab-duo.test.ts +412 -0
- package/test/provider/provider.test.ts +2494 -0
- package/test/provider/transform.test.ts +2944 -0
- package/test/pty/pty-output-isolation.test.ts +141 -0
- package/test/pty/pty-session.test.ts +92 -0
- package/test/pty/pty-shell.test.ts +59 -0
- package/test/question/question.test.ts +453 -0
- package/test/server/global-session-list.test.ts +89 -0
- package/test/server/project-init-git.test.ts +121 -0
- package/test/server/session-actions.test.ts +83 -0
- package/test/server/session-list.test.ts +98 -0
- package/test/server/session-messages.test.ts +159 -0
- package/test/server/session-select.test.ts +84 -0
- package/test/session/compaction.test.ts +683 -0
- package/test/session/continuity-handover.test.ts +620 -0
- package/test/session/deterministic-handover.test.ts +328 -0
- package/test/session/doom-protection.test.ts +247 -0
- package/test/session/hard-reset.test.ts +179 -0
- package/test/session/instruction.test.ts +286 -0
- package/test/session/llm/monitor.test.ts +53 -0
- package/test/session/llm-sanitizer.test.ts +90 -0
- package/test/session/llm-zones-e2e.test.ts +61 -0
- package/test/session/llm.test.ts +1308 -0
- package/test/session/mcpx-normalization.test.ts +86 -0
- package/test/session/mcpx-syntax-recovery.test.ts +28 -0
- package/test/session/message-v2.test.ts +957 -0
- package/test/session/messages-pagination.test.ts +885 -0
- package/test/session/processor-effect.test.ts +805 -0
- package/test/session/prompt/builder.test.ts +71 -0
- package/test/session/prompt/engine-loop.test.ts +80 -0
- package/test/session/prompt/orchestrator.test.ts +108 -0
- package/test/session/prompt/resolver.test.ts +211 -0
- package/test/session/prompt/router.test.ts +84 -0
- package/test/session/prompt/state.test.ts +57 -0
- package/test/session/prompt-effect.test.ts +1241 -0
- package/test/session/prompt.test.ts +522 -0
- package/test/session/refactor-system-zones.test.ts +241 -0
- package/test/session/retry.test.ts +232 -0
- package/test/session/revert-compact.test.ts +621 -0
- package/test/session/sanitizer.test.ts +61 -0
- package/test/session/session.test.ts +142 -0
- package/test/session/snapshot-tool-race.test.ts +242 -0
- package/test/session/structured-output-integration.test.ts +233 -0
- package/test/session/structured-output.test.ts +391 -0
- package/test/session/system.test.ts +59 -0
- package/test/session/telemetry.test.ts +35 -0
- package/test/shell/shell.test.ts +73 -0
- package/test/skill/discovery.test.ts +116 -0
- package/test/skill/skill.test.ts +392 -0
- package/test/snapshot/snapshot.test.ts +1404 -0
- package/test/storage/db.test.ts +14 -0
- package/test/storage/json-migration.test.ts +791 -0
- package/test/storage/storage.test.ts +295 -0
- package/test/sync/index.test.ts +191 -0
- package/test/tool/__snapshots__/tool.test.ts.snap +9 -0
- package/test/tool/apply_patch.test.ts +567 -0
- package/test/tool/bash.test.ts +1099 -0
- package/test/tool/edit.test.ts +681 -0
- package/test/tool/external-directory.test.ts +198 -0
- package/test/tool/fixtures/large-image.png +0 -0
- package/test/tool/fixtures/models-api.json +65179 -0
- package/test/tool/grep.test.ts +111 -0
- package/test/tool/question.test.ts +126 -0
- package/test/tool/read.test.ts +468 -0
- package/test/tool/registry.test.ts +126 -0
- package/test/tool/skill.test.ts +167 -0
- package/test/tool/task.test.ts +49 -0
- package/test/tool/tool-define.test.ts +101 -0
- package/test/tool/truncation.test.ts +161 -0
- package/test/tool/webfetch.test.ts +101 -0
- package/test/tool/write.test.ts +354 -0
- package/test/util/data-url.test.ts +14 -0
- package/test/util/effect-zod.test.ts +61 -0
- package/test/util/error.test.ts +38 -0
- package/test/util/filesystem.test.ts +656 -0
- package/test/util/flock.test.ts +383 -0
- package/test/util/format.test.ts +59 -0
- package/test/util/glob.test.ts +164 -0
- package/test/util/iife.test.ts +36 -0
- package/test/util/lazy.test.ts +50 -0
- package/test/util/lock.test.ts +72 -0
- package/test/util/log-parser.test.ts +61 -0
- package/test/util/module.test.ts +59 -0
- package/test/util/process.test.ts +128 -0
- package/test/util/telemetry-integration.test.ts +104 -0
- package/test/util/timeout.test.ts +21 -0
- package/test/util/which.test.ts +100 -0
- package/test/util/wildcard.test.ts +90 -0
- package/test-regex.js +50 -0
- package/tsconfig.json +23 -0
- package/LICENSE +0 -21
- /package/{postinstall.mjs → script/postinstall.mjs} +0 -0
|
@@ -0,0 +1,179 @@
|
|
|
1
|
+
import { afterEach, describe, expect, mock, test } from "bun:test"
|
|
2
|
+
import { Effect, Layer, ServiceMap } from "effect"
|
|
3
|
+
import * as Stream from "effect/Stream"
|
|
4
|
+
import path from "path"
|
|
5
|
+
import fs from "fs/promises"
|
|
6
|
+
|
|
7
|
+
process.env.EPOCHCLI_DB = ":memory:"
|
|
8
|
+
process.env.EPOCHCLI_DISABLE_CHANNEL_DB = "true"
|
|
9
|
+
|
|
10
|
+
import { Bus } from "../../src/bus"
|
|
11
|
+
import { Config } from "../../src/config/config"
|
|
12
|
+
import { Agent } from "../../src/agent/agent"
|
|
13
|
+
import { LLM } from "../../src/session/llm"
|
|
14
|
+
import { SessionCompaction } from "../../src/session/compaction"
|
|
15
|
+
import { Log } from "../../src/util/log"
|
|
16
|
+
import { Permission } from "../../src/permission"
|
|
17
|
+
import { Plugin } from "../../src/plugin"
|
|
18
|
+
import { provideTmpdirInstance } from "../fixture/fixture"
|
|
19
|
+
import { Session } from "../../src/session"
|
|
20
|
+
import { MessageID, PartID } from "../../src/session/schema"
|
|
21
|
+
import { SessionStatus } from "../../src/session/status"
|
|
22
|
+
import { ModelID, ProviderID } from "../../src/provider/schema"
|
|
23
|
+
import { MessageV2 } from "../../src/session/message-v2"
|
|
24
|
+
import { SessionEngine } from "../../src/session/prompt/engine"
|
|
25
|
+
import { InputResolver } from "../../src/session/prompt/resolver"
|
|
26
|
+
import { ToolOrchestrator } from "../../src/session/prompt/orchestrator"
|
|
27
|
+
import { SessionProcessor } from "../../src/session/processor"
|
|
28
|
+
import { Instruction } from "../../src/session/instruction"
|
|
29
|
+
import { Snapshot } from "../../src/snapshot"
|
|
30
|
+
import { Todo } from "../../src/session/todo"
|
|
31
|
+
import * as CrossSpawnSpawner from "../../src/effect/cross-spawn-spawner"
|
|
32
|
+
import { MCP } from "../../src/mcp"
|
|
33
|
+
import { ToolRegistry } from "../../src/tool/registry"
|
|
34
|
+
import { Truncate } from "../../src/tool/truncate"
|
|
35
|
+
import { AppFileSystem } from "../../src/filesystem"
|
|
36
|
+
import { Git } from "../../src/git"
|
|
37
|
+
import { testEffect } from "../lib/effect"
|
|
38
|
+
import { NodeFileSystem, NodePath } from "@effect/platform-node"
|
|
39
|
+
import { LSP } from "../../src/lsp"
|
|
40
|
+
import { FileTime } from "../../src/file/time"
|
|
41
|
+
import { InstanceState } from "../../src/effect/instance-state"
|
|
42
|
+
|
|
43
|
+
Log.init({ print: false })
|
|
44
|
+
|
|
45
|
+
function createModel(opts: { context: number; output: number; input?: number }): any {
|
|
46
|
+
return {
|
|
47
|
+
id: "test-model",
|
|
48
|
+
providerID: "test",
|
|
49
|
+
name: "Test",
|
|
50
|
+
limit: { context: opts.context, input: opts.input, output: opts.output },
|
|
51
|
+
cost: { input: 0, output: 0, cache: { read: 0, write: 0 } },
|
|
52
|
+
capabilities: {
|
|
53
|
+
toolcall: true, attachment: false, reasoning: false, temperature: true, interleaved: false,
|
|
54
|
+
input: { text: true, image: false, audio: false, video: false, pdf: false },
|
|
55
|
+
output: { text: true, image: false, audio: false, video: false, pdf: false },
|
|
56
|
+
},
|
|
57
|
+
api: { id: "test-model", url: "https://example.com", npm: "@ai-sdk/openai" },
|
|
58
|
+
options: {},
|
|
59
|
+
}
|
|
60
|
+
}
|
|
61
|
+
|
|
62
|
+
class ProviderService extends ServiceMap.Service<ProviderService, any>()("@epochcli/Provider") {}
|
|
63
|
+
|
|
64
|
+
mock.module("../../src/provider/provider", () => ({
|
|
65
|
+
Provider: {
|
|
66
|
+
getSideModel: async () => ({ providerID: "test", modelID: "test-model" }),
|
|
67
|
+
getLanguage: async () => ({
|
|
68
|
+
specificationVersion: "v3",
|
|
69
|
+
modelId: "test-side-model",
|
|
70
|
+
doGenerate: async () => ({
|
|
71
|
+
text: "NONE",
|
|
72
|
+
content: [{ type: "text", text: "NONE" }],
|
|
73
|
+
finishReason: "stop",
|
|
74
|
+
usage: { promptTokens: 10, completionTokens: 10, inputTokens: 10, outputTokens: 10 },
|
|
75
|
+
}),
|
|
76
|
+
}),
|
|
77
|
+
getModel: async () => createModel({ context: 16000, output: 4000 }),
|
|
78
|
+
Service: ProviderService,
|
|
79
|
+
},
|
|
80
|
+
}))
|
|
81
|
+
|
|
82
|
+
function llm() {
|
|
83
|
+
const queue: any[] = []
|
|
84
|
+
return {
|
|
85
|
+
clear() { queue.length = 0 },
|
|
86
|
+
push(stream: any) { queue.push(stream) },
|
|
87
|
+
layer: Layer.succeed(
|
|
88
|
+
LLM.Service,
|
|
89
|
+
LLM.Service.of({
|
|
90
|
+
stream: (input: any) => {
|
|
91
|
+
const item = queue.shift()
|
|
92
|
+
if (!item) return Stream.fail(new Error("Queue empty - stop test"))
|
|
93
|
+
return item.pipe(Stream.mapEffect((event: any) => Effect.succeed(event)))
|
|
94
|
+
},
|
|
95
|
+
} as any),
|
|
96
|
+
),
|
|
97
|
+
}
|
|
98
|
+
}
|
|
99
|
+
|
|
100
|
+
function liveLayer(llmLayer: Layer.Layer<LLM.Service>) {
|
|
101
|
+
const bus = Bus.layer
|
|
102
|
+
const status = SessionStatus.layer.pipe(Layer.provide(bus))
|
|
103
|
+
const providers = Layer.mergeAll(
|
|
104
|
+
status, bus, Config.defaultLayer, Session.defaultLayer, Agent.defaultLayer, Plugin.defaultLayer,
|
|
105
|
+
Snapshot.defaultLayer, Instruction.defaultLayer, Permission.defaultLayer, CrossSpawnSpawner.defaultLayer,
|
|
106
|
+
AppFileSystem.defaultLayer, Git.defaultLayer, Todo.defaultLayer,
|
|
107
|
+
Layer.succeed(MCP.Service, { clients: () => Effect.succeed({}), tools: () => Effect.succeed({}) } as any),
|
|
108
|
+
Layer.succeed(ToolRegistry.Service, { resolve: () => Effect.succeed({}), list: () => Effect.succeed([]), get: () => Effect.succeed(undefined), tools: () => Effect.succeed([]) } as any),
|
|
109
|
+
Layer.succeed(Truncate.Service, { output: (text: string) => Effect.succeed({ text, truncated: false }) } as any),
|
|
110
|
+
Layer.succeed(LSP.Service, {} as any),
|
|
111
|
+
Layer.succeed(FileTime.Service, { get: () => Effect.succeed(0) } as any),
|
|
112
|
+
Layer.succeed(ProviderService, { getSideModel: () => Effect.succeed({ providerID: "test", modelID: "test-model" }), getLanguage: () => Effect.succeed({}), getModel: () => Effect.succeed(createModel({ context: 16000, output: 4000 })) } as any),
|
|
113
|
+
llmLayer, NodeFileSystem.layer, NodePath.layer,
|
|
114
|
+
)
|
|
115
|
+
let l: Layer.Layer<any, any, any> = providers
|
|
116
|
+
l = Layer.provideMerge(InputResolver.layer, l)
|
|
117
|
+
l = Layer.provideMerge(ToolOrchestrator.layer, l)
|
|
118
|
+
l = Layer.provideMerge(SessionProcessor.layer, l)
|
|
119
|
+
l = Layer.provideMerge(SessionCompaction.layer, l)
|
|
120
|
+
l = Layer.provideMerge(SessionEngine.layer, l)
|
|
121
|
+
return l
|
|
122
|
+
}
|
|
123
|
+
|
|
124
|
+
describe("Hard Reset Doom Loop Protection", () => {
|
|
125
|
+
const l = llm()
|
|
126
|
+
const layer = liveLayer(l.layer)
|
|
127
|
+
const { live: it } = testEffect(layer as any)
|
|
128
|
+
afterEach(() => { l.clear() })
|
|
129
|
+
|
|
130
|
+
it("triggers a Hard Reset (transition) when stall score hits 20", () =>
|
|
131
|
+
provideTmpdirInstance((tmpPath) =>
|
|
132
|
+
Effect.gen(function* () {
|
|
133
|
+
const session = yield* Session.Service
|
|
134
|
+
const engine = yield* SessionEngine.Service
|
|
135
|
+
|
|
136
|
+
const chat = yield* session.create({})
|
|
137
|
+
const parentId = MessageID.ascending()
|
|
138
|
+
yield* session.updateMessage({
|
|
139
|
+
id: parentId,
|
|
140
|
+
sessionID: chat.id,
|
|
141
|
+
role: "user",
|
|
142
|
+
time: { created: Date.now() },
|
|
143
|
+
agent: "build",
|
|
144
|
+
model: { providerID: ProviderID.make("test"), modelID: ModelID.make("test-model") },
|
|
145
|
+
})
|
|
146
|
+
|
|
147
|
+
// Create continuity file to prevent ENOENT during transition
|
|
148
|
+
const fsNode = yield* Effect.promise(() => import("fs/promises"))
|
|
149
|
+
yield* Effect.promise(() => fsNode.writeFile(path.join(tmpPath, ".epoch-continuity.toon"), "test"))
|
|
150
|
+
|
|
151
|
+
// Simulate 3 loops of empty text in YOLO mode (10 points each, meaning 2 loops hits 20)
|
|
152
|
+
for (let i = 0; i < 3; i++) {
|
|
153
|
+
l.push(Stream.make(
|
|
154
|
+
{ type: "text-start" } as any,
|
|
155
|
+
{ type: "text-delta", text: "I am thinking..." } as any,
|
|
156
|
+
{ type: "text-end" } as any,
|
|
157
|
+
{ type: "finish-step", finishReason: "stop", usage: { totalTokens: 10 } } as any,
|
|
158
|
+
).pipe(Stream.tap(() => Effect.sync(() => console.log("Stream", i, "consumed")))))
|
|
159
|
+
}
|
|
160
|
+
|
|
161
|
+
// Push a task_complete to gracefully exit the YOLO loop after the transition
|
|
162
|
+
l.push(Stream.make(
|
|
163
|
+
{ type: "tool-call", toolCallId: "1", toolName: "task_complete", args: {} } as any,
|
|
164
|
+
{ type: "finish-step", finishReason: "stop", usage: { totalTokens: 10 } } as any,
|
|
165
|
+
).pipe(Stream.tap(() => Effect.sync(() => console.log("Task Complete Stream consumed")))))
|
|
166
|
+
|
|
167
|
+
console.log("Starting runLoop")
|
|
168
|
+
const result = yield* engine.runLoop(chat.id, true) // yolo = true
|
|
169
|
+
console.log("runLoop finished")
|
|
170
|
+
|
|
171
|
+
// Read all messages to verify transition occurred
|
|
172
|
+
const msgs = yield* MessageV2.filterByEpochEffect(chat.id)
|
|
173
|
+
const transitionParts = msgs.flatMap(m => m.parts).filter(p => p.type === "transition")
|
|
174
|
+
|
|
175
|
+
expect(transitionParts.length).toBeGreaterThan(0)
|
|
176
|
+
}),
|
|
177
|
+
),
|
|
178
|
+
)
|
|
179
|
+
})
|
|
@@ -0,0 +1,286 @@
|
|
|
1
|
+
import { afterEach, beforeEach, describe, expect, test } from "bun:test"
|
|
2
|
+
import path from "path"
|
|
3
|
+
import { ModelID, ProviderID } from "../../src/provider/schema"
|
|
4
|
+
import { Instruction } from "../../src/session/instruction"
|
|
5
|
+
import type { MessageV2 } from "../../src/session/message-v2"
|
|
6
|
+
import { Instance } from "../../src/project/instance"
|
|
7
|
+
import { MessageID, PartID, SessionID } from "../../src/session/schema"
|
|
8
|
+
import { Global } from "../../src/global"
|
|
9
|
+
import { tmpdir } from "../fixture/fixture"
|
|
10
|
+
|
|
11
|
+
function loaded(filepath: string): MessageV2.WithParts[] {
|
|
12
|
+
const sessionID = SessionID.make("session-loaded-1")
|
|
13
|
+
const messageID = MessageID.make("message-loaded-1")
|
|
14
|
+
|
|
15
|
+
return [
|
|
16
|
+
{
|
|
17
|
+
info: {
|
|
18
|
+
id: messageID,
|
|
19
|
+
sessionID,
|
|
20
|
+
role: "user",
|
|
21
|
+
time: { created: 0 },
|
|
22
|
+
agent: "build",
|
|
23
|
+
model: {
|
|
24
|
+
providerID: ProviderID.make("anthropic"),
|
|
25
|
+
modelID: ModelID.make("claude-sonnet-4-20250514"),
|
|
26
|
+
},
|
|
27
|
+
},
|
|
28
|
+
parts: [
|
|
29
|
+
{
|
|
30
|
+
id: PartID.make("part-loaded-1"),
|
|
31
|
+
messageID,
|
|
32
|
+
sessionID,
|
|
33
|
+
type: "tool",
|
|
34
|
+
callID: "call-loaded-1",
|
|
35
|
+
tool: "read",
|
|
36
|
+
state: {
|
|
37
|
+
status: "completed",
|
|
38
|
+
input: {},
|
|
39
|
+
output: "done",
|
|
40
|
+
title: "Read",
|
|
41
|
+
metadata: { loaded: [filepath] },
|
|
42
|
+
time: { start: 0, end: 1 },
|
|
43
|
+
},
|
|
44
|
+
},
|
|
45
|
+
],
|
|
46
|
+
},
|
|
47
|
+
]
|
|
48
|
+
}
|
|
49
|
+
|
|
50
|
+
describe("Instruction.resolve", () => {
|
|
51
|
+
test("returns empty when AGENTS.md is at project root (already in systemPaths)", async () => {
|
|
52
|
+
await using tmp = await tmpdir({
|
|
53
|
+
init: async (dir) => {
|
|
54
|
+
await Bun.write(path.join(dir, "AGENTS.md"), "# Root Instructions")
|
|
55
|
+
await Bun.write(path.join(dir, "src", "file.ts"), "const x = 1")
|
|
56
|
+
},
|
|
57
|
+
})
|
|
58
|
+
await Instance.provide({
|
|
59
|
+
directory: tmp.path,
|
|
60
|
+
fn: async () => {
|
|
61
|
+
const system = await Instruction.systemPaths()
|
|
62
|
+
expect(system.has(path.join(tmp.path, "AGENTS.md"))).toBe(true)
|
|
63
|
+
|
|
64
|
+
const results = await Instruction.resolve(
|
|
65
|
+
[],
|
|
66
|
+
path.join(tmp.path, "src", "file.ts"),
|
|
67
|
+
MessageID.make("message-test-1"),
|
|
68
|
+
)
|
|
69
|
+
expect(results).toEqual([])
|
|
70
|
+
},
|
|
71
|
+
})
|
|
72
|
+
})
|
|
73
|
+
|
|
74
|
+
test("returns AGENTS.md from subdirectory (not in systemPaths)", async () => {
|
|
75
|
+
await using tmp = await tmpdir({
|
|
76
|
+
init: async (dir) => {
|
|
77
|
+
await Bun.write(path.join(dir, "subdir", "AGENTS.md"), "# Subdir Instructions")
|
|
78
|
+
await Bun.write(path.join(dir, "subdir", "nested", "file.ts"), "const x = 1")
|
|
79
|
+
},
|
|
80
|
+
})
|
|
81
|
+
await Instance.provide({
|
|
82
|
+
directory: tmp.path,
|
|
83
|
+
fn: async () => {
|
|
84
|
+
const system = await Instruction.systemPaths()
|
|
85
|
+
expect(system.has(path.join(tmp.path, "subdir", "AGENTS.md"))).toBe(false)
|
|
86
|
+
|
|
87
|
+
const results = await Instruction.resolve(
|
|
88
|
+
[],
|
|
89
|
+
path.join(tmp.path, "subdir", "nested", "file.ts"),
|
|
90
|
+
MessageID.make("message-test-2"),
|
|
91
|
+
)
|
|
92
|
+
expect(results.length).toBe(1)
|
|
93
|
+
expect(results[0].filepath).toBe(path.join(tmp.path, "subdir", "AGENTS.md"))
|
|
94
|
+
},
|
|
95
|
+
})
|
|
96
|
+
})
|
|
97
|
+
|
|
98
|
+
test("doesn't reload AGENTS.md when reading it directly", async () => {
|
|
99
|
+
await using tmp = await tmpdir({
|
|
100
|
+
init: async (dir) => {
|
|
101
|
+
await Bun.write(path.join(dir, "subdir", "AGENTS.md"), "# Subdir Instructions")
|
|
102
|
+
await Bun.write(path.join(dir, "subdir", "nested", "file.ts"), "const x = 1")
|
|
103
|
+
},
|
|
104
|
+
})
|
|
105
|
+
await Instance.provide({
|
|
106
|
+
directory: tmp.path,
|
|
107
|
+
fn: async () => {
|
|
108
|
+
const filepath = path.join(tmp.path, "subdir", "AGENTS.md")
|
|
109
|
+
const system = await Instruction.systemPaths()
|
|
110
|
+
expect(system.has(filepath)).toBe(false)
|
|
111
|
+
|
|
112
|
+
const results = await Instruction.resolve([], filepath, MessageID.make("message-test-3"))
|
|
113
|
+
expect(results).toEqual([])
|
|
114
|
+
},
|
|
115
|
+
})
|
|
116
|
+
})
|
|
117
|
+
|
|
118
|
+
test("does not reattach the same nearby instructions twice for one message", async () => {
|
|
119
|
+
await using tmp = await tmpdir({
|
|
120
|
+
init: async (dir) => {
|
|
121
|
+
await Bun.write(path.join(dir, "subdir", "AGENTS.md"), "# Subdir Instructions")
|
|
122
|
+
await Bun.write(path.join(dir, "subdir", "nested", "file.ts"), "const x = 1")
|
|
123
|
+
},
|
|
124
|
+
})
|
|
125
|
+
await Instance.provide({
|
|
126
|
+
directory: tmp.path,
|
|
127
|
+
fn: async () => {
|
|
128
|
+
const filepath = path.join(tmp.path, "subdir", "nested", "file.ts")
|
|
129
|
+
const id = MessageID.make("message-claim-1")
|
|
130
|
+
|
|
131
|
+
const first = await Instruction.resolve([], filepath, id)
|
|
132
|
+
const second = await Instruction.resolve([], filepath, id)
|
|
133
|
+
|
|
134
|
+
expect(first).toHaveLength(1)
|
|
135
|
+
expect(first[0].filepath).toBe(path.join(tmp.path, "subdir", "AGENTS.md"))
|
|
136
|
+
expect(second).toEqual([])
|
|
137
|
+
},
|
|
138
|
+
})
|
|
139
|
+
})
|
|
140
|
+
|
|
141
|
+
test("clear allows nearby instructions to be attached again for the same message", async () => {
|
|
142
|
+
await using tmp = await tmpdir({
|
|
143
|
+
init: async (dir) => {
|
|
144
|
+
await Bun.write(path.join(dir, "subdir", "AGENTS.md"), "# Subdir Instructions")
|
|
145
|
+
await Bun.write(path.join(dir, "subdir", "nested", "file.ts"), "const x = 1")
|
|
146
|
+
},
|
|
147
|
+
})
|
|
148
|
+
await Instance.provide({
|
|
149
|
+
directory: tmp.path,
|
|
150
|
+
fn: async () => {
|
|
151
|
+
const filepath = path.join(tmp.path, "subdir", "nested", "file.ts")
|
|
152
|
+
const id = MessageID.make("message-claim-2")
|
|
153
|
+
|
|
154
|
+
const first = await Instruction.resolve([], filepath, id)
|
|
155
|
+
await Instruction.clear(id)
|
|
156
|
+
const second = await Instruction.resolve([], filepath, id)
|
|
157
|
+
|
|
158
|
+
expect(first).toHaveLength(1)
|
|
159
|
+
expect(second).toHaveLength(1)
|
|
160
|
+
expect(second[0].filepath).toBe(path.join(tmp.path, "subdir", "AGENTS.md"))
|
|
161
|
+
},
|
|
162
|
+
})
|
|
163
|
+
})
|
|
164
|
+
|
|
165
|
+
test("skips instructions already reported by prior read metadata", async () => {
|
|
166
|
+
await using tmp = await tmpdir({
|
|
167
|
+
init: async (dir) => {
|
|
168
|
+
await Bun.write(path.join(dir, "subdir", "AGENTS.md"), "# Subdir Instructions")
|
|
169
|
+
await Bun.write(path.join(dir, "subdir", "nested", "file.ts"), "const x = 1")
|
|
170
|
+
},
|
|
171
|
+
})
|
|
172
|
+
await Instance.provide({
|
|
173
|
+
directory: tmp.path,
|
|
174
|
+
fn: async () => {
|
|
175
|
+
const agents = path.join(tmp.path, "subdir", "AGENTS.md")
|
|
176
|
+
const filepath = path.join(tmp.path, "subdir", "nested", "file.ts")
|
|
177
|
+
const id = MessageID.make("message-claim-3")
|
|
178
|
+
|
|
179
|
+
const results = await Instruction.resolve(loaded(agents), filepath, id)
|
|
180
|
+
|
|
181
|
+
expect(results).toEqual([])
|
|
182
|
+
},
|
|
183
|
+
})
|
|
184
|
+
})
|
|
185
|
+
|
|
186
|
+
test.todo("fetches remote instructions from config URLs via HttpClient", () => {})
|
|
187
|
+
})
|
|
188
|
+
|
|
189
|
+
describe("Instruction.systemPaths EPOCHCLI_CONFIG_DIR", () => {
|
|
190
|
+
let originalConfigDir: string | undefined
|
|
191
|
+
|
|
192
|
+
beforeEach(() => {
|
|
193
|
+
originalConfigDir = process.env["EPOCHCLI_CONFIG_DIR"]
|
|
194
|
+
})
|
|
195
|
+
|
|
196
|
+
afterEach(() => {
|
|
197
|
+
if (originalConfigDir === undefined) {
|
|
198
|
+
delete process.env["EPOCHCLI_CONFIG_DIR"]
|
|
199
|
+
} else {
|
|
200
|
+
process.env["EPOCHCLI_CONFIG_DIR"] = originalConfigDir
|
|
201
|
+
}
|
|
202
|
+
})
|
|
203
|
+
|
|
204
|
+
test("prefers EPOCHCLI_CONFIG_DIR AGENTS.md over global when both exist", async () => {
|
|
205
|
+
await using profileTmp = await tmpdir({
|
|
206
|
+
init: async (dir) => {
|
|
207
|
+
await Bun.write(path.join(dir, "AGENTS.md"), "# Profile Instructions")
|
|
208
|
+
},
|
|
209
|
+
})
|
|
210
|
+
await using globalTmp = await tmpdir({
|
|
211
|
+
init: async (dir) => {
|
|
212
|
+
await Bun.write(path.join(dir, "AGENTS.md"), "# Global Instructions")
|
|
213
|
+
},
|
|
214
|
+
})
|
|
215
|
+
await using projectTmp = await tmpdir()
|
|
216
|
+
|
|
217
|
+
process.env["EPOCHCLI_CONFIG_DIR"] = profileTmp.path
|
|
218
|
+
const originalGlobalConfig = Global.Path.config
|
|
219
|
+
;(Global.Path as { config: string }).config = globalTmp.path
|
|
220
|
+
|
|
221
|
+
try {
|
|
222
|
+
await Instance.provide({
|
|
223
|
+
directory: projectTmp.path,
|
|
224
|
+
fn: async () => {
|
|
225
|
+
const paths = await Instruction.systemPaths()
|
|
226
|
+
expect(paths.has(path.join(profileTmp.path, "AGENTS.md"))).toBe(true)
|
|
227
|
+
expect(paths.has(path.join(globalTmp.path, "AGENTS.md"))).toBe(false)
|
|
228
|
+
},
|
|
229
|
+
})
|
|
230
|
+
} finally {
|
|
231
|
+
;(Global.Path as { config: string }).config = originalGlobalConfig
|
|
232
|
+
}
|
|
233
|
+
})
|
|
234
|
+
|
|
235
|
+
test("falls back to global AGENTS.md when EPOCHCLI_CONFIG_DIR has no AGENTS.md", async () => {
|
|
236
|
+
await using profileTmp = await tmpdir()
|
|
237
|
+
await using globalTmp = await tmpdir({
|
|
238
|
+
init: async (dir) => {
|
|
239
|
+
await Bun.write(path.join(dir, "AGENTS.md"), "# Global Instructions")
|
|
240
|
+
},
|
|
241
|
+
})
|
|
242
|
+
await using projectTmp = await tmpdir()
|
|
243
|
+
|
|
244
|
+
process.env["EPOCHCLI_CONFIG_DIR"] = profileTmp.path
|
|
245
|
+
const originalGlobalConfig = Global.Path.config
|
|
246
|
+
;(Global.Path as { config: string }).config = globalTmp.path
|
|
247
|
+
|
|
248
|
+
try {
|
|
249
|
+
await Instance.provide({
|
|
250
|
+
directory: projectTmp.path,
|
|
251
|
+
fn: async () => {
|
|
252
|
+
const paths = await Instruction.systemPaths()
|
|
253
|
+
expect(paths.has(path.join(profileTmp.path, "AGENTS.md"))).toBe(false)
|
|
254
|
+
expect(paths.has(path.join(globalTmp.path, "AGENTS.md"))).toBe(true)
|
|
255
|
+
},
|
|
256
|
+
})
|
|
257
|
+
} finally {
|
|
258
|
+
;(Global.Path as { config: string }).config = originalGlobalConfig
|
|
259
|
+
}
|
|
260
|
+
})
|
|
261
|
+
|
|
262
|
+
test("uses global AGENTS.md when EPOCHCLI_CONFIG_DIR is not set", async () => {
|
|
263
|
+
await using globalTmp = await tmpdir({
|
|
264
|
+
init: async (dir) => {
|
|
265
|
+
await Bun.write(path.join(dir, "AGENTS.md"), "# Global Instructions")
|
|
266
|
+
},
|
|
267
|
+
})
|
|
268
|
+
await using projectTmp = await tmpdir()
|
|
269
|
+
|
|
270
|
+
delete process.env["EPOCHCLI_CONFIG_DIR"]
|
|
271
|
+
const originalGlobalConfig = Global.Path.config
|
|
272
|
+
;(Global.Path as { config: string }).config = globalTmp.path
|
|
273
|
+
|
|
274
|
+
try {
|
|
275
|
+
await Instance.provide({
|
|
276
|
+
directory: projectTmp.path,
|
|
277
|
+
fn: async () => {
|
|
278
|
+
const paths = await Instruction.systemPaths()
|
|
279
|
+
expect(paths.has(path.join(globalTmp.path, "AGENTS.md"))).toBe(true)
|
|
280
|
+
},
|
|
281
|
+
})
|
|
282
|
+
} finally {
|
|
283
|
+
;(Global.Path as { config: string }).config = originalGlobalConfig
|
|
284
|
+
}
|
|
285
|
+
})
|
|
286
|
+
})
|
|
@@ -0,0 +1,53 @@
|
|
|
1
|
+
import { describe, it, expect } from "bun:test"
|
|
2
|
+
import { StreamingMonitor } from "../../../src/session/llm/monitor"
|
|
3
|
+
|
|
4
|
+
describe("StreamingMonitor", () => {
|
|
5
|
+
it("should not detect loops in normal text", () => {
|
|
6
|
+
const monitor = new StreamingMonitor()
|
|
7
|
+
expect(monitor.push("This is a normal sentence.")).toBe(false)
|
|
8
|
+
expect(monitor.push("It does not have any long repeating patterns.")).toBe(false)
|
|
9
|
+
expect(monitor.push("The quick brown fox jumps over the lazy dog.")).toBe(false)
|
|
10
|
+
})
|
|
11
|
+
|
|
12
|
+
it("should detect simple repeating loops", () => {
|
|
13
|
+
const monitor = new StreamingMonitor({ minMatchLength: 5, maxOccurrences: 3 })
|
|
14
|
+
const pattern = "Repeat me! "
|
|
15
|
+
expect(monitor.push(pattern)).toBe(false)
|
|
16
|
+
expect(monitor.push(pattern)).toBe(false)
|
|
17
|
+
expect(monitor.push(pattern)).toBe(true)
|
|
18
|
+
expect(monitor.getOffendingText()).toBe(pattern.trim())
|
|
19
|
+
})
|
|
20
|
+
|
|
21
|
+
it("should respect the minimum match length", () => {
|
|
22
|
+
const monitor = new StreamingMonitor({ minMatchLength: 10, maxOccurrences: 3 })
|
|
23
|
+
const shortPattern = "abc "
|
|
24
|
+
expect(monitor.push(shortPattern)).toBe(false)
|
|
25
|
+
expect(monitor.push(shortPattern)).toBe(false)
|
|
26
|
+
expect(monitor.push(shortPattern)).toBe(false)
|
|
27
|
+
expect(monitor.push(shortPattern)).toBe(false)
|
|
28
|
+
|
|
29
|
+
const longPattern = "This is a long pattern "
|
|
30
|
+
expect(monitor.push(longPattern)).toBe(false)
|
|
31
|
+
expect(monitor.push(longPattern)).toBe(false)
|
|
32
|
+
expect(monitor.push(longPattern)).toBe(true)
|
|
33
|
+
})
|
|
34
|
+
|
|
35
|
+
it("should detect loops even if they are pushed in small chunks", () => {
|
|
36
|
+
const monitor = new StreamingMonitor({ minMatchLength: 10, maxOccurrences: 3 })
|
|
37
|
+
const pattern = "Thinking about the next step... "
|
|
38
|
+
const fullText = pattern + pattern + pattern
|
|
39
|
+
|
|
40
|
+
for (let i = 0; i < fullText.length - 1; i++) {
|
|
41
|
+
expect(monitor.push(fullText[i])).toBe(false)
|
|
42
|
+
}
|
|
43
|
+
expect(monitor.push(fullText[fullText.length - 1])).toBe(true)
|
|
44
|
+
})
|
|
45
|
+
|
|
46
|
+
it("should return the longest repeating suffix as offending text", () => {
|
|
47
|
+
const monitor = new StreamingMonitor({ minMatchLength: 5, maxOccurrences: 2 })
|
|
48
|
+
const pattern = "Longer pattern that repeats"
|
|
49
|
+
monitor.push(pattern)
|
|
50
|
+
monitor.push(pattern)
|
|
51
|
+
expect(monitor.getOffendingText()).toBe(pattern)
|
|
52
|
+
})
|
|
53
|
+
})
|
|
@@ -0,0 +1,90 @@
|
|
|
1
|
+
import { describe, expect, test } from "bun:test"
|
|
2
|
+
import { SanitizerMiddleware } from "../../src/session/sanitizer"
|
|
3
|
+
|
|
4
|
+
describe("LLM Sanitizer Middleware Logic", () => {
|
|
5
|
+
test("strips unhelpful providerOptions from prompt messages and parts", () => {
|
|
6
|
+
const rawMessages = [
|
|
7
|
+
{
|
|
8
|
+
role: "user",
|
|
9
|
+
providerOptions: {
|
|
10
|
+
anthropic: { cacheControl: { type: "ephemeral" } },
|
|
11
|
+
openrouter: { cacheControl: { type: "ephemeral" } },
|
|
12
|
+
bedrock: { cachePoint: { type: "default" } },
|
|
13
|
+
openaiCompatible: { cache_control: { type: "ephemeral" } },
|
|
14
|
+
copilot: { copilot_cache_control: { type: "ephemeral" } },
|
|
15
|
+
},
|
|
16
|
+
content: [
|
|
17
|
+
{
|
|
18
|
+
type: "text",
|
|
19
|
+
text: "Hi",
|
|
20
|
+
providerOptions: {
|
|
21
|
+
anthropic: { cacheControl: { type: "ephemeral" } },
|
|
22
|
+
openrouter: { cacheControl: { type: "ephemeral" } },
|
|
23
|
+
bedrock: { cachePoint: { type: "default" } },
|
|
24
|
+
},
|
|
25
|
+
},
|
|
26
|
+
],
|
|
27
|
+
},
|
|
28
|
+
]
|
|
29
|
+
|
|
30
|
+
const result = SanitizerMiddleware.stripProviderOptions(rawMessages, "anthropic")
|
|
31
|
+
|
|
32
|
+
// Message level
|
|
33
|
+
expect(result[0].providerOptions).toEqual({
|
|
34
|
+
anthropic: { cacheControl: { type: "ephemeral" } },
|
|
35
|
+
})
|
|
36
|
+
|
|
37
|
+
// Part level
|
|
38
|
+
expect(result[0].content[0].providerOptions).toEqual({
|
|
39
|
+
anthropic: { cacheControl: { type: "ephemeral" } },
|
|
40
|
+
})
|
|
41
|
+
})
|
|
42
|
+
|
|
43
|
+
test("handles azure provider by keeping both openai and azure keys", () => {
|
|
44
|
+
const rawMessages = [
|
|
45
|
+
{
|
|
46
|
+
role: "user",
|
|
47
|
+
providerOptions: {
|
|
48
|
+
openai: { cache_control: { type: "ephemeral" } },
|
|
49
|
+
azure: { cache_control: { type: "ephemeral" } },
|
|
50
|
+
anthropic: { cacheControl: { type: "ephemeral" } },
|
|
51
|
+
},
|
|
52
|
+
content: "Hello",
|
|
53
|
+
},
|
|
54
|
+
]
|
|
55
|
+
|
|
56
|
+
const result = SanitizerMiddleware.stripProviderOptions(rawMessages, "azure", true)
|
|
57
|
+
|
|
58
|
+
expect(result[0].providerOptions).toEqual({
|
|
59
|
+
openai: { cache_control: { type: "ephemeral" } },
|
|
60
|
+
azure: { cache_control: { type: "ephemeral" } },
|
|
61
|
+
})
|
|
62
|
+
expect(result[0].providerOptions.anthropic).toBeUndefined()
|
|
63
|
+
})
|
|
64
|
+
|
|
65
|
+
test("removes providerOptions entirely if empty after stripping", () => {
|
|
66
|
+
const rawMessages = [
|
|
67
|
+
{
|
|
68
|
+
role: "user",
|
|
69
|
+
providerOptions: {
|
|
70
|
+
anthropic: { cacheControl: { type: "ephemeral" } },
|
|
71
|
+
},
|
|
72
|
+
content: [
|
|
73
|
+
{
|
|
74
|
+
type: "text",
|
|
75
|
+
text: "Hi",
|
|
76
|
+
providerOptions: {
|
|
77
|
+
anthropic: { cacheControl: { type: "ephemeral" } },
|
|
78
|
+
},
|
|
79
|
+
},
|
|
80
|
+
],
|
|
81
|
+
},
|
|
82
|
+
]
|
|
83
|
+
|
|
84
|
+
// Target a key that is NOT present
|
|
85
|
+
const result = SanitizerMiddleware.stripProviderOptions(rawMessages, "google")
|
|
86
|
+
|
|
87
|
+
expect(result[0].providerOptions).toBeUndefined()
|
|
88
|
+
expect(result[0].content[0].providerOptions).toBeUndefined()
|
|
89
|
+
})
|
|
90
|
+
})
|
|
@@ -0,0 +1,61 @@
|
|
|
1
|
+
import { describe, it, expect } from "bun:test"
|
|
2
|
+
import * as fs from "fs"
|
|
3
|
+
import * as path from "path"
|
|
4
|
+
|
|
5
|
+
describe("LLM Payload Zones E2E Verification", () => {
|
|
6
|
+
it("should have correct contents in Zone 1, Zone 2, and Zone 3", () => {
|
|
7
|
+
// We are verifying against the explicit e2e test output mentioned by the user
|
|
8
|
+
const payloadPath = path.resolve(
|
|
9
|
+
__dirname,
|
|
10
|
+
"../../../../e2e_testing/results/suite_2026-04-13T23-32-46-621Z/system-prompt-hi-run-1/initial_payload.json",
|
|
11
|
+
)
|
|
12
|
+
|
|
13
|
+
// Skip if the file doesn't exist (e.g. in CI where this specific run isn't present)
|
|
14
|
+
if (!fs.existsSync(payloadPath)) {
|
|
15
|
+
console.warn("Skipping e2e payload test because payload file is missing.")
|
|
16
|
+
return
|
|
17
|
+
}
|
|
18
|
+
|
|
19
|
+
const data = JSON.parse(fs.readFileSync(payloadPath, "utf-8"))
|
|
20
|
+
const payload = data.payload
|
|
21
|
+
|
|
22
|
+
// 1. Verify Zone 1 (System Role)
|
|
23
|
+
const systemMsg = payload.find((m: any) => m.role === "system")
|
|
24
|
+
expect(systemMsg).toBeDefined()
|
|
25
|
+
// Verify it contains the Operational Facts and active agent context (Zone 1)
|
|
26
|
+
expect(systemMsg.content).toContain("Current Phase: [PLAN]")
|
|
27
|
+
expect(systemMsg.content).toContain("The environment context limit is strictly 32K tokens.")
|
|
28
|
+
expect(systemMsg.content).toContain("The codebase primary language is JavaScript/TypeScript.")
|
|
29
|
+
|
|
30
|
+
// 2. Verify Zone 2 (Assistant Role)
|
|
31
|
+
const assistantMsg = payload.find((m: any) => m.role === "assistant")
|
|
32
|
+
expect(assistantMsg).toBeDefined()
|
|
33
|
+
// Verify it contains the Behavioral Rules (Zone 2)
|
|
34
|
+
const assistantContent = assistantMsg.content[0].text
|
|
35
|
+
expect(assistantContent).toContain("=== BEHAVIORAL RULES & GENERAL CONTEXT ===")
|
|
36
|
+
|
|
37
|
+
// Verify core_interaction_pack rules ARE present
|
|
38
|
+
expect(assistantContent).toContain("Trigger: Generating code in response to a user prompt.") // comm_01
|
|
39
|
+
expect(assistantContent).toContain("Trigger: User asks a non-coding general knowledge question.") // comm_05
|
|
40
|
+
expect(assistantContent).toContain("Trigger: User implies the AI can execute, compile, or host code.") // self_01
|
|
41
|
+
|
|
42
|
+
// Verify context_mgmt_pack rules are NOT present
|
|
43
|
+
expect(assistantContent).not.toContain("Token context approaches maximum limit during a long session") // mem_02
|
|
44
|
+
expect(assistantContent).not.toContain("Session begins.") // pers_01
|
|
45
|
+
|
|
46
|
+
// Verify new_feature_pack rules are NOT present
|
|
47
|
+
expect(assistantContent).not.toContain("User asks to integrate a new file into the broader project") // epi_03
|
|
48
|
+
expect(assistantContent).not.toContain("A major feature is completed or refactored") // pers_02
|
|
49
|
+
|
|
50
|
+
// 3. Verify Zone 3 (User Role)
|
|
51
|
+
const userMsg = payload.find((m: any) => m.role === "user")
|
|
52
|
+
expect(userMsg).toBeDefined()
|
|
53
|
+
// Verify it contains the appended Project-Specific Rules (Zone 3)
|
|
54
|
+
const userContentText = userMsg.content.map((c: any) => c.text).join("")
|
|
55
|
+
expect(userContentText).toContain("Hi") // The original user message
|
|
56
|
+
expect(userContentText).toContain("ZONE 3: PROJECT-SPECIFIC RULES (Context-Aware Gaps)")
|
|
57
|
+
expect(userContentText).toContain("project_specific_pack:")
|
|
58
|
+
expect(userContentText).toContain("Adhere strictly to the detected stack conventions (Solid.js, Effect-ts")
|
|
59
|
+
expect(userContentText).toContain("Extracted from AGENTS.md:")
|
|
60
|
+
})
|
|
61
|
+
})
|