agent-afk 5.117.0 → 5.118.0

package/dist/agent/providers/anthropic-direct/tracing-fetch.d.ts

@@ -3,4 +3,8 @@ export interface ThrottleInfo {
     status: number;
     retryAfterMs?: number;
 }
-export declare function makeTracingFetch(writer: TraceSink | undefined, baseFetch?: typeof fetch, onThrottle?: (info: ThrottleInfo) => void, onQuota?: (headers: Headers) => void): typeof fetch;
+export interface RateLimitGate {
+    acquirePermit(estimatedInputTokens: number, signal?: AbortSignal): Promise<void>;
+    freeze(retryAfterMs: number): void;
+}
+export declare function makeTracingFetch(writer: TraceSink | undefined, baseFetch?: typeof fetch, onThrottle?: (info: ThrottleInfo) => void, onQuota?: (headers: Headers) => void, onRateLimit?: (headers: Headers) => void, gate?: RateLimitGate): typeof fetch;

package/dist/agent/providers/openai-compatible/query/client.d.ts

@@ -3,6 +3,8 @@ export type OpenAIClientFactory = (opts: {
     apiKey: string;
     baseURL?: string;
     defaultHeaders?: Record<string, string>;
+    fetch?: typeof globalThis.fetch;
 }) => OpenAI;
 export declare function __setOpenAIClientFactory(factory: OpenAIClientFactory | null): void;
 export declare function resolveClientFactory(): OpenAIClientFactory;
+export declare function buildOpenAIAdmissionFetch(baseURL: string | undefined): typeof fetch | undefined;

package/dist/agent/providers/openai-compatible/tracing-fetch.d.ts

@@ -0,0 +1,6 @@
+import type { ThrottleInfo } from '../anthropic-direct/tracing-fetch.js';
+export interface OpenAIRateLimitGate {
+    acquirePermit(estimatedInputTokens: number, signal?: AbortSignal): Promise<void>;
+    freeze(retryAfterMs: number): void;
+}
+export declare function makeOpenAITracingFetch(baseFetch?: typeof fetch, onThrottle?: (info: ThrottleInfo) => void, onRateLimit?: (headers: Headers) => void, gate?: OpenAIRateLimitGate): typeof fetch;

package/dist/agent/providers/shared/rate-limit-bucket.d.ts

@@ -0,0 +1,28 @@
+export interface RateLimitSnapshot {
+    requestsRemaining?: number;
+    requestsLimit?: number;
+    requestsResetAt?: number;
+    inputTokensRemaining?: number;
+    inputTokensLimit?: number;
+    inputTokensResetAt?: number;
+    outputTokensRemaining?: number;
+    outputTokensResetAt?: number;
+}
+export declare function estimateInputTokens(body: unknown): number;
+export declare class RateLimitBucket {
+    private requestsRemaining;
+    private requestsResetAt;
+    private requestsLimit;
+    private inputTokensRemaining;
+    private inputTokensResetAt;
+    private inputTokensLimit;
+    private outputTokensRemainingVal;
+    private frozenUntil;
+    resetForTests(): void;
+    get outputTokensRemaining(): number;
+    update(snap: RateLimitSnapshot): void;
+    freeze(retryAfterMs: number): void;
+    private maybeResetWindows;
+    acquirePermit(estimatedInputTokens: number, signal?: AbortSignal): Promise<void>;
+}
+export declare const globalRateLimitBucket: RateLimitBucket;

package/dist/agent/providers/shared/rate-limit-headers.d.ts

@@ -0,0 +1,3 @@
+import type { RateLimitSnapshot } from './rate-limit-bucket.js';
+export declare function parseAnthropicRateLimitHeaders(headers: Headers): RateLimitSnapshot | undefined;
+export declare function parseOpenAIRateLimitHeaders(headers: Headers): RateLimitSnapshot | undefined;