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.
87 lines
2.7 KiB
TypeScript
87 lines
2.7 KiB
TypeScript
import test from "node:test";
|
|
import assert from "node:assert/strict";
|
|
|
|
import { resolveForcedConnectionForCredentialPool } from "../../src/sse/services/sessionAffinityPin.ts";
|
|
|
|
const conn = (id: string, rateLimitedUntil: string | null = null) => ({
|
|
id,
|
|
rateLimitedUntil,
|
|
});
|
|
|
|
test("resolveForcedConnectionForCredentialPool drops forced id when excluded after 429 fallback", () => {
|
|
const excluded = new Set(["dead-account"]);
|
|
assert.equal(
|
|
resolveForcedConnectionForCredentialPool({
|
|
forcedConnectionId: "dead-account",
|
|
excludedConnectionIds: excluded,
|
|
connections: [conn("dead-account"), conn("healthy-account")],
|
|
allowRateLimitedConnections: false,
|
|
bypassQuotaPolicy: false,
|
|
isQuotaExhausted: () => false,
|
|
isQuotaPolicyBlocked: () => false,
|
|
}),
|
|
null
|
|
);
|
|
});
|
|
|
|
test("resolveForcedConnectionForCredentialPool keeps forced id when eligible", () => {
|
|
assert.equal(
|
|
resolveForcedConnectionForCredentialPool({
|
|
forcedConnectionId: "healthy-account",
|
|
excludedConnectionIds: new Set(),
|
|
connections: [conn("healthy-account"), conn("other-account")],
|
|
allowRateLimitedConnections: false,
|
|
bypassQuotaPolicy: false,
|
|
isQuotaExhausted: () => false,
|
|
isQuotaPolicyBlocked: () => false,
|
|
}),
|
|
"healthy-account"
|
|
);
|
|
});
|
|
|
|
test("resolveForcedConnectionForCredentialPool drops forced id on cooldown", () => {
|
|
const future = new Date(Date.now() + 60_000).toISOString();
|
|
assert.equal(
|
|
resolveForcedConnectionForCredentialPool({
|
|
forcedConnectionId: "cooling-account",
|
|
excludedConnectionIds: new Set(),
|
|
connections: [conn("cooling-account", future)],
|
|
allowRateLimitedConnections: false,
|
|
bypassQuotaPolicy: false,
|
|
isQuotaExhausted: () => false,
|
|
isQuotaPolicyBlocked: () => false,
|
|
}),
|
|
null
|
|
);
|
|
});
|
|
|
|
test("resolveForcedConnectionForCredentialPool drops forced id when quota exhausted", () => {
|
|
assert.equal(
|
|
resolveForcedConnectionForCredentialPool({
|
|
forcedConnectionId: "exhausted-account",
|
|
excludedConnectionIds: new Set(),
|
|
connections: [conn("exhausted-account")],
|
|
allowRateLimitedConnections: false,
|
|
bypassQuotaPolicy: false,
|
|
isQuotaExhausted: (id) => id === "exhausted-account",
|
|
isQuotaPolicyBlocked: () => false,
|
|
}),
|
|
null
|
|
);
|
|
});
|
|
|
|
test("resolveForcedConnectionForCredentialPool with empty connections only checks exclusion", () => {
|
|
assert.equal(
|
|
resolveForcedConnectionForCredentialPool({
|
|
forcedConnectionId: "pinned-account",
|
|
excludedConnectionIds: new Set(),
|
|
connections: [],
|
|
allowRateLimitedConnections: false,
|
|
bypassQuotaPolicy: false,
|
|
isQuotaExhausted: () => true,
|
|
isQuotaPolicyBlocked: () => true,
|
|
}),
|
|
"pinned-account"
|
|
);
|
|
});
|