mirror of
https://github.com/diegosouzapw/OmniRoute.git
synced 2026-09-14 02:42:24 +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.
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
|
|
});
|