Files
OmniRoute/tests/unit/model-parse.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

65 lines
2.4 KiB
TypeScript

import { test } from "node:test";
import assert from "node:assert/strict";
import {
normalizeCrossProxyModelId,
parseModel,
stripContextWindowSuffix,
} from "../../open-sse/services/model.ts";
test("context-window suffix helper strips client tags without changing the base model", () => {
assert.strictEqual(stripContextWindowSuffix("my-glm52[500k]"), "my-glm52");
});
// [1m] extended context suffix — PR #311 (DavyMassoneto)
test("[1m] suffix: strips suffix and sets extendedContext=true", () => {
const result = parseModel("claude-sonnet-4-6[1m]");
assert.strictEqual(result.model, "claude-sonnet-4-6");
assert.strictEqual(result.extendedContext, true);
});
test("[1M] suffix: preserves the extended-context flag case-insensitively", () => {
const result = parseModel("claude-sonnet-4-6[1M]");
assert.strictEqual(result.model, "claude-sonnet-4-6");
assert.strictEqual(result.extendedContext, true);
});
test("[1m] suffix: normal model has extendedContext=false", () => {
const result = parseModel("claude-sonnet-4-6");
assert.strictEqual(result.model, "claude-sonnet-4-6");
assert.strictEqual(result.extendedContext, false);
});
test("[1m] suffix: works with provider prefix", () => {
const result = parseModel("claude/claude-sonnet-4-6[1m]");
assert.strictEqual(result.model, "claude-sonnet-4-6");
assert.strictEqual(result.extendedContext, true);
});
test("parseModel trims provider prefix and model id", () => {
const result = parseModel(" cx / gpt-5.6-sol ");
assert.strictEqual(result.providerAlias, "cx");
assert.strictEqual(result.provider, "codex");
assert.strictEqual(result.model, "gpt-5.6-sol");
});
test("parseModel treats exact slashful model ids as models, not provider prefixes", () => {
const result = parseModel("openai/gpt-oss-120b");
assert.strictEqual(result.provider, null);
assert.strictEqual(result.providerAlias, null);
assert.strictEqual(result.isAlias, true);
assert.strictEqual(result.model, "openai/gpt-oss-120b");
});
test("normalizeCrossProxyModelId maps supported external dialects to canonical ids", () => {
assert.deepEqual(normalizeCrossProxyModelId("gpt-oss:120b"), {
modelId: "gpt-oss-120b",
applied: true,
original: "gpt-oss:120b",
});
assert.deepEqual(normalizeCrossProxyModelId("qwen3-coder:480b"), {
modelId: "Qwen/Qwen3-Coder-480B-A35B-Instruct",
applied: true,
original: "qwen3-coder:480b",
});
});