mirror of
https://github.com/diegosouzapw/OmniRoute.git
synced 2026-08-07 15:52:52 +03:00
92e8960f77
* fix(models): preserve static registry models not covered by synced discovery * feat(models): add functional gateway mirror synthesizer * feat(models): add functional gateway mirror gate predicate * feat(models): add functional gateway mirror DB gate * feat(models): wire functional gateway mirrors into /v1/models * refactor(models): extract synced-coverage helper to pure leaf (file-size gate) * fix(db): re-export functional gateway mirrors gate from localDb (db-rules) * fix(i18n): translate functional gateway mirror flag for Vietnamese (locale completeness) --------- Co-authored-by: diegosouzapw <diegosouzapw@users.noreply.github.com>
84 lines
2.9 KiB
TypeScript
84 lines
2.9 KiB
TypeScript
import { test } from "node:test";
|
|
import assert from "node:assert/strict";
|
|
|
|
import { appendFunctionalGatewayMirrors } from "../../open-sse/utils/functionalGatewayMirrors.ts";
|
|
|
|
interface CatalogEntry {
|
|
id: string;
|
|
owned_by?: string;
|
|
root?: string;
|
|
name?: string;
|
|
[key: string]: unknown;
|
|
}
|
|
|
|
// Simulate: canonical owner "deepseek" has no eligible connection; passthrough
|
|
// gateway "agentrouter" (alias "agentrouter") has one and routes arbitrary models.
|
|
const deps = {
|
|
gatewayProviderIds: ["agentrouter", "openrouter"],
|
|
isGateway: (p: string) => p === "agentrouter" || p === "openrouter",
|
|
gatewayAlias: (p: string) => p, // agentrouter has no distinct alias
|
|
gatewayCovers: (p: string, modelId: string) => p === "agentrouter", // routes anything
|
|
gatewayHasConnection: (p: string) => p === "agentrouter",
|
|
canonicalOwnerHasConnection: (owner: string) => owner !== "deepseek",
|
|
};
|
|
|
|
test("synthesizes a gateway-alias mirror when canonical owner has no connection", () => {
|
|
const models: CatalogEntry[] = [
|
|
{
|
|
id: "deepseek/deepseek-v4-flash",
|
|
owned_by: "deepseek",
|
|
root: "deepseek-v4-flash",
|
|
name: "DeepSeek V4 Flash",
|
|
},
|
|
];
|
|
const out = appendFunctionalGatewayMirrors(models, deps);
|
|
|
|
assert.ok(out.some((m) => m.id === "agentrouter/deepseek/deepseek-v4-flash"));
|
|
const mirror = out.find((m) => m.id === "agentrouter/deepseek/deepseek-v4-flash");
|
|
assert.equal(mirror!.root, "deepseek/deepseek-v4-flash");
|
|
assert.equal(mirror!.owned_by, "agentrouter");
|
|
assert.equal(mirror!.display_name, "DeepSeek V4 Flash (via agentrouter)");
|
|
});
|
|
|
|
test("does NOT mirror when the canonical owner already has a connection", () => {
|
|
const models: CatalogEntry[] = [
|
|
{ id: "kimi/kimi-k2.7-code", owned_by: "kimi", root: "kimi-k2.7-code" },
|
|
];
|
|
const out = appendFunctionalGatewayMirrors(models, {
|
|
gatewayProviderIds: ["agentrouter"],
|
|
isGateway: () => false,
|
|
gatewayAlias: (p) => p,
|
|
gatewayCovers: () => false,
|
|
gatewayHasConnection: () => true,
|
|
canonicalOwnerHasConnection: () => true,
|
|
});
|
|
assert.equal(out.length, 1); // unchanged
|
|
});
|
|
|
|
test("does NOT mirror when the gateway has no connection", () => {
|
|
const models: CatalogEntry[] = [
|
|
{ id: "deepseek/deepseek-v4-flash", owned_by: "deepseek", root: "deepseek-v4-flash" },
|
|
];
|
|
const out = appendFunctionalGatewayMirrors(models, {
|
|
gatewayProviderIds: ["agentrouter"],
|
|
isGateway: () => true,
|
|
gatewayAlias: (p) => p,
|
|
gatewayCovers: () => true,
|
|
gatewayHasConnection: () => false, // no gateway credential
|
|
canonicalOwnerHasConnection: () => false,
|
|
});
|
|
assert.equal(out.length, 1); // unchanged
|
|
});
|
|
|
|
test("never mirrors ids that already carry the gateway alias prefix", () => {
|
|
const models: CatalogEntry[] = [
|
|
{
|
|
id: "agentrouter/deepseek/deepseek-v4-flash",
|
|
owned_by: "deepseek",
|
|
root: "deepseek-v4-flash",
|
|
},
|
|
];
|
|
const out = appendFunctionalGatewayMirrors(models, deps);
|
|
assert.equal(out.length, 1); // unchanged
|
|
});
|