chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,75 @@
|
||||
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-alias-seed-"));
|
||||
process.env.DATA_DIR = TEST_DATA_DIR;
|
||||
|
||||
const core = await import("../../src/lib/db/core.ts");
|
||||
const modelsDb = await import("../../src/lib/db/models.ts");
|
||||
const sseModelService = await import("../../src/sse/services/model.ts");
|
||||
const { DEFAULT_MODEL_ALIAS_SEED, seedDefaultModelAliases } =
|
||||
await import("../../src/lib/modelAliasSeed.ts");
|
||||
|
||||
async function resetStorage() {
|
||||
core.resetDbInstance();
|
||||
fs.rmSync(TEST_DATA_DIR, { recursive: true, force: true });
|
||||
fs.mkdirSync(TEST_DATA_DIR, { recursive: true });
|
||||
}
|
||||
|
||||
test.beforeEach(async () => {
|
||||
await resetStorage();
|
||||
});
|
||||
|
||||
test.after(async () => {
|
||||
core.resetDbInstance();
|
||||
fs.rmSync(TEST_DATA_DIR, { recursive: true, force: true });
|
||||
});
|
||||
|
||||
test("default model alias seed writes missing aliases and is idempotent", async () => {
|
||||
const first = await seedDefaultModelAliases();
|
||||
const aliases = await modelsDb.getModelAliases();
|
||||
|
||||
assert.deepEqual(first.failed, []);
|
||||
assert.equal(first.applied.length, Object.keys(DEFAULT_MODEL_ALIAS_SEED).length);
|
||||
assert.equal(aliases["gemini-3-pro-high"], "agy/gemini-3.1-pro-high");
|
||||
assert.equal(aliases["gemini-3-pro-low"], "agy/gemini-3.1-pro-low");
|
||||
assert.equal(aliases["gemini-3-pro-preview"], "agy/gemini-pro-agent");
|
||||
assert.equal(aliases["gemini-3.1-pro-preview"], "agy/gemini-pro-agent");
|
||||
assert.equal(aliases["gemini-3-flash-preview"], "agy/gemini-3.5-flash-medium");
|
||||
|
||||
const routed = await sseModelService.getModelInfo("gemini-3-pro-high");
|
||||
assert.deepEqual(routed, {
|
||||
provider: "agy",
|
||||
model: "gemini-3.1-pro-high",
|
||||
extendedContext: false,
|
||||
});
|
||||
|
||||
const second = await seedDefaultModelAliases();
|
||||
assert.equal(second.applied.length, 0);
|
||||
assert.equal(second.failed.length, 0);
|
||||
assert.equal(second.skipped.length, Object.keys(DEFAULT_MODEL_ALIAS_SEED).length);
|
||||
});
|
||||
|
||||
test("default model alias seed preserves existing aliases and skips invalid entries", async () => {
|
||||
await modelsDb.setModelAlias("gemini-3-pro-high", "custom/provider-model");
|
||||
|
||||
const warnings = [];
|
||||
const result = await seedDefaultModelAliases({
|
||||
logger: {
|
||||
warn: (message) => warnings.push(String(message)),
|
||||
},
|
||||
seedMap: {
|
||||
...DEFAULT_MODEL_ALIAS_SEED,
|
||||
"broken-entry": null,
|
||||
},
|
||||
});
|
||||
const aliases = await modelsDb.getModelAliases();
|
||||
|
||||
assert.equal(aliases["gemini-3-pro-high"], "custom/provider-model");
|
||||
assert.ok(result.skipped.includes("gemini-3-pro-high"));
|
||||
assert.ok(result.failed.includes("broken-entry"));
|
||||
assert.ok(warnings.some((message) => message.includes("broken-entry")));
|
||||
});
|
||||
Reference in New Issue
Block a user