@cortexkit/pi-anthropic-auth 2.4.11 → 2.4.12

package/dist/stream.d.ts

@@ -1,5 +1,5 @@
-import { type ApiKeyAccount, CacheKeepManager, type MidConversationEffortTransition, type QuotaManagerOptions } from '@cortexkit/anthropic-auth-core';
-import { type Api, type AssistantMessageEventStream, type Context, type Model, type SimpleStreamOptions } from '@earendil-works/pi-ai';
+import { type ApiKeyAccount, CacheKeepManager, type MidConversationEffortTransition, type NativeAccountView, type QuotaManagerOptions } from '@cortexkit/anthropic-auth-core';
+import { type Api, type AssistantMessageEventStream, type Context, type Model, type SimpleStreamOptions, type Tool } from '@earendil-works/pi-ai';
 /** Set or clear the test replacement for the wait between quota usage polls. */
 export declare function __setPiQuotaCallPauseForTest(pause: QuotaManagerOptions['pauseBetweenQuotaCalls']): void;
 export declare function clearPiStickyRoutingSession(storagePath: string, sessionId: string): Promise<void>;
@@ -20,6 +20,22 @@ type AnthropicEvent = {
 export declare function buildExplicitBaseMessagesUrl(baseURL: string): URL;
 export declare function configureApiRouteHeaders(account: ApiKeyAccount, fastMode: boolean): Headers;
 export declare function parseSse(response: Response): AsyncGenerator<AnthropicEvent>;
+declare function sendAnthropicRequest(options: {
+    model: Model<Api>;
+    context: Context;
+    streamOptions?: SimpleStreamOptions;
+    apiAccount?: NativeAccountView;
+    storagePath: string;
+    oauthAccountId?: string;
+    route?: string;
+    effortTransitions?: readonly MidConversationEffortTransition[];
+    onResolvedTools?: (tools: Tool[]) => void;
+}, deps?: {
+    fetch?: typeof fetch;
+    cacheKeep?: CacheKeepManager;
+}): Promise<Response>;
+/** Exercise one native send with isolated transport and scheduling in tests. */
+export declare const __sendPiAnthropicRequestForTest: typeof sendAnthropicRequest;
 export declare function primaryResponseAllowsApiFallback(preflight: Response | string): boolean;
 /**
  * Classify a successful stream by its first event without a tee: the bytes

package/dist/stream.js

@@ -335,7 +335,9 @@ export async function* parseSse(response) {
         reader.releaseLock();
     }
 }
-async function sendAnthropicRequest(options) {
+async function sendAnthropicRequest(options, deps = {}) {
+    const fetchImpl = deps.fetch ?? globalThis.fetch;
+    const keep = deps.cacheKeep ?? cacheKeepManager;
     const native = getPiNativeRuntime(options.storagePath);
     // Capture the serving service while authorization is open. Shutdown rejects
     // new serving work immediately but drains writes for already-served attempts.
@@ -385,8 +387,10 @@ async function sendAnthropicRequest(options) {
         signal: options.streamOptions?.signal,
         ...(!options.apiAccount && { redirect: 'error' }),
     };
-    if (!options.apiAccount)
-        await cacheKeepManager.track({
+    const trackAcceptedRequest = () => {
+        if (options.apiAccount || init.signal?.aborted)
+            return;
+        keep.track({
             sessionId: relayAffinity,
             url: input.toString(),
             headers,
@@ -397,6 +401,7 @@ async function sendAnthropicRequest(options) {
             oauthAccountIdentity: verifiedUuid,
             accountStoragePath: options.storagePath,
         });
+    };
     const reported = new Set();
     let retried401 = false;
     let lastTransport;
@@ -455,13 +460,13 @@ async function sendAnthropicRequest(options) {
             // OAuth bodies go gzipped as Claude Code sends them; `bodyText` stays
             // the signed identity text for dumps and CacheKeep.
             if (options.apiAccount || typeof init.body !== 'string')
-                return scope.redactResponse(await fetch(input, init));
+                return scope.redactResponse(await fetchImpl(input, init));
             return scope.redactResponse(await sendWithRequestGzip({
                 url: input,
                 body: init.body,
                 headers,
                 latch: getRequestGzipLatch(getRequestGzipLatchPath(options.storagePath)),
-                send: (wireBody, wireHeaders) => fetch(input, { ...init, body: wireBody, headers: wireHeaders }),
+                send: (wireBody, wireHeaders) => fetchImpl(input, { ...init, body: wireBody, headers: wireHeaders }),
             }));
         };
         let response;
@@ -537,8 +542,10 @@ async function sendAnthropicRequest(options) {
             const subject = apiSubject;
             native.usagePublications.publish(() => servingRuntime.publishApi(subject, { lastUsed: Date.now() }));
         }
-        if (attempt && response.ok)
+        if (attempt && response.ok) {
+            trackAcceptedRequest();
             markServed(attempt);
+        }
         return response;
     };
     if (options.apiAccount)
@@ -601,10 +608,14 @@ async function sendAnthropicRequest(options) {
     // If the request was cancelled since, end it as cancelled here.
     if (relay401Attempt && response.status === 401)
         await endIfCancelled(response, init.signal);
-    if (response.ok && lastTransport === 'relay' && relayAttempt)
+    if (response.ok && lastTransport === 'relay' && relayAttempt) {
+        trackAcceptedRequest();
         markServed(relayAttempt);
+    }
     return response;
 }
+/** Exercise one native send with isolated transport and scheduling in tests. */
+export const __sendPiAnthropicRequestForTest = sendAnthropicRequest;
 /**
  * If the request's signal is aborted, cancels the response body and throws the
  * signal's abort error; otherwise does nothing. Call it before a rejected

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cortexkit/pi-anthropic-auth",
-  "version": "2.4.11",
+  "version": "2.4.12",
   "type": "module",
   "repository": {
     "type": "git",
@@ -36,7 +36,7 @@
     "prepublishOnly": "bun run build"
   },
   "dependencies": {
-    "@cortexkit/anthropic-auth-core": "2.4.11"
+    "@cortexkit/anthropic-auth-core": "2.4.12"
   },
   "peerDependencies": {
     "@earendil-works/pi-ai": ">=0.86.1 <2",