mirror of
https://github.com/diegosouzapw/OmniRoute.git
synced 2026-09-18 21:02:50 +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.
105 lines
3.1 KiB
TypeScript
105 lines
3.1 KiB
TypeScript
import test from "node:test";
|
|
import assert from "node:assert/strict";
|
|
|
|
import {
|
|
HordeImageCatalog,
|
|
parseHordeImageModels,
|
|
resetAiHordeImageCatalog,
|
|
getCachedAiHordeImageCatalogEntries,
|
|
aiHordeImageCatalog,
|
|
} from "../../open-sse/services/aihordeImageCatalog.ts";
|
|
import {
|
|
parseImageModel,
|
|
getImageProvider,
|
|
getAllImageModels,
|
|
} from "../../open-sse/config/imageRegistry.ts";
|
|
|
|
const HORDE_MODELS = [
|
|
{
|
|
name: "FLUX.1-schnell",
|
|
count: 6,
|
|
queued: 2,
|
|
eta: 12,
|
|
performance: 18.5,
|
|
type: "image",
|
|
},
|
|
{
|
|
name: "AlbedoBase XL (SDXL)",
|
|
count: 1,
|
|
queued: 0,
|
|
eta: 4,
|
|
type: "image",
|
|
},
|
|
{ name: "DeadModel", count: 0, type: "image" },
|
|
{ name: "koboldcpp/Gemma", count: 3, type: "text" },
|
|
];
|
|
|
|
test.afterEach(() => {
|
|
resetAiHordeImageCatalog();
|
|
});
|
|
|
|
test("parseHordeImageModels drops zero-worker and text models", () => {
|
|
const models = parseHordeImageModels(HORDE_MODELS);
|
|
assert.deepEqual(
|
|
models.map((model) => model.name),
|
|
["AlbedoBase XL (SDXL)", "FLUX.1-schnell"]
|
|
);
|
|
assert.equal(models[1].count, 6);
|
|
});
|
|
|
|
test("parseHordeImageModels rejects a non-array payload", () => {
|
|
assert.throws(() => parseHordeImageModels({ name: "nope" }), /JSON array/);
|
|
});
|
|
|
|
test("refresh keeps the last good snapshot when Horde is down", async () => {
|
|
let calls = 0;
|
|
const catalog = new HordeImageCatalog({
|
|
pollMs: 5_000,
|
|
fetchImpl: async () => {
|
|
calls += 1;
|
|
if (calls === 1) {
|
|
return new Response(JSON.stringify(HORDE_MODELS), { status: 200 });
|
|
}
|
|
return new Response(JSON.stringify({ message: "maintenance" }), { status: 503 });
|
|
},
|
|
});
|
|
|
|
await catalog.refresh();
|
|
assert.equal(catalog.isServed("FLUX.1-schnell"), true);
|
|
assert.equal(catalog.snapshot.lastError, null);
|
|
|
|
await catalog.refresh();
|
|
assert.equal(catalog.isServed("FLUX.1-schnell"), true);
|
|
assert.equal(catalog.stale, true);
|
|
assert.ok(catalog.snapshot.lastError);
|
|
assert.equal(
|
|
catalog.listModels().some((model) => model.name === "DeadModel"),
|
|
false
|
|
);
|
|
});
|
|
|
|
test("live catalog entries use exact Horde names and the aihorde prefix", async () => {
|
|
aiHordeImageCatalog.setFetch(
|
|
async () => new Response(JSON.stringify(HORDE_MODELS), { status: 200 })
|
|
);
|
|
await aiHordeImageCatalog.refresh();
|
|
const ids = getCachedAiHordeImageCatalogEntries().map((model) => model.id);
|
|
assert.deepEqual(ids, ["aihorde/AlbedoBase XL (SDXL)", "aihorde/FLUX.1-schnell"]);
|
|
const listed = getAllImageModels().map((model) => model.id);
|
|
assert.ok(listed.includes("aihorde/FLUX.1-schnell"));
|
|
assert.equal(listed.includes("aihorde/DeadModel"), false);
|
|
});
|
|
|
|
test("parseImageModel accepts aihorde/ and horde/ prefixes for live names", () => {
|
|
assert.deepEqual(parseImageModel("aihorde/Flux.1-Schnell fp8 (Compact)"), {
|
|
provider: "aihorde",
|
|
model: "Flux.1-Schnell fp8 (Compact)",
|
|
});
|
|
assert.deepEqual(parseImageModel("horde/AlbedoBase XL (SDXL)"), {
|
|
provider: "aihorde",
|
|
model: "AlbedoBase XL (SDXL)",
|
|
});
|
|
assert.equal(getImageProvider("aihorde")?.format, "aihorde");
|
|
assert.equal(getImageProvider("aihorde")?.alias, "horde");
|
|
});
|