Spaces:
Runtime error
Runtime error
File size: 3,239 Bytes
cd8bd0a | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 | import { describe, it } from "node:test";
import assert from "node:assert/strict";
import { compressAggressive } from "../../../open-sse/services/compression/aggressive.ts";
import type { AggressiveConfig } from "../../../open-sse/services/compression/types.ts";
function makeMessages(count: number): Array<{ role: string; content: string }> {
return Array.from({ length: count }, (_, i) => ({
role: i % 2 === 0 ? "user" : "assistant",
content: `Message ${i}: ${"x".repeat(200)}`,
}));
}
describe("compressAggressive", () => {
it("returns messages unchanged for empty input", () => {
const result = compressAggressive([]);
assert.equal(result.messages.length, 0);
assert.equal(result.stats.mode, "aggressive");
});
it("returns messages for single message (no compression needed)", () => {
const msgs = [{ role: "user", content: "Hello" }];
const result = compressAggressive(msgs);
assert.equal(result.messages.length, 1);
});
it("compresses 20-message conversation and returns stats", () => {
const msgs = makeMessages(20);
const result = compressAggressive(msgs);
assert.ok(result.messages.length > 0);
assert.equal(result.stats.mode, "aggressive");
assert.ok(typeof result.stats.originalTokens === "number");
assert.ok(typeof result.stats.compressedTokens === "number");
});
it("skips messages with [COMPRESSED: prefix (recursion guard)", () => {
const msgs = [{ role: "assistant", content: "[COMPRESSED:aging:fullSummary] prior summary" }];
const result = compressAggressive(msgs);
const content =
typeof result.messages[0].content === "string" ? result.messages[0].content : "";
assert.ok(content.startsWith("[COMPRESSED:"));
assert.ok(!content.includes("[COMPRESSED:aging:fullSummary][COMPRESSED:"));
});
it("step failure triggers downgrade, not crash", () => {
const msgs = makeMessages(10);
const config: Partial<AggressiveConfig> = {
toolStrategies: {
fileContent: false,
grepSearch: false,
shellOutput: false,
json: false,
errorMessage: false,
},
};
const result = compressAggressive(msgs, config);
assert.ok(result.messages.length > 0);
assert.ok(typeof result.stats.savingsPercent === "number");
});
it("config merge overrides defaults", () => {
const msgs = makeMessages(10);
const config: Partial<AggressiveConfig> = {
maxTokensPerMessage: 100,
minSavingsThreshold: 0.5,
};
const result = compressAggressive(msgs, config);
assert.ok(result.messages.length > 0);
});
it("aggressive stats breakdown is populated", () => {
const msgs = makeMessages(20);
const result = compressAggressive(msgs);
assert.ok(result.stats.aggressive !== undefined);
assert.ok(typeof result.stats.aggressive!.summarizerSavings === "number");
assert.ok(typeof result.stats.aggressive!.toolResultSavings === "number");
assert.ok(typeof result.stats.aggressive!.agingSavings === "number");
});
it("techniquesUsed lists applied strategies", () => {
const msgs = makeMessages(20);
const result = compressAggressive(msgs);
assert.ok(Array.isArray(result.stats.techniquesUsed));
});
});
|