@oxagen/cli 0.5.0 → 0.6.1

package/dist/agent/__tests__/pipeline.test.js

@@ -0,0 +1,235 @@
+/**
+ * Turn pipeline — proves the orchestration: evaluate → enhance → route → execute
+ * → judge → revise, the trace it assembles, model selection (evaluator ⟂ router,
+ * manual pin), the auto-revise loop and its guards (readOnly, maxReviseRounds),
+ * bare mode, stage emission, usage accumulation, and the gateway-key guard. Every
+ * stage module is mocked, so no gateway or filesystem is touched.
+ */
+import { describe, it, expect, beforeEach, vi } from "vitest";
+vi.mock("../../lib/config.js", () => ({ readConfig: () => ({}) }));
+vi.mock("../env.js", () => {
+    class MissingGatewayKeyError extends Error {
+        constructor() {
+            super("missing");
+            this.name = "MissingGatewayKeyError";
+        }
+    }
+    return { ensureGatewayKey: vi.fn(() => "key"), MissingGatewayKeyError };
+});
+vi.mock("../loop.js", () => ({ runAgent: vi.fn() }));
+vi.mock("../evaluator.js", () => ({ evaluatePrompt: vi.fn() }));
+vi.mock("../judge.js", () => ({
+    judgeCompleteness: vi.fn(),
+    buildRevisionPrompt: vi.fn(() => "REVISION PROMPT"),
+}));
+vi.mock("../prompt-enhancer.js", () => ({ enhancePrompt: vi.fn() }));
+import { runTurn, MissingGatewayKeyError } from "../pipeline.js";
+import { ensureGatewayKey } from "../env.js";
+import { runAgent } from "../loop.js";
+import { evaluatePrompt } from "../evaluator.js";
+import { judgeCompleteness, buildRevisionPrompt } from "../judge.js";
+import { enhancePrompt } from "../prompt-enhancer.js";
+const mockEnsure = ensureGatewayKey;
+const mockRun = runAgent;
+const mockEval = evaluatePrompt;
+const mockJudge = judgeCompleteness;
+const mockEnhance = enhancePrompt;
+const mockBuildRevision = buildRevisionPrompt;
+function evaluation(over = {}) {
+    return {
+        completeness: 70,
+        complexity: 50,
+        recommendedTier: "balanced",
+        missing: [],
+        contextQueries: ["Foo"],
+        refinedPrompt: "refined ask",
+        removed: [],
+        reasoning: "r",
+        fallback: false,
+        model: "anthropic/claude-haiku-4.5",
+        usage: { inputTokens: 5, outputTokens: 1, costUsd: 0.001 },
+        ...over,
+    };
+}
+function verdict(over = {}) {
+    return {
+        complete: true,
+        confidence: 90,
+        findings: [],
+        remainingWork: [],
+        reasoning: "looks done",
+        model: "anthropic/claude-opus-4.8",
+        fallback: false,
+        usage: { inputTokens: 3, outputTokens: 1, costUsd: 0.002 },
+        ...over,
+    };
+}
+beforeEach(() => {
+    delete process.env["OXAGEN_LLM_FAST"];
+    delete process.env["OXAGEN_LLM_BALANCED"];
+    delete process.env["OXAGEN_LLM_PRECISE"];
+    mockEnsure.mockReturnValue("key");
+    mockEval.mockResolvedValue(evaluation());
+    mockEnhance.mockResolvedValue({
+        prompt: "refined ask\n\n## context",
+        context: "## context",
+        resolved: ["Foo"],
+        lessons: [],
+    });
+    mockJudge.mockResolvedValue(verdict());
+    mockBuildRevision.mockReturnValue("REVISION PROMPT");
+    // Default executor: streams text, touches a file, returns a message + usage.
+    mockRun.mockImplementation(async (opts) => {
+        opts.onToolCall?.("write_file", { path: "src/x.ts" });
+        opts.onText?.("hi");
+        return {
+            text: "done",
+            steps: 2,
+            messages: [{ role: "assistant", content: "done" }],
+            usage: { inputTokens: 100, outputTokens: 50 },
+        };
+    });
+});
+describe("runTurn — happy path", () => {
+    it("runs all stages, selects the model, and assembles a complete trace", async () => {
+        const stages = [];
+        const res = await runTurn({ prompt: "fix Foo", cwd: "/x", onStage: (s) => stages.push(s) });
+        expect(mockEval).toHaveBeenCalledOnce();
+        expect(mockEnhance).toHaveBeenCalledOnce();
+        expect(mockRun).toHaveBeenCalledOnce();
+        expect(mockJudge).toHaveBeenCalledOnce();
+        // The executor saw the enhanced prompt on the balanced (sonnet) model.
+        expect(mockRun.mock.calls[0]?.[0].prompt).toContain("## context");
+        expect(mockRun.mock.calls[0]?.[0].model).toContain("sonnet");
+        const t = res.trace;
+        expect(t.originalPrompt).toBe("fix Foo");
+        expect(t.selectedTier).toBe("balanced");
+        expect(t.selectionRationale).toContain("evaluator chose");
+        expect(t.judgeRounds).toHaveLength(1);
+        expect(t.finalComplete).toBe(true);
+        expect(t.filesTouched).toEqual(["src/x.ts"]);
+        expect(t.enhancement.source).toBe("code-graph");
+        // Usage accumulates evaluator + executor + judge token counts.
+        expect(t.usage.inputTokens).toBe(108);
+        expect(t.usage.outputTokens).toBe(52);
+        expect(t.usage.costUsd).toBeGreaterThan(0);
+        // Stages are emitted in order.
+        expect(stages.map((s) => s.kind)).toEqual([
+            "evaluate",
+            "enhance",
+            "route",
+            "execute",
+            "judge",
+            "complete",
+        ]);
+    });
+    it("forwards the evaluator's context queries to the enhancer", async () => {
+        await runTurn({ prompt: "fix Foo", cwd: "/x" });
+        expect(mockEnhance.mock.calls[0]?.[0].extraQueries).toEqual(["Foo"]);
+    });
+});
+describe("runTurn — completeness revise loop", () => {
+    it("sends the agent back when judged incomplete, then re-judges", async () => {
+        mockJudge
+            .mockResolvedValueOnce(verdict({ complete: false, findings: ["no test"] }))
+            .mockResolvedValueOnce(verdict({ complete: true }));
+        const stages = [];
+        const res = await runTurn({ prompt: "add x", cwd: "/x", onStage: (s) => stages.push(s) });
+        expect(mockRun).toHaveBeenCalledTimes(2);
+        expect(mockBuildRevision).toHaveBeenCalledOnce();
+        // The second execution runs the revision prompt.
+        expect(mockRun.mock.calls[1]?.[0].prompt).toBe("REVISION PROMPT");
+        expect(res.trace.judgeRounds).toHaveLength(2);
+        expect(res.trace.finalComplete).toBe(true);
+        expect(stages.some((s) => s.kind === "revise")).toBe(true);
+    });
+    it("does not revise in read-only mode", async () => {
+        mockJudge.mockResolvedValue(verdict({ complete: false, findings: ["x"] }));
+        const res = await runTurn({ prompt: "add x", cwd: "/x", readOnly: true });
+        expect(mockRun).toHaveBeenCalledOnce();
+        expect(res.trace.finalComplete).toBe(false);
+    });
+    it("does not revise when maxReviseRounds is 0", async () => {
+        mockJudge.mockResolvedValue(verdict({ complete: false, findings: ["x"] }));
+        const res = await runTurn({ prompt: "add x", cwd: "/x", maxReviseRounds: 0 });
+        expect(mockRun).toHaveBeenCalledOnce();
+        expect(res.trace.judgeRounds).toHaveLength(1);
+    });
+});
+describe("runTurn — model selection", () => {
+    it("pins a manually overridden model and skips routing", async () => {
+        const res = await runTurn({ prompt: "x", cwd: "/x", model: "openai/gpt-5" });
+        expect(res.trace.selectedModel).toBe("openai/gpt-5");
+        expect(res.trace.selectionRationale).toBe("pinned model");
+        expect(mockRun.mock.calls[0]?.[0].model).toBe("openai/gpt-5");
+    });
+    it("raises the safety floor above the evaluator on high-stakes signals", async () => {
+        mockEval.mockResolvedValue(evaluation({ recommendedTier: "fast", refinedPrompt: "implement the stripe billing refund" }));
+        const res = await runTurn({ prompt: "billing", cwd: "/x" });
+        expect(res.trace.selectedTier).toBe("precise");
+        expect(res.trace.selectedModel).toContain("opus");
+        expect(res.trace.selectionRationale).toContain("safety floor");
+    });
+    it("trusts the evaluator's cheap choice on non-high-stakes work (no upward escalation)", async () => {
+        // "refactor" is a design signal the old maxTier router escalated to balanced;
+        // now the Haiku evaluator is the authority and its 'fast' pick stands.
+        mockEval.mockResolvedValue(evaluation({ recommendedTier: "fast", refinedPrompt: "refactor the date helper" }));
+        const res = await runTurn({ prompt: "tidy helper", cwd: "/x" });
+        expect(res.trace.selectedTier).toBe("fast");
+        expect(res.trace.selectedModel).toContain("haiku");
+        expect(res.trace.selectionRationale).toContain("evaluator chose");
+    });
+});
+describe("runTurn — bare mode", () => {
+    it("skips eval/enhance/judge and still produces a trace", async () => {
+        const res = await runTurn({ prompt: "x", cwd: "/x", bare: true });
+        expect(mockEval).not.toHaveBeenCalled();
+        expect(mockEnhance).not.toHaveBeenCalled();
+        expect(mockJudge).not.toHaveBeenCalled();
+        expect(mockRun).toHaveBeenCalledOnce();
+        expect(res.trace.judgeRounds).toHaveLength(0);
+        expect(res.trace.selectionRationale).toBe("bare mode (no routing)");
+        expect(res.trace.finalComplete).toBe(true);
+    });
+});
+describe("runTurn — fleet memory", () => {
+    it("records a gotcha lesson when the turn had to be revised", async () => {
+        const record = vi.fn();
+        const fleetMemory = { record, recall: () => [], all: () => [] };
+        mockJudge
+            .mockResolvedValueOnce(verdict({ complete: false, findings: ["no test"] }))
+            .mockResolvedValueOnce(verdict({ complete: true }));
+        await runTurn({ prompt: "add x", cwd: "/x", fleetMemory });
+        expect(record).toHaveBeenCalledOnce();
+        expect(record.mock.calls[0]?.[0].kind).toBe("gotcha");
+        expect(record.mock.calls[0]?.[0].weight).toBe("high");
+    });
+    it("records a routine-change lesson when files changed without revision", async () => {
+        const record = vi.fn();
+        const fleetMemory = { record, recall: () => [], all: () => [] };
+        await runTurn({ prompt: "tweak x", cwd: "/x", fleetMemory });
+        expect(record).toHaveBeenCalledOnce();
+        expect(record.mock.calls[0]?.[0].kind).toBe("routine-change");
+        expect(record.mock.calls[0]?.[0].files).toContain("src/x.ts");
+    });
+    it("records nothing when no files changed and no revision was needed", async () => {
+        const record = vi.fn();
+        const fleetMemory = { record, recall: () => [], all: () => [] };
+        mockRun.mockImplementationOnce(async () => ({
+            text: "explained",
+            steps: 1,
+            messages: [],
+            usage: { inputTokens: 1, outputTokens: 1 },
+        }));
+        await runTurn({ prompt: "explain x", cwd: "/x", fleetMemory });
+        expect(record).not.toHaveBeenCalled();
+    });
+});
+describe("runTurn — guards", () => {
+    it("throws MissingGatewayKeyError when no gateway key resolves", async () => {
+        mockEnsure.mockReturnValue(null);
+        await expect(runTurn({ prompt: "x", cwd: "/x" })).rejects.toBeInstanceOf(MissingGatewayKeyError);
+        expect(mockRun).not.toHaveBeenCalled();
+    });
+});
+//# sourceMappingURL=pipeline.test.js.map

