mirror of
https://github.com/diegosouzapw/OmniRoute.git
synced 2026-07-31 20:32:20 +03:00
a7233d1bf1
Centralize compression config schemas across settings, preview, RTK, and combo APIs to enforce consistent validation for stacked pipelines and engine-specific options. Expand caveman and stacked compression behavior by applying default combos at runtime, surfacing validation and fallback metadata, and exposing aggressive and ultra adapter schemas for configuration UI and tests. Persist MCP description compression snapshots into analytics without counting them as provider usage, and extend the dashboard with the new RTK controls and localized labels.
86 lines
3.0 KiB
TypeScript
86 lines
3.0 KiB
TypeScript
import { describe, it, beforeEach } from "node:test";
|
|
import assert from "node:assert/strict";
|
|
|
|
import {
|
|
aggressiveEngine,
|
|
cavemanEngine,
|
|
ultraEngine,
|
|
} from "../../../open-sse/services/compression/engines/cavemanAdapter.ts";
|
|
import { rtkEngine as realRtkEngine } from "../../../open-sse/services/compression/engines/rtk/index.ts";
|
|
import {
|
|
clearCompressionEngineRegistry,
|
|
getEngine,
|
|
getEngineEntry,
|
|
listEnabledEngines,
|
|
listEngines,
|
|
registerEngine,
|
|
setEngineEnabled,
|
|
updateEngineConfig,
|
|
} from "../../../open-sse/services/compression/index.ts";
|
|
|
|
describe("compression engine registry contract", () => {
|
|
beforeEach(() => {
|
|
clearCompressionEngineRegistry();
|
|
});
|
|
|
|
it("registers and retrieves an engine by id", () => {
|
|
registerEngine(cavemanEngine);
|
|
|
|
assert.equal(getEngine("caveman"), cavemanEngine);
|
|
assert.equal(getEngine("missing"), null);
|
|
});
|
|
|
|
it("lists engine entries and filters enabled entries", () => {
|
|
registerEngine(cavemanEngine);
|
|
registerEngine(realRtkEngine);
|
|
|
|
assert.equal(listEngines().length, 2);
|
|
assert.equal(listEnabledEngines().length, 2);
|
|
|
|
assert.equal(setEngineEnabled("rtk", false), true);
|
|
assert.equal(
|
|
listEnabledEngines()
|
|
.map((entry) => entry.engine.id)
|
|
.join(","),
|
|
"caveman"
|
|
);
|
|
});
|
|
|
|
it("updates config only after engine validation passes", () => {
|
|
registerEngine(realRtkEngine);
|
|
|
|
const invalid = updateEngineConfig("rtk", { intensity: "extreme" });
|
|
assert.equal(invalid.valid, false);
|
|
assert.match(invalid.errors.join(" "), /intensity/);
|
|
|
|
const valid = updateEngineConfig("rtk", {
|
|
intensity: "aggressive",
|
|
applyToCodeBlocks: true,
|
|
});
|
|
assert.equal(valid.valid, true);
|
|
assert.deepEqual(getEngineEntry("rtk")?.config, {
|
|
intensity: "aggressive",
|
|
applyToCodeBlocks: true,
|
|
});
|
|
});
|
|
|
|
it("exposes schema and validation for built-in adapters", () => {
|
|
const cavemanSchema = cavemanEngine.getConfigSchema();
|
|
const rtkSchema = realRtkEngine.getConfigSchema();
|
|
const aggressiveSchema = aggressiveEngine.getConfigSchema();
|
|
const ultraSchema = ultraEngine.getConfigSchema();
|
|
|
|
assert.ok(cavemanSchema.some((field) => field.key === "intensity"));
|
|
assert.ok(rtkSchema.some((field) => field.key === "applyToCodeBlocks"));
|
|
assert.ok(aggressiveSchema.some((field) => field.key === "maxTokensPerMessage"));
|
|
assert.ok(ultraSchema.some((field) => field.key === "compressionRate"));
|
|
assert.equal(cavemanEngine.validateConfig({ intensity: "full" }).valid, true);
|
|
assert.equal(cavemanEngine.validateConfig({ intensity: "bad" }).valid, false);
|
|
assert.equal(realRtkEngine.validateConfig({ maxLinesPerResult: 20 }).valid, true);
|
|
assert.equal(aggressiveEngine.validateConfig({ maxTokensPerMessage: 2048 }).valid, true);
|
|
assert.equal(aggressiveEngine.validateConfig({ maxTokensPerMessage: 10 }).valid, false);
|
|
assert.equal(ultraEngine.validateConfig({ compressionRate: 0.4 }).valid, true);
|
|
assert.equal(ultraEngine.validateConfig({ compressionRate: 4 }).valid, false);
|
|
});
|
|
});
|