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.
54 lines
1.6 KiB
TypeScript
54 lines
1.6 KiB
TypeScript
import test from "node:test";
|
|
import assert from "node:assert/strict";
|
|
|
|
import { checkFallbackError, getProviderProfile } from "../../open-sse/services/accountFallback.ts";
|
|
import { RateLimitReason } from "../../open-sse/config/constants.ts";
|
|
|
|
const ANTIGRAVITY_RESET_TEXT =
|
|
"Individual quota reached. Contact your administrator to enable overages. Resets in 164h27m24s.";
|
|
const RESET_164H = (164 * 3600 + 27 * 60 + 24) * 1000;
|
|
|
|
function antigravityProfile(useUpstreamRetryHints: boolean) {
|
|
return {
|
|
...getProviderProfile("antigravity"),
|
|
baseCooldownMs: 125,
|
|
transientCooldown: 125,
|
|
rateLimitCooldown: useUpstreamRetryHints ? 0 : 125,
|
|
useUpstreamRetryHints,
|
|
};
|
|
}
|
|
|
|
test("checkFallbackError ignores body reset text when upstream retry hints are disabled", () => {
|
|
const result = checkFallbackError(
|
|
429,
|
|
ANTIGRAVITY_RESET_TEXT,
|
|
0,
|
|
"gemini-3.7-flash-high",
|
|
"antigravity",
|
|
null,
|
|
antigravityProfile(false)
|
|
);
|
|
|
|
assert.equal(result.shouldFallback, true);
|
|
assert.equal(result.reason, RateLimitReason.QUOTA_EXHAUSTED);
|
|
assert.equal(result.usedUpstreamRetryHint, false);
|
|
assert.equal(result.cooldownMs, 125);
|
|
});
|
|
|
|
test("checkFallbackError honors body reset text when upstream retry hints are enabled", () => {
|
|
const result = checkFallbackError(
|
|
429,
|
|
ANTIGRAVITY_RESET_TEXT,
|
|
0,
|
|
"gemini-3.7-flash-high",
|
|
"antigravity",
|
|
null,
|
|
antigravityProfile(true)
|
|
);
|
|
|
|
assert.equal(result.shouldFallback, true);
|
|
assert.equal(result.reason, RateLimitReason.QUOTA_EXHAUSTED);
|
|
assert.equal(result.usedUpstreamRetryHint, true);
|
|
assert.equal(result.cooldownMs, RESET_164H);
|
|
});
|