package/dist/agent/__tests__/pipeline.test.js.map

@@ -0,0 +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,iBAAiB,CAAC,CAAC;QAC1D,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,oEAAoE,EAAE,KAAK,IAAI,EAAE;QAClF,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,cAAc,CAAC,CAAC;IACjE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oFAAoF,EAAE,KAAK,IAAI,EAAE;QAClG,8EAA8E;QAC9E,uEAAuE;QACvE,QAAQ,CAAC,iBAAiB,CACxB,UAAU,CAAC,EAAE,eAAe,EAAE,MAAM,EAAE,aAAa,EAAE,0BAA0B,EAAE,CAAC,CACnF,CAAC;QACF,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,EAAE,MAAM,EAAE,aAAa,EAAE,GAAG,EAAE,IAAI,EAAE,CAAC,CAAC;QAChE,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC5C,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QACnD,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC;IACpE,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"}

package/dist/agent/__tests__/planner.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=planner.test.d.ts.map

package/dist/agent/__tests__/planner.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"planner.test.d.ts","sourceRoot":"","sources":["../../../src/agent/__tests__/planner.test.ts"],"names":[],"mappings":""}

package/dist/agent/__tests__/planner.test.js

@@ -0,0 +1,98 @@
+/**
+ * Task planner — proves a goal decomposes into a valid plan: unique task ids,
+ * dependencies pruned to real tasks, and each task's tier reconciled UP with the
+ * cost router so a high-stakes task is never under-spent. The model call and the
+ * prompt enhancer are mocked, so no gateway or filesystem is touched.
+ */
+import { describe, it, expect, beforeEach, vi } from "vitest";
+import { tmpdir } from "node:os";
+vi.mock("ai", () => ({ generateObject: vi.fn() }));
+vi.mock("../../lib/config.js", () => ({ readConfig: () => ({}) }));
+vi.mock("../prompt-enhancer.js", () => ({
+    enhancePrompt: async ({ prompt }) => ({
+        prompt,
+        context: "",
+        resolved: [],
+        lessons: [],
+    }),
+}));
+import { generateObject } from "ai";
+import { planTasks } from "../planner.js";
+import { MissingGatewayKeyError } from "../env.js";
+const mockGen = generateObject;
+function resolveWith(tasks) {
+    mockGen.mockResolvedValueOnce({
+        object: { tasks },
+        usage: { inputTokens: 10, outputTokens: 20 },
+    });
+}
+beforeEach(() => {
+    process.env["AI_GATEWAY_API_KEY"] = "test-key";
+    delete process.env["OXAGEN_LLM_FAST"];
+    delete process.env["OXAGEN_LLM_BALANCED"];
+    delete process.env["OXAGEN_LLM_PRECISE"];
+    delete process.env["OXAGEN_MODEL"];
+});
+describe("planTasks", () => {
+    it("builds a draft plan from the model's tasks", async () => {
+        resolveWith([
+            { id: "a", title: "Do A", description: "first", dependsOn: [], files: ["a.ts"], tier: "fast" },
+            { id: "b", title: "Do B", description: "second", dependsOn: ["a"], files: ["b.ts"], tier: "balanced" },
+        ]);
+        const plan = await planTasks({ goal: "do the thing", cwd: tmpdir() });
+        expect(plan.status).toBe("draft");
+        expect(plan.goal).toBe("do the thing");
+        expect(plan.tasks).toHaveLength(2);
+        expect(plan.tasks.every((t) => t.status === "queued")).toBe(true);
+        expect(plan.tasks[1]?.dependsOn).toEqual(["a"]);
+        expect(plan.tasks[0]?.model).toContain("haiku");
+        expect(plan.tasks[0]?.usage).toEqual({ inputTokens: 0, outputTokens: 0, costUsd: 0 });
+    });
+    it("escalates a task's tier when its description is high-stakes", async () => {
+        resolveWith([
+            {
+                id: "pay",
+                title: "Add refund",
+                description: "implement the stripe billing refund flow",
+                dependsOn: [],
+                files: ["pay.ts"],
+                tier: "fast", // model under-estimated
+            },
+        ]);
+        const plan = await planTasks({ goal: "billing", cwd: tmpdir() });
+        expect(plan.tasks[0]?.tier).toBe("precise");
+        expect(plan.tasks[0]?.model).toContain("opus");
+    });
+    it("never downgrades a precise task the model already flagged", async () => {
+        resolveWith([
+            { id: "x", title: "tiny", description: "rename a local var", dependsOn: [], files: [], tier: "precise" },
+        ]);
+        const plan = await planTasks({ goal: "g", cwd: tmpdir() });
+        expect(plan.tasks[0]?.tier).toBe("precise");
+    });
+    it("de-duplicates repeated ids", async () => {
+        resolveWith([
+            { id: "x", title: "One", description: "d", dependsOn: [], files: [], tier: "fast" },
+            { id: "x", title: "Two", description: "d", dependsOn: [], files: [], tier: "fast" },
+        ]);
+        const plan = await planTasks({ goal: "g", cwd: tmpdir() });
+        const ids = plan.tasks.map((t) => t.id);
+        expect(new Set(ids).size).toBe(2);
+        expect(ids).toContain("x");
+    });
+    it("drops dependencies that point at nonexistent or self ids", async () => {
+        resolveWith([
+            { id: "a", title: "A", description: "d", dependsOn: ["ghost", "b", "a"], files: [], tier: "fast" },
+            { id: "b", title: "B", description: "d", dependsOn: [], files: [], tier: "fast" },
+        ]);
+        const plan = await planTasks({ goal: "g", cwd: tmpdir() });
+        expect(plan.tasks[0]?.dependsOn).toEqual(["b"]);
+    });
+    it("throws MissingGatewayKeyError when no gateway key is resolvable", async () => {
+        delete process.env["AI_GATEWAY_API_KEY"];
+        // cwd in the OS temp dir → no .env.local up the tree, and config is mocked empty.
+        await expect(planTasks({ goal: "g", cwd: tmpdir() })).rejects.toBeInstanceOf(MissingGatewayKeyError);
+        expect(mockGen).not.toHaveBeenCalled();
+    });
+});
+//# sourceMappingURL=planner.test.js.map

