pjandro/OmniRoute
1
0
Fork 0
mirror of https://github.com/diegosouzapw/OmniRoute.git synced 2026-08-26 09:02:11 +03:00
Code Issues Packages Projects Releases Wiki Activity

test(kimi): stop drawing the refresh window inside the assertion (#11380)

Browse Source 
Validado em lote combinado (batch-0824g, junto de #11388/#11397/#11415/#11418) contra o tip de release/v3.8.50: typecheck:core limpo, file-size/changelog/complexity/cognitive-complexity OK, 62/62 testes focados passando.

Diagnóstico correto e bem documentado: a falha do nightly Node 26 era um teste que sorteia um número e depende do resultado, não uma quebra de compatibilidade. Comportamento de produção inalterado (a janela de jitter continua aleatória; só o teste ganhou controle sobre ela). Obrigado pela investigação detalhada!
This commit is contained in:
Nguyen Thanh Dat
2026-08-25 03:23:27 +07:00
committed by GitHub
parent 8bbe92c692
commit d5d730c845
3 changed files with 86 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
src/lib/tokenHealthCheckKimi.ts
View File

@@ -2,6 +2,21 @@ import { isKimiTokenExpiringSoon } from "@omniroute/open-sse/utils/kimiJwt.ts";
import { exchangeKimiRefreshToken } from "@/lib/kimi/tokenRefresh";
import { updateProviderConnection } from "@/lib/db/providers";
/**
* Refresh window, spread over [60, 240) seconds before expiry so a fleet of
* connections does not stampede the token endpoint at the same instant.
*
* Kept as a named export rather than inline: it is the only nondeterminism in this
* path, and a caller that needs a decision it can predict — a test — has to be able
* to replace it. `tests/unit/token-health-check-kimi.test.ts` used a token expiring
* in 90 s and asserted a refresh, which is a coin the draw loses 1 in 6 times
* (a refresh needs `jitter >= 90`, i.e. 150 of the 180 possible values). It failed
* that way on the Node 26 nightly and was triaged as a Node-compat break.
*/
export function defaultKimiRefreshJitterSec(): number {
return 60 + Math.floor(Math.random() * 180);
}
export async function checkKimiWebConnectionIfNeeded(params: {
conn: any;
now: string;
@@ -12,6 +27,12 @@ export async function checkKimiWebConnectionIfNeeded(params: {
logPrefix: string;
exchangeFn?: typeof exchangeKimiRefreshToken;
persistFn?: typeof updateProviderConnection;
/**
* Seconds before expiry at which a refresh is triggered. Defaults to the random
* spread below; injectable so a caller — a test above all — can decide the window
* instead of drawing it.
*/
jitterSecFn?: () => number;
}): Promise<boolean> {
const { conn, log, logWarn, getConnectionLogLabel, logPrefix } = params;
const provider = String(conn?.provider || "").toLowerCase();
@@ -21,20 +42,23 @@ export async function checkKimiWebConnectionIfNeeded(params: {
if (!refreshToken) return true; // Handled, but cannot refresh without refresh_token
const token = conn.apiKey || conn.accessToken;
// Calculate jitter: random value between 60 and 240 seconds (1 to 4 min before expiry)
const jitterSec = 60 + Math.floor(Math.random() * 180);
const jitterSec = (params.jitterSecFn ?? defaultKimiRefreshJitterSec)();
const expiringSoon = isKimiTokenExpiringSoon(token, jitterSec);
if (!expiringSoon) return true;
log(`${logPrefix} Kimi Web connection ${getConnectionLogLabel(conn)} token expiring soon; refreshing in background...`);
log(
`${logPrefix} Kimi Web connection ${getConnectionLogLabel(conn)} token expiring soon; refreshing in background...`
);
const exchange = params.exchangeFn || exchangeKimiRefreshToken;
const persist = params.persistFn || updateProviderConnection;
const res = await exchange(refreshToken);
if (res.success && res.accessToken) {
log(`${logPrefix} Kimi Web connection ${getConnectionLogLabel(conn)} token refreshed successfully.`);
log(
`${logPrefix} Kimi Web connection ${getConnectionLogLabel(conn)} token refreshed successfully.`
);
await persist(conn.id, {
apiKey: res.accessToken,
accessToken: res.accessToken,