@cortexkit/anthropic-auth-core 1.15.1 → 1.17.0

package/dist/quotas.js

@@ -1,4 +1,5 @@
 import { isOAuthAccount } from "./accounts.js";
+import { formatOAuthAccountTier } from "./oauth-profile.js";
 export const CLAUDE_QUOTAS_COMMAND_NAME = 'claude-quota';
 const WINDOW_LABELS = {
     five_hour: '5h',
@@ -43,14 +44,28 @@ function formatReset(resetsAt, now) {
         return '';
     return `, resets ${formatResetDuration(resetsAt, now)}`;
 }
-function formatWindow(key, window, now) {
+function formatWindow(key, window, now, bindingWindow) {
     if (!window)
         return `  - ${WINDOW_LABELS[key]}: unknown`;
-    return [
+    const line = [
         `  - ${WINDOW_LABELS[key]}: ${formatPercent(window.remainingPercent)} remaining`,
         ` (${formatPercent(window.usedPercent)} used`,
         `${formatReset(window.resetsAt, now)}, checked ${formatAge(window.checkedAt, now)})`,
     ].join('');
+    return bindingWindow === key ? `${line} •` : line;
+}
+export function formatQuotaMoney(money) {
+    try {
+        return new Intl.NumberFormat('en-US', {
+            style: 'currency',
+            currency: money.currency,
+            minimumFractionDigits: money.exponent,
+            maximumFractionDigits: money.exponent,
+        }).format(money.amountMinor / 10 ** money.exponent);
+    }
+    catch {
+        return `${money.amountMinor} ${money.currency}`;
+    }
 }
 function formatScopedWindow(window, now) {
     return [
@@ -75,6 +90,9 @@ export function buildFallbackQuotaSummaries(storage, errors = new Map()) {
             role: 'fallback',
             enabled: account.enabled !== false,
             quota: account.quota,
+            ...(formatOAuthAccountTier(account.profile) && {
+                tierLabel: formatOAuthAccountTier(account.profile),
+            }),
             lastRefreshedAt: account.lastRefreshedAt,
             ...(error && { error }),
         };
@@ -94,16 +112,28 @@ export function buildClaudeQuotaSummary(input) {
         const role = account.role === 'main' ? 'main' : 'fallback';
         const disabled = account.enabled === false ? ' disabled' : '';
         lines.push(`### ${account.name} (${role}${disabled})`);
+        if (account.tierLabel)
+            lines.push(`  - Tier: ${account.tierLabel}`);
         if (account.lastRefreshedAt) {
             lines.push(`  - Last token refresh: ${formatAge(account.lastRefreshedAt, now)}`);
         }
         if (account.error) {
             lines.push(`  - Error: ${account.error}`);
         }
-        lines.push(formatWindow('five_hour', account.quota?.five_hour, now));
-        lines.push(formatWindow('seven_day', account.quota?.seven_day, now));
+        lines.push(formatWindow('five_hour', account.quota?.five_hour, now, account.quota?.bindingWindow));
+        lines.push(formatWindow('seven_day', account.quota?.seven_day, now, account.quota?.bindingWindow));
         for (const window of account.quota?.scoped ?? []) {
-            lines.push(formatScopedWindow(window, now));
+            const line = formatScopedWindow(window, now);
+            lines.push(account.quota?.bindingWindow === window.id ? `${line} •` : line);
+        }
+        const extraUsage = account.quota?.extraUsage;
+        if (extraUsage) {
+            const used = formatQuotaMoney(extraUsage.used);
+            const limit = formatQuotaMoney(extraUsage.limit);
+            lines.push(`  - credits ${used}/${limit}${extraUsage.exhausted ? ' · exhausted' : ''}`);
+        }
+        if (account.quota?.fallbackAdvised === true) {
+            lines.push('  - → fallback advised');
         }
         lines.push('');
     }

package/dist/relay.d.ts

@@ -24,5 +24,12 @@ export declare function sendViaRelay(options: {
     fallback: () => Promise<Response>;
     affinity?: string | null;
     optimisticResponse?: boolean;
+    /**
+     * Observes genuine upstream headers once per delivered response. WebSocket
+     * retries before downstream bytes replace staged headers, so the callback
+     * receives only the final accepted response; duplicate response_start frames
+     * within an attempt are ignored.
+     */
+    onResponseHeaders?: (headers: Headers) => void;
 }): Promise<Response>;
 export declare const WORKER_SCRIPT = "\nasync function hashBody(body) {\n  const bytes = new TextEncoder().encode(body)\n  const digest = await crypto.subtle.digest('SHA-256', bytes)\n  const hex = [...new Uint8Array(digest)].map((b) => b.toString(16).padStart(2, '0')).join('')\n  return 'sha256:' + hex\n}\n\nfunction applyPatch(base, patch) {\n  return base.slice(0, patch.start) + patch.insert + base.slice(patch.start + patch.deleteCount)\n}\n\nfunction applyPatchSet(base, patch) {\n  if (!Array.isArray(patch)) return applyPatch(base, patch)\n  if (patch.length === 0) return base\n\n  let result = ''\n  let cursor = 0\n  for (const item of patch) {\n    if (item.start < cursor) return { error: 'overlapping patch', status: 400 }\n    result += base.slice(cursor, item.start)\n    result += item.insert\n    cursor = item.start + item.deleteCount\n  }\n  return result + base.slice(cursor)\n}\n\nasync function readState(env, affinity) {\n  const raw = await env.RELAY_STATE.get('session:' + affinity)\n  return raw ? JSON.parse(raw) : null\n}\n\nasync function writeState(env, affinity, state) {\n  await env.RELAY_STATE.put('session:' + affinity, JSON.stringify(state), { expirationTtl: 86400 })\n}\n\nasync function resolveBody(env, payload) {\n  if (payload.mode === 'full_sync') return payload.body\n  if (payload.mode === 'patch') {\n    const state = await readState(env, payload.affinity)\n    if (!state || state.hash !== payload.base_hash) {\n      return { error: 'state mismatch', status: 409 }\n    }\n    return applyPatchSet(state.body, payload.patch)\n  }\n  return { error: 'unknown mode', status: 400 }\n}\n\nasync function prepareUpstream(env, payload) {\n  if ((payload.protocol !== 1 && payload.protocol !== 2) || payload.type !== 'request' || !payload.affinity || !payload.upstream?.url || !payload.next_hash) {\n    return { error: 'invalid payload', status: 400 }\n  }\n\n  const body = await resolveBody(env, payload)\n  if (body && typeof body === 'object' && 'error' in body) return body\n\n  if (typeof body !== 'string' || (await hashBody(body)) !== payload.next_hash) {\n    return { error: 'hash mismatch', status: 409 }\n  }\n\n  const stateWrite = writeState(env, payload.affinity, { body, hash: payload.next_hash, revision: payload.revision }).catch(() => {})\n  console.log(JSON.stringify({\n    relay: 'opencode-anthropic-auth',\n    transport: 'relay',\n    mode: payload.mode,\n    revision: payload.revision,\n    affinity: String(payload.affinity).slice(0, 12),\n    bodyBytes: body.length,\n  }))\n\n  return { body, stateWrite }\n}\n\nfunction resolveBodyFromState(state, payload) {\n  if (payload.mode === 'full_sync') return payload.body\n  if (payload.mode === 'patch') {\n    if (!state || state.hash !== payload.base_hash) {\n      return { error: 'state mismatch', status: 409 }\n    }\n    return applyPatchSet(state.body, payload.patch)\n  }\n  return { error: 'unknown mode', status: 400 }\n}\n\nfunction deferWorkerTask(task) {\n  return new Promise((resolve) => setTimeout(resolve, 0)).then(task)\n}\n\nconst WEBSOCKET_STREAM_FLUSH_BYTES = 1024\nconst WEBSOCKET_STREAM_FLUSH_MS = 20\n\nfunction createWebSocketStreamCoalescer(socket, options = {}) {\n  const flushBytes = options.flushBytes || WEBSOCKET_STREAM_FLUSH_BYTES\n  const flushMs = options.flushMs ?? WEBSOCKET_STREAM_FLUSH_MS\n  let chunks = []\n  let byteLength = 0\n  let flushTimer = null\n  const stats = {\n    upstreamChunks: 0,\n    upstreamBytes: 0,\n    frames: 0,\n    frameBytes: 0,\n  }\n\n  const clearFlushTimer = () => {\n    if (flushTimer === null) return\n    clearTimeout(flushTimer)\n    flushTimer = null\n  }\n\n  const mergeChunks = () => {\n    if (chunks.length === 1) return chunks[0]\n    const frame = new Uint8Array(byteLength)\n    let offset = 0\n    for (const chunk of chunks) {\n      frame.set(chunk, offset)\n      offset += chunk.byteLength\n    }\n    return frame\n  }\n\n  const flush = () => {\n    clearFlushTimer()\n    if (byteLength === 0) return\n    const frame = mergeChunks()\n    chunks = []\n    byteLength = 0\n    socket.send(frame)\n    stats.frames += 1\n    stats.frameBytes += frame.byteLength\n  }\n\n  const scheduleFlush = () => {\n    if (flushTimer !== null) return\n    flushTimer = setTimeout(flush, flushMs)\n  }\n\n  return {\n    push(value) {\n      const chunk = value instanceof Uint8Array ? value : new Uint8Array(value)\n      if (chunk.byteLength === 0) return\n      stats.upstreamChunks += 1\n      stats.upstreamBytes += chunk.byteLength\n      chunks.push(chunk)\n      byteLength += chunk.byteLength\n      if (byteLength >= flushBytes) flush()\n      else scheduleFlush()\n    },\n    flush,\n    stats() {\n      return { ...stats, pendingBytes: byteLength }\n    },\n  }\n}\n\nfunction checkpointWebSocketState(env, payload, body, nextState) {\n  return deferWorkerTask(async () => {\n    try {\n      if ((await hashBody(body)) !== payload.next_hash) {\n        throw new Error('hash mismatch')\n      }\n      await writeState(env, payload.affinity, nextState)\n    } catch (error) {\n      console.log(JSON.stringify({\n        relay: 'opencode-anthropic-auth',\n        transport: 'websocket',\n        event: 'checkpoint_write_failed',\n        affinity: String(payload.affinity).slice(0, 12),\n        message: error instanceof Error ? error.message : String(error),\n      }))\n    }\n  })\n}\n\nasync function prepareWebSocketUpstream(env, state, payload) {\n  if (payload.protocol !== 2 || payload.type !== 'request' || !payload.id || !payload.affinity || !payload.upstream?.url || !payload.next_hash) {\n    return { error: 'invalid payload', status: 400 }\n  }\n\n  const body = resolveBodyFromState(state, payload)\n  if (body && typeof body === 'object' && 'error' in body) return body\n  if (typeof body !== 'string') return { error: 'invalid body', status: 400 }\n  if ((await hashBody(body)) !== payload.next_hash) {\n    return { error: 'hash mismatch', status: 409 }\n  }\n\n  const nextState = { body, hash: payload.next_hash, revision: payload.revision }\n  const checkpoint = checkpointWebSocketState(env, payload, body, nextState)\n  const logAccepted = () => console.log(JSON.stringify({\n    relay: 'opencode-anthropic-auth',\n    transport: 'websocket',\n    mode: payload.mode,\n    revision: payload.revision,\n    affinity: String(payload.affinity).slice(0, 12),\n    bodyBytes: body.length,\n  }))\n\n  return { body, state: nextState, checkpoint, logAccepted }\n}\n\nfunction headersToObject(headers) {\n  const result = {}\n  for (const [key, value] of headers.entries()) result[key] = value\n  return result\n}\n\nasync function handleRelayPayload(env, payload) {\n  const prepared = await prepareUpstream(env, payload)\n  if (prepared.error) return prepared\n  const upstream = await fetch(payload.upstream.url, {\n    method: payload.upstream.method || 'POST',\n    headers: payload.upstream.headers,\n    body: prepared.body,\n  })\n  return { upstream, stateWrite: prepared.stateWrite }\n}\n\nasync function handleWebSocket(socket, env, ctx, payload, getState, setState) {\n  const heartbeat = setInterval(() => {\n    try {\n      socket.send(JSON.stringify({ protocol: 2, type: 'keepalive' }))\n    } catch {\n      clearInterval(heartbeat)\n    }\n  }, 15000)\n  let streamCoalescer = null\n  try {\n    const result = await prepareWebSocketUpstream(env, getState(), payload)\n    if (result.error) {\n      socket.send(JSON.stringify({ protocol: 2, type: 'error', id: payload.id, status: result.status, message: result.error }))\n      return\n    }\n\n    setState(result.state)\n    socket.send(JSON.stringify({ protocol: 2, type: 'accepted', id: payload.id, hash: result.state.hash, revision: result.state.revision }))\n    ctx?.waitUntil?.(deferWorkerTask(result.logAccepted))\n\n    const upstreamPromise = fetch(payload.upstream.url, {\n      method: payload.upstream.method || 'POST',\n      headers: payload.upstream.headers,\n      body: result.body,\n    })\n    ctx?.waitUntil?.(result.checkpoint)\n    const upstream = await upstreamPromise\n    socket.send(JSON.stringify({\n      protocol: 2,\n      type: 'response_start',\n      id: payload.id,\n      status: upstream.status,\n      statusText: upstream.statusText,\n      headers: headersToObject(upstream.headers),\n    }))\n\n    const reader = upstream.body?.getReader()\n    if (reader) {\n      streamCoalescer = createWebSocketStreamCoalescer(socket)\n      while (true) {\n        const { done, value } = await reader.read()\n        if (done) break\n        streamCoalescer.push(value)\n      }\n      streamCoalescer.flush()\n    }\n    const stats = streamCoalescer?.stats?.() || {}\n    socket.send(JSON.stringify({ protocol: 2, type: 'done', id: payload.id, ...stats }))\n  } catch (error) {\n    try {\n      streamCoalescer?.flush?.()\n    } catch {}\n    socket.send(JSON.stringify({ protocol: 2, type: 'error', id: payload.id, status: 500, message: error instanceof Error ? error.message : String(error) }))\n  } finally {\n    clearInterval(heartbeat)\n  }\n}\n\nexport default {\n  async fetch(request, env, ctx) {\n    if (request.headers.get('Upgrade') === 'websocket') {\n      const url = new URL(request.url)\n      const token = url.searchParams.get('token')\n      const affinity = url.searchParams.get('affinity')\n      if (token !== env.RELAY_TOKEN) return new Response('unauthorized', { status: 401 })\n      if (!affinity) return new Response('missing affinity', { status: 400 })\n      const pair = new WebSocketPair()\n      const client = pair[0]\n      const server = pair[1]\n      server.binaryType = 'arraybuffer'\n      let state = null\n      let ready = false\n      server.accept()\n      const loadState = readState(env, affinity).then((loadedState) => {\n        state = loadedState\n        ready = true\n        server.send(JSON.stringify({\n          protocol: 2,\n          type: 'ready',\n          state: state ? { hash: state.hash, revision: state.revision } : null,\n        }))\n      }).catch((error) => {\n        server.send(JSON.stringify({ protocol: 2, type: 'error', status: 500, message: error instanceof Error ? error.message : String(error) }))\n        server.close(1011, 'state load failed')\n      })\n      ctx?.waitUntil?.(loadState)\n      if (!ctx?.waitUntil) void loadState\n      let busy = false\n      server.addEventListener('message', (event) => {\n        if (!ready) {\n          server.send(JSON.stringify({ protocol: 2, type: 'error', status: 425, message: 'relay state is not ready' }))\n          return\n        }\n        if (busy) {\n          server.send(JSON.stringify({ protocol: 2, type: 'error', status: 429, message: 'request already in flight' }))\n          return\n        }\n        let payload\n        try {\n          payload = JSON.parse(event.data)\n        } catch {\n          server.send(JSON.stringify({ protocol: 2, type: 'error', status: 400, message: 'invalid JSON payload' }))\n          return\n        }\n        payload.affinity = affinity\n        busy = true\n        const run = handleWebSocket(server, env, ctx, payload, () => state, (nextState) => { state = nextState }).finally(() => { busy = false })\n        ctx?.waitUntil?.(run)\n        if (!ctx?.waitUntil) void run\n      })\n      return new Response(null, { status: 101, webSocket: client })\n    }\n\n    if (request.method === 'GET') {\n      return Response.json({ status: 'ok', transports: ['http', 'websocket'] })\n    }\n    if (request.method !== 'POST') return new Response('method not allowed', { status: 405 })\n    if (request.headers.get('x-relay-token') !== env.RELAY_TOKEN) {\n      return new Response('unauthorized', { status: 401 })\n    }\n\n    try {\n      const payload = await request.json()\n      const result = await handleRelayPayload(env, payload)\n      if (result.error) return Response.json({ error: result.error }, { status: result.status })\n\n      if (result.stateWrite) ctx.waitUntil(result.stateWrite)\n\n      const upstream = result.upstream\n      return new Response(upstream.body, {\n        status: upstream.status,\n        statusText: upstream.statusText,\n        headers: upstream.headers,\n      })\n    } catch (error) {\n      return Response.json(\n        { error: error instanceof Error ? error.message : 'internal relay error' },\n        { status: 502 },\n      )\n    }\n  },\n}\n";

package/dist/relay.js

@@ -323,12 +323,12 @@ class PersistentRelaySession {
         this.serverState = undefined;
         this.connecting = undefined;
     }
-    send(payload, bodyText, optimisticResponse = false) {
+    send(payload, bodyText, optimisticResponse = false, onResponseHeaders) {
         this.touch();
         const enqueuedAt = perfNowMs();
         const start = this.queue
             .catch(() => { })
-            .then(() => this.startQueued(payload, bodyText, enqueuedAt, optimisticResponse));
+            .then(() => this.startQueued(payload, bodyText, enqueuedAt, optimisticResponse, onResponseHeaders));
         const result = start.then(async ({ response, getPayload }) => ({
             response: await response,
             payload: getPayload(),
@@ -339,7 +339,7 @@ class PersistentRelaySession {
         this.queue = start.then(({ done }) => done.catch(() => { }), () => { });
         return result;
     }
-    async startQueued(payload, bodyText, enqueuedAt, optimisticResponse) {
+    async startQueued(payload, bodyText, enqueuedAt, optimisticResponse, onResponseHeaders) {
         const connectStart = perfNowMs();
         await this.ensureConnected();
         const connectedAt = perfNowMs();
@@ -363,7 +363,7 @@ class PersistentRelaySession {
         }
         relayLog(`perf websocket send_start session=${shortAffinity(this.affinity)} request=${requestPayload.id} mode=${requestPayload.mode} queueMs=${formatMs(connectedAt - enqueuedAt)} connectMs=${formatMs(connectedAt - connectStart)} bodyBytes=${bodyText.length} relayBytes=${JSON.stringify(requestPayload).length}`);
         let activePayload = requestPayload;
-        const first = this.sendPayload(requestPayload, bodyText, optimisticResponse);
+        const first = this.sendPayload(requestPayload, bodyText, optimisticResponse, onResponseHeaders);
         void first.done.catch(() => { });
         let activeDone = first.done;
         const response = first.response.catch((error) => {
@@ -376,7 +376,7 @@ class PersistentRelaySession {
             fullSync.id = createRequestId();
             fullSync.revision = (this.serverState?.revision ?? 0) + 1;
             activePayload = fullSync;
-            const retry = this.sendPayload(fullSync, bodyText, optimisticResponse);
+            const retry = this.sendPayload(fullSync, bodyText, optimisticResponse, onResponseHeaders);
             void retry.done.catch(() => { });
             activeDone = retry.done;
             return retry.response;
@@ -457,7 +457,7 @@ class PersistentRelaySession {
         });
         await this.connecting;
     }
-    sendPayload(payload, bodyText, optimisticResponse) {
+    sendPayload(payload, bodyText, optimisticResponse, onResponseHeaders) {
         const socket = this.socket;
         if (!socket || socket.readyState !== WebSocket.OPEN) {
             throw new Error('relay websocket is not connected');
@@ -486,6 +486,9 @@ class PersistentRelaySession {
                 streamByteCount: 0,
                 retryAttempts: 0,
                 retryingBeforeResponse: false,
+                onResponseHeaders,
+                stagedResponseHeaders: undefined,
+                responseHeadersDelivered: false,
             };
             this.pending = pending;
             this.resetPendingTimeout(pending);
@@ -540,6 +543,8 @@ class PersistentRelaySession {
                 pending.accepted = false;
                 pending.acceptedAt = undefined;
                 pending.responseStartedAt = undefined;
+                pending.stagedResponseHeaders = undefined;
+                pending.responseHeadersDelivered = false;
                 pending.sentAt = perfNowMs();
                 pending.retryingBeforeResponse = false;
                 const socket = this.socket;
@@ -559,10 +564,21 @@ class PersistentRelaySession {
         return true;
     }
     enqueueStreamChunk(pending, chunk) {
+        this.deliverResponseHeaders(pending);
         pending.streamChunkCount += 1;
         pending.streamByteCount += chunk.byteLength;
         pending.streamController?.enqueue(chunk);
     }
+    deliverResponseHeaders(pending) {
+        if (pending.responseHeadersDelivered ||
+            pending.stagedResponseHeaders == null)
+            return;
+        pending.responseHeadersDelivered = true;
+        try {
+            pending.onResponseHeaders?.(pending.stagedResponseHeaders);
+        }
+        catch { }
+    }
     handleBinaryChunk(data) {
         const pending = this.pending;
         if (!pending?.responseStarted)
@@ -609,12 +625,17 @@ class PersistentRelaySession {
             return;
         }
         if (message.type === 'response_start') {
+            if (pending.responseStartedAt != null)
+                return;
             const responseStartedAt = perfNowMs();
             pending.responseStartedAt = responseStartedAt;
             relayLog(`perf websocket response_start session=${shortAffinity(this.affinity)} request=${pending.payload.id} sentMs=${formatMs(responseStartedAt - pending.sentAt)} upstreamMs=${pending.acceptedAt == null ? 'unknown' : formatMs(responseStartedAt - pending.acceptedAt)} status=${message.status}`);
             clearTimeout(pending.timeout);
+            if (message.headers)
+                pending.stagedResponseHeaders = new Headers(message.headers);
             this.resolvePendingResponse(pending, message.status, message.statusText, message.headers);
             if (pending.optimisticResponse && message.status >= 400) {
+                this.deliverResponseHeaders(pending);
                 pending.streamController?.enqueue(new TextEncoder().encode(`event: error\ndata: ${JSON.stringify({ type: 'error', error: { type: 'relay_upstream_error', message: `Relay upstream returned HTTP ${message.status}` } })}\n\n`));
                 this.finishPending();
             }
@@ -635,6 +656,7 @@ class PersistentRelaySession {
             // inject an SSE error event and close cleanly.
             if (pending.optimisticResponse && pending.responseStartedAt != null) {
                 relayLog(`websocket relay error during optimistic response session=${shortAffinity(this.affinity)} request=${pending.payload.id} status=${message.status} message=${message.message || 'unknown'} chunks=${pending.streamChunkCount} bytes=${pending.streamByteCount}`);
+                this.deliverResponseHeaders(pending);
                 pending.streamController?.enqueue(new TextEncoder().encode(`event: error\ndata: ${JSON.stringify({ type: 'error', error: { type: 'relay_error', message: message.message || 'relay error' } })}\n\n`));
                 this.finishPending();
                 this.socket?.close();
@@ -661,6 +683,7 @@ class PersistentRelaySession {
         relayLog(`perf websocket done session=${shortAffinity(this.affinity)} request=${pending.payload.id} sentMs=${formatMs(finishedAt - pending.sentAt)} streamMs=${pending.responseStartedAt == null ? 'unknown' : formatMs(finishedAt - pending.responseStartedAt)} chunks=${pending.streamChunkCount} bytes=${pending.streamByteCount}${workerStats}`);
         clearTimeout(pending.timeout);
         if (!pending.streamDone) {
+            this.deliverResponseHeaders(pending);
             pending.streamDone = true;
             pending.streamController?.close();
         }
@@ -727,7 +750,7 @@ function getPersistentRelaySession(config, affinity) {
     return session;
 }
 export async function sendViaRelay(options) {
-    const { config, input, init, headers, body, fallback, affinity: explicitAffinity, optimisticResponse, } = options;
+    const { config, input, init, headers, body, fallback, affinity: explicitAffinity, optimisticResponse, onResponseHeaders, } = options;
     if (!config || !isRelayableAnthropicRequest(input, body))
         return fallback();
     const affinity = explicitAffinity ||
@@ -769,7 +792,7 @@ export async function sendViaRelay(options) {
         if (config.transport === 'websocket') {
             try {
                 const session = getPersistentRelaySession(config, affinity);
-                result = await session.send(payload, bodyText, optimisticResponse === true);
+                result = await session.send(payload, bodyText, optimisticResponse === true, onResponseHeaders);
             }
             catch (error) {
                 relayLog(`websocket relay failed session=${shortAffinity(affinity)}; trying http relay: ${error instanceof Error ? error.message : String(error)}`);
@@ -802,6 +825,12 @@ export async function sendViaRelay(options) {
         });
         if (!result.usedRelay)
             return result.response;
+        if (result.transport === 'http') {
+            try {
+                onResponseHeaders?.(result.response.headers);
+            }
+            catch { }
+        }
         if (result.transport === 'http') {
             updateLocalRelayState(affinity, bodyText, result.payload.next_hash, result.payload.revision);
         }

package/dist/routing.d.ts

@@ -1,12 +1,14 @@
 import { type AccountStorage, type RoutingMode } from './accounts.ts';
 export declare const CLAUDE_ROUTING_COMMAND_NAME = "claude-routing";
-export declare const ROUTING_MODES: readonly ["main-first", "fallback-first"];
+export declare const ROUTING_MODES: readonly ["main-first", "fallback-first", "sticky-balanced"];
 export declare const DEFAULT_ROUTING_MODE: RoutingMode;
 type RoutingCommandAction = {
     type: 'status';
 } | {
     type: 'mode';
     mode: RoutingMode;
+} | {
+    type: 'reset';
 } | {
     type: 'usage';
 };

package/dist/routing.js

@@ -1,8 +1,12 @@
 import { createEmptyStorage, getAccountStoragePath, loadAccounts, saveAccounts, } from "./accounts.js";
 export const CLAUDE_ROUTING_COMMAND_NAME = 'claude-routing';
-export const ROUTING_MODES = ['main-first', 'fallback-first'];
+export const ROUTING_MODES = [
+    'main-first',
+    'fallback-first',
+    'sticky-balanced',
+];
 export const DEFAULT_ROUTING_MODE = 'main-first';
-const ROUTING_USAGE = 'Usage: `/claude-routing`, `/claude-routing main-first`, or `/claude-routing fallback-first`.';
+const ROUTING_USAGE = 'Usage: `/claude-routing`, `/claude-routing main-first`, `/claude-routing fallback-first`, `/claude-routing sticky-balanced`, or `/claude-routing reset`.';
 function normalizeRoutingMode(value) {
     return typeof value === 'string' &&
         ROUTING_MODES.includes(value)
@@ -25,6 +29,8 @@ export function parseRoutingCommandAction(argumentsText) {
     const tokens = argumentsText.trim().toLowerCase().split(/\s+/).filter(Boolean);
     if (tokens.length === 0)
         return { type: 'status' };
+    if (tokens.length === 1 && tokens[0] === 'reset')
+        return { type: 'reset' };
     if (tokens.length === 1 && ROUTING_MODES.includes(tokens[0])) {
         return { type: 'mode', mode: tokens[0] };
     }
@@ -39,6 +45,9 @@ function routingDescription(mode) {
     if (mode === 'fallback-first') {
         return 'Try usable fallback accounts before the main account. If no fallback succeeds, try the main account.';
     }
+    if (mode === 'sticky-balanced') {
+        return 'Assign each session to a quota-weighted OAuth account, keep it sticky across transient failures, and migrate only for confirmed long-lived exhaustion or permanent account failure.';
+    }
     return 'Try the main account first. Use fallback accounts only when quota policy or fallback errors require it.';
 }
 export function buildRoutingStatusSummary(input) {
@@ -66,6 +75,15 @@ export function executeRoutingCommand(input) {
             ...buildRoutingStatusSummary({ mode: action.mode }).split('\n').slice(2),
         ].join('\n');
     }
+    if (action.type === 'reset') {
+        return [
+            '## Claude Routing Assignment Reset',
+            '',
+            'The current session will be assigned again on its next request.',
+            '',
+            ...buildRoutingStatusSummary({ mode }).split('\n').slice(2),
+        ].join('\n');
+    }
     return [
         '## Claude Routing Usage',
         '',

package/dist/sticky-routing.d.ts

@@ -0,0 +1,86 @@
+import { type AccountStorage, type OAuthQuotaSnapshot } from './accounts.ts';
+export declare const STICKY_ROUTING_MAIN_ACCOUNT_ID = "main";
+export declare const STICKY_ROUTING_SHORT_RESET_GRACE_MS: number;
+export declare const STICKY_ROUTING_ASSIGNMENT_TTL_MS: number;
+export type StickyRouteFamily = 'fable' | 'opus' | 'general';
+export type StickyRouteCandidate = {
+    accountId: string;
+    quota: OAuthQuotaSnapshot;
+    order: number;
+};
+export type StickyRouteAssignment = {
+    accountId: string;
+    family: StickyRouteFamily;
+    assignedAt: number;
+    lastSeenAt: number;
+    initialInputBytes: number;
+    quotaCheckedAt: number;
+};
+export type StickyRouteResolution = {
+    accountId: string;
+    assignment: StickyRouteAssignment;
+    created: boolean;
+    migrated: boolean;
+};
+export type StickyQuotaFailureDecision = {
+    action: 'retain';
+    reason: 'not-exhausted' | 'unknown';
+} | {
+    action: 'hold';
+    reason: 'five-hour-short-reset';
+    retryAfterSeconds: number;
+} | {
+    action: 'migrate';
+    reason: 'five-hour' | 'seven-day' | 'model-scoped';
+};
+export declare function stickyQuotaSnapshotIsFresh(quota: OAuthQuotaSnapshot | undefined, storage: AccountStorage | null, now?: number, modelId?: string): boolean;
+export declare function stickyRouteFamilyForModel(model: unknown): StickyRouteFamily;
+export declare function stickyRouteCandidateWeight(input: {
+    candidate: StickyRouteCandidate;
+    family: StickyRouteFamily;
+    modelId?: string;
+    storage: AccountStorage | null;
+    now?: number;
+}): number;
+export declare function stickyRouteKnownFableExhausted(candidate: StickyRouteCandidate, storage: AccountStorage | null, now?: number): boolean;
+export declare function decideStickyQuotaFailure(input: {
+    quota: OAuthQuotaSnapshot | undefined;
+    modelId?: string;
+    now?: number;
+}): StickyQuotaFailureDecision;
+export declare function getStickyRoutingStatePath(accountStoragePath: string): string;
+export declare function stickyRetryAfterWithJitter(sessionId: string, retryAfterSeconds: number): number;
+export declare class StickySessionRouter {
+    private readonly options;
+    private loaded;
+    private state;
+    private stateMtimeNs;
+    private touchChain;
+    constructor(options: {
+        path: string;
+        now?: () => number;
+        assignmentTtlMs?: number;
+    });
+    private now;
+    private readState;
+    private readStateMtimeNs;
+    private refreshStateIfChanged;
+    private acquireLock;
+    private assignmentIsActive;
+    private prune;
+    private writeState;
+    private scheduleTouch;
+    private selectCandidate;
+    resolve(input: {
+        sessionId: string;
+        family: StickyRouteFamily;
+        modelId?: string;
+        candidates: readonly StickyRouteCandidate[];
+        retainAccountIds: ReadonlySet<string>;
+        storage: AccountStorage | null;
+        inputBytes: number;
+        preferredAccountId?: string;
+        excludeAccountIds?: ReadonlySet<string>;
+    }): Promise<StickyRouteResolution | null>;
+    clear(sessionId: string): Promise<void>;
+}