package/dist/agent/__tests__/planner.test.js.map

@@ -0,0 +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"}

package/dist/agent/__tests__/prompt-enhancer.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=prompt-enhancer.test.d.ts.map

package/dist/agent/__tests__/prompt-enhancer.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"prompt-enhancer.test.d.ts","sourceRoot":"","sources":["../../../src/agent/__tests__/prompt-enhancer.test.ts"],"names":[],"mappings":""}

package/dist/agent/__tests__/prompt-enhancer.test.js

@@ -0,0 +1,107 @@
+/**
+ * Prompt enhancement — proves a raw prompt is enriched with real code-graph
+ * context (symbol definitions, file symbols, dependents) and recalled lessons,
+ * and that it degrades to a no-op when there is nothing to add.
+ */
+import { describe, it, expect, beforeAll, afterAll } from "vitest";
+import { promises as fs } from "node:fs";
+import { tmpdir } from "node:os";
+import { join } from "node:path";
+import { extractCandidates, enhancePrompt } from "../prompt-enhancer.js";
+import { clearCodeGraphCache } from "../code-graph.js";
+let root = "";
+beforeAll(async () => {
+    root = await fs.mkdtemp(join(tmpdir(), "oxagen-pe-"));
+    await fs.writeFile(join(root, "alpha.ts"), ["export function computeAlpha() {", "  return 1;", "}"].join("\n"));
+    await fs.writeFile(join(root, "beta.ts"), [
+        'import { computeAlpha } from "./alpha.js";',
+        "export class BetaEngine {}",
+        "export function runBeta() {",
+        "  return computeAlpha();",
+        "}",
+    ].join("\n"));
+    await fs.mkdir(join(root, "sub"), { recursive: true });
+    await fs.writeFile(join(root, "sub", "gamma.ts"), ['import { BetaEngine } from "../beta.js";', "export const g = new BetaEngine();"].join("\n"));
+});
+afterAll(async () => {
+    clearCodeGraphCache();
+    if (root)
+        await fs.rm(root, { recursive: true, force: true });
+});
+describe("extractCandidates", () => {
+    it("pulls backticked symbols and paths, CamelCase and snake_case", () => {
+        const { symbols, paths } = extractCandidates("where is `computeAlpha` defined? look at `src/agent/loop.ts` and the GraphNode type and build_tools fn and edit alpha.ts");
+        expect(symbols).toContain("computeAlpha");
+        expect(symbols).toContain("GraphNode");
+        expect(symbols).toContain("build_tools");
+        expect(paths).toContain("src/agent/loop.ts");
+        expect(paths).toContain("alpha.ts");
+    });
+});
+const noMemory = {
+    record: () => undefined,
+    recall: () => [],
+    all: () => [],
+};
+describe("enhancePrompt", () => {
+    it("retrieves symbol definitions, file symbols and dependents", async () => {
+        const res = await enhancePrompt({
+            prompt: "where is `computeAlpha` defined and what imports `beta.ts`?",
+            cwd: root,
+            memory: noMemory,
+        });
+        expect(res.resolved).toContain("computeAlpha");
+        expect(res.resolved).toContain("beta.ts");
+        expect(res.context).toContain("alpha.ts"); // computeAlpha's definition site
+        expect(res.context).toContain("Symbols in beta.ts");
+        expect(res.context).toContain("sub/gamma.ts"); // dependent of beta.ts
+        // The enhanced prompt keeps the original text and appends the context block.
+        expect(res.prompt.startsWith("where is")).toBe(true);
+        expect(res.prompt).toContain("Relevant code context");
+    });
+    it("injects recalled lessons as a context section", async () => {
+        const lesson = {
+            id: "m1",
+            createdAt: 1,
+            kind: "gotcha",
+            weight: "critical",
+            lesson: "beta.ts must keep its default export",
+            files: ["beta.ts"],
+            outcome: "success",
+        };
+        const memory = { record: () => undefined, recall: () => [lesson], all: () => [] };
+        const res = await enhancePrompt({ prompt: "touch beta.ts", cwd: root, memory });
+        expect(res.lessons).toHaveLength(1);
+        expect(res.context).toContain("Lessons from past work");
+        expect(res.context).toContain("must keep its default export");
+    });
+    it("resolves context from extraQueries even when the prompt names nothing", async () => {
+        const res = await enhancePrompt({
+            prompt: "do the thing",
+            cwd: root,
+            memory: noMemory,
+            extraQueries: ["computeAlpha", "beta.ts"],
+        });
+        expect(res.resolved).toContain("computeAlpha");
+        expect(res.resolved).toContain("beta.ts");
+        expect(res.context).toContain("Relevant code context");
+        // extraQueries sharpen retrieval but never pollute the visible prompt text.
+        expect(res.prompt.startsWith("do the thing")).toBe(true);
+        expect(res.prompt).not.toContain("computeAlpha\n");
+    });
+    it("is a no-op when nothing is found", async () => {
+        const res = await enhancePrompt({ prompt: "hello there friend", cwd: root, memory: noMemory });
+        expect(res.context).toBe("");
+        expect(res.prompt).toBe("hello there friend");
+        expect(res.resolved).toHaveLength(0);
+    });
+    it("never throws and degrades when the cwd is unreadable", async () => {
+        const res = await enhancePrompt({
+            prompt: "look at `Thing`",
+            cwd: join(root, "does-not-exist"),
+            memory: noMemory,
+        });
+        expect(res.prompt).toContain("look at");
+    });
+});
+//# sourceMappingURL=prompt-enhancer.test.js.map

