mirror of
https://github.com/diegosouzapw/OmniRoute.git
synced 2026-09-13 18:32:12 +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.
56 lines
2.2 KiB
TypeScript
56 lines
2.2 KiB
TypeScript
/**
|
|
* A synced model delete removes the current discovery row only. If the upstream
|
|
* provider advertises that model again, the next sync restores it. Persistent
|
|
* exclusion is represented by isHidden, not a separate deletion tombstone.
|
|
*/
|
|
import test, { before, after } from "node:test";
|
|
import assert from "node:assert/strict";
|
|
import os from "node:os";
|
|
import path from "node:path";
|
|
import fs from "node:fs";
|
|
|
|
// Hermetic DB: this test mutates the syncedAvailableModels key_value namespace.
|
|
const tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), "omniroute-test-synced-del-"));
|
|
process.env.DATA_DIR = tmpDir;
|
|
|
|
const { replaceSyncedAvailableModelsForConnection, getSyncedAvailableModels } =
|
|
await import("../../src/lib/localDb.ts");
|
|
const { resetDbInstance } = await import("../../src/lib/db/core.ts");
|
|
|
|
before(() => {
|
|
resetDbInstance();
|
|
});
|
|
|
|
after(() => {
|
|
resetDbInstance();
|
|
fs.rmSync(tmpDir, { recursive: true, force: true });
|
|
});
|
|
|
|
test("a deleted synced model is restored when upstream advertises it again", async () => {
|
|
const provider = "llama-cpp";
|
|
const connectionId = "conn-3199";
|
|
|
|
// Initial sync brings in two models.
|
|
await replaceSyncedAvailableModelsForConnection(provider, connectionId, [
|
|
{ id: "model-keep", name: "Keep" },
|
|
{ id: "model-del", name: "Delete me" },
|
|
]);
|
|
let synced = (await getSyncedAvailableModels(provider)).map((m) => m.id);
|
|
assert.ok(synced.includes("model-del"), "both models present after first sync");
|
|
|
|
const { removeSyncedAvailableModel } = await import("../../src/lib/localDb.ts");
|
|
assert.equal(await removeSyncedAvailableModel(provider, "model-del"), true);
|
|
synced = (await getSyncedAvailableModels(provider)).map((m) => m.id);
|
|
assert.ok(!synced.includes("model-del"), "delete removes the current synced row");
|
|
|
|
// Auto-fetch re-imports the SAME upstream list (still advertising model-del).
|
|
await replaceSyncedAvailableModelsForConnection(provider, connectionId, [
|
|
{ id: "model-keep", name: "Keep" },
|
|
{ id: "model-del", name: "Delete me" },
|
|
]);
|
|
|
|
synced = (await getSyncedAvailableModels(provider)).map((m) => m.id);
|
|
assert.ok(synced.includes("model-keep"), "unrelated model stays");
|
|
assert.ok(synced.includes("model-del"), "upstream re-sync restores the deleted model");
|
|
});
|