@oxagen/cli 0.5.0 → 0.6.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +66 -3
- package/dist/agent/__tests__/code-graph.test.d.ts +2 -0
- package/dist/agent/__tests__/code-graph.test.d.ts.map +1 -0
- package/dist/agent/__tests__/code-graph.test.js +83 -0
- package/dist/agent/__tests__/code-graph.test.js.map +1 -0
- package/dist/agent/__tests__/evaluator.test.d.ts +2 -0
- package/dist/agent/__tests__/evaluator.test.d.ts.map +1 -0
- package/dist/agent/__tests__/evaluator.test.js +96 -0
- package/dist/agent/__tests__/evaluator.test.js.map +1 -0
- package/dist/agent/__tests__/fleet-memory.test.d.ts +2 -0
- package/dist/agent/__tests__/fleet-memory.test.d.ts.map +1 -0
- package/dist/agent/__tests__/fleet-memory.test.js +107 -0
- package/dist/agent/__tests__/fleet-memory.test.js.map +1 -0
- package/dist/agent/__tests__/fleet-store.test.d.ts +2 -0
- package/dist/agent/__tests__/fleet-store.test.d.ts.map +1 -0
- package/dist/agent/__tests__/fleet-store.test.js +93 -0
- package/dist/agent/__tests__/fleet-store.test.js.map +1 -0
- package/dist/agent/__tests__/git-isolation.test.d.ts +2 -0
- package/dist/agent/__tests__/git-isolation.test.d.ts.map +1 -0
- package/dist/agent/__tests__/git-isolation.test.js +119 -0
- package/dist/agent/__tests__/git-isolation.test.js.map +1 -0
- package/dist/agent/__tests__/judge.test.d.ts +2 -0
- package/dist/agent/__tests__/judge.test.d.ts.map +1 -0
- package/dist/agent/__tests__/judge.test.js +135 -0
- package/dist/agent/__tests__/judge.test.js.map +1 -0
- package/dist/agent/__tests__/loop-errors.test.d.ts +2 -0
- package/dist/agent/__tests__/loop-errors.test.d.ts.map +1 -0
- package/dist/agent/__tests__/loop-errors.test.js +44 -0
- package/dist/agent/__tests__/loop-errors.test.js.map +1 -0
- package/dist/agent/__tests__/model-router.test.d.ts +2 -0
- package/dist/agent/__tests__/model-router.test.d.ts.map +1 -0
- package/dist/agent/__tests__/model-router.test.js +122 -0
- package/dist/agent/__tests__/model-router.test.js.map +1 -0
- package/dist/agent/__tests__/orchestrator-isolation.test.d.ts +2 -0
- package/dist/agent/__tests__/orchestrator-isolation.test.d.ts.map +1 -0
- package/dist/agent/__tests__/orchestrator-isolation.test.js +134 -0
- package/dist/agent/__tests__/orchestrator-isolation.test.js.map +1 -0
- package/dist/agent/__tests__/orchestrator.test.d.ts +2 -0
- package/dist/agent/__tests__/orchestrator.test.d.ts.map +1 -0
- package/dist/agent/__tests__/orchestrator.test.js +201 -0
- package/dist/agent/__tests__/orchestrator.test.js.map +1 -0
- package/dist/agent/__tests__/pipeline.test.d.ts +2 -0
- package/dist/agent/__tests__/pipeline.test.d.ts.map +1 -0
- package/dist/agent/__tests__/pipeline.test.js +226 -0
- package/dist/agent/__tests__/pipeline.test.js.map +1 -0
- package/dist/agent/__tests__/planner.test.d.ts +2 -0
- package/dist/agent/__tests__/planner.test.d.ts.map +1 -0
- package/dist/agent/__tests__/planner.test.js +98 -0
- package/dist/agent/__tests__/planner.test.js.map +1 -0
- package/dist/agent/__tests__/prompt-enhancer.test.d.ts +2 -0
- package/dist/agent/__tests__/prompt-enhancer.test.d.ts.map +1 -0
- package/dist/agent/__tests__/prompt-enhancer.test.js +107 -0
- package/dist/agent/__tests__/prompt-enhancer.test.js.map +1 -0
- package/dist/agent/__tests__/trace-format.test.d.ts +2 -0
- package/dist/agent/__tests__/trace-format.test.d.ts.map +1 -0
- package/dist/agent/__tests__/trace-format.test.js +104 -0
- package/dist/agent/__tests__/trace-format.test.js.map +1 -0
- package/dist/agent/__tests__/trace-store.test.d.ts +2 -0
- package/dist/agent/__tests__/trace-store.test.d.ts.map +1 -0
- package/dist/agent/__tests__/trace-store.test.js +113 -0
- package/dist/agent/__tests__/trace-store.test.js.map +1 -0
- package/dist/agent/code-graph.d.ts +18 -0
- package/dist/agent/code-graph.d.ts.map +1 -0
- package/dist/agent/code-graph.js +119 -0
- package/dist/agent/code-graph.js.map +1 -0
- package/dist/agent/env.d.ts +11 -0
- package/dist/agent/env.d.ts.map +1 -0
- package/dist/agent/env.js +82 -0
- package/dist/agent/env.js.map +1 -0
- package/dist/agent/evaluator.d.ts +13 -0
- package/dist/agent/evaluator.d.ts.map +1 -0
- package/dist/agent/evaluator.js +146 -0
- package/dist/agent/evaluator.js.map +1 -0
- package/dist/agent/fleet/git-isolation.d.ts +142 -0
- package/dist/agent/fleet/git-isolation.d.ts.map +1 -0
- package/dist/agent/fleet/git-isolation.js +290 -0
- package/dist/agent/fleet/git-isolation.js.map +1 -0
- package/dist/agent/fleet/memory.d.ts +21 -0
- package/dist/agent/fleet/memory.d.ts.map +1 -0
- package/dist/agent/fleet/memory.js +129 -0
- package/dist/agent/fleet/memory.js.map +1 -0
- package/dist/agent/fleet/orchestrator.d.ts +103 -0
- package/dist/agent/fleet/orchestrator.d.ts.map +1 -0
- package/dist/agent/fleet/orchestrator.js +355 -0
- package/dist/agent/fleet/orchestrator.js.map +1 -0
- package/dist/agent/fleet/store.d.ts +13 -0
- package/dist/agent/fleet/store.d.ts.map +1 -0
- package/dist/agent/fleet/store.js +79 -0
- package/dist/agent/fleet/store.js.map +1 -0
- package/dist/agent/fleet/types.d.ts +105 -0
- package/dist/agent/fleet/types.d.ts.map +1 -0
- package/dist/agent/fleet/types.js +17 -0
- package/dist/agent/fleet/types.js.map +1 -0
- package/dist/agent/judge.d.ts +38 -0
- package/dist/agent/judge.d.ts.map +1 -0
- package/dist/agent/judge.js +170 -0
- package/dist/agent/judge.js.map +1 -0
- package/dist/agent/loop.d.ts +61 -0
- package/dist/agent/loop.d.ts.map +1 -0
- package/dist/agent/loop.js +134 -0
- package/dist/agent/loop.js.map +1 -0
- package/dist/agent/memory.d.ts +14 -0
- package/dist/agent/memory.d.ts.map +1 -0
- package/dist/agent/memory.js +118 -0
- package/dist/agent/memory.js.map +1 -0
- package/dist/agent/model-router.d.ts +79 -0
- package/dist/agent/model-router.d.ts.map +1 -0
- package/dist/agent/model-router.js +141 -0
- package/dist/agent/model-router.js.map +1 -0
- package/dist/agent/model.d.ts +9 -0
- package/dist/agent/model.d.ts.map +1 -0
- package/dist/agent/model.js +24 -0
- package/dist/agent/model.js.map +1 -0
- package/dist/agent/pipeline.d.ts +82 -0
- package/dist/agent/pipeline.d.ts.map +1 -0
- package/dist/agent/pipeline.js +320 -0
- package/dist/agent/pipeline.js.map +1 -0
- package/dist/agent/planner.d.ts +16 -0
- package/dist/agent/planner.d.ts.map +1 -0
- package/dist/agent/planner.js +126 -0
- package/dist/agent/planner.js.map +1 -0
- package/dist/agent/project-context.d.ts +13 -0
- package/dist/agent/project-context.d.ts.map +1 -0
- package/dist/agent/project-context.js +66 -0
- package/dist/agent/project-context.js.map +1 -0
- package/dist/agent/prompt-enhancer.d.ts +37 -0
- package/dist/agent/prompt-enhancer.d.ts.map +1 -0
- package/dist/agent/prompt-enhancer.js +115 -0
- package/dist/agent/prompt-enhancer.js.map +1 -0
- package/dist/agent/system-prompt.d.ts +9 -0
- package/dist/agent/system-prompt.d.ts.map +1 -0
- package/dist/agent/system-prompt.js +38 -0
- package/dist/agent/system-prompt.js.map +1 -0
- package/dist/agent/tools.d.ts +19 -0
- package/dist/agent/tools.d.ts.map +1 -0
- package/dist/agent/tools.js +323 -0
- package/dist/agent/tools.js.map +1 -0
- package/dist/agent/trace-format.d.ts +6 -0
- package/dist/agent/trace-format.d.ts.map +1 -0
- package/dist/agent/trace-format.js +74 -0
- package/dist/agent/trace-format.js.map +1 -0
- package/dist/agent/trace-store.d.ts +19 -0
- package/dist/agent/trace-store.d.ts.map +1 -0
- package/dist/agent/trace-store.js +82 -0
- package/dist/agent/trace-store.js.map +1 -0
- package/dist/agent/trace.d.ts +121 -0
- package/dist/agent/trace.d.ts.map +1 -0
- package/dist/agent/trace.js +2 -0
- package/dist/agent/trace.js.map +1 -0
- package/dist/commands/__tests__/replay.test.d.ts +2 -0
- package/dist/commands/__tests__/replay.test.d.ts.map +1 -0
- package/dist/commands/__tests__/replay.test.js +76 -0
- package/dist/commands/__tests__/replay.test.js.map +1 -0
- package/dist/commands/agent.skill.load.d.ts +3 -0
- package/dist/commands/agent.skill.load.d.ts.map +1 -0
- package/dist/commands/agent.skill.load.js +57 -0
- package/dist/commands/agent.skill.load.js.map +1 -0
- package/dist/commands/code.d.ts +14 -0
- package/dist/commands/code.d.ts.map +1 -0
- package/dist/commands/code.js +100 -0
- package/dist/commands/code.js.map +1 -0
- package/dist/commands/config.d.ts +2 -0
- package/dist/commands/config.d.ts.map +1 -0
- package/dist/commands/config.js +66 -0
- package/dist/commands/config.js.map +1 -0
- package/dist/commands/env.d.ts +19 -0
- package/dist/commands/env.d.ts.map +1 -0
- package/dist/commands/env.js +64 -0
- package/dist/commands/env.js.map +1 -0
- package/dist/commands/graph.search.d.ts +10 -0
- package/dist/commands/graph.search.d.ts.map +1 -0
- package/dist/commands/graph.search.js +25 -0
- package/dist/commands/graph.search.js.map +1 -0
- package/dist/commands/mcp.add.d.ts +13 -0
- package/dist/commands/mcp.add.d.ts.map +1 -0
- package/dist/commands/mcp.add.js +110 -0
- package/dist/commands/mcp.add.js.map +1 -0
- package/dist/commands/mcp.auth.d.ts +10 -0
- package/dist/commands/mcp.auth.d.ts.map +1 -0
- package/dist/commands/mcp.auth.js +132 -0
- package/dist/commands/mcp.auth.js.map +1 -0
- package/dist/commands/mcp.check.d.ts +10 -0
- package/dist/commands/mcp.check.d.ts.map +1 -0
- package/dist/commands/mcp.check.js +114 -0
- package/dist/commands/mcp.check.js.map +1 -0
- package/dist/commands/mcp.list.d.ts +9 -0
- package/dist/commands/mcp.list.d.ts.map +1 -0
- package/dist/commands/mcp.list.js +93 -0
- package/dist/commands/mcp.list.js.map +1 -0
- package/dist/commands/mcp.permit.d.ts +12 -0
- package/dist/commands/mcp.permit.d.ts.map +1 -0
- package/dist/commands/mcp.permit.js +117 -0
- package/dist/commands/mcp.permit.js.map +1 -0
- package/dist/commands/mcp.remove.d.ts +9 -0
- package/dist/commands/mcp.remove.d.ts.map +1 -0
- package/dist/commands/mcp.remove.js +65 -0
- package/dist/commands/mcp.remove.js.map +1 -0
- package/dist/commands/plugin.org.install_bulk.js +1 -1
- package/dist/commands/plugin.org.install_bulk.js.map +1 -1
- package/dist/commands/privacy.erase.test.js +7 -0
- package/dist/commands/privacy.erase.test.js.map +1 -1
- package/dist/commands/replay.d.ts +5 -0
- package/dist/commands/replay.d.ts.map +1 -0
- package/dist/commands/replay.js +28 -0
- package/dist/commands/replay.js.map +1 -0
- package/dist/commands/schema/schema.config.d.ts +3 -0
- package/dist/commands/schema/schema.config.d.ts.map +1 -0
- package/dist/commands/schema/schema.config.js +34 -0
- package/dist/commands/schema/schema.config.js.map +1 -0
- package/dist/commands/schema/schema.disable.d.ts +3 -0
- package/dist/commands/schema/schema.disable.d.ts.map +1 -0
- package/dist/commands/schema/schema.disable.js +22 -0
- package/dist/commands/schema/schema.disable.js.map +1 -0
- package/dist/commands/schema/schema.enable.d.ts +3 -0
- package/dist/commands/schema/schema.enable.d.ts.map +1 -0
- package/dist/commands/schema/schema.enable.js +22 -0
- package/dist/commands/schema/schema.enable.js.map +1 -0
- package/dist/commands/schema/schema.export.d.ts +3 -0
- package/dist/commands/schema/schema.export.d.ts.map +1 -0
- package/dist/commands/schema/schema.export.js +31 -0
- package/dist/commands/schema/schema.export.js.map +1 -0
- package/dist/commands/schema/schema.get.d.ts +3 -0
- package/dist/commands/schema/schema.get.d.ts.map +1 -0
- package/dist/commands/schema/schema.get.js +23 -0
- package/dist/commands/schema/schema.get.js.map +1 -0
- package/dist/commands/schema/schema.label.d.ts +5 -0
- package/dist/commands/schema/schema.label.d.ts.map +1 -0
- package/dist/commands/schema/schema.label.js +60 -0
- package/dist/commands/schema/schema.label.js.map +1 -0
- package/dist/commands/schema/schema.list.d.ts +3 -0
- package/dist/commands/schema/schema.list.d.ts.map +1 -0
- package/dist/commands/schema/schema.list.js +30 -0
- package/dist/commands/schema/schema.list.js.map +1 -0
- package/dist/commands/schema/schema.prop.d.ts +5 -0
- package/dist/commands/schema/schema.prop.d.ts.map +1 -0
- package/dist/commands/schema/schema.prop.js +72 -0
- package/dist/commands/schema/schema.prop.js.map +1 -0
- package/dist/commands/schema/schema.reconcile.d.ts +10 -0
- package/dist/commands/schema/schema.reconcile.d.ts.map +1 -0
- package/dist/commands/schema/schema.reconcile.js +65 -0
- package/dist/commands/schema/schema.reconcile.js.map +1 -0
- package/dist/commands/schema/schema.rel.d.ts +5 -0
- package/dist/commands/schema/schema.rel.d.ts.map +1 -0
- package/dist/commands/schema/schema.rel.js +65 -0
- package/dist/commands/schema/schema.rel.js.map +1 -0
- package/dist/commands/schema/schema.version.d.ts +7 -0
- package/dist/commands/schema/schema.version.d.ts.map +1 -0
- package/dist/commands/schema/schema.version.js +96 -0
- package/dist/commands/schema/schema.version.js.map +1 -0
- package/dist/commands/secret.d.ts +23 -0
- package/dist/commands/secret.d.ts.map +1 -0
- package/dist/commands/secret.js +90 -0
- package/dist/commands/secret.js.map +1 -0
- package/dist/commands/skill.create.d.ts +3 -0
- package/dist/commands/skill.create.d.ts.map +1 -0
- package/dist/commands/skill.create.js +52 -0
- package/dist/commands/skill.create.js.map +1 -0
- package/dist/commands/skill.enable.d.ts +3 -0
- package/dist/commands/skill.enable.d.ts.map +1 -0
- package/dist/commands/skill.enable.js +31 -0
- package/dist/commands/skill.enable.js.map +1 -0
- package/dist/commands.test.js +1291 -281
- package/dist/commands.test.js.map +1 -1
- package/dist/components/DevStatus.d.ts.map +1 -1
- package/dist/components/DevStatus.js +3 -2
- package/dist/components/DevStatus.js.map +1 -1
- package/dist/daemon/client.d.ts +30 -0
- package/dist/daemon/client.d.ts.map +1 -0
- package/dist/daemon/client.js +97 -0
- package/dist/daemon/client.js.map +1 -0
- package/dist/daemon/code-graph/builder.d.ts +6 -0
- package/dist/daemon/code-graph/builder.d.ts.map +1 -0
- package/dist/daemon/code-graph/builder.js +215 -0
- package/dist/daemon/code-graph/builder.js.map +1 -0
- package/dist/daemon/code-graph/query.d.ts +29 -0
- package/dist/daemon/code-graph/query.d.ts.map +1 -0
- package/dist/daemon/code-graph/query.js +98 -0
- package/dist/daemon/code-graph/query.js.map +1 -0
- package/dist/daemon/code-graph/types.d.ts +37 -0
- package/dist/daemon/code-graph/types.d.ts.map +1 -0
- package/dist/daemon/code-graph/types.js +5 -0
- package/dist/daemon/code-graph/types.js.map +1 -0
- package/dist/daemon/code-graph/watcher.d.ts +37 -0
- package/dist/daemon/code-graph/watcher.d.ts.map +1 -0
- package/dist/daemon/code-graph/watcher.js +79 -0
- package/dist/daemon/code-graph/watcher.js.map +1 -0
- package/dist/daemon/lifecycle.d.ts +6 -0
- package/dist/daemon/lifecycle.d.ts.map +1 -0
- package/dist/daemon/lifecycle.js +132 -0
- package/dist/daemon/lifecycle.js.map +1 -0
- package/dist/daemon/protocol.d.ts +113 -0
- package/dist/daemon/protocol.d.ts.map +1 -0
- package/dist/daemon/protocol.js +16 -0
- package/dist/daemon/protocol.js.map +1 -0
- package/dist/daemon/server.d.ts +26 -0
- package/dist/daemon/server.d.ts.map +1 -0
- package/dist/daemon/server.js +168 -0
- package/dist/daemon/server.js.map +1 -0
- package/dist/index.d.ts +1 -1
- package/dist/index.js +262 -356
- package/dist/index.js.map +1 -1
- package/dist/lib/api.d.ts +5 -0
- package/dist/lib/api.d.ts.map +1 -0
- package/dist/lib/api.js +52 -0
- package/dist/lib/api.js.map +1 -0
- package/dist/lib/config.d.ts +3 -0
- package/dist/lib/config.d.ts.map +1 -1
- package/dist/lib/config.js +8 -2
- package/dist/lib/config.js.map +1 -1
- package/dist/lib/differential-context.d.ts +46 -0
- package/dist/lib/differential-context.d.ts.map +1 -0
- package/dist/lib/differential-context.js +89 -0
- package/dist/lib/differential-context.js.map +1 -0
- package/dist/lib/resolve.d.ts +3 -0
- package/dist/lib/resolve.d.ts.map +1 -0
- package/dist/lib/resolve.js +29 -0
- package/dist/lib/resolve.js.map +1 -0
- package/dist/lib/structured-tool-io.d.ts +31 -0
- package/dist/lib/structured-tool-io.d.ts.map +1 -0
- package/dist/lib/structured-tool-io.js +56 -0
- package/dist/lib/structured-tool-io.js.map +1 -0
- package/dist/repl/__tests__/_queue_demo.test.d.ts +2 -0
- package/dist/repl/__tests__/_queue_demo.test.d.ts.map +1 -0
- package/dist/repl/__tests__/_queue_demo.test.js +40 -0
- package/dist/repl/__tests__/_queue_demo.test.js.map +1 -0
- package/dist/repl/__tests__/components.test.d.ts +2 -0
- package/dist/repl/__tests__/components.test.d.ts.map +1 -0
- package/dist/repl/__tests__/components.test.js +38 -0
- package/dist/repl/__tests__/components.test.js.map +1 -0
- package/dist/repl/__tests__/interactive.queue.test.d.ts +2 -0
- package/dist/repl/__tests__/interactive.queue.test.d.ts.map +1 -0
- package/dist/repl/__tests__/interactive.queue.test.js +124 -0
- package/dist/repl/__tests__/interactive.queue.test.js.map +1 -0
- package/dist/repl/components.d.ts +59 -0
- package/dist/repl/components.d.ts.map +1 -0
- package/dist/repl/components.js +152 -0
- package/dist/repl/components.js.map +1 -0
- package/dist/repl/interactive.d.ts +12 -0
- package/dist/repl/interactive.d.ts.map +1 -0
- package/dist/repl/interactive.js +326 -0
- package/dist/repl/interactive.js.map +1 -0
- package/dist/repl/one-shot.d.ts +9 -0
- package/dist/repl/one-shot.d.ts.map +1 -0
- package/dist/repl/one-shot.js +83 -0
- package/dist/repl/one-shot.js.map +1 -0
- package/dist/tui/__tests__/app.test.d.ts +2 -0
- package/dist/tui/__tests__/app.test.d.ts.map +1 -0
- package/dist/tui/__tests__/app.test.js +136 -0
- package/dist/tui/__tests__/app.test.js.map +1 -0
- package/dist/tui/__tests__/banner.test.d.ts +2 -0
- package/dist/tui/__tests__/banner.test.d.ts.map +1 -0
- package/dist/tui/__tests__/banner.test.js +15 -0
- package/dist/tui/__tests__/banner.test.js.map +1 -0
- package/dist/tui/__tests__/command-form.test.d.ts +2 -0
- package/dist/tui/__tests__/command-form.test.d.ts.map +1 -0
- package/dist/tui/__tests__/command-form.test.js +96 -0
- package/dist/tui/__tests__/command-form.test.js.map +1 -0
- package/dist/tui/__tests__/command-tree.test.d.ts +2 -0
- package/dist/tui/__tests__/command-tree.test.d.ts.map +1 -0
- package/dist/tui/__tests__/command-tree.test.js +55 -0
- package/dist/tui/__tests__/command-tree.test.js.map +1 -0
- package/dist/tui/__tests__/runner.test.d.ts +2 -0
- package/dist/tui/__tests__/runner.test.d.ts.map +1 -0
- package/dist/tui/__tests__/runner.test.js +38 -0
- package/dist/tui/__tests__/runner.test.js.map +1 -0
- package/dist/tui/__tests__/theme.test.d.ts +2 -0
- package/dist/tui/__tests__/theme.test.d.ts.map +1 -0
- package/dist/tui/__tests__/theme.test.js +11 -0
- package/dist/tui/__tests__/theme.test.js.map +1 -0
- package/dist/tui/agent-view/activity-feed.d.ts +3 -0
- package/dist/tui/agent-view/activity-feed.d.ts.map +1 -0
- package/dist/tui/agent-view/activity-feed.js +34 -0
- package/dist/tui/agent-view/activity-feed.js.map +1 -0
- package/dist/tui/agent-view/budget-bar.d.ts +3 -0
- package/dist/tui/agent-view/budget-bar.d.ts.map +1 -0
- package/dist/tui/agent-view/budget-bar.js +53 -0
- package/dist/tui/agent-view/budget-bar.js.map +1 -0
- package/dist/tui/agent-view/compile-panel.d.ts +3 -0
- package/dist/tui/agent-view/compile-panel.d.ts.map +1 -0
- package/dist/tui/agent-view/compile-panel.js +34 -0
- package/dist/tui/agent-view/compile-panel.js.map +1 -0
- package/dist/tui/agent-view/index.d.ts +4 -0
- package/dist/tui/agent-view/index.d.ts.map +1 -0
- package/dist/tui/agent-view/index.js +31 -0
- package/dist/tui/agent-view/index.js.map +1 -0
- package/dist/tui/agent-view/memory-panel.d.ts +3 -0
- package/dist/tui/agent-view/memory-panel.d.ts.map +1 -0
- package/dist/tui/agent-view/memory-panel.js +80 -0
- package/dist/tui/agent-view/memory-panel.js.map +1 -0
- package/dist/tui/agent-view/session-panel.d.ts +3 -0
- package/dist/tui/agent-view/session-panel.d.ts.map +1 -0
- package/dist/tui/agent-view/session-panel.js +44 -0
- package/dist/tui/agent-view/session-panel.js.map +1 -0
- package/dist/tui/agent-view/status-bar.d.ts +7 -0
- package/dist/tui/agent-view/status-bar.d.ts.map +1 -0
- package/dist/tui/agent-view/status-bar.js +22 -0
- package/dist/tui/agent-view/status-bar.js.map +1 -0
- package/dist/tui/app.d.ts +9 -0
- package/dist/tui/app.d.ts.map +1 -0
- package/dist/tui/app.js +115 -0
- package/dist/tui/app.js.map +1 -0
- package/dist/tui/banner.d.ts +5 -0
- package/dist/tui/banner.d.ts.map +1 -0
- package/dist/tui/banner.js +17 -0
- package/dist/tui/banner.js.map +1 -0
- package/dist/tui/command-form.d.ts +9 -0
- package/dist/tui/command-form.d.ts.map +1 -0
- package/dist/tui/command-form.js +76 -0
- package/dist/tui/command-form.js.map +1 -0
- package/dist/tui/command-tree.d.ts +30 -0
- package/dist/tui/command-tree.d.ts.map +1 -0
- package/dist/tui/command-tree.js +43 -0
- package/dist/tui/command-tree.js.map +1 -0
- package/dist/tui/fleet-view/agent-row.d.ts +10 -0
- package/dist/tui/fleet-view/agent-row.d.ts.map +1 -0
- package/dist/tui/fleet-view/agent-row.js +80 -0
- package/dist/tui/fleet-view/agent-row.js.map +1 -0
- package/dist/tui/fleet-view/dispatch-input.d.ts +5 -0
- package/dist/tui/fleet-view/dispatch-input.d.ts.map +1 -0
- package/dist/tui/fleet-view/dispatch-input.js +36 -0
- package/dist/tui/fleet-view/dispatch-input.js.map +1 -0
- package/dist/tui/fleet-view/fleet-app.d.ts +11 -0
- package/dist/tui/fleet-view/fleet-app.d.ts.map +1 -0
- package/dist/tui/fleet-view/fleet-app.js +95 -0
- package/dist/tui/fleet-view/fleet-app.js.map +1 -0
- package/dist/tui/fleet-view/fleet-summary.d.ts +6 -0
- package/dist/tui/fleet-view/fleet-summary.d.ts.map +1 -0
- package/dist/tui/fleet-view/fleet-summary.js +19 -0
- package/dist/tui/fleet-view/fleet-summary.js.map +1 -0
- package/dist/tui/fleet-view/index.d.ts +16 -0
- package/dist/tui/fleet-view/index.d.ts.map +1 -0
- package/dist/tui/fleet-view/index.js +61 -0
- package/dist/tui/fleet-view/index.js.map +1 -0
- package/dist/tui/runner.d.ts +7 -0
- package/dist/tui/runner.d.ts.map +1 -0
- package/dist/tui/runner.js +36 -0
- package/dist/tui/runner.js.map +1 -0
- package/dist/tui/theme.d.ts +8 -0
- package/dist/tui/theme.d.ts.map +1 -0
- package/dist/tui/theme.js +10 -0
- package/dist/tui/theme.js.map +1 -0
- package/package.json +12 -7
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"orchestrator-isolation.test.js","sourceRoot":"","sources":["../../../src/agent/__tests__/orchestrator-isolation.test.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AACH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAElD,EAAE,CAAC,IAAI,CAAC,uBAAuB,EAAE,GAAG,EAAE,CAAC,CAAC;IACtC,aAAa,EAAE,KAAK,EAAE,EAAE,MAAM,EAAsB,EAAE,EAAE,CAAC,CAAC;QACxD,MAAM;QACN,OAAO,EAAE,EAAE;QACX,QAAQ,EAAE,EAAE;QACZ,OAAO,EAAE,EAAE;KACZ,CAAC;CACH,CAAC,CAAC,CAAC;AAEJ,OAAO,EAAE,KAAK,EAAoB,MAAM,0BAA0B,CAAC;AAInE,SAAS,IAAI,CAAC,EAAU,EAAE,OAAsB,EAAE;IAChD,OAAO;QACL,EAAE;QACF,KAAK,EAAE,EAAE;QACT,WAAW,EAAE,EAAE;QACf,MAAM,EAAE,QAAQ;QAChB,SAAS,EAAE,EAAE;QACb,KAAK,EAAE,EAAE;QACT,IAAI,EAAE,MAAM;QACZ,KAAK,EAAE,4BAA4B;QACnC,SAAS,EAAE,CAAC;QACZ,KAAK,EAAE,EAAE,WAAW,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE;QACtD,GAAG,IAAI;KACR,CAAC;AACJ,CAAC;AAED,SAAS,IAAI,CAAC,KAAa;IACzB,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,EAAE,SAAS,EAAE,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC;AACvE,CAAC;AAED,SAAS,QAAQ;IACf,IAAI,OAAwB,CAAC;IAC7B,MAAM,OAAO,GAAG,IAAI,OAAO,CAAI,CAAC,GAAG,EAAE,EAAE;QACrC,OAAO,GAAG,GAAG,CAAC;IAChB,CAAC,CAAC,CAAC;IACH,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;AAC9B,CAAC;AAED,yEAAyE;AACzE,SAAS,aAAa,CACpB,SAA6C;IAE7C,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,MAAM,EAAE,GAAG,CAAC,EAAU,EAAc,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,IAAI,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC,CAAC;IAC3F,OAAO;QACL,KAAK;QACL,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,KAAK,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC;QACzC,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE;YAClB,KAAK,CAAC,IAAI,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;YAC1B,OAAO,OAAO,EAAE,EAAE,CAAC;QACrB,CAAC;QACD,UAAU,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE;YACvB,KAAK,CAAC,IAAI,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC;YAC/B,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;QAChB,CAAC;QACD,eAAe,EAAE,GAAG,EAAE,CAAC,EAAE;QACzB,SAAS,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE;YACtB,KAAK,CAAC,IAAI,CAAC,aAAa,EAAE,EAAE,CAAC,CAAC;YAC9B,OAAO,SAAS,EAAE,CAAC,EAAE,CAAC,IAAI,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,UAAU,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;QAC7E,CAAC;QACD,cAAc,EAAE,GAAG,EAAE,CAAC,IAAI;QAC1B,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,KAAK,KAAK,CAAC,IAAI,CAAC,WAAW,EAAE,SAAS,IAAI,EAAE,IAAI,IAAI,KAAK,EAAE,CAAC;QACzF,UAAU,EAAE,KAAK,IAAI,EAAE,CAAC,KAAK,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC;KACtD,CAAC;AACJ,CAAC;AAED,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;IACxC,EAAE,CAAC,wEAAwE,EAAE,KAAK,IAAI,EAAE;QACtF,MAAM,GAAG,GAAG,aAAa,EAAE,CAAC;QAC5B,IAAI,OAAO,GAAG,EAAE,CAAC;QACjB,MAAM,MAAM,GAAgB,KAAK,EAAE,CAAC,EAAE,EAAE;YACtC,OAAO,GAAG,CAAC,CAAC,GAAG,CAAC;YAChB,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,WAAW,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,EAAE,CAAC;QAChF,CAAC,CAAC;QAEF,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,GAAG,EAAE,OAAO,EAAE,SAAS,EAAE,GAAG,EAAE,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;QAChF,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;QACnC,MAAM,KAAK,CAAC,KAAK,EAAE,CAAC;QAEpB,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,iCAAiC;QACjE,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC;YACxB,UAAU;YACV,eAAe;YACf,cAAc;YACd,uBAAuB;SACxB,CAAC,CAAC;QACH,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAC1D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QACvE,MAAM,GAAG,GAAG,aAAa,CAAC,GAAG,EAAE,CAAC,CAAC;YAC/B,EAAE,EAAE,KAAK;YACT,SAAS,EAAE,CAAC,UAAU,CAAC;YACvB,UAAU,EAAE,MAAM;YAClB,OAAO,EAAE,KAAK;SACf,CAAC,CAAC,CAAC;QACJ,MAAM,MAAM,GAAoD,EAAE,CAAC;QACnE,MAAM,MAAM,GAAgB,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;QAE7E,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,GAAG,EAAE,OAAO,EAAE,SAAS,EAAE,GAAG,EAAE,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;QAChF,KAAK,CAAC,EAAE,CAAC,UAAU,EAAE,CAAC,CAA2C,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;QACtF,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;QACnC,MAAM,KAAK,CAAC,KAAK,EAAE,CAAC;QAEpB,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QACnD,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,sBAAsB,CAAC,CAAC,CAAC,sBAAsB;QAC3E,MAAM,IAAI,GAAG,KAAK,CAAC,QAAQ,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;QACxC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAClC,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC,OAAO,CAAC,sBAAsB,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+EAA+E,EAAE,KAAK,IAAI,EAAE;QAC7F,MAAM,GAAG,GAAG,aAAa,EAAE,CAAC;QAC5B,IAAI,MAAM,GAAG,CAAC,CAAC;QACf,IAAI,SAAS,GAAG,CAAC,CAAC;QAClB,MAAM,OAAO,GAAG,QAAQ,EAAE,CAAC;QAC3B,MAAM,WAAW,GAAG,QAAQ,EAAE,CAAC;QAC/B,MAAM,MAAM,GAAgB,KAAK,IAAI,EAAE;YACrC,MAAM,EAAE,CAAC;YACT,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;YACxC,IAAI,MAAM,KAAK,CAAC;gBAAE,WAAW,CAAC,OAAO,EAAE,CAAC;YACxC,MAAM,OAAO,CAAC,OAAO,CAAC;YACtB,MAAM,EAAE,CAAC;YACT,OAAO,EAAE,IAAI,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;QAC5C,CAAC,CAAC;QAEF,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,GAAG,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC,EAAE,SAAS,EAAE,GAAG,EAAE,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;QAChG,uEAAuE;QACvE,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,IAAI,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QACzF,MAAM,IAAI,GAAG,KAAK,CAAC,KAAK,EAAE,CAAC;QAE3B,MAAM,WAAW,CAAC,OAAO,CAAC;QAC1B,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC1B,OAAO,CAAC,OAAO,EAAE,CAAC;QAClB,MAAM,IAAI,CAAC;IACb,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"orchestrator.test.d.ts","sourceRoot":"","sources":["../../../src/agent/__tests__/orchestrator.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,201 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Fleet orchestrator — proves the army schedules safely: dependencies gate
|
|
3
|
+
* ordering, overlapping file ownership serializes agents, the concurrency cap is
|
|
4
|
+
* honoured, a failed dependency blocks its dependents, and every finished task
|
|
5
|
+
* records a weighted memory. The coding loop is injected as a fake runner, so no
|
|
6
|
+
* gateway or filesystem is touched. The prompt enhancer is mocked to a no-op.
|
|
7
|
+
*/
|
|
8
|
+
import { describe, it, expect, vi } from "vitest";
|
|
9
|
+
vi.mock("../prompt-enhancer.js", () => ({
|
|
10
|
+
enhancePrompt: async ({ prompt }) => ({
|
|
11
|
+
prompt,
|
|
12
|
+
context: "",
|
|
13
|
+
resolved: [],
|
|
14
|
+
lessons: [],
|
|
15
|
+
}),
|
|
16
|
+
}));
|
|
17
|
+
import { Fleet } from "../fleet/orchestrator.js";
|
|
18
|
+
function task(id, over = {}) {
|
|
19
|
+
return {
|
|
20
|
+
id,
|
|
21
|
+
title: id,
|
|
22
|
+
description: id, // the enhancer is mocked, so the runner sees this as its prompt
|
|
23
|
+
status: "queued",
|
|
24
|
+
dependsOn: [],
|
|
25
|
+
files: [],
|
|
26
|
+
tier: "fast",
|
|
27
|
+
model: "anthropic/claude-haiku-4.5",
|
|
28
|
+
createdAt: 1,
|
|
29
|
+
usage: { inputTokens: 0, outputTokens: 0, costUsd: 0 },
|
|
30
|
+
...over,
|
|
31
|
+
};
|
|
32
|
+
}
|
|
33
|
+
function plan(tasks) {
|
|
34
|
+
return { id: "p1", goal: "g", createdAt: 1, tasks, status: "draft" };
|
|
35
|
+
}
|
|
36
|
+
const flush = async () => {
|
|
37
|
+
for (let i = 0; i < 6; i++)
|
|
38
|
+
await new Promise((r) => setImmediate(r));
|
|
39
|
+
};
|
|
40
|
+
function deferred() {
|
|
41
|
+
let resolve;
|
|
42
|
+
const promise = new Promise((res) => {
|
|
43
|
+
resolve = res;
|
|
44
|
+
});
|
|
45
|
+
return { promise, resolve };
|
|
46
|
+
}
|
|
47
|
+
function fakeMemory() {
|
|
48
|
+
return { record: vi.fn(), recall: () => [], all: () => [] };
|
|
49
|
+
}
|
|
50
|
+
function statusOf(fleet, id) {
|
|
51
|
+
return fleet.snapshot().agents.find((a) => a.taskId === id)?.status;
|
|
52
|
+
}
|
|
53
|
+
describe("Fleet scheduling", () => {
|
|
54
|
+
it("runs a dependent task only after its dependency completes", async () => {
|
|
55
|
+
const events = [];
|
|
56
|
+
const runner = async (o) => {
|
|
57
|
+
events.push(`start:${o.prompt}`);
|
|
58
|
+
await Promise.resolve();
|
|
59
|
+
events.push(`end:${o.prompt}`);
|
|
60
|
+
return { text: `did ${o.prompt}`, steps: 2, usage: { inputTokens: 100, outputTokens: 50 } };
|
|
61
|
+
};
|
|
62
|
+
const fleet = new Fleet({ cwd: "/x", runner, concurrency: 4 });
|
|
63
|
+
fleet.loadPlan(plan([task("t1"), task("t2", { dependsOn: ["t1"] })]));
|
|
64
|
+
await fleet.start();
|
|
65
|
+
expect(events.indexOf("end:t1")).toBeLessThan(events.indexOf("start:t2"));
|
|
66
|
+
expect(statusOf(fleet, "t1")).toBe("done");
|
|
67
|
+
expect(statusOf(fleet, "t2")).toBe("done");
|
|
68
|
+
});
|
|
69
|
+
it("respects the concurrency cap", async () => {
|
|
70
|
+
let active = 0;
|
|
71
|
+
let maxActive = 0;
|
|
72
|
+
const gates = new Map();
|
|
73
|
+
const runner = async (o) => {
|
|
74
|
+
active++;
|
|
75
|
+
maxActive = Math.max(maxActive, active);
|
|
76
|
+
const g = gates.get(o.prompt) ?? deferred();
|
|
77
|
+
gates.set(o.prompt, g);
|
|
78
|
+
await g.promise;
|
|
79
|
+
active--;
|
|
80
|
+
return { text: "ok", steps: 1, usage: { inputTokens: 1, outputTokens: 1 } };
|
|
81
|
+
};
|
|
82
|
+
// Pre-create gates so resolving them releases task "c" too, even though it
|
|
83
|
+
// only starts (and would otherwise create its gate) after a/b complete.
|
|
84
|
+
for (const id of ["a", "b", "c"])
|
|
85
|
+
gates.set(id, deferred());
|
|
86
|
+
const fleet = new Fleet({ cwd: "/x", runner, concurrency: 2 });
|
|
87
|
+
fleet.loadPlan(plan([task("a"), task("b"), task("c")]));
|
|
88
|
+
const done = fleet.start();
|
|
89
|
+
await flush();
|
|
90
|
+
expect(fleet.snapshot().runningCount).toBe(2);
|
|
91
|
+
for (const g of gates.values())
|
|
92
|
+
g.resolve(undefined);
|
|
93
|
+
await done;
|
|
94
|
+
expect(maxActive).toBe(2);
|
|
95
|
+
expect(fleet.snapshot().doneCount).toBe(3);
|
|
96
|
+
});
|
|
97
|
+
it("serializes tasks whose files overlap", async () => {
|
|
98
|
+
let active = 0;
|
|
99
|
+
let maxActive = 0;
|
|
100
|
+
const gates = new Map();
|
|
101
|
+
const runner = async (o) => {
|
|
102
|
+
active++;
|
|
103
|
+
maxActive = Math.max(maxActive, active);
|
|
104
|
+
const g = gates.get(o.prompt) ?? deferred();
|
|
105
|
+
gates.set(o.prompt, g);
|
|
106
|
+
await g.promise;
|
|
107
|
+
active--;
|
|
108
|
+
return { text: "ok", steps: 1, usage: { inputTokens: 1, outputTokens: 1 } };
|
|
109
|
+
};
|
|
110
|
+
const fleet = new Fleet({ cwd: "/x", runner, concurrency: 4 });
|
|
111
|
+
fleet.loadPlan(plan([task("t1", { files: ["shared.ts"] }), task("t2", { files: ["shared.ts"] })]));
|
|
112
|
+
const done = fleet.start();
|
|
113
|
+
await flush();
|
|
114
|
+
// Only one of the two file-conflicting tasks may run at a time.
|
|
115
|
+
expect(fleet.snapshot().runningCount).toBe(1);
|
|
116
|
+
gates.get("t1")?.resolve(undefined);
|
|
117
|
+
await flush();
|
|
118
|
+
expect(fleet.snapshot().runningCount).toBe(1); // now t2
|
|
119
|
+
gates.get("t2")?.resolve(undefined);
|
|
120
|
+
await done;
|
|
121
|
+
expect(maxActive).toBe(1);
|
|
122
|
+
expect(fleet.snapshot().doneCount).toBe(2);
|
|
123
|
+
});
|
|
124
|
+
it("blocks dependents when a dependency fails, and records the failure", async () => {
|
|
125
|
+
const memory = fakeMemory();
|
|
126
|
+
const runner = async (o) => {
|
|
127
|
+
if (o.prompt === "t1")
|
|
128
|
+
throw new Error("kaboom");
|
|
129
|
+
return { text: "ok", steps: 1, usage: { inputTokens: 1, outputTokens: 1 } };
|
|
130
|
+
};
|
|
131
|
+
const fleet = new Fleet({ cwd: "/x", runner, memory });
|
|
132
|
+
fleet.loadPlan(plan([task("t1"), task("t2", { dependsOn: ["t1"] })]));
|
|
133
|
+
await fleet.start();
|
|
134
|
+
expect(statusOf(fleet, "t1")).toBe("failed");
|
|
135
|
+
expect(statusOf(fleet, "t2")).toBe("blocked");
|
|
136
|
+
expect(fleet.snapshot().failedCount).toBe(2); // failed + blocked
|
|
137
|
+
expect(memory.record).toHaveBeenCalledWith(expect.objectContaining({ outcome: "failure", kind: "gotcha" }));
|
|
138
|
+
});
|
|
139
|
+
it("records a success memory and accumulates token cost", async () => {
|
|
140
|
+
const memory = fakeMemory();
|
|
141
|
+
const runner = async () => ({
|
|
142
|
+
text: "rewired the thing",
|
|
143
|
+
steps: 3,
|
|
144
|
+
usage: { inputTokens: 1_000_000, outputTokens: 0 },
|
|
145
|
+
});
|
|
146
|
+
const fleet = new Fleet({ cwd: "/x", runner, memory });
|
|
147
|
+
fleet.loadPlan(plan([task("fix-bug", { title: "fix the bug", description: "fix the bug" })]));
|
|
148
|
+
await fleet.start();
|
|
149
|
+
expect(memory.record).toHaveBeenCalledWith(expect.objectContaining({ outcome: "success", kind: "bug-root-cause" }));
|
|
150
|
+
// 1M Haiku input tokens = $1.00.
|
|
151
|
+
expect(fleet.snapshot().totals.costUsd).toBeCloseTo(1, 6);
|
|
152
|
+
});
|
|
153
|
+
it("dispatches an ad-hoc prompt and routes its model", async () => {
|
|
154
|
+
const runner = async (o) => ({
|
|
155
|
+
text: `did ${o.prompt}`,
|
|
156
|
+
steps: 1,
|
|
157
|
+
usage: { inputTokens: 10, outputTokens: 5 },
|
|
158
|
+
});
|
|
159
|
+
const fleet = new Fleet({ cwd: "/x", runner });
|
|
160
|
+
const id = fleet.dispatchPrompt("rename a local variable");
|
|
161
|
+
await fleet.start();
|
|
162
|
+
const agent = fleet.snapshot().agents.find((a) => a.taskId === id);
|
|
163
|
+
expect(agent?.status).toBe("done");
|
|
164
|
+
expect(agent?.model).toContain("haiku"); // trivial → fast tier
|
|
165
|
+
});
|
|
166
|
+
it("cancelAll aborts in-flight agents and settles", async () => {
|
|
167
|
+
const gates = new Map();
|
|
168
|
+
const runner = async (o) => {
|
|
169
|
+
const g = gates.get(o.prompt) ?? deferred();
|
|
170
|
+
gates.set(o.prompt, g);
|
|
171
|
+
await new Promise((resolve, reject) => {
|
|
172
|
+
o.signal?.addEventListener("abort", () => reject(new Error("aborted")));
|
|
173
|
+
void g.promise.then(() => resolve());
|
|
174
|
+
});
|
|
175
|
+
return { text: "ok", steps: 1, usage: { inputTokens: 1, outputTokens: 1 } };
|
|
176
|
+
};
|
|
177
|
+
const fleet = new Fleet({ cwd: "/x", runner, concurrency: 2 });
|
|
178
|
+
fleet.loadPlan(plan([task("a", { files: ["a.ts"] }), task("b", { files: ["b.ts"] })]));
|
|
179
|
+
const done = fleet.start();
|
|
180
|
+
await flush();
|
|
181
|
+
expect(fleet.snapshot().runningCount).toBe(2);
|
|
182
|
+
fleet.cancelAll();
|
|
183
|
+
await done;
|
|
184
|
+
expect(fleet.snapshot().doneCount).toBe(0);
|
|
185
|
+
expect(fleet.snapshot().agents.every((a) => a.status === "cancelled")).toBe(true);
|
|
186
|
+
});
|
|
187
|
+
it("emits an update snapshot on changes", async () => {
|
|
188
|
+
const runner = async () => ({
|
|
189
|
+
text: "ok",
|
|
190
|
+
steps: 1,
|
|
191
|
+
usage: { inputTokens: 1, outputTokens: 1 },
|
|
192
|
+
});
|
|
193
|
+
const fleet = new Fleet({ cwd: "/x", runner });
|
|
194
|
+
const seen = [];
|
|
195
|
+
fleet.on("update", (snap) => seen.push(snap.agents.length));
|
|
196
|
+
fleet.loadPlan(plan([task("a")]));
|
|
197
|
+
await fleet.start();
|
|
198
|
+
expect(seen.length).toBeGreaterThan(0);
|
|
199
|
+
});
|
|
200
|
+
});
|
|
201
|
+
//# sourceMappingURL=orchestrator.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"orchestrator.test.js","sourceRoot":"","sources":["../../../src/agent/__tests__/orchestrator.test.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AACH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAElD,EAAE,CAAC,IAAI,CAAC,uBAAuB,EAAE,GAAG,EAAE,CAAC,CAAC;IACtC,aAAa,EAAE,KAAK,EAAE,EAAE,MAAM,EAAsB,EAAE,EAAE,CAAC,CAAC;QACxD,MAAM;QACN,OAAO,EAAE,EAAE;QACX,QAAQ,EAAE,EAAE;QACZ,OAAO,EAAE,EAAE;KACZ,CAAC;CACH,CAAC,CAAC,CAAC;AAEJ,OAAO,EAAE,KAAK,EAAoB,MAAM,0BAA0B,CAAC;AAInE,SAAS,IAAI,CAAC,EAAU,EAAE,OAAsB,EAAE;IAChD,OAAO;QACL,EAAE;QACF,KAAK,EAAE,EAAE;QACT,WAAW,EAAE,EAAE,EAAE,gEAAgE;QACjF,MAAM,EAAE,QAAQ;QAChB,SAAS,EAAE,EAAE;QACb,KAAK,EAAE,EAAE;QACT,IAAI,EAAE,MAAM;QACZ,KAAK,EAAE,4BAA4B;QACnC,SAAS,EAAE,CAAC;QACZ,KAAK,EAAE,EAAE,WAAW,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE;QACtD,GAAG,IAAI;KACR,CAAC;AACJ,CAAC;AAED,SAAS,IAAI,CAAC,KAAa;IACzB,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,EAAE,SAAS,EAAE,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC;AACvE,CAAC;AAED,MAAM,KAAK,GAAG,KAAK,IAAmB,EAAE;IACtC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE;QAAE,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC;AACxE,CAAC,CAAC;AAEF,SAAS,QAAQ;IACf,IAAI,OAAwB,CAAC;IAC7B,MAAM,OAAO,GAAG,IAAI,OAAO,CAAI,CAAC,GAAG,EAAE,EAAE;QACrC,OAAO,GAAG,GAAG,CAAC;IAChB,CAAC,CAAC,CAAC;IACH,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;AAC9B,CAAC;AAED,SAAS,UAAU;IACjB,OAAO,EAAE,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;AAC9D,CAAC;AAED,SAAS,QAAQ,CAAC,KAAY,EAAE,EAAU;IACxC,OAAO,KAAK,CAAC,QAAQ,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,EAAE,CAAC,EAAE,MAAM,CAAC;AACtE,CAAC;AAED,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;QACzE,MAAM,MAAM,GAAa,EAAE,CAAC;QAC5B,MAAM,MAAM,GAAgB,KAAK,EAAE,CAAC,EAAE,EAAE;YACtC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;YACjC,MAAM,OAAO,CAAC,OAAO,EAAE,CAAC;YACxB,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;YAC/B,OAAO,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,WAAW,EAAE,GAAG,EAAE,YAAY,EAAE,EAAE,EAAE,EAAE,CAAC;QAC9F,CAAC,CAAC;QACF,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC,EAAE,CAAC,CAAC;QAC/D,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,CAAC,IAAI,EAAE,EAAE,SAAS,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QACtE,MAAM,KAAK,CAAC,KAAK,EAAE,CAAC;QAEpB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,YAAY,CAAC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC;QAC1E,MAAM,CAAC,QAAQ,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC3C,MAAM,CAAC,QAAQ,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;QAC5C,IAAI,MAAM,GAAG,CAAC,CAAC;QACf,IAAI,SAAS,GAAG,CAAC,CAAC;QAClB,MAAM,KAAK,GAAG,IAAI,GAAG,EAAuC,CAAC;QAC7D,MAAM,MAAM,GAAgB,KAAK,EAAE,CAAC,EAAE,EAAE;YACtC,MAAM,EAAE,CAAC;YACT,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;YACxC,MAAM,CAAC,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,QAAQ,EAAE,CAAC;YAC5C,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;YACvB,MAAM,CAAC,CAAC,OAAO,CAAC;YAChB,MAAM,EAAE,CAAC;YACT,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,WAAW,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,EAAE,CAAC;QAC9E,CAAC,CAAC;QACF,2EAA2E;QAC3E,wEAAwE;QACxE,KAAK,MAAM,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC;YAAE,KAAK,CAAC,GAAG,CAAC,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAC;QAC5D,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC,EAAE,CAAC,CAAC;QAC/D,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACxD,MAAM,IAAI,GAAG,KAAK,CAAC,KAAK,EAAE,CAAC;QAC3B,MAAM,KAAK,EAAE,CAAC;QAEd,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC9C,KAAK,MAAM,CAAC,IAAI,KAAK,CAAC,MAAM,EAAE;YAAE,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACrD,MAAM,IAAI,CAAC;QACX,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC1B,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;QACpD,IAAI,MAAM,GAAG,CAAC,CAAC;QACf,IAAI,SAAS,GAAG,CAAC,CAAC;QAClB,MAAM,KAAK,GAAG,IAAI,GAAG,EAAuC,CAAC;QAC7D,MAAM,MAAM,GAAgB,KAAK,EAAE,CAAC,EAAE,EAAE;YACtC,MAAM,EAAE,CAAC;YACT,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;YACxC,MAAM,CAAC,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,QAAQ,EAAE,CAAC;YAC5C,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;YACvB,MAAM,CAAC,CAAC,OAAO,CAAC;YAChB,MAAM,EAAE,CAAC;YACT,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,WAAW,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,EAAE,CAAC;QAC9E,CAAC,CAAC;QACF,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC,EAAE,CAAC,CAAC;QAC/D,KAAK,CAAC,QAAQ,CACZ,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC,EAAE,IAAI,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC,CAAC,CACnF,CAAC;QACF,MAAM,IAAI,GAAG,KAAK,CAAC,KAAK,EAAE,CAAC;QAC3B,MAAM,KAAK,EAAE,CAAC;QAEd,gEAAgE;QAChE,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC9C,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC;QACpC,MAAM,KAAK,EAAE,CAAC;QACd,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS;QACxD,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC;QACpC,MAAM,IAAI,CAAC;QACX,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC1B,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oEAAoE,EAAE,KAAK,IAAI,EAAE;QAClF,MAAM,MAAM,GAAG,UAAU,EAAE,CAAC;QAC5B,MAAM,MAAM,GAAgB,KAAK,EAAE,CAAC,EAAE,EAAE;YACtC,IAAI,CAAC,CAAC,MAAM,KAAK,IAAI;gBAAE,MAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,CAAC;YACjD,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,WAAW,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,EAAE,CAAC;QAC9E,CAAC,CAAC;QACF,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;QACvD,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,CAAC,IAAI,EAAE,EAAE,SAAS,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QACtE,MAAM,KAAK,CAAC,KAAK,EAAE,CAAC;QAEpB,MAAM,CAAC,QAAQ,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC7C,MAAM,CAAC,QAAQ,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC9C,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,mBAAmB;QACjE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,oBAAoB,CACxC,MAAM,CAAC,gBAAgB,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAChE,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACnE,MAAM,MAAM,GAAG,UAAU,EAAE,CAAC;QAC5B,MAAM,MAAM,GAAgB,KAAK,IAAI,EAAE,CAAC,CAAC;YACvC,IAAI,EAAE,mBAAmB;YACzB,KAAK,EAAE,CAAC;YACR,KAAK,EAAE,EAAE,WAAW,EAAE,SAAS,EAAE,YAAY,EAAE,CAAC,EAAE;SACnD,CAAC,CAAC;QACH,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;QACvD,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,SAAS,EAAE,EAAE,KAAK,EAAE,aAAa,EAAE,WAAW,EAAE,aAAa,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QAC9F,MAAM,KAAK,CAAC,KAAK,EAAE,CAAC;QAEpB,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,oBAAoB,CACxC,MAAM,CAAC,gBAAgB,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,IAAI,EAAE,gBAAgB,EAAE,CAAC,CACxE,CAAC;QACF,iCAAiC;QACjC,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,WAAW,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;QAChE,MAAM,MAAM,GAAgB,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;YACxC,IAAI,EAAE,OAAO,CAAC,CAAC,MAAM,EAAE;YACvB,KAAK,EAAE,CAAC;YACR,KAAK,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE;SAC5C,CAAC,CAAC;QACH,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;QAC/C,MAAM,EAAE,GAAG,KAAK,CAAC,cAAc,CAAC,yBAAyB,CAAC,CAAC;QAC3D,MAAM,KAAK,CAAC,KAAK,EAAE,CAAC;QACpB,MAAM,KAAK,GAAG,KAAK,CAAC,QAAQ,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,EAAE,CAAC,CAAC;QACnE,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACnC,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,sBAAsB;IACjE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;QAC7D,MAAM,KAAK,GAAG,IAAI,GAAG,EAAuC,CAAC;QAC7D,MAAM,MAAM,GAAgB,KAAK,EAAE,CAAC,EAAE,EAAE;YACtC,MAAM,CAAC,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,QAAQ,EAAE,CAAC;YAC5C,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;YACvB,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;gBAC1C,CAAC,CAAC,MAAM,EAAE,gBAAgB,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;gBACxE,KAAK,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC;YACvC,CAAC,CAAC,CAAC;YACH,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,WAAW,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,EAAE,CAAC;QAC9E,CAAC,CAAC;QACF,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC,EAAE,CAAC,CAAC;QAC/D,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,EAAE,KAAK,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,KAAK,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QACvF,MAAM,IAAI,GAAG,KAAK,CAAC,KAAK,EAAE,CAAC;QAC3B,MAAM,KAAK,EAAE,CAAC;QACd,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAE9C,KAAK,CAAC,SAAS,EAAE,CAAC;QAClB,MAAM,IAAI,CAAC;QACX,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC3C,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACpF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;QACnD,MAAM,MAAM,GAAgB,KAAK,IAAI,EAAE,CAAC,CAAC;YACvC,IAAI,EAAE,IAAI;YACV,KAAK,EAAE,CAAC;YACR,KAAK,EAAE,EAAE,WAAW,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE;SAC3C,CAAC,CAAC;QACH,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;QAC/C,MAAM,IAAI,GAAa,EAAE,CAAC;QAC1B,KAAK,CAAC,EAAE,CAAC,QAAQ,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;QAC5D,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QAClC,MAAM,KAAK,CAAC,KAAK,EAAE,CAAC;QACpB,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"pipeline.test.d.ts","sourceRoot":"","sources":["../../../src/agent/__tests__/pipeline.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,226 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Turn pipeline — proves the orchestration: evaluate → enhance → route → execute
|
|
3
|
+
* → judge → revise, the trace it assembles, model selection (evaluator ⟂ router,
|
|
4
|
+
* manual pin), the auto-revise loop and its guards (readOnly, maxReviseRounds),
|
|
5
|
+
* bare mode, stage emission, usage accumulation, and the gateway-key guard. Every
|
|
6
|
+
* stage module is mocked, so no gateway or filesystem is touched.
|
|
7
|
+
*/
|
|
8
|
+
import { describe, it, expect, beforeEach, vi } from "vitest";
|
|
9
|
+
vi.mock("../../lib/config.js", () => ({ readConfig: () => ({}) }));
|
|
10
|
+
vi.mock("../env.js", () => {
|
|
11
|
+
class MissingGatewayKeyError extends Error {
|
|
12
|
+
constructor() {
|
|
13
|
+
super("missing");
|
|
14
|
+
this.name = "MissingGatewayKeyError";
|
|
15
|
+
}
|
|
16
|
+
}
|
|
17
|
+
return { ensureGatewayKey: vi.fn(() => "key"), MissingGatewayKeyError };
|
|
18
|
+
});
|
|
19
|
+
vi.mock("../loop.js", () => ({ runAgent: vi.fn() }));
|
|
20
|
+
vi.mock("../evaluator.js", () => ({ evaluatePrompt: vi.fn() }));
|
|
21
|
+
vi.mock("../judge.js", () => ({
|
|
22
|
+
judgeCompleteness: vi.fn(),
|
|
23
|
+
buildRevisionPrompt: vi.fn(() => "REVISION PROMPT"),
|
|
24
|
+
}));
|
|
25
|
+
vi.mock("../prompt-enhancer.js", () => ({ enhancePrompt: vi.fn() }));
|
|
26
|
+
import { runTurn, MissingGatewayKeyError } from "../pipeline.js";
|
|
27
|
+
import { ensureGatewayKey } from "../env.js";
|
|
28
|
+
import { runAgent } from "../loop.js";
|
|
29
|
+
import { evaluatePrompt } from "../evaluator.js";
|
|
30
|
+
import { judgeCompleteness, buildRevisionPrompt } from "../judge.js";
|
|
31
|
+
import { enhancePrompt } from "../prompt-enhancer.js";
|
|
32
|
+
const mockEnsure = ensureGatewayKey;
|
|
33
|
+
const mockRun = runAgent;
|
|
34
|
+
const mockEval = evaluatePrompt;
|
|
35
|
+
const mockJudge = judgeCompleteness;
|
|
36
|
+
const mockEnhance = enhancePrompt;
|
|
37
|
+
const mockBuildRevision = buildRevisionPrompt;
|
|
38
|
+
function evaluation(over = {}) {
|
|
39
|
+
return {
|
|
40
|
+
completeness: 70,
|
|
41
|
+
complexity: 50,
|
|
42
|
+
recommendedTier: "balanced",
|
|
43
|
+
missing: [],
|
|
44
|
+
contextQueries: ["Foo"],
|
|
45
|
+
refinedPrompt: "refined ask",
|
|
46
|
+
removed: [],
|
|
47
|
+
reasoning: "r",
|
|
48
|
+
fallback: false,
|
|
49
|
+
model: "anthropic/claude-haiku-4.5",
|
|
50
|
+
usage: { inputTokens: 5, outputTokens: 1, costUsd: 0.001 },
|
|
51
|
+
...over,
|
|
52
|
+
};
|
|
53
|
+
}
|
|
54
|
+
function verdict(over = {}) {
|
|
55
|
+
return {
|
|
56
|
+
complete: true,
|
|
57
|
+
confidence: 90,
|
|
58
|
+
findings: [],
|
|
59
|
+
remainingWork: [],
|
|
60
|
+
reasoning: "looks done",
|
|
61
|
+
model: "anthropic/claude-opus-4.8",
|
|
62
|
+
fallback: false,
|
|
63
|
+
usage: { inputTokens: 3, outputTokens: 1, costUsd: 0.002 },
|
|
64
|
+
...over,
|
|
65
|
+
};
|
|
66
|
+
}
|
|
67
|
+
beforeEach(() => {
|
|
68
|
+
delete process.env["OXAGEN_LLM_FAST"];
|
|
69
|
+
delete process.env["OXAGEN_LLM_BALANCED"];
|
|
70
|
+
delete process.env["OXAGEN_LLM_PRECISE"];
|
|
71
|
+
mockEnsure.mockReturnValue("key");
|
|
72
|
+
mockEval.mockResolvedValue(evaluation());
|
|
73
|
+
mockEnhance.mockResolvedValue({
|
|
74
|
+
prompt: "refined ask\n\n## context",
|
|
75
|
+
context: "## context",
|
|
76
|
+
resolved: ["Foo"],
|
|
77
|
+
lessons: [],
|
|
78
|
+
});
|
|
79
|
+
mockJudge.mockResolvedValue(verdict());
|
|
80
|
+
mockBuildRevision.mockReturnValue("REVISION PROMPT");
|
|
81
|
+
// Default executor: streams text, touches a file, returns a message + usage.
|
|
82
|
+
mockRun.mockImplementation(async (opts) => {
|
|
83
|
+
opts.onToolCall?.("write_file", { path: "src/x.ts" });
|
|
84
|
+
opts.onText?.("hi");
|
|
85
|
+
return {
|
|
86
|
+
text: "done",
|
|
87
|
+
steps: 2,
|
|
88
|
+
messages: [{ role: "assistant", content: "done" }],
|
|
89
|
+
usage: { inputTokens: 100, outputTokens: 50 },
|
|
90
|
+
};
|
|
91
|
+
});
|
|
92
|
+
});
|
|
93
|
+
describe("runTurn — happy path", () => {
|
|
94
|
+
it("runs all stages, selects the model, and assembles a complete trace", async () => {
|
|
95
|
+
const stages = [];
|
|
96
|
+
const res = await runTurn({ prompt: "fix Foo", cwd: "/x", onStage: (s) => stages.push(s) });
|
|
97
|
+
expect(mockEval).toHaveBeenCalledOnce();
|
|
98
|
+
expect(mockEnhance).toHaveBeenCalledOnce();
|
|
99
|
+
expect(mockRun).toHaveBeenCalledOnce();
|
|
100
|
+
expect(mockJudge).toHaveBeenCalledOnce();
|
|
101
|
+
// The executor saw the enhanced prompt on the balanced (sonnet) model.
|
|
102
|
+
expect(mockRun.mock.calls[0]?.[0].prompt).toContain("## context");
|
|
103
|
+
expect(mockRun.mock.calls[0]?.[0].model).toContain("sonnet");
|
|
104
|
+
const t = res.trace;
|
|
105
|
+
expect(t.originalPrompt).toBe("fix Foo");
|
|
106
|
+
expect(t.selectedTier).toBe("balanced");
|
|
107
|
+
expect(t.selectionRationale).toContain("evaluator recommended");
|
|
108
|
+
expect(t.judgeRounds).toHaveLength(1);
|
|
109
|
+
expect(t.finalComplete).toBe(true);
|
|
110
|
+
expect(t.filesTouched).toEqual(["src/x.ts"]);
|
|
111
|
+
expect(t.enhancement.source).toBe("code-graph");
|
|
112
|
+
// Usage accumulates evaluator + executor + judge token counts.
|
|
113
|
+
expect(t.usage.inputTokens).toBe(108);
|
|
114
|
+
expect(t.usage.outputTokens).toBe(52);
|
|
115
|
+
expect(t.usage.costUsd).toBeGreaterThan(0);
|
|
116
|
+
// Stages are emitted in order.
|
|
117
|
+
expect(stages.map((s) => s.kind)).toEqual([
|
|
118
|
+
"evaluate",
|
|
119
|
+
"enhance",
|
|
120
|
+
"route",
|
|
121
|
+
"execute",
|
|
122
|
+
"judge",
|
|
123
|
+
"complete",
|
|
124
|
+
]);
|
|
125
|
+
});
|
|
126
|
+
it("forwards the evaluator's context queries to the enhancer", async () => {
|
|
127
|
+
await runTurn({ prompt: "fix Foo", cwd: "/x" });
|
|
128
|
+
expect(mockEnhance.mock.calls[0]?.[0].extraQueries).toEqual(["Foo"]);
|
|
129
|
+
});
|
|
130
|
+
});
|
|
131
|
+
describe("runTurn — completeness revise loop", () => {
|
|
132
|
+
it("sends the agent back when judged incomplete, then re-judges", async () => {
|
|
133
|
+
mockJudge
|
|
134
|
+
.mockResolvedValueOnce(verdict({ complete: false, findings: ["no test"] }))
|
|
135
|
+
.mockResolvedValueOnce(verdict({ complete: true }));
|
|
136
|
+
const stages = [];
|
|
137
|
+
const res = await runTurn({ prompt: "add x", cwd: "/x", onStage: (s) => stages.push(s) });
|
|
138
|
+
expect(mockRun).toHaveBeenCalledTimes(2);
|
|
139
|
+
expect(mockBuildRevision).toHaveBeenCalledOnce();
|
|
140
|
+
// The second execution runs the revision prompt.
|
|
141
|
+
expect(mockRun.mock.calls[1]?.[0].prompt).toBe("REVISION PROMPT");
|
|
142
|
+
expect(res.trace.judgeRounds).toHaveLength(2);
|
|
143
|
+
expect(res.trace.finalComplete).toBe(true);
|
|
144
|
+
expect(stages.some((s) => s.kind === "revise")).toBe(true);
|
|
145
|
+
});
|
|
146
|
+
it("does not revise in read-only mode", async () => {
|
|
147
|
+
mockJudge.mockResolvedValue(verdict({ complete: false, findings: ["x"] }));
|
|
148
|
+
const res = await runTurn({ prompt: "add x", cwd: "/x", readOnly: true });
|
|
149
|
+
expect(mockRun).toHaveBeenCalledOnce();
|
|
150
|
+
expect(res.trace.finalComplete).toBe(false);
|
|
151
|
+
});
|
|
152
|
+
it("does not revise when maxReviseRounds is 0", async () => {
|
|
153
|
+
mockJudge.mockResolvedValue(verdict({ complete: false, findings: ["x"] }));
|
|
154
|
+
const res = await runTurn({ prompt: "add x", cwd: "/x", maxReviseRounds: 0 });
|
|
155
|
+
expect(mockRun).toHaveBeenCalledOnce();
|
|
156
|
+
expect(res.trace.judgeRounds).toHaveLength(1);
|
|
157
|
+
});
|
|
158
|
+
});
|
|
159
|
+
describe("runTurn — model selection", () => {
|
|
160
|
+
it("pins a manually overridden model and skips routing", async () => {
|
|
161
|
+
const res = await runTurn({ prompt: "x", cwd: "/x", model: "openai/gpt-5" });
|
|
162
|
+
expect(res.trace.selectedModel).toBe("openai/gpt-5");
|
|
163
|
+
expect(res.trace.selectionRationale).toBe("pinned model");
|
|
164
|
+
expect(mockRun.mock.calls[0]?.[0].model).toBe("openai/gpt-5");
|
|
165
|
+
});
|
|
166
|
+
it("escalates above the evaluator on high-stakes signals", async () => {
|
|
167
|
+
mockEval.mockResolvedValue(evaluation({ recommendedTier: "fast", refinedPrompt: "implement the stripe billing refund" }));
|
|
168
|
+
const res = await runTurn({ prompt: "billing", cwd: "/x" });
|
|
169
|
+
expect(res.trace.selectedTier).toBe("precise");
|
|
170
|
+
expect(res.trace.selectedModel).toContain("opus");
|
|
171
|
+
expect(res.trace.selectionRationale).toContain("router escalated");
|
|
172
|
+
});
|
|
173
|
+
});
|
|
174
|
+
describe("runTurn — bare mode", () => {
|
|
175
|
+
it("skips eval/enhance/judge and still produces a trace", async () => {
|
|
176
|
+
const res = await runTurn({ prompt: "x", cwd: "/x", bare: true });
|
|
177
|
+
expect(mockEval).not.toHaveBeenCalled();
|
|
178
|
+
expect(mockEnhance).not.toHaveBeenCalled();
|
|
179
|
+
expect(mockJudge).not.toHaveBeenCalled();
|
|
180
|
+
expect(mockRun).toHaveBeenCalledOnce();
|
|
181
|
+
expect(res.trace.judgeRounds).toHaveLength(0);
|
|
182
|
+
expect(res.trace.selectionRationale).toBe("bare mode (no routing)");
|
|
183
|
+
expect(res.trace.finalComplete).toBe(true);
|
|
184
|
+
});
|
|
185
|
+
});
|
|
186
|
+
describe("runTurn — fleet memory", () => {
|
|
187
|
+
it("records a gotcha lesson when the turn had to be revised", async () => {
|
|
188
|
+
const record = vi.fn();
|
|
189
|
+
const fleetMemory = { record, recall: () => [], all: () => [] };
|
|
190
|
+
mockJudge
|
|
191
|
+
.mockResolvedValueOnce(verdict({ complete: false, findings: ["no test"] }))
|
|
192
|
+
.mockResolvedValueOnce(verdict({ complete: true }));
|
|
193
|
+
await runTurn({ prompt: "add x", cwd: "/x", fleetMemory });
|
|
194
|
+
expect(record).toHaveBeenCalledOnce();
|
|
195
|
+
expect(record.mock.calls[0]?.[0].kind).toBe("gotcha");
|
|
196
|
+
expect(record.mock.calls[0]?.[0].weight).toBe("high");
|
|
197
|
+
});
|
|
198
|
+
it("records a routine-change lesson when files changed without revision", async () => {
|
|
199
|
+
const record = vi.fn();
|
|
200
|
+
const fleetMemory = { record, recall: () => [], all: () => [] };
|
|
201
|
+
await runTurn({ prompt: "tweak x", cwd: "/x", fleetMemory });
|
|
202
|
+
expect(record).toHaveBeenCalledOnce();
|
|
203
|
+
expect(record.mock.calls[0]?.[0].kind).toBe("routine-change");
|
|
204
|
+
expect(record.mock.calls[0]?.[0].files).toContain("src/x.ts");
|
|
205
|
+
});
|
|
206
|
+
it("records nothing when no files changed and no revision was needed", async () => {
|
|
207
|
+
const record = vi.fn();
|
|
208
|
+
const fleetMemory = { record, recall: () => [], all: () => [] };
|
|
209
|
+
mockRun.mockImplementationOnce(async () => ({
|
|
210
|
+
text: "explained",
|
|
211
|
+
steps: 1,
|
|
212
|
+
messages: [],
|
|
213
|
+
usage: { inputTokens: 1, outputTokens: 1 },
|
|
214
|
+
}));
|
|
215
|
+
await runTurn({ prompt: "explain x", cwd: "/x", fleetMemory });
|
|
216
|
+
expect(record).not.toHaveBeenCalled();
|
|
217
|
+
});
|
|
218
|
+
});
|
|
219
|
+
describe("runTurn — guards", () => {
|
|
220
|
+
it("throws MissingGatewayKeyError when no gateway key resolves", async () => {
|
|
221
|
+
mockEnsure.mockReturnValue(null);
|
|
222
|
+
await expect(runTurn({ prompt: "x", cwd: "/x" })).rejects.toBeInstanceOf(MissingGatewayKeyError);
|
|
223
|
+
expect(mockRun).not.toHaveBeenCalled();
|
|
224
|
+
});
|
|
225
|
+
});
|
|
226
|
+
//# sourceMappingURL=pipeline.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"pipeline.test.js","sourceRoot":"","sources":["../../../src/agent/__tests__/pipeline.test.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AACH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE,EAAa,MAAM,QAAQ,CAAC;AAGzE,EAAE,CAAC,IAAI,CAAC,qBAAqB,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,UAAU,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;AACnE,EAAE,CAAC,IAAI,CAAC,WAAW,EAAE,GAAG,EAAE;IACxB,MAAM,sBAAuB,SAAQ,KAAK;QACxC;YACE,KAAK,CAAC,SAAS,CAAC,CAAC;YACjB,IAAI,CAAC,IAAI,GAAG,wBAAwB,CAAC;QACvC,CAAC;KACF;IACD,OAAO,EAAE,gBAAgB,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,EAAE,sBAAsB,EAAE,CAAC;AAC1E,CAAC,CAAC,CAAC;AACH,EAAE,CAAC,IAAI,CAAC,YAAY,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;AACrD,EAAE,CAAC,IAAI,CAAC,iBAAiB,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;AAChE,EAAE,CAAC,IAAI,CAAC,aAAa,EAAE,GAAG,EAAE,CAAC,CAAC;IAC5B,iBAAiB,EAAE,EAAE,CAAC,EAAE,EAAE;IAC1B,mBAAmB,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,iBAAiB,CAAC;CACpD,CAAC,CAAC,CAAC;AACJ,EAAE,CAAC,IAAI,CAAC,uBAAuB,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,aAAa,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;AAErE,OAAO,EAAE,OAAO,EAAE,sBAAsB,EAAE,MAAM,gBAAgB,CAAC;AACjE,OAAO,EAAE,gBAAgB,EAAE,MAAM,WAAW,CAAC;AAC7C,OAAO,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAC;AACtC,OAAO,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AACjD,OAAO,EAAE,iBAAiB,EAAE,mBAAmB,EAAE,MAAM,aAAa,CAAC;AACrE,OAAO,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AAEtD,MAAM,UAAU,GAAG,gBAAmC,CAAC;AACvD,MAAM,OAAO,GAAG,QAA2B,CAAC;AAC5C,MAAM,QAAQ,GAAG,cAAiC,CAAC;AACnD,MAAM,SAAS,GAAG,iBAAoC,CAAC;AACvD,MAAM,WAAW,GAAG,aAAgC,CAAC;AACrD,MAAM,iBAAiB,GAAG,mBAAsC,CAAC;AAEjE,SAAS,UAAU,CAAC,OAAkC,EAAE;IACtD,OAAO;QACL,YAAY,EAAE,EAAE;QAChB,UAAU,EAAE,EAAE;QACd,eAAe,EAAE,UAAU;QAC3B,OAAO,EAAE,EAAE;QACX,cAAc,EAAE,CAAC,KAAK,CAAC;QACvB,aAAa,EAAE,aAAa;QAC5B,OAAO,EAAE,EAAE;QACX,SAAS,EAAE,GAAG;QACd,QAAQ,EAAE,KAAK;QACf,KAAK,EAAE,4BAA4B;QACnC,KAAK,EAAE,EAAE,WAAW,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE;QAC1D,GAAG,IAAI;KACR,CAAC;AACJ,CAAC;AAED,SAAS,OAAO,CAAC,OAA8B,EAAE;IAC/C,OAAO;QACL,QAAQ,EAAE,IAAI;QACd,UAAU,EAAE,EAAE;QACd,QAAQ,EAAE,EAAE;QACZ,aAAa,EAAE,EAAE;QACjB,SAAS,EAAE,YAAY;QACvB,KAAK,EAAE,2BAA2B;QAClC,QAAQ,EAAE,KAAK;QACf,KAAK,EAAE,EAAE,WAAW,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE;QAC1D,GAAG,IAAI;KACR,CAAC;AACJ,CAAC;AAED,UAAU,CAAC,GAAG,EAAE;IACd,OAAO,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAC;IACtC,OAAO,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAC;IAC1C,OAAO,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC;IACzC,UAAU,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;IAClC,QAAQ,CAAC,iBAAiB,CAAC,UAAU,EAAE,CAAC,CAAC;IACzC,WAAW,CAAC,iBAAiB,CAAC;QAC5B,MAAM,EAAE,2BAA2B;QACnC,OAAO,EAAE,YAAY;QACrB,QAAQ,EAAE,CAAC,KAAK,CAAC;QACjB,OAAO,EAAE,EAAE;KACZ,CAAC,CAAC;IACH,SAAS,CAAC,iBAAiB,CAAC,OAAO,EAAE,CAAC,CAAC;IACvC,iBAAiB,CAAC,eAAe,CAAC,iBAAiB,CAAC,CAAC;IACrD,6EAA6E;IAC7E,OAAO,CAAC,kBAAkB,CAAC,KAAK,EAAE,IAA6B,EAAE,EAAE;QAChE,IAAI,CAAC,UAA8C,EAAE,CAAC,YAAY,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,CAAC;QAC1F,IAAI,CAAC,MAA8B,EAAE,CAAC,IAAI,CAAC,CAAC;QAC7C,OAAO;YACL,IAAI,EAAE,MAAM;YACZ,KAAK,EAAE,CAAC;YACR,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;YAClD,KAAK,EAAE,EAAE,WAAW,EAAE,GAAG,EAAE,YAAY,EAAE,EAAE,EAAE;SAC9C,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;IACpC,EAAE,CAAC,oEAAoE,EAAE,KAAK,IAAI,EAAE;QAClF,MAAM,MAAM,GAAiB,EAAE,CAAC;QAChC,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAE5F,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,EAAE,CAAC;QACxC,MAAM,CAAC,WAAW,CAAC,CAAC,oBAAoB,EAAE,CAAC;QAC3C,MAAM,CAAC,OAAO,CAAC,CAAC,oBAAoB,EAAE,CAAC;QACvC,MAAM,CAAC,SAAS,CAAC,CAAC,oBAAoB,EAAE,CAAC;QAEzC,uEAAuE;QACvE,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;QAClE,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QAE7D,MAAM,CAAC,GAAG,GAAG,CAAC,KAAK,CAAC;QACpB,MAAM,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACzC,MAAM,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACxC,MAAM,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC,SAAS,CAAC,uBAAuB,CAAC,CAAC;QAChE,MAAM,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACtC,MAAM,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACnC,MAAM,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,OAAO,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC;QAC7C,MAAM,CAAC,CAAC,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAChD,+DAA+D;QAC/D,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACtC,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACtC,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAE3C,+BAA+B;QAC/B,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC;YACxC,UAAU;YACV,SAAS;YACT,OAAO;YACP,SAAS;YACT,OAAO;YACP,UAAU;SACX,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;QACxE,MAAM,OAAO,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,EAAE,IAAI,EAAE,CAAC,CAAC;QAChD,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,OAAO,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;IACvE,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,oCAAoC,EAAE,GAAG,EAAE;IAClD,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;QAC3E,SAAS;aACN,qBAAqB,CAAC,OAAO,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;aAC1E,qBAAqB,CAAC,OAAO,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;QAEtD,MAAM,MAAM,GAAiB,EAAE,CAAC;QAChC,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAE1F,MAAM,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACzC,MAAM,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,EAAE,CAAC;QACjD,iDAAiD;QACjD,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAClE,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC9C,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC3C,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC7D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;QACjD,SAAS,CAAC,iBAAiB,CAAC,OAAO,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;QAC3E,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;QAC1E,MAAM,CAAC,OAAO,CAAC,CAAC,oBAAoB,EAAE,CAAC;QACvC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;QACzD,SAAS,CAAC,iBAAiB,CAAC,OAAO,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;QAC3E,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC,EAAE,CAAC,CAAC;QAC9E,MAAM,CAAC,OAAO,CAAC,CAAC,oBAAoB,EAAE,CAAC;QACvC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,2BAA2B,EAAE,GAAG,EAAE;IACzC,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QAClE,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,GAAG,EAAE,IAAI,EAAE,KAAK,EAAE,cAAc,EAAE,CAAC,CAAC;QAC7E,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACrD,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAC1D,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IAChE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;QACpE,QAAQ,CAAC,iBAAiB,CACxB,UAAU,CAAC,EAAE,eAAe,EAAE,MAAM,EAAE,aAAa,EAAE,qCAAqC,EAAE,CAAC,CAC9F,CAAC;QACF,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,EAAE,IAAI,EAAE,CAAC,CAAC;QAC5D,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC/C,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QAClD,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC;IACrE,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACnC,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACnE,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,GAAG,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QAClE,MAAM,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACxC,MAAM,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC3C,MAAM,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACzC,MAAM,CAAC,OAAO,CAAC,CAAC,oBAAoB,EAAE,CAAC;QACvC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC9C,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;QACpE,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;IACtC,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QACvE,MAAM,MAAM,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACvB,MAAM,WAAW,GAAG,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAChE,SAAS;aACN,qBAAqB,CAAC,OAAO,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;aAC1E,qBAAqB,CAAC,OAAO,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;QACtD,MAAM,OAAO,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC,CAAC;QAC3D,MAAM,CAAC,MAAM,CAAC,CAAC,oBAAoB,EAAE,CAAC;QACtC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACtD,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;QACnF,MAAM,MAAM,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACvB,MAAM,WAAW,GAAG,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAChE,MAAM,OAAO,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC,CAAC;QAC7D,MAAM,CAAC,MAAM,CAAC,CAAC,oBAAoB,EAAE,CAAC;QACtC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QAC9D,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;IAChE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;QAChF,MAAM,MAAM,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACvB,MAAM,WAAW,GAAG,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAChE,OAAO,CAAC,sBAAsB,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;YAC1C,IAAI,EAAE,WAAW;YACjB,KAAK,EAAE,CAAC;YACR,QAAQ,EAAE,EAAE;YACZ,KAAK,EAAE,EAAE,WAAW,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE;SAC3C,CAAC,CAAC,CAAC;QACJ,MAAM,OAAO,CAAC,EAAE,MAAM,EAAE,WAAW,EAAE,GAAG,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC,CAAC;QAC/D,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IACxC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;QAC1E,UAAU,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QACjC,MAAM,MAAM,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,GAAG,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,cAAc,CACtE,sBAAsB,CACvB,CAAC;QACF,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IACzC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"planner.test.d.ts","sourceRoot":"","sources":["../../../src/agent/__tests__/planner.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,98 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Task planner — proves a goal decomposes into a valid plan: unique task ids,
|
|
3
|
+
* dependencies pruned to real tasks, and each task's tier reconciled UP with the
|
|
4
|
+
* cost router so a high-stakes task is never under-spent. The model call and the
|
|
5
|
+
* prompt enhancer are mocked, so no gateway or filesystem is touched.
|
|
6
|
+
*/
|
|
7
|
+
import { describe, it, expect, beforeEach, vi } from "vitest";
|
|
8
|
+
import { tmpdir } from "node:os";
|
|
9
|
+
vi.mock("ai", () => ({ generateObject: vi.fn() }));
|
|
10
|
+
vi.mock("../../lib/config.js", () => ({ readConfig: () => ({}) }));
|
|
11
|
+
vi.mock("../prompt-enhancer.js", () => ({
|
|
12
|
+
enhancePrompt: async ({ prompt }) => ({
|
|
13
|
+
prompt,
|
|
14
|
+
context: "",
|
|
15
|
+
resolved: [],
|
|
16
|
+
lessons: [],
|
|
17
|
+
}),
|
|
18
|
+
}));
|
|
19
|
+
import { generateObject } from "ai";
|
|
20
|
+
import { planTasks } from "../planner.js";
|
|
21
|
+
import { MissingGatewayKeyError } from "../env.js";
|
|
22
|
+
const mockGen = generateObject;
|
|
23
|
+
function resolveWith(tasks) {
|
|
24
|
+
mockGen.mockResolvedValueOnce({
|
|
25
|
+
object: { tasks },
|
|
26
|
+
usage: { inputTokens: 10, outputTokens: 20 },
|
|
27
|
+
});
|
|
28
|
+
}
|
|
29
|
+
beforeEach(() => {
|
|
30
|
+
process.env["AI_GATEWAY_API_KEY"] = "test-key";
|
|
31
|
+
delete process.env["OXAGEN_LLM_FAST"];
|
|
32
|
+
delete process.env["OXAGEN_LLM_BALANCED"];
|
|
33
|
+
delete process.env["OXAGEN_LLM_PRECISE"];
|
|
34
|
+
delete process.env["OXAGEN_MODEL"];
|
|
35
|
+
});
|
|
36
|
+
describe("planTasks", () => {
|
|
37
|
+
it("builds a draft plan from the model's tasks", async () => {
|
|
38
|
+
resolveWith([
|
|
39
|
+
{ id: "a", title: "Do A", description: "first", dependsOn: [], files: ["a.ts"], tier: "fast" },
|
|
40
|
+
{ id: "b", title: "Do B", description: "second", dependsOn: ["a"], files: ["b.ts"], tier: "balanced" },
|
|
41
|
+
]);
|
|
42
|
+
const plan = await planTasks({ goal: "do the thing", cwd: tmpdir() });
|
|
43
|
+
expect(plan.status).toBe("draft");
|
|
44
|
+
expect(plan.goal).toBe("do the thing");
|
|
45
|
+
expect(plan.tasks).toHaveLength(2);
|
|
46
|
+
expect(plan.tasks.every((t) => t.status === "queued")).toBe(true);
|
|
47
|
+
expect(plan.tasks[1]?.dependsOn).toEqual(["a"]);
|
|
48
|
+
expect(plan.tasks[0]?.model).toContain("haiku");
|
|
49
|
+
expect(plan.tasks[0]?.usage).toEqual({ inputTokens: 0, outputTokens: 0, costUsd: 0 });
|
|
50
|
+
});
|
|
51
|
+
it("escalates a task's tier when its description is high-stakes", async () => {
|
|
52
|
+
resolveWith([
|
|
53
|
+
{
|
|
54
|
+
id: "pay",
|
|
55
|
+
title: "Add refund",
|
|
56
|
+
description: "implement the stripe billing refund flow",
|
|
57
|
+
dependsOn: [],
|
|
58
|
+
files: ["pay.ts"],
|
|
59
|
+
tier: "fast", // model under-estimated
|
|
60
|
+
},
|
|
61
|
+
]);
|
|
62
|
+
const plan = await planTasks({ goal: "billing", cwd: tmpdir() });
|
|
63
|
+
expect(plan.tasks[0]?.tier).toBe("precise");
|
|
64
|
+
expect(plan.tasks[0]?.model).toContain("opus");
|
|
65
|
+
});
|
|
66
|
+
it("never downgrades a precise task the model already flagged", async () => {
|
|
67
|
+
resolveWith([
|
|
68
|
+
{ id: "x", title: "tiny", description: "rename a local var", dependsOn: [], files: [], tier: "precise" },
|
|
69
|
+
]);
|
|
70
|
+
const plan = await planTasks({ goal: "g", cwd: tmpdir() });
|
|
71
|
+
expect(plan.tasks[0]?.tier).toBe("precise");
|
|
72
|
+
});
|
|
73
|
+
it("de-duplicates repeated ids", async () => {
|
|
74
|
+
resolveWith([
|
|
75
|
+
{ id: "x", title: "One", description: "d", dependsOn: [], files: [], tier: "fast" },
|
|
76
|
+
{ id: "x", title: "Two", description: "d", dependsOn: [], files: [], tier: "fast" },
|
|
77
|
+
]);
|
|
78
|
+
const plan = await planTasks({ goal: "g", cwd: tmpdir() });
|
|
79
|
+
const ids = plan.tasks.map((t) => t.id);
|
|
80
|
+
expect(new Set(ids).size).toBe(2);
|
|
81
|
+
expect(ids).toContain("x");
|
|
82
|
+
});
|
|
83
|
+
it("drops dependencies that point at nonexistent or self ids", async () => {
|
|
84
|
+
resolveWith([
|
|
85
|
+
{ id: "a", title: "A", description: "d", dependsOn: ["ghost", "b", "a"], files: [], tier: "fast" },
|
|
86
|
+
{ id: "b", title: "B", description: "d", dependsOn: [], files: [], tier: "fast" },
|
|
87
|
+
]);
|
|
88
|
+
const plan = await planTasks({ goal: "g", cwd: tmpdir() });
|
|
89
|
+
expect(plan.tasks[0]?.dependsOn).toEqual(["b"]);
|
|
90
|
+
});
|
|
91
|
+
it("throws MissingGatewayKeyError when no gateway key is resolvable", async () => {
|
|
92
|
+
delete process.env["AI_GATEWAY_API_KEY"];
|
|
93
|
+
// cwd in the OS temp dir → no .env.local up the tree, and config is mocked empty.
|
|
94
|
+
await expect(planTasks({ goal: "g", cwd: tmpdir() })).rejects.toBeInstanceOf(MissingGatewayKeyError);
|
|
95
|
+
expect(mockGen).not.toHaveBeenCalled();
|
|
96
|
+
});
|
|
97
|
+
});
|
|
98
|
+
//# sourceMappingURL=planner.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"planner.test.js","sourceRoot":"","sources":["../../../src/agent/__tests__/planner.test.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AACH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE,EAAa,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AAEjC,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;AACnD,EAAE,CAAC,IAAI,CAAC,qBAAqB,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,UAAU,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;AACnE,EAAE,CAAC,IAAI,CAAC,uBAAuB,EAAE,GAAG,EAAE,CAAC,CAAC;IACtC,aAAa,EAAE,KAAK,EAAE,EAAE,MAAM,EAAsB,EAAE,EAAE,CAAC,CAAC;QACxD,MAAM;QACN,OAAO,EAAE,EAAE;QACX,QAAQ,EAAE,EAAE;QACZ,OAAO,EAAE,EAAE;KACZ,CAAC;CACH,CAAC,CAAC,CAAC;AAEJ,OAAO,EAAE,cAAc,EAAE,MAAM,IAAI,CAAC;AACpC,OAAO,EAAE,SAAS,EAAE,MAAM,eAAe,CAAC;AAC1C,OAAO,EAAE,sBAAsB,EAAE,MAAM,WAAW,CAAC;AAEnD,MAAM,OAAO,GAAG,cAAiC,CAAC;AAWlD,SAAS,WAAW,CAAC,KAAgB;IACnC,OAAO,CAAC,qBAAqB,CAAC;QAC5B,MAAM,EAAE,EAAE,KAAK,EAAE;QACjB,KAAK,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE;KAC7C,CAAC,CAAC;AACL,CAAC;AAED,UAAU,CAAC,GAAG,EAAE;IACd,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC,GAAG,UAAU,CAAC;IAC/C,OAAO,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAC;IACtC,OAAO,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAC;IAC1C,OAAO,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC;IACzC,OAAO,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;AACrC,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;IACzB,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;QAC1D,WAAW,CAAC;YACV,EAAE,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,MAAM,EAAE,WAAW,EAAE,OAAO,EAAE,SAAS,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE;YAC9F,EAAE,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,MAAM,EAAE,WAAW,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC,GAAG,CAAC,EAAE,KAAK,EAAE,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE;SACvG,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,MAAM,SAAS,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC;QACtE,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAClC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACvC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACnC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClE,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;QAChD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QAChD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,OAAO,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;IACxF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;QAC3E,WAAW,CAAC;YACV;gBACE,EAAE,EAAE,KAAK;gBACT,KAAK,EAAE,YAAY;gBACnB,WAAW,EAAE,0CAA0C;gBACvD,SAAS,EAAE,EAAE;gBACb,KAAK,EAAE,CAAC,QAAQ,CAAC;gBACjB,IAAI,EAAE,MAAM,EAAE,wBAAwB;aACvC;SACF,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,MAAM,SAAS,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC;QACjE,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC5C,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;QACzE,WAAW,CAAC;YACV,EAAE,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,MAAM,EAAE,WAAW,EAAE,oBAAoB,EAAE,SAAS,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE;SACzG,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,MAAM,SAAS,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC;QAC3D,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;QAC1C,WAAW,CAAC;YACV,EAAE,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,KAAK,EAAE,WAAW,EAAE,GAAG,EAAE,SAAS,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE;YACnF,EAAE,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,KAAK,EAAE,WAAW,EAAE,GAAG,EAAE,SAAS,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE;SACpF,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,MAAM,SAAS,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC;QAC3D,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QACxC,MAAM,CAAC,IAAI,GAAG,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClC,MAAM,CAAC,GAAG,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;QACxE,WAAW,CAAC;YACV,EAAE,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,GAAG,EAAE,WAAW,EAAE,GAAG,EAAE,SAAS,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE,GAAG,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE;YAClG,EAAE,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,GAAG,EAAE,WAAW,EAAE,GAAG,EAAE,SAAS,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE;SAClF,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,MAAM,SAAS,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC;QAC3D,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iEAAiE,EAAE,KAAK,IAAI,EAAE;QAC/E,OAAO,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC;QACzC,kFAAkF;QAClF,MAAM,MAAM,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,cAAc,CAC1E,sBAAsB,CACvB,CAAC;QACF,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IACzC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"prompt-enhancer.test.d.ts","sourceRoot":"","sources":["../../../src/agent/__tests__/prompt-enhancer.test.ts"],"names":[],"mappings":""}
|