package/dist/agent/__tests__/prompt-enhancer.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"prompt-enhancer.test.js","sourceRoot":"","sources":["../../../src/agent/__tests__/prompt-enhancer.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,QAAQ,CAAC;AACnE,OAAO,EAAE,QAAQ,IAAI,EAAE,EAAE,MAAM,SAAS,CAAC;AACzC,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AACjC,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,iBAAiB,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AACzE,OAAO,EAAE,mBAAmB,EAAE,MAAM,kBAAkB,CAAC;AAIvD,IAAI,IAAI,GAAG,EAAE,CAAC;AAEd,SAAS,CAAC,KAAK,IAAI,EAAE;IACnB,IAAI,GAAG,MAAM,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,YAAY,CAAC,CAAC,CAAC;IACtD,MAAM,EAAE,CAAC,SAAS,CAChB,IAAI,CAAC,IAAI,EAAE,UAAU,CAAC,EACtB,CAAC,kCAAkC,EAAE,aAAa,EAAE,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CACpE,CAAC;IACF,MAAM,EAAE,CAAC,SAAS,CAChB,IAAI,CAAC,IAAI,EAAE,SAAS,CAAC,EACrB;QACE,4CAA4C;QAC5C,4BAA4B;QAC5B,6BAA6B;QAC7B,0BAA0B;QAC1B,GAAG;KACJ,CAAC,IAAI,CAAC,IAAI,CAAC,CACb,CAAC;IACF,MAAM,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IACvD,MAAM,EAAE,CAAC,SAAS,CAChB,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,UAAU,CAAC,EAC7B,CAAC,0CAA0C,EAAE,oCAAoC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAC9F,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,KAAK,IAAI,EAAE;IAClB,mBAAmB,EAAE,CAAC;IACtB,IAAI,IAAI;QAAE,MAAM,EAAE,CAAC,EAAE,CAAC,IAAI,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;AAChE,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;IACjC,EAAE,CAAC,8DAA8D,EAAE,GAAG,EAAE;QACtE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,iBAAiB,CAC1C,0HAA0H,CAC3H,CAAC;QACF,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;QAC1C,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;QACvC,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;QACzC,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;QAC7C,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,MAAM,QAAQ,GAAgB;IAC5B,MAAM,EAAE,GAAG,EAAE,CAAC,SAAS;IACvB,MAAM,EAAE,GAAG,EAAE,CAAC,EAAE;IAChB,GAAG,EAAE,GAAG,EAAE,CAAC,EAAE;CACd,CAAC;AAEF,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;QACzE,MAAM,GAAG,GAAG,MAAM,aAAa,CAAC;YAC9B,MAAM,EAAE,6DAA6D;YACrE,GAAG,EAAE,IAAI;YACT,MAAM,EAAE,QAAQ;SACjB,CAAC,CAAC;QACH,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;QAC/C,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QAC1C,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC,CAAC,iCAAiC;QAC5E,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,oBAAoB,CAAC,CAAC;QACpD,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC,CAAC,uBAAuB;QACtE,6EAA6E;QAC7E,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACrD,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,uBAAuB,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;QAC7D,MAAM,MAAM,GAAiB;YAC3B,EAAE,EAAE,IAAI;YACR,SAAS,EAAE,CAAC;YACZ,IAAI,EAAE,QAAQ;YACd,MAAM,EAAE,UAAU;YAClB,MAAM,EAAE,sCAAsC;YAC9C,KAAK,EAAE,CAAC,SAAS,CAAC;YAClB,OAAO,EAAE,SAAS;SACnB,CAAC;QACF,MAAM,MAAM,GAAgB,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,SAAS,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAC/F,MAAM,GAAG,GAAG,MAAM,aAAa,CAAC,EAAE,MAAM,EAAE,eAAe,EAAE,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;QAChF,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACpC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,wBAAwB,CAAC,CAAC;QACxD,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,8BAA8B,CAAC,CAAC;IAChE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;QACrF,MAAM,GAAG,GAAG,MAAM,aAAa,CAAC;YAC9B,MAAM,EAAE,cAAc;YACtB,GAAG,EAAE,IAAI;YACT,MAAM,EAAE,QAAQ;YAChB,YAAY,EAAE,CAAC,cAAc,EAAE,SAAS,CAAC;SAC1C,CAAC,CAAC;QACH,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;QAC/C,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QAC1C,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,uBAAuB,CAAC,CAAC;QACvD,4EAA4E;QAC5E,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzD,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;QAChD,MAAM,GAAG,GAAG,MAAM,aAAa,CAAC,EAAE,MAAM,EAAE,oBAAoB,EAAE,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC,CAAC;QAC/F,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC7B,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QAC9C,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;QACpE,MAAM,GAAG,GAAG,MAAM,aAAa,CAAC;YAC9B,MAAM,EAAE,iBAAiB;YACzB,GAAG,EAAE,IAAI,CAAC,IAAI,EAAE,gBAAgB,CAAC;YACjC,MAAM,EAAE,QAAQ;SACjB,CAAC,CAAC;QACH,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/agent/__tests__/rate-card.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=rate-card.test.d.ts.map

package/dist/agent/__tests__/rate-card.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"rate-card.test.d.ts","sourceRoot":"","sources":["../../../src/agent/__tests__/rate-card.test.ts"],"names":[],"mappings":""}

