orynacode-ai 1.16.2
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/AGENTS.md +131 -0
- package/BUN_SHELL_MIGRATION_PLAN.md +136 -0
- package/Dockerfile +18 -0
- package/README.md +15 -0
- package/bin/orynacode +199 -0
- package/bunfig.toml +7 -0
- package/git +0 -0
- package/migration/20260511173437_session-metadata/migration.sql +1 -0
- package/migration/20260511173437_session-metadata/snapshot.json +1500 -0
- package/package.json +155 -0
- package/parsers-config.ts +386 -0
- package/script/bench-search.ts +115 -0
- package/script/bench-test-suite.ts +52 -0
- package/script/build.ts +244 -0
- package/script/generate.ts +14 -0
- package/script/httpapi-exercise.ts +1 -0
- package/script/postinstall.mjs +189 -0
- package/script/profile-test-files.ts +42 -0
- package/script/publish.ts +213 -0
- package/script/run-workspace-server +106 -0
- package/script/schema.ts +77 -0
- package/script/time.ts +6 -0
- package/script/trace-imports.ts +153 -0
- package/specs/effect/error-boundaries-plan.md +235 -0
- package/specs/effect/errors.md +207 -0
- package/specs/effect/facades.md +218 -0
- package/specs/effect/guide.md +247 -0
- package/specs/effect/instance-context.md +13 -0
- package/specs/effect/loose-ends.md +30 -0
- package/specs/effect/migration.md +62 -0
- package/specs/effect/routes.md +61 -0
- package/specs/effect/schema.md +88 -0
- package/specs/effect/server-package.md +58 -0
- package/specs/effect/todo.md +241 -0
- package/specs/effect/tools.md +88 -0
- package/specs/openapi-translation-cleanup.md +204 -0
- package/specs/tui-plugins.md +544 -0
- package/specs/v2/api.ts +67 -0
- package/specs/v2/message-shape.md +136 -0
- package/specs/v2/notifications.md +13 -0
- package/specs/v2/tui-command-shim.md +67 -0
- package/src/account/account.ts +459 -0
- package/src/account/repo.ts +170 -0
- package/src/account/schema.ts +99 -0
- package/src/account/url.ts +8 -0
- package/src/acp/agent.ts +95 -0
- package/src/acp/config-option.ts +203 -0
- package/src/acp/content.ts +250 -0
- package/src/acp/directory.ts +210 -0
- package/src/acp/error.ts +90 -0
- package/src/acp/event.ts +344 -0
- package/src/acp/permission.ts +145 -0
- package/src/acp/profile.ts +42 -0
- package/src/acp/service.ts +1062 -0
- package/src/acp/session.ts +231 -0
- package/src/acp/tool.ts +317 -0
- package/src/acp/usage.ts +239 -0
- package/src/agent/agent.ts +433 -0
- package/src/agent/generate.txt +75 -0
- package/src/agent/prompt/compaction.txt +9 -0
- package/src/agent/prompt/explore.txt +18 -0
- package/src/agent/prompt/summary.txt +11 -0
- package/src/agent/prompt/title.txt +44 -0
- package/src/agent/subagent-permissions.ts +35 -0
- package/src/audio.d.ts +14 -0
- package/src/auth/index.ts +96 -0
- package/src/background/job.ts +36 -0
- package/src/bus/global.ts +22 -0
- package/src/cli/bootstrap.ts +11 -0
- package/src/cli/cmd/account.ts +264 -0
- package/src/cli/cmd/acp.ts +76 -0
- package/src/cli/cmd/agent.ts +259 -0
- package/src/cli/cmd/cmd.ts +7 -0
- package/src/cli/cmd/db.ts +62 -0
- package/src/cli/cmd/debug/agent.handler.ts +193 -0
- package/src/cli/cmd/debug/agent.ts +27 -0
- package/src/cli/cmd/debug/config.ts +14 -0
- package/src/cli/cmd/debug/file.ts +87 -0
- package/src/cli/cmd/debug/index.ts +87 -0
- package/src/cli/cmd/debug/lsp.ts +51 -0
- package/src/cli/cmd/debug/ripgrep.ts +99 -0
- package/src/cli/cmd/debug/scrap.ts +18 -0
- package/src/cli/cmd/debug/skill.ts +15 -0
- package/src/cli/cmd/debug/snapshot.ts +50 -0
- package/src/cli/cmd/debug/startup.ts +11 -0
- package/src/cli/cmd/debug/v2.ts +46 -0
- package/src/cli/cmd/export.ts +292 -0
- package/src/cli/cmd/generate.ts +54 -0
- package/src/cli/cmd/github.handler.ts +1593 -0
- package/src/cli/cmd/github.shared.ts +30 -0
- package/src/cli/cmd/github.ts +42 -0
- package/src/cli/cmd/import.ts +224 -0
- package/src/cli/cmd/mcp.ts +846 -0
- package/src/cli/cmd/models.ts +66 -0
- package/src/cli/cmd/plug.ts +230 -0
- package/src/cli/cmd/pr.ts +115 -0
- package/src/cli/cmd/prompt-display.ts +48 -0
- package/src/cli/cmd/providers.ts +506 -0
- package/src/cli/cmd/run/demo.ts +1274 -0
- package/src/cli/cmd/run/entry.body.ts +194 -0
- package/src/cli/cmd/run/footer.command.tsx +899 -0
- package/src/cli/cmd/run/footer.menu.tsx +306 -0
- package/src/cli/cmd/run/footer.permission.tsx +475 -0
- package/src/cli/cmd/run/footer.prompt.tsx +1207 -0
- package/src/cli/cmd/run/footer.question.tsx +579 -0
- package/src/cli/cmd/run/footer.subagent.tsx +171 -0
- package/src/cli/cmd/run/footer.ts +1092 -0
- package/src/cli/cmd/run/footer.view.tsx +935 -0
- package/src/cli/cmd/run/otel.ts +117 -0
- package/src/cli/cmd/run/permission.shared.ts +256 -0
- package/src/cli/cmd/run/prompt.shared.ts +147 -0
- package/src/cli/cmd/run/question.shared.ts +340 -0
- package/src/cli/cmd/run/runtime.boot.ts +210 -0
- package/src/cli/cmd/run/runtime.lifecycle.ts +369 -0
- package/src/cli/cmd/run/runtime.queue.ts +347 -0
- package/src/cli/cmd/run/runtime.shared.ts +17 -0
- package/src/cli/cmd/run/runtime.stdin.ts +37 -0
- package/src/cli/cmd/run/runtime.ts +879 -0
- package/src/cli/cmd/run/scrollback.shared.ts +92 -0
- package/src/cli/cmd/run/scrollback.surface.ts +435 -0
- package/src/cli/cmd/run/scrollback.writer.tsx +335 -0
- package/src/cli/cmd/run/session-data.ts +1113 -0
- package/src/cli/cmd/run/session-replay.ts +301 -0
- package/src/cli/cmd/run/session.shared.ts +196 -0
- package/src/cli/cmd/run/splash.ts +310 -0
- package/src/cli/cmd/run/stream.transport.ts +1465 -0
- package/src/cli/cmd/run/stream.ts +175 -0
- package/src/cli/cmd/run/subagent-data.ts +844 -0
- package/src/cli/cmd/run/theme.ts +603 -0
- package/src/cli/cmd/run/tool.ts +1489 -0
- package/src/cli/cmd/run/trace.ts +94 -0
- package/src/cli/cmd/run/types.ts +342 -0
- package/src/cli/cmd/run/variant.shared.ts +215 -0
- package/src/cli/cmd/run.ts +879 -0
- package/src/cli/cmd/serve.ts +24 -0
- package/src/cli/cmd/session.ts +147 -0
- package/src/cli/cmd/stats.ts +393 -0
- package/src/cli/cmd/tui/app.tsx +1113 -0
- package/src/cli/cmd/tui/attach.ts +103 -0
- package/src/cli/cmd/tui/attention.ts +262 -0
- package/src/cli/cmd/tui/component/bg-pulse-render.ts +436 -0
- package/src/cli/cmd/tui/component/bg-pulse.tsx +99 -0
- package/src/cli/cmd/tui/component/border.tsx +21 -0
- package/src/cli/cmd/tui/component/command-palette.tsx +79 -0
- package/src/cli/cmd/tui/component/dialog-agent.tsx +31 -0
- package/src/cli/cmd/tui/component/dialog-console-org.tsx +103 -0
- package/src/cli/cmd/tui/component/dialog-mcp.tsx +85 -0
- package/src/cli/cmd/tui/component/dialog-model.tsx +185 -0
- package/src/cli/cmd/tui/component/dialog-move-session.tsx +240 -0
- package/src/cli/cmd/tui/component/dialog-provider.tsx +687 -0
- package/src/cli/cmd/tui/component/dialog-retry-action.tsx +160 -0
- package/src/cli/cmd/tui/component/dialog-session-delete-failed.tsx +99 -0
- package/src/cli/cmd/tui/component/dialog-session-list.tsx +318 -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 +47 -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-create.tsx +313 -0
- package/src/cli/cmd/tui/component/dialog-workspace-file-changes.tsx +144 -0
- package/src/cli/cmd/tui/component/dialog-workspace-list.tsx +112 -0
- package/src/cli/cmd/tui/component/dialog-workspace-unavailable.tsx +69 -0
- package/src/cli/cmd/tui/component/error-component.tsx +81 -0
- package/src/cli/cmd/tui/component/logo.tsx +885 -0
- package/src/cli/cmd/tui/component/plugin-route-missing.tsx +14 -0
- package/src/cli/cmd/tui/component/prompt/autocomplete.tsx +799 -0
- package/src/cli/cmd/tui/component/prompt/cwd.ts +0 -0
- package/src/cli/cmd/tui/component/prompt/frecency.tsx +90 -0
- package/src/cli/cmd/tui/component/prompt/history.tsx +117 -0
- package/src/cli/cmd/tui/component/prompt/index.tsx +1725 -0
- package/src/cli/cmd/tui/component/prompt/move.tsx +192 -0
- package/src/cli/cmd/tui/component/prompt/part.ts +31 -0
- package/src/cli/cmd/tui/component/prompt/stash.tsx +101 -0
- package/src/cli/cmd/tui/component/prompt/traits.ts +35 -0
- package/src/cli/cmd/tui/component/prompt/workspace.tsx +137 -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/todo-item.tsx +32 -0
- package/src/cli/cmd/tui/component/use-connected.tsx +9 -0
- package/src/cli/cmd/tui/component/workspace-label.tsx +19 -0
- package/src/cli/cmd/tui/config/cwd.ts +5 -0
- package/src/cli/cmd/tui/config/keybind.ts +467 -0
- package/src/cli/cmd/tui/config/tui-migrate.ts +154 -0
- package/src/cli/cmd/tui/config/tui-schema.ts +88 -0
- package/src/cli/cmd/tui/config/tui.ts +308 -0
- package/src/cli/cmd/tui/context/agent.tsx +11 -0
- package/src/cli/cmd/tui/context/aggregate-failures.ts +51 -0
- package/src/cli/cmd/tui/context/args.tsx +15 -0
- package/src/cli/cmd/tui/context/directory.ts +15 -0
- package/src/cli/cmd/tui/context/editor-zed.ts +287 -0
- package/src/cli/cmd/tui/context/editor.ts +469 -0
- package/src/cli/cmd/tui/context/event.ts +38 -0
- package/src/cli/cmd/tui/context/exit.tsx +42 -0
- package/src/cli/cmd/tui/context/helper.tsx +25 -0
- package/src/cli/cmd/tui/context/kv.tsx +76 -0
- package/src/cli/cmd/tui/context/local.tsx +510 -0
- package/src/cli/cmd/tui/context/path-format.tsx +39 -0
- package/src/cli/cmd/tui/context/project.tsx +111 -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 +142 -0
- package/src/cli/cmd/tui/context/sync-v2.tsx +447 -0
- package/src/cli/cmd/tui/context/sync.tsx +628 -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 +230 -0
- package/src/cli/cmd/tui/context/theme/catppuccin-macchiato.json +230 -0
- package/src/cli/cmd/tui/context/theme/catppuccin.json +112 -0
- package/src/cli/cmd/tui/context/theme/cobalt2.json +225 -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/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 +234 -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/opencode.json +245 -0
- package/src/cli/cmd/tui/context/theme/orng.json +249 -0
- package/src/cli/cmd/tui/context/theme/oryna.json +95 -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 +1341 -0
- package/src/cli/cmd/tui/context/thinking.ts +67 -0
- package/src/cli/cmd/tui/context/tui-config.tsx +9 -0
- package/src/cli/cmd/tui/event.ts +53 -0
- package/src/cli/cmd/tui/feature-plugins/home/footer.tsx +98 -0
- package/src/cli/cmd/tui/feature-plugins/home/tips-view.tsx +288 -0
- package/src/cli/cmd/tui/feature-plugins/home/tips.tsx +59 -0
- package/src/cli/cmd/tui/feature-plugins/session/dialog.tsx +356 -0
- package/src/cli/cmd/tui/feature-plugins/session/index.tsx +32 -0
- package/src/cli/cmd/tui/feature-plugins/session/preview-pane.tsx +288 -0
- package/src/cli/cmd/tui/feature-plugins/session/util.tsx +54 -0
- package/src/cli/cmd/tui/feature-plugins/sidebar/context.tsx +65 -0
- package/src/cli/cmd/tui/feature-plugins/sidebar/files.tsx +70 -0
- package/src/cli/cmd/tui/feature-plugins/sidebar/footer.tsx +96 -0
- package/src/cli/cmd/tui/feature-plugins/sidebar/lsp.tsx +65 -0
- package/src/cli/cmd/tui/feature-plugins/sidebar/mcp.tsx +97 -0
- package/src/cli/cmd/tui/feature-plugins/sidebar/todo.tsx +49 -0
- package/src/cli/cmd/tui/feature-plugins/system/diff-viewer-file-tree-utils.ts +232 -0
- package/src/cli/cmd/tui/feature-plugins/system/diff-viewer-file-tree.tsx +162 -0
- package/src/cli/cmd/tui/feature-plugins/system/diff-viewer-ui.tsx +103 -0
- package/src/cli/cmd/tui/feature-plugins/system/diff-viewer.tsx +1058 -0
- package/src/cli/cmd/tui/feature-plugins/system/notifications.ts +94 -0
- package/src/cli/cmd/tui/feature-plugins/system/plugins.tsx +269 -0
- package/src/cli/cmd/tui/feature-plugins/system/session-v2.tsx +1184 -0
- package/src/cli/cmd/tui/feature-plugins/system/which-key.tsx +608 -0
- package/src/cli/cmd/tui/keymap.tsx +283 -0
- package/src/cli/cmd/tui/layer.ts +6 -0
- package/src/cli/cmd/tui/plugin/api.tsx +390 -0
- package/src/cli/cmd/tui/plugin/command-shim.ts +109 -0
- package/src/cli/cmd/tui/plugin/internal.ts +42 -0
- package/src/cli/cmd/tui/plugin/runtime.ts +1131 -0
- package/src/cli/cmd/tui/plugin/slots.tsx +60 -0
- package/src/cli/cmd/tui/routes/home/session-destination.tsx +39 -0
- package/src/cli/cmd/tui/routes/home.tsx +149 -0
- package/src/cli/cmd/tui/routes/session/dialog-fork-from-timeline.tsx +76 -0
- package/src/cli/cmd/tui/routes/session/dialog-message.tsx +108 -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 +2629 -0
- package/src/cli/cmd/tui/routes/session/permission.tsx +729 -0
- package/src/cli/cmd/tui/routes/session/question.tsx +514 -0
- package/src/cli/cmd/tui/routes/session/sidebar.tsx +102 -0
- package/src/cli/cmd/tui/routes/session/subagent-footer.tsx +132 -0
- package/src/cli/cmd/tui/thread.ts +264 -0
- package/src/cli/cmd/tui/ui/dialog-alert.tsx +66 -0
- package/src/cli/cmd/tui/ui/dialog-confirm.tsx +108 -0
- package/src/cli/cmd/tui/ui/dialog-export-options.tsx +217 -0
- package/src/cli/cmd/tui/ui/dialog-help.tsx +40 -0
- package/src/cli/cmd/tui/ui/dialog-prompt.tsx +126 -0
- package/src/cli/cmd/tui/ui/dialog-select.tsx +712 -0
- package/src/cli/cmd/tui/ui/dialog.tsx +218 -0
- package/src/cli/cmd/tui/ui/link.tsx +34 -0
- package/src/cli/cmd/tui/ui/spinner.ts +368 -0
- package/src/cli/cmd/tui/ui/toast.tsx +102 -0
- package/src/cli/cmd/tui/util/audio.ts +58 -0
- package/src/cli/cmd/tui/util/clipboard.ts +181 -0
- package/src/cli/cmd/tui/util/collapse-tool-output.ts +19 -0
- package/src/cli/cmd/tui/util/editor.ts +43 -0
- package/src/cli/cmd/tui/util/layout.ts +25 -0
- package/src/cli/cmd/tui/util/model.ts +23 -0
- package/src/cli/cmd/tui/util/provider-origin.ts +7 -0
- package/src/cli/cmd/tui/util/revert-diff.ts +18 -0
- package/src/cli/cmd/tui/util/scroll.ts +25 -0
- package/src/cli/cmd/tui/util/selection.ts +73 -0
- package/src/cli/cmd/tui/util/signal.ts +41 -0
- package/src/cli/cmd/tui/util/transcript.ts +112 -0
- package/src/cli/cmd/tui/validate-session.ts +29 -0
- package/src/cli/cmd/tui/win32.ts +130 -0
- package/src/cli/cmd/tui/worker.ts +99 -0
- package/src/cli/cmd/uninstall.ts +353 -0
- package/src/cli/cmd/upgrade.ts +74 -0
- package/src/cli/cmd/web.ts +84 -0
- package/src/cli/effect/prompt.ts +37 -0
- package/src/cli/effect-cmd.ts +96 -0
- package/src/cli/error.ts +118 -0
- package/src/cli/heap.ts +59 -0
- package/src/cli/logo.ts +21 -0
- package/src/cli/network.ts +64 -0
- package/src/cli/ui.ts +132 -0
- package/src/cli/upgrade.ts +53 -0
- package/src/command/index.ts +181 -0
- package/src/command/template/initialize.txt +66 -0
- package/src/command/template/review.txt +101 -0
- package/src/config/agent.ts +68 -0
- package/src/config/command.ts +45 -0
- package/src/config/config.ts +679 -0
- package/src/config/entry-name.ts +19 -0
- package/src/config/managed.ts +77 -0
- package/src/config/markdown.ts +36 -0
- package/src/config/parse.ts +79 -0
- package/src/config/paths.ts +45 -0
- package/src/config/plugin.ts +79 -0
- package/src/config/reference.ts +48 -0
- package/src/config/variable.ts +91 -0
- package/src/control-plane/adapters/index.ts +41 -0
- package/src/control-plane/adapters/worktree.ts +96 -0
- package/src/control-plane/dev/README.md +19 -0
- package/src/control-plane/dev/debug-workspace-plugin.ts +73 -0
- package/src/control-plane/types.ts +59 -0
- package/src/control-plane/util.ts +39 -0
- package/src/control-plane/workspace-adapter-runtime.ts +51 -0
- package/src/control-plane/workspace-context.ts +26 -0
- package/src/control-plane/workspace.ts +1075 -0
- package/src/effect/app-runtime.ts +133 -0
- package/src/effect/bootstrap-runtime.ts +23 -0
- package/src/effect/bridge.ts +84 -0
- package/src/effect/config-service.ts +67 -0
- package/src/effect/instance-ref.ts +11 -0
- package/src/effect/instance-registry.ts +12 -0
- package/src/effect/instance-state.ts +72 -0
- package/src/effect/promise.ts +17 -0
- package/src/effect/run-service.ts +47 -0
- package/src/effect/runner.ts +217 -0
- package/src/effect/runtime-flags.ts +76 -0
- package/src/env/index.ts +40 -0
- package/src/event-v2-bridge.ts +76 -0
- package/src/format/formatter.ts +404 -0
- package/src/format/index.ts +212 -0
- package/src/git/index.ts +347 -0
- package/src/id/id.ts +80 -0
- package/src/ide/index.ts +70 -0
- package/src/image/image.ts +177 -0
- package/src/index.ts +208 -0
- package/src/installation/index.ts +349 -0
- package/src/lsp/client.ts +686 -0
- package/src/lsp/diagnostic.ts +29 -0
- package/src/lsp/language.ts +121 -0
- package/src/lsp/launch.ts +21 -0
- package/src/lsp/lsp.ts +517 -0
- package/src/lsp/server.ts +2064 -0
- package/src/markdown.d.ts +4 -0
- package/src/mcp/auth.ts +171 -0
- package/src/mcp/index.ts +982 -0
- package/src/mcp/oauth-callback.ts +232 -0
- package/src/mcp/oauth-provider.ts +217 -0
- package/src/node.ts +5 -0
- package/src/oryna/agent.ts +112 -0
- package/src/oryna/reply-service.ts +8 -0
- package/src/patch/index.ts +689 -0
- package/src/permission/arity.ts +163 -0
- package/src/permission/evaluate.ts +1 -0
- package/src/permission/index.ts +230 -0
- package/src/plugin/azure.ts +26 -0
- package/src/plugin/cloudflare.ts +76 -0
- package/src/plugin/digitalocean.ts +391 -0
- package/src/plugin/github-copilot/copilot.ts +417 -0
- package/src/plugin/github-copilot/models.ts +246 -0
- package/src/plugin/index.ts +323 -0
- package/src/plugin/install.ts +439 -0
- package/src/plugin/loader.ts +237 -0
- package/src/plugin/meta.ts +188 -0
- package/src/plugin/openai/README.md +31 -0
- package/src/plugin/openai/codex.ts +647 -0
- package/src/plugin/openai/ws-pool.ts +290 -0
- package/src/plugin/openai/ws.ts +381 -0
- package/src/plugin/oryna.ts +349 -0
- package/src/plugin/shared.ts +323 -0
- package/src/plugin/xai.ts +742 -0
- package/src/project/bootstrap-service.ts +9 -0
- package/src/project/bootstrap.ts +80 -0
- package/src/project/instance-context.ts +24 -0
- package/src/project/instance-layer.ts +11 -0
- package/src/project/instance-runtime.ts +16 -0
- package/src/project/instance-store.ts +207 -0
- package/src/project/project.ts +520 -0
- package/src/project/vcs.ts +435 -0
- package/src/provider/auth.ts +230 -0
- package/src/provider/error.ts +188 -0
- package/src/provider/model-status.ts +8 -0
- package/src/provider/provider.ts +2009 -0
- package/src/provider/transform.ts +1363 -0
- package/src/pty-preparation.ts +30 -0
- package/src/question/index.ts +229 -0
- package/src/question/schema.ts +10 -0
- package/src/reference/reference.ts +239 -0
- package/src/reference/repository-cache.ts +320 -0
- package/src/server/auth.ts +48 -0
- package/src/server/cors.ts +34 -0
- package/src/server/event.ts +13 -0
- package/src/server/global-lifecycle.ts +37 -0
- package/src/server/init-projectors.ts +3 -0
- package/src/server/mdns.ts +60 -0
- package/src/server/projectors.ts +1 -0
- package/src/server/proxy-util.ts +48 -0
- package/src/server/routes/instance/httpapi/AGENTS.md +39 -0
- package/src/server/routes/instance/httpapi/api.ts +78 -0
- package/src/server/routes/instance/httpapi/errors.ts +193 -0
- package/src/server/routes/instance/httpapi/groups/config.ts +65 -0
- package/src/server/routes/instance/httpapi/groups/control-plane.ts +35 -0
- package/src/server/routes/instance/httpapi/groups/control.ts +76 -0
- package/src/server/routes/instance/httpapi/groups/event.ts +29 -0
- package/src/server/routes/instance/httpapi/groups/experimental.ts +260 -0
- package/src/server/routes/instance/httpapi/groups/file.ts +172 -0
- package/src/server/routes/instance/httpapi/groups/global.ts +138 -0
- package/src/server/routes/instance/httpapi/groups/instance.ts +206 -0
- package/src/server/routes/instance/httpapi/groups/mcp.ts +156 -0
- package/src/server/routes/instance/httpapi/groups/metadata.ts +18 -0
- package/src/server/routes/instance/httpapi/groups/permission.ts +61 -0
- package/src/server/routes/instance/httpapi/groups/project-copy.ts +88 -0
- package/src/server/routes/instance/httpapi/groups/project.ts +93 -0
- package/src/server/routes/instance/httpapi/groups/provider.ts +101 -0
- package/src/server/routes/instance/httpapi/groups/pty.ts +172 -0
- package/src/server/routes/instance/httpapi/groups/query.ts +12 -0
- package/src/server/routes/instance/httpapi/groups/question.ts +74 -0
- package/src/server/routes/instance/httpapi/groups/session.ts +462 -0
- package/src/server/routes/instance/httpapi/groups/sync.ts +113 -0
- package/src/server/routes/instance/httpapi/groups/tui.ts +208 -0
- package/src/server/routes/instance/httpapi/groups/workspace.ts +141 -0
- package/src/server/routes/instance/httpapi/handlers/config.ts +151 -0
- package/src/server/routes/instance/httpapi/handlers/control-plane.ts +37 -0
- package/src/server/routes/instance/httpapi/handlers/control.ts +40 -0
- package/src/server/routes/instance/httpapi/handlers/event.ts +102 -0
- package/src/server/routes/instance/httpapi/handlers/experimental.ts +187 -0
- package/src/server/routes/instance/httpapi/handlers/file.ts +103 -0
- package/src/server/routes/instance/httpapi/handlers/global.ts +157 -0
- package/src/server/routes/instance/httpapi/handlers/instance.ts +110 -0
- package/src/server/routes/instance/httpapi/handlers/mcp.ts +111 -0
- package/src/server/routes/instance/httpapi/handlers/permission.ts +41 -0
- package/src/server/routes/instance/httpapi/handlers/project-copy.ts +157 -0
- package/src/server/routes/instance/httpapi/handlers/project.ts +63 -0
- package/src/server/routes/instance/httpapi/handlers/provider.ts +160 -0
- package/src/server/routes/instance/httpapi/handlers/pty.ts +258 -0
- package/src/server/routes/instance/httpapi/handlers/question.ts +54 -0
- package/src/server/routes/instance/httpapi/handlers/session-errors.ts +21 -0
- package/src/server/routes/instance/httpapi/handlers/session.ts +442 -0
- package/src/server/routes/instance/httpapi/handlers/sync.ts +95 -0
- package/src/server/routes/instance/httpapi/handlers/tui.ts +131 -0
- package/src/server/routes/instance/httpapi/handlers/workspace.ts +102 -0
- package/src/server/routes/instance/httpapi/lifecycle.ts +57 -0
- package/src/server/routes/instance/httpapi/middleware/authorization.ts +147 -0
- package/src/server/routes/instance/httpapi/middleware/compression.ts +64 -0
- package/src/server/routes/instance/httpapi/middleware/cors-vary.ts +29 -0
- package/src/server/routes/instance/httpapi/middleware/error.ts +36 -0
- package/src/server/routes/instance/httpapi/middleware/fence.ts +25 -0
- package/src/server/routes/instance/httpapi/middleware/instance-context.ts +43 -0
- package/src/server/routes/instance/httpapi/middleware/proxy.ts +108 -0
- package/src/server/routes/instance/httpapi/middleware/schema-error.ts +42 -0
- package/src/server/routes/instance/httpapi/middleware/workspace-routing.ts +250 -0
- package/src/server/routes/instance/httpapi/public.ts +535 -0
- package/src/server/routes/instance/httpapi/server.ts +277 -0
- package/src/server/routes/instance/httpapi/websocket-tracker.ts +57 -0
- package/src/server/server.ts +218 -0
- package/src/server/shared/fence.ts +68 -0
- package/src/server/shared/pty-ticket.ts +15 -0
- package/src/server/shared/public-ui.ts +12 -0
- package/src/server/shared/tui-control.ts +28 -0
- package/src/server/shared/ui.ts +108 -0
- package/src/server/shared/workspace-routing.ts +38 -0
- package/src/session/compaction.ts +609 -0
- package/src/session/instruction.ts +237 -0
- package/src/session/llm/AGENTS.md +90 -0
- package/src/session/llm/ai-sdk.ts +288 -0
- package/src/session/llm/native-request.ts +196 -0
- package/src/session/llm/native-runtime.ts +195 -0
- package/src/session/llm/request.ts +215 -0
- package/src/session/llm.ts +402 -0
- package/src/session/message-error.ts +14 -0
- package/src/session/message-v2.ts +745 -0
- package/src/session/message.ts +148 -0
- package/src/session/overflow.ts +34 -0
- package/src/session/processor.ts +1063 -0
- package/src/session/prompt/anthropic.txt +105 -0
- package/src/session/prompt/beast.txt +147 -0
- package/src/session/prompt/build-switch.txt +5 -0
- package/src/session/prompt/codex.txt +79 -0
- package/src/session/prompt/copilot-gpt-5.txt +143 -0
- package/src/session/prompt/default.txt +95 -0
- package/src/session/prompt/gemini.txt +155 -0
- package/src/session/prompt/gpt.txt +107 -0
- package/src/session/prompt/kimi.txt +95 -0
- package/src/session/prompt/max-steps.txt +16 -0
- package/src/session/prompt/plan-mode.txt +70 -0
- package/src/session/prompt/plan-reminder-anthropic.txt +67 -0
- package/src/session/prompt/plan.txt +26 -0
- package/src/session/prompt/reference.ts +72 -0
- package/src/session/prompt/trinity.txt +97 -0
- package/src/session/prompt.ts +1755 -0
- package/src/session/reminders.ts +92 -0
- package/src/session/retry.ts +201 -0
- package/src/session/revert.ts +153 -0
- package/src/session/run-state.ts +153 -0
- package/src/session/schema.ts +26 -0
- package/src/session/session.ts +1116 -0
- package/src/session/status.ts +94 -0
- package/src/session/summary.ts +162 -0
- package/src/session/system.ts +84 -0
- package/src/session/todo.ts +87 -0
- package/src/session/tools.ts +211 -0
- package/src/share/session.ts +58 -0
- package/src/share/share-next.ts +379 -0
- package/src/shell/shell.ts +215 -0
- package/src/skill/discovery.ts +115 -0
- package/src/skill/index.ts +357 -0
- package/src/snapshot/index.ts +759 -0
- package/src/sql.d.ts +4 -0
- package/src/storage/schema.ts +5 -0
- package/src/storage/storage.ts +329 -0
- package/src/sync/README.md +179 -0
- package/src/sync/schema.ts +11 -0
- package/src/temporary.ts +33 -0
- package/src/tool/apply_patch.ts +313 -0
- package/src/tool/apply_patch.txt +33 -0
- package/src/tool/edit.ts +737 -0
- package/src/tool/edit.txt +10 -0
- package/src/tool/external-directory.ts +49 -0
- package/src/tool/glob.ts +84 -0
- package/src/tool/glob.txt +6 -0
- package/src/tool/grep.ts +140 -0
- package/src/tool/grep.txt +8 -0
- package/src/tool/invalid.ts +21 -0
- package/src/tool/json-schema.ts +164 -0
- package/src/tool/lsp.ts +113 -0
- package/src/tool/lsp.txt +24 -0
- package/src/tool/mcp-websearch.ts +96 -0
- package/src/tool/plan-enter.txt +14 -0
- package/src/tool/plan-exit.txt +13 -0
- package/src/tool/plan.ts +79 -0
- package/src/tool/question.ts +44 -0
- package/src/tool/question.txt +10 -0
- package/src/tool/read.ts +392 -0
- package/src/tool/read.txt +14 -0
- package/src/tool/registry.ts +475 -0
- package/src/tool/reply.ts +29 -0
- package/src/tool/schema.ts +14 -0
- package/src/tool/shell/id.ts +19 -0
- package/src/tool/shell/prompt.ts +307 -0
- package/src/tool/shell/shell.txt +21 -0
- package/src/tool/shell.ts +660 -0
- package/src/tool/skill.ts +72 -0
- package/src/tool/skill.txt +5 -0
- package/src/tool/task.ts +338 -0
- package/src/tool/task.txt +19 -0
- package/src/tool/todo.ts +57 -0
- package/src/tool/todowrite.txt +44 -0
- package/src/tool/tool.ts +183 -0
- package/src/tool/truncate.ts +160 -0
- package/src/tool/truncation-dir.ts +4 -0
- package/src/tool/webfetch.ts +192 -0
- package/src/tool/webfetch.txt +13 -0
- package/src/tool/websearch.ts +143 -0
- package/src/tool/websearch.txt +14 -0
- package/src/tool/write.ts +104 -0
- package/src/tool/write.txt +8 -0
- package/src/util/archive.ts +17 -0
- package/src/util/bom.ts +27 -0
- package/src/util/data-url.ts +9 -0
- package/src/util/defer.ts +10 -0
- package/src/util/effect-http-client.ts +11 -0
- package/src/util/error.ts +88 -0
- package/src/util/filesystem.ts +251 -0
- package/src/util/format.ts +20 -0
- package/src/util/iife.ts +3 -0
- package/src/util/lan-scan.ts +90 -0
- package/src/util/lazy.ts +20 -0
- package/src/util/local-context.ts +25 -0
- package/src/util/locale.ts +86 -0
- package/src/util/media.ts +26 -0
- package/src/util/process.ts +176 -0
- package/src/util/proxy-env.ts +72 -0
- package/src/util/queue.ts +32 -0
- package/src/util/record.ts +3 -0
- package/src/util/repository.ts +232 -0
- package/src/util/rpc.ts +66 -0
- package/src/util/signal.ts +12 -0
- package/src/util/timeout.ts +13 -0
- package/src/util/token.ts +1 -0
- package/src/util/wildcard.ts +59 -0
- package/src/worktree/index.ts +645 -0
- package/sst-env.d.ts +10 -0
- package/test/AGENTS.md +204 -0
- package/test/EFFECT_TEST_MIGRATION.md +169 -0
- package/test/account/repo.test.ts +353 -0
- package/test/account/service.test.ts +453 -0
- package/test/acp/config-option.test.ts +229 -0
- package/test/acp/content.test.ts +201 -0
- package/test/acp/directory.test.ts +186 -0
- package/test/acp/error.test.ts +67 -0
- package/test/acp/event.test.ts +711 -0
- package/test/acp/permission.test.ts +273 -0
- package/test/acp/service-session.test.ts +1174 -0
- package/test/acp/session.test.ts +200 -0
- package/test/acp/tool.test.ts +210 -0
- package/test/acp/usage.test.ts +315 -0
- package/test/agent/agent.test.ts +710 -0
- package/test/agent/plan-mode-subagent-bypass.test.ts +213 -0
- package/test/agent/plugin-agent-regression.test.ts +62 -0
- package/test/auth/auth.test.ts +77 -0
- package/test/background/job.test.ts +243 -0
- package/test/cli/account.test.ts +30 -0
- package/test/cli/acp/acp-test-client.ts +97 -0
- package/test/cli/acp/config-options.test.ts +103 -0
- package/test/cli/acp/helpers.ts +96 -0
- package/test/cli/acp/initialize-auth.test.ts +61 -0
- package/test/cli/acp/lifecycle.test.ts +118 -0
- package/test/cli/acp/prompt-content.test.ts +97 -0
- package/test/cli/acp/skills.test.ts +38 -0
- package/test/cli/cmd/tui/aggregate-failures.test.ts +93 -0
- package/test/cli/cmd/tui/attention.test.ts +484 -0
- package/test/cli/cmd/tui/dialog-workspace-create.test.ts +28 -0
- package/test/cli/cmd/tui/model-options.test.ts +30 -0
- package/test/cli/cmd/tui/notifications.test.ts +267 -0
- package/test/cli/cmd/tui/prompt-history.test.ts +44 -0
- package/test/cli/cmd/tui/prompt-part.test.ts +77 -0
- package/test/cli/cmd/tui/prompt-traits.test.ts +29 -0
- package/test/cli/cmd/tui/provider-options.test.ts +29 -0
- package/test/cli/cmd/tui/sync-fixture.tsx +64 -0
- package/test/cli/cmd/tui/sync-live-hydration.test.tsx +278 -0
- package/test/cli/cmd/tui/sync-undefined-messages.test.tsx +47 -0
- package/test/cli/cmd/tui/sync.test.tsx +70 -0
- package/test/cli/effect-cmd-instance-als.test.ts +39 -0
- package/test/cli/error.test.ts +95 -0
- package/test/cli/github-action.test.ts +199 -0
- package/test/cli/github-remote.test.ts +90 -0
- package/test/cli/help/__snapshots__/help-snapshots.test.ts.snap +631 -0
- package/test/cli/help/help-snapshots.test.ts +137 -0
- package/test/cli/import.test.ts +54 -0
- package/test/cli/mcp-add.test.ts +74 -0
- package/test/cli/plugin-auth-picker.test.ts +120 -0
- package/test/cli/run/entry.body.test.ts +536 -0
- package/test/cli/run/footer.menu.test.ts +43 -0
- package/test/cli/run/footer.view.test.tsx +927 -0
- package/test/cli/run/permission.shared.test.ts +144 -0
- package/test/cli/run/prompt.shared.test.ts +133 -0
- package/test/cli/run/question.shared.test.ts +115 -0
- package/test/cli/run/run-process.test.ts +84 -0
- package/test/cli/run/runtime.boot.test.ts +282 -0
- package/test/cli/run/runtime.queue.test.ts +465 -0
- package/test/cli/run/runtime.stdin.test.ts +71 -0
- package/test/cli/run/scrollback.surface.test.ts +1048 -0
- package/test/cli/run/session-data.test.ts +595 -0
- package/test/cli/run/session-replay.test.ts +456 -0
- package/test/cli/run/session.shared.test.ts +247 -0
- package/test/cli/run/stream.test.ts +56 -0
- package/test/cli/run/stream.transport.test.ts +2363 -0
- package/test/cli/run/subagent-data.test.ts +456 -0
- package/test/cli/run/theme.test.ts +152 -0
- package/test/cli/run/variant.shared.test.ts +217 -0
- package/test/cli/serve/serve-process.test.ts +61 -0
- package/test/cli/smokes/read-only.test.ts +115 -0
- package/test/cli/tui/__snapshots__/inline-tool-wrap-snapshot.test.tsx.snap +72 -0
- package/test/cli/tui/app-lifecycle.test.ts +261 -0
- package/test/cli/tui/dialog-prompt.test.tsx +146 -0
- package/test/cli/tui/diff-viewer-file-tree-utils.test.ts +323 -0
- package/test/cli/tui/diff-viewer-file-tree.test.tsx +197 -0
- package/test/cli/tui/diff-viewer.test.tsx +230 -0
- package/test/cli/tui/editor-context-zed.test.ts +384 -0
- package/test/cli/tui/editor-context.test.tsx +288 -0
- package/test/cli/tui/inline-tool-wrap-snapshot.test.tsx +232 -0
- package/test/cli/tui/keymap.test.tsx +136 -0
- package/test/cli/tui/plugin-add.test.ts +110 -0
- package/test/cli/tui/plugin-install.test.ts +87 -0
- package/test/cli/tui/plugin-lifecycle.test.ts +224 -0
- package/test/cli/tui/plugin-loader-entrypoint.test.ts +485 -0
- package/test/cli/tui/plugin-loader-pure.test.ts +72 -0
- package/test/cli/tui/plugin-loader.test.ts +1332 -0
- package/test/cli/tui/plugin-toggle.test.ts +264 -0
- package/test/cli/tui/prompt-submit-race.test.ts +98 -0
- package/test/cli/tui/revert-diff.test.ts +35 -0
- package/test/cli/tui/slot-replace.test.tsx +50 -0
- package/test/cli/tui/sync-v2.test.tsx +558 -0
- package/test/cli/tui/theme-store.test.ts +76 -0
- package/test/cli/tui/thinking.test.ts +36 -0
- package/test/cli/tui/thread.test.ts +28 -0
- package/test/cli/tui/transcript.test.ts +426 -0
- package/test/cli/tui/use-event.test.tsx +145 -0
- package/test/config/agent-color.test.ts +47 -0
- package/test/config/config.test.ts +1991 -0
- package/test/config/entry-name.test.ts +57 -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/lsp.test.ts +69 -0
- package/test/config/markdown.test.ts +228 -0
- package/test/config/plugin.test.ts +0 -0
- package/test/config/tui.test.ts +878 -0
- package/test/control-plane/adapters.test.ts +71 -0
- package/test/control-plane/workspace.test.ts +1704 -0
- package/test/effect/app-runtime-logger.test.ts +105 -0
- package/test/effect/config-service.test.ts +65 -0
- package/test/effect/instance-state.test.ts +391 -0
- package/test/effect/run-service.test.ts +89 -0
- package/test/effect/runner.test.ts +514 -0
- package/test/effect/runtime-flags.test.ts +373 -0
- package/test/fake/account.ts +9 -0
- package/test/fake/auth.ts +8 -0
- package/test/fake/npm.ts +8 -0
- package/test/fake/provider.ts +82 -0
- package/test/fake/skill.ts +8 -0
- package/test/filesystem/filesystem.test.ts +319 -0
- package/test/fixture/agent-plugin.constants.ts +6 -0
- package/test/fixture/agent-plugin.ts +12 -0
- package/test/fixture/config.ts +23 -0
- package/test/fixture/db.ts +11 -0
- package/test/fixture/fixture.test.ts +26 -0
- package/test/fixture/fixture.ts +224 -0
- package/test/fixture/flag.ts +20 -0
- package/test/fixture/flock-worker.ts +72 -0
- package/test/fixture/lsp/fake-lsp-server.js +249 -0
- package/test/fixture/plug-worker.ts +93 -0
- package/test/fixture/plugin-meta-worker.ts +19 -0
- package/test/fixture/plugin.ts +10 -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 +355 -0
- package/test/fixture/tui-runtime.ts +64 -0
- package/test/fixture/tui-sdk.ts +82 -0
- package/test/fixture/workspace.ts +30 -0
- package/test/fixtures/recordings/session/native-anthropic-tool-loop.json +49 -0
- package/test/fixtures/recordings/session/native-openai-oauth-tool-loop.json +45 -0
- package/test/fixtures/recordings/session/native-zen-tool-loop.json +49 -0
- package/test/format/format.test.ts +228 -0
- package/test/git/git.test.ts +178 -0
- package/test/ide/ide.test.ts +82 -0
- package/test/image/fixtures/picture-5mb-base64.png +0 -0
- package/test/image/image.test.ts +123 -0
- package/test/installation/installation.test.ts +230 -0
- package/test/lib/cli-process.ts +459 -0
- package/test/lib/effect.ts +177 -0
- package/test/lib/filesystem.ts +10 -0
- package/test/lib/llm-server.ts +771 -0
- package/test/lib/snapshot.ts +73 -0
- package/test/lib/test-provider.ts +37 -0
- package/test/lib/websocket.ts +46 -0
- package/test/lsp/client.test.ts +493 -0
- package/test/lsp/index.test.ts +232 -0
- package/test/lsp/launch.test.ts +22 -0
- package/test/lsp/lifecycle.test.ts +160 -0
- package/test/mcp/auth.test.ts +78 -0
- package/test/mcp/headers.test.ts +126 -0
- package/test/mcp/lifecycle.test.ts +888 -0
- package/test/mcp/oauth-auto-connect.test.ts +236 -0
- package/test/mcp/oauth-browser.test.ts +228 -0
- package/test/mcp/oauth-callback.test.ts +34 -0
- package/test/mcp/oauth-provider.test.ts +61 -0
- package/test/patch/patch.test.ts +383 -0
- package/test/permission/arity.test.ts +33 -0
- package/test/permission/next.test.ts +1176 -0
- package/test/permission-task.test.ts +318 -0
- package/test/plugin/auth-override.test.ts +105 -0
- package/test/plugin/cloudflare.test.ts +68 -0
- package/test/plugin/codex.test.ts +247 -0
- package/test/plugin/github-copilot-models.test.ts +332 -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 +1303 -0
- package/test/plugin/meta.test.ts +137 -0
- package/test/plugin/openai-rollout.test.ts +17 -0
- package/test/plugin/openai-ws.test.ts +877 -0
- package/test/plugin/shared.test.ts +88 -0
- package/test/plugin/trigger.test.ts +120 -0
- package/test/plugin/workspace-adapter.test.ts +137 -0
- package/test/plugin/xai.test.ts +634 -0
- package/test/preload.ts +95 -0
- package/test/project/instance-bootstrap.test.ts +110 -0
- package/test/project/instance.test.ts +245 -0
- package/test/project/migrate-global.test.ts +170 -0
- package/test/project/project-directory.test.ts +169 -0
- package/test/project/project.test.ts +818 -0
- package/test/project/vcs.test.ts +336 -0
- package/test/project/worktree-remove.test.ts +126 -0
- package/test/project/worktree.test.ts +320 -0
- package/test/provider/amazon-bedrock.test.ts +360 -0
- package/test/provider/cf-ai-gateway-e2e.test.ts +132 -0
- package/test/provider/digitalocean.test.ts +123 -0
- package/test/provider/gitlab-duo.test.ts +412 -0
- package/test/provider/header-timeout.test.ts +233 -0
- package/test/provider/model-status.test.ts +61 -0
- package/test/provider/provider.test.ts +1793 -0
- package/test/provider/transform.test.ts +3937 -0
- package/test/pty/pty-shell.test.ts +102 -0
- package/test/question/question.test.ts +465 -0
- package/test/reference/reference.test.ts +310 -0
- package/test/server/AGENTS.md +15 -0
- package/test/server/auth.test.ts +59 -0
- package/test/server/global-bus.ts +31 -0
- package/test/server/global-session-list.test.ts +107 -0
- package/test/server/httpapi-authorization.test.ts +174 -0
- package/test/server/httpapi-compression.test.ts +154 -0
- package/test/server/httpapi-config.test.ts +113 -0
- package/test/server/httpapi-control-plane.test.ts +63 -0
- package/test/server/httpapi-cors-vary.test.ts +66 -0
- package/test/server/httpapi-cors.test.ts +122 -0
- package/test/server/httpapi-error-middleware.test.ts +96 -0
- package/test/server/httpapi-event.test.ts +97 -0
- package/test/server/httpapi-exercise/assertions.ts +64 -0
- package/test/server/httpapi-exercise/backend.ts +144 -0
- package/test/server/httpapi-exercise/dsl.ts +210 -0
- package/test/server/httpapi-exercise/environment.ts +40 -0
- package/test/server/httpapi-exercise/index.ts +1535 -0
- package/test/server/httpapi-exercise/report.ts +66 -0
- package/test/server/httpapi-exercise/routing.ts +96 -0
- package/test/server/httpapi-exercise/runner.ts +267 -0
- package/test/server/httpapi-exercise/runtime.ts +52 -0
- package/test/server/httpapi-exercise/types.ts +123 -0
- package/test/server/httpapi-experimental.test.ts +300 -0
- package/test/server/httpapi-file.test.ts +76 -0
- package/test/server/httpapi-global.test.ts +66 -0
- package/test/server/httpapi-instance-context.test.ts +347 -0
- package/test/server/httpapi-instance-route-auth.test.ts +84 -0
- package/test/server/httpapi-instance.test.ts +265 -0
- package/test/server/httpapi-layer.ts +33 -0
- package/test/server/httpapi-listen.test.ts +415 -0
- package/test/server/httpapi-mcp-oauth.test.ts +73 -0
- package/test/server/httpapi-mcp.test.ts +226 -0
- package/test/server/httpapi-mdns.test.ts +82 -0
- package/test/server/httpapi-promptasync-context.test.ts +222 -0
- package/test/server/httpapi-provider.test.ts +403 -0
- package/test/server/httpapi-pty.test.ts +275 -0
- package/test/server/httpapi-public-openapi.test.ts +297 -0
- package/test/server/httpapi-query-schema-drift.test.ts +330 -0
- package/test/server/httpapi-schema-error-body.test.ts +165 -0
- package/test/server/httpapi-sdk.test.ts +909 -0
- package/test/server/httpapi-session.test.ts +1013 -0
- package/test/server/httpapi-sync.test.ts +154 -0
- package/test/server/httpapi-ui.test.ts +456 -0
- package/test/server/httpapi-v2-location.test.ts +85 -0
- package/test/server/httpapi-workspace-routing.test.ts +554 -0
- package/test/server/httpapi-workspace.test.ts +515 -0
- package/test/server/negative-tokens-regression.test.ts +83 -0
- package/test/server/project-copy.test.ts +101 -0
- package/test/server/project-init-git.test.ts +117 -0
- package/test/server/proxy-util.test.ts +113 -0
- package/test/server/sdk-error-shape.test.ts +84 -0
- package/test/server/sdk-v1-smoke.test.ts +60 -0
- package/test/server/session-actions.test.ts +112 -0
- package/test/server/session-diff-missing-patch.test.ts +99 -0
- package/test/server/session-list.test.ts +314 -0
- package/test/server/session-messages.test.ts +182 -0
- package/test/server/session-select.test.ts +69 -0
- package/test/server/workspace-proxy.test.ts +181 -0
- package/test/server/workspace-routing.test.ts +94 -0
- package/test/server/worktree-endpoint-repro.test.ts +307 -0
- package/test/session/compaction.test.ts +1835 -0
- package/test/session/instruction.test.ts +256 -0
- package/test/session/llm-native-recorded.test.ts +433 -0
- package/test/session/llm-native.test.ts +760 -0
- package/test/session/llm.test.ts +1932 -0
- package/test/session/message-v2.test.ts +1661 -0
- package/test/session/messages-pagination.test.ts +1059 -0
- package/test/session/processor-effect.test.ts +1101 -0
- package/test/session/prompt.test.ts +2318 -0
- package/test/session/retry.test.ts +439 -0
- package/test/session/revert-compact.test.ts +642 -0
- package/test/session/schema-decoding.test.ts +313 -0
- package/test/session/session-schema.test.ts +78 -0
- package/test/session/session.test.ts +251 -0
- package/test/session/snapshot-tool-race.test.ts +280 -0
- package/test/session/structured-output-integration.test.ts +235 -0
- package/test/session/structured-output.test.ts +387 -0
- package/test/session/system.test.ts +84 -0
- package/test/share/share-next.test.ts +344 -0
- package/test/shell/shell.test.ts +99 -0
- package/test/skill/discovery.test.ts +139 -0
- package/test/skill/skill.test.ts +571 -0
- package/test/snapshot/snapshot.test.ts +1121 -0
- package/test/storage/storage.test.ts +296 -0
- package/test/storage/workspace-time-migration.test.ts +50 -0
- package/test/tool/__snapshots__/parameters.test.ts.snap +484 -0
- package/test/tool/__snapshots__/tool.test.ts.snap +9 -0
- package/test/tool/apply_patch.test.ts +533 -0
- package/test/tool/edit.test.ts +578 -0
- package/test/tool/external-directory.test.ts +155 -0
- package/test/tool/fixtures/large-image.png +0 -0
- package/test/tool/fixtures/models-api.json +117299 -0
- package/test/tool/glob.test.ts +188 -0
- package/test/tool/grep.test.ts +266 -0
- package/test/tool/lsp.test.ts +181 -0
- package/test/tool/parameters.test.ts +293 -0
- package/test/tool/question.test.ts +138 -0
- package/test/tool/read.test.ts +654 -0
- package/test/tool/registry.test.ts +539 -0
- package/test/tool/shell.test.ts +1238 -0
- package/test/tool/skill.test.ts +132 -0
- package/test/tool/task.test.ts +901 -0
- package/test/tool/tool-define.test.ts +153 -0
- package/test/tool/truncation.test.ts +266 -0
- package/test/tool/webfetch.test.ts +113 -0
- package/test/tool/websearch.test.ts +99 -0
- package/test/tool/write.test.ts +276 -0
- package/test/util/data-url.test.ts +14 -0
- package/test/util/error.test.ts +64 -0
- package/test/util/filesystem.test.ts +656 -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/log.test.ts +77 -0
- package/test/util/module.test.ts +59 -0
- package/test/util/process.test.ts +128 -0
- package/test/util/repository.test.ts +93 -0
- package/test/util/timeout.test.ts +21 -0
- package/test/util/wildcard.test.ts +90 -0
- package/test/v2/session-message-updater.test.ts +270 -0
- package/tsconfig.json +17 -0
|
@@ -0,0 +1,2318 @@
|
|
|
1
|
+
import { NodeFileSystem } from "@effect/platform-node"
|
|
2
|
+
import { ConfigV1 } from "@opencode-ai/core/v1/config/config"
|
|
3
|
+
import { SessionV1 } from "@opencode-ai/core/v1/session"
|
|
4
|
+
import { Database } from "@opencode-ai/core/database/database"
|
|
5
|
+
import { eq } from "drizzle-orm"
|
|
6
|
+
import { EventV2Bridge } from "@/event-v2-bridge"
|
|
7
|
+
import { FetchHttpClient } from "effect/unstable/http"
|
|
8
|
+
import { expect } from "bun:test"
|
|
9
|
+
import { Cause, Deferred, Duration, Effect, Exit, Fiber, Layer } from "effect"
|
|
10
|
+
import path from "path"
|
|
11
|
+
import { fileURLToPath, pathToFileURL } from "url"
|
|
12
|
+
import { NamedError } from "@opencode-ai/core/util/error"
|
|
13
|
+
import { Agent as AgentSvc } from "../../src/agent/agent"
|
|
14
|
+
import { BackgroundJob } from "@/background/job"
|
|
15
|
+
import { Command } from "../../src/command"
|
|
16
|
+
import { Config } from "@/config/config"
|
|
17
|
+
import { LSP } from "@/lsp/lsp"
|
|
18
|
+
import { MCP } from "../../src/mcp"
|
|
19
|
+
import { Permission } from "../../src/permission"
|
|
20
|
+
import { Plugin } from "../../src/plugin"
|
|
21
|
+
import { Provider as ProviderSvc } from "@/provider/provider"
|
|
22
|
+
import { Env } from "../../src/env"
|
|
23
|
+
import { Git } from "../../src/git"
|
|
24
|
+
import { Image } from "../../src/image/image"
|
|
25
|
+
|
|
26
|
+
import { Question } from "../../src/question"
|
|
27
|
+
import { Todo } from "../../src/session/todo"
|
|
28
|
+
import { Session } from "@/session/session"
|
|
29
|
+
import { SessionMessageTable } from "@opencode-ai/core/session/sql"
|
|
30
|
+
import { LLM } from "../../src/session/llm"
|
|
31
|
+
import { MessageV2 } from "../../src/session/message-v2"
|
|
32
|
+
import { FSUtil } from "@opencode-ai/core/fs-util"
|
|
33
|
+
import { SessionCompaction } from "../../src/session/compaction"
|
|
34
|
+
import { SessionSummary } from "../../src/session/summary"
|
|
35
|
+
import { Instruction } from "../../src/session/instruction"
|
|
36
|
+
import { SessionProcessor } from "../../src/session/processor"
|
|
37
|
+
import { SessionPrompt } from "../../src/session/prompt"
|
|
38
|
+
import { SessionRevert } from "../../src/session/revert"
|
|
39
|
+
import { SessionRunState } from "../../src/session/run-state"
|
|
40
|
+
import { MessageID, PartID, SessionID } from "../../src/session/schema"
|
|
41
|
+
import { SessionStatus } from "../../src/session/status"
|
|
42
|
+
import { SessionV2 } from "@opencode-ai/core/session"
|
|
43
|
+
import { Skill } from "../../src/skill"
|
|
44
|
+
import { SystemPrompt } from "../../src/session/system"
|
|
45
|
+
import { Shell } from "../../src/shell/shell"
|
|
46
|
+
import { Snapshot } from "../../src/snapshot"
|
|
47
|
+
import { ToolRegistry } from "@/tool/registry"
|
|
48
|
+
import { Truncate } from "@/tool/truncate"
|
|
49
|
+
import * as Log from "@opencode-ai/core/util/log"
|
|
50
|
+
import { CrossSpawnSpawner } from "@opencode-ai/core/cross-spawn-spawner"
|
|
51
|
+
import { Search } from "@opencode-ai/core/filesystem/search"
|
|
52
|
+
import { Format } from "../../src/format"
|
|
53
|
+
import { Reference } from "../../src/reference/reference"
|
|
54
|
+
import { RepositoryCache } from "../../src/reference/repository-cache"
|
|
55
|
+
import { TestInstance } from "../fixture/fixture"
|
|
56
|
+
import { awaitWithTimeout, pollWithTimeout, testEffect } from "../lib/effect"
|
|
57
|
+
import { reply, TestLLMServer } from "../lib/llm-server"
|
|
58
|
+
import { RuntimeFlags } from "@/effect/runtime-flags"
|
|
59
|
+
import { ProviderV2 } from "@opencode-ai/core/provider"
|
|
60
|
+
import { ModelV2 } from "@opencode-ai/core/model"
|
|
61
|
+
|
|
62
|
+
void Log.init({ print: false })
|
|
63
|
+
|
|
64
|
+
const summary = Layer.succeed(
|
|
65
|
+
SessionSummary.Service,
|
|
66
|
+
SessionSummary.Service.of({
|
|
67
|
+
summarize: () => Effect.void,
|
|
68
|
+
diff: () => Effect.succeed([]),
|
|
69
|
+
computeDiff: () => Effect.succeed([]),
|
|
70
|
+
}),
|
|
71
|
+
)
|
|
72
|
+
|
|
73
|
+
const ref = {
|
|
74
|
+
providerID: ProviderV2.ID.make("test"),
|
|
75
|
+
modelID: ModelV2.ID.make("test-model"),
|
|
76
|
+
}
|
|
77
|
+
|
|
78
|
+
function withSh<A, E, R>(fx: () => Effect.Effect<A, E, R>) {
|
|
79
|
+
return Effect.acquireUseRelease(
|
|
80
|
+
Effect.sync(() => {
|
|
81
|
+
const prev = process.env.SHELL
|
|
82
|
+
process.env.SHELL = "/bin/sh"
|
|
83
|
+
Shell.preferred.reset()
|
|
84
|
+
return prev
|
|
85
|
+
}),
|
|
86
|
+
() => fx(),
|
|
87
|
+
(prev) =>
|
|
88
|
+
Effect.sync(() => {
|
|
89
|
+
if (prev === undefined) delete process.env.SHELL
|
|
90
|
+
else process.env.SHELL = prev
|
|
91
|
+
Shell.preferred.reset()
|
|
92
|
+
}),
|
|
93
|
+
)
|
|
94
|
+
}
|
|
95
|
+
|
|
96
|
+
function toolPart(parts: SessionV1.Part[]) {
|
|
97
|
+
return parts.find((part): part is SessionV1.ToolPart => part.type === "tool")
|
|
98
|
+
}
|
|
99
|
+
|
|
100
|
+
type CompletedToolPart = SessionV1.ToolPart & { state: SessionV1.ToolStateCompleted }
|
|
101
|
+
type ErrorToolPart = SessionV1.ToolPart & { state: SessionV1.ToolStateError }
|
|
102
|
+
|
|
103
|
+
function completedTool(parts: SessionV1.Part[]) {
|
|
104
|
+
const part = toolPart(parts)
|
|
105
|
+
expect(part?.state.status).toBe("completed")
|
|
106
|
+
return part?.state.status === "completed" ? (part as CompletedToolPart) : undefined
|
|
107
|
+
}
|
|
108
|
+
|
|
109
|
+
function errorTool(parts: SessionV1.Part[]) {
|
|
110
|
+
const part = toolPart(parts)
|
|
111
|
+
expect(part?.state.status).toBe("error")
|
|
112
|
+
return part?.state.status === "error" ? (part as ErrorToolPart) : undefined
|
|
113
|
+
}
|
|
114
|
+
|
|
115
|
+
const mcp = Layer.succeed(
|
|
116
|
+
MCP.Service,
|
|
117
|
+
MCP.Service.of({
|
|
118
|
+
status: () => Effect.succeed({}),
|
|
119
|
+
clients: () => Effect.succeed({}),
|
|
120
|
+
tools: () => Effect.succeed({}),
|
|
121
|
+
prompts: () => Effect.succeed({}),
|
|
122
|
+
resources: () => Effect.succeed({}),
|
|
123
|
+
add: () => Effect.succeed({ status: { status: "disabled" as const } }),
|
|
124
|
+
connect: () => Effect.void,
|
|
125
|
+
disconnect: () => Effect.void,
|
|
126
|
+
getPrompt: () => Effect.succeed(undefined),
|
|
127
|
+
readResource: () => Effect.succeed(undefined),
|
|
128
|
+
startAuth: () => Effect.die("unexpected MCP auth in prompt-effect tests"),
|
|
129
|
+
authenticate: () => Effect.die("unexpected MCP auth in prompt-effect tests"),
|
|
130
|
+
finishAuth: () => Effect.die("unexpected MCP auth in prompt-effect tests"),
|
|
131
|
+
removeAuth: () => Effect.void,
|
|
132
|
+
supportsOAuth: () => Effect.succeed(false),
|
|
133
|
+
hasStoredTokens: () => Effect.succeed(false),
|
|
134
|
+
getAuthStatus: () => Effect.succeed("not_authenticated" as const),
|
|
135
|
+
}),
|
|
136
|
+
)
|
|
137
|
+
|
|
138
|
+
const lsp = Layer.succeed(
|
|
139
|
+
LSP.Service,
|
|
140
|
+
LSP.Service.of({
|
|
141
|
+
init: () => Effect.void,
|
|
142
|
+
status: () => Effect.succeed([]),
|
|
143
|
+
hasClients: () => Effect.succeed(false),
|
|
144
|
+
touchFile: () => Effect.void,
|
|
145
|
+
diagnostics: () => Effect.succeed({}),
|
|
146
|
+
hover: () => Effect.succeed(undefined),
|
|
147
|
+
definition: () => Effect.succeed([]),
|
|
148
|
+
references: () => Effect.succeed([]),
|
|
149
|
+
implementation: () => Effect.succeed([]),
|
|
150
|
+
documentSymbol: () => Effect.succeed([]),
|
|
151
|
+
workspaceSymbol: () => Effect.succeed([]),
|
|
152
|
+
prepareCallHierarchy: () => Effect.succeed([]),
|
|
153
|
+
incomingCalls: () => Effect.succeed([]),
|
|
154
|
+
outgoingCalls: () => Effect.succeed([]),
|
|
155
|
+
}),
|
|
156
|
+
)
|
|
157
|
+
|
|
158
|
+
const status = SessionStatus.layer.pipe(Layer.provideMerge(EventV2Bridge.defaultLayer))
|
|
159
|
+
const run = SessionRunState.layer.pipe(Layer.provide(status))
|
|
160
|
+
const infra = Layer.mergeAll(NodeFileSystem.layer, CrossSpawnSpawner.defaultLayer)
|
|
161
|
+
|
|
162
|
+
const processorCreateStarted: Array<() => void> = []
|
|
163
|
+
const blockingProcessor = Layer.succeed(
|
|
164
|
+
SessionProcessor.Service,
|
|
165
|
+
SessionProcessor.Service.of({
|
|
166
|
+
create: () => Effect.sync(() => processorCreateStarted.shift()?.()).pipe(Effect.andThen(Effect.never)),
|
|
167
|
+
}),
|
|
168
|
+
)
|
|
169
|
+
|
|
170
|
+
function makePrompt(input?: { processor?: "blocking" }) {
|
|
171
|
+
const deps = Layer.mergeAll(
|
|
172
|
+
Session.defaultLayer,
|
|
173
|
+
Snapshot.defaultLayer,
|
|
174
|
+
LLM.defaultLayer,
|
|
175
|
+
Env.defaultLayer,
|
|
176
|
+
AgentSvc.defaultLayer,
|
|
177
|
+
Command.defaultLayer,
|
|
178
|
+
Permission.defaultLayer,
|
|
179
|
+
Plugin.defaultLayer,
|
|
180
|
+
Config.defaultLayer,
|
|
181
|
+
ProviderSvc.defaultLayer,
|
|
182
|
+
lsp,
|
|
183
|
+
mcp,
|
|
184
|
+
FSUtil.defaultLayer,
|
|
185
|
+
BackgroundJob.defaultLayer,
|
|
186
|
+
status,
|
|
187
|
+
Database.defaultLayer,
|
|
188
|
+
EventV2Bridge.defaultLayer,
|
|
189
|
+
).pipe(Layer.provideMerge(infra))
|
|
190
|
+
const question = Question.layer.pipe(Layer.provideMerge(deps))
|
|
191
|
+
const todo = Todo.layer.pipe(Layer.provideMerge(deps))
|
|
192
|
+
const registry = ToolRegistry.layer.pipe(
|
|
193
|
+
Layer.provide(Skill.defaultLayer),
|
|
194
|
+
Layer.provide(FetchHttpClient.layer),
|
|
195
|
+
Layer.provide(CrossSpawnSpawner.defaultLayer),
|
|
196
|
+
Layer.provide(RepositoryCache.defaultLayer),
|
|
197
|
+
Layer.provide(Git.defaultLayer),
|
|
198
|
+
Layer.provide(Reference.defaultLayer),
|
|
199
|
+
Layer.provide(Search.defaultLayer),
|
|
200
|
+
Layer.provide(Format.defaultLayer),
|
|
201
|
+
Layer.provide(RuntimeFlags.layer({ experimentalEventSystem: true })),
|
|
202
|
+
Layer.provideMerge(todo),
|
|
203
|
+
Layer.provideMerge(question),
|
|
204
|
+
Layer.provideMerge(deps),
|
|
205
|
+
)
|
|
206
|
+
const trunc = Truncate.layer.pipe(Layer.provideMerge(deps))
|
|
207
|
+
const proc =
|
|
208
|
+
input?.processor === "blocking"
|
|
209
|
+
? blockingProcessor
|
|
210
|
+
: SessionProcessor.layer.pipe(
|
|
211
|
+
Layer.provide(summary),
|
|
212
|
+
Layer.provide(Image.defaultLayer),
|
|
213
|
+
Layer.provide(RuntimeFlags.layer({ experimentalEventSystem: true })),
|
|
214
|
+
Layer.provideMerge(deps),
|
|
215
|
+
)
|
|
216
|
+
const compact = SessionCompaction.layer.pipe(
|
|
217
|
+
Layer.provide(RuntimeFlags.layer({ experimentalEventSystem: true })),
|
|
218
|
+
Layer.provideMerge(proc),
|
|
219
|
+
Layer.provideMerge(deps),
|
|
220
|
+
)
|
|
221
|
+
return SessionPrompt.layer.pipe(
|
|
222
|
+
Layer.provide(SessionRevert.defaultLayer),
|
|
223
|
+
Layer.provide(Image.defaultLayer),
|
|
224
|
+
Layer.provide(Reference.defaultLayer),
|
|
225
|
+
Layer.provide(summary),
|
|
226
|
+
Layer.provideMerge(run),
|
|
227
|
+
Layer.provideMerge(compact),
|
|
228
|
+
Layer.provideMerge(proc),
|
|
229
|
+
Layer.provideMerge(registry),
|
|
230
|
+
Layer.provideMerge(trunc),
|
|
231
|
+
Layer.provide(Instruction.defaultLayer),
|
|
232
|
+
Layer.provide(SystemPrompt.defaultLayer),
|
|
233
|
+
Layer.provide(RuntimeFlags.layer({ experimentalEventSystem: true })),
|
|
234
|
+
Layer.provideMerge(deps),
|
|
235
|
+
Layer.provide(summary),
|
|
236
|
+
)
|
|
237
|
+
}
|
|
238
|
+
|
|
239
|
+
function makeHttp(input?: { processor?: "blocking" }) {
|
|
240
|
+
return Layer.mergeAll(TestLLMServer.layer, makePrompt(input))
|
|
241
|
+
}
|
|
242
|
+
|
|
243
|
+
function makeHttpNoLLMServer(input?: { processor?: "blocking" }) {
|
|
244
|
+
return makePrompt(input)
|
|
245
|
+
}
|
|
246
|
+
|
|
247
|
+
const it = testEffect(makeHttp())
|
|
248
|
+
const noLLMServer = testEffect(makeHttpNoLLMServer())
|
|
249
|
+
const raceNoLLMServer = testEffect(makeHttpNoLLMServer({ processor: "blocking" }))
|
|
250
|
+
const unix = process.platform !== "win32" ? it.instance : it.instance.skip
|
|
251
|
+
const unixNoLLMServer = process.platform !== "win32" ? noLLMServer.instance : noLLMServer.instance.skip
|
|
252
|
+
|
|
253
|
+
// Config that registers a custom "test" provider with a "test-model" model
|
|
254
|
+
// so provider model lookup succeeds inside the loop.
|
|
255
|
+
const cfg = {
|
|
256
|
+
provider: {
|
|
257
|
+
test: {
|
|
258
|
+
name: "Test",
|
|
259
|
+
id: "test",
|
|
260
|
+
env: [],
|
|
261
|
+
npm: "@ai-sdk/openai-compatible",
|
|
262
|
+
models: {
|
|
263
|
+
"test-model": {
|
|
264
|
+
id: "test-model",
|
|
265
|
+
name: "Test Model",
|
|
266
|
+
attachment: false,
|
|
267
|
+
reasoning: false,
|
|
268
|
+
temperature: false,
|
|
269
|
+
tool_call: true,
|
|
270
|
+
release_date: "2025-01-01",
|
|
271
|
+
limit: { context: 100000, output: 10000 },
|
|
272
|
+
cost: { input: 0, output: 0 },
|
|
273
|
+
options: {},
|
|
274
|
+
},
|
|
275
|
+
},
|
|
276
|
+
options: {
|
|
277
|
+
apiKey: "test-key",
|
|
278
|
+
baseURL: "http://localhost:1/v1",
|
|
279
|
+
},
|
|
280
|
+
},
|
|
281
|
+
},
|
|
282
|
+
}
|
|
283
|
+
|
|
284
|
+
function providerCfg(url: string) {
|
|
285
|
+
return {
|
|
286
|
+
...cfg,
|
|
287
|
+
provider: {
|
|
288
|
+
...cfg.provider,
|
|
289
|
+
test: {
|
|
290
|
+
...cfg.provider.test,
|
|
291
|
+
options: {
|
|
292
|
+
...cfg.provider.test.options,
|
|
293
|
+
baseURL: url,
|
|
294
|
+
},
|
|
295
|
+
},
|
|
296
|
+
},
|
|
297
|
+
}
|
|
298
|
+
}
|
|
299
|
+
|
|
300
|
+
const writeText = Effect.fn("test.writeText")(function* (file: string, text: string) {
|
|
301
|
+
const fs = yield* FSUtil.Service
|
|
302
|
+
yield* fs.writeWithDirs(file, text)
|
|
303
|
+
})
|
|
304
|
+
|
|
305
|
+
const ensureDir = Effect.fn("test.ensureDir")(function* (dir: string) {
|
|
306
|
+
const fs = yield* FSUtil.Service
|
|
307
|
+
yield* fs.ensureDir(dir)
|
|
308
|
+
})
|
|
309
|
+
|
|
310
|
+
const writeConfig = Effect.fn("test.writeConfig")(function* (dir: string, config: Partial<ConfigV1.Info>) {
|
|
311
|
+
yield* writeText(
|
|
312
|
+
path.join(dir, "opencode.json"),
|
|
313
|
+
JSON.stringify({ $schema: "https://opencode.ai/config.json", ...config }),
|
|
314
|
+
)
|
|
315
|
+
})
|
|
316
|
+
|
|
317
|
+
const useServerConfig = Effect.fn("test.useServerConfig")(function* (config: (url: string) => Partial<ConfigV1.Info>) {
|
|
318
|
+
const { directory: dir } = yield* TestInstance
|
|
319
|
+
const llm = yield* TestLLMServer
|
|
320
|
+
yield* writeConfig(dir, config(llm.url))
|
|
321
|
+
return { dir, llm }
|
|
322
|
+
})
|
|
323
|
+
|
|
324
|
+
// Wait for a session's runner to enter a busy state. SessionStatus is flipped
|
|
325
|
+
// inside Runner.startShell's serialized transition, so cancel can't no-op once
|
|
326
|
+
// we observe it.
|
|
327
|
+
const waitForBusy = (sessionID: SessionID, duration: Duration.Input = "2 seconds") =>
|
|
328
|
+
pollWithTimeout(
|
|
329
|
+
Effect.gen(function* () {
|
|
330
|
+
const status = yield* SessionStatus.Service
|
|
331
|
+
const s = yield* status.get(sessionID)
|
|
332
|
+
return s.type === "busy" ? (true as const) : undefined
|
|
333
|
+
}),
|
|
334
|
+
`session ${sessionID} never became busy`,
|
|
335
|
+
duration,
|
|
336
|
+
)
|
|
337
|
+
|
|
338
|
+
const hasBash = Effect.sync(() => Bun.which("bash") !== null)
|
|
339
|
+
|
|
340
|
+
const deferredAsPromise = <A>(deferred: Deferred.Deferred<A>): PromiseLike<A> => ({
|
|
341
|
+
then: (onfulfilled, onrejected) => {
|
|
342
|
+
Effect.runFork(
|
|
343
|
+
Deferred.await(deferred).pipe(
|
|
344
|
+
Effect.match({
|
|
345
|
+
onFailure: (error) => {
|
|
346
|
+
onrejected?.(error)
|
|
347
|
+
},
|
|
348
|
+
onSuccess: (value) => {
|
|
349
|
+
onfulfilled?.(value)
|
|
350
|
+
},
|
|
351
|
+
}),
|
|
352
|
+
),
|
|
353
|
+
)
|
|
354
|
+
return deferredAsPromise(deferred) as PromiseLike<never>
|
|
355
|
+
},
|
|
356
|
+
})
|
|
357
|
+
|
|
358
|
+
function defer<T>() {
|
|
359
|
+
let resolve!: (value: T | PromiseLike<T>) => void
|
|
360
|
+
const promise = new Promise<T>((done) => {
|
|
361
|
+
resolve = done
|
|
362
|
+
})
|
|
363
|
+
return { promise, resolve }
|
|
364
|
+
}
|
|
365
|
+
|
|
366
|
+
const succeedVoid = (deferred: Deferred.Deferred<void>) => {
|
|
367
|
+
Effect.runSync(Deferred.succeed(deferred, void 0).pipe(Effect.ignore))
|
|
368
|
+
}
|
|
369
|
+
|
|
370
|
+
const user = Effect.fn("test.user")(function* (sessionID: SessionID, text: string) {
|
|
371
|
+
const session = yield* Session.Service
|
|
372
|
+
const msg = yield* session.updateMessage({
|
|
373
|
+
id: MessageID.ascending(),
|
|
374
|
+
role: "user",
|
|
375
|
+
sessionID,
|
|
376
|
+
agent: "build",
|
|
377
|
+
model: ref,
|
|
378
|
+
time: { created: Date.now() },
|
|
379
|
+
})
|
|
380
|
+
yield* session.updatePart({
|
|
381
|
+
id: PartID.ascending(),
|
|
382
|
+
messageID: msg.id,
|
|
383
|
+
sessionID,
|
|
384
|
+
type: "text",
|
|
385
|
+
text,
|
|
386
|
+
})
|
|
387
|
+
return msg
|
|
388
|
+
})
|
|
389
|
+
|
|
390
|
+
const seed = Effect.fn("test.seed")(function* (sessionID: SessionID, opts?: { finish?: string }) {
|
|
391
|
+
const session = yield* Session.Service
|
|
392
|
+
const msg = yield* user(sessionID, "hello")
|
|
393
|
+
const assistant: SessionV1.Assistant = {
|
|
394
|
+
id: MessageID.ascending(),
|
|
395
|
+
role: "assistant",
|
|
396
|
+
parentID: msg.id,
|
|
397
|
+
sessionID,
|
|
398
|
+
mode: "build",
|
|
399
|
+
agent: "build",
|
|
400
|
+
cost: 0,
|
|
401
|
+
path: { cwd: "/tmp", root: "/tmp" },
|
|
402
|
+
tokens: { input: 0, output: 0, reasoning: 0, cache: { read: 0, write: 0 } },
|
|
403
|
+
modelID: ref.modelID,
|
|
404
|
+
providerID: ref.providerID,
|
|
405
|
+
time: { created: Date.now() },
|
|
406
|
+
...(opts?.finish ? { finish: opts.finish } : {}),
|
|
407
|
+
}
|
|
408
|
+
yield* session.updateMessage(assistant)
|
|
409
|
+
yield* session.updatePart({
|
|
410
|
+
id: PartID.ascending(),
|
|
411
|
+
messageID: assistant.id,
|
|
412
|
+
sessionID,
|
|
413
|
+
type: "text",
|
|
414
|
+
text: "hi there",
|
|
415
|
+
})
|
|
416
|
+
return { user: msg, assistant }
|
|
417
|
+
})
|
|
418
|
+
|
|
419
|
+
const addSubtask = (sessionID: SessionID, messageID: MessageID, model = ref) =>
|
|
420
|
+
Effect.gen(function* () {
|
|
421
|
+
const session = yield* Session.Service
|
|
422
|
+
yield* session.updatePart({
|
|
423
|
+
id: PartID.ascending(),
|
|
424
|
+
messageID,
|
|
425
|
+
sessionID,
|
|
426
|
+
type: "subtask",
|
|
427
|
+
prompt: "look into the cache key path",
|
|
428
|
+
description: "inspect bug",
|
|
429
|
+
agent: "general",
|
|
430
|
+
model,
|
|
431
|
+
})
|
|
432
|
+
})
|
|
433
|
+
|
|
434
|
+
const boot = Effect.fn("test.boot")(function* (input?: { title?: string }) {
|
|
435
|
+
const config = yield* Config.Service
|
|
436
|
+
const prompt = yield* SessionPrompt.Service
|
|
437
|
+
const run = yield* SessionRunState.Service
|
|
438
|
+
const sessions = yield* Session.Service
|
|
439
|
+
yield* config.get()
|
|
440
|
+
const chat = yield* sessions.create(input ?? { title: "Pinned" })
|
|
441
|
+
return { prompt, run, sessions, chat }
|
|
442
|
+
})
|
|
443
|
+
|
|
444
|
+
// Loop semantics
|
|
445
|
+
|
|
446
|
+
noLLMServer.instance(
|
|
447
|
+
"loop exits immediately when last assistant has stop finish",
|
|
448
|
+
() =>
|
|
449
|
+
Effect.gen(function* () {
|
|
450
|
+
const prompt = yield* SessionPrompt.Service
|
|
451
|
+
const sessions = yield* Session.Service
|
|
452
|
+
const chat = yield* sessions.create({ title: "Pinned" })
|
|
453
|
+
yield* seed(chat.id, { finish: "stop" })
|
|
454
|
+
|
|
455
|
+
const result = yield* prompt.loop({ sessionID: chat.id })
|
|
456
|
+
expect(result.info.role).toBe("assistant")
|
|
457
|
+
if (result.info.role === "assistant") expect(result.info.finish).toBe("stop")
|
|
458
|
+
}),
|
|
459
|
+
{ config: cfg },
|
|
460
|
+
)
|
|
461
|
+
|
|
462
|
+
it.instance("loop exits without an LLM request for interrupted orphan tool calls", () =>
|
|
463
|
+
Effect.gen(function* () {
|
|
464
|
+
const { llm } = yield* useServerConfig(providerCfg)
|
|
465
|
+
const prompt = yield* SessionPrompt.Service
|
|
466
|
+
const sessions = yield* Session.Service
|
|
467
|
+
const chat = yield* sessions.create({ title: "Pinned" })
|
|
468
|
+
const seeded = yield* seed(chat.id, { finish: "stop" })
|
|
469
|
+
yield* sessions.updatePart({
|
|
470
|
+
id: PartID.ascending(),
|
|
471
|
+
messageID: seeded.assistant.id,
|
|
472
|
+
sessionID: chat.id,
|
|
473
|
+
type: "tool",
|
|
474
|
+
callID: "interrupted-call",
|
|
475
|
+
tool: "edit",
|
|
476
|
+
state: {
|
|
477
|
+
status: "error",
|
|
478
|
+
input: {},
|
|
479
|
+
error: "Tool execution aborted",
|
|
480
|
+
metadata: { interrupted: true },
|
|
481
|
+
time: { start: 1, end: 2 },
|
|
482
|
+
},
|
|
483
|
+
})
|
|
484
|
+
|
|
485
|
+
const result = yield* prompt.loop({ sessionID: chat.id })
|
|
486
|
+
expect(result.info.id).toBe(seeded.assistant.id)
|
|
487
|
+
expect(yield* llm.hits).toHaveLength(0)
|
|
488
|
+
}),
|
|
489
|
+
)
|
|
490
|
+
|
|
491
|
+
it.instance("loop calls LLM and returns assistant message", () =>
|
|
492
|
+
Effect.gen(function* () {
|
|
493
|
+
const { llm } = yield* useServerConfig(providerCfg)
|
|
494
|
+
const prompt = yield* SessionPrompt.Service
|
|
495
|
+
const sessions = yield* Session.Service
|
|
496
|
+
const chat = yield* sessions.create({
|
|
497
|
+
title: "Pinned",
|
|
498
|
+
permission: [{ permission: "*", pattern: "*", action: "allow" }],
|
|
499
|
+
})
|
|
500
|
+
yield* prompt.prompt({
|
|
501
|
+
sessionID: chat.id,
|
|
502
|
+
agent: "build",
|
|
503
|
+
noReply: true,
|
|
504
|
+
parts: [{ type: "text", text: "hello" }],
|
|
505
|
+
})
|
|
506
|
+
yield* llm.text("world")
|
|
507
|
+
|
|
508
|
+
const result = yield* prompt.loop({ sessionID: chat.id })
|
|
509
|
+
expect(result.info.role).toBe("assistant")
|
|
510
|
+
const parts = result.parts.filter((p) => p.type === "text")
|
|
511
|
+
expect(parts.some((p) => p.type === "text" && p.text === "world")).toBe(true)
|
|
512
|
+
expect(yield* llm.hits).toHaveLength(1)
|
|
513
|
+
}),
|
|
514
|
+
)
|
|
515
|
+
|
|
516
|
+
it.instance("loop stops provider overflow instead of auto-compacting when disabled", () =>
|
|
517
|
+
Effect.gen(function* () {
|
|
518
|
+
const { llm } = yield* useServerConfig((url) => ({
|
|
519
|
+
...providerCfg(url),
|
|
520
|
+
compaction: { auto: false },
|
|
521
|
+
}))
|
|
522
|
+
const prompt = yield* SessionPrompt.Service
|
|
523
|
+
const sessions = yield* Session.Service
|
|
524
|
+
const chat = yield* sessions.create({ title: "Pinned" })
|
|
525
|
+
|
|
526
|
+
yield* llm.error(413, { error: { message: "request entity too large" } })
|
|
527
|
+
yield* prompt.prompt({
|
|
528
|
+
sessionID: chat.id,
|
|
529
|
+
agent: "build",
|
|
530
|
+
noReply: true,
|
|
531
|
+
parts: [{ type: "text", text: "hello" }],
|
|
532
|
+
})
|
|
533
|
+
|
|
534
|
+
const result = yield* prompt.loop({ sessionID: chat.id })
|
|
535
|
+
const messages = yield* sessions.messages({ sessionID: chat.id })
|
|
536
|
+
|
|
537
|
+
expect(result.info.role).toBe("assistant")
|
|
538
|
+
if (result.info.role === "assistant") {
|
|
539
|
+
expect(result.info.error?.name).toBe("ContextOverflowError")
|
|
540
|
+
expect(result.info.finish).toBe("error")
|
|
541
|
+
}
|
|
542
|
+
expect(messages.some((message) => message.parts.some((part) => part.type === "compaction"))).toBe(false)
|
|
543
|
+
}),
|
|
544
|
+
)
|
|
545
|
+
|
|
546
|
+
noLLMServer.instance.skip(
|
|
547
|
+
"prompt emits v2 prompted and synthetic events (v2 projector disabled)",
|
|
548
|
+
() =>
|
|
549
|
+
Effect.gen(function* () {
|
|
550
|
+
const prompt = yield* SessionPrompt.Service
|
|
551
|
+
const sessions = yield* Session.Service
|
|
552
|
+
const chat = yield* sessions.create({ title: "Pinned" })
|
|
553
|
+
|
|
554
|
+
yield* prompt.prompt({
|
|
555
|
+
sessionID: chat.id,
|
|
556
|
+
agent: "build",
|
|
557
|
+
noReply: true,
|
|
558
|
+
parts: [
|
|
559
|
+
{ type: "text", text: "hello v2" },
|
|
560
|
+
{
|
|
561
|
+
type: "file",
|
|
562
|
+
mime: "text/plain",
|
|
563
|
+
filename: "note.txt",
|
|
564
|
+
url: "data:text/plain;base64,bm90ZSBjb250ZW50",
|
|
565
|
+
},
|
|
566
|
+
],
|
|
567
|
+
})
|
|
568
|
+
|
|
569
|
+
const messages = yield* SessionV2.Service.use((session) => session.messages({ sessionID: chat.id })).pipe(
|
|
570
|
+
Effect.provide(SessionV2.defaultLayer),
|
|
571
|
+
)
|
|
572
|
+
const { db } = yield* Database.Service
|
|
573
|
+
const row = yield* db
|
|
574
|
+
.select()
|
|
575
|
+
.from(SessionMessageTable)
|
|
576
|
+
.where(eq(SessionMessageTable.session_id, chat.id))
|
|
577
|
+
.get()
|
|
578
|
+
.pipe(Effect.orDie)
|
|
579
|
+
expect(messages.find((message) => message.type === "user")).toMatchObject({ type: "user", text: "hello v2" })
|
|
580
|
+
expect(typeof row?.data.time.created).toBe("number")
|
|
581
|
+
expect(messages).toEqual(
|
|
582
|
+
expect.arrayContaining([
|
|
583
|
+
expect.objectContaining({ type: "synthetic", text: expect.stringContaining("Called the Read tool") }),
|
|
584
|
+
expect.objectContaining({ type: "synthetic", text: "note content" }),
|
|
585
|
+
]),
|
|
586
|
+
)
|
|
587
|
+
}),
|
|
588
|
+
{ config: cfg },
|
|
589
|
+
)
|
|
590
|
+
|
|
591
|
+
it.instance("static loop returns assistant text through local provider", () =>
|
|
592
|
+
Effect.gen(function* () {
|
|
593
|
+
const { llm } = yield* useServerConfig(providerCfg)
|
|
594
|
+
const prompt = yield* SessionPrompt.Service
|
|
595
|
+
const sessions = yield* Session.Service
|
|
596
|
+
const session = yield* sessions.create({
|
|
597
|
+
title: "Prompt provider",
|
|
598
|
+
permission: [{ permission: "*", pattern: "*", action: "allow" }],
|
|
599
|
+
})
|
|
600
|
+
|
|
601
|
+
yield* prompt.prompt({
|
|
602
|
+
sessionID: session.id,
|
|
603
|
+
agent: "build",
|
|
604
|
+
noReply: true,
|
|
605
|
+
parts: [{ type: "text", text: "hello" }],
|
|
606
|
+
})
|
|
607
|
+
|
|
608
|
+
yield* llm.text("world")
|
|
609
|
+
|
|
610
|
+
const result = yield* prompt.loop({ sessionID: session.id })
|
|
611
|
+
expect(result.info.role).toBe("assistant")
|
|
612
|
+
expect(result.parts.some((part) => part.type === "text" && part.text === "world")).toBe(true)
|
|
613
|
+
expect(yield* llm.hits).toHaveLength(1)
|
|
614
|
+
expect(yield* llm.pending).toBe(0)
|
|
615
|
+
}),
|
|
616
|
+
)
|
|
617
|
+
|
|
618
|
+
it.instance("static loop consumes queued replies across turns", () =>
|
|
619
|
+
Effect.gen(function* () {
|
|
620
|
+
const { llm } = yield* useServerConfig(providerCfg)
|
|
621
|
+
const prompt = yield* SessionPrompt.Service
|
|
622
|
+
const sessions = yield* Session.Service
|
|
623
|
+
const session = yield* sessions.create({
|
|
624
|
+
title: "Prompt provider turns",
|
|
625
|
+
permission: [{ permission: "*", pattern: "*", action: "allow" }],
|
|
626
|
+
})
|
|
627
|
+
|
|
628
|
+
yield* prompt.prompt({
|
|
629
|
+
sessionID: session.id,
|
|
630
|
+
agent: "build",
|
|
631
|
+
noReply: true,
|
|
632
|
+
parts: [{ type: "text", text: "hello one" }],
|
|
633
|
+
})
|
|
634
|
+
|
|
635
|
+
yield* llm.text("world one")
|
|
636
|
+
|
|
637
|
+
const first = yield* prompt.loop({ sessionID: session.id })
|
|
638
|
+
expect(first.info.role).toBe("assistant")
|
|
639
|
+
expect(first.parts.some((part) => part.type === "text" && part.text === "world one")).toBe(true)
|
|
640
|
+
|
|
641
|
+
yield* prompt.prompt({
|
|
642
|
+
sessionID: session.id,
|
|
643
|
+
agent: "build",
|
|
644
|
+
noReply: true,
|
|
645
|
+
parts: [{ type: "text", text: "hello two" }],
|
|
646
|
+
})
|
|
647
|
+
|
|
648
|
+
yield* llm.text("world two")
|
|
649
|
+
|
|
650
|
+
const second = yield* prompt.loop({ sessionID: session.id })
|
|
651
|
+
expect(second.info.role).toBe("assistant")
|
|
652
|
+
expect(second.parts.some((part) => part.type === "text" && part.text === "world two")).toBe(true)
|
|
653
|
+
|
|
654
|
+
expect(yield* llm.hits).toHaveLength(2)
|
|
655
|
+
expect(yield* llm.pending).toBe(0)
|
|
656
|
+
}),
|
|
657
|
+
)
|
|
658
|
+
|
|
659
|
+
it.instance("loop continues when finish is tool-calls", () =>
|
|
660
|
+
Effect.gen(function* () {
|
|
661
|
+
const { llm } = yield* useServerConfig(providerCfg)
|
|
662
|
+
const prompt = yield* SessionPrompt.Service
|
|
663
|
+
const sessions = yield* Session.Service
|
|
664
|
+
const session = yield* sessions.create({
|
|
665
|
+
title: "Pinned",
|
|
666
|
+
permission: [{ permission: "*", pattern: "*", action: "allow" }],
|
|
667
|
+
})
|
|
668
|
+
yield* prompt.prompt({
|
|
669
|
+
sessionID: session.id,
|
|
670
|
+
agent: "build",
|
|
671
|
+
noReply: true,
|
|
672
|
+
parts: [{ type: "text", text: "hello" }],
|
|
673
|
+
})
|
|
674
|
+
yield* llm.tool("first", { value: "first" })
|
|
675
|
+
yield* llm.text("second")
|
|
676
|
+
|
|
677
|
+
const result = yield* prompt.loop({ sessionID: session.id })
|
|
678
|
+
expect(yield* llm.calls).toBe(2)
|
|
679
|
+
expect(result.info.role).toBe("assistant")
|
|
680
|
+
if (result.info.role === "assistant") {
|
|
681
|
+
expect(result.parts.some((part) => part.type === "text" && part.text === "second")).toBe(true)
|
|
682
|
+
expect(result.info.finish).toBe("stop")
|
|
683
|
+
}
|
|
684
|
+
}),
|
|
685
|
+
)
|
|
686
|
+
|
|
687
|
+
it.instance("glob tool keeps instance context during prompt runs", () =>
|
|
688
|
+
Effect.gen(function* () {
|
|
689
|
+
const { dir, llm } = yield* useServerConfig(providerCfg)
|
|
690
|
+
const prompt = yield* SessionPrompt.Service
|
|
691
|
+
const sessions = yield* Session.Service
|
|
692
|
+
const session = yield* sessions.create({
|
|
693
|
+
title: "Glob context",
|
|
694
|
+
permission: [{ permission: "*", pattern: "*", action: "allow" }],
|
|
695
|
+
})
|
|
696
|
+
const file = path.join(dir, "probe.txt")
|
|
697
|
+
yield* writeText(file, "probe")
|
|
698
|
+
|
|
699
|
+
yield* prompt.prompt({
|
|
700
|
+
sessionID: session.id,
|
|
701
|
+
agent: "build",
|
|
702
|
+
noReply: true,
|
|
703
|
+
parts: [{ type: "text", text: "find text files" }],
|
|
704
|
+
})
|
|
705
|
+
yield* llm.tool("glob", { pattern: "**/*.txt" })
|
|
706
|
+
yield* llm.text("done")
|
|
707
|
+
|
|
708
|
+
const result = yield* prompt.loop({ sessionID: session.id })
|
|
709
|
+
expect(result.info.role).toBe("assistant")
|
|
710
|
+
|
|
711
|
+
const msgs = yield* MessageV2.filterCompactedEffect(session.id)
|
|
712
|
+
const tool = msgs
|
|
713
|
+
.flatMap((msg) => msg.parts)
|
|
714
|
+
.find(
|
|
715
|
+
(part): part is CompletedToolPart =>
|
|
716
|
+
part.type === "tool" && part.tool === "glob" && part.state.status === "completed",
|
|
717
|
+
)
|
|
718
|
+
if (!tool) return
|
|
719
|
+
|
|
720
|
+
expect(tool.state.output).toContain(file)
|
|
721
|
+
expect(tool.state.output).not.toContain("No context found for instance")
|
|
722
|
+
expect(result.parts.some((part) => part.type === "text" && part.text === "done")).toBe(true)
|
|
723
|
+
}),
|
|
724
|
+
)
|
|
725
|
+
|
|
726
|
+
it.instance("loop continues when finish is stop but assistant has tool parts", () =>
|
|
727
|
+
Effect.gen(function* () {
|
|
728
|
+
const { llm } = yield* useServerConfig(providerCfg)
|
|
729
|
+
const prompt = yield* SessionPrompt.Service
|
|
730
|
+
const sessions = yield* Session.Service
|
|
731
|
+
const session = yield* sessions.create({
|
|
732
|
+
title: "Pinned",
|
|
733
|
+
permission: [{ permission: "*", pattern: "*", action: "allow" }],
|
|
734
|
+
})
|
|
735
|
+
yield* prompt.prompt({
|
|
736
|
+
sessionID: session.id,
|
|
737
|
+
agent: "build",
|
|
738
|
+
noReply: true,
|
|
739
|
+
parts: [{ type: "text", text: "hello" }],
|
|
740
|
+
})
|
|
741
|
+
yield* llm.push(reply().tool("first", { value: "first" }).stop())
|
|
742
|
+
yield* llm.text("second")
|
|
743
|
+
|
|
744
|
+
const result = yield* prompt.loop({ sessionID: session.id })
|
|
745
|
+
expect(yield* llm.calls).toBe(2)
|
|
746
|
+
expect(result.info.role).toBe("assistant")
|
|
747
|
+
if (result.info.role === "assistant") {
|
|
748
|
+
expect(result.parts.some((part) => part.type === "text" && part.text === "second")).toBe(true)
|
|
749
|
+
expect(result.info.finish).toBe("stop")
|
|
750
|
+
}
|
|
751
|
+
}),
|
|
752
|
+
)
|
|
753
|
+
|
|
754
|
+
it.instance("failed subtask preserves metadata on error tool state", () =>
|
|
755
|
+
Effect.gen(function* () {
|
|
756
|
+
const { llm } = yield* useServerConfig((url) => ({
|
|
757
|
+
...providerCfg(url),
|
|
758
|
+
agent: {
|
|
759
|
+
general: {
|
|
760
|
+
model: "test/missing-model",
|
|
761
|
+
},
|
|
762
|
+
},
|
|
763
|
+
}))
|
|
764
|
+
const prompt = yield* SessionPrompt.Service
|
|
765
|
+
const sessions = yield* Session.Service
|
|
766
|
+
const chat = yield* sessions.create({ title: "Pinned" })
|
|
767
|
+
yield* llm.tool("task", {
|
|
768
|
+
description: "inspect bug",
|
|
769
|
+
prompt: "look into the cache key path",
|
|
770
|
+
subagent_type: "general",
|
|
771
|
+
})
|
|
772
|
+
yield* llm.text("done")
|
|
773
|
+
const msg = yield* user(chat.id, "hello")
|
|
774
|
+
yield* addSubtask(chat.id, msg.id)
|
|
775
|
+
|
|
776
|
+
const result = yield* prompt.loop({ sessionID: chat.id })
|
|
777
|
+
expect(result.info.role).toBe("assistant")
|
|
778
|
+
expect(yield* llm.calls).toBe(2)
|
|
779
|
+
|
|
780
|
+
const msgs = yield* MessageV2.filterCompactedEffect(chat.id)
|
|
781
|
+
const taskMsg = msgs.find((item) => item.info.role === "assistant" && item.info.agent === "general")
|
|
782
|
+
expect(taskMsg?.info.role).toBe("assistant")
|
|
783
|
+
if (!taskMsg || taskMsg.info.role !== "assistant") return
|
|
784
|
+
|
|
785
|
+
const tool = errorTool(taskMsg.parts)
|
|
786
|
+
if (!tool) return
|
|
787
|
+
|
|
788
|
+
expect(tool.state.error).toContain("Tool execution failed")
|
|
789
|
+
expect(tool.state.metadata).toBeDefined()
|
|
790
|
+
expect(tool.state.metadata?.sessionId).toBeDefined()
|
|
791
|
+
expect(tool.state.metadata?.model).toEqual({
|
|
792
|
+
providerID: ProviderV2.ID.make("test"),
|
|
793
|
+
modelID: ModelV2.ID.make("missing-model"),
|
|
794
|
+
})
|
|
795
|
+
}),
|
|
796
|
+
)
|
|
797
|
+
|
|
798
|
+
it.instance(
|
|
799
|
+
"running subtask preserves metadata after tool-call transition",
|
|
800
|
+
() =>
|
|
801
|
+
Effect.gen(function* () {
|
|
802
|
+
const { llm } = yield* useServerConfig(providerCfg)
|
|
803
|
+
const prompt = yield* SessionPrompt.Service
|
|
804
|
+
const sessions = yield* Session.Service
|
|
805
|
+
const chat = yield* sessions.create({ title: "Pinned" })
|
|
806
|
+
yield* llm.hang
|
|
807
|
+
const msg = yield* user(chat.id, "hello")
|
|
808
|
+
yield* addSubtask(chat.id, msg.id)
|
|
809
|
+
|
|
810
|
+
const fiber = yield* prompt.loop({ sessionID: chat.id }).pipe(Effect.forkChild)
|
|
811
|
+
|
|
812
|
+
const tool = yield* pollWithTimeout(
|
|
813
|
+
Effect.gen(function* () {
|
|
814
|
+
const msgs = yield* MessageV2.filterCompactedEffect(chat.id)
|
|
815
|
+
const taskMsg = msgs.find((item) => item.info.role === "assistant" && item.info.agent === "general")
|
|
816
|
+
const tool = taskMsg?.parts.find((part): part is SessionV1.ToolPart => part.type === "tool")
|
|
817
|
+
if (tool?.state.status === "running" && tool.state.metadata?.sessionId) return tool
|
|
818
|
+
}),
|
|
819
|
+
"timed out waiting for running subtask metadata",
|
|
820
|
+
)
|
|
821
|
+
|
|
822
|
+
if (tool.state.status !== "running") return
|
|
823
|
+
expect(typeof tool.state.metadata?.sessionId).toBe("string")
|
|
824
|
+
expect(tool.state.title).toBeDefined()
|
|
825
|
+
expect(tool.state.metadata?.model).toBeDefined()
|
|
826
|
+
|
|
827
|
+
yield* prompt.cancel(chat.id)
|
|
828
|
+
yield* Fiber.await(fiber)
|
|
829
|
+
}),
|
|
830
|
+
5_000,
|
|
831
|
+
)
|
|
832
|
+
|
|
833
|
+
it.instance(
|
|
834
|
+
"running task tool preserves metadata after tool-call transition",
|
|
835
|
+
() =>
|
|
836
|
+
Effect.gen(function* () {
|
|
837
|
+
const { llm } = yield* useServerConfig(providerCfg)
|
|
838
|
+
const prompt = yield* SessionPrompt.Service
|
|
839
|
+
const sessions = yield* Session.Service
|
|
840
|
+
const chat = yield* sessions.create({
|
|
841
|
+
title: "Pinned",
|
|
842
|
+
permission: [{ permission: "*", pattern: "*", action: "allow" }],
|
|
843
|
+
})
|
|
844
|
+
yield* llm.tool("task", {
|
|
845
|
+
description: "inspect bug",
|
|
846
|
+
prompt: "look into the cache key path",
|
|
847
|
+
subagent_type: "general",
|
|
848
|
+
})
|
|
849
|
+
yield* llm.hang
|
|
850
|
+
yield* user(chat.id, "hello")
|
|
851
|
+
|
|
852
|
+
const fiber = yield* prompt.loop({ sessionID: chat.id }).pipe(Effect.forkChild)
|
|
853
|
+
|
|
854
|
+
const tool = yield* pollWithTimeout(
|
|
855
|
+
Effect.gen(function* () {
|
|
856
|
+
const msgs = yield* MessageV2.filterCompactedEffect(chat.id)
|
|
857
|
+
const assistant = msgs.findLast((item) => item.info.role === "assistant" && item.info.agent === "build")
|
|
858
|
+
const tool = assistant?.parts.find(
|
|
859
|
+
(part): part is SessionV1.ToolPart => part.type === "tool" && part.tool === "task",
|
|
860
|
+
)
|
|
861
|
+
if (tool?.state.status === "running" && tool.state.metadata?.sessionId) return tool
|
|
862
|
+
}),
|
|
863
|
+
"timed out waiting for running task metadata",
|
|
864
|
+
)
|
|
865
|
+
|
|
866
|
+
if (tool.state.status !== "running") return
|
|
867
|
+
expect(typeof tool.state.metadata?.sessionId).toBe("string")
|
|
868
|
+
expect(tool.state.title).toBe("inspect bug")
|
|
869
|
+
expect(tool.state.metadata?.model).toBeDefined()
|
|
870
|
+
|
|
871
|
+
yield* prompt.cancel(chat.id)
|
|
872
|
+
yield* Fiber.await(fiber)
|
|
873
|
+
}),
|
|
874
|
+
10_000,
|
|
875
|
+
)
|
|
876
|
+
|
|
877
|
+
it.instance(
|
|
878
|
+
"loop sets status to busy then idle",
|
|
879
|
+
() =>
|
|
880
|
+
Effect.gen(function* () {
|
|
881
|
+
const { llm } = yield* useServerConfig(providerCfg)
|
|
882
|
+
const prompt = yield* SessionPrompt.Service
|
|
883
|
+
const sessions = yield* Session.Service
|
|
884
|
+
const status = yield* SessionStatus.Service
|
|
885
|
+
|
|
886
|
+
yield* llm.hang
|
|
887
|
+
|
|
888
|
+
const chat = yield* sessions.create({})
|
|
889
|
+
yield* user(chat.id, "hi")
|
|
890
|
+
|
|
891
|
+
const fiber = yield* prompt.loop({ sessionID: chat.id }).pipe(Effect.forkChild)
|
|
892
|
+
yield* llm.wait(1)
|
|
893
|
+
expect((yield* status.get(chat.id)).type).toBe("busy")
|
|
894
|
+
yield* prompt.cancel(chat.id)
|
|
895
|
+
yield* Fiber.await(fiber)
|
|
896
|
+
expect((yield* status.get(chat.id)).type).toBe("idle")
|
|
897
|
+
}),
|
|
898
|
+
3_000,
|
|
899
|
+
)
|
|
900
|
+
|
|
901
|
+
// Cancel semantics
|
|
902
|
+
|
|
903
|
+
it.instance(
|
|
904
|
+
"cancel interrupts loop and resolves with an assistant message",
|
|
905
|
+
() =>
|
|
906
|
+
Effect.gen(function* () {
|
|
907
|
+
const { llm } = yield* useServerConfig(providerCfg)
|
|
908
|
+
const prompt = yield* SessionPrompt.Service
|
|
909
|
+
const sessions = yield* Session.Service
|
|
910
|
+
const chat = yield* sessions.create({ title: "Pinned" })
|
|
911
|
+
yield* seed(chat.id)
|
|
912
|
+
|
|
913
|
+
yield* llm.hang
|
|
914
|
+
|
|
915
|
+
yield* user(chat.id, "more")
|
|
916
|
+
|
|
917
|
+
const fiber = yield* prompt.loop({ sessionID: chat.id }).pipe(Effect.forkChild)
|
|
918
|
+
yield* llm.wait(1)
|
|
919
|
+
yield* prompt.cancel(chat.id)
|
|
920
|
+
const exit = yield* Fiber.await(fiber)
|
|
921
|
+
expect(Exit.isSuccess(exit)).toBe(true)
|
|
922
|
+
if (Exit.isSuccess(exit)) {
|
|
923
|
+
expect(exit.value.info.role).toBe("assistant")
|
|
924
|
+
}
|
|
925
|
+
}),
|
|
926
|
+
3_000,
|
|
927
|
+
)
|
|
928
|
+
|
|
929
|
+
it.instance(
|
|
930
|
+
"cancel records MessageAbortedError on interrupted process",
|
|
931
|
+
() =>
|
|
932
|
+
Effect.gen(function* () {
|
|
933
|
+
const { llm } = yield* useServerConfig(providerCfg)
|
|
934
|
+
const prompt = yield* SessionPrompt.Service
|
|
935
|
+
const sessions = yield* Session.Service
|
|
936
|
+
const chat = yield* sessions.create({ title: "Pinned" })
|
|
937
|
+
yield* llm.hang
|
|
938
|
+
yield* user(chat.id, "hello")
|
|
939
|
+
|
|
940
|
+
const fiber = yield* prompt.loop({ sessionID: chat.id }).pipe(Effect.forkChild)
|
|
941
|
+
yield* llm.wait(1)
|
|
942
|
+
yield* prompt.cancel(chat.id)
|
|
943
|
+
const exit = yield* Fiber.await(fiber)
|
|
944
|
+
expect(Exit.isSuccess(exit)).toBe(true)
|
|
945
|
+
if (Exit.isSuccess(exit)) {
|
|
946
|
+
const info = exit.value.info
|
|
947
|
+
if (info.role === "assistant") {
|
|
948
|
+
expect(info.error?.name).toBe("MessageAbortedError")
|
|
949
|
+
}
|
|
950
|
+
}
|
|
951
|
+
}),
|
|
952
|
+
3_000,
|
|
953
|
+
)
|
|
954
|
+
|
|
955
|
+
raceNoLLMServer.instance(
|
|
956
|
+
"finalizes assistant when cancelled before processor creation completes",
|
|
957
|
+
() =>
|
|
958
|
+
Effect.gen(function* () {
|
|
959
|
+
processorCreateStarted.length = 0
|
|
960
|
+
yield* Effect.addFinalizer(() =>
|
|
961
|
+
Effect.sync(() => {
|
|
962
|
+
processorCreateStarted.length = 0
|
|
963
|
+
}),
|
|
964
|
+
)
|
|
965
|
+
|
|
966
|
+
const prompt = yield* SessionPrompt.Service
|
|
967
|
+
const sessions = yield* Session.Service
|
|
968
|
+
const chat = yield* sessions.create({ title: "Processor creation race" })
|
|
969
|
+
|
|
970
|
+
yield* prompt.prompt({
|
|
971
|
+
sessionID: chat.id,
|
|
972
|
+
agent: "build",
|
|
973
|
+
noReply: true,
|
|
974
|
+
parts: [{ type: "text", text: "first" }],
|
|
975
|
+
})
|
|
976
|
+
|
|
977
|
+
const firstCreate = defer<void>()
|
|
978
|
+
processorCreateStarted.push(firstCreate.resolve)
|
|
979
|
+
const first = yield* prompt.loop({ sessionID: chat.id }).pipe(Effect.forkChild)
|
|
980
|
+
yield* Effect.promise(() => firstCreate.promise)
|
|
981
|
+
|
|
982
|
+
yield* prompt.cancel(chat.id)
|
|
983
|
+
const firstExit = yield* Fiber.await(first)
|
|
984
|
+
expect(Exit.isSuccess(firstExit)).toBe(true)
|
|
985
|
+
|
|
986
|
+
let messages = yield* sessions.messages({ sessionID: chat.id })
|
|
987
|
+
const firstInterrupted = messages.at(-1)
|
|
988
|
+
expect(firstInterrupted?.info.role).toBe("assistant")
|
|
989
|
+
expect(firstInterrupted?.parts).toHaveLength(0)
|
|
990
|
+
if (firstInterrupted?.info.role === "assistant") {
|
|
991
|
+
expect(firstInterrupted.info.finish).toBeUndefined()
|
|
992
|
+
expect(firstInterrupted.info.time.completed).toBeNumber()
|
|
993
|
+
expect(firstInterrupted.info.error?.name).toBe("MessageAbortedError")
|
|
994
|
+
}
|
|
995
|
+
|
|
996
|
+
yield* prompt.prompt({
|
|
997
|
+
sessionID: chat.id,
|
|
998
|
+
agent: "build",
|
|
999
|
+
noReply: true,
|
|
1000
|
+
parts: [{ type: "text", text: "second" }],
|
|
1001
|
+
})
|
|
1002
|
+
|
|
1003
|
+
const secondCreate = defer<void>()
|
|
1004
|
+
processorCreateStarted.push(secondCreate.resolve)
|
|
1005
|
+
const second = yield* prompt.loop({ sessionID: chat.id }).pipe(Effect.forkChild)
|
|
1006
|
+
yield* Effect.promise(() => secondCreate.promise)
|
|
1007
|
+
|
|
1008
|
+
yield* prompt.cancel(chat.id)
|
|
1009
|
+
const secondExit = yield* Fiber.await(second)
|
|
1010
|
+
expect(Exit.isSuccess(secondExit)).toBe(true)
|
|
1011
|
+
|
|
1012
|
+
messages = yield* sessions.messages({ sessionID: chat.id })
|
|
1013
|
+
const poisonMessages = messages.filter(
|
|
1014
|
+
(message) =>
|
|
1015
|
+
message.info.role === "assistant" &&
|
|
1016
|
+
message.parts.length === 0 &&
|
|
1017
|
+
!message.info.finish &&
|
|
1018
|
+
!message.info.time.completed &&
|
|
1019
|
+
!message.info.error,
|
|
1020
|
+
)
|
|
1021
|
+
expect(poisonMessages).toHaveLength(0)
|
|
1022
|
+
|
|
1023
|
+
const interruptedMessages = messages.filter(
|
|
1024
|
+
(message) =>
|
|
1025
|
+
message.info.role === "assistant" &&
|
|
1026
|
+
message.parts.length === 0 &&
|
|
1027
|
+
message.info.time.completed &&
|
|
1028
|
+
message.info.error?.name === "MessageAbortedError",
|
|
1029
|
+
)
|
|
1030
|
+
expect(interruptedMessages).toHaveLength(2)
|
|
1031
|
+
|
|
1032
|
+
const lastUser = messages.at(-2)
|
|
1033
|
+
const lastAssistant = messages.at(-1)
|
|
1034
|
+
expect(lastUser?.info.role).toBe("user")
|
|
1035
|
+
expect(lastAssistant?.info.role).toBe("assistant")
|
|
1036
|
+
if (lastUser?.info.role === "user" && lastAssistant?.info.role === "assistant") {
|
|
1037
|
+
expect(lastAssistant.info.parentID).toBe(lastUser?.info.id)
|
|
1038
|
+
}
|
|
1039
|
+
}),
|
|
1040
|
+
{ config: cfg },
|
|
1041
|
+
3_000,
|
|
1042
|
+
)
|
|
1043
|
+
|
|
1044
|
+
noLLMServer.instance(
|
|
1045
|
+
"cancel finalizes subtask tool state",
|
|
1046
|
+
() =>
|
|
1047
|
+
Effect.gen(function* () {
|
|
1048
|
+
const ready = yield* Deferred.make<void>()
|
|
1049
|
+
const aborted = yield* Deferred.make<void>()
|
|
1050
|
+
const registry = yield* ToolRegistry.Service
|
|
1051
|
+
const { task } = yield* registry.named()
|
|
1052
|
+
const original = task.execute
|
|
1053
|
+
task.execute = (_args, ctx) =>
|
|
1054
|
+
Effect.callback<never>((_resume) => {
|
|
1055
|
+
ctx.abort.addEventListener("abort", () => succeedVoid(aborted), { once: true })
|
|
1056
|
+
if (ctx.abort.aborted) succeedVoid(aborted)
|
|
1057
|
+
succeedVoid(ready)
|
|
1058
|
+
return Effect.sync(() => succeedVoid(aborted))
|
|
1059
|
+
})
|
|
1060
|
+
yield* Effect.addFinalizer(() => Effect.sync(() => void (task.execute = original)))
|
|
1061
|
+
|
|
1062
|
+
const { prompt, chat } = yield* boot()
|
|
1063
|
+
const msg = yield* user(chat.id, "hello")
|
|
1064
|
+
yield* addSubtask(chat.id, msg.id)
|
|
1065
|
+
|
|
1066
|
+
const fiber = yield* prompt.loop({ sessionID: chat.id }).pipe(Effect.forkChild)
|
|
1067
|
+
yield* awaitWithTimeout(Deferred.await(ready), "timed out waiting for task tool to start", "10 seconds")
|
|
1068
|
+
yield* prompt.cancel(chat.id)
|
|
1069
|
+
|
|
1070
|
+
const exit = yield* Fiber.await(fiber)
|
|
1071
|
+
expect(Exit.isSuccess(exit)).toBe(true)
|
|
1072
|
+
yield* awaitWithTimeout(Deferred.await(aborted), "timed out waiting for task tool abort", "10 seconds")
|
|
1073
|
+
|
|
1074
|
+
const msgs = yield* MessageV2.filterCompactedEffect(chat.id)
|
|
1075
|
+
const taskMsg = msgs.find((item) => item.info.role === "assistant" && item.info.agent === "general")
|
|
1076
|
+
expect(taskMsg?.info.role).toBe("assistant")
|
|
1077
|
+
if (!taskMsg || taskMsg.info.role !== "assistant") return
|
|
1078
|
+
|
|
1079
|
+
const tool = toolPart(taskMsg.parts)
|
|
1080
|
+
expect(tool?.type).toBe("tool")
|
|
1081
|
+
if (!tool) return
|
|
1082
|
+
|
|
1083
|
+
expect(tool.state.status).not.toBe("running")
|
|
1084
|
+
expect(taskMsg.info.time.completed).toBeDefined()
|
|
1085
|
+
expect(taskMsg.info.finish).toBeDefined()
|
|
1086
|
+
}),
|
|
1087
|
+
{ config: cfg },
|
|
1088
|
+
30_000,
|
|
1089
|
+
)
|
|
1090
|
+
|
|
1091
|
+
it.instance(
|
|
1092
|
+
"cancel propagates from slash command subtask to child session",
|
|
1093
|
+
() =>
|
|
1094
|
+
Effect.gen(function* () {
|
|
1095
|
+
const { llm } = yield* useServerConfig(providerCfg)
|
|
1096
|
+
const prompt = yield* SessionPrompt.Service
|
|
1097
|
+
const sessions = yield* Session.Service
|
|
1098
|
+
const status = yield* SessionStatus.Service
|
|
1099
|
+
const chat = yield* sessions.create({ title: "Pinned" })
|
|
1100
|
+
yield* llm.hang
|
|
1101
|
+
const msg = yield* user(chat.id, "hello")
|
|
1102
|
+
yield* addSubtask(chat.id, msg.id)
|
|
1103
|
+
|
|
1104
|
+
const fiber = yield* prompt.loop({ sessionID: chat.id }).pipe(Effect.forkChild)
|
|
1105
|
+
yield* llm.wait(1)
|
|
1106
|
+
|
|
1107
|
+
const msgs = yield* MessageV2.filterCompactedEffect(chat.id)
|
|
1108
|
+
const taskMsg = msgs.find((item) => item.info.role === "assistant" && item.info.agent === "general")
|
|
1109
|
+
const tool = taskMsg ? toolPart(taskMsg.parts) : undefined
|
|
1110
|
+
const sessionID = tool?.state.status === "running" ? tool.state.metadata?.sessionId : undefined
|
|
1111
|
+
expect(typeof sessionID).toBe("string")
|
|
1112
|
+
if (typeof sessionID !== "string") throw new Error("missing child session id")
|
|
1113
|
+
const childID = SessionID.make(sessionID)
|
|
1114
|
+
expect((yield* status.get(childID)).type).toBe("busy")
|
|
1115
|
+
|
|
1116
|
+
yield* prompt.cancel(chat.id)
|
|
1117
|
+
const exit = yield* Fiber.await(fiber)
|
|
1118
|
+
expect(Exit.isSuccess(exit)).toBe(true)
|
|
1119
|
+
|
|
1120
|
+
expect((yield* status.get(chat.id)).type).toBe("idle")
|
|
1121
|
+
expect((yield* status.get(childID)).type).toBe("idle")
|
|
1122
|
+
}),
|
|
1123
|
+
10_000,
|
|
1124
|
+
)
|
|
1125
|
+
|
|
1126
|
+
it.instance(
|
|
1127
|
+
"cancel with queued callers resolves all cleanly",
|
|
1128
|
+
() =>
|
|
1129
|
+
Effect.gen(function* () {
|
|
1130
|
+
const { llm } = yield* useServerConfig(providerCfg)
|
|
1131
|
+
const prompt = yield* SessionPrompt.Service
|
|
1132
|
+
const sessions = yield* Session.Service
|
|
1133
|
+
const chat = yield* sessions.create({ title: "Pinned" })
|
|
1134
|
+
yield* llm.hang
|
|
1135
|
+
yield* user(chat.id, "hello")
|
|
1136
|
+
|
|
1137
|
+
const a = yield* prompt.loop({ sessionID: chat.id }).pipe(Effect.forkChild)
|
|
1138
|
+
yield* llm.wait(1)
|
|
1139
|
+
const b = yield* prompt.loop({ sessionID: chat.id }).pipe(Effect.forkChild)
|
|
1140
|
+
yield* Effect.sleep(50)
|
|
1141
|
+
|
|
1142
|
+
yield* prompt.cancel(chat.id)
|
|
1143
|
+
const [exitA, exitB] = yield* Effect.all([Fiber.await(a), Fiber.await(b)])
|
|
1144
|
+
expect(Exit.isSuccess(exitA)).toBe(true)
|
|
1145
|
+
expect(Exit.isSuccess(exitB)).toBe(true)
|
|
1146
|
+
if (Exit.isSuccess(exitA) && Exit.isSuccess(exitB)) {
|
|
1147
|
+
expect(exitA.value.info.id).toBe(exitB.value.info.id)
|
|
1148
|
+
}
|
|
1149
|
+
}),
|
|
1150
|
+
{ git: true },
|
|
1151
|
+
3_000,
|
|
1152
|
+
)
|
|
1153
|
+
|
|
1154
|
+
// Queue semantics
|
|
1155
|
+
|
|
1156
|
+
noLLMServer.instance("concurrent loop callers get same result", () =>
|
|
1157
|
+
Effect.gen(function* () {
|
|
1158
|
+
const { prompt, run, chat } = yield* boot()
|
|
1159
|
+
yield* seed(chat.id, { finish: "stop" })
|
|
1160
|
+
|
|
1161
|
+
const [a, b] = yield* Effect.all([prompt.loop({ sessionID: chat.id }), prompt.loop({ sessionID: chat.id })], {
|
|
1162
|
+
concurrency: "unbounded",
|
|
1163
|
+
})
|
|
1164
|
+
|
|
1165
|
+
expect(a.info.id).toBe(b.info.id)
|
|
1166
|
+
expect(a.info.role).toBe("assistant")
|
|
1167
|
+
yield* run.assertNotBusy(chat.id)
|
|
1168
|
+
}),
|
|
1169
|
+
)
|
|
1170
|
+
|
|
1171
|
+
it.instance(
|
|
1172
|
+
"concurrent loop callers all receive same error result",
|
|
1173
|
+
() =>
|
|
1174
|
+
Effect.gen(function* () {
|
|
1175
|
+
const { llm } = yield* useServerConfig(providerCfg)
|
|
1176
|
+
const prompt = yield* SessionPrompt.Service
|
|
1177
|
+
const sessions = yield* Session.Service
|
|
1178
|
+
const chat = yield* sessions.create({ title: "Pinned" })
|
|
1179
|
+
|
|
1180
|
+
yield* llm.fail("boom")
|
|
1181
|
+
yield* user(chat.id, "hello")
|
|
1182
|
+
|
|
1183
|
+
const [a, b] = yield* Effect.all([prompt.loop({ sessionID: chat.id }), prompt.loop({ sessionID: chat.id })], {
|
|
1184
|
+
concurrency: "unbounded",
|
|
1185
|
+
})
|
|
1186
|
+
expect(a.info.id).toBe(b.info.id)
|
|
1187
|
+
expect(a.info.role).toBe("assistant")
|
|
1188
|
+
}),
|
|
1189
|
+
3_000,
|
|
1190
|
+
)
|
|
1191
|
+
|
|
1192
|
+
it.instance(
|
|
1193
|
+
"prompt submitted during an active run is included in the next LLM input",
|
|
1194
|
+
() =>
|
|
1195
|
+
Effect.gen(function* () {
|
|
1196
|
+
const { llm } = yield* useServerConfig(providerCfg)
|
|
1197
|
+
const gate = yield* Deferred.make<void>()
|
|
1198
|
+
const prompt = yield* SessionPrompt.Service
|
|
1199
|
+
const sessions = yield* Session.Service
|
|
1200
|
+
const chat = yield* sessions.create({ title: "Pinned" })
|
|
1201
|
+
|
|
1202
|
+
yield* llm.hold("first", deferredAsPromise(gate))
|
|
1203
|
+
yield* llm.text("second")
|
|
1204
|
+
|
|
1205
|
+
const a = yield* prompt
|
|
1206
|
+
.prompt({
|
|
1207
|
+
sessionID: chat.id,
|
|
1208
|
+
agent: "build",
|
|
1209
|
+
model: ref,
|
|
1210
|
+
parts: [{ type: "text", text: "first" }],
|
|
1211
|
+
})
|
|
1212
|
+
.pipe(Effect.forkChild)
|
|
1213
|
+
|
|
1214
|
+
yield* llm.wait(1)
|
|
1215
|
+
|
|
1216
|
+
const id = MessageID.ascending()
|
|
1217
|
+
const b = yield* prompt
|
|
1218
|
+
.prompt({
|
|
1219
|
+
sessionID: chat.id,
|
|
1220
|
+
messageID: id,
|
|
1221
|
+
agent: "build",
|
|
1222
|
+
model: ref,
|
|
1223
|
+
parts: [{ type: "text", text: "second" }],
|
|
1224
|
+
})
|
|
1225
|
+
.pipe(Effect.forkChild)
|
|
1226
|
+
|
|
1227
|
+
yield* pollWithTimeout(
|
|
1228
|
+
sessions
|
|
1229
|
+
.messages({ sessionID: chat.id })
|
|
1230
|
+
.pipe(
|
|
1231
|
+
Effect.map((msgs) =>
|
|
1232
|
+
msgs.some((msg) => msg.info.role === "user" && msg.info.id === id) ? true : undefined,
|
|
1233
|
+
),
|
|
1234
|
+
),
|
|
1235
|
+
"timed out waiting for second prompt to save",
|
|
1236
|
+
)
|
|
1237
|
+
|
|
1238
|
+
yield* Deferred.succeed(gate, void 0)
|
|
1239
|
+
|
|
1240
|
+
const [ea, eb] = yield* Effect.all([Fiber.await(a), Fiber.await(b)])
|
|
1241
|
+
expect(Exit.isSuccess(ea)).toBe(true)
|
|
1242
|
+
expect(Exit.isSuccess(eb)).toBe(true)
|
|
1243
|
+
expect(yield* llm.calls).toBe(2)
|
|
1244
|
+
|
|
1245
|
+
const msgs = yield* sessions.messages({ sessionID: chat.id })
|
|
1246
|
+
const assistants = msgs.filter((msg) => msg.info.role === "assistant")
|
|
1247
|
+
expect(assistants).toHaveLength(2)
|
|
1248
|
+
const last = assistants.at(-1)
|
|
1249
|
+
if (!last || last.info.role !== "assistant") throw new Error("expected second assistant")
|
|
1250
|
+
expect(last.info.parentID).toBe(id)
|
|
1251
|
+
expect(last.parts.some((part) => part.type === "text" && part.text === "second")).toBe(true)
|
|
1252
|
+
|
|
1253
|
+
const inputs = yield* llm.inputs
|
|
1254
|
+
expect(inputs).toHaveLength(2)
|
|
1255
|
+
expect(JSON.stringify(inputs.at(-1)?.messages)).toContain("second")
|
|
1256
|
+
}),
|
|
1257
|
+
3_000,
|
|
1258
|
+
)
|
|
1259
|
+
|
|
1260
|
+
it.instance(
|
|
1261
|
+
"assertNotBusy fails with BusyError when loop running",
|
|
1262
|
+
() =>
|
|
1263
|
+
Effect.gen(function* () {
|
|
1264
|
+
const { llm } = yield* useServerConfig(providerCfg)
|
|
1265
|
+
const prompt = yield* SessionPrompt.Service
|
|
1266
|
+
const run = yield* SessionRunState.Service
|
|
1267
|
+
const sessions = yield* Session.Service
|
|
1268
|
+
yield* llm.hang
|
|
1269
|
+
|
|
1270
|
+
const chat = yield* sessions.create({})
|
|
1271
|
+
yield* user(chat.id, "hi")
|
|
1272
|
+
|
|
1273
|
+
const fiber = yield* prompt.loop({ sessionID: chat.id }).pipe(Effect.forkChild)
|
|
1274
|
+
yield* llm.wait(1)
|
|
1275
|
+
|
|
1276
|
+
const exit = yield* run.assertNotBusy(chat.id).pipe(Effect.exit)
|
|
1277
|
+
expect(Exit.isFailure(exit)).toBe(true)
|
|
1278
|
+
if (Exit.isFailure(exit)) {
|
|
1279
|
+
expect(Cause.squash(exit.cause)).toBeInstanceOf(Session.BusyError)
|
|
1280
|
+
expect(Cause.squash(exit.cause)).toMatchObject({ _tag: "SessionBusyError", sessionID: chat.id })
|
|
1281
|
+
}
|
|
1282
|
+
|
|
1283
|
+
yield* prompt.cancel(chat.id)
|
|
1284
|
+
yield* Fiber.await(fiber)
|
|
1285
|
+
}),
|
|
1286
|
+
3_000,
|
|
1287
|
+
)
|
|
1288
|
+
|
|
1289
|
+
noLLMServer.instance("assertNotBusy succeeds when idle", () =>
|
|
1290
|
+
Effect.gen(function* () {
|
|
1291
|
+
const run = yield* SessionRunState.Service
|
|
1292
|
+
const sessions = yield* Session.Service
|
|
1293
|
+
|
|
1294
|
+
const chat = yield* sessions.create({})
|
|
1295
|
+
const exit = yield* run.assertNotBusy(chat.id).pipe(Effect.exit)
|
|
1296
|
+
expect(Exit.isSuccess(exit)).toBe(true)
|
|
1297
|
+
}),
|
|
1298
|
+
)
|
|
1299
|
+
|
|
1300
|
+
// Shell semantics
|
|
1301
|
+
|
|
1302
|
+
it.instance(
|
|
1303
|
+
"shell rejects with BusyError when loop running",
|
|
1304
|
+
() =>
|
|
1305
|
+
Effect.gen(function* () {
|
|
1306
|
+
const { llm } = yield* useServerConfig(providerCfg)
|
|
1307
|
+
const prompt = yield* SessionPrompt.Service
|
|
1308
|
+
const sessions = yield* Session.Service
|
|
1309
|
+
const chat = yield* sessions.create({ title: "Pinned" })
|
|
1310
|
+
yield* llm.hang
|
|
1311
|
+
yield* user(chat.id, "hi")
|
|
1312
|
+
|
|
1313
|
+
const fiber = yield* prompt.loop({ sessionID: chat.id }).pipe(Effect.forkChild)
|
|
1314
|
+
yield* llm.wait(1)
|
|
1315
|
+
|
|
1316
|
+
const exit = yield* prompt.shell({ sessionID: chat.id, agent: "build", command: "echo hi" }).pipe(Effect.exit)
|
|
1317
|
+
expect(Exit.isFailure(exit)).toBe(true)
|
|
1318
|
+
if (Exit.isFailure(exit)) {
|
|
1319
|
+
expect(Cause.squash(exit.cause)).toBeInstanceOf(Session.BusyError)
|
|
1320
|
+
expect(Cause.squash(exit.cause)).toMatchObject({ _tag: "SessionBusyError", sessionID: chat.id })
|
|
1321
|
+
}
|
|
1322
|
+
|
|
1323
|
+
yield* prompt.cancel(chat.id)
|
|
1324
|
+
yield* Fiber.await(fiber)
|
|
1325
|
+
}),
|
|
1326
|
+
3_000,
|
|
1327
|
+
)
|
|
1328
|
+
|
|
1329
|
+
unixNoLLMServer(
|
|
1330
|
+
"shell captures stdout and stderr in completed tool output",
|
|
1331
|
+
() =>
|
|
1332
|
+
Effect.gen(function* () {
|
|
1333
|
+
const { prompt, run, chat } = yield* boot()
|
|
1334
|
+
const result = yield* prompt.shell({
|
|
1335
|
+
sessionID: chat.id,
|
|
1336
|
+
agent: "build",
|
|
1337
|
+
command: "printf out && printf err >&2",
|
|
1338
|
+
})
|
|
1339
|
+
|
|
1340
|
+
expect(result.info.role).toBe("assistant")
|
|
1341
|
+
const tool = completedTool(result.parts)
|
|
1342
|
+
if (!tool) return
|
|
1343
|
+
|
|
1344
|
+
expect(tool.state.output).toContain("out")
|
|
1345
|
+
expect(tool.state.output).toContain("err")
|
|
1346
|
+
expect(tool.state.metadata.output).toContain("out")
|
|
1347
|
+
expect(tool.state.metadata.output).toContain("err")
|
|
1348
|
+
yield* run.assertNotBusy(chat.id)
|
|
1349
|
+
}),
|
|
1350
|
+
{ config: cfg },
|
|
1351
|
+
)
|
|
1352
|
+
|
|
1353
|
+
unixNoLLMServer(
|
|
1354
|
+
"shell completes a fast command on the preferred shell",
|
|
1355
|
+
() =>
|
|
1356
|
+
Effect.gen(function* () {
|
|
1357
|
+
const { directory: dir } = yield* TestInstance
|
|
1358
|
+
const { prompt, run, chat } = yield* boot()
|
|
1359
|
+
const result = yield* prompt.shell({
|
|
1360
|
+
sessionID: chat.id,
|
|
1361
|
+
agent: "build",
|
|
1362
|
+
command: "pwd",
|
|
1363
|
+
})
|
|
1364
|
+
|
|
1365
|
+
expect(result.info.role).toBe("assistant")
|
|
1366
|
+
const tool = completedTool(result.parts)
|
|
1367
|
+
if (!tool) return
|
|
1368
|
+
|
|
1369
|
+
expect(tool.state.input.command).toBe("pwd")
|
|
1370
|
+
expect(tool.state.output).toContain(dir)
|
|
1371
|
+
expect(tool.state.metadata.output).toContain(dir)
|
|
1372
|
+
yield* run.assertNotBusy(chat.id)
|
|
1373
|
+
}),
|
|
1374
|
+
{ config: cfg },
|
|
1375
|
+
)
|
|
1376
|
+
|
|
1377
|
+
unixNoLLMServer(
|
|
1378
|
+
"shell uses configured shell over env shell",
|
|
1379
|
+
() =>
|
|
1380
|
+
withSh(() =>
|
|
1381
|
+
Effect.gen(function* () {
|
|
1382
|
+
if (!(yield* hasBash)) return
|
|
1383
|
+
|
|
1384
|
+
const { prompt, chat } = yield* boot()
|
|
1385
|
+
const result = yield* prompt.shell({
|
|
1386
|
+
sessionID: chat.id,
|
|
1387
|
+
agent: "build",
|
|
1388
|
+
command: "[[ 1 -eq 1 ]] && printf configured",
|
|
1389
|
+
})
|
|
1390
|
+
|
|
1391
|
+
const tool = completedTool(result.parts)
|
|
1392
|
+
if (!tool) return
|
|
1393
|
+
expect(tool.state.output).toContain("configured")
|
|
1394
|
+
}),
|
|
1395
|
+
),
|
|
1396
|
+
{ config: { ...cfg, shell: "bash" } },
|
|
1397
|
+
30_000,
|
|
1398
|
+
)
|
|
1399
|
+
|
|
1400
|
+
unixNoLLMServer(
|
|
1401
|
+
"shell commands can change directory after startup",
|
|
1402
|
+
() =>
|
|
1403
|
+
withSh(() =>
|
|
1404
|
+
Effect.gen(function* () {
|
|
1405
|
+
const { directory: dir } = yield* TestInstance
|
|
1406
|
+
const { prompt, run, chat } = yield* boot()
|
|
1407
|
+
const parent = path.dirname(dir)
|
|
1408
|
+
const result = yield* prompt.shell({
|
|
1409
|
+
sessionID: chat.id,
|
|
1410
|
+
agent: "build",
|
|
1411
|
+
command: "cd .. && pwd",
|
|
1412
|
+
})
|
|
1413
|
+
|
|
1414
|
+
expect(result.info.role).toBe("assistant")
|
|
1415
|
+
const tool = completedTool(result.parts)
|
|
1416
|
+
if (!tool) return
|
|
1417
|
+
|
|
1418
|
+
expect(tool.state.output).toContain(parent)
|
|
1419
|
+
expect(tool.state.metadata.output).toContain(parent)
|
|
1420
|
+
yield* run.assertNotBusy(chat.id)
|
|
1421
|
+
}),
|
|
1422
|
+
),
|
|
1423
|
+
{ config: cfg },
|
|
1424
|
+
)
|
|
1425
|
+
|
|
1426
|
+
unixNoLLMServer(
|
|
1427
|
+
"shell lists files from the project directory",
|
|
1428
|
+
() =>
|
|
1429
|
+
Effect.gen(function* () {
|
|
1430
|
+
const { directory: dir } = yield* TestInstance
|
|
1431
|
+
const { prompt, run, chat } = yield* boot()
|
|
1432
|
+
yield* writeText(path.join(dir, "README.md"), "# e2e\n")
|
|
1433
|
+
|
|
1434
|
+
const result = yield* prompt.shell({
|
|
1435
|
+
sessionID: chat.id,
|
|
1436
|
+
agent: "build",
|
|
1437
|
+
command: "command ls",
|
|
1438
|
+
})
|
|
1439
|
+
|
|
1440
|
+
expect(result.info.role).toBe("assistant")
|
|
1441
|
+
const tool = completedTool(result.parts)
|
|
1442
|
+
if (!tool) return
|
|
1443
|
+
|
|
1444
|
+
expect(tool.state.input.command).toBe("command ls")
|
|
1445
|
+
expect(tool.state.output).toContain("README.md")
|
|
1446
|
+
expect(tool.state.metadata.output).toContain("README.md")
|
|
1447
|
+
yield* run.assertNotBusy(chat.id)
|
|
1448
|
+
}),
|
|
1449
|
+
{ config: cfg },
|
|
1450
|
+
)
|
|
1451
|
+
|
|
1452
|
+
unixNoLLMServer(
|
|
1453
|
+
"shell captures stderr from a failing command",
|
|
1454
|
+
() =>
|
|
1455
|
+
Effect.gen(function* () {
|
|
1456
|
+
const { prompt, run, chat } = yield* boot()
|
|
1457
|
+
const result = yield* prompt.shell({
|
|
1458
|
+
sessionID: chat.id,
|
|
1459
|
+
agent: "build",
|
|
1460
|
+
command: "command -v __nonexistent_cmd_e2e__ || echo 'not found' >&2; exit 1",
|
|
1461
|
+
})
|
|
1462
|
+
|
|
1463
|
+
expect(result.info.role).toBe("assistant")
|
|
1464
|
+
const tool = completedTool(result.parts)
|
|
1465
|
+
if (!tool) return
|
|
1466
|
+
|
|
1467
|
+
expect(tool.state.output).toContain("not found")
|
|
1468
|
+
expect(tool.state.metadata.output).toContain("not found")
|
|
1469
|
+
yield* run.assertNotBusy(chat.id)
|
|
1470
|
+
}),
|
|
1471
|
+
{ config: cfg },
|
|
1472
|
+
)
|
|
1473
|
+
|
|
1474
|
+
unixNoLLMServer(
|
|
1475
|
+
"shell updates running metadata before process exit",
|
|
1476
|
+
() =>
|
|
1477
|
+
withSh(() =>
|
|
1478
|
+
Effect.gen(function* () {
|
|
1479
|
+
const { prompt, chat } = yield* boot()
|
|
1480
|
+
|
|
1481
|
+
const fiber = yield* prompt
|
|
1482
|
+
.shell({ sessionID: chat.id, agent: "build", command: "printf first && sleep 0.2 && printf second" })
|
|
1483
|
+
.pipe(Effect.forkChild)
|
|
1484
|
+
|
|
1485
|
+
yield* pollWithTimeout(
|
|
1486
|
+
Effect.gen(function* () {
|
|
1487
|
+
const msgs = yield* MessageV2.filterCompactedEffect(chat.id)
|
|
1488
|
+
const taskMsg = msgs.find((item) => item.info.role === "assistant")
|
|
1489
|
+
const tool = taskMsg ? toolPart(taskMsg.parts) : undefined
|
|
1490
|
+
if (tool?.state.status === "running" && tool.state.metadata?.output.includes("first")) return true
|
|
1491
|
+
}),
|
|
1492
|
+
"timed out waiting for running shell metadata",
|
|
1493
|
+
)
|
|
1494
|
+
|
|
1495
|
+
const exit = yield* Fiber.await(fiber)
|
|
1496
|
+
expect(Exit.isSuccess(exit)).toBe(true)
|
|
1497
|
+
}),
|
|
1498
|
+
),
|
|
1499
|
+
{ config: cfg },
|
|
1500
|
+
30_000,
|
|
1501
|
+
)
|
|
1502
|
+
|
|
1503
|
+
it.instance(
|
|
1504
|
+
"loop waits while shell runs and starts after shell exits",
|
|
1505
|
+
() =>
|
|
1506
|
+
Effect.gen(function* () {
|
|
1507
|
+
const { llm } = yield* useServerConfig(providerCfg)
|
|
1508
|
+
const prompt = yield* SessionPrompt.Service
|
|
1509
|
+
const sessions = yield* Session.Service
|
|
1510
|
+
const chat = yield* sessions.create({
|
|
1511
|
+
title: "Pinned",
|
|
1512
|
+
permission: [{ permission: "*", pattern: "*", action: "allow" }],
|
|
1513
|
+
})
|
|
1514
|
+
yield* llm.text("after-shell")
|
|
1515
|
+
|
|
1516
|
+
const sh = yield* prompt
|
|
1517
|
+
.shell({ sessionID: chat.id, agent: "build", command: "sleep 0.2" })
|
|
1518
|
+
.pipe(Effect.forkChild)
|
|
1519
|
+
yield* waitForBusy(chat.id)
|
|
1520
|
+
|
|
1521
|
+
const loop = yield* prompt.loop({ sessionID: chat.id }).pipe(Effect.forkChild)
|
|
1522
|
+
yield* Effect.sleep(50)
|
|
1523
|
+
|
|
1524
|
+
expect(yield* llm.calls).toBe(0)
|
|
1525
|
+
|
|
1526
|
+
yield* Fiber.await(sh)
|
|
1527
|
+
const exit = yield* Fiber.await(loop)
|
|
1528
|
+
|
|
1529
|
+
expect(Exit.isSuccess(exit)).toBe(true)
|
|
1530
|
+
if (Exit.isSuccess(exit)) {
|
|
1531
|
+
expect(exit.value.info.role).toBe("assistant")
|
|
1532
|
+
expect(exit.value.parts.some((part) => part.type === "text" && part.text === "after-shell")).toBe(true)
|
|
1533
|
+
}
|
|
1534
|
+
expect(yield* llm.calls).toBe(1)
|
|
1535
|
+
}),
|
|
1536
|
+
{ git: true },
|
|
1537
|
+
3_000,
|
|
1538
|
+
)
|
|
1539
|
+
|
|
1540
|
+
it.instance(
|
|
1541
|
+
"shell completion resumes queued loop callers",
|
|
1542
|
+
() =>
|
|
1543
|
+
Effect.gen(function* () {
|
|
1544
|
+
const { llm } = yield* useServerConfig(providerCfg)
|
|
1545
|
+
const prompt = yield* SessionPrompt.Service
|
|
1546
|
+
const sessions = yield* Session.Service
|
|
1547
|
+
const chat = yield* sessions.create({
|
|
1548
|
+
title: "Pinned",
|
|
1549
|
+
permission: [{ permission: "*", pattern: "*", action: "allow" }],
|
|
1550
|
+
})
|
|
1551
|
+
yield* llm.text("done")
|
|
1552
|
+
|
|
1553
|
+
const sh = yield* prompt
|
|
1554
|
+
.shell({ sessionID: chat.id, agent: "build", command: "sleep 0.2" })
|
|
1555
|
+
.pipe(Effect.forkChild)
|
|
1556
|
+
yield* waitForBusy(chat.id)
|
|
1557
|
+
|
|
1558
|
+
const a = yield* prompt.loop({ sessionID: chat.id }).pipe(Effect.forkChild)
|
|
1559
|
+
const b = yield* prompt.loop({ sessionID: chat.id }).pipe(Effect.forkChild)
|
|
1560
|
+
yield* Effect.sleep(50)
|
|
1561
|
+
|
|
1562
|
+
expect(yield* llm.calls).toBe(0)
|
|
1563
|
+
|
|
1564
|
+
yield* Fiber.await(sh)
|
|
1565
|
+
const [ea, eb] = yield* Effect.all([Fiber.await(a), Fiber.await(b)])
|
|
1566
|
+
|
|
1567
|
+
expect(Exit.isSuccess(ea)).toBe(true)
|
|
1568
|
+
expect(Exit.isSuccess(eb)).toBe(true)
|
|
1569
|
+
if (Exit.isSuccess(ea) && Exit.isSuccess(eb)) {
|
|
1570
|
+
expect(ea.value.info.id).toBe(eb.value.info.id)
|
|
1571
|
+
expect(ea.value.info.role).toBe("assistant")
|
|
1572
|
+
}
|
|
1573
|
+
expect(yield* llm.calls).toBe(1)
|
|
1574
|
+
}),
|
|
1575
|
+
{ git: true },
|
|
1576
|
+
3_000,
|
|
1577
|
+
)
|
|
1578
|
+
|
|
1579
|
+
unix(
|
|
1580
|
+
"command ! expansion uses configured shell over env shell",
|
|
1581
|
+
() =>
|
|
1582
|
+
withSh(() =>
|
|
1583
|
+
Effect.gen(function* () {
|
|
1584
|
+
if (!(yield* hasBash)) return
|
|
1585
|
+
const { llm } = yield* useServerConfig((url) => ({
|
|
1586
|
+
...providerCfg(url),
|
|
1587
|
+
shell: "bash",
|
|
1588
|
+
command: {
|
|
1589
|
+
probe: {
|
|
1590
|
+
template: "Probe: !`[[ 1 -eq 1 ]] && printf configured`",
|
|
1591
|
+
},
|
|
1592
|
+
},
|
|
1593
|
+
}))
|
|
1594
|
+
|
|
1595
|
+
const { prompt, chat } = yield* boot()
|
|
1596
|
+
yield* llm.text("done")
|
|
1597
|
+
|
|
1598
|
+
const result = yield* prompt.command({
|
|
1599
|
+
sessionID: chat.id,
|
|
1600
|
+
command: "probe",
|
|
1601
|
+
arguments: "",
|
|
1602
|
+
})
|
|
1603
|
+
|
|
1604
|
+
expect(result.info.role).toBe("assistant")
|
|
1605
|
+
const inputs = yield* llm.inputs
|
|
1606
|
+
expect(JSON.stringify(inputs.at(-1)?.messages)).toContain("configured")
|
|
1607
|
+
}),
|
|
1608
|
+
),
|
|
1609
|
+
30_000,
|
|
1610
|
+
)
|
|
1611
|
+
|
|
1612
|
+
unixNoLLMServer(
|
|
1613
|
+
"cancel interrupts shell and resolves cleanly",
|
|
1614
|
+
() =>
|
|
1615
|
+
withSh(() =>
|
|
1616
|
+
Effect.gen(function* () {
|
|
1617
|
+
const { prompt, run, chat } = yield* boot()
|
|
1618
|
+
|
|
1619
|
+
const sh = yield* prompt
|
|
1620
|
+
.shell({ sessionID: chat.id, agent: "build", command: "sleep 30" })
|
|
1621
|
+
.pipe(Effect.forkChild)
|
|
1622
|
+
yield* waitForBusy(chat.id)
|
|
1623
|
+
|
|
1624
|
+
yield* prompt.cancel(chat.id)
|
|
1625
|
+
|
|
1626
|
+
const status = yield* SessionStatus.Service
|
|
1627
|
+
expect((yield* status.get(chat.id)).type).toBe("idle")
|
|
1628
|
+
const busy = yield* run.assertNotBusy(chat.id).pipe(Effect.exit)
|
|
1629
|
+
expect(Exit.isSuccess(busy)).toBe(true)
|
|
1630
|
+
|
|
1631
|
+
const exit = yield* Fiber.await(sh)
|
|
1632
|
+
expect(Exit.isSuccess(exit)).toBe(true)
|
|
1633
|
+
if (Exit.isSuccess(exit)) {
|
|
1634
|
+
expect(exit.value.info.role).toBe("assistant")
|
|
1635
|
+
const tool = completedTool(exit.value.parts)
|
|
1636
|
+
if (tool) {
|
|
1637
|
+
expect(tool.state.output).toContain("User aborted the command")
|
|
1638
|
+
}
|
|
1639
|
+
}
|
|
1640
|
+
}),
|
|
1641
|
+
),
|
|
1642
|
+
{ git: true, config: cfg },
|
|
1643
|
+
30_000,
|
|
1644
|
+
)
|
|
1645
|
+
|
|
1646
|
+
unixNoLLMServer(
|
|
1647
|
+
"cancel persists aborted shell result when shell ignores TERM",
|
|
1648
|
+
() =>
|
|
1649
|
+
withSh(() =>
|
|
1650
|
+
Effect.gen(function* () {
|
|
1651
|
+
const { prompt, chat } = yield* boot()
|
|
1652
|
+
const { directory: dir } = yield* TestInstance
|
|
1653
|
+
const afs = yield* FSUtil.Service
|
|
1654
|
+
const ready = path.join(dir, ".trap-ready")
|
|
1655
|
+
|
|
1656
|
+
const sh = yield* prompt
|
|
1657
|
+
.shell({
|
|
1658
|
+
sessionID: chat.id,
|
|
1659
|
+
agent: "build",
|
|
1660
|
+
// Touch marker AFTER trap installs so the test waits for the actual
|
|
1661
|
+
// ignore-TERM state before cancelling; otherwise SIGTERM can arrive
|
|
1662
|
+
// before `trap` runs and the escalation path is never exercised.
|
|
1663
|
+
command: `trap '' TERM; touch "${ready}"; sleep 30`,
|
|
1664
|
+
})
|
|
1665
|
+
.pipe(Effect.forkChild)
|
|
1666
|
+
|
|
1667
|
+
yield* Effect.gen(function* () {
|
|
1668
|
+
while (!(yield* afs.existsSafe(ready))) {
|
|
1669
|
+
yield* Effect.sleep(Duration.millis(10))
|
|
1670
|
+
}
|
|
1671
|
+
}).pipe(Effect.timeout(Duration.seconds(5)))
|
|
1672
|
+
|
|
1673
|
+
yield* prompt.cancel(chat.id)
|
|
1674
|
+
|
|
1675
|
+
const exit = yield* Fiber.await(sh)
|
|
1676
|
+
expect(Exit.isSuccess(exit)).toBe(true)
|
|
1677
|
+
if (Exit.isSuccess(exit)) {
|
|
1678
|
+
expect(exit.value.info.role).toBe("assistant")
|
|
1679
|
+
const tool = completedTool(exit.value.parts)
|
|
1680
|
+
if (tool) {
|
|
1681
|
+
expect(tool.state.output).toContain("User aborted the command")
|
|
1682
|
+
}
|
|
1683
|
+
}
|
|
1684
|
+
}),
|
|
1685
|
+
),
|
|
1686
|
+
{ git: true, config: cfg },
|
|
1687
|
+
30_000,
|
|
1688
|
+
)
|
|
1689
|
+
|
|
1690
|
+
unix(
|
|
1691
|
+
"cancel finalizes interrupted bash tool output through normal truncation",
|
|
1692
|
+
() =>
|
|
1693
|
+
Effect.gen(function* () {
|
|
1694
|
+
const { dir, llm } = yield* useServerConfig(providerCfg)
|
|
1695
|
+
const prompt = yield* SessionPrompt.Service
|
|
1696
|
+
const sessions = yield* Session.Service
|
|
1697
|
+
const chat = yield* sessions.create({
|
|
1698
|
+
title: "Interrupted bash truncation",
|
|
1699
|
+
permission: [{ permission: "*", pattern: "*", action: "allow" }],
|
|
1700
|
+
})
|
|
1701
|
+
|
|
1702
|
+
yield* prompt.prompt({
|
|
1703
|
+
sessionID: chat.id,
|
|
1704
|
+
agent: "build",
|
|
1705
|
+
noReply: true,
|
|
1706
|
+
parts: [{ type: "text", text: "run bash" }],
|
|
1707
|
+
})
|
|
1708
|
+
|
|
1709
|
+
yield* llm.tool("bash", {
|
|
1710
|
+
command:
|
|
1711
|
+
'i=0; while [ "$i" -lt 4000 ]; do printf "xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx %05d\\n" "$i"; i=$((i + 1)); done; printf truncation-ready; sleep 30',
|
|
1712
|
+
description: "Print many lines",
|
|
1713
|
+
timeout: 30_000,
|
|
1714
|
+
workdir: path.resolve(dir),
|
|
1715
|
+
})
|
|
1716
|
+
|
|
1717
|
+
const run = yield* prompt.loop({ sessionID: chat.id }).pipe(Effect.forkChild)
|
|
1718
|
+
yield* llm.wait(1)
|
|
1719
|
+
yield* pollWithTimeout(
|
|
1720
|
+
Effect.gen(function* () {
|
|
1721
|
+
const msgs = yield* MessageV2.filterCompactedEffect(chat.id)
|
|
1722
|
+
const assistant = msgs.findLast((item) => item.info.role === "assistant")
|
|
1723
|
+
const tool = assistant ? toolPart(assistant.parts) : undefined
|
|
1724
|
+
if (tool?.state.status === "running" && tool.state.metadata?.output.includes("truncation-ready")) return true
|
|
1725
|
+
}),
|
|
1726
|
+
"timed out waiting for truncated shell output",
|
|
1727
|
+
)
|
|
1728
|
+
yield* prompt.cancel(chat.id)
|
|
1729
|
+
|
|
1730
|
+
const exit = yield* Fiber.await(run)
|
|
1731
|
+
expect(Exit.isSuccess(exit)).toBe(true)
|
|
1732
|
+
if (Exit.isFailure(exit)) return
|
|
1733
|
+
|
|
1734
|
+
const tool = completedTool(exit.value.parts)
|
|
1735
|
+
if (!tool) return
|
|
1736
|
+
|
|
1737
|
+
expect(tool.state.metadata.truncated).toBe(true)
|
|
1738
|
+
expect(typeof tool.state.metadata.outputPath).toBe("string")
|
|
1739
|
+
expect(tool.state.output).toMatch(/\.\.\.output truncated\.\.\./)
|
|
1740
|
+
expect(tool.state.output).toMatch(/Full output saved to:\s+\S+/)
|
|
1741
|
+
expect(tool.state.output).not.toContain("Tool execution aborted")
|
|
1742
|
+
}),
|
|
1743
|
+
{ git: true },
|
|
1744
|
+
30_000,
|
|
1745
|
+
)
|
|
1746
|
+
|
|
1747
|
+
unixNoLLMServer(
|
|
1748
|
+
"cancel interrupts loop queued behind shell",
|
|
1749
|
+
() =>
|
|
1750
|
+
Effect.gen(function* () {
|
|
1751
|
+
const { prompt, chat } = yield* boot()
|
|
1752
|
+
|
|
1753
|
+
const sh = yield* prompt.shell({ sessionID: chat.id, agent: "build", command: "sleep 30" }).pipe(Effect.forkChild)
|
|
1754
|
+
yield* waitForBusy(chat.id)
|
|
1755
|
+
|
|
1756
|
+
const loop = yield* prompt.loop({ sessionID: chat.id }).pipe(Effect.forkChild)
|
|
1757
|
+
yield* Effect.sleep(50)
|
|
1758
|
+
|
|
1759
|
+
yield* prompt.cancel(chat.id)
|
|
1760
|
+
|
|
1761
|
+
const exit = yield* Fiber.await(loop)
|
|
1762
|
+
expect(Exit.isSuccess(exit)).toBe(true)
|
|
1763
|
+
if (Exit.isSuccess(exit)) {
|
|
1764
|
+
const tool = completedTool(exit.value.parts)
|
|
1765
|
+
expect(tool?.state.output).toContain("User aborted the command")
|
|
1766
|
+
}
|
|
1767
|
+
|
|
1768
|
+
yield* Fiber.await(sh)
|
|
1769
|
+
}),
|
|
1770
|
+
{ git: true, config: cfg },
|
|
1771
|
+
30_000,
|
|
1772
|
+
)
|
|
1773
|
+
|
|
1774
|
+
unixNoLLMServer(
|
|
1775
|
+
"shell rejects when another shell is already running",
|
|
1776
|
+
() =>
|
|
1777
|
+
withSh(() =>
|
|
1778
|
+
Effect.gen(function* () {
|
|
1779
|
+
const { prompt, chat } = yield* boot()
|
|
1780
|
+
|
|
1781
|
+
const a = yield* prompt
|
|
1782
|
+
.shell({ sessionID: chat.id, agent: "build", command: "sleep 30" })
|
|
1783
|
+
.pipe(Effect.forkChild)
|
|
1784
|
+
yield* waitForBusy(chat.id)
|
|
1785
|
+
|
|
1786
|
+
const exit = yield* prompt.shell({ sessionID: chat.id, agent: "build", command: "echo hi" }).pipe(Effect.exit)
|
|
1787
|
+
expect(Exit.isFailure(exit)).toBe(true)
|
|
1788
|
+
if (Exit.isFailure(exit)) {
|
|
1789
|
+
expect(Cause.squash(exit.cause)).toBeInstanceOf(Session.BusyError)
|
|
1790
|
+
}
|
|
1791
|
+
|
|
1792
|
+
yield* prompt.cancel(chat.id)
|
|
1793
|
+
yield* Fiber.await(a)
|
|
1794
|
+
}),
|
|
1795
|
+
),
|
|
1796
|
+
{ git: true, config: cfg },
|
|
1797
|
+
30_000,
|
|
1798
|
+
)
|
|
1799
|
+
|
|
1800
|
+
// Abort signal propagation tests for inline tool execution
|
|
1801
|
+
|
|
1802
|
+
function hangUntilAborted(tool: { execute: (...args: any[]) => any }) {
|
|
1803
|
+
return Effect.gen(function* () {
|
|
1804
|
+
const ready = yield* Deferred.make<void>()
|
|
1805
|
+
const aborted = yield* Deferred.make<void>()
|
|
1806
|
+
const original = tool.execute
|
|
1807
|
+
tool.execute = (_args: any, ctx: any) => {
|
|
1808
|
+
ctx.abort.addEventListener("abort", () => succeedVoid(aborted), { once: true })
|
|
1809
|
+
if (ctx.abort.aborted) succeedVoid(aborted)
|
|
1810
|
+
succeedVoid(ready)
|
|
1811
|
+
return Effect.callback<never>(() => Effect.sync(() => succeedVoid(aborted)))
|
|
1812
|
+
}
|
|
1813
|
+
const restore = Effect.addFinalizer(() => Effect.sync(() => void (tool.execute = original)))
|
|
1814
|
+
return { ready, aborted, restore }
|
|
1815
|
+
})
|
|
1816
|
+
}
|
|
1817
|
+
|
|
1818
|
+
noLLMServer.instance(
|
|
1819
|
+
"interrupt propagates abort signal to read tool via file part (text/plain)",
|
|
1820
|
+
() =>
|
|
1821
|
+
Effect.gen(function* () {
|
|
1822
|
+
const { directory: dir } = yield* TestInstance
|
|
1823
|
+
const registry = yield* ToolRegistry.Service
|
|
1824
|
+
const { read } = yield* registry.named()
|
|
1825
|
+
const { ready, restore } = yield* hangUntilAborted(read)
|
|
1826
|
+
yield* restore
|
|
1827
|
+
|
|
1828
|
+
const prompt = yield* SessionPrompt.Service
|
|
1829
|
+
const sessions = yield* Session.Service
|
|
1830
|
+
const chat = yield* sessions.create({ title: "Abort Test" })
|
|
1831
|
+
|
|
1832
|
+
const testFile = path.join(dir, "test.txt")
|
|
1833
|
+
yield* writeText(testFile, "hello world")
|
|
1834
|
+
|
|
1835
|
+
const fiber = yield* prompt
|
|
1836
|
+
.prompt({
|
|
1837
|
+
sessionID: chat.id,
|
|
1838
|
+
agent: "build",
|
|
1839
|
+
parts: [
|
|
1840
|
+
{ type: "text", text: "read this" },
|
|
1841
|
+
{ type: "file", url: `file://${testFile}`, filename: "test.txt", mime: "text/plain" },
|
|
1842
|
+
],
|
|
1843
|
+
})
|
|
1844
|
+
.pipe(Effect.forkChild)
|
|
1845
|
+
|
|
1846
|
+
yield* awaitWithTimeout(Deferred.await(ready), "timed out waiting for read tool to start", "10 seconds")
|
|
1847
|
+
yield* prompt.cancel(chat.id)
|
|
1848
|
+
yield* Fiber.interrupt(fiber)
|
|
1849
|
+
const exit = yield* Fiber.await(fiber)
|
|
1850
|
+
expect(Exit.isFailure(exit)).toBe(true)
|
|
1851
|
+
}),
|
|
1852
|
+
{ config: cfg },
|
|
1853
|
+
30_000,
|
|
1854
|
+
)
|
|
1855
|
+
|
|
1856
|
+
noLLMServer.instance(
|
|
1857
|
+
"interrupt propagates abort signal to read tool via file part (directory)",
|
|
1858
|
+
() =>
|
|
1859
|
+
Effect.gen(function* () {
|
|
1860
|
+
const { directory: dir } = yield* TestInstance
|
|
1861
|
+
const registry = yield* ToolRegistry.Service
|
|
1862
|
+
const { read } = yield* registry.named()
|
|
1863
|
+
const { ready, restore } = yield* hangUntilAborted(read)
|
|
1864
|
+
yield* restore
|
|
1865
|
+
|
|
1866
|
+
const prompt = yield* SessionPrompt.Service
|
|
1867
|
+
const sessions = yield* Session.Service
|
|
1868
|
+
const chat = yield* sessions.create({ title: "Abort Test" })
|
|
1869
|
+
|
|
1870
|
+
const fiber = yield* prompt
|
|
1871
|
+
.prompt({
|
|
1872
|
+
sessionID: chat.id,
|
|
1873
|
+
agent: "build",
|
|
1874
|
+
parts: [
|
|
1875
|
+
{ type: "text", text: "read this" },
|
|
1876
|
+
{ type: "file", url: `file://${dir}`, filename: "dir", mime: "application/x-directory" },
|
|
1877
|
+
],
|
|
1878
|
+
})
|
|
1879
|
+
.pipe(Effect.forkChild)
|
|
1880
|
+
|
|
1881
|
+
yield* awaitWithTimeout(Deferred.await(ready), "timed out waiting for read tool to start", "10 seconds")
|
|
1882
|
+
yield* prompt.cancel(chat.id)
|
|
1883
|
+
yield* Fiber.interrupt(fiber)
|
|
1884
|
+
const exit = yield* Fiber.await(fiber)
|
|
1885
|
+
expect(Exit.isFailure(exit)).toBe(true)
|
|
1886
|
+
}),
|
|
1887
|
+
{ config: cfg },
|
|
1888
|
+
30_000,
|
|
1889
|
+
)
|
|
1890
|
+
|
|
1891
|
+
// Missing file handling
|
|
1892
|
+
|
|
1893
|
+
noLLMServer.instance(
|
|
1894
|
+
"does not fail the prompt when a file part is missing",
|
|
1895
|
+
() =>
|
|
1896
|
+
Effect.gen(function* () {
|
|
1897
|
+
const { directory: dir } = yield* TestInstance
|
|
1898
|
+
const prompt = yield* SessionPrompt.Service
|
|
1899
|
+
const sessions = yield* Session.Service
|
|
1900
|
+
const session = yield* sessions.create({})
|
|
1901
|
+
|
|
1902
|
+
const missing = path.join(dir, "does-not-exist.ts")
|
|
1903
|
+
const msg = yield* prompt.prompt({
|
|
1904
|
+
sessionID: session.id,
|
|
1905
|
+
agent: "build",
|
|
1906
|
+
noReply: true,
|
|
1907
|
+
parts: [
|
|
1908
|
+
{ type: "text", text: "please review @does-not-exist.ts" },
|
|
1909
|
+
{
|
|
1910
|
+
type: "file",
|
|
1911
|
+
mime: "text/plain",
|
|
1912
|
+
url: `file://${missing}`,
|
|
1913
|
+
filename: "does-not-exist.ts",
|
|
1914
|
+
},
|
|
1915
|
+
],
|
|
1916
|
+
})
|
|
1917
|
+
|
|
1918
|
+
if (msg.info.role !== "user") throw new Error("expected user message")
|
|
1919
|
+
const hasFailure = msg.parts.some(
|
|
1920
|
+
(part) => part.type === "text" && part.synthetic && part.text.includes("Read tool failed to read"),
|
|
1921
|
+
)
|
|
1922
|
+
expect(hasFailure).toBe(true)
|
|
1923
|
+
|
|
1924
|
+
yield* sessions.remove(session.id)
|
|
1925
|
+
}),
|
|
1926
|
+
{ config: cfg },
|
|
1927
|
+
)
|
|
1928
|
+
|
|
1929
|
+
noLLMServer.instance(
|
|
1930
|
+
"keeps stored part order stable when file resolution is async",
|
|
1931
|
+
() =>
|
|
1932
|
+
Effect.gen(function* () {
|
|
1933
|
+
const { directory: dir } = yield* TestInstance
|
|
1934
|
+
const prompt = yield* SessionPrompt.Service
|
|
1935
|
+
const sessions = yield* Session.Service
|
|
1936
|
+
const session = yield* sessions.create({})
|
|
1937
|
+
|
|
1938
|
+
const missing = path.join(dir, "still-missing.ts")
|
|
1939
|
+
const msg = yield* prompt.prompt({
|
|
1940
|
+
sessionID: session.id,
|
|
1941
|
+
agent: "build",
|
|
1942
|
+
noReply: true,
|
|
1943
|
+
parts: [
|
|
1944
|
+
{
|
|
1945
|
+
type: "file",
|
|
1946
|
+
mime: "text/plain",
|
|
1947
|
+
url: `file://${missing}`,
|
|
1948
|
+
filename: "still-missing.ts",
|
|
1949
|
+
},
|
|
1950
|
+
{ type: "text", text: "after-file" },
|
|
1951
|
+
],
|
|
1952
|
+
})
|
|
1953
|
+
|
|
1954
|
+
if (msg.info.role !== "user") throw new Error("expected user message")
|
|
1955
|
+
|
|
1956
|
+
const stored = yield* MessageV2.get({
|
|
1957
|
+
sessionID: session.id,
|
|
1958
|
+
messageID: msg.info.id,
|
|
1959
|
+
})
|
|
1960
|
+
const text = stored.parts.filter((part) => part.type === "text").map((part) => part.text)
|
|
1961
|
+
|
|
1962
|
+
expect(text[0]?.startsWith("Called the Read tool with the following input:")).toBe(true)
|
|
1963
|
+
expect(text[1]?.includes("Read tool failed to read")).toBe(true)
|
|
1964
|
+
expect(text[2]).toBe("after-file")
|
|
1965
|
+
|
|
1966
|
+
yield* sessions.remove(session.id)
|
|
1967
|
+
}),
|
|
1968
|
+
{ config: cfg },
|
|
1969
|
+
)
|
|
1970
|
+
|
|
1971
|
+
noLLMServer.instance(
|
|
1972
|
+
"resolves configured reference mentions to one root directory attachment",
|
|
1973
|
+
() =>
|
|
1974
|
+
Effect.gen(function* () {
|
|
1975
|
+
const { directory: dir } = yield* TestInstance
|
|
1976
|
+
const docs = path.join(dir, "external-docs")
|
|
1977
|
+
yield* ensureDir(path.join(docs, "guide"))
|
|
1978
|
+
yield* ensureDir(path.join(dir, "docs"))
|
|
1979
|
+
yield* writeText(path.join(docs, "README.md"), "reference readme")
|
|
1980
|
+
yield* writeText(path.join(docs, "guide", "intro.md"), "reference intro")
|
|
1981
|
+
yield* writeText(path.join(dir, "docs", "README.md"), "workspace readme")
|
|
1982
|
+
|
|
1983
|
+
const prompt = yield* SessionPrompt.Service
|
|
1984
|
+
const parts = yield* prompt.resolvePromptParts(
|
|
1985
|
+
"Use @docs and @docs/README.md and @docs/guide and @docs/missing.md and @docs/README.md and @build",
|
|
1986
|
+
)
|
|
1987
|
+
const files = parts.filter((part): part is SessionV1.FilePartInput => part.type === "file")
|
|
1988
|
+
const agents = parts.filter((part): part is SessionV1.AgentPartInput => part.type === "agent")
|
|
1989
|
+
const text = parts.find((part): part is SessionV1.TextPartInput => part.type === "text" && !part.synthetic)
|
|
1990
|
+
|
|
1991
|
+
expect(text?.text).toContain("@docs")
|
|
1992
|
+
expect(files).toHaveLength(1)
|
|
1993
|
+
expect(files[0]).toMatchObject({
|
|
1994
|
+
filename: "docs",
|
|
1995
|
+
mime: "application/x-directory",
|
|
1996
|
+
source: { type: "file", path: "docs", text: { value: "@docs" } },
|
|
1997
|
+
})
|
|
1998
|
+
expect(fileURLToPath(files[0].url)).toBe(docs)
|
|
1999
|
+
expect(agents.map((agent) => agent.name)).toEqual(["build"])
|
|
2000
|
+
}),
|
|
2001
|
+
{
|
|
2002
|
+
config: {
|
|
2003
|
+
...cfg,
|
|
2004
|
+
reference: {
|
|
2005
|
+
docs: "./external-docs",
|
|
2006
|
+
},
|
|
2007
|
+
},
|
|
2008
|
+
},
|
|
2009
|
+
)
|
|
2010
|
+
|
|
2011
|
+
noLLMServer.instance(
|
|
2012
|
+
"stores raw reference mentions alongside directory attachments",
|
|
2013
|
+
() =>
|
|
2014
|
+
Effect.gen(function* () {
|
|
2015
|
+
const { directory: dir } = yield* TestInstance
|
|
2016
|
+
const docs = path.join(dir, "external-docs")
|
|
2017
|
+
yield* ensureDir(docs)
|
|
2018
|
+
|
|
2019
|
+
const prompt = yield* SessionPrompt.Service
|
|
2020
|
+
const sessions = yield* Session.Service
|
|
2021
|
+
const session = yield* sessions.create({})
|
|
2022
|
+
const message = yield* prompt.prompt({
|
|
2023
|
+
sessionID: session.id,
|
|
2024
|
+
noReply: true,
|
|
2025
|
+
parts: [{ type: "text", text: "Use @docs for context" }],
|
|
2026
|
+
})
|
|
2027
|
+
|
|
2028
|
+
const stored = yield* MessageV2.get({ sessionID: session.id, messageID: message.info.id })
|
|
2029
|
+
const synthetic = stored.parts.filter(
|
|
2030
|
+
(part): part is SessionV1.TextPart => part.type === "text" && part.synthetic === true,
|
|
2031
|
+
)
|
|
2032
|
+
const files = stored.parts.filter((part): part is SessionV1.FilePart => part.type === "file")
|
|
2033
|
+
const text = stored.parts.find((part): part is SessionV1.TextPart => part.type === "text" && !part.synthetic)
|
|
2034
|
+
|
|
2035
|
+
expect(text?.text).toBe("Use @docs for context")
|
|
2036
|
+
expect(synthetic.some((part) => part.text.includes(JSON.stringify({ filePath: docs })))).toBe(true)
|
|
2037
|
+
expect(files).toHaveLength(1)
|
|
2038
|
+
expect(files[0]).toMatchObject({
|
|
2039
|
+
filename: "docs",
|
|
2040
|
+
mime: "application/x-directory",
|
|
2041
|
+
source: { type: "file", path: "docs", text: { value: "@docs", start: 4, end: 9 } },
|
|
2042
|
+
})
|
|
2043
|
+
expect(fileURLToPath(files[0].url)).toBe(docs)
|
|
2044
|
+
|
|
2045
|
+
yield* sessions.remove(session.id)
|
|
2046
|
+
}),
|
|
2047
|
+
{
|
|
2048
|
+
config: {
|
|
2049
|
+
...cfg,
|
|
2050
|
+
reference: {
|
|
2051
|
+
docs: "./external-docs",
|
|
2052
|
+
},
|
|
2053
|
+
},
|
|
2054
|
+
},
|
|
2055
|
+
)
|
|
2056
|
+
|
|
2057
|
+
// Special characters in filenames
|
|
2058
|
+
|
|
2059
|
+
noLLMServer.instance(
|
|
2060
|
+
"handles filenames with # character",
|
|
2061
|
+
() =>
|
|
2062
|
+
Effect.gen(function* () {
|
|
2063
|
+
const { directory: dir } = yield* TestInstance
|
|
2064
|
+
yield* writeText(path.join(dir, "file#name.txt"), "special content\n")
|
|
2065
|
+
|
|
2066
|
+
const prompt = yield* SessionPrompt.Service
|
|
2067
|
+
const sessions = yield* Session.Service
|
|
2068
|
+
const session = yield* sessions.create({})
|
|
2069
|
+
const parts = yield* prompt.resolvePromptParts("Read @file#name.txt")
|
|
2070
|
+
const fileParts = parts.filter((part) => part.type === "file")
|
|
2071
|
+
|
|
2072
|
+
expect(fileParts.length).toBe(1)
|
|
2073
|
+
expect(fileParts[0].filename).toBe("file#name.txt")
|
|
2074
|
+
expect(fileParts[0].url).toContain("%23")
|
|
2075
|
+
|
|
2076
|
+
const decodedPath = fileURLToPath(fileParts[0].url)
|
|
2077
|
+
expect(decodedPath).toBe(path.join(dir, "file#name.txt"))
|
|
2078
|
+
|
|
2079
|
+
const message = yield* prompt.prompt({
|
|
2080
|
+
sessionID: session.id,
|
|
2081
|
+
parts,
|
|
2082
|
+
noReply: true,
|
|
2083
|
+
})
|
|
2084
|
+
const stored = yield* MessageV2.get({ sessionID: session.id, messageID: message.info.id })
|
|
2085
|
+
const textParts = stored.parts.filter((part) => part.type === "text")
|
|
2086
|
+
const hasContent = textParts.some((part) => part.text.includes("special content"))
|
|
2087
|
+
expect(hasContent).toBe(true)
|
|
2088
|
+
|
|
2089
|
+
yield* sessions.remove(session.id)
|
|
2090
|
+
}),
|
|
2091
|
+
{ git: true, config: cfg },
|
|
2092
|
+
)
|
|
2093
|
+
|
|
2094
|
+
// Regression: empty assistant turn loop
|
|
2095
|
+
|
|
2096
|
+
it.instance("does not loop empty assistant turns for a simple reply", () =>
|
|
2097
|
+
Effect.gen(function* () {
|
|
2098
|
+
const { llm } = yield* useServerConfig(providerCfg)
|
|
2099
|
+
const prompt = yield* SessionPrompt.Service
|
|
2100
|
+
const sessions = yield* Session.Service
|
|
2101
|
+
const session = yield* sessions.create({ title: "Prompt regression" })
|
|
2102
|
+
|
|
2103
|
+
yield* llm.text("packages/opencode/src/session/processor.ts")
|
|
2104
|
+
|
|
2105
|
+
const result = yield* prompt.prompt({
|
|
2106
|
+
sessionID: session.id,
|
|
2107
|
+
agent: "build",
|
|
2108
|
+
parts: [{ type: "text", text: "Where is SessionProcessor?" }],
|
|
2109
|
+
})
|
|
2110
|
+
|
|
2111
|
+
expect(result.info.role).toBe("assistant")
|
|
2112
|
+
expect(result.parts.some((part) => part.type === "text" && part.text.includes("processor.ts"))).toBe(true)
|
|
2113
|
+
|
|
2114
|
+
const msgs = yield* sessions.messages({ sessionID: session.id })
|
|
2115
|
+
expect(msgs.filter((msg) => msg.info.role === "assistant")).toHaveLength(1)
|
|
2116
|
+
expect(yield* llm.calls).toBe(1)
|
|
2117
|
+
}),
|
|
2118
|
+
)
|
|
2119
|
+
|
|
2120
|
+
it.instance(
|
|
2121
|
+
"records aborted errors when prompt is cancelled mid-stream",
|
|
2122
|
+
() =>
|
|
2123
|
+
Effect.gen(function* () {
|
|
2124
|
+
const { llm } = yield* useServerConfig(providerCfg)
|
|
2125
|
+
const prompt = yield* SessionPrompt.Service
|
|
2126
|
+
const sessions = yield* Session.Service
|
|
2127
|
+
const session = yield* sessions.create({ title: "Prompt cancel regression" })
|
|
2128
|
+
|
|
2129
|
+
yield* llm.hang
|
|
2130
|
+
|
|
2131
|
+
const fiber = yield* prompt
|
|
2132
|
+
.prompt({
|
|
2133
|
+
sessionID: session.id,
|
|
2134
|
+
agent: "build",
|
|
2135
|
+
parts: [{ type: "text", text: "Cancel me" }],
|
|
2136
|
+
})
|
|
2137
|
+
.pipe(Effect.forkChild)
|
|
2138
|
+
|
|
2139
|
+
yield* llm.wait(1)
|
|
2140
|
+
yield* prompt.cancel(session.id)
|
|
2141
|
+
|
|
2142
|
+
const exit = yield* Fiber.await(fiber)
|
|
2143
|
+
expect(Exit.isSuccess(exit)).toBe(true)
|
|
2144
|
+
if (Exit.isSuccess(exit)) {
|
|
2145
|
+
expect(exit.value.info.role).toBe("assistant")
|
|
2146
|
+
if (exit.value.info.role === "assistant") {
|
|
2147
|
+
expect(exit.value.info.error?.name).toBe("MessageAbortedError")
|
|
2148
|
+
}
|
|
2149
|
+
}
|
|
2150
|
+
|
|
2151
|
+
const msgs = yield* sessions.messages({ sessionID: session.id })
|
|
2152
|
+
const last = msgs.findLast((msg) => msg.info.role === "assistant")
|
|
2153
|
+
expect(last?.info.role).toBe("assistant")
|
|
2154
|
+
if (last?.info.role === "assistant") {
|
|
2155
|
+
expect(last.info.error?.name).toBe("MessageAbortedError")
|
|
2156
|
+
}
|
|
2157
|
+
}),
|
|
2158
|
+
3_000,
|
|
2159
|
+
)
|
|
2160
|
+
|
|
2161
|
+
// Agent variant
|
|
2162
|
+
|
|
2163
|
+
noLLMServer.instance(
|
|
2164
|
+
"applies agent variant only when using agent model",
|
|
2165
|
+
() =>
|
|
2166
|
+
Effect.gen(function* () {
|
|
2167
|
+
const prompt = yield* SessionPrompt.Service
|
|
2168
|
+
const sessions = yield* Session.Service
|
|
2169
|
+
const session = yield* sessions.create({})
|
|
2170
|
+
|
|
2171
|
+
const other = yield* prompt.prompt({
|
|
2172
|
+
sessionID: session.id,
|
|
2173
|
+
agent: "build",
|
|
2174
|
+
model: { providerID: ProviderV2.ID.make("opencode"), modelID: ModelV2.ID.make("kimi-k2.5-free") },
|
|
2175
|
+
noReply: true,
|
|
2176
|
+
parts: [{ type: "text", text: "hello" }],
|
|
2177
|
+
})
|
|
2178
|
+
if (other.info.role !== "user") throw new Error("expected user message")
|
|
2179
|
+
expect(other.info.model.variant).toBeUndefined()
|
|
2180
|
+
|
|
2181
|
+
const match = yield* prompt.prompt({
|
|
2182
|
+
sessionID: session.id,
|
|
2183
|
+
agent: "build",
|
|
2184
|
+
noReply: true,
|
|
2185
|
+
parts: [{ type: "text", text: "hello again" }],
|
|
2186
|
+
})
|
|
2187
|
+
if (match.info.role !== "user") throw new Error("expected user message")
|
|
2188
|
+
expect(match.info.model).toEqual({
|
|
2189
|
+
providerID: ProviderV2.ID.make("test"),
|
|
2190
|
+
modelID: ModelV2.ID.make("test-model"),
|
|
2191
|
+
variant: "xhigh",
|
|
2192
|
+
})
|
|
2193
|
+
expect(match.info.model.variant).toBe("xhigh")
|
|
2194
|
+
|
|
2195
|
+
const override = yield* prompt.prompt({
|
|
2196
|
+
sessionID: session.id,
|
|
2197
|
+
agent: "build",
|
|
2198
|
+
noReply: true,
|
|
2199
|
+
variant: "high",
|
|
2200
|
+
parts: [{ type: "text", text: "hello third" }],
|
|
2201
|
+
})
|
|
2202
|
+
if (override.info.role !== "user") throw new Error("expected user message")
|
|
2203
|
+
expect(override.info.model.variant).toBe("high")
|
|
2204
|
+
|
|
2205
|
+
yield* sessions.remove(session.id)
|
|
2206
|
+
}),
|
|
2207
|
+
{
|
|
2208
|
+
config: {
|
|
2209
|
+
...cfg,
|
|
2210
|
+
provider: {
|
|
2211
|
+
...cfg.provider,
|
|
2212
|
+
test: {
|
|
2213
|
+
...cfg.provider.test,
|
|
2214
|
+
models: {
|
|
2215
|
+
"test-model": {
|
|
2216
|
+
...cfg.provider.test.models["test-model"],
|
|
2217
|
+
variants: { xhigh: {}, high: {} },
|
|
2218
|
+
},
|
|
2219
|
+
},
|
|
2220
|
+
},
|
|
2221
|
+
},
|
|
2222
|
+
agent: {
|
|
2223
|
+
build: {
|
|
2224
|
+
model: "test/test-model",
|
|
2225
|
+
variant: "xhigh",
|
|
2226
|
+
},
|
|
2227
|
+
},
|
|
2228
|
+
},
|
|
2229
|
+
},
|
|
2230
|
+
)
|
|
2231
|
+
|
|
2232
|
+
// Agent / command resolution errors
|
|
2233
|
+
|
|
2234
|
+
noLLMServer.instance(
|
|
2235
|
+
"unknown agent throws typed error",
|
|
2236
|
+
() =>
|
|
2237
|
+
Effect.gen(function* () {
|
|
2238
|
+
const prompt = yield* SessionPrompt.Service
|
|
2239
|
+
const sessions = yield* Session.Service
|
|
2240
|
+
const session = yield* sessions.create({})
|
|
2241
|
+
const exit = yield* prompt
|
|
2242
|
+
.prompt({
|
|
2243
|
+
sessionID: session.id,
|
|
2244
|
+
agent: "nonexistent-agent-xyz",
|
|
2245
|
+
noReply: true,
|
|
2246
|
+
parts: [{ type: "text", text: "hello" }],
|
|
2247
|
+
})
|
|
2248
|
+
.pipe(Effect.exit)
|
|
2249
|
+
|
|
2250
|
+
expect(Exit.isFailure(exit)).toBe(true)
|
|
2251
|
+
if (Exit.isFailure(exit)) {
|
|
2252
|
+
const err = Cause.squash(exit.cause)
|
|
2253
|
+
expect(err).not.toBeInstanceOf(TypeError)
|
|
2254
|
+
expect(NamedError.Unknown.isInstance(err)).toBe(true)
|
|
2255
|
+
if (NamedError.Unknown.isInstance(err)) {
|
|
2256
|
+
expect(err.data.message).toContain('Agent not found: "nonexistent-agent-xyz"')
|
|
2257
|
+
}
|
|
2258
|
+
}
|
|
2259
|
+
}),
|
|
2260
|
+
30_000,
|
|
2261
|
+
)
|
|
2262
|
+
|
|
2263
|
+
noLLMServer.instance(
|
|
2264
|
+
"unknown agent error includes available agent names",
|
|
2265
|
+
() =>
|
|
2266
|
+
Effect.gen(function* () {
|
|
2267
|
+
const prompt = yield* SessionPrompt.Service
|
|
2268
|
+
const sessions = yield* Session.Service
|
|
2269
|
+
const session = yield* sessions.create({})
|
|
2270
|
+
const exit = yield* prompt
|
|
2271
|
+
.prompt({
|
|
2272
|
+
sessionID: session.id,
|
|
2273
|
+
agent: "nonexistent-agent-xyz",
|
|
2274
|
+
noReply: true,
|
|
2275
|
+
parts: [{ type: "text", text: "hello" }],
|
|
2276
|
+
})
|
|
2277
|
+
.pipe(Effect.exit)
|
|
2278
|
+
|
|
2279
|
+
expect(Exit.isFailure(exit)).toBe(true)
|
|
2280
|
+
if (Exit.isFailure(exit)) {
|
|
2281
|
+
const err = Cause.squash(exit.cause)
|
|
2282
|
+
expect(NamedError.Unknown.isInstance(err)).toBe(true)
|
|
2283
|
+
if (NamedError.Unknown.isInstance(err)) {
|
|
2284
|
+
expect(err.data.message).toContain("build")
|
|
2285
|
+
}
|
|
2286
|
+
}
|
|
2287
|
+
}),
|
|
2288
|
+
30_000,
|
|
2289
|
+
)
|
|
2290
|
+
|
|
2291
|
+
noLLMServer.instance(
|
|
2292
|
+
"unknown command throws typed error with available names",
|
|
2293
|
+
() =>
|
|
2294
|
+
Effect.gen(function* () {
|
|
2295
|
+
const prompt = yield* SessionPrompt.Service
|
|
2296
|
+
const sessions = yield* Session.Service
|
|
2297
|
+
const session = yield* sessions.create({})
|
|
2298
|
+
const exit = yield* prompt
|
|
2299
|
+
.command({
|
|
2300
|
+
sessionID: session.id,
|
|
2301
|
+
command: "nonexistent-command-xyz",
|
|
2302
|
+
arguments: "",
|
|
2303
|
+
})
|
|
2304
|
+
.pipe(Effect.exit)
|
|
2305
|
+
|
|
2306
|
+
expect(Exit.isFailure(exit)).toBe(true)
|
|
2307
|
+
if (Exit.isFailure(exit)) {
|
|
2308
|
+
const err = Cause.squash(exit.cause)
|
|
2309
|
+
expect(err).not.toBeInstanceOf(TypeError)
|
|
2310
|
+
expect(NamedError.Unknown.isInstance(err)).toBe(true)
|
|
2311
|
+
if (NamedError.Unknown.isInstance(err)) {
|
|
2312
|
+
expect(err.data.message).toContain('Command not found: "nonexistent-command-xyz"')
|
|
2313
|
+
expect(err.data.message).toContain("init")
|
|
2314
|
+
}
|
|
2315
|
+
}
|
|
2316
|
+
}),
|
|
2317
|
+
30_000,
|
|
2318
|
+
)
|