mirror of
https://github.com/diegosouzapw/OmniRoute.git
synced 2026-09-15 11:22:15 +03:00
65e81158ab
Landed with the design call resolved per the owner's pick — **option 1**: the synced store is now endpoint-agnostic (persistDiscoveredModels and managedModelImport no longer drop non-chat models at write time), and chat selectability moved to read time (auto-pool expansion in autoStrategy applies filterChatSelectableModels; the models-route projection already had its chatOnly filter). Your discovery test now passes end-to-end (3/3): /api/show capabilities persist per connection and image/embedding requests route through the advertising host. Reconciliation notes: conflicted areas merged onto the current tip (adobe discovery import, requestedModel preflight signature, resolvedProvider fast-path coexists with the synced-route override — explicit resolution wins); carried base-red drains (#10055 memoization, #11071 test variants) dropped as already-landed; the managed-model-import exclusion test was propagated to the new contract (image/video models persist; the read filter still hides them from chat pickers — pinned by a new assertion). Full battery: 205/206 focused (the one red is a confirmed periodic-timer timing flake on the loaded devbox — 20/20 isolated), autoCombo vitest 30/30, combo suites 46/46, gates + typecheck clean. Thank you @yourspraveen — the capability probe + routing design was right; it just needed the store contract opened up. Fixes #11087.
113 lines
4.6 KiB
TypeScript
113 lines
4.6 KiB
TypeScript
import test from "node:test";
|
|
import assert from "node:assert/strict";
|
|
|
|
const {
|
|
resolveModelAlias,
|
|
getDeprecationNotice,
|
|
isDeprecated,
|
|
setCustomAliases,
|
|
getCustomAliases,
|
|
addCustomAlias,
|
|
removeCustomAlias,
|
|
getAllAliases,
|
|
getBuiltInAliases,
|
|
} = await import("../../open-sse/services/modelDeprecation.ts");
|
|
|
|
// ─── resolveModelAlias ──────────────────────────────────────────────────────
|
|
|
|
test("resolveModelAlias: returns original for non-deprecated model", () => {
|
|
assert.equal(resolveModelAlias("claude-opus-4-6"), "claude-opus-4-6");
|
|
});
|
|
|
|
test("resolveModelAlias: resolves deprecated Gemini model", () => {
|
|
assert.equal(resolveModelAlias("gemini-pro"), "gemini-2.5-pro");
|
|
assert.equal(resolveModelAlias("gemini-1.5-pro"), "gemini-2.5-pro");
|
|
assert.equal(resolveModelAlias("gemini-1.5-flash"), "gemini-2.5-flash");
|
|
// Retired 2.0 Flash-Lite (Google shutdown 2026-06-01) + renamed flash-lite preview
|
|
// both forward to the live GA gemini-3.1-flash-lite.
|
|
assert.equal(resolveModelAlias("gemini-2.0-flash-lite"), "gemini-3.1-flash-lite");
|
|
assert.equal(resolveModelAlias("gemini-3.1-flash-lite-preview"), "gemini-3.1-flash-lite");
|
|
// Retired free Gemma (was in the gemini-free pool) forwards to the current
|
|
// gemini-free model instead of erroring with model-not-found.
|
|
assert.equal(resolveModelAlias("gemma-4"), "gemini-3.1-flash-lite");
|
|
});
|
|
|
|
test("resolveModelAlias: resolves deprecated Claude model", () => {
|
|
assert.equal(resolveModelAlias("claude-3-opus-20240229"), "claude-opus-4-20250514");
|
|
assert.equal(resolveModelAlias("claude-3-5-sonnet-latest"), "claude-sonnet-4-20250514");
|
|
});
|
|
|
|
test("resolveModelAlias: does not silently reroute retired OpenAI models", () => {
|
|
assert.equal(resolveModelAlias("gpt-4-turbo-preview"), "gpt-4-turbo-preview");
|
|
assert.equal(resolveModelAlias("gpt-3.5-turbo-0125"), "gpt-3.5-turbo-0125");
|
|
});
|
|
|
|
test("resolveModelAlias: handles null/empty", () => {
|
|
assert.equal(resolveModelAlias(""), "");
|
|
assert.equal(resolveModelAlias(null), null);
|
|
assert.equal(resolveModelAlias(undefined), undefined);
|
|
});
|
|
|
|
// ─── getDeprecationNotice ───────────────────────────────────────────────────
|
|
|
|
test("getDeprecationNotice: returns message for deprecated model", () => {
|
|
const notice = getDeprecationNotice("gemini-pro");
|
|
assert.ok(notice);
|
|
assert.ok(notice.includes("deprecated"));
|
|
assert.ok(notice.includes("gemini-2.5-pro"));
|
|
});
|
|
|
|
test("getDeprecationNotice: returns null for non-deprecated model", () => {
|
|
assert.equal(getDeprecationNotice("claude-opus-4-6"), null);
|
|
});
|
|
|
|
test("getDeprecationNotice: returns null for empty/null", () => {
|
|
assert.equal(getDeprecationNotice(""), null);
|
|
assert.equal(getDeprecationNotice(null), null);
|
|
});
|
|
|
|
// ─── isDeprecated ───────────────────────────────────────────────────────────
|
|
|
|
test("isDeprecated: true for deprecated model", () => {
|
|
assert.equal(isDeprecated("gemini-pro"), true);
|
|
});
|
|
|
|
test("isDeprecated: false for current model", () => {
|
|
assert.equal(isDeprecated("claude-opus-4-6"), false);
|
|
});
|
|
|
|
// ─── Custom Aliases ─────────────────────────────────────────────────────────
|
|
|
|
test("custom aliases override built-in", () => {
|
|
setCustomAliases({ "gemini-pro": "gemini-3.1-pro" });
|
|
assert.equal(resolveModelAlias("gemini-pro"), "gemini-3.1-pro"); // custom wins
|
|
setCustomAliases({}); // reset
|
|
});
|
|
|
|
test("addCustomAlias and removeCustomAlias", () => {
|
|
addCustomAlias("my-old-model", "my-new-model");
|
|
assert.equal(resolveModelAlias("my-old-model"), "my-new-model");
|
|
|
|
const removed = removeCustomAlias("my-old-model");
|
|
assert.equal(removed, true);
|
|
assert.equal(resolveModelAlias("my-old-model"), "my-old-model");
|
|
});
|
|
|
|
test("removeCustomAlias: returns false for non-existent", () => {
|
|
assert.equal(removeCustomAlias("nonexistent"), false);
|
|
});
|
|
|
|
test("getAllAliases: includes both built-in and custom", () => {
|
|
addCustomAlias("test-from", "test-to");
|
|
const all = getAllAliases();
|
|
assert.ok(all["gemini-pro"]); // built-in
|
|
assert.equal(all["test-from"], "test-to"); // custom
|
|
removeCustomAlias("test-from"); // cleanup
|
|
});
|
|
|
|
test("getBuiltInAliases: returns built-in aliases", () => {
|
|
const builtIn = getBuiltInAliases();
|
|
assert.ok(builtIn["gemini-pro"]);
|
|
assert.ok(builtIn["claude-3-opus-20240229"]);
|
|
});
|