Files
2026-07-13 13:39:12 +08:00

1523 lines
62 KiB
TypeScript

/**
* T-07 config-hook backward-compat shim tests.
*
* Covers `createOmniRouteConfigHook(opts, deps)`:
* - happy path: valid auth.json → mutates input.provider[id] with the
* stripped per-model shape (mirroring `@omniroute/opencode-provider`).
* - no-op paths: missing auth.json, malformed JSON, missing apiKey,
* missing baseURL, existing input.provider[id] (manual override).
* - fail-open: /v1/models error → stub `models: {}`; /api/combos error →
* models-only static catalog.
* - baseURL resolution: opts.baseURL → auth.json.baseURL fallback.
* - multi-instance: two plugins with different providerIds publish to
* their own keys without collision.
* - cache sharing: provider hook + config hook on the same Map dedupe
* fetcher invocations.
* - sibling-shape parity: emitted entries carry only
* `{name, attachment?, reasoning?, temperature?, tool_call?, limit?}`
* — never the rich ModelV2 nested capabilities tree.
*
* Mocking strategy mirrors `provider.test.ts` and `combos.test.ts`: every
* dependency (`readAuthJson`, `fetcher`, `combosFetcher`, `now`, `cache`,
* `logger`) is dependency-injected at hook construction. No global
* `fs/promises` or `fetch` monkey-patch needed.
*/
import test from "node:test";
import assert from "node:assert/strict";
import type { Config } from "@opencode-ai/plugin";
import {
buildStaticProviderEntry,
createOmniRouteConfigHook,
createOmniRouteProviderHook,
OmniRoutePlugin,
resolveOmniRoutePluginOptions,
type OmniRouteCombosFetcher,
type OmniRouteEnrichmentEntry,
type OmniRouteEnrichmentFetcher,
type OmniRouteEnrichmentMap,
type OmniRouteFetchCache,
type OmniRouteModelsFetcher,
type OmniRouteProviderConnection,
type OmniRouteProvidersFetcher,
type OmniRouteRawCombo,
type OmniRouteRawModelEntry,
type OmniRouteReadAuthJson,
type OmniRouteStaticProviderEntry,
} from "../src/index.js";
// ────────────────────────────────────────────────────────────────────────────
// Fixtures
// ────────────────────────────────────────────────────────────────────────────
const MODEL_CLAUDE: OmniRouteRawModelEntry = {
id: "claude-sonnet-4-6",
capabilities: {
tool_calling: true,
reasoning: true,
vision: true,
thinking: false,
temperature: true,
},
context_length: 200_000,
max_output_tokens: 64_000,
max_input_tokens: 180_000,
input_modalities: ["text", "image"],
output_modalities: ["text"],
};
const MODEL_GEMINI: OmniRouteRawModelEntry = {
id: "gemini-3-flash",
capabilities: { tool_calling: true, reasoning: false, vision: true, thinking: false },
context_length: 1_000_000,
max_output_tokens: 8_192,
input_modalities: ["text", "image"],
output_modalities: ["text"],
};
const COMBO_CLAUDE_TIER: OmniRouteRawCombo = {
id: "combo-claude-tier",
name: "Claude Tier",
models: [
{ id: "s1", kind: "model", model: "claude-sonnet-4-6", weight: 100 },
{ id: "s2", kind: "model", model: "gemini-3-flash", weight: 50 },
],
};
// ────────────────────────────────────────────────────────────────────────────
// Helpers (DI stubs — mirrors patterns in provider.test.ts / combos.test.ts)
// ────────────────────────────────────────────────────────────────────────────
function stubReadAuthJson(
value: Record<string, unknown> | undefined | null
): OmniRouteReadAuthJson & { callCount: () => number } {
let n = 0;
const f: OmniRouteReadAuthJson = async () => {
n++;
return value as never;
};
return Object.assign(f, { callCount: () => n });
}
function throwingReadAuthJson(): OmniRouteReadAuthJson & { callCount: () => number } {
let n = 0;
const f: OmniRouteReadAuthJson = async () => {
n++;
throw new Error("EACCES");
};
return Object.assign(f, { callCount: () => n });
}
function stubModelsFetcher(
payload: OmniRouteRawModelEntry[]
): OmniRouteModelsFetcher & { callCount: () => number; callsBy: () => Array<[string, string]> } {
let n = 0;
const calls: Array<[string, string]> = [];
const f: OmniRouteModelsFetcher = async (baseURL, apiKey) => {
n++;
calls.push([baseURL, apiKey]);
return payload;
};
return Object.assign(f, { callCount: () => n, callsBy: () => calls });
}
function stubCombosFetcher(
payload: OmniRouteRawCombo[]
): OmniRouteCombosFetcher & { callCount: () => number; callsBy: () => Array<[string, string]> } {
let n = 0;
const calls: Array<[string, string]> = [];
const f: OmniRouteCombosFetcher = async (baseURL, apiKey) => {
n++;
calls.push([baseURL, apiKey]);
return payload;
};
return Object.assign(f, { callCount: () => n, callsBy: () => calls });
}
function throwingModelsFetcher(): OmniRouteModelsFetcher & { callCount: () => number } {
let n = 0;
const f: OmniRouteModelsFetcher = async () => {
n++;
throw new Error("ECONNREFUSED");
};
return Object.assign(f, { callCount: () => n });
}
function throwingCombosFetcher(): OmniRouteCombosFetcher & { callCount: () => number } {
let n = 0;
const f: OmniRouteCombosFetcher = async () => {
n++;
throw new Error("403 Forbidden");
};
return Object.assign(f, { callCount: () => n });
}
function stubEnrichmentFetcher(payload: OmniRouteEnrichmentMap): OmniRouteEnrichmentFetcher & {
callCount: () => number;
callsBy: () => Array<[string, string]>;
} {
let n = 0;
const calls: Array<[string, string]> = [];
const f: OmniRouteEnrichmentFetcher = async (baseURL, apiKey) => {
n++;
calls.push([baseURL, apiKey]);
return payload;
};
return Object.assign(f, { callCount: () => n, callsBy: () => calls });
}
function throwingEnrichmentFetcher(): OmniRouteEnrichmentFetcher & { callCount: () => number } {
let n = 0;
const f: OmniRouteEnrichmentFetcher = async () => {
n++;
throw new Error("ETIMEDOUT");
};
return Object.assign(f, { callCount: () => n });
}
interface WarnCapture {
warn: (...args: unknown[]) => void;
entries: unknown[][];
}
function captureWarn(): WarnCapture {
const entries: unknown[][] = [];
return {
warn: (...args: unknown[]) => {
entries.push(args);
},
entries,
};
}
function makeInput(initialProvider: Record<string, unknown> = {}): Config {
// Config = Omit<SDKConfig, "plugin"> & {plugin?: ...}. We only touch the
// `provider` slot, so a partial cast is acceptable for these tests.
return { provider: initialProvider } as unknown as Config;
}
// ────────────────────────────────────────────────────────────────────────────
// 1. Happy path — valid auth.json + apiKey + baseURL → mutates input.provider
// ────────────────────────────────────────────────────────────────────────────
test("config: with valid auth.json + apiKey + baseURL → mutates input.provider[id] with stripped models block", async () => {
const readAuthJson = stubReadAuthJson({
"opencode-omniroute": { type: "api", key: "sk-test-1", baseURL: "https://or.example.com/v1" },
});
const fetcher = stubModelsFetcher([MODEL_CLAUDE, MODEL_GEMINI]);
const combosFetcher = stubCombosFetcher([COMBO_CLAUDE_TIER]);
const logger = captureWarn();
const hook = createOmniRouteConfigHook(
{ providerId: "omniroute" },
{ readAuthJson, fetcher, combosFetcher, logger }
);
const input = makeInput();
await hook(input);
const provider = (input as { provider: Record<string, OmniRouteStaticProviderEntry> }).provider;
const entry = provider["opencode-omniroute"];
assert.ok(entry, "input.provider['opencode-omniroute'] set");
assert.equal(entry.npm, "@ai-sdk/openai-compatible");
assert.equal(entry.name, "OmniRoute");
assert.equal(entry.options.baseURL, "https://or.example.com/v1");
assert.equal(entry.options.apiKey, "sk-test-1");
// Stripped per-model shape: name + cap flags + modalities + (optional)
// cost. OC's SDK static schema accepts only `limit.{context,output}` —
// `limit.input` is NOT in the SDK shape and gets dropped silently.
const claude = entry.models["opencode-omniroute/claude-sonnet-4-6"];
assert.ok(claude, "claude model surfaced");
assert.equal(claude.name, "claude-sonnet-4-6");
assert.equal(claude.attachment, true);
assert.equal(claude.reasoning, true);
assert.equal(claude.temperature, true);
assert.equal(claude.tool_call, true);
assert.equal(claude.limit?.context, 200_000);
assert.equal(claude.limit?.output, 64_000);
assert.equal(
(claude.limit as Record<string, unknown>).input,
undefined,
"limit.input is NOT in OC's SDK schema — must not emit"
);
// Modalities — without this field OC defaults `input.image: false`
// even when `attachment: true`, blocking clipboard paste in the TUI.
assert.deepEqual(claude.modalities?.input, ["text", "image"]);
assert.deepEqual(claude.modalities?.output, ["text"]);
// Combo surfaces under bare key + LCD'd
// (gemini's reasoning=false → combo reasoning=false).
const combo = entry.models["opencode-omniroute/claude-tier"];
assert.ok(combo, "combo surfaced under bare key");
assert.equal(combo.name, "Claude Tier");
assert.equal(combo.reasoning, false, "LCD: any member reasoning=false → combo reasoning=false");
assert.equal(combo.tool_call, true);
assert.equal(combo.limit?.context, 200_000, "LCD: min(200_000, 1_000_000)");
});
// ────────────────────────────────────────────────────────────────────────────
// 1b. Dual-key fallback (#5027) — auth.json stored under the BARE providerId
// (pre-auto-prefix login) must still resolve when the active providerId is
// prefixed (`opencode-omniroute`). Without the fallback the lookup misses
// the stored key and the user is forced to re-auth.
// ────────────────────────────────────────────────────────────────────────────
test("config: auth.json under bare key (pre-prefix login) resolves via dual-key fallback", async () => {
// Stored under bare `omniroute` (the key OC wrote before the auto-prefix fix),
// but the resolved providerId is now `opencode-omniroute`.
const readAuthJson = stubReadAuthJson({
omniroute: { type: "api", key: "sk-bare-1", baseURL: "https://or.example.com/v1" },
});
const fetcher = stubModelsFetcher([MODEL_CLAUDE]);
const combosFetcher = stubCombosFetcher([]);
const logger = captureWarn();
const hook = createOmniRouteConfigHook(
{ providerId: "omniroute" }, // resolves to opencode-omniroute internally
{ readAuthJson, fetcher, combosFetcher, logger }
);
const input = makeInput();
await hook(input);
const provider = (input as { provider: Record<string, OmniRouteStaticProviderEntry> }).provider;
const entry = provider["opencode-omniroute"];
assert.ok(entry, "provider entry published from bare-key apiKey");
assert.equal(entry.options.apiKey, "sk-bare-1", "apiKey resolved from the bare auth.json key");
assert.equal(entry.options.baseURL, "https://or.example.com/v1");
});
test("config: prefixed key wins over bare key when both present (dual-key precedence)", async () => {
const readAuthJson = stubReadAuthJson({
"opencode-omniroute": { type: "api", key: "sk-prefixed", baseURL: "https://pref.example/v1" },
omniroute: { type: "api", key: "sk-bare", baseURL: "https://bare.example/v1" },
});
const fetcher = stubModelsFetcher([MODEL_CLAUDE]);
const combosFetcher = stubCombosFetcher([]);
const logger = captureWarn();
const hook = createOmniRouteConfigHook(
{ providerId: "omniroute" },
{ readAuthJson, fetcher, combosFetcher, logger }
);
const input = makeInput();
await hook(input);
const entry = (input as { provider: Record<string, OmniRouteStaticProviderEntry> }).provider[
"opencode-omniroute"
];
assert.ok(entry);
assert.equal(
entry.options.apiKey,
"sk-prefixed",
"prefixed key takes precedence (looked up first)"
);
assert.equal(entry.options.baseURL, "https://pref.example/v1");
});
// ────────────────────────────────────────────────────────────────────────────
// 2. Missing auth.json → no-op, no throw, no mutation
// ────────────────────────────────────────────────────────────────────────────
test("config: missing auth.json file → no-op, no throw, no input mutation", async () => {
const readAuthJson = stubReadAuthJson(undefined);
const fetcher = stubModelsFetcher([MODEL_CLAUDE]);
const combosFetcher = stubCombosFetcher([]);
const logger = captureWarn();
const hook = createOmniRouteConfigHook(
{ providerId: "omniroute" },
{ readAuthJson, fetcher, combosFetcher, logger }
);
const input = makeInput();
await hook(input);
assert.deepEqual((input as { provider: Record<string, unknown> }).provider, {});
assert.equal(fetcher.callCount(), 0, "no fetch on missing auth.json");
assert.equal(combosFetcher.callCount(), 0, "no combos fetch on missing auth.json");
// One breadcrumb — the missing-apiKey path.
assert.ok(
logger.entries.some((e) => String(e[0]).includes("no apiKey")),
"breadcrumb emitted"
);
});
// ────────────────────────────────────────────────────────────────────────────
// 3. Malformed auth.json → no-op + warn once
// ────────────────────────────────────────────────────────────────────────────
test("config: malformed auth.json → no-op + warn once", async () => {
// stubReadAuthJson returns `null` to signal malformed JSON (matches
// defaultReadAuthJson's contract).
const readAuthJson = stubReadAuthJson(null);
const fetcher = stubModelsFetcher([MODEL_CLAUDE]);
const combosFetcher = stubCombosFetcher([]);
const logger = captureWarn();
const hook = createOmniRouteConfigHook(
{ providerId: "omniroute" },
{ readAuthJson, fetcher, combosFetcher, logger }
);
const input = makeInput();
await hook(input);
assert.deepEqual((input as { provider: Record<string, unknown> }).provider, {});
assert.equal(fetcher.callCount(), 0);
// First warn = "failed to parse"; second warn = "no apiKey".
assert.ok(
logger.entries.some((e) => String(e[0]).includes("failed to parse")),
"parse-failure breadcrumb emitted"
);
});
// ────────────────────────────────────────────────────────────────────────────
// 4. Existing input.provider[id] → no overwrite (respect manual override)
// ────────────────────────────────────────────────────────────────────────────
test("config: existing input.provider[id] → no overwrite (respect manual override)", async () => {
const manual = {
npm: "@ai-sdk/openai-compatible",
name: "Manual OmniRoute",
options: { baseURL: "http://manual/v1", apiKey: "manual-key" },
models: { "manual-model": { name: "manual-model" } },
};
const readAuthJson = stubReadAuthJson({
"opencode-omniroute": { type: "api", key: "sk-test", baseURL: "https://or.example/v1" },
});
const fetcher = stubModelsFetcher([MODEL_CLAUDE]);
const combosFetcher = stubCombosFetcher([]);
const logger = captureWarn();
const hook = createOmniRouteConfigHook(
{ providerId: "omniroute" },
{ readAuthJson, fetcher, combosFetcher, logger }
);
const input = makeInput({ "opencode-omniroute": manual });
await hook(input);
const provider = (input as { provider: Record<string, unknown> }).provider;
assert.equal(provider["opencode-omniroute"], manual, "manual override preserved by reference");
assert.equal(fetcher.callCount(), 0, "no fetch — short-circuited before I/O");
assert.equal(readAuthJson.callCount(), 0, "no auth.json read either");
assert.ok(
logger.entries.some((e) => String(e[0]).includes("already set")),
"override breadcrumb emitted"
);
});
// ────────────────────────────────────────────────────────────────────────────
// 5. fetchers throw → warn + emit stub entry with `models: {}`
// ────────────────────────────────────────────────────────────────────────────
test("config: fetchers throw → warn + emit stub entry with models: {}", async () => {
const readAuthJson = stubReadAuthJson({
"opencode-omniroute": { type: "api", key: "sk-test", baseURL: "https://or.example/v1" },
});
const fetcher = throwingModelsFetcher();
const combosFetcher = throwingCombosFetcher();
const logger = captureWarn();
// Opt-out of disk-cache fallback for this test — we want to assert the
// pure stub path, not the disk-cache-recovery path (covered by its own
// test below).
const hook = createOmniRouteConfigHook(
{ providerId: "omniroute", features: { diskCache: false } },
{ readAuthJson, fetcher, combosFetcher, logger }
);
const input = makeInput();
await hook(input);
const entry = (input as { provider: Record<string, OmniRouteStaticProviderEntry> }).provider[
"opencode-omniroute"
];
assert.ok(entry, "stub provider entry published even when fetchers fail");
assert.equal(entry.npm, "@ai-sdk/openai-compatible");
assert.deepEqual(entry.models, {}, "models stub is empty object");
assert.equal(entry.options.baseURL, "https://or.example/v1");
assert.equal(entry.options.apiKey, "sk-test");
// Both warns fired.
assert.ok(
logger.entries.some((e) => String(e[0]).includes("/v1/models fetch failed")),
"models-fetch breadcrumb emitted"
);
assert.ok(
logger.entries.some((e) => String(e[0]).includes("/api/combos fetch failed")),
"combos-fetch breadcrumb emitted"
);
});
// ────────────────────────────────────────────────────────────────────────────
// 6. Combos fetcher throws → models-only catalog (no combos in models block)
// ────────────────────────────────────────────────────────────────────────────
test("config: combos fetcher throws → emit models-only catalog (no combos in models block)", async () => {
const readAuthJson = stubReadAuthJson({
"opencode-omniroute": { type: "api", key: "sk-test", baseURL: "https://or.example/v1" },
});
const fetcher = stubModelsFetcher([MODEL_CLAUDE, MODEL_GEMINI]);
const combosFetcher = throwingCombosFetcher();
const logger = captureWarn();
const hook = createOmniRouteConfigHook(
{ providerId: "omniroute" },
{ readAuthJson, fetcher, combosFetcher, logger }
);
const input = makeInput();
await hook(input);
const entry = (input as { provider: Record<string, OmniRouteStaticProviderEntry> }).provider[
"opencode-omniroute"
];
assert.ok(entry);
const ids = Object.keys(entry.models).sort();
assert.deepEqual(ids, [
"opencode-omniroute/claude-sonnet-4-6",
"opencode-omniroute/gemini-3-flash",
]);
assert.equal(entry.models["opencode-omniroute/claude-tier"], undefined, "no combo entry");
assert.ok(
logger.entries.some((e) => String(e[0]).includes("/api/combos fetch failed")),
"combos-fetch breadcrumb emitted"
);
});
// ────────────────────────────────────────────────────────────────────────────
// 7. baseURL from auth.json takes precedence when opts.baseURL absent
// ────────────────────────────────────────────────────────────────────────────
test("config: baseURL from auth.json takes precedence when opts.baseURL absent", async () => {
const readAuthJson = stubReadAuthJson({
"opencode-omniroute": { type: "api", key: "sk-test", baseURL: "https://creds.example/v1" },
});
const fetcher = stubModelsFetcher([MODEL_CLAUDE]);
const combosFetcher = stubCombosFetcher([]);
const logger = captureWarn();
const hook = createOmniRouteConfigHook(
{ providerId: "omniroute" }, // NO opts.baseURL
{ readAuthJson, fetcher, combosFetcher, logger }
);
const input = makeInput();
await hook(input);
assert.equal(fetcher.callsBy()[0][0], "https://creds.example/v1");
const entry = (input as { provider: Record<string, OmniRouteStaticProviderEntry> }).provider[
"opencode-omniroute"
];
assert.equal(entry.options.baseURL, "https://creds.example/v1");
});
test("config: opts.baseURL wins over auth.json's stored baseURL", async () => {
const readAuthJson = stubReadAuthJson({
"opencode-omniroute": { type: "api", key: "sk-test", baseURL: "https://creds.example/v1" },
});
const fetcher = stubModelsFetcher([MODEL_CLAUDE]);
const combosFetcher = stubCombosFetcher([]);
const logger = captureWarn();
const hook = createOmniRouteConfigHook(
{ providerId: "omniroute", baseURL: "https://opts.example/v1" },
{ readAuthJson, fetcher, combosFetcher, logger }
);
const input = makeInput();
await hook(input);
assert.equal(fetcher.callsBy()[0][0], "https://opts.example/v1");
const entry = (input as { provider: Record<string, OmniRouteStaticProviderEntry> }).provider[
"opencode-omniroute"
];
assert.equal(entry.options.baseURL, "https://opts.example/v1");
});
test("config: no baseURL resolvable (no opts, no auth.json baseURL) → no-op", async () => {
const readAuthJson = stubReadAuthJson({
"opencode-omniroute": { type: "api", key: "sk-test" }, // NO baseURL on the credential
});
const fetcher = stubModelsFetcher([MODEL_CLAUDE]);
const combosFetcher = stubCombosFetcher([]);
const logger = captureWarn();
const hook = createOmniRouteConfigHook(
{ providerId: "omniroute" }, // NO opts.baseURL
{ readAuthJson, fetcher, combosFetcher, logger }
);
const input = makeInput();
await hook(input);
assert.deepEqual((input as { provider: Record<string, unknown> }).provider, {});
assert.equal(fetcher.callCount(), 0);
assert.ok(
logger.entries.some((e) => String(e[0]).includes("no baseURL")),
"no-baseURL breadcrumb emitted"
);
});
// ────────────────────────────────────────────────────────────────────────────
// 8. Multi-instance: two plugins with different providerIds publish to
// their own keys without collision.
// ────────────────────────────────────────────────────────────────────────────
test("config: multi-instance — two plugins with different providerIds publish to their own keys without collision", async () => {
const readAuthJson = stubReadAuthJson({
"opencode-omniroute-prod": {
type: "api",
key: "sk-prod",
baseURL: "https://prod.example/v1",
},
"opencode-omniroute-preprod": {
type: "api",
key: "sk-preprod",
baseURL: "https://preprod.example/v1",
},
});
const fetcher = stubModelsFetcher([MODEL_CLAUDE]);
const combosFetcher = stubCombosFetcher([]);
const logger = captureWarn();
const hookA = createOmniRouteConfigHook(
{ providerId: "omniroute-prod" },
{ readAuthJson, fetcher, combosFetcher, logger }
);
const hookB = createOmniRouteConfigHook(
{ providerId: "omniroute-preprod" },
{ readAuthJson, fetcher, combosFetcher, logger }
);
const input = makeInput();
await hookA(input);
await hookB(input);
const provider = (input as { provider: Record<string, OmniRouteStaticProviderEntry> }).provider;
assert.ok(provider["opencode-omniroute-prod"], "prod block present");
assert.ok(provider["opencode-omniroute-preprod"], "preprod block present");
assert.equal(provider["opencode-omniroute-prod"].options.apiKey, "sk-prod");
assert.equal(provider["opencode-omniroute-preprod"].options.apiKey, "sk-preprod");
assert.equal(provider["opencode-omniroute-prod"].options.baseURL, "https://prod.example/v1");
assert.equal(
provider["opencode-omniroute-preprod"].options.baseURL,
"https://preprod.example/v1"
);
assert.notEqual(
provider["opencode-omniroute-prod"],
provider["opencode-omniroute-preprod"],
"blocks are distinct references"
);
});
// ────────────────────────────────────────────────────────────────────────────
// 9. Cache sharing: provider hook + config hook on the same Map dedupe
// fetcher invocations.
// ────────────────────────────────────────────────────────────────────────────
test("config + provider share cache: second call uses cached fetch result (single fetch per TTL)", async () => {
const readAuthJson = stubReadAuthJson({
"opencode-omniroute": { type: "api", key: "sk-shared", baseURL: "https://or.example/v1" },
});
const fetcher = stubModelsFetcher([MODEL_CLAUDE]);
const combosFetcher = stubCombosFetcher([COMBO_CLAUDE_TIER]);
const sharedCache: OmniRouteFetchCache = new Map();
const logger = captureWarn();
const configHook = createOmniRouteConfigHook(
{ providerId: "omniroute", baseURL: "https://or.example/v1", modelCacheTtl: 60_000 },
{ readAuthJson, fetcher, combosFetcher, cache: sharedCache, logger }
);
const providerHook = createOmniRouteProviderHook(
{ providerId: "omniroute", baseURL: "https://or.example/v1", modelCacheTtl: 60_000 },
{ fetcher, combosFetcher, cache: sharedCache }
);
// Simulate OC ≥1.14.49 cold start: config fires first, populates cache,
// then provider.models() reuses the cached raw results.
const input = makeInput();
await configHook(input);
assert.equal(fetcher.callCount(), 1, "config fired the only models fetch");
assert.equal(combosFetcher.callCount(), 1, "config fired the only combos fetch");
// provider hook then runs — should hit the shared cache, NOT refetch.
const apiAuth = { type: "api", key: "sk-shared" };
await providerHook.models!({} as never, { auth: apiAuth as never });
assert.equal(fetcher.callCount(), 1, "provider reused cached models");
assert.equal(combosFetcher.callCount(), 1, "provider reused cached combos");
});
test("provider → config order also dedupes (cache populated by provider, consumed by config)", async () => {
const readAuthJson = stubReadAuthJson({
"opencode-omniroute": { type: "api", key: "sk-reverse", baseURL: "https://or.example/v1" },
});
const fetcher = stubModelsFetcher([MODEL_CLAUDE]);
const combosFetcher = stubCombosFetcher([]);
const sharedCache: OmniRouteFetchCache = new Map();
const logger = captureWarn();
const configHook = createOmniRouteConfigHook(
{ providerId: "omniroute", baseURL: "https://or.example/v1", modelCacheTtl: 60_000 },
{ readAuthJson, fetcher, combosFetcher, cache: sharedCache, logger }
);
const providerHook = createOmniRouteProviderHook(
{ providerId: "omniroute", baseURL: "https://or.example/v1", modelCacheTtl: 60_000 },
{ fetcher, combosFetcher, cache: sharedCache }
);
await providerHook.models!({} as never, {
auth: { type: "api", key: "sk-reverse" } as never,
});
assert.equal(fetcher.callCount(), 1);
const input = makeInput();
await configHook(input);
assert.equal(fetcher.callCount(), 1, "config reused cached models");
});
// ────────────────────────────────────────────────────────────────────────────
// 10. Stripped models shape matches sibling provider spec
// (`{name, attachment, reasoning, tool_call, temperature, limit?}`).
// ────────────────────────────────────────────────────────────────────────────
test("buildStaticProviderEntry: stripped per-model shape matches sibling @omniroute/opencode-provider", () => {
const resolved = resolveOmniRoutePluginOptions({
providerId: "omniroute",
displayName: "OmniRoute",
});
const block = buildStaticProviderEntry(
[MODEL_CLAUDE, MODEL_GEMINI],
[],
resolved,
"https://or.example/v1",
"sk-test"
);
// Top-level provider entry shape — ONLY these four keys.
assert.deepEqual(Object.keys(block).sort(), ["models", "name", "npm", "options"]);
assert.equal(block.npm, "@ai-sdk/openai-compatible");
assert.equal(block.name, "OmniRoute");
assert.deepEqual(Object.keys(block.options).sort(), ["apiKey", "baseURL"]);
// Per-model entry shape — the keys OC's SDK static schema accepts
// (see @opencode-ai/sdk types.gen.d.ts). NO nested capabilities tree,
// NO providerID/api fields from ModelV2 (those belong on the dynamic
// hook path).
const allowedKeys = new Set([
"name",
"release_date",
"attachment",
"reasoning",
"temperature",
"tool_call",
"cost",
"limit",
"modalities",
"providerID",
]);
for (const [id, entry] of Object.entries(block.models)) {
for (const key of Object.keys(entry)) {
assert.ok(allowedKeys.has(key), `${id}.${key} is not in the SDK static shape`);
}
// capabilities (ModelV2-only) must NOT leak — that's the dynamic-
// hook nested shape, not the static SDK schema.
assert.equal(
(entry as Record<string, unknown>).capabilities,
undefined,
`${id} must not carry nested capabilities tree`
);
}
// Sanity: claude entry has all expected stripped fields.
const claude = block.models["opencode-omniroute/claude-sonnet-4-6"];
assert.equal(typeof claude.name, "string");
assert.equal(typeof claude.attachment, "boolean");
assert.equal(typeof claude.reasoning, "boolean");
assert.equal(typeof claude.temperature, "boolean");
assert.equal(typeof claude.tool_call, "boolean");
assert.equal(typeof claude.limit?.context, "number");
});
test("buildStaticProviderEntry: empty fetch results → stub block with models: {}", () => {
const resolved = resolveOmniRoutePluginOptions({ providerId: "omniroute" });
const block = buildStaticProviderEntry([], [], resolved, "https://or.example/v1", "sk-test");
assert.deepEqual(block.models, {});
assert.equal(block.options.apiKey, "sk-test");
});
test("buildStaticProviderEntry: hidden combos are excluded", () => {
const resolved = resolveOmniRoutePluginOptions({ providerId: "omniroute" });
const block = buildStaticProviderEntry(
[MODEL_CLAUDE],
[{ ...COMBO_CLAUDE_TIER, isHidden: true }],
resolved,
"https://or.example/v1",
"sk-test"
);
assert.equal(block.models["opencode-omniroute/claude-tier"], undefined);
assert.ok(block.models["opencode-omniroute/claude-sonnet-4-6"]);
});
// ────────────────────────────────────────────────────────────────────────────
// Schema parity (modalities / cost / release_date / limit cleanup)
// ────────────────────────────────────────────────────────────────────────────
test("buildStaticProviderEntry: emits modalities.input from raw.input_modalities", () => {
const resolved = resolveOmniRoutePluginOptions({ providerId: "omniroute" });
const block = buildStaticProviderEntry(
[MODEL_CLAUDE, MODEL_GEMINI],
[],
resolved,
"https://or.example/v1",
"sk-test"
);
const claude = block.models["opencode-omniroute/claude-sonnet-4-6"];
assert.deepEqual(claude.modalities?.input, ["text", "image"]);
assert.deepEqual(claude.modalities?.output, ["text"]);
});
test("buildStaticProviderEntry: never emits limit.input (OC SDK rejects it)", () => {
const resolved = resolveOmniRoutePluginOptions({ providerId: "omniroute" });
const block = buildStaticProviderEntry(
[MODEL_CLAUDE],
[],
resolved,
"https://or.example/v1",
"sk-test"
);
const claude = block.models["opencode-omniroute/claude-sonnet-4-6"];
assert.equal((claude.limit as Record<string, unknown>).input, undefined);
assert.equal(typeof claude.limit?.context, "number");
assert.equal(typeof claude.limit?.output, "number");
});
test("buildStaticProviderEntry: emits cost when enrichment carries pricing", () => {
const resolved = resolveOmniRoutePluginOptions({ providerId: "omniroute" });
const enrichment = new Map([
[
"claude-sonnet-4-6",
{
name: "Claude Sonnet 4.6",
providerAlias: "cc",
providerCanonical: "claude",
providerDisplayName: "Claude",
pricing: { input: 3, output: 15, cacheRead: 0.3, cacheWrite: 3.75 },
},
],
]);
const block = buildStaticProviderEntry(
[MODEL_CLAUDE],
[],
resolved,
"https://or.example/v1",
"sk-test",
enrichment
);
const claude = block.models["opencode-omniroute/claude-sonnet-4-6"];
assert.equal(claude.cost?.input, 3);
assert.equal(claude.cost?.output, 15);
assert.equal(claude.cost?.cache_read, 0.3);
assert.equal(claude.cost?.cache_write, 3.75);
});
test("buildStaticProviderEntry: emits release_date when raw carries it; omits when null", () => {
const resolved = resolveOmniRoutePluginOptions({ providerId: "omniroute" });
const withDate: OmniRouteRawModelEntry = {
...MODEL_CLAUDE,
id: "claude-with-date",
release_date: "2026-02-19",
};
const block = buildStaticProviderEntry(
[withDate, MODEL_GEMINI],
[],
resolved,
"https://or.example/v1",
"sk-test"
);
assert.equal(block.models["opencode-omniroute/claude-with-date"].release_date, "2026-02-19");
assert.equal(block.models["opencode-omniroute/gemini-3-flash"].release_date, undefined);
});
test("buildStaticProviderEntry: combo modalities = intersection of members (LCD)", () => {
const resolved = resolveOmniRoutePluginOptions({ providerId: "omniroute" });
const TEXT_ONLY: OmniRouteRawModelEntry = {
id: "text-only",
capabilities: { tool_calling: true, reasoning: false, vision: false, thinking: false },
context_length: 100_000,
max_output_tokens: 4_096,
input_modalities: ["text"],
output_modalities: ["text"],
};
const block = buildStaticProviderEntry(
[MODEL_CLAUDE, TEXT_ONLY],
[
{
id: "combo-mixed",
name: "Mixed Tier",
models: [
{ id: "s1", kind: "model", model: "claude-sonnet-4-6", weight: 100 },
{ id: "s2", kind: "model", model: "text-only", weight: 50 },
],
},
],
resolved,
"https://or.example/v1",
"sk-test"
);
const combo = block.models["opencode-omniroute/mixed-tier"];
assert.ok(combo, "combo emitted under slug key");
// claude has text+image, text-only has text → intersection drops image.
assert.deepEqual(combo.modalities?.input, ["text"]);
assert.deepEqual(combo.modalities?.output, ["text"]);
});
// ────────────────────────────────────────────────────────────────────────────
// Integration: OmniRoutePlugin factory now exposes config hook
// ────────────────────────────────────────────────────────────────────────────
test("OmniRoutePlugin factory exposes config hook alongside auth + provider", async () => {
const hooks = await OmniRoutePlugin({} as never, { providerId: "omniroute" });
assert.equal(typeof hooks.config, "function", "config hook present");
assert.ok(hooks.auth, "auth hook present");
assert.ok(hooks.provider, "provider hook present");
});
// ────────────────────────────────────────────────────────────────────────────
// Edge cases / robustness
// ────────────────────────────────────────────────────────────────────────────
test("config: auth.json entry of wrong type (oauth) → no-op", async () => {
const readAuthJson = stubReadAuthJson({
"opencode-omniroute": { type: "oauth", refresh: "r", access: "a", expires: 0 },
});
const fetcher = stubModelsFetcher([MODEL_CLAUDE]);
const combosFetcher = stubCombosFetcher([]);
const logger = captureWarn();
const hook = createOmniRouteConfigHook(
{ providerId: "omniroute", baseURL: "https://or.example/v1" },
{ readAuthJson, fetcher, combosFetcher, logger }
);
const input = makeInput();
await hook(input);
assert.deepEqual((input as { provider: Record<string, unknown> }).provider, {});
assert.equal(fetcher.callCount(), 0);
});
test("config: readAuthJson throws → treat as missing file (silent fallback)", async () => {
const readAuthJson = throwingReadAuthJson();
const fetcher = stubModelsFetcher([MODEL_CLAUDE]);
const combosFetcher = stubCombosFetcher([]);
const logger = captureWarn();
const hook = createOmniRouteConfigHook(
{ providerId: "omniroute", baseURL: "https://or.example/v1" },
{ readAuthJson, fetcher, combosFetcher, logger }
);
const input = makeInput();
await hook(input);
assert.deepEqual((input as { provider: Record<string, unknown> }).provider, {});
assert.equal(readAuthJson.callCount(), 1);
assert.equal(fetcher.callCount(), 0);
});
test("config: initialises input.provider when undefined", async () => {
const readAuthJson = stubReadAuthJson({
"opencode-omniroute": { type: "api", key: "sk", baseURL: "https://or.example/v1" },
});
const fetcher = stubModelsFetcher([MODEL_CLAUDE]);
const combosFetcher = stubCombosFetcher([]);
const logger = captureWarn();
const hook = createOmniRouteConfigHook(
{ providerId: "omniroute" },
{ readAuthJson, fetcher, combosFetcher, logger }
);
// input with NO provider field at all
const input = {} as Config;
await hook(input);
const provider = (input as { provider?: Record<string, unknown> }).provider;
assert.ok(provider, "provider bag initialised");
assert.ok(provider!["opencode-omniroute"]);
});
// ────────────────────────────────────────────────────────────────────────────
// Enrichment overlay — eager fetch on config-hook so OC ≤1.15.5 TUI sees
// human display names instead of raw provider/model ids.
// ────────────────────────────────────────────────────────────────────────────
test("config: enrichment fetched + name overlaid on raw-model entries", async () => {
const readAuthJson = stubReadAuthJson({
"opencode-omniroute": { type: "api", key: "sk-test", baseURL: "https://or.example/v1" },
});
const fetcher = stubModelsFetcher([MODEL_CLAUDE, MODEL_GEMINI]);
const combosFetcher = stubCombosFetcher([COMBO_CLAUDE_TIER]);
const enrichmentFetcher = stubEnrichmentFetcher(
new Map<string, OmniRouteEnrichmentEntry>([
["claude-sonnet-4-6", { name: "Claude Sonnet 4.6" }],
["gemini-3-flash", { name: "Gemini 3 Flash" }],
])
);
const logger = captureWarn();
const hook = createOmniRouteConfigHook(
{ providerId: "omniroute" },
{ readAuthJson, fetcher, combosFetcher, enrichmentFetcher, logger }
);
const input = makeInput();
await hook(input);
const entry = (input as { provider: Record<string, OmniRouteStaticProviderEntry> }).provider[
"opencode-omniroute"
];
assert.ok(entry);
assert.equal(entry.models["opencode-omniroute/claude-sonnet-4-6"].name, "Claude Sonnet 4.6");
assert.equal(entry.models["opencode-omniroute/gemini-3-flash"].name, "Gemini 3 Flash");
// Combo names still come from /api/combos — enrichment overlay does NOT touch combos.
assert.equal(entry.models["opencode-omniroute/claude-tier"].name, "Claude Tier");
assert.equal(enrichmentFetcher.callCount(), 1);
});
test("config: features.enrichment=false skips enrichment fetch + keeps raw-id names", async () => {
const readAuthJson = stubReadAuthJson({
"opencode-omniroute": { type: "api", key: "sk-test", baseURL: "https://or.example/v1" },
});
const fetcher = stubModelsFetcher([MODEL_CLAUDE]);
const combosFetcher = stubCombosFetcher([]);
const enrichmentFetcher = stubEnrichmentFetcher(
new Map<string, OmniRouteEnrichmentEntry>([
["claude-sonnet-4-6", { name: "Claude Sonnet 4.6" }],
])
);
const logger = captureWarn();
const hook = createOmniRouteConfigHook(
{ providerId: "omniroute", features: { enrichment: false } },
{ readAuthJson, fetcher, combosFetcher, enrichmentFetcher, logger }
);
const input = makeInput();
await hook(input);
const entry = (input as { provider: Record<string, OmniRouteStaticProviderEntry> }).provider[
"opencode-omniroute"
];
assert.ok(entry);
assert.equal(enrichmentFetcher.callCount(), 0, "enrichment fetch suppressed by feature flag");
assert.equal(
entry.models["opencode-omniroute/claude-sonnet-4-6"].name,
"claude-sonnet-4-6",
"raw id retained"
);
});
test("config: enrichment fetcher throws → soft-fail (warn + raw-id static catalog)", async () => {
const readAuthJson = stubReadAuthJson({
"opencode-omniroute": { type: "api", key: "sk-test", baseURL: "https://or.example/v1" },
});
const fetcher = stubModelsFetcher([MODEL_CLAUDE]);
const combosFetcher = stubCombosFetcher([]);
const enrichmentFetcher = throwingEnrichmentFetcher();
const logger = captureWarn();
const hook = createOmniRouteConfigHook(
{ providerId: "omniroute" },
{ readAuthJson, fetcher, combosFetcher, enrichmentFetcher, logger }
);
const input = makeInput();
await hook(input);
const entry = (input as { provider: Record<string, OmniRouteStaticProviderEntry> }).provider[
"opencode-omniroute"
];
assert.ok(entry, "static block still published on enrichment failure");
assert.equal(
entry.models["opencode-omniroute/claude-sonnet-4-6"].name,
"claude-sonnet-4-6",
"raw id retained"
);
assert.equal(enrichmentFetcher.callCount(), 1);
assert.ok(
logger.entries.some((e) => String(e[0]).includes("/api/pricing/models fetch failed")),
"enrichment-fetch breadcrumb emitted"
);
});
function stubProvidersFetcher(payload: OmniRouteProviderConnection[]): OmniRouteProvidersFetcher & {
callCount: () => number;
} {
let n = 0;
const f: OmniRouteProvidersFetcher = async () => {
n++;
return payload;
};
return Object.assign(f, { callCount: () => n });
}
function throwingProvidersFetcher(): OmniRouteProvidersFetcher & { callCount: () => number } {
let n = 0;
const f: OmniRouteProvidersFetcher = async () => {
n++;
throw new Error("ETIMEDOUT");
};
return Object.assign(f, { callCount: () => n });
}
const MODEL_CC_OPUS: OmniRouteRawModelEntry = {
id: "cc/claude-opus-4-7",
capabilities: { tool_calling: true, reasoning: true, temperature: true },
context_length: 200_000,
};
const MODEL_NV_LLAMA: OmniRouteRawModelEntry = {
id: "nvidia/llama-3-70b",
capabilities: { tool_calling: true, temperature: true },
context_length: 128_000,
};
test("config: usableOnly=false → no filter (existing behavior)", async () => {
const readAuthJson = stubReadAuthJson({
"opencode-omniroute": { type: "api", key: "sk-test", baseURL: "https://or.example/v1" },
});
const fetcher = stubModelsFetcher([MODEL_CC_OPUS, MODEL_NV_LLAMA]);
const combosFetcher = stubCombosFetcher([]);
const providersFetcher = stubProvidersFetcher([
{ id: "c1", provider: "claude", isActive: true, testStatus: "active" },
]);
const enrichmentFetcher = stubEnrichmentFetcher(
new Map<string, OmniRouteEnrichmentEntry>([
["cc/claude-opus-4-7", { name: "Claude Opus 4.7" }],
["nvidia/llama-3-70b", { name: "Llama 3 70B" }],
])
);
const hook = createOmniRouteConfigHook(
{ providerId: "omniroute", baseURL: "https://or.example/v1" },
{ readAuthJson, fetcher, combosFetcher, enrichmentFetcher, providersFetcher }
);
const input = makeInput();
await hook(input);
const entry = (input as { provider: Record<string, OmniRouteStaticProviderEntry> }).provider[
"opencode-omniroute"
];
assert.ok(entry.models["cc/claude-opus-4-7"], "claude kept");
assert.ok(entry.models["nvidia/llama-3-70b"], "nvidia kept (filter off)");
assert.equal(providersFetcher.callCount(), 0, "providers fetch not called when feature off");
});
test("config: usableOnly=true → drops models for non-usable providers, keeps usable + unknown", async () => {
const readAuthJson = stubReadAuthJson({
"opencode-omniroute": { type: "api", key: "sk-test", baseURL: "https://or.example/v1" },
});
const fetcher = stubModelsFetcher([
MODEL_CC_OPUS,
MODEL_NV_LLAMA,
// Unknown prefix not in pricing-models nor connections — must pass through.
{
id: "agentrouter/synthetic-1",
capabilities: { temperature: true },
context_length: 100_000,
},
]);
const combosFetcher = stubCombosFetcher([]);
const providersFetcher = stubProvidersFetcher([
// Claude is usable.
{ id: "c1", provider: "claude", isActive: true, testStatus: "active" },
// Nvidia provisioned but errored → not usable.
{ id: "c2", provider: "nvidia", isActive: true, testStatus: "error" },
]);
const enrichmentFetcher = stubEnrichmentFetcher(
new Map<string, OmniRouteEnrichmentEntry>([
[
"cc/claude-opus-4-7",
{ name: "Claude Opus 4.7", providerAlias: "cc", providerCanonical: "claude" },
],
[
"nvidia/llama-3-70b",
{ name: "Llama 3 70B", providerAlias: "nvidia", providerCanonical: "nvidia" },
],
])
);
const hook = createOmniRouteConfigHook(
{ providerId: "omniroute", baseURL: "https://or.example/v1", features: { usableOnly: true } },
{ readAuthJson, fetcher, combosFetcher, enrichmentFetcher, providersFetcher }
);
const input = makeInput();
await hook(input);
const entry = (input as { provider: Record<string, OmniRouteStaticProviderEntry> }).provider[
"opencode-omniroute"
];
assert.ok(entry.models["cc/claude-opus-4-7"], "claude kept (active)");
assert.equal(entry.models["nvidia/llama-3-70b"], undefined, "nvidia dropped (error status)");
assert.ok(entry.models["agentrouter/synthetic-1"], "unknown prefix kept (subtract-filter)");
assert.equal(providersFetcher.callCount(), 1);
});
test("config: usableOnly=true + providers fetch fails → soft-fail keeps everything", async () => {
const readAuthJson = stubReadAuthJson({
"opencode-omniroute": { type: "api", key: "sk-test", baseURL: "https://or.example/v1" },
});
const fetcher = stubModelsFetcher([MODEL_CC_OPUS, MODEL_NV_LLAMA]);
const combosFetcher = stubCombosFetcher([]);
const providersFetcher = throwingProvidersFetcher();
const enrichmentFetcher = stubEnrichmentFetcher(
new Map<string, OmniRouteEnrichmentEntry>([
["cc/claude-opus-4-7", { name: "Claude Opus 4.7" }],
["nvidia/llama-3-70b", { name: "Llama 3 70B" }],
])
);
const logger = captureWarn();
const hook = createOmniRouteConfigHook(
{ providerId: "omniroute", baseURL: "https://or.example/v1", features: { usableOnly: true } },
{ readAuthJson, fetcher, combosFetcher, enrichmentFetcher, providersFetcher, logger }
);
const input = makeInput();
await hook(input);
const entry = (input as { provider: Record<string, OmniRouteStaticProviderEntry> }).provider[
"opencode-omniroute"
];
assert.ok(entry.models["cc/claude-opus-4-7"]);
assert.ok(entry.models["nvidia/llama-3-70b"], "soft-fail keeps both");
assert.ok(
logger.entries.some((e) => String(e[0]).includes("/api/providers fetch failed")),
"providers-fetch breadcrumb emitted"
);
});
test("config: diskCache hydrates stale snapshot when /v1/models throws", async () => {
const readAuthJson = stubReadAuthJson({
"opencode-omniroute": { type: "api", key: "sk-test", baseURL: "https://or.example/v1" },
});
const fetcher = throwingModelsFetcher();
const combosFetcher = stubCombosFetcher([]);
const logger = captureWarn();
// Disk reader returns a stale snapshot — emulates the last-known-good
// catalog written on a healthy refresh.
const diskSnapshotReader: typeof import("../src/index.js").defaultDiskSnapshotReader =
async () => ({
rawModels: [MODEL_CLAUDE],
rawCombos: [],
rawEnrichment: new Map([["claude-sonnet-4-6", { name: "Claude Sonnet 4.6 (cached)" }]]),
rawCompressionCombos: [],
rawConnections: [],
});
let writes = 0;
const diskSnapshotWriter: typeof import("../src/index.js").defaultDiskSnapshotWriter =
async () => {
writes++;
};
const hook = createOmniRouteConfigHook(
{ providerId: "omniroute", features: { diskCache: true } },
{
readAuthJson,
fetcher,
combosFetcher,
diskSnapshotReader,
diskSnapshotWriter,
logger,
}
);
const input = makeInput();
await hook(input);
const entry = (input as { provider: Record<string, OmniRouteStaticProviderEntry> }).provider[
"opencode-omniroute"
];
assert.ok(
entry.models["opencode-omniroute/claude-sonnet-4-6"],
"stale snapshot hydrated into static block"
);
assert.equal(
entry.models["opencode-omniroute/claude-sonnet-4-6"].name,
"Claude Sonnet 4.6 (cached)",
"stale enrichment also reused"
);
assert.equal(writes, 0, "disk write skipped when live fetch failed");
assert.ok(
logger.entries.some((e) => String(e[0]).includes("using stale disk cache")),
"disk-cache hydration breadcrumb emitted"
);
});
test("config: cached rawEnrichment from earlier provider hook is reused (no refetch)", async () => {
const readAuthJson = stubReadAuthJson({
"opencode-omniroute": { type: "api", key: "sk-shared", baseURL: "https://or.example/v1" },
});
const fetcher = stubModelsFetcher([MODEL_CLAUDE]);
const combosFetcher = stubCombosFetcher([]);
const enrichmentFetcher = stubEnrichmentFetcher(
new Map<string, OmniRouteEnrichmentEntry>([
["claude-sonnet-4-6", { name: "Claude Sonnet 4.6" }],
])
);
const sharedCache: OmniRouteFetchCache = new Map();
const logger = captureWarn();
const providerHook = createOmniRouteProviderHook(
{ providerId: "omniroute", baseURL: "https://or.example/v1", modelCacheTtl: 60_000 },
{ fetcher, combosFetcher, enrichmentFetcher, cache: sharedCache }
);
const configHook = createOmniRouteConfigHook(
{ providerId: "omniroute", baseURL: "https://or.example/v1", modelCacheTtl: 60_000 },
{ readAuthJson, fetcher, combosFetcher, enrichmentFetcher, cache: sharedCache, logger }
);
// Provider hook fires first (e.g. eager cache warm-up), populates rawEnrichment.
await providerHook.models!({} as never, {
auth: { type: "api", key: "sk-shared" } as never,
});
assert.equal(enrichmentFetcher.callCount(), 1);
// Config hook then fires — must reuse cached enrichment, not refetch.
const input = makeInput();
await configHook(input);
assert.equal(enrichmentFetcher.callCount(), 1, "config reused cached enrichment");
const entry = (input as { provider: Record<string, OmniRouteStaticProviderEntry> }).provider[
"opencode-omniroute"
];
assert.equal(entry.models["opencode-omniroute/claude-sonnet-4-6"].name, "Claude Sonnet 4.6");
});
// ─────────────────────────────────────────────────────────────────────
// Provider-tag suffix (Option E) — append upstream provider label to
// enriched model names so the picker can differentiate `cc/claude-opus-4-7`
// (Anthropic) from `kr/claude-opus-4-7` (Kiro) etc.
// ─────────────────────────────────────────────────────────────────────
test("config: providerTag (default-on) prepends '<provider> - ' to enriched raw-model names", async () => {
const readAuthJson = stubReadAuthJson({
"opencode-omniroute": { type: "api", key: "sk-test", baseURL: "https://or.example/v1" },
});
const fetcher = stubModelsFetcher([MODEL_CLAUDE, MODEL_GEMINI]);
const combosFetcher = stubCombosFetcher([COMBO_CLAUDE_TIER]);
const enrichmentFetcher = stubEnrichmentFetcher(
new Map<string, OmniRouteEnrichmentEntry>([
[
"claude-sonnet-4-6",
{
name: "Claude Sonnet 4.6",
providerAlias: "cc",
providerCanonical: "claude",
providerDisplayName: "Claude",
},
],
[
"gemini-3-flash",
{
name: "Gemini 3 Flash",
providerAlias: "gemini",
providerCanonical: "gemini",
providerDisplayName: "Gemini",
},
],
])
);
const logger = captureWarn();
const hook = createOmniRouteConfigHook(
{ providerId: "omniroute" },
{ readAuthJson, fetcher, combosFetcher, enrichmentFetcher, logger }
);
const input = makeInput();
await hook(input);
const entry = (input as { provider: Record<string, OmniRouteStaticProviderEntry> }).provider[
"opencode-omniroute"
];
assert.ok(entry);
assert.equal(
entry.models["opencode-omniroute/claude-sonnet-4-6"].name,
"Claude - Claude Sonnet 4.6"
);
assert.equal(entry.models["opencode-omniroute/gemini-3-flash"].name, "Gemini - Gemini 3 Flash");
// Combos stay untouched — `Combo: ` prefix already conveys multi-upstream.
assert.equal(entry.models["opencode-omniroute/claude-tier"].name, "Claude Tier");
});
test("config: providerTag=false suppresses the suffix", async () => {
const readAuthJson = stubReadAuthJson({
"opencode-omniroute": { type: "api", key: "sk-test", baseURL: "https://or.example/v1" },
});
const fetcher = stubModelsFetcher([MODEL_CLAUDE]);
const combosFetcher = stubCombosFetcher([]);
const enrichmentFetcher = stubEnrichmentFetcher(
new Map<string, OmniRouteEnrichmentEntry>([
["claude-sonnet-4-6", { name: "Claude Sonnet 4.6", providerDisplayName: "Claude" }],
])
);
const logger = captureWarn();
const hook = createOmniRouteConfigHook(
{ providerId: "omniroute", features: { providerTag: false } },
{ readAuthJson, fetcher, combosFetcher, enrichmentFetcher, logger }
);
const input = makeInput();
await hook(input);
const entry = (input as { provider: Record<string, OmniRouteStaticProviderEntry> }).provider[
"opencode-omniroute"
];
assert.equal(
entry.models["opencode-omniroute/claude-sonnet-4-6"].name,
"Claude Sonnet 4.6",
"enriched name kept, provider tag suppressed"
);
});
test("config: providerTag falls back to UPPER(alias) when providerDisplayName missing", async () => {
const readAuthJson = stubReadAuthJson({
"opencode-omniroute": { type: "api", key: "sk-test", baseURL: "https://or.example/v1" },
});
const fetcher = stubModelsFetcher([MODEL_CLAUDE]);
const combosFetcher = stubCombosFetcher([]);
// Enrichment has the friendly name but NO providerDisplayName — e.g.
// a slot OmniRoute hasn't curated a human label for yet. We still
// have the alias though, so the prefix uses UPPER(alias) = "CC".
const enrichmentFetcher = stubEnrichmentFetcher(
new Map<string, OmniRouteEnrichmentEntry>([
["claude-sonnet-4-6", { name: "Claude Sonnet 4.6", providerAlias: "cc" }],
])
);
const logger = captureWarn();
const hook = createOmniRouteConfigHook(
{ providerId: "omniroute" },
{ readAuthJson, fetcher, combosFetcher, enrichmentFetcher, logger }
);
const input = makeInput();
await hook(input);
const entry = (input as { provider: Record<string, OmniRouteStaticProviderEntry> }).provider[
"opencode-omniroute"
];
assert.equal(entry.models["opencode-omniroute/claude-sonnet-4-6"].name, "CC - Claude Sonnet 4.6");
});
test("config: providerTag skipped entirely when neither providerDisplayName nor providerAlias set", async () => {
const readAuthJson = stubReadAuthJson({
"opencode-omniroute": { type: "api", key: "sk-test", baseURL: "https://or.example/v1" },
});
const fetcher = stubModelsFetcher([MODEL_CLAUDE]);
const combosFetcher = stubCombosFetcher([]);
// No metadata at all — defensive case, e.g. legacy enrichment payload.
const enrichmentFetcher = stubEnrichmentFetcher(
new Map<string, OmniRouteEnrichmentEntry>([
["claude-sonnet-4-6", { name: "Claude Sonnet 4.6" }],
])
);
const logger = captureWarn();
const hook = createOmniRouteConfigHook(
{ providerId: "omniroute" },
{ readAuthJson, fetcher, combosFetcher, enrichmentFetcher, logger }
);
const input = makeInput();
await hook(input);
const entry = (input as { provider: Record<string, OmniRouteStaticProviderEntry> }).provider[
"opencode-omniroute"
];
assert.equal(entry.models["opencode-omniroute/claude-sonnet-4-6"].name, "Claude Sonnet 4.6");
});
test("config: providerTag is idempotent — second hook call doesn't double-suffix", async () => {
const readAuthJson = stubReadAuthJson({
"opencode-omniroute": { type: "api", key: "sk-test", baseURL: "https://or.example/v1" },
});
const fetcher = stubModelsFetcher([MODEL_CLAUDE]);
const combosFetcher = stubCombosFetcher([]);
const enrichmentFetcher = stubEnrichmentFetcher(
new Map<string, OmniRouteEnrichmentEntry>([
["claude-sonnet-4-6", { name: "Claude Sonnet 4.6", providerDisplayName: "Claude" }],
])
);
const logger = captureWarn();
const sharedCache = new Map();
const hook = createOmniRouteConfigHook(
{ providerId: "omniroute", modelCacheTtl: 60_000 },
{ readAuthJson, fetcher, combosFetcher, enrichmentFetcher, cache: sharedCache, logger }
);
const inputA = makeInput();
await hook(inputA);
const entryA = (inputA as { provider: Record<string, OmniRouteStaticProviderEntry> }).provider[
"opencode-omniroute"
];
assert.equal(
entryA.models["opencode-omniroute/claude-sonnet-4-6"].name,
"Claude - Claude Sonnet 4.6"
);
// Second invocation (cache hit) — name must still be single-suffixed.
const inputB = makeInput();
await hook(inputB);
const entryB = (inputB as { provider: Record<string, OmniRouteStaticProviderEntry> }).provider[
"opencode-omniroute"
];
assert.equal(
entryB.models["opencode-omniroute/claude-sonnet-4-6"].name,
"Claude - Claude Sonnet 4.6"
);
});
// ────────────────────────────────────────────────────────────────────────────
// T-NN: nested combo-ref resolution in the static catalog
// (mirrors the dynamic-catalog fix in combos.test.ts)
// ────────────────────────────────────────────────────────────────────────────
test("buildStaticProviderEntry: nested combo-ref context is the bottleneck across the graph", () => {
const resolved = resolveOmniRoutePluginOptions({ providerId: "omniroute" });
const rawModels: OmniRouteRawModelEntry[] = [
{
id: "raw-big",
context_length: 200_000,
max_output_tokens: 64_000,
capabilities: { tool_calling: true, reasoning: true, vision: false, temperature: true },
input_modalities: ["text"],
output_modalities: ["text"],
},
{
id: "raw-tiny",
context_length: 8_000,
max_output_tokens: 4_000,
capabilities: { tool_calling: false, reasoning: false, vision: false, temperature: true },
input_modalities: ["text"],
output_modalities: ["text"],
},
];
const rawCombos: OmniRouteRawCombo[] = [
{
id: "tiny-combo",
name: "TinyCombo",
models: [{ id: "m1", kind: "model", model: "raw-tiny", weight: 100 }],
},
{
id: "parent",
name: "Parent",
models: [
{ id: "p1", kind: "model", model: "raw-big", weight: 50 },
{ id: "p2", kind: "combo-ref", comboName: "TinyCombo", weight: 50 },
],
},
];
const block = buildStaticProviderEntry(
rawModels,
rawCombos,
resolved,
"https://or.example/v1",
"sk-test"
);
// Pre-fix: Parent would advertise 200_000 (only raw-big counted).
// Post-fix: Parent should advertise 8_000 (TinyCombo bottleneck).
const parent = block.models["opencode-omniroute/parent"];
assert.ok(parent, "Parent combo must be in the static catalog");
assert.equal(parent.limit?.context, 8_000);
});