package/dist/agent/__tests__/rate-card.test.js

@@ -0,0 +1,69 @@
+/**
+ * Rate card — proves pricing is correct, vendor-agnostic, and projectable: the
+ * longest-prefix family match, dollar math per 1M tokens, the fallback for
+ * unknown slugs, and the compare-every-model projection that backs `oxagen cost`.
+ */
+import { describe, it, expect } from "vitest";
+import { rateFor, entryFor, estimateCostUsd, formatUsd, projectCost, compareModels, listRateCard, FALLBACK_RATE, } from "../rate-card.js";
+describe("rateFor", () => {
+    it("matches a family across the vendor/ prefix and version suffix", () => {
+        expect(rateFor("anthropic/claude-opus-4.8")).toEqual({ inputPer1M: 15, outputPer1M: 75 });
+        expect(rateFor("anthropic/claude-haiku-4.5")).toEqual({ inputPer1M: 1, outputPer1M: 5 });
+    });
+    it("prefers the more specific gpt-4o-mini over gpt-4o", () => {
+        expect(rateFor("openai/gpt-4o-mini")).toEqual({ inputPer1M: 0.15, outputPer1M: 0.6 });
+        expect(rateFor("openai/gpt-4o")).toEqual({ inputPer1M: 2.5, outputPer1M: 10 });
+    });
+    it("falls back to the Sonnet-equivalent rate for unknown slugs (never zero)", () => {
+        expect(rateFor("somevendor/mystery-model")).toEqual(FALLBACK_RATE);
+        expect(entryFor("somevendor/mystery-model")).toBeUndefined();
+    });
+});
+describe("estimateCostUsd", () => {
+    it("prices input and output independently per 1M tokens", () => {
+        // 1M in @ $15 + 1M out @ $75 = $90 on opus.
+        expect(estimateCostUsd("anthropic/claude-opus-4.8", { inputTokens: 1_000_000, outputTokens: 1_000_000 })).toBeCloseTo(90);
+        // Half a million haiku output @ $5/1M = $2.50.
+        expect(estimateCostUsd("anthropic/claude-haiku-4.5", { outputTokens: 500_000 })).toBeCloseTo(2.5);
+    });
+    it("treats missing token directions as zero", () => {
+        expect(estimateCostUsd("openai/gpt-4o", {})).toBe(0);
+    });
+});
+describe("formatUsd", () => {
+    it("renders small, sub-cent, and dollar amounts distinctly", () => {
+        expect(formatUsd(0)).toBe("$0");
+        expect(formatUsd(0.00001)).toBe("<$0.0001");
+        expect(formatUsd(0.0042)).toBe("$0.0042");
+        expect(formatUsd(1.2)).toBe("$1.20");
+    });
+});
+describe("projectCost", () => {
+    it("breaks cost into input/output/total with vendor metadata", () => {
+        const p = projectCost("anthropic/claude-sonnet-4.6", { inputTokens: 1_000_000, outputTokens: 100_000 });
+        expect(p.vendor).toBe("anthropic");
+        expect(p.label).toBe("Claude Sonnet");
+        expect(p.inputCostUsd).toBeCloseTo(3);
+        expect(p.outputCostUsd).toBeCloseTo(1.5);
+        expect(p.totalUsd).toBeCloseTo(4.5);
+        expect(p.fallback).toBe(false);
+    });
+    it("flags the fallback for an unpriced model", () => {
+        const p = projectCost("acme/unknown", { inputTokens: 1000 });
+        expect(p.fallback).toBe(true);
+    });
+});
+describe("compareModels", () => {
+    it("prices the same usage across every family, cheapest first", () => {
+        const rows = compareModels({ inputTokens: 1_000_000, outputTokens: 1_000_000 });
+        expect(rows.length).toBe(listRateCard().length);
+        // Sorted ascending by total.
+        for (let i = 1; i < rows.length; i++) {
+            expect(rows[i].totalUsd).toBeGreaterThanOrEqual(rows[i - 1].totalUsd);
+        }
+        // gpt-4o-mini is the cheapest family in the card; opus the dearest.
+        expect(rows[0].label).toBe("GPT-4o mini");
+        expect(rows[rows.length - 1].label).toBe("Claude Opus");
+    });
+});
+//# sourceMappingURL=rate-card.test.js.map

