Files
OmniRoute/tests/unit/token-kiosk-provider.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

38 lines
1.8 KiB
TypeScript

import test from "node:test";
import assert from "node:assert/strict";
const { APIKEY_PROVIDERS } = await import("../../src/shared/constants/providers.ts");
const { PROVIDER_ENDPOINTS } = await import("../../src/shared/constants/config.ts");
const { REGISTRY: providerRegistry } = await import("../../open-sse/config/providerRegistry.ts");
const { isNamedOpenAIStyleProvider } = await import("../../src/app/api/providers/[id]/models/discovery/providerSets.ts");
test("Token Kiosk is registered as an API-key gateway provider", () => {
const entry = APIKEY_PROVIDERS["token-kiosk"];
assert.ok(entry, "APIKEY_PROVIDERS['token-kiosk'] must be defined");
assert.equal(entry.id, "token-kiosk");
assert.equal(entry.alias, "tk");
assert.equal(entry.name, "Token Kiosk");
assert.equal(entry.website, "https://agent-router.gaib.ai");
});
test("Token Kiosk exposes the OpenAI-compatible chat completions URL", () => {
assert.equal(PROVIDER_ENDPOINTS["token-kiosk"], "https://agent-router.gaib.ai/v1/chat/completions");
});
test("Token Kiosk registry entry uses OpenAI format with bearer apikey auth", () => {
const entry = providerRegistry["token-kiosk"];
assert.ok(entry, "providerRegistry['token-kiosk'] must be defined");
assert.equal(entry.id, "token-kiosk");
assert.equal(entry.alias, "tk");
assert.equal(entry.format, "openai");
assert.equal(entry.executor, "default");
assert.equal(entry.authType, "apikey");
assert.equal(entry.authHeader, "bearer");
assert.equal(entry.baseUrl, "https://agent-router.gaib.ai/v1/chat/completions");
assert.equal(entry.modelsUrl, "https://agent-router.gaib.ai/v1/models");
});
test("Token Kiosk is registered as a named OpenAI-style provider for live model discovery", () => {
assert.ok(isNamedOpenAIStyleProvider("token-kiosk"));
});