@wildix/xbees-conversations-utils 1.1.75 → 1.1.77

package/dist-es/rateLimits/generatedMethodNames.js

@@ -0,0 +1,317 @@
+export const STREAM_METHOD_NAMES = [
+    '_addChannelConfig',
+    '_buildWSPayload',
+    '_callClientListeners',
+    '_createImport',
+    '_createImportURL',
+    '_deleteUserMessageReference',
+    '_enrichAxiosOptions',
+    '_getConnectionID',
+    '_getImport',
+    '_getToken',
+    '_handleClientEvent',
+    '_handleUserEvent',
+    '_hasConnectionID',
+    '_isUsingServerAuth',
+    '_listImports',
+    '_logApiError',
+    '_logApiRequest',
+    '_logApiResponse',
+    '_muteStatus',
+    '_normalizeDate',
+    '_normalizeExpiration',
+    '_queryFlagReports',
+    '_queryFlags',
+    '_reviewFlagReport',
+    '_sayHi',
+    '_setToken',
+    '_setupConnection',
+    '_setUser',
+    '_startCleaning',
+    '_unblockMessage',
+    '_updateMemberWatcherReferences',
+    '_updateUserMessageReferences',
+    '_updateUserReferences',
+    '_validateAndGetMessageId',
+    'addDevice',
+    'banUser',
+    'channel',
+    'closeConnection',
+    'commitMessage',
+    'connect',
+    'connectAnonymousUser',
+    'connectUser',
+    'createAbortControllerForNextRequest',
+    'createBlockList',
+    'createCampaign',
+    'createChannelType',
+    'createCommand',
+    'createPermission',
+    'createRole',
+    'createSegment',
+    'createToken',
+    'deactivateUser',
+    'deactivateUsers',
+    'delete',
+    'deleteBlockList',
+    'deleteCampaign',
+    'deleteChannels',
+    'deleteChannelType',
+    'deleteCommand',
+    'deleteMessage',
+    'deletePermission',
+    'deletePushProvider',
+    'deleteRole',
+    'deleteSegment',
+    'deleteUser',
+    'deleteUsers',
+    'devToken',
+    'disconnect',
+    'disconnectUser',
+    'dispatchEvent',
+    'doAxiosRequest',
+    'enrichURL',
+    'errorFromResponse',
+    'exportChannel',
+    'exportChannels',
+    'exportUser',
+    'exportUsers',
+    'flagMessage',
+    'flagUser',
+    'get',
+    'getAppSettings',
+    'getAuthType',
+    'getBlockList',
+    'getCallToken',
+    'getChannelById',
+    'getChannelByMembers',
+    'getChannelType',
+    'getCommand',
+    'getDevices',
+    'getExportChannelStatus',
+    'getMessage',
+    'getPermission',
+    'getRateLimits',
+    'getTask',
+    'getUnreadCount',
+    'getUserAgent',
+    'handleEvent',
+    'handleResponse',
+    'hydrateActiveChannels',
+    'listBlockLists',
+    'listChannelTypes',
+    'listCommands',
+    'listPermissions',
+    'listPushProviders',
+    'listRoles',
+    'markAllRead',
+    'markChannelsRead',
+    'muteUser',
+    'off',
+    'on',
+    'openConnection',
+    'partialUpdateMessage',
+    'partialUpdateUser',
+    'partialUpdateUsers',
+    'patch',
+    'pinMessage',
+    'post',
+    'put',
+    'queryBannedUsers',
+    'queryCampaigns',
+    'queryChannels',
+    'queryMessageFlags',
+    'queryRecipients',
+    'querySegments',
+    'queryUsers',
+    'reactivateUser',
+    'reactivateUsers',
+    'recoverState',
+    'removeDevice',
+    'removeShadowBan',
+    'restoreUsers',
+    'resumeCampaign',
+    'revokeTokens',
+    'revokeUsersToken',
+    'revokeUserToken',
+    'scheduleCampaign',
+    'search',
+    'sendFile',
+    'sendUserCustomEvent',
+    'setAnonymousUser',
+    'setBaseURL',
+    'setGuestUser',
+    'setLocalDevice',
+    'setUser',
+    'setUserAgent',
+    'shadowBan',
+    'stopCampaign',
+    'sync',
+    'testCampaign',
+    'testPushSettings',
+    'testSQSSettings',
+    'translateMessage',
+    'unbanUser',
+    'unblockMessage',
+    'unflagMessage',
+    'unflagUser',
+    'unmuteUser',
+    'unpinMessage',
+    'updateAppSettings',
+    'updateBlockList',
+    'updateCampaign',
+    'updateChannelType',
+    'updateCommand',
+    'updateMessage',
+    'updatePermission',
+    'updateSegment',
+    'updateUser',
+    'updateUsers',
+    'upsertPushProvider',
+    'upsertUser',
+    'upsertUsers',
+    'userMuteStatus',
+    'verifyWebhook',
+];
+export const STREAM_SYNC_METHOD_NAMES = [
+    '_addChannelConfig',
+    '_buildWSPayload',
+    '_callClientListeners',
+    '_deleteUserMessageReference',
+    '_enrichAxiosOptions',
+    '_getConnectionID',
+    '_getToken',
+    '_handleClientEvent',
+    '_handleUserEvent',
+    '_hasConnectionID',
+    '_isUsingServerAuth',
+    '_logApiError',
+    '_logApiRequest',
+    '_logApiResponse',
+    '_muteStatus',
+    '_normalizeDate',
+    '_normalizeExpiration',
+    '_sayHi',
+    '_setUser',
+    '_startCleaning',
+    '_updateMemberWatcherReferences',
+    '_updateUserMessageReferences',
+    '_updateUserReferences',
+    '_validateAndGetMessageId',
+    'createAbortControllerForNextRequest',
+    'createToken',
+    'devToken',
+    'dispatchEvent',
+    'errorFromResponse',
+    'getAuthType',
+    'getUserAgent',
+    'handleEvent',
+    'handleResponse',
+    'off',
+    'on',
+    'setBaseURL',
+    'setLocalDevice',
+    'setUserAgent',
+    'userMuteStatus',
+    'verifyWebhook',
+];
+export const STREAM_CHANNEL_RESULT_METHOD_NAMES = [
+    'channel',
+    'getChannelById',
+    'getChannelByMembers',
+    'hydrateActiveChannels',
+    'queryChannels',
+];
+export const STREAM_SYNC_CHANNEL_RESULT_METHOD_NAMES = [
+    'channel',
+    'getChannelById',
+    'getChannelByMembers',
+    'hydrateActiveChannels',
+];
+export const CHANNEL_METHOD_NAMES = [
+    '_callChannelListeners',
+    '_channelURL',
+    '_checkInitialized',
+    '_countMessageAsUnread',
+    '_disconnect',
+    '_extendEventWithOwnReactions',
+    '_handleChannelEvent',
+    '_initializeState',
+    '_update',
+    'acceptInvite',
+    'addMembers',
+    'addModerators',
+    'assignRoles',
+    'banUser',
+    'clean',
+    'countUnread',
+    'countUnreadMentions',
+    'create',
+    'createCall',
+    'delete',
+    'deleteFile',
+    'deleteImage',
+    'deleteReaction',
+    'demoteModerators',
+    'disableSlowMode',
+    'enableSlowMode',
+    'getClient',
+    'getConfig',
+    'getMessagesById',
+    'getPinnedMessages',
+    'getReactions',
+    'getReplies',
+    'hide',
+    'inviteMembers',
+    'keystroke',
+    'lastMessage',
+    'lastRead',
+    'markRead',
+    'markUnread',
+    'mute',
+    'muteStatus',
+    'off',
+    'on',
+    'query',
+    'queryMembers',
+    'rejectInvite',
+    'removeMembers',
+    'removeShadowBan',
+    'search',
+    'sendAction',
+    'sendEvent',
+    'sendFile',
+    'sendImage',
+    'sendMessage',
+    'sendReaction',
+    'shadowBan',
+    'show',
+    'stopTyping',
+    'stopWatching',
+    'truncate',
+    'unbanUser',
+    'unmute',
+    'update',
+    'updatePartial',
+    'watch',
+];
+export const CHANNEL_SYNC_METHOD_NAMES = [
+    '_callChannelListeners',
+    '_channelURL',
+    '_checkInitialized',
+    '_countMessageAsUnread',
+    '_disconnect',
+    '_extendEventWithOwnReactions',
+    '_handleChannelEvent',
+    '_initializeState',
+    'clean',
+    'countUnread',
+    'countUnreadMentions',
+    'getConfig',
+    'lastMessage',
+    'lastRead',
+    'muteStatus',
+    'off',
+    'on',
+];
+export const CHANNEL_SYNC_PROXY_RESULT_METHOD_NAMES = ['getClient'];

