@omnicross/core 0.1.1 → 0.1.3
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/ApiConverter.cjs +3 -778
- package/dist/ApiConverter.js +12 -751
- package/dist/{CompletionService-DtOF_War.d.cts → CompletionService-D3a47vdU.d.cts} +2 -1
- package/dist/{CompletionService-7fCmKAP3.d.ts → CompletionService-DdWaAxq0.d.ts} +2 -1
- package/dist/auth/GeminiCodeAssistProjectResolver.cjs +32 -60
- package/dist/auth/GeminiCodeAssistProjectResolver.js +11 -10
- package/dist/builtin-web-fetch-77JZUZGM.cjs +100 -0
- package/dist/builtin-web-fetch-IGLBX2XY.js +100 -0
- package/dist/chunk-2DCNB7DF.cjs +514 -0
- package/dist/chunk-2FMI2B6V.cjs +6196 -0
- package/dist/chunk-3OPYJG76.cjs +266 -0
- package/dist/chunk-3QOSDDNI.cjs +130 -0
- package/dist/chunk-46RLHI7D.js +0 -0
- package/dist/chunk-46ZIFWYG.cjs +277 -0
- package/dist/chunk-4DECIFU7.js +6196 -0
- package/dist/chunk-4IH4EL7M.cjs +9 -0
- package/dist/chunk-4VAVYUZS.cjs +991 -0
- package/dist/chunk-5ENKBSWO.js +311 -0
- package/dist/chunk-5ERIZW3D.js +44 -0
- package/dist/chunk-5HTVET6E.js +514 -0
- package/dist/chunk-5ZQBEOMD.js +764 -0
- package/dist/chunk-6RPZADX3.cjs +92 -0
- package/dist/chunk-7VU7V2E4.js +0 -0
- package/dist/chunk-7YN22L2H.cjs +87 -0
- package/dist/chunk-A2YMUCBT.cjs +105 -0
- package/dist/chunk-AEOZIDEB.cjs +32 -0
- package/dist/chunk-AQ5TNQB7.js +13 -0
- package/dist/chunk-AVYJ4YBN.js +58 -0
- package/dist/chunk-BO4RFWBR.js +89 -0
- package/dist/chunk-C72S6TTV.cjs +49 -0
- package/dist/chunk-C7PSERMH.js +87 -0
- package/dist/chunk-CDHS2QDC.js +179 -0
- package/dist/chunk-CWM2ZDU5.js +130 -0
- package/dist/chunk-CZHZGRU7.js +383 -0
- package/dist/chunk-DK4A7DLE.js +266 -0
- package/dist/chunk-DQ5VAYZ6.cjs +42 -0
- package/dist/chunk-DRWHTIIZ.js +13 -0
- package/dist/chunk-DWKQZ2SK.cjs +254 -0
- package/dist/chunk-EYZYXJTJ.cjs +1 -0
- package/dist/chunk-FFALUUZZ.cjs +235 -0
- package/dist/chunk-FMOBD2OQ.js +39 -0
- package/dist/chunk-FTQ7IQUC.js +991 -0
- package/dist/chunk-FZ46M34D.cjs +1 -0
- package/dist/chunk-G2UTNSGU.js +277 -0
- package/dist/chunk-GBHYAGX3.cjs +26 -0
- package/dist/chunk-GMMT7RVN.cjs +39 -0
- package/dist/chunk-H5JUT3KV.js +105 -0
- package/dist/chunk-HQP3K7PU.cjs +431 -0
- package/dist/chunk-HUZZZ3JL.js +32 -0
- package/dist/chunk-IAXWICZU.js +69 -0
- package/dist/chunk-JMJBSACD.cjs +147 -0
- package/dist/chunk-JO5NLTLY.js +46 -0
- package/dist/chunk-JWC6APFQ.js +23 -0
- package/dist/chunk-KUU2RNG6.cjs +46 -0
- package/dist/chunk-L74VDN4A.cjs +32 -0
- package/dist/chunk-MQNDWYPN.js +235 -0
- package/dist/chunk-MZNPGW5Q.js +26 -0
- package/dist/chunk-N5QFKSDO.js +56 -0
- package/dist/chunk-NYUNU75A.cjs +44 -0
- package/dist/chunk-NZYN7C3P.cjs +179 -0
- package/dist/chunk-Q27JY5RG.cjs +126 -0
- package/dist/chunk-QBZJ7P2T.cjs +13 -0
- package/dist/chunk-QWBWGODS.cjs +69 -0
- package/dist/chunk-QXD4MV2X.cjs +23 -0
- package/dist/chunk-R3HK6R4B.cjs +13 -0
- package/dist/chunk-RPII3E6Z.cjs +145 -0
- package/dist/chunk-RSJ3TWJQ.cjs +186 -0
- package/dist/chunk-S4WBV25P.js +913 -0
- package/dist/chunk-S6JGLF3G.cjs +89 -0
- package/dist/chunk-SUEGJ75D.cjs +764 -0
- package/dist/chunk-SVNDB62D.cjs +58 -0
- package/dist/chunk-SVXC36MU.cjs +383 -0
- package/dist/chunk-TKJMZBAH.js +32 -0
- package/dist/chunk-UBWEBKRC.js +42 -0
- package/dist/chunk-UNEFIWXI.cjs +32 -0
- package/dist/chunk-VQY5W4YW.cjs +311 -0
- package/dist/chunk-VVEHS2LI.js +431 -0
- package/dist/chunk-WNY4WMRF.js +32 -0
- package/dist/chunk-XEKD23J5.js +9 -0
- package/dist/chunk-XTEI64OU.js +145 -0
- package/dist/chunk-XX6NQJMA.js +92 -0
- package/dist/chunk-Y6OM7EV7.cjs +56 -0
- package/dist/chunk-Y7FO65VT.js +147 -0
- package/dist/chunk-YGXJGRFI.cjs +913 -0
- package/dist/chunk-YMFAJFQT.cjs +173 -0
- package/dist/chunk-Z7LUNECE.js +49 -0
- package/dist/chunk-Z7Y2NPHY.js +186 -0
- package/dist/chunk-ZJZAOKI7.js +254 -0
- package/dist/chunk-ZLX3KGXX.js +173 -0
- package/dist/chunk-ZY6KU6P3.js +126 -0
- package/dist/completion/ApiKeyPoolService.cjs +23 -48
- package/dist/completion/ApiKeyPoolService.d.cts +150 -2
- package/dist/completion/ApiKeyPoolService.d.ts +150 -2
- package/dist/completion/BuiltinToolExecutor.cjs +9 -324
- package/dist/completion/BuiltinToolExecutor.js +6 -290
- package/dist/completion/CompletionService.cjs +45 -3487
- package/dist/completion/CompletionService.d.cts +3 -2
- package/dist/completion/CompletionService.d.ts +3 -2
- package/dist/completion/CompletionService.js +41 -3457
- package/dist/completion/NativeSearchInjector.cjs +8 -192
- package/dist/completion/NativeSearchInjector.js +7 -162
- package/dist/completion/ProviderSearchInjector.cjs +12 -39
- package/dist/completion/native-search-types.cjs +13 -66
- package/dist/completion/native-search-types.js +7 -31
- package/dist/completion/openrouter-headers.cjs +12 -71
- package/dist/completion/openrouter-headers.d.cts +9 -12
- package/dist/completion/openrouter-headers.d.ts +9 -12
- package/dist/completion/openrouter-headers.js +7 -36
- package/dist/completion/openrouter-models.cjs +11 -67
- package/dist/completion/openrouter-models.js +5 -34
- package/dist/completion/types.cjs +1 -18
- package/dist/completion/url-builder.cjs +24 -138
- package/dist/completion/url-builder.js +12 -92
- package/dist/completion.cjs +62 -3989
- package/dist/completion.d.cts +3 -2
- package/dist/completion.d.ts +3 -2
- package/dist/completion.js +70 -3953
- package/dist/index.cjs +319 -10429
- package/dist/index.d.cts +49 -12
- package/dist/index.d.ts +49 -12
- package/dist/index.js +248 -10269
- package/dist/logger-4GvQNzhE.d.cts +22 -0
- package/dist/logger-4GvQNzhE.d.ts +22 -0
- package/dist/outbound-api/routeResolver.cjs +10 -217
- package/dist/outbound-api/routeResolver.d.cts +10 -3
- package/dist/outbound-api/routeResolver.d.ts +10 -3
- package/dist/outbound-api/routeResolver.js +9 -187
- package/dist/outbound-api/subscriptionRegistryPort.cjs +7 -37
- package/dist/outbound-api/subscriptionRegistryPort.d.cts +2 -1
- package/dist/outbound-api/subscriptionRegistryPort.d.ts +2 -1
- package/dist/outbound-api/subscriptionRegistryPort.js +4 -8
- package/dist/outbound-api/types.cjs +5 -17
- package/dist/outbound-api/types.d.cts +12 -144
- package/dist/outbound-api/types.d.ts +12 -144
- package/dist/outbound-api/types.js +6 -0
- package/dist/outbound-api.cjs +197 -5265
- package/dist/outbound-api.d.cts +787 -25
- package/dist/outbound-api.d.ts +787 -25
- package/dist/outbound-api.js +173 -5195
- package/dist/pipeline/AuthSource.cjs +1 -18
- package/dist/pipeline/AuthSource.d.cts +23 -1
- package/dist/pipeline/AuthSource.d.ts +23 -1
- package/dist/pipeline/LlmConfigProviderAuth.cjs +5 -166
- package/dist/pipeline/LlmConfigProviderAuth.d.cts +2 -1
- package/dist/pipeline/LlmConfigProviderAuth.d.ts +2 -1
- package/dist/pipeline/LlmConfigProviderAuth.js +5 -139
- package/dist/pipeline/SubscriptionAccountHealth.cjs +22 -0
- package/dist/pipeline/SubscriptionAccountHealth.d.cts +235 -0
- package/dist/pipeline/SubscriptionAccountHealth.d.ts +235 -0
- package/dist/pipeline/SubscriptionAccountHealth.js +22 -0
- package/dist/pipeline/SubscriptionAuthSource.cjs +6 -128
- package/dist/pipeline/SubscriptionAuthSource.d.cts +3 -2
- package/dist/pipeline/SubscriptionAuthSource.d.ts +3 -2
- package/dist/pipeline/SubscriptionAuthSource.js +5 -99
- package/dist/pipeline/SubscriptionAuthStrategy.cjs +1 -18
- package/dist/pipeline/SubscriptionAuthStrategy.d.cts +24 -1
- package/dist/pipeline/SubscriptionAuthStrategy.d.ts +24 -1
- package/dist/pipeline/auditSink.cjs +14 -0
- package/dist/pipeline/auditSink.d.cts +48 -0
- package/dist/pipeline/auditSink.d.ts +48 -0
- package/dist/pipeline/auditSink.js +14 -0
- package/dist/pipeline/billingEmit.cjs +14 -0
- package/dist/pipeline/billingEmit.d.cts +49 -0
- package/dist/pipeline/billingEmit.d.ts +49 -0
- package/dist/pipeline/billingEmit.js +14 -0
- package/dist/pipeline/executeProviderCall.cjs +5 -69
- package/dist/pipeline/executeProviderCall.js +3 -42
- package/dist/pipeline/resolveProviderChain.cjs +5 -46
- package/dist/pipeline/resolveProviderChain.js +3 -19
- package/dist/pipeline/resolveSubscriptionChain.cjs +5 -67
- package/dist/pipeline/resolveSubscriptionChain.js +3 -40
- package/dist/ports/gemini-code-assist-resolver.cjs +7 -37
- package/dist/ports/gemini-code-assist-resolver.js +4 -8
- package/dist/ports/pricing-store.cjs +1 -0
- package/dist/ports/pricing-store.d.cts +50 -0
- package/dist/ports/pricing-store.d.ts +50 -0
- package/dist/ports/pricing-store.js +0 -0
- package/dist/ports/provider-config-source.cjs +1 -18
- package/dist/ports/usage-event-store.cjs +1 -0
- package/dist/ports/usage-event-store.d.cts +54 -0
- package/dist/ports/usage-event-store.d.ts +54 -0
- package/dist/ports/usage-event-store.js +0 -0
- package/dist/ports/web-search-backend.cjs +1 -18
- package/dist/ports.cjs +1 -18
- package/dist/ports.d.cts +13 -2
- package/dist/ports.d.ts +13 -2
- package/dist/provider-proxy/ProviderProxy.cjs +47 -4643
- package/dist/provider-proxy/ProviderProxy.d.cts +2 -1
- package/dist/provider-proxy/ProviderProxy.d.ts +2 -1
- package/dist/provider-proxy/ProviderProxy.js +42 -4613
- package/dist/provider-proxy/identity/SubscriptionIdentityStore.cjs +15 -0
- package/dist/provider-proxy/identity/SubscriptionIdentityStore.d.cts +121 -0
- package/dist/provider-proxy/identity/SubscriptionIdentityStore.d.ts +121 -0
- package/dist/provider-proxy/identity/SubscriptionIdentityStore.js +15 -0
- package/dist/provider-proxy/identity/fingerprintHeaders.cjs +20 -0
- package/dist/provider-proxy/identity/fingerprintHeaders.d.cts +89 -0
- package/dist/provider-proxy/identity/fingerprintHeaders.d.ts +89 -0
- package/dist/provider-proxy/identity/fingerprintHeaders.js +20 -0
- package/dist/provider-proxy/ingress/providerProxyShared.cjs +42 -2941
- package/dist/provider-proxy/ingress/providerProxyShared.d.cts +13 -3
- package/dist/provider-proxy/ingress/providerProxyShared.d.ts +13 -3
- package/dist/provider-proxy/ingress/providerProxyShared.js +48 -2914
- package/dist/provider-proxy/matchText.cjs +13 -72
- package/dist/provider-proxy/matchText.d.cts +14 -1
- package/dist/provider-proxy/matchText.d.ts +14 -1
- package/dist/provider-proxy/matchText.js +8 -39
- package/dist/provider-proxy/types.cjs +1 -18
- package/dist/provider-proxy/types.d.cts +25 -1
- package/dist/provider-proxy/types.d.ts +25 -1
- package/dist/provider-proxy.cjs +42 -4655
- package/dist/provider-proxy.d.cts +2 -1
- package/dist/provider-proxy.d.ts +2 -1
- package/dist/provider-proxy.js +46 -4628
- package/dist/{routeResolver-HE-ZO0fO.d.ts → routeResolver-B1NglTfe.d.ts} +36 -10
- package/dist/{routeResolver-BrbK6ja9.d.cts → routeResolver-CbG5uBgm.d.cts} +36 -10
- package/dist/serializeError.cjs +5 -81
- package/dist/serializeError.js +3 -54
- package/dist/sse-parser.cjs +7 -455
- package/dist/sse-parser.js +4 -426
- package/dist/transformer/TransformerChainExecutor.cjs +4 -318
- package/dist/transformer/TransformerChainExecutor.js +4 -291
- package/dist/transformer/TransformerService.cjs +5 -289
- package/dist/transformer/TransformerService.js +3 -262
- package/dist/transformer/anthropicBetaInject.cjs +7 -50
- package/dist/transformer/anthropicBetaInject.js +4 -21
- package/dist/transformer/transformers/AnthropicTransformer.cjs +3 -1014
- package/dist/transformer/transformers/AnthropicTransformer.js +3 -987
- package/dist/transformer/transformers/GeminiCodeAssistTransformer.cjs +4 -1106
- package/dist/transformer/transformers/GeminiCodeAssistTransformer.js +7 -1079
- package/dist/transformer/transformers/GeminiTransformer.cjs +4 -1010
- package/dist/transformer/transformers/GeminiTransformer.js +4 -983
- package/dist/transformer/transformers/OpenAIResponseTransformer.cjs +3 -535
- package/dist/transformer/transformers/OpenAIResponseTransformer.js +3 -510
- package/dist/transformer/transformers/OpenCodeGoTransformer.cjs +5 -72
- package/dist/transformer/transformers/OpenCodeGoTransformer.js +3 -45
- package/dist/transformer/transformers/ReasoningTransformer.cjs +3 -270
- package/dist/transformer/transformers/ReasoningTransformer.js +2 -249
- package/dist/transformer/transformers.cjs +10 -3177
- package/dist/transformer/transformers.js +23 -3158
- package/dist/transformer/types.cjs +1 -18
- package/dist/transformer.cjs +13 -3726
- package/dist/transformer.js +17 -3706
- package/dist/types-6OUY_xfA.d.ts +830 -0
- package/dist/types-aBFFXoBz.d.cts +830 -0
- package/dist/usage/pricing-engine.cjs +6 -0
- package/dist/usage/pricing-engine.d.cts +83 -0
- package/dist/usage/pricing-engine.d.ts +83 -0
- package/dist/usage/pricing-engine.js +6 -0
- package/dist/usage/usage-recorder.cjs +6 -0
- package/dist/usage/usage-recorder.d.cts +91 -0
- package/dist/usage/usage-recorder.d.ts +91 -0
- package/dist/usage/usage-recorder.js +6 -0
- package/dist/usage.cjs +11 -0
- package/dist/usage.d.cts +8 -0
- package/dist/usage.d.ts +8 -0
- package/dist/usage.js +11 -0
- package/package.json +4 -2
- package/dist/ApiKeyPoolService-BmMkau07.d.cts +0 -170
- package/dist/ApiKeyPoolService-BmMkau07.d.ts +0 -170
|
@@ -0,0 +1,311 @@
|
|
|
1
|
+
// src/pipeline/SubscriptionAccountHealth.ts
|
|
2
|
+
var OVERLOAD_TTL_MS = 10 * 6e4;
|
|
3
|
+
var OVERLOAD_ENABLED_DEFAULT = true;
|
|
4
|
+
var AUTH_ERROR_TTL_MS = 30 * 6e4;
|
|
5
|
+
var SERVER_ERROR_TTL_MS = 5 * 6e4;
|
|
6
|
+
var BAN_BODY_MARKERS = [
|
|
7
|
+
"this organization has been disabled",
|
|
8
|
+
"oauth authentication is currently not allowed"
|
|
9
|
+
];
|
|
10
|
+
function readHeader(headers, name) {
|
|
11
|
+
if (typeof headers.get === "function") {
|
|
12
|
+
return headers.get(name);
|
|
13
|
+
}
|
|
14
|
+
const rec = headers;
|
|
15
|
+
return rec[name] ?? rec[name.toLowerCase()] ?? null;
|
|
16
|
+
}
|
|
17
|
+
function parseIntHeader(value) {
|
|
18
|
+
if (value == null) return null;
|
|
19
|
+
const n = Number.parseInt(value.trim(), 10);
|
|
20
|
+
return Number.isFinite(n) ? n : null;
|
|
21
|
+
}
|
|
22
|
+
function resolveResetSeconds(providerId, headers) {
|
|
23
|
+
const unified = parseIntHeader(readHeader(headers, "anthropic-ratelimit-unified-reset"));
|
|
24
|
+
if (providerId === "claude") {
|
|
25
|
+
return { resetHeaderSeconds: unified, retryAfterSeconds: null };
|
|
26
|
+
}
|
|
27
|
+
const providerReset = parseIntHeader(readHeader(headers, "x-ratelimit-reset")) ?? parseIntHeader(readHeader(headers, "ratelimit-reset"));
|
|
28
|
+
const retryAfter = parseIntHeader(readHeader(headers, "retry-after"));
|
|
29
|
+
return {
|
|
30
|
+
resetHeaderSeconds: unified ?? providerReset,
|
|
31
|
+
retryAfterSeconds: retryAfter
|
|
32
|
+
};
|
|
33
|
+
}
|
|
34
|
+
function isBanBody(bodyText) {
|
|
35
|
+
if (!bodyText) return false;
|
|
36
|
+
const lower = bodyText.toLowerCase();
|
|
37
|
+
return BAN_BODY_MARKERS.some((marker) => lower.includes(marker));
|
|
38
|
+
}
|
|
39
|
+
function isRecordEmpty(record) {
|
|
40
|
+
return record.rateLimitEndAt === void 0 && record.overloadUntil === void 0 && record.tempUnavailableUntil === void 0 && !record.blocked;
|
|
41
|
+
}
|
|
42
|
+
var KEY_SEP = "\0";
|
|
43
|
+
var SubscriptionAccountHealth = class {
|
|
44
|
+
records = /* @__PURE__ */ new Map();
|
|
45
|
+
listeners = /* @__PURE__ */ new Set();
|
|
46
|
+
anomalyListeners = /* @__PURE__ */ new Set();
|
|
47
|
+
now;
|
|
48
|
+
overloadEnabled;
|
|
49
|
+
overloadTtlMs;
|
|
50
|
+
authErrorTtlMs;
|
|
51
|
+
serverErrorTtlMs;
|
|
52
|
+
constructor(opts = {}) {
|
|
53
|
+
this.now = opts.now ?? Date.now;
|
|
54
|
+
this.overloadEnabled = opts.overloadEnabled ?? OVERLOAD_ENABLED_DEFAULT;
|
|
55
|
+
this.overloadTtlMs = opts.overloadTtlMs ?? OVERLOAD_TTL_MS;
|
|
56
|
+
this.authErrorTtlMs = opts.authErrorTtlMs ?? AUTH_ERROR_TTL_MS;
|
|
57
|
+
this.serverErrorTtlMs = opts.serverErrorTtlMs ?? SERVER_ERROR_TTL_MS;
|
|
58
|
+
}
|
|
59
|
+
/**
|
|
60
|
+
* Re-apply cooldown config to the LIVE shared instance (subscription-account-
|
|
61
|
+
* health, LEAD OQ1). `buildDaemon` is synchronous so it cannot await the
|
|
62
|
+
* persisted server config; the async `start.ts` path loads it and applies the
|
|
63
|
+
* `accountHealth` segment here — every strategy/relay/sweeper that already
|
|
64
|
+
* captured this instance picks the new values up. Only defined fields override.
|
|
65
|
+
*/
|
|
66
|
+
configure(opts) {
|
|
67
|
+
if (opts.overloadEnabled !== void 0) this.overloadEnabled = opts.overloadEnabled;
|
|
68
|
+
if (opts.overloadTtlMs !== void 0) this.overloadTtlMs = opts.overloadTtlMs;
|
|
69
|
+
if (opts.authErrorTtlMs !== void 0) this.authErrorTtlMs = opts.authErrorTtlMs;
|
|
70
|
+
if (opts.serverErrorTtlMs !== void 0) this.serverErrorTtlMs = opts.serverErrorTtlMs;
|
|
71
|
+
}
|
|
72
|
+
/**
|
|
73
|
+
* Whether an account may be scheduled RIGHT NOW. `blocked` → false; any timer
|
|
74
|
+
* still in the future → false. Expired timers are lazily deleted on read (CRS
|
|
75
|
+
* lazy-clear parity) so an elapsed cooldown restores the account WITHOUT a
|
|
76
|
+
* timer — but the recovery SIGNAL is the sweeper's job (`sweepRecoveries`), so
|
|
77
|
+
* this read never emits (correctness is independent of the tick).
|
|
78
|
+
*/
|
|
79
|
+
isSchedulable(providerId, accountId, now = this.now()) {
|
|
80
|
+
const key = this.key(providerId, accountId);
|
|
81
|
+
const record = this.records.get(key);
|
|
82
|
+
if (!record) return true;
|
|
83
|
+
if (record.blocked) return false;
|
|
84
|
+
const active = this.clearExpired(record, now);
|
|
85
|
+
if (!active && isRecordEmpty(record)) this.records.delete(key);
|
|
86
|
+
return !active;
|
|
87
|
+
}
|
|
88
|
+
/**
|
|
89
|
+
* The single marking entry point (design D3). Faithful to CRS status semantics:
|
|
90
|
+
* - 429 + authoritative reset → `rateLimitEndAt`; a non-claude 429 may use
|
|
91
|
+
* `retryAfterSeconds`; a bare 429 (no resolvable reset) is NOT marked;
|
|
92
|
+
* - 529 → `overloadUntil` (gated by `overloadEnabled`);
|
|
93
|
+
* - 403-ban body → `blocked` (permanent); plain 403 → transient;
|
|
94
|
+
* - final-401 → transient; 5xx / thrown(null) → transient;
|
|
95
|
+
* - non-429 4xx → NEUTRAL (never marked);
|
|
96
|
+
* - 2xx → clear rate-limit + transient state (keep `blocked`).
|
|
97
|
+
*/
|
|
98
|
+
recordUpstreamOutcome(providerId, accountId, input) {
|
|
99
|
+
const now = input.now ?? this.now();
|
|
100
|
+
const key = this.key(providerId, accountId);
|
|
101
|
+
const preExisting = this.records.get(key);
|
|
102
|
+
const record = preExisting ?? {};
|
|
103
|
+
const status = input.status;
|
|
104
|
+
const wasUnhealthyBefore = preExisting ? this.isUnhealthyAt(preExisting, now) : false;
|
|
105
|
+
let anomalyState;
|
|
106
|
+
let recovered = false;
|
|
107
|
+
if (status === null) {
|
|
108
|
+
record.tempUnavailableUntil = now + this.serverErrorTtlMs;
|
|
109
|
+
} else if (status >= 200 && status < 300) {
|
|
110
|
+
const wasUnhealthy = record.rateLimitEndAt !== void 0 || record.overloadUntil !== void 0 || record.tempUnavailableUntil !== void 0;
|
|
111
|
+
delete record.rateLimitEndAt;
|
|
112
|
+
delete record.overloadUntil;
|
|
113
|
+
delete record.tempUnavailableUntil;
|
|
114
|
+
recovered = wasUnhealthy && !record.blocked;
|
|
115
|
+
} else if (status === 429) {
|
|
116
|
+
if (input.resetHeaderSeconds != null) {
|
|
117
|
+
record.rateLimitEndAt = input.resetHeaderSeconds * 1e3;
|
|
118
|
+
anomalyState = "rate_limited";
|
|
119
|
+
} else if (input.retryAfterSeconds != null) {
|
|
120
|
+
record.rateLimitEndAt = now + input.retryAfterSeconds * 1e3;
|
|
121
|
+
anomalyState = "rate_limited";
|
|
122
|
+
}
|
|
123
|
+
} else if (status === 529) {
|
|
124
|
+
if (this.overloadEnabled) {
|
|
125
|
+
record.overloadUntil = now + this.overloadTtlMs;
|
|
126
|
+
anomalyState = "overloaded";
|
|
127
|
+
}
|
|
128
|
+
} else if (status === 403) {
|
|
129
|
+
if (isBanBody(input.bodyText)) {
|
|
130
|
+
record.blocked = true;
|
|
131
|
+
anomalyState = "blocked";
|
|
132
|
+
} else {
|
|
133
|
+
record.tempUnavailableUntil = now + this.authErrorTtlMs;
|
|
134
|
+
anomalyState = "unauthorized";
|
|
135
|
+
}
|
|
136
|
+
} else if (status === 401) {
|
|
137
|
+
record.tempUnavailableUntil = now + this.authErrorTtlMs;
|
|
138
|
+
anomalyState = "unauthorized";
|
|
139
|
+
} else if (status >= 500) {
|
|
140
|
+
record.tempUnavailableUntil = now + this.serverErrorTtlMs;
|
|
141
|
+
}
|
|
142
|
+
if (isRecordEmpty(record)) this.records.delete(key);
|
|
143
|
+
else this.records.set(key, record);
|
|
144
|
+
if (recovered) this.emit({ providerId, accountId, at: now, kind: "rateLimitRecovery" });
|
|
145
|
+
if (anomalyState && !wasUnhealthyBefore) {
|
|
146
|
+
this.emitAnomaly({ providerId, accountId, at: now, state: anomalyState });
|
|
147
|
+
}
|
|
148
|
+
}
|
|
149
|
+
/** Whether a record is unhealthy at `now`: blocked, or any timer still future. */
|
|
150
|
+
isUnhealthyAt(record, now) {
|
|
151
|
+
if (record.blocked) return true;
|
|
152
|
+
if (record.rateLimitEndAt !== void 0 && now < record.rateLimitEndAt) return true;
|
|
153
|
+
if (record.overloadUntil !== void 0 && now < record.overloadUntil) return true;
|
|
154
|
+
if (record.tempUnavailableUntil !== void 0 && now < record.tempUnavailableUntil) return true;
|
|
155
|
+
return false;
|
|
156
|
+
}
|
|
157
|
+
/**
|
|
158
|
+
* Clear ONLY the auth/transient unavailability mark (`tempUnavailableUntil`) —
|
|
159
|
+
* the narrow healing a background PROBE is allowed to do (subscription-account-
|
|
160
|
+
* probe #8, review M1). A probe 2xx from the lightweight `GET /v1/models` attests
|
|
161
|
+
* the TOKEN works (so a final-401 / plain-403 / 5xx transient may be healed), but
|
|
162
|
+
* it does NOT prove the traffic endpoint's rate-limit / overload has recovered —
|
|
163
|
+
* so it MUST NOT touch `rateLimitEndAt` / `overloadUntil` (those clear only on the
|
|
164
|
+
* upstream's authoritative reset, or a REAL-traffic 2xx). `blocked` (a ban) is
|
|
165
|
+
* never cleared here. Distinct from `recordUpstreamOutcome`'s 2xx path, which is
|
|
166
|
+
* UNCHANGED and still clears rate + transient together because it IS traffic.
|
|
167
|
+
* Returns whether a transient mark was actually cleared. Does NOT emit a recovery
|
|
168
|
+
* signal (a probe is internal maintenance; the account is schedulable-on-read).
|
|
169
|
+
*/
|
|
170
|
+
clearTransientMark(providerId, accountId) {
|
|
171
|
+
const key = this.key(providerId, accountId);
|
|
172
|
+
const record = this.records.get(key);
|
|
173
|
+
if (!record || record.tempUnavailableUntil === void 0) return false;
|
|
174
|
+
delete record.tempUnavailableUntil;
|
|
175
|
+
if (isRecordEmpty(record)) this.records.delete(key);
|
|
176
|
+
else this.records.set(key, record);
|
|
177
|
+
return true;
|
|
178
|
+
}
|
|
179
|
+
/**
|
|
180
|
+
* Proactively surface accounts that transitioned unhealthy → schedulable since
|
|
181
|
+
* the last sweep (design D6). Clears their expired timers, emits a recovery
|
|
182
|
+
* signal for each through `onRecovered`, and returns the list. A `blocked`
|
|
183
|
+
* account never recovers here. Correctness does NOT depend on this running —
|
|
184
|
+
* `isSchedulable` already restores an elapsed cooldown lazily; this adds the
|
|
185
|
+
* proactive signal (#5/#8) + lets the daemon nudge a token refresh.
|
|
186
|
+
*/
|
|
187
|
+
sweepRecoveries(now = this.now()) {
|
|
188
|
+
const recovered = [];
|
|
189
|
+
for (const [key, record] of this.records) {
|
|
190
|
+
if (record.blocked) continue;
|
|
191
|
+
const hadTimer = record.rateLimitEndAt !== void 0 || record.overloadUntil !== void 0 || record.tempUnavailableUntil !== void 0;
|
|
192
|
+
const active = this.clearExpired(record, now);
|
|
193
|
+
if (hadTimer && !active && isRecordEmpty(record)) {
|
|
194
|
+
const [pid, aid] = this.parseKey(key);
|
|
195
|
+
recovered.push({ providerId: pid, accountId: aid, at: now, kind: "rateLimitRecovery" });
|
|
196
|
+
this.records.delete(key);
|
|
197
|
+
}
|
|
198
|
+
}
|
|
199
|
+
for (const event of recovered) this.emit(event);
|
|
200
|
+
return recovered;
|
|
201
|
+
}
|
|
202
|
+
/** Admin-facing coarse status + cooldown-until for one account (secret-free). */
|
|
203
|
+
getStatus(providerId, accountId, now = this.now()) {
|
|
204
|
+
const record = this.records.get(this.key(providerId, accountId));
|
|
205
|
+
if (!record) return { state: "healthy" };
|
|
206
|
+
if (record.blocked) return { state: "blocked" };
|
|
207
|
+
if (record.rateLimitEndAt !== void 0 && now < record.rateLimitEndAt) {
|
|
208
|
+
return { state: "rate_limited", cooldownUntil: record.rateLimitEndAt };
|
|
209
|
+
}
|
|
210
|
+
if (record.overloadUntil !== void 0 && now < record.overloadUntil) {
|
|
211
|
+
return { state: "overloaded", cooldownUntil: record.overloadUntil };
|
|
212
|
+
}
|
|
213
|
+
if (record.tempUnavailableUntil !== void 0 && now < record.tempUnavailableUntil) {
|
|
214
|
+
return { state: "transient", cooldownUntil: record.tempUnavailableUntil };
|
|
215
|
+
}
|
|
216
|
+
return { state: "healthy" };
|
|
217
|
+
}
|
|
218
|
+
/**
|
|
219
|
+
* Register a recovery listener (the OQ3 seam #5 webhooks + #8 health-cron
|
|
220
|
+
* consume). Returns an unsubscribe function. Two edges fire it:
|
|
221
|
+
* - TRAFFIC-DRIVEN: a `recordUpstreamOutcome` 2xx that clears an unhealthy
|
|
222
|
+
* account (guaranteed — the sweep would miss it, as the 2xx deletes the
|
|
223
|
+
* record first); and
|
|
224
|
+
* - PURE-IDLE: `sweepRecoveries` when an idle account's cooldown elapsed with
|
|
225
|
+
* no traffic (best-effort, on the 60s tick).
|
|
226
|
+
* A lazy `isSchedulable` read that heals an account does NOT emit (a request is
|
|
227
|
+
* already using it) — so consumers never double-count a recovery.
|
|
228
|
+
*/
|
|
229
|
+
onRecovered(listener) {
|
|
230
|
+
this.listeners.add(listener);
|
|
231
|
+
return () => this.listeners.delete(listener);
|
|
232
|
+
}
|
|
233
|
+
emit(event) {
|
|
234
|
+
for (const listener of this.listeners) {
|
|
235
|
+
try {
|
|
236
|
+
listener(event);
|
|
237
|
+
} catch {
|
|
238
|
+
}
|
|
239
|
+
}
|
|
240
|
+
}
|
|
241
|
+
/**
|
|
242
|
+
* Register an anomaly listener (the ADDITIVE seam #5 webhooks consumes; the
|
|
243
|
+
* frozen `kind` discriminator on {@link AccountRecoveryEvent} anticipated it).
|
|
244
|
+
* Fires ONLY on the healthy→unhealthy EDGE inside `recordUpstreamOutcome`
|
|
245
|
+
* (de-duped: an already-unhealthy account escalating does NOT re-fire). Returns
|
|
246
|
+
* an unsubscribe function. Existing marking/recovery behavior is UNCHANGED.
|
|
247
|
+
*/
|
|
248
|
+
onAnomaly(listener) {
|
|
249
|
+
this.anomalyListeners.add(listener);
|
|
250
|
+
return () => this.anomalyListeners.delete(listener);
|
|
251
|
+
}
|
|
252
|
+
emitAnomaly(event) {
|
|
253
|
+
for (const listener of this.anomalyListeners) {
|
|
254
|
+
try {
|
|
255
|
+
listener(event);
|
|
256
|
+
} catch {
|
|
257
|
+
}
|
|
258
|
+
}
|
|
259
|
+
}
|
|
260
|
+
/**
|
|
261
|
+
* Delete every EXPIRED timer field on the record in place; return whether any
|
|
262
|
+
* timer is STILL active (in the future). `blocked` is not a timer and is left
|
|
263
|
+
* untouched here (only an explicit reset / restart clears a ban).
|
|
264
|
+
*/
|
|
265
|
+
clearExpired(record, now) {
|
|
266
|
+
let active = false;
|
|
267
|
+
if (record.rateLimitEndAt !== void 0) {
|
|
268
|
+
if (now >= record.rateLimitEndAt) delete record.rateLimitEndAt;
|
|
269
|
+
else active = true;
|
|
270
|
+
}
|
|
271
|
+
if (record.overloadUntil !== void 0) {
|
|
272
|
+
if (now >= record.overloadUntil) delete record.overloadUntil;
|
|
273
|
+
else active = true;
|
|
274
|
+
}
|
|
275
|
+
if (record.tempUnavailableUntil !== void 0) {
|
|
276
|
+
if (now >= record.tempUnavailableUntil) delete record.tempUnavailableUntil;
|
|
277
|
+
else active = true;
|
|
278
|
+
}
|
|
279
|
+
return active;
|
|
280
|
+
}
|
|
281
|
+
key(providerId, accountId) {
|
|
282
|
+
return `${providerId}${KEY_SEP}${accountId}`;
|
|
283
|
+
}
|
|
284
|
+
parseKey(key) {
|
|
285
|
+
const idx = key.indexOf(KEY_SEP);
|
|
286
|
+
return [key.slice(0, idx), key.slice(idx + 1)];
|
|
287
|
+
}
|
|
288
|
+
};
|
|
289
|
+
var _shared = null;
|
|
290
|
+
function getSharedAccountHealth() {
|
|
291
|
+
if (!_shared) _shared = new SubscriptionAccountHealth();
|
|
292
|
+
return _shared;
|
|
293
|
+
}
|
|
294
|
+
function setSharedAccountHealth(instance) {
|
|
295
|
+
_shared = instance;
|
|
296
|
+
}
|
|
297
|
+
function __resetSharedAccountHealthForTests() {
|
|
298
|
+
_shared = null;
|
|
299
|
+
}
|
|
300
|
+
|
|
301
|
+
export {
|
|
302
|
+
OVERLOAD_TTL_MS,
|
|
303
|
+
OVERLOAD_ENABLED_DEFAULT,
|
|
304
|
+
AUTH_ERROR_TTL_MS,
|
|
305
|
+
SERVER_ERROR_TTL_MS,
|
|
306
|
+
resolveResetSeconds,
|
|
307
|
+
SubscriptionAccountHealth,
|
|
308
|
+
getSharedAccountHealth,
|
|
309
|
+
setSharedAccountHealth,
|
|
310
|
+
__resetSharedAccountHealthForTests
|
|
311
|
+
};
|
|
@@ -0,0 +1,44 @@
|
|
|
1
|
+
// src/pipeline/resolveSubscriptionChain.ts
|
|
2
|
+
function resolveOne(transformerService, name) {
|
|
3
|
+
const t = transformerService.getTransformer(name);
|
|
4
|
+
if (!t) return null;
|
|
5
|
+
return typeof t === "function" ? new t() : t;
|
|
6
|
+
}
|
|
7
|
+
function resolveSubscriptionChain(profile, transformerService, fallbackEndpoint, overrideNames) {
|
|
8
|
+
const providerNames = overrideNames ?? profile.providerTransformerNames ?? [];
|
|
9
|
+
const modelNames = profile.modelTransformerNames ?? [];
|
|
10
|
+
if (providerNames.length === 0) {
|
|
11
|
+
return { providerTransformers: [fallbackEndpoint], modelTransformers: [] };
|
|
12
|
+
}
|
|
13
|
+
if (!transformerService) {
|
|
14
|
+
throw new Error(
|
|
15
|
+
"[resolveSubscriptionChain] profile declares providerTransformerNames but no TransformerService is wired"
|
|
16
|
+
);
|
|
17
|
+
}
|
|
18
|
+
const providerTransformers = [];
|
|
19
|
+
for (const name of providerNames) {
|
|
20
|
+
const instance = resolveOne(transformerService, name);
|
|
21
|
+
if (!instance) {
|
|
22
|
+
console.warn(`[resolveSubscriptionChain] Transformer not registered: ${name}`);
|
|
23
|
+
continue;
|
|
24
|
+
}
|
|
25
|
+
providerTransformers.push(instance);
|
|
26
|
+
}
|
|
27
|
+
const modelTransformers = [];
|
|
28
|
+
for (const name of modelNames) {
|
|
29
|
+
const instance = resolveOne(transformerService, name);
|
|
30
|
+
if (!instance) {
|
|
31
|
+
console.warn(`[resolveSubscriptionChain] Model transformer not registered: ${name}`);
|
|
32
|
+
continue;
|
|
33
|
+
}
|
|
34
|
+
modelTransformers.push(instance);
|
|
35
|
+
}
|
|
36
|
+
if (providerTransformers.length === 0) {
|
|
37
|
+
return { providerTransformers: [fallbackEndpoint], modelTransformers };
|
|
38
|
+
}
|
|
39
|
+
return { providerTransformers, modelTransformers };
|
|
40
|
+
}
|
|
41
|
+
|
|
42
|
+
export {
|
|
43
|
+
resolveSubscriptionChain
|
|
44
|
+
};
|