mirror of
https://github.com/diegosouzapw/OmniRoute.git
synced 2026-09-14 02:42:24 +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.
39 lines
1.4 KiB
TypeScript
39 lines
1.4 KiB
TypeScript
import { test } from "node:test";
|
|
import assert from "node:assert/strict";
|
|
import Bottleneck from "bottleneck";
|
|
|
|
const { runAsProbe, isProbeContext } = await import("../../src/shared/utils/probeOrigin.ts");
|
|
|
|
test("runAsProbe propagates through nested async/await; false outside", async () => {
|
|
assert.equal(isProbeContext(), false);
|
|
await runAsProbe(async () => {
|
|
assert.equal(isProbeContext(), true);
|
|
await Promise.resolve();
|
|
const inner = async () => {
|
|
await Promise.resolve();
|
|
return isProbeContext();
|
|
};
|
|
assert.equal(await inner(), true);
|
|
await new Promise((r) => setTimeout(r, 5));
|
|
assert.equal(isProbeContext(), true);
|
|
});
|
|
assert.equal(isProbeContext(), false);
|
|
});
|
|
|
|
test("queued scheduler job wrapped in runAsProbe keeps the probe context", async () => {
|
|
const limiter = new Bottleneck({ maxConcurrent: 1, minTime: 50 });
|
|
let insideQueuedJob: boolean | null = null;
|
|
// Schedule job 1 WITHOUT awaiting it so job 2 really queues behind it
|
|
// (maxConcurrent 1 + minTime 50) — the queued job must execute with the
|
|
// probe context established at scheduling time.
|
|
const job1 = limiter.schedule(() => new Promise((r) => setTimeout(r, 30)));
|
|
const job2 = limiter.schedule(() =>
|
|
runAsProbe(() => {
|
|
insideQueuedJob = isProbeContext();
|
|
return Promise.resolve();
|
|
})
|
|
);
|
|
await Promise.all([job1, job2]);
|
|
assert.equal(insideQueuedJob, true);
|
|
});
|