package/dist-es/rateLimits/helpers/RedisCooldown.js

@@ -53,6 +53,22 @@ export class RedisCooldown {
             });
         }
     }
+    static async setRemainingMsExact(cooldownKey, cooldownMs, context, options) {
+        if (cooldownMs <= 0) {
+            return;
+        }
+        const { redis, logger } = context;
+        try {
+            await redis.set(RedisCooldown.resolveRedisKey(cooldownKey, options), '1', 'PX', cooldownMs);
+        }
+        catch (error) {
+            logger.warn('Failed to write exact stream rate-limit cooldown to Redis', {
+                cooldownKey,
+                cooldownMs,
+                error,
+            });
+        }
+    }
     static async clear(cooldownKey, context, options) {
         const { redis, logger } = context;
         try {

package/dist-es/rateLimits/helpers/createRateLimitExceededException.js

@@ -1,11 +1,11 @@
 import { RateLimitExceededException } from '@wildix/xbees-conversations-client';
 import { DEFAULT_RETRY_AFTER_MS } from '../constants';
+const RETRY_AFTER_HEADER_PADDING_MS = 3000;
 function toPositiveMilliseconds(milliseconds) {
     return Math.max(0, Math.ceil(milliseconds));
 }
-function toHttpDateFromDelayMs(delayMs) {
-    const positiveDelayMs = toPositiveMilliseconds(delayMs);
-    return new Date(Date.now() + positiveDelayMs).toUTCString();
+function toEpochSecondsFromNow(delayMs) {
+    return Math.ceil((Date.now() + toPositiveMilliseconds(delayMs)) / 1000);
 }
 export function createRateLimitExceededException(options = {}) {
     const { rateLimit, message, operation, retryAfterMs = DEFAULT_RETRY_AFTER_MS } = options;
@@ -13,15 +13,15 @@ export function createRateLimitExceededException(options = {}) {
     const retryAfterFromResetMs = rateLimit?.reset ? rateLimit.reset - Date.now() : undefined;
     const effectiveRetryAfterMs = retryAfterFromResetMs ?? retryAfterMs;
     const retryAfterMilliseconds = toPositiveMilliseconds(effectiveRetryAfterMs);
-    const retryAfterHttpDate = toHttpDateFromDelayMs(effectiveRetryAfterMs);
     const rateLimitResetMilliseconds = rateLimit?.reset
         ? toPositiveMilliseconds(retryAfterFromResetMs ?? 0)
         : retryAfterMilliseconds;
+    const retryAfterEpochSeconds = toEpochSecondsFromNow(effectiveRetryAfterMs + RETRY_AFTER_HEADER_PADDING_MS);
     return new RateLimitExceededException({
         $metadata: {},
         message: message || defaultMessage,
         rateLimitRemaining: rateLimit?.remaining || 0,
-        retryAfter: retryAfterHttpDate,
+        retryAfter: retryAfterEpochSeconds,
         rateLimit: rateLimit?.limit?.toString() || 'unknown',
         rateLimitReset: rateLimitResetMilliseconds,
     });

package/dist-es/rateLimits/helpers/executeWithRateLimitHandling.js

@@ -1,35 +1,12 @@
 import { setTimeout as sleep } from 'node:timers/promises';
-import { DEFAULT_JITTER_MS, DEFAULT_MAX_DELAY_MS, DEFAULT_RETRY_AFTER_MS, RETRY_AFTER_BUFFER_MS, STREAM_BUDGET_COOLDOWN_KEY, STREAM_BUDGET_COOLDOWN_KEY_PREFIX, STREAM_RATE_LIMIT_COOLDOWN_KEY_PREFIX, STREAM_RATE_LIMIT_SOFT_COOLDOWN_KEY, } from '../constants';
+import { BUDGET_COOLDOWN_SLEEP_CHUNK_MS, BUDGET_DELAY_MAX_MS, DEFAULT_JITTER_MS, DEFAULT_MAX_DELAY_MS, DEFAULT_RETRY_AFTER_MS, RETRY_AFTER_BUFFER_MS, STREAM_BUDGET_COOLDOWN_KEY, STREAM_BUDGET_COOLDOWN_KEY_PREFIX, STREAM_RATE_LIMIT_COOLDOWN_KEY_PREFIX, STREAM_RATE_LIMIT_SOFT_COOLDOWN_KEY, } from '../constants';
 import { createRateLimitExceededException } from './createRateLimitExceededException';
 import { processStreamRateLimitException } from './processStreamRateLimitException';
 import { RedisCooldown } from './RedisCooldown';
-function normalizeRetryAfterToDelayMs(retryAfter) {
-    if (typeof retryAfter === 'string') {
-        const asDateMs = Date.parse(retryAfter);
-        if (!Number.isNaN(asDateMs)) {
-            return Math.max(0, Math.round(asDateMs - Date.now()));
-        }
-        const asSeconds = Number(retryAfter);
-        if (Number.isFinite(asSeconds) && asSeconds > 0) {
-            return Math.max(0, Math.round(asSeconds * 1000));
-        }
-        return 0;
-    }
-    if (typeof retryAfter !== 'number' || !Number.isFinite(retryAfter) || retryAfter <= 0) {
-        return 0;
-    }
-    if (retryAfter > 1000000000) {
-        return Math.max(0, Math.round(retryAfter * 1000 - Date.now()));
-    }
-    if (retryAfter >= 1000) {
-        return Math.max(0, Math.round(retryAfter));
-    }
-    return Math.max(0, Math.round(retryAfter * 1000));
-}
 function calculateRateLimitDelayMs(error, maxDelayMs) {
-    const retryAfterMsFromError = normalizeRetryAfterToDelayMs(error.retryAfter);
-    const hasServerRetryAfter = retryAfterMsFromError > 0;
-    const baseDelayMs = hasServerRetryAfter ? retryAfterMsFromError + RETRY_AFTER_BUFFER_MS : DEFAULT_RETRY_AFTER_MS;
+    const retryAfterMs = typeof error.rateLimitReset === 'number' && error.rateLimitReset > 0 ? error.rateLimitReset : 0;
+    const hasServerRetryAfter = retryAfterMs > 0;
+    const baseDelayMs = hasServerRetryAfter ? retryAfterMs + RETRY_AFTER_BUFFER_MS : DEFAULT_RETRY_AFTER_MS;
     if (baseDelayMs > 0) {
         const jitterMs = Math.floor(Math.random() * DEFAULT_JITTER_MS);
         const computedDelayMs = baseDelayMs + jitterMs;
@@ -45,6 +22,14 @@ export async function executeWithRateLimitHandling(execute, operation, context,
     const { enableCooldown, maxAttempts, maxDelayMs, maxRetryableDelayMs } = options;
     let attempt = 1;
     if (enableCooldown) {
+        const cooldownMs = await RedisCooldown.getRemainingMs(operation, { redis, logger });
+        if (cooldownMs > 0) {
+            logger.warn('Skipping stream request due to active rate-limit cooldown', {
+                operation,
+                cooldown: cooldownMs,
+            });
+            throw createRateLimitExceededException({ operation, retryAfterMs: cooldownMs });
+        }
         const softCooldownMs = await RedisCooldown.getRemainingMs(STREAM_RATE_LIMIT_SOFT_COOLDOWN_KEY, { redis, logger }, { keyPrefix: STREAM_RATE_LIMIT_COOLDOWN_KEY_PREFIX });
         if (softCooldownMs > 0) {
             logger.debug('Delaying stream request due to active soft rate-limit cooldown', {
@@ -59,15 +44,16 @@ export async function executeWithRateLimitHandling(execute, operation, context,
                 operation,
                 cooldown: budgetCooldownMs,
             });
-            await sleep(budgetCooldownMs);
-        }
-        const cooldownMs = await RedisCooldown.getRemainingMs(operation, { redis, logger });
-        if (cooldownMs > 0) {
-            logger.warn('Skipping stream request due to active rate-limit cooldown', {
-                operation,
-                cooldown: cooldownMs,
-            });
-            throw createRateLimitExceededException({ operation, retryAfterMs: cooldownMs });
+            let totalBudgetWaitMs = 0;
+            for (;;) {
+                const waitMs = await RedisCooldown.getRemainingMs(STREAM_BUDGET_COOLDOWN_KEY, { redis, logger }, { keyPrefix: STREAM_BUDGET_COOLDOWN_KEY_PREFIX });
+                if (waitMs <= 0 || totalBudgetWaitMs >= BUDGET_DELAY_MAX_MS) {
+                    break;
+                }
+                const chunkMs = Math.min(waitMs, BUDGET_COOLDOWN_SLEEP_CHUNK_MS);
+                await sleep(chunkMs);
+                totalBudgetWaitMs += chunkMs;
+            }
         }
     }
     for (;;) {

package/dist-es/rateLimits/helpers/headerUtils.js

@@ -0,0 +1,34 @@
+export function getHeaderCaseInsensitive(headers, key) {
+    if (!headers) {
+        return;
+    }
+    const direct = headers[key];
+    if (direct !== undefined) {
+        return direct;
+    }
+    const normalizedKey = key.toLowerCase();
+    for (const [headerKey, headerValue] of Object.entries(headers)) {
+        if (headerKey.toLowerCase() === normalizedKey) {
+            return headerValue;
+        }
+    }
+}
+export function parseHeaderNumber(headers, key) {
+    const raw = getHeaderCaseInsensitive(headers, key);
+    if (typeof raw === 'number' && Number.isFinite(raw)) {
+        return raw;
+    }
+    if (typeof raw === 'string') {
+        const parsed = Number(raw);
+        if (Number.isFinite(parsed)) {
+            return parsed;
+        }
+    }
+}
+export function normalizeResetToEpochMs(reset) {
+    if (reset < 10000)
+        return Date.now() + Math.round(reset * 1000);
+    if (reset < 10000000000)
+        return Math.round(reset * 1000);
+    return Math.round(reset);
+}

package/dist-es/rateLimits/helpers/isStreamChannelLike.js

@@ -1,11 +1,19 @@
+import { Channel } from 'stream-chat';
 function hasFunctionProperty(value, key) {
     return key in value && typeof value[key] === 'function';
 }
+function hasStringProperty(value, key) {
+    return key in value && typeof value[key] === 'string';
+}
 export function isStreamChannelLike(value) {
     if (typeof value !== 'object' || value === null) {
         return false;
     }
-    return (hasFunctionProperty(value, 'sendMessage') &&
+    if (value instanceof Channel) {
+        return true;
+    }
+    return (hasStringProperty(value, 'cid') &&
+        hasFunctionProperty(value, 'sendMessage') &&
         hasFunctionProperty(value, 'queryMembers') &&
         hasFunctionProperty(value, 'updatePartial'));
 }

package/dist-es/rateLimits/helpers/processStreamBudgetHeaders.js

@@ -1,36 +1,11 @@
-import { BUDGET_DELAY_MAX_MS, BUDGET_DELAY_MIN_MS, BUDGET_THRESHOLD, STREAM_BUDGET_COOLDOWN_KEY, STREAM_BUDGET_COOLDOWN_KEY_PREFIX, } from '../constants';
+import { BUDGET_CLEAR_USAGE_THRESHOLD, BUDGET_DELAY_MAX_MS, BUDGET_DELAY_MIN_MS, BUDGET_MINIMAL_DELAY_MAX_MS, BUDGET_MINIMAL_DELAY_MIN_MS, BUDGET_RELAXED_DELAY_MAX_MS, BUDGET_RELAXED_DELAY_MIN_MS, BUDGET_RELAXED_USAGE_THRESHOLD, BUDGET_THRESHOLD, STREAM_BUDGET_COOLDOWN_KEY, STREAM_BUDGET_COOLDOWN_KEY_PREFIX, } from '../constants';
+import { parseHeaderNumber } from './headerUtils';
 import { RedisCooldown } from './RedisCooldown';
-function parseHeaderNumber(value) {
-    if (typeof value === 'number' && Number.isFinite(value)) {
-        return value;
-    }
-    if (typeof value === 'string') {
-        const parsed = Number(value);
-        if (Number.isFinite(parsed)) {
-            return parsed;
-        }
-    }
-}
-function getHeaderCaseInsensitive(headers, key) {
-    if (!headers) {
-        return;
-    }
-    const direct = headers[key];
-    if (direct !== undefined) {
-        return direct;
-    }
-    const normalizedKey = key.toLowerCase();
-    for (const [headerKey, headerValue] of Object.entries(headers)) {
-        if (headerKey.toLowerCase() === normalizedKey) {
-            return headerValue;
-        }
-    }
-}
 function extractBudgetHeaders(headers) {
     return {
-        limitMs: parseHeaderNumber(getHeaderCaseInsensitive(headers, 'x-budget-limit-ms')),
-        remainingMs: parseHeaderNumber(getHeaderCaseInsensitive(headers, 'x-budget-remaining-ms')),
-        usedMs: parseHeaderNumber(getHeaderCaseInsensitive(headers, 'x-budget-used-ms')),
+        limitMs: parseHeaderNumber(headers ?? {}, 'x-budget-limit-ms'),
+        remainingMs: parseHeaderNumber(headers ?? {}, 'x-budget-remaining-ms'),
+        usedMs: parseHeaderNumber(headers ?? {}, 'x-budget-used-ms'),
     };
 }
 function isBudgetThresholdReached(limitMs, remainingMs, usedMs) {
@@ -46,8 +21,20 @@ function isBudgetThresholdReached(limitMs, remainingMs, usedMs) {
     }
     return false;
 }
-function getBudgetDelayMs() {
-    return Math.round(BUDGET_DELAY_MIN_MS + Math.random() * (BUDGET_DELAY_MAX_MS - BUDGET_DELAY_MIN_MS));
+function getRandomDelayInRange(minMs, maxMs) {
+    return Math.round(minMs + Math.random() * (maxMs - minMs));
+}
+function getBudgetUsage(limitMs, remainingMs, usedMs) {
+    if (limitMs <= 0) {
+        return 0;
+    }
+    if (typeof usedMs === 'number' && Number.isFinite(usedMs) && usedMs >= 0) {
+        return usedMs / limitMs;
+    }
+    if (typeof remainingMs === 'number' && Number.isFinite(remainingMs)) {
+        return Math.max(0, 1 - remainingMs / limitMs);
+    }
+    return 0;
 }
 export async function processStreamBudgetHeaders(headers, context) {
     const { logger } = context;
@@ -58,26 +45,45 @@ export async function processStreamBudgetHeaders(headers, context) {
     if (typeof limitMs !== 'number' || !Number.isFinite(limitMs)) {
         return;
     }
+    const usage = getBudgetUsage(limitMs, remainingMs, usedMs);
     const thresholdReached = isBudgetThresholdReached(limitMs, remainingMs, usedMs);
     if (thresholdReached) {
-        const delayMs = getBudgetDelayMs();
-        await RedisCooldown.setRemainingMs(STREAM_BUDGET_COOLDOWN_KEY, delayMs, context, {
+        const delayMs = getRandomDelayInRange(BUDGET_DELAY_MIN_MS, BUDGET_DELAY_MAX_MS);
+        await RedisCooldown.setRemainingMsExact(STREAM_BUDGET_COOLDOWN_KEY, delayMs, context, {
             keyPrefix: STREAM_BUDGET_COOLDOWN_KEY_PREFIX,
         });
         logger.warn('Stream budget threshold reached, enabling cooldown', {
             budgetLimitMs: limitMs,
             budgetRemainingMs: remainingMs,
             budgetUsedMs: usedMs,
+            budgetUsage: usage,
             cooldownMs: delayMs,
         });
         return;
     }
-    await RedisCooldown.clear(STREAM_BUDGET_COOLDOWN_KEY, context, {
+    if (usage < BUDGET_CLEAR_USAGE_THRESHOLD) {
+        await RedisCooldown.clear(STREAM_BUDGET_COOLDOWN_KEY, context, {
+            keyPrefix: STREAM_BUDGET_COOLDOWN_KEY_PREFIX,
+        });
+        logger.info('Stream budget is low, clearing cooldown', {
+            budgetLimitMs: limitMs,
+            budgetRemainingMs: remainingMs,
+            budgetUsedMs: usedMs,
+            budgetUsage: usage,
+        });
+        return;
+    }
+    const delayMs = usage >= BUDGET_RELAXED_USAGE_THRESHOLD
+        ? getRandomDelayInRange(BUDGET_RELAXED_DELAY_MIN_MS, BUDGET_RELAXED_DELAY_MAX_MS)
+        : getRandomDelayInRange(BUDGET_MINIMAL_DELAY_MIN_MS, BUDGET_MINIMAL_DELAY_MAX_MS);
+    await RedisCooldown.setRemainingMsExact(STREAM_BUDGET_COOLDOWN_KEY, delayMs, context, {
         keyPrefix: STREAM_BUDGET_COOLDOWN_KEY_PREFIX,
     });
-    logger.warn('Stream budget is below threshold, disabling cooldown', {
+    logger.info('Stream budget is below threshold, reducing cooldown', {
         budgetLimitMs: limitMs,
         budgetRemainingMs: remainingMs,
         budgetUsedMs: usedMs,
+        budgetUsage: usage,
+        cooldownMs: delayMs,
     });
 }

package/dist-es/rateLimits/helpers/processStreamRateLimitException.js

@@ -1,18 +1,7 @@
 import { createRateLimitExceededException } from './createRateLimitExceededException';
-function parseHeaderNumber(headers, key) {
-    const raw = headers[key];
-    if (typeof raw === 'number' && Number.isFinite(raw)) {
-        return raw;
-    }
-    if (typeof raw === 'string') {
-        const parsed = Number(raw);
-        if (Number.isFinite(parsed)) {
-            return parsed;
-        }
-    }
-}
+import { getHeaderCaseInsensitive, normalizeResetToEpochMs, parseHeaderNumber } from './headerUtils';
 function parseRetryAfterMs(headers) {
-    const retryAfter = headers['retry-after'];
+    const retryAfter = getHeaderCaseInsensitive(headers, 'retry-after');
     if (typeof retryAfter === 'number' && Number.isFinite(retryAfter) && retryAfter >= 0) {
         return Math.round(retryAfter * 1000);
     }
@@ -31,14 +20,14 @@ function parseRetryAfterMs(headers) {
 function getRateLimitFromHeaders(headers) {
     const limit = parseHeaderNumber(headers, 'x-ratelimit-limit');
     const remaining = parseHeaderNumber(headers, 'x-ratelimit-remaining');
-    const resetSeconds = parseHeaderNumber(headers, 'x-ratelimit-reset');
-    if (limit === undefined || remaining === undefined || resetSeconds === undefined) {
+    const resetRaw = parseHeaderNumber(headers, 'x-ratelimit-reset');
+    if (limit === undefined || remaining === undefined || resetRaw === undefined) {
         return;
     }
     return {
         limit,
         remaining,
-        reset: Math.round(resetSeconds * 1000),
+        reset: normalizeResetToEpochMs(resetRaw),
     };
 }
 function isStreamHttpRateLimitError(error) {