mirror of
https://github.com/diegosouzapw/OmniRoute.git
synced 2026-07-31 20:32:20 +03:00
01dd72cf1f
Introduce a custom CLI tools card that generates OpenAI-compatible env vars and JSON config, and add a cost overview tab with pricing source visibility. Add persisted custom eval suites with authenticated CRUD routes and validation, plus new translator stream transformation tooling and richer live monitor routing details. Improve localization coverage across dashboard flows and add unit tests for custom CLI config, pricing sources, eval suites, and stream transformation.
87 lines
2.7 KiB
TypeScript
87 lines
2.7 KiB
TypeScript
import test from "node:test";
|
|
import assert from "node:assert/strict";
|
|
import fs from "node:fs";
|
|
import os from "node:os";
|
|
import path from "node:path";
|
|
|
|
const TEST_DATA_DIR = fs.mkdtempSync(path.join(os.tmpdir(), "omniroute-pricing-route-"));
|
|
process.env.DATA_DIR = TEST_DATA_DIR;
|
|
|
|
interface PricingWithSourcesPayload {
|
|
pricing: Record<string, Record<string, Record<string, number>>>;
|
|
sourceMap: Record<string, Record<string, "default" | "litellm" | "modelsDev" | "user">>;
|
|
}
|
|
|
|
const core = await import("../../src/lib/db/core.ts");
|
|
const settingsDb = await import("../../src/lib/db/settings.ts");
|
|
const pricingRoute = await import("../../src/app/api/pricing/route.ts");
|
|
|
|
function resetDb() {
|
|
core.resetDbInstance();
|
|
fs.rmSync(TEST_DATA_DIR, { recursive: true, force: true });
|
|
fs.mkdirSync(TEST_DATA_DIR, { recursive: true });
|
|
}
|
|
|
|
test.beforeEach(() => {
|
|
resetDb();
|
|
});
|
|
|
|
test.after(() => {
|
|
core.resetDbInstance();
|
|
fs.rmSync(TEST_DATA_DIR, { recursive: true, force: true });
|
|
});
|
|
|
|
test("pricing GET keeps legacy payload by default and exposes source metadata on demand", async () => {
|
|
const db = core.getDbInstance();
|
|
|
|
db.prepare("INSERT INTO key_value (namespace, key, value) VALUES (?, ?, ?)").run(
|
|
"pricing_synced",
|
|
"route-provider",
|
|
JSON.stringify({
|
|
"model-litellm": { prompt: 1 },
|
|
"model-user": { prompt: 2 },
|
|
})
|
|
);
|
|
db.prepare("INSERT INTO key_value (namespace, key, value) VALUES (?, ?, ?)").run(
|
|
"models_dev_pricing",
|
|
"route-provider",
|
|
JSON.stringify({
|
|
"model-modelsdev": { prompt: 3 },
|
|
"model-user": { completion: 4 },
|
|
})
|
|
);
|
|
|
|
await settingsDb.updatePricing({
|
|
"route-provider": {
|
|
"model-user": { cached: 5 },
|
|
},
|
|
});
|
|
|
|
const legacyResponse = await pricingRoute.GET(new Request("http://localhost/api/pricing"));
|
|
assert.equal(legacyResponse.status, 200);
|
|
const legacyPayload = await legacyResponse.json();
|
|
|
|
assert.equal("sourceMap" in legacyPayload, false);
|
|
assert.deepEqual(legacyPayload["route-provider"]["model-user"], {
|
|
prompt: 2,
|
|
completion: 4,
|
|
cached: 5,
|
|
});
|
|
|
|
const sourceResponse = await pricingRoute.GET(
|
|
new Request("http://localhost/api/pricing?includeSources=1")
|
|
);
|
|
assert.equal(sourceResponse.status, 200);
|
|
|
|
const payload = (await sourceResponse.json()) as PricingWithSourcesPayload;
|
|
assert.deepEqual(payload.pricing["route-provider"]["model-user"], {
|
|
prompt: 2,
|
|
completion: 4,
|
|
cached: 5,
|
|
});
|
|
assert.equal(payload.sourceMap["route-provider"]["model-litellm"], "litellm");
|
|
assert.equal(payload.sourceMap["route-provider"]["model-modelsdev"], "modelsDev");
|
|
assert.equal(payload.sourceMap["route-provider"]["model-user"], "user");
|
|
assert.equal(payload.sourceMap.openai["gpt-4o"], "default");
|
|
});
|