package/dist/agent/__tests__/rate-card.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"rate-card.test.js","sourceRoot":"","sources":["../../../src/agent/__tests__/rate-card.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EACL,OAAO,EACP,QAAQ,EACR,eAAe,EACf,SAAS,EACT,WAAW,EACX,aAAa,EACb,YAAY,EACZ,aAAa,GACd,MAAM,iBAAiB,CAAC;AAEzB,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;IACvB,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;QACvE,MAAM,CAAC,OAAO,CAAC,2BAA2B,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,UAAU,EAAE,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC,CAAC;QAC1F,MAAM,CAAC,OAAO,CAAC,4BAA4B,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE,CAAC,CAAC;IAC3F,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;QAC3D,MAAM,CAAC,OAAO,CAAC,oBAAoB,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE,WAAW,EAAE,GAAG,EAAE,CAAC,CAAC;QACtF,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,UAAU,EAAE,GAAG,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC,CAAC;IACjF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yEAAyE,EAAE,GAAG,EAAE;QACjF,MAAM,CAAC,OAAO,CAAC,0BAA0B,CAAC,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;QACnE,MAAM,CAAC,QAAQ,CAAC,0BAA0B,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;IAC/D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,4CAA4C;QAC5C,MAAM,CAAC,eAAe,CAAC,2BAA2B,EAAE,EAAE,WAAW,EAAE,SAAS,EAAE,YAAY,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;QAC1H,+CAA+C;QAC/C,MAAM,CAAC,eAAe,CAAC,4BAA4B,EAAE,EAAE,YAAY,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;IACpG,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;QACjD,MAAM,CAAC,eAAe,CAAC,eAAe,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;IACzB,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;QAChE,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAChC,MAAM,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC5C,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC1C,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;IAC3B,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;QAClE,MAAM,CAAC,GAAG,WAAW,CAAC,6BAA6B,EAAE,EAAE,WAAW,EAAE,SAAS,EAAE,YAAY,EAAE,OAAO,EAAE,CAAC,CAAC;QACxG,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACnC,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QACtC,MAAM,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;QACtC,MAAM,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;QACzC,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;QACpC,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACjC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QAClD,MAAM,CAAC,GAAG,WAAW,CAAC,cAAc,EAAE,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC,CAAC;QAC7D,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAChC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,EAAE,CAAC,2DAA2D,EAAE,GAAG,EAAE;QACnE,MAAM,IAAI,GAAG,aAAa,CAAC,EAAE,WAAW,EAAE,SAAS,EAAE,YAAY,EAAE,SAAS,EAAE,CAAC,CAAC;QAChF,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC,MAAM,CAAC,CAAC;QAChD,6BAA6B;QAC7B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACrC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAE,CAAC,QAAQ,CAAC,CAAC,sBAAsB,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAE,CAAC,QAAQ,CAAC,CAAC;QAC1E,CAAC;QACD,oEAAoE;QACpE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAE,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAC3C,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAE,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IAC3D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/agent/__tests__/system-prompt.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=system-prompt.test.d.ts.map

package/dist/agent/__tests__/system-prompt.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"system-prompt.test.d.ts","sourceRoot":"","sources":["../../../src/agent/__tests__/system-prompt.test.ts"],"names":[],"mappings":""}

