Files
OmniRoute/tests/unit/functional-gateway-mirrors-db.test.ts
Praveen K Palaniswamy 65e81158ab fix(ollama): route models by advertised capability (#11088)
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.
2026-08-23 11:45:01 -03:00

53 lines
1.8 KiB
TypeScript

import { test, after } from "node:test";
import assert from "node:assert/strict";
import { resetDbInstance } from "../../src/lib/db/core.ts";
import {
removeFeatureFlagOverride,
setFeatureFlagOverride,
} from "../../src/lib/db/featureFlags.ts";
import {
getFunctionalGatewayGlobalState,
getFunctionalGatewayProviderSetting,
getFunctionalGatewayModelSetting,
setFunctionalGatewayProviderSetting,
setFunctionalGatewayModelSetting,
getFunctionalGatewaySettingsBulk,
} from "../../src/lib/db/functionalGatewayMirrors.ts";
const FLAG_KEY = "EXPOSE_FUNCTIONAL_GATEWAY_MIRRORS";
after(() => {
removeFeatureFlagOverride(FLAG_KEY);
resetDbInstance();
});
test("global state defaults to off", () => {
const { enabled } = getFunctionalGatewayGlobalState();
assert.equal(enabled, false);
});
test("can flip global on via feature-flag override (the dashboard/env mechanism)", () => {
setFeatureFlagOverride(FLAG_KEY, "true");
const { enabled, source } = getFunctionalGatewayGlobalState();
assert.equal(enabled, true);
assert.equal(source, "db");
});
test("provider and model settings default to null", () => {
assert.equal(getFunctionalGatewayProviderSetting("agentrouter"), null);
assert.equal(
getFunctionalGatewayModelSetting("agentrouter/deepseek/deepseek-v4-flash"),
null
);
});
test("bulk settings reflect provider/model overrides", () => {
setFunctionalGatewayProviderSetting("agentrouter", "on");
setFunctionalGatewayModelSetting("openrouter/deepseek/deepseek-v4-flash", "off");
const { providers, models } = getFunctionalGatewaySettingsBulk();
assert.equal(providers.get("agentrouter"), "on");
assert.equal(models.get("openrouter/deepseek/deepseek-v4-flash"), "off");
setFunctionalGatewayProviderSetting("agentrouter", null);
setFunctionalGatewayModelSetting("openrouter/deepseek/deepseek-v4-flash", null);
});