mirror of
https://github.com/diegosouzapw/OmniRoute.git
synced 2026-09-15 19:32:20 +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.
124 lines
5.3 KiB
TypeScript
124 lines
5.3 KiB
TypeScript
import test from "node:test";
|
|
import assert from "node:assert/strict";
|
|
import {
|
|
getAllEmbeddingModels,
|
|
getEmbeddingProvider,
|
|
parseEmbeddingModel,
|
|
} from "../../open-sse/config/embeddingRegistry.ts";
|
|
import {
|
|
getAllRerankModels,
|
|
getRerankProvider,
|
|
parseRerankModel,
|
|
} from "../../open-sse/config/rerankRegistry.ts";
|
|
|
|
test("voyage-ai embedding registry exposes current embedding models", () => {
|
|
const provider = getEmbeddingProvider("voyage-ai");
|
|
|
|
assert.ok(provider);
|
|
assert.equal(provider.baseUrl, "https://api.voyageai.com/v1/embeddings");
|
|
assert.ok(provider.models.some((model) => model.id === "voyage-4-large"));
|
|
assert.ok(provider.models.some((model) => model.id === "voyage-code-3"));
|
|
assert.ok(provider.models.some((model) => model.id === "voyage-4"));
|
|
|
|
const parsed = parseEmbeddingModel("voyage-ai/voyage-4-large");
|
|
assert.equal(parsed.provider, "voyage-ai");
|
|
assert.equal(parsed.model, "voyage-4-large");
|
|
|
|
const all = getAllEmbeddingModels().filter((model) => model.provider === "voyage-ai");
|
|
assert.ok(all.length >= 3);
|
|
});
|
|
|
|
// #7351: voyage-multilingual-3.5 does not exist in the Voyage AI API — it was
|
|
// added to the registry by mistake and caused HTTP 400 on every test/request.
|
|
// voyage-multilingual-2 is the real model and was missing from the registry.
|
|
test("#7351 voyage-ai registry does not list non-existent voyage-multilingual-3.5", () => {
|
|
const provider = getEmbeddingProvider("voyage-ai");
|
|
assert.ok(provider);
|
|
assert.ok(
|
|
!provider.models.some((model) => model.id === "voyage-multilingual-3.5"),
|
|
"voyage-multilingual-3.5 must not be in the registry — it does not exist in the Voyage AI API"
|
|
);
|
|
});
|
|
|
|
test("#7351 voyage-ai registry includes voyage-multilingual-2 (real model)", () => {
|
|
const provider = getEmbeddingProvider("voyage-ai");
|
|
assert.ok(provider);
|
|
assert.ok(provider.models.some((model) => model.id === "voyage-multilingual-2"));
|
|
});
|
|
|
|
test("#7351 voyage-ai registry includes voyage-3.5 and voyage-3.5-lite", () => {
|
|
const provider = getEmbeddingProvider("voyage-ai");
|
|
assert.ok(provider);
|
|
assert.ok(provider.models.some((model) => model.id === "voyage-3.5"));
|
|
assert.ok(provider.models.some((model) => model.id === "voyage-3.5-lite"));
|
|
});
|
|
|
|
test("voyage-ai and jina-ai rerank registries expose supported models", () => {
|
|
const voyage = getRerankProvider("voyage-ai");
|
|
const jina = getRerankProvider("jina-ai");
|
|
|
|
assert.ok(voyage);
|
|
assert.equal(voyage.baseUrl, "https://api.voyageai.com/v1/rerank");
|
|
assert.ok(voyage.models.some((model) => model.id === "rerank-2.5"));
|
|
assert.ok(voyage.models.some((model) => model.id === "rerank-2.5-lite"));
|
|
|
|
assert.ok(jina);
|
|
assert.equal(jina.baseUrl, "https://api.jina.ai/v1/rerank");
|
|
assert.ok(jina.models.some((model) => model.id === "jina-reranker-v3.5"));
|
|
assert.ok(jina.models.some((model) => model.id === "jina-reranker-v3"));
|
|
assert.ok(jina.models.some((model) => model.id === "jina-reranker-m0"));
|
|
|
|
const parsedVoyage = parseRerankModel("voyage-ai/rerank-2.5");
|
|
assert.equal(parsedVoyage.provider, "voyage-ai");
|
|
assert.equal(parsedVoyage.model, "rerank-2.5");
|
|
|
|
const parsedJina = parseRerankModel("jina-ai/jina-reranker-v3");
|
|
assert.equal(parsedJina.provider, "jina-ai");
|
|
assert.equal(parsedJina.model, "jina-reranker-v3");
|
|
|
|
const parsedJinaAlias = parseRerankModel("jina/jina-reranker-v3");
|
|
assert.equal(parsedJinaAlias.provider, "jina-ai");
|
|
assert.equal(parsedJinaAlias.model, "jina-reranker-v3");
|
|
|
|
const all = getAllRerankModels();
|
|
assert.ok(all.some((model) => model.id === "voyage-ai/rerank-2.5"));
|
|
assert.ok(all.some((model) => model.id === "jina-ai/jina-reranker-v3"));
|
|
});
|
|
|
|
test("upstage embedding registry exposes current embedding models", () => {
|
|
const provider = getEmbeddingProvider("upstage");
|
|
|
|
assert.ok(provider);
|
|
assert.equal(provider.baseUrl, "https://api.upstage.ai/v1/embeddings");
|
|
assert.ok(provider.models.some((model) => model.id === "embedding-query"));
|
|
assert.ok(provider.models.some((model) => model.id === "embedding-passage"));
|
|
|
|
const parsed = parseEmbeddingModel("upstage/embedding-query");
|
|
assert.equal(parsed.provider, "upstage");
|
|
assert.equal(parsed.model, "embedding-query");
|
|
|
|
const all = getAllEmbeddingModels().filter((model) => model.provider === "upstage");
|
|
assert.deepEqual(
|
|
all.map((model) => model.id),
|
|
["upstage/embedding-query", "upstage/embedding-passage"]
|
|
);
|
|
});
|
|
|
|
test("nvidia embedding and rerank parsing preserves provider-prefixed upstream model IDs", () => {
|
|
const parsedEmbedding = parseEmbeddingModel("nvidia/nv-embedqa-e5-v5");
|
|
assert.equal(parsedEmbedding.provider, "nvidia");
|
|
assert.equal(parsedEmbedding.model, "nvidia/nv-embedqa-e5-v5");
|
|
|
|
const parsedDoublePrefixedEmbedding = parseEmbeddingModel("nvidia/nvidia/nv-embedqa-e5-v5");
|
|
assert.equal(parsedDoublePrefixedEmbedding.provider, "nvidia");
|
|
assert.equal(parsedDoublePrefixedEmbedding.model, "nvidia/nv-embedqa-e5-v5");
|
|
|
|
const parsedRerank = parseRerankModel("nvidia/nv-rerankqa-mistral-4b-v3");
|
|
assert.equal(parsedRerank.provider, "nvidia");
|
|
assert.equal(parsedRerank.model, "nvidia/nv-rerankqa-mistral-4b-v3");
|
|
|
|
const parsedDoublePrefixedRerank = parseRerankModel("nvidia/nvidia/nv-rerankqa-mistral-4b-v3");
|
|
assert.equal(parsedDoublePrefixedRerank.provider, "nvidia");
|
|
assert.equal(parsedDoublePrefixedRerank.model, "nvidia/nv-rerankqa-mistral-4b-v3");
|
|
});
|