package/dist/agent/__tests__/system-prompt.test.js

@@ -0,0 +1,29 @@
+import { describe, it, expect } from "vitest";
+import { buildSystemPrompt } from "../system-prompt.js";
+describe("buildSystemPrompt", () => {
+    it("uses the default oxagen identity with no agent", () => {
+        const prompt = buildSystemPrompt({ cwd: "/repo" });
+        expect(prompt).toContain("You are oxagen");
+        expect(prompt).toContain("Operating rules:");
+    });
+    it("replaces the identity with the agent's system prompt", () => {
+        const prompt = buildSystemPrompt({
+            cwd: "/repo",
+            agent: { name: "reviewer", systemPrompt: "You are a meticulous code reviewer." },
+        });
+        expect(prompt).toContain("You are a meticulous code reviewer.");
+        expect(prompt).not.toContain("You are oxagen");
+        // Operating rules + environment are still present.
+        expect(prompt).toContain("Operating rules:");
+        expect(prompt).toContain("cwd: /repo");
+    });
+    it("still appends the read-only notice for an agent", () => {
+        const prompt = buildSystemPrompt({
+            cwd: "/repo",
+            readOnly: true,
+            agent: { name: "x", systemPrompt: "Persona." },
+        });
+        expect(prompt).toContain("READ-ONLY MODE");
+    });
+});
+//# sourceMappingURL=system-prompt.test.js.map

package/dist/agent/__tests__/system-prompt.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"system-prompt.test.js","sourceRoot":"","sources":["../../../src/agent/__tests__/system-prompt.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AAExD,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;IACjC,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;QACxD,MAAM,MAAM,GAAG,iBAAiB,CAAC,EAAE,GAAG,EAAE,OAAO,EAAE,CAAC,CAAC;QACnD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;QAC3C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,MAAM,MAAM,GAAG,iBAAiB,CAAC;YAC/B,GAAG,EAAE,OAAO;YACZ,KAAK,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,YAAY,EAAE,qCAAqC,EAAE;SACjF,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,qCAAqC,CAAC,CAAC;QAChE,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;QAC/C,mDAAmD;QACnD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC;QAC7C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;QACzD,MAAM,MAAM,GAAG,iBAAiB,CAAC;YAC/B,GAAG,EAAE,OAAO;YACZ,QAAQ,EAAE,IAAI;YACd,KAAK,EAAE,EAAE,IAAI,EAAE,GAAG,EAAE,YAAY,EAAE,UAAU,EAAE;SAC/C,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/agent/__tests__/trace-format-verbose.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=trace-format-verbose.test.d.ts.map