feat(canvas): 云端 Agent - 缩减画布目录上下文并展示实时用量

web/test/agent-context-usage.test.ts

@@ -0,0 +1,107 @@
+import { describe, expect, it } from "bun:test";
+import { contextInputTokens, contextPressureRatio, emptyAgentContextUsage, presentAgentContextUsage, reduceAgentContextUsage } from "@/lib/canvas/agent-context-usage";
+
+const event = (runId: string, type: string, payload: Record<string, unknown>, seq = 1) => ({ runId, type, payload, seq });
+
+describe("Agent context usage events", () => {
+    it("keeps provider's previous measurement separate from projected next request", () => {
+        const state = reduceAgentContextUsage(emptyAgentContextUsage("run-1"), event("run-1", "context_pressure", {
+            modelLimitConfigured: true,
+            usableInputTokens: 80_000,
+            pressureTokens: 40_000,
+            projectedNextInputTokens: 48_000,
+            projectedPressureRatio: 0.6,
+            tokenSource: "provider",
+        }));
+        expect(contextInputTokens(state.reading)).toBe(48_000);
+        expect(contextPressureRatio(state.reading)).toBe(0.6);
+    });
+
+    it("does not invent percentages when the model window is unknown", () => {
+        const state = reduceAgentContextUsage(emptyAgentContextUsage("run-1"), event("run-1", "context_pressure", {
+            estimatedInputTokens: 12_000,
+            modelLimitConfigured: false,
+        }));
+        expect(contextInputTokens(state.reading)).toBe(12_000);
+        expect(contextPressureRatio(state.reading)).toBeUndefined();
+    });
+
+    it("marks pre-compaction readings stale until a fresh pressure event arrives", () => {
+        let state = reduceAgentContextUsage(emptyAgentContextUsage("run-1"), event("run-1", "context_pressure", { pressureRatio: 0.8, modelLimitConfigured: true, usableInputTokens: 100 }, 2));
+        state = reduceAgentContextUsage(state, event("run-1", "context_compaction_requested", { basis: "tokens" }, 3));
+        expect(state.compactionPending).toEqual({ basis: "tokens" });
+        state = reduceAgentContextUsage(state, event("run-1", "context_compacted", { mode: "checkpoint", resume: true }, 4));
+        expect(state.readingStale).toBe(true);
+        expect(state.compactionPending).toBeNull();
+        expect(state.lastCompaction).toEqual({ mode: "checkpoint", resume: true });
+        state = reduceAgentContextUsage(state, event("run-1", "context_pressure", { pressureRatio: 0.25, modelLimitConfigured: true, usableInputTokens: 100 }, 5));
+        expect(state.readingStale).toBe(false);
+        expect(contextPressureRatio(state.reading)).toBe(0.25);
+    });
+
+    it("fills the ring against the compaction line, not the raw window", () => {
+        const state = reduceAgentContextUsage(emptyAgentContextUsage("run-1"), event("run-1", "context_pressure", {
+            modelLimitConfigured: true,
+            tokenSource: "provider",
+            usableInputTokens: 100_000,
+            compactAtTokens: 85_000,
+            contextWindowTokens: 128_000,
+            projectedNextInputTokens: 42_500,
+            projectedPressureRatio: 0.425,
+            breakdown: { buckets: [
+                { key: "system", label: "系统提示（含画布摘要）", tokens: 8_000, scaledTokens: 9_000, bytes: 12_000 },
+                { key: "tools", tokens: 4_000, scaledTokens: 4_500, bytes: 18_000 },
+                { key: "messages", tokens: 20_000, scaledTokens: 22_000, bytes: 228_000 },
+            ], envelopeBytes: 558 },
+        }));
+        const view = presentAgentContextUsage(state);
+        expect(view.phase).toBe("ok");
+        expect(view.ring).toBeCloseTo(0.5, 5);
+        expect(view.label).toBe("43%");
+        expect(view.inputTokens).toBe(42_500);
+        expect(view.remainingTokens).toBe(57_500);
+        expect(view.protocolBytes).toBe(558);
+        expect(view.breakdown.map((item) => item.label)).toEqual(["系统提示（含画布摘要）", "工具 schema", "会话消息（含工具结果）"]);
+        expect(view.breakdown[0]?.tokens).toBe(9_000);
+        expect(view.breakdown[0]?.bytes).toBe(12_000);
+    });
+
+    it("treats the compaction line as full and surfaces compacting before a stale reading", () => {
+        let state = reduceAgentContextUsage(emptyAgentContextUsage("run-1"), event("run-1", "context_pressure", {
+            modelLimitConfigured: true,
+            usableInputTokens: 100,
+            compactAtTokens: 85,
+            estimatedInputTokens: 90,
+            pressureRatio: 0.9,
+        }));
+        expect(presentAgentContextUsage(state).phase).toBe("compress");
+        expect(presentAgentContextUsage(state).ring).toBe(1);
+        state = reduceAgentContextUsage(state, event("run-1", "context_compaction_requested", { basis: "tokens" }, 2));
+        expect(presentAgentContextUsage(state).phase).toBe("compacting");
+        state = reduceAgentContextUsage(state, event("run-1", "context_compacted", { mode: "fallback", resume: true }, 3));
+        expect(presentAgentContextUsage(state).phase).toBe("stale");
+    });
+
+    it("does not draw a ring when the window is unknown or unread", () => {
+        expect(presentAgentContextUsage(emptyAgentContextUsage("")).phase).toBe("idle");
+        const state = reduceAgentContextUsage(emptyAgentContextUsage("run-1"), event("run-1", "context_pressure", {
+            estimatedInputTokens: 12_000,
+            modelLimitConfigured: false,
+        }));
+        const view = presentAgentContextUsage(state);
+        expect(view.phase).toBe("unknown");
+        expect(view.ratio).toBeUndefined();
+        expect(view.ring).toBe(0);
+    });
+
+    it("isolates usage across runs", () => {
+        const state = reduceAgentContextUsage({
+            ...emptyAgentContextUsage("run-1"),
+            reading: { estimatedInputTokens: 300 },
+            lastCompaction: { mode: "checkpoint" },
+        }, event("run-2", "assistant_message", {}));
+        expect(state.runId).toBe("run-2");
+        expect(state.reading).toBeNull();
+        expect(state.lastCompaction).toBeNull();
+    });
+});