@code-yeongyu/senpi 2026.8.28 → 2026.8.30-2

package/dist/core/credential-pool/rotation-stream.js

@@ -1,9 +1,11 @@
 import { createHash, randomUUID } from "node:crypto";
 import { rendezvousOrder } from "@earendil-works/pi-ai/auth/pool/select";
 import { listSlots as listCredentialSlots } from "@earendil-works/pi-ai/auth/pool/slots";
+import { resolveConfigValue } from "../resolve-config-value.js";
 import { classifyCredentialFailure } from "./classify.js";
 import { discoverEnvSlots } from "./env-slots.js";
 import { runCredentialFailover } from "./failover.js";
+import { acquireHalfOpenLease } from "./state-store.js";
 /** The exact hash the claude-sdk-oauth affinity oracle uses, so pools never remap. */
 export const sha256SlotHasher = (input) => createHash("sha256").update(input).digest().readBigUInt64BE(0);
 function overlayState(slot, state) {
@@ -14,6 +16,7 @@ function overlayState(slot, state) {
         ...(state.blockedUntil === undefined ? {} : { blockedUntil: state.blockedUntil }),
         ...(state.blockReason === undefined ? {} : { blockReason: state.blockReason }),
         ...(state.failureCount === undefined ? {} : { failureCount: state.failureCount }),
+        ...(state.lease === undefined ? {} : { lease: state.lease }),
     };
 }
 /**
@@ -23,26 +26,87 @@ function overlayState(slot, state) {
  * persisted health applies only while its HMAC revision still matches the
  * current env value, so rotating a key in place clears its own stale block.
  */
-export async function listRotationSlots(sources) {
+export async function listRotationSlots(sources, options = {}) {
+    const acquireLeases = options.acquireLeases !== false;
     const { providerId, credential, env, repository } = sources;
+    const policySlots = Object.entries(sources.policy?.slots ?? {}).flatMap(([name, ref]) => {
+        const envVarName = ref.env ?? `models.json:${name}`;
+        const key = ref.env !== undefined ? env(ref.env) : ref.value !== undefined ? resolveConfigValue(ref.value, {}) : undefined;
+        if (!key)
+            return [];
+        return [{ name, envVarName, key, source: "env" }];
+    });
     if (credential) {
         const state = await repository.listSlots(providerId, "stored");
-        return listCredentialSlots(credential).map((slot) => overlayState({ name: slot.name, lane: "stored" }, state[slot.name]));
+        const slots = [];
+        for (const slot of listCredentialSlots(credential)) {
+            const current = state[slot.name];
+            if (acquireLeases &&
+                current?.blockedUntil !== undefined &&
+                current.blockedUntil <= (sources.now ?? Date.now)()) {
+                const lease = await acquireHalfOpenLease(repository, providerId, "stored", slot.name, {
+                    now: (sources.now ?? Date.now)(),
+                });
+                if (!lease)
+                    continue;
+                const leased = await repository.listSlots(providerId, "stored");
+                slots.push(overlayState({
+                    name: slot.name,
+                    lane: "stored",
+                    pinned: credential.pinned === slot.name,
+                }, leased[slot.name]));
+                continue;
+            }
+            slots.push(overlayState({
+                name: slot.name,
+                lane: "stored",
+                pinned: credential.pinned === slot.name,
+            }, current));
+        }
+        if (policySlots.length === 0)
+            return slots;
+        const namedSources = {
+            ...sources,
+            credential: undefined,
+            policy: { ...sources.policy, slots: {} },
+        };
+        const namedSlots = await listEnvRotationSlots(namedSources, policySlots, acquireLeases);
+        return [...slots, ...namedSlots];
     }
-    const envSlots = discoverEnvSlots(providerId, env);
+    const envSlots = [...discoverEnvSlots(providerId, env), ...policySlots];
+    return listEnvRotationSlots(sources, envSlots, acquireLeases);
+}
+async function listEnvRotationSlots(sources, envSlots, acquireLeases = true) {
     if (envSlots.length === 0)
         return [];
+    const { providerId, repository } = sources;
     const state = await repository.listSlots(providerId, "env");
     const slots = [];
     for (const slot of envSlots) {
         const persisted = state[slot.name];
         const revision = await repository.envCredentialRevision(slot.envVarName, slot.key);
-        const applicable = persisted?.credentialRevision === revision ? persisted : undefined;
-        slots.push(overlayState({ name: slot.name, lane: "env", envKey: slot.key, envVarName: slot.envVarName }, applicable));
+        let applicable = persisted?.credentialRevision === revision ? persisted : undefined;
+        if (acquireLeases &&
+            applicable?.blockedUntil !== undefined &&
+            applicable.blockedUntil <= (sources.now ?? Date.now)()) {
+            const lease = await acquireHalfOpenLease(repository, providerId, "env", slot.name, {
+                now: (sources.now ?? Date.now)(),
+            });
+            if (!lease)
+                continue;
+            const leased = await repository.listSlots(providerId, "env");
+            applicable = leased[slot.name];
+        }
+        slots.push(overlayState({
+            name: slot.name,
+            lane: "env",
+            envKey: slot.key,
+            envVarName: slot.envVarName,
+        }, applicable));
     }
     return slots;
 }
-function blockPatch(block, current, now, credentialRevision) {
+function blockPatch(block, current, now, credentialRevision, policy) {
     const failureCount = (current?.failureCount ?? 0) + 1;
     const base = {
         failureCount,
@@ -50,7 +114,11 @@ function blockPatch(block, current, now, credentialRevision) {
         ...(current?.lastSuccessAt === undefined ? {} : { lastSuccessAt: current.lastSuccessAt }),
     };
     if (block.reason === "rate_limit") {
-        return { ...base, blockedUntil: now + block.cooldownMs, blockReason: "rate_limit" };
+        return {
+            ...base,
+            blockedUntil: now + Math.min(policy?.cooldownCapMs ?? block.cooldownMs, block.cooldownMs),
+            blockReason: "rate_limit",
+        };
     }
     return { ...base, blockReason: block.reason };
 }
@@ -70,11 +138,15 @@ export function streamWithCredentialRotation(options) {
     const { sources, runAttempt } = options;
     const hasher = options.hasher ?? sha256SlotHasher;
     const affinityKey = options.affinityKey ?? randomUUID();
+    const useAffinity = sources.policy?.affinity !== false;
     const now = sources.now ?? Date.now;
     return runCredentialFailover({
         listSlots: () => listRotationSlots(sources),
         select: (candidates) => {
-            const ordered = rendezvousOrder(affinityKey, candidates, hasher);
+            const pinned = candidates.find((candidate) => candidate.pinned === true);
+            if (pinned)
+                return pinned;
+            const ordered = useAffinity ? rendezvousOrder(affinityKey, candidates, hasher) : candidates;
             const winner = ordered[0];
             if (!winner)
                 throw new Error("credential rotation selected from an empty candidate set");
@@ -83,12 +155,27 @@ export function streamWithCredentialRotation(options) {
         runAttempt,
         isCommittedOutput: (event) => event.type !== "start",
         errorFromEvent,
-        classify: classifyCredentialFailure,
+        classify: (error, context) => classifyCredentialFailure(error, {
+            ...context,
+            cooldownBaseMs: sources.policy?.cooldownBaseMs,
+            cooldownCapMs: sources.policy?.cooldownCapMs,
+        }),
+        onSuccess: async (slot) => {
+            await sources.repository.mutateSlotState(sources.providerId, slot.lane, slot.name, (current) => current
+                ? {
+                    ...current,
+                    lastSuccessAt: now(),
+                    lease: undefined,
+                    blockedUntil: undefined,
+                    blockReason: undefined,
+                }
+                : undefined);
+        },
         persistBlock: async (slot, block) => {
             const revision = slot.lane === "env" && slot.envVarName !== undefined && slot.envKey !== undefined
                 ? await sources.repository.envCredentialRevision(slot.envVarName, slot.envKey)
                 : undefined;
-            await sources.repository.mutateSlotState(sources.providerId, slot.lane, slot.name, (current) => blockPatch(block, current, now(), revision));
+            await sources.repository.mutateSlotState(sources.providerId, slot.lane, slot.name, (current) => blockPatch(block, current, now(), revision, sources.policy));
         },
         now,
     });

package/dist/core/credential-pool/rotation-stream.js.map

@@ -1 +1 @@
-{"version":3,"file":"rotation-stream.js","sourceRoot":"","sources":["../../../src/core/credential-pool/rotation-stream.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAErD,OAAO,EAAE,eAAe,EAAmB,MAAM,wCAAwC,CAAC;AAC1F,OAAO,EAAE,SAAS,IAAI,mBAAmB,EAAE,MAAM,uCAAuC,CAAC;AACzF,OAAO,EAAwB,yBAAyB,EAAE,MAAM,eAAe,CAAC;AAChF,OAAO,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAClD,OAAO,EAAgB,qBAAqB,EAAE,MAAM,eAAe,CAAC;AAGpE,sFAAsF;AACtF,MAAM,CAAC,MAAM,gBAAgB,GAAe,CAAC,KAAK,EAAE,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,MAAM,EAAE,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;AAmBtH,SAAS,YAAY,CAAC,IAAkB,EAAE,KAAsC;IAC/E,IAAI,CAAC,KAAK;QAAE,OAAO,IAAI,CAAC;IACxB,OAAO;QACN,GAAG,IAAI;QACP,GAAG,CAAC,KAAK,CAAC,YAAY,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,KAAK,CAAC,YAAY,EAAE,CAAC;QACjF,GAAG,CAAC,KAAK,CAAC,WAAW,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,KAAK,CAAC,WAAW,EAAE,CAAC;QAC9E,GAAG,CAAC,KAAK,CAAC,YAAY,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,KAAK,CAAC,YAAY,EAAE,CAAC;KACjF,CAAC;AACH,CAAC;AAED;;;;;;GAMG;AACH,MAAM,CAAC,KAAK,UAAU,iBAAiB,CAAC,OAAwB;IAC/D,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,GAAG,EAAE,UAAU,EAAE,GAAG,OAAO,CAAC;IAC5D,IAAI,UAAU,EAAE,CAAC;QAChB,MAAM,KAAK,GAAG,MAAM,UAAU,CAAC,SAAS,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QAC/D,OAAO,mBAAmB,CAAC,UAAU,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CACnD,YAAY,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CACnE,CAAC;IACH,CAAC;IACD,MAAM,QAAQ,GAAG,gBAAgB,CAAC,UAAU,EAAE,GAAG,CAAC,CAAC;IACnD,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,EAAE,CAAC;IACrC,MAAM,KAAK,GAAG,MAAM,UAAU,CAAC,SAAS,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;IAC5D,MAAM,KAAK,GAAmB,EAAE,CAAC;IACjC,KAAK,MAAM,IAAI,IAAI,QAAQ,EAAE,CAAC;QAC7B,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACnC,MAAM,QAAQ,GAAG,MAAM,UAAU,CAAC,qBAAqB,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC;QACnF,MAAM,UAAU,GAAG,SAAS,EAAE,kBAAkB,KAAK,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC;QACtF,KAAK,CAAC,IAAI,CACT,YAAY,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,CAAC,GAAG,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,EAAE,UAAU,CAAC,CACzG,CAAC;IACH,CAAC;IACD,OAAO,KAAK,CAAC;AACd,CAAC;AAED,SAAS,UAAU,CAClB,KAAsB,EACtB,OAAwC,EACxC,GAAW,EACX,kBAA2B;IAE3B,MAAM,YAAY,GAAG,CAAC,OAAO,EAAE,YAAY,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;IACtD,MAAM,IAAI,GAAG;QACZ,YAAY;QACZ,GAAG,CAAC,kBAAkB,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,kBAAkB,EAAE,CAAC;QACnE,GAAG,CAAC,OAAO,EAAE,aAAa,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,OAAO,CAAC,aAAa,EAAE,CAAC;KACzF,CAAC;IACF,IAAI,KAAK,CAAC,MAAM,KAAK,YAAY,EAAE,CAAC;QACnC,OAAO,EAAE,GAAG,IAAI,EAAE,YAAY,EAAE,GAAG,GAAG,KAAK,CAAC,UAAU,EAAE,WAAW,EAAE,YAAY,EAAE,CAAC;IACrF,CAAC;IACD,OAAO,EAAE,GAAG,IAAI,EAAE,WAAW,EAAE,KAAK,CAAC,MAAM,EAAE,CAAC;AAC/C,CAAC;AAYD,SAAS,cAAc,CAAC,KAA4B;IACnD,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO;QAAE,OAAO,SAAS,CAAC;IAC7C,MAAM,OAAO,GAAG,KAAK,CAAC,KAAK,CAAC,YAAY,IAAI,uBAAuB,CAAC;IACpE,OAAO,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC;AAC3B,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,4BAA4B,CAC3C,OAAkC;IAElC,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,GAAG,OAAO,CAAC;IACxC,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,gBAAgB,CAAC;IAClD,MAAM,WAAW,GAAG,OAAO,CAAC,WAAW,IAAI,UAAU,EAAE,CAAC;IACxD,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC;IAEpC,OAAO,qBAAqB,CAAsC;QACjE,SAAS,EAAE,GAAG,EAAE,CAAC,iBAAiB,CAAC,OAAO,CAAC;QAC3C,MAAM,EAAE,CAAC,UAAU,EAAE,EAAE;YACtB,MAAM,OAAO,GAAG,eAAe,CAAC,WAAW,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC;YACjE,MAAM,MAAM,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;YAC1B,IAAI,CAAC,MAAM;gBAAE,MAAM,IAAI,KAAK,CAAC,0DAA0D,CAAC,CAAC;YACzF,OAAO,MAAM,CAAC;QACf,CAAC;QACD,UAAU;QACV,iBAAiB,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,OAAO;QACpD,cAAc;QACd,QAAQ,EAAE,yBAAyB;QACnC,YAAY,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE;YACnC,MAAM,QAAQ,GACb,IAAI,CAAC,IAAI,KAAK,KAAK,IAAI,IAAI,CAAC,UAAU,KAAK,SAAS,IAAI,IAAI,CAAC,MAAM,KAAK,SAAS;gBAChF,CAAC,CAAC,MAAM,OAAO,CAAC,UAAU,CAAC,qBAAqB,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC;gBAC9E,CAAC,CAAC,SAAS,CAAC;YACd,MAAM,OAAO,CAAC,UAAU,CAAC,eAAe,CAAC,OAAO,CAAC,UAAU,EAAE,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,OAAO,EAAE,EAAE,CAC9F,UAAU,CAAC,KAAK,EAAE,OAAO,EAAE,GAAG,EAAE,EAAE,QAAQ,CAAC,CAC3C,CAAC;QACH,CAAC;QACD,GAAG;KACH,CAAC,CAAC;AACJ,CAAC","sourcesContent":["import { createHash, randomUUID } from \"node:crypto\";\nimport type { AssistantMessageEvent, Credential } from \"@earendil-works/pi-ai\";\nimport { rendezvousOrder, type SlotHasher } from \"@earendil-works/pi-ai/auth/pool/select\";\nimport { listSlots as listCredentialSlots } from \"@earendil-works/pi-ai/auth/pool/slots\";\nimport { type CredentialBlock, classifyCredentialFailure } from \"./classify.ts\";\nimport { discoverEnvSlots } from \"./env-slots.ts\";\nimport { type RunSlot, runCredentialFailover } from \"./failover.ts\";\nimport type { CredentialSlotRepository, CredentialSlotState } from \"./state-store.ts\";\n\n/** The exact hash the claude-sdk-oauth affinity oracle uses, so pools never remap. */\nexport const sha256SlotHasher: SlotHasher = (input) => createHash(\"sha256\").update(input).digest().readBigUInt64BE(0);\n\nexport type RotationLane = \"stored\" | \"env\";\n\nexport type RotationSlot = RunSlot & {\n\tlane: RotationLane;\n\t/** Env-lane key material for the attempt; never serialized or persisted. */\n\tenvKey?: string;\n\tenvVarName?: string;\n};\n\nexport type RotationSources = {\n\tproviderId: string;\n\tcredential: Credential | undefined;\n\tenv: (name: string) => string | undefined;\n\trepository: CredentialSlotRepository;\n\tnow?: () => number;\n};\n\nfunction overlayState(slot: RotationSlot, state: CredentialSlotState | undefined): RotationSlot {\n\tif (!state) return slot;\n\treturn {\n\t\t...slot,\n\t\t...(state.blockedUntil === undefined ? {} : { blockedUntil: state.blockedUntil }),\n\t\t...(state.blockReason === undefined ? {} : { blockReason: state.blockReason }),\n\t\t...(state.failureCount === undefined ? {} : { failureCount: state.failureCount }),\n\t};\n}\n\n/**\n * Lists the provider's rotation slots with sidecar health overlaid. Stored\n * credentials own the lane when present; env slots participate only when\n * nothing is stored, preserving today's resolution precedence. An env slot's\n * persisted health applies only while its HMAC revision still matches the\n * current env value, so rotating a key in place clears its own stale block.\n */\nexport async function listRotationSlots(sources: RotationSources): Promise<RotationSlot[]> {\n\tconst { providerId, credential, env, repository } = sources;\n\tif (credential) {\n\t\tconst state = await repository.listSlots(providerId, \"stored\");\n\t\treturn listCredentialSlots(credential).map((slot) =>\n\t\t\toverlayState({ name: slot.name, lane: \"stored\" }, state[slot.name]),\n\t\t);\n\t}\n\tconst envSlots = discoverEnvSlots(providerId, env);\n\tif (envSlots.length === 0) return [];\n\tconst state = await repository.listSlots(providerId, \"env\");\n\tconst slots: RotationSlot[] = [];\n\tfor (const slot of envSlots) {\n\t\tconst persisted = state[slot.name];\n\t\tconst revision = await repository.envCredentialRevision(slot.envVarName, slot.key);\n\t\tconst applicable = persisted?.credentialRevision === revision ? persisted : undefined;\n\t\tslots.push(\n\t\t\toverlayState({ name: slot.name, lane: \"env\", envKey: slot.key, envVarName: slot.envVarName }, applicable),\n\t\t);\n\t}\n\treturn slots;\n}\n\nfunction blockPatch(\n\tblock: CredentialBlock,\n\tcurrent: CredentialSlotState | undefined,\n\tnow: number,\n\tcredentialRevision?: string,\n): Omit<CredentialSlotState, \"stateVersion\"> {\n\tconst failureCount = (current?.failureCount ?? 0) + 1;\n\tconst base = {\n\t\tfailureCount,\n\t\t...(credentialRevision === undefined ? {} : { credentialRevision }),\n\t\t...(current?.lastSuccessAt === undefined ? {} : { lastSuccessAt: current.lastSuccessAt }),\n\t};\n\tif (block.reason === \"rate_limit\") {\n\t\treturn { ...base, blockedUntil: now + block.cooldownMs, blockReason: \"rate_limit\" };\n\t}\n\treturn { ...base, blockReason: block.reason };\n}\n\nexport type CredentialRotationOptions = {\n\tsources: RotationSources;\n\t/** Stable session key keeps a session on its slot; absent, each request distributes. */\n\taffinityKey?: string;\n\thasher?: SlotHasher;\n\trunAttempt: (\n\t\tslot: RotationSlot,\n\t) => AsyncIterable<AssistantMessageEvent> | Promise<AsyncIterable<AssistantMessageEvent>>;\n};\n\nfunction errorFromEvent(event: AssistantMessageEvent): unknown {\n\tif (event.type !== \"error\") return undefined;\n\tconst message = event.error.errorMessage ?? \"provider stream error\";\n\treturn new Error(message);\n}\n\n/**\n * In-lane credential rotation for one provider request. Selection follows the\n * HRW order for the affinity key; only the `start` bookkeeping event counts as\n * pre-commit, so any delta bars silent rotation (default-DENY) and failures\n * after output carry the turn-retry suppression marker.\n */\nexport function streamWithCredentialRotation(\n\toptions: CredentialRotationOptions,\n): AsyncGenerator<AssistantMessageEvent> {\n\tconst { sources, runAttempt } = options;\n\tconst hasher = options.hasher ?? sha256SlotHasher;\n\tconst affinityKey = options.affinityKey ?? randomUUID();\n\tconst now = sources.now ?? Date.now;\n\n\treturn runCredentialFailover<AssistantMessageEvent, RotationSlot>({\n\t\tlistSlots: () => listRotationSlots(sources),\n\t\tselect: (candidates) => {\n\t\t\tconst ordered = rendezvousOrder(affinityKey, candidates, hasher);\n\t\t\tconst winner = ordered[0];\n\t\t\tif (!winner) throw new Error(\"credential rotation selected from an empty candidate set\");\n\t\t\treturn winner;\n\t\t},\n\t\trunAttempt,\n\t\tisCommittedOutput: (event) => event.type !== \"start\",\n\t\terrorFromEvent,\n\t\tclassify: classifyCredentialFailure,\n\t\tpersistBlock: async (slot, block) => {\n\t\t\tconst revision =\n\t\t\t\tslot.lane === \"env\" && slot.envVarName !== undefined && slot.envKey !== undefined\n\t\t\t\t\t? await sources.repository.envCredentialRevision(slot.envVarName, slot.envKey)\n\t\t\t\t\t: undefined;\n\t\t\tawait sources.repository.mutateSlotState(sources.providerId, slot.lane, slot.name, (current) =>\n\t\t\t\tblockPatch(block, current, now(), revision),\n\t\t\t);\n\t\t},\n\t\tnow,\n\t});\n}\n"]}
+{"version":3,"file":"rotation-stream.js","sourceRoot":"","sources":["../../../src/core/credential-pool/rotation-stream.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAErD,OAAO,EAAE,eAAe,EAAmB,MAAM,wCAAwC,CAAC;AAC1F,OAAO,EAAE,SAAS,IAAI,mBAAmB,EAAyB,MAAM,uCAAuC,CAAC;AAChH,OAAO,EAAE,kBAAkB,EAAE,MAAM,4BAA4B,CAAC;AAChE,OAAO,EAAwB,yBAAyB,EAAE,MAAM,eAAe,CAAC;AAChF,OAAO,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAClD,OAAO,EAAgB,qBAAqB,EAAE,MAAM,eAAe,CAAC;AACpE,OAAO,EAAE,oBAAoB,EAA2D,MAAM,kBAAkB,CAAC;AAEjH,sFAAsF;AACtF,MAAM,CAAC,MAAM,gBAAgB,GAAe,CAAC,KAAK,EAAE,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,MAAM,EAAE,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;AAyBtH,SAAS,YAAY,CAAC,IAAkB,EAAE,KAAsC;IAC/E,IAAI,CAAC,KAAK;QAAE,OAAO,IAAI,CAAC;IACxB,OAAO;QACN,GAAG,IAAI;QACP,GAAG,CAAC,KAAK,CAAC,YAAY,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,KAAK,CAAC,YAAY,EAAE,CAAC;QACjF,GAAG,CAAC,KAAK,CAAC,WAAW,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,KAAK,CAAC,WAAW,EAAE,CAAC;QAC9E,GAAG,CAAC,KAAK,CAAC,YAAY,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,KAAK,CAAC,YAAY,EAAE,CAAC;QACjF,GAAG,CAAC,KAAK,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,CAAC;KAC5D,CAAC;AACH,CAAC;AAED;;;;;;GAMG;AACH,MAAM,CAAC,KAAK,UAAU,iBAAiB,CACtC,OAAwB,EACxB,UAAuC,EAAE;IAEzC,MAAM,aAAa,GAAG,OAAO,CAAC,aAAa,KAAK,KAAK,CAAC;IACtD,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,GAAG,EAAE,UAAU,EAAE,GAAG,OAAO,CAAC;IAC5D,MAAM,WAAW,GAAG,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE;QACvF,MAAM,UAAU,GAAG,GAAG,CAAC,GAAG,IAAI,eAAe,IAAI,EAAE,CAAC;QACpD,MAAM,GAAG,GACR,GAAG,CAAC,GAAG,KAAK,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,kBAAkB,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAChH,IAAI,CAAC,GAAG;YAAE,OAAO,EAAE,CAAC;QACpB,OAAO,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,GAAG,EAAE,MAAM,EAAE,KAAc,EAAE,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;IACH,IAAI,UAAU,EAAE,CAAC;QAChB,MAAM,KAAK,GAAG,MAAM,UAAU,CAAC,SAAS,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QAC/D,MAAM,KAAK,GAAmB,EAAE,CAAC;QACjC,KAAK,MAAM,IAAI,IAAI,mBAAmB,CAAC,UAAU,CAAC,EAAE,CAAC;YACpD,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACjC,IACC,aAAa;gBACb,OAAO,EAAE,YAAY,KAAK,SAAS;gBACnC,OAAO,CAAC,YAAY,IAAI,CAAC,OAAO,CAAC,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC,EAAE,EAClD,CAAC;gBACF,MAAM,KAAK,GAAG,MAAM,oBAAoB,CAAC,UAAU,EAAE,UAAU,EAAE,QAAQ,EAAE,IAAI,CAAC,IAAI,EAAE;oBACrF,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC,EAAE;iBAChC,CAAC,CAAC;gBACH,IAAI,CAAC,KAAK;oBAAE,SAAS;gBACrB,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,SAAS,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;gBAChE,KAAK,CAAC,IAAI,CACT,YAAY,CACX;oBACC,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,IAAI,EAAE,QAAQ;oBACd,MAAM,EAAG,UAA+B,CAAC,MAAM,KAAK,IAAI,CAAC,IAAI;iBAC7D,EACD,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CACjB,CACD,CAAC;gBACF,SAAS;YACV,CAAC;YACD,KAAK,CAAC,IAAI,CACT,YAAY,CACX;gBACC,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,IAAI,EAAE,QAAQ;gBACd,MAAM,EAAG,UAA+B,CAAC,MAAM,KAAK,IAAI,CAAC,IAAI;aAC7D,EACD,OAAO,CACP,CACD,CAAC;QACH,CAAC;QACD,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,KAAK,CAAC;QAC3C,MAAM,YAAY,GAAoB;YACrC,GAAG,OAAO;YACV,UAAU,EAAE,SAAS;YACrB,MAAM,EAAE,EAAE,GAAG,OAAO,CAAC,MAAM,EAAE,KAAK,EAAE,EAAE,EAAE;SACxC,CAAC;QACF,MAAM,UAAU,GAAG,MAAM,oBAAoB,CAAC,YAAY,EAAE,WAAW,EAAE,aAAa,CAAC,CAAC;QACxF,OAAO,CAAC,GAAG,KAAK,EAAE,GAAG,UAAU,CAAC,CAAC;IAClC,CAAC;IACD,MAAM,QAAQ,GAAG,CAAC,GAAG,gBAAgB,CAAC,UAAU,EAAE,GAAG,CAAC,EAAE,GAAG,WAAW,CAAC,CAAC;IACxE,OAAO,oBAAoB,CAAC,OAAO,EAAE,QAAQ,EAAE,aAAa,CAAC,CAAC;AAC/D,CAAC;AAED,KAAK,UAAU,oBAAoB,CAClC,OAAwB,EACxB,QAAsE,EACtE,aAAa,GAAG,IAAI;IAEpB,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,EAAE,CAAC;IACrC,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,GAAG,OAAO,CAAC;IAC3C,MAAM,KAAK,GAAG,MAAM,UAAU,CAAC,SAAS,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;IAC5D,MAAM,KAAK,GAAmB,EAAE,CAAC;IACjC,KAAK,MAAM,IAAI,IAAI,QAAQ,EAAE,CAAC;QAC7B,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACnC,MAAM,QAAQ,GAAG,MAAM,UAAU,CAAC,qBAAqB,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC;QACnF,IAAI,UAAU,GAAG,SAAS,EAAE,kBAAkB,KAAK,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC;QACpF,IACC,aAAa;YACb,UAAU,EAAE,YAAY,KAAK,SAAS;YACtC,UAAU,CAAC,YAAY,IAAI,CAAC,OAAO,CAAC,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC,EAAE,EACrD,CAAC;YACF,MAAM,KAAK,GAAG,MAAM,oBAAoB,CAAC,UAAU,EAAE,UAAU,EAAE,KAAK,EAAE,IAAI,CAAC,IAAI,EAAE;gBAClF,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC,EAAE;aAChC,CAAC,CAAC;YACH,IAAI,CAAC,KAAK;gBAAE,SAAS;YACrB,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,SAAS,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;YAC7D,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAChC,CAAC;QACD,KAAK,CAAC,IAAI,CACT,YAAY,CACX;YACC,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,IAAI,EAAE,KAAK;YACX,MAAM,EAAE,IAAI,CAAC,GAAG;YAChB,UAAU,EAAE,IAAI,CAAC,UAAU;SAC3B,EACD,UAAU,CACV,CACD,CAAC;IACH,CAAC;IACD,OAAO,KAAK,CAAC;AACd,CAAC;AAED,SAAS,UAAU,CAClB,KAAsB,EACtB,OAAwC,EACxC,GAAW,EACX,kBAA2B,EAC3B,MAA4D;IAE5D,MAAM,YAAY,GAAG,CAAC,OAAO,EAAE,YAAY,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;IACtD,MAAM,IAAI,GAAG;QACZ,YAAY;QACZ,GAAG,CAAC,kBAAkB,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,kBAAkB,EAAE,CAAC;QACnE,GAAG,CAAC,OAAO,EAAE,aAAa,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,OAAO,CAAC,aAAa,EAAE,CAAC;KACzF,CAAC;IACF,IAAI,KAAK,CAAC,MAAM,KAAK,YAAY,EAAE,CAAC;QACnC,OAAO;YACN,GAAG,IAAI;YACP,YAAY,EAAE,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,aAAa,IAAI,KAAK,CAAC,UAAU,EAAE,KAAK,CAAC,UAAU,CAAC;YACzF,WAAW,EAAE,YAAY;SACzB,CAAC;IACH,CAAC;IACD,OAAO,EAAE,GAAG,IAAI,EAAE,WAAW,EAAE,KAAK,CAAC,MAAM,EAAE,CAAC;AAC/C,CAAC;AAYD,SAAS,cAAc,CAAC,KAA4B;IACnD,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO;QAAE,OAAO,SAAS,CAAC;IAC7C,MAAM,OAAO,GAAG,KAAK,CAAC,KAAK,CAAC,YAAY,IAAI,uBAAuB,CAAC;IACpE,OAAO,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC;AAC3B,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,4BAA4B,CAC3C,OAAkC;IAElC,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,GAAG,OAAO,CAAC;IACxC,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,gBAAgB,CAAC;IAClD,MAAM,WAAW,GAAG,OAAO,CAAC,WAAW,IAAI,UAAU,EAAE,CAAC;IACxD,MAAM,WAAW,GAAG,OAAO,CAAC,MAAM,EAAE,QAAQ,KAAK,KAAK,CAAC;IACvD,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC;IAEpC,OAAO,qBAAqB,CAAsC;QACjE,SAAS,EAAE,GAAG,EAAE,CAAC,iBAAiB,CAAC,OAAO,CAAC;QAC3C,MAAM,EAAE,CAAC,UAAU,EAAE,EAAE;YACtB,MAAM,MAAM,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,CAAC,MAAM,KAAK,IAAI,CAAC,CAAC;YACzE,IAAI,MAAM;gBAAE,OAAO,MAAM,CAAC;YAC1B,MAAM,OAAO,GAAG,WAAW,CAAC,CAAC,CAAC,eAAe,CAAC,WAAW,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC;YAE5F,MAAM,MAAM,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;YAC1B,IAAI,CAAC,MAAM;gBAAE,MAAM,IAAI,KAAK,CAAC,0DAA0D,CAAC,CAAC;YACzF,OAAO,MAAM,CAAC;QACf,CAAC;QACD,UAAU;QACV,iBAAiB,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,OAAO;QACpD,cAAc;QACd,QAAQ,EAAE,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE,CAC5B,yBAAyB,CAAC,KAAK,EAAE;YAChC,GAAG,OAAO;YACV,cAAc,EAAE,OAAO,CAAC,MAAM,EAAE,cAAc;YAC9C,aAAa,EAAE,OAAO,CAAC,MAAM,EAAE,aAAa;SAC5C,CAAC;QACH,SAAS,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YACzB,MAAM,OAAO,CAAC,UAAU,CAAC,eAAe,CAAC,OAAO,CAAC,UAAU,EAAE,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,OAAO,EAAE,EAAE,CAC9F,OAAO;gBACN,CAAC,CAAC;oBACA,GAAG,OAAO;oBACV,aAAa,EAAE,GAAG,EAAE;oBACpB,KAAK,EAAE,SAAS;oBAChB,YAAY,EAAE,SAAS;oBACvB,WAAW,EAAE,SAAS;iBACtB;gBACF,CAAC,CAAC,SAAS,CACZ,CAAC;QACH,CAAC;QACD,YAAY,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE;YACnC,MAAM,QAAQ,GACb,IAAI,CAAC,IAAI,KAAK,KAAK,IAAI,IAAI,CAAC,UAAU,KAAK,SAAS,IAAI,IAAI,CAAC,MAAM,KAAK,SAAS;gBAChF,CAAC,CAAC,MAAM,OAAO,CAAC,UAAU,CAAC,qBAAqB,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC;gBAC9E,CAAC,CAAC,SAAS,CAAC;YACd,MAAM,OAAO,CAAC,UAAU,CAAC,eAAe,CAAC,OAAO,CAAC,UAAU,EAAE,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,OAAO,EAAE,EAAE,CAC9F,UAAU,CAAC,KAAK,EAAE,OAAO,EAAE,GAAG,EAAE,EAAE,QAAQ,EAAE,OAAO,CAAC,MAAM,CAAC,CAC3D,CAAC;QACH,CAAC;QACD,GAAG;KACH,CAAC,CAAC;AACJ,CAAC","sourcesContent":["import { createHash, randomUUID } from \"node:crypto\";\nimport type { AssistantMessageEvent, Credential } from \"@earendil-works/pi-ai\";\nimport { rendezvousOrder, type SlotHasher } from \"@earendil-works/pi-ai/auth/pool/select\";\nimport { listSlots as listCredentialSlots, type PooledCredential } from \"@earendil-works/pi-ai/auth/pool/slots\";\nimport { resolveConfigValue } from \"../resolve-config-value.ts\";\nimport { type CredentialBlock, classifyCredentialFailure } from \"./classify.ts\";\nimport { discoverEnvSlots } from \"./env-slots.ts\";\nimport { type RunSlot, runCredentialFailover } from \"./failover.ts\";\nimport { acquireHalfOpenLease, type CredentialSlotRepository, type CredentialSlotState } from \"./state-store.ts\";\n\n/** The exact hash the claude-sdk-oauth affinity oracle uses, so pools never remap. */\nexport const sha256SlotHasher: SlotHasher = (input) => createHash(\"sha256\").update(input).digest().readBigUInt64BE(0);\n\nexport type RotationLane = \"stored\" | \"env\";\n\nexport type RotationSlot = RunSlot & {\n\tlane: RotationLane;\n\t/** Env-lane key material for the attempt; never serialized or persisted. */\n\tenvKey?: string;\n\tenvVarName?: string;\n};\n\nexport type RotationSources = {\n\tproviderId: string;\n\tcredential: Credential | undefined;\n\tenv: (name: string) => string | undefined;\n\trepository: CredentialSlotRepository;\n\tpolicy?: {\n\t\taffinity?: boolean;\n\t\tcooldownBaseMs?: number;\n\t\tcooldownCapMs?: number;\n\t\tslots?: Record<string, { env?: string; value?: string }>;\n\t};\n\tnow?: () => number;\n};\n\nfunction overlayState(slot: RotationSlot, state: CredentialSlotState | undefined): RotationSlot {\n\tif (!state) return slot;\n\treturn {\n\t\t...slot,\n\t\t...(state.blockedUntil === undefined ? {} : { blockedUntil: state.blockedUntil }),\n\t\t...(state.blockReason === undefined ? {} : { blockReason: state.blockReason }),\n\t\t...(state.failureCount === undefined ? {} : { failureCount: state.failureCount }),\n\t\t...(state.lease === undefined ? {} : { lease: state.lease }),\n\t};\n}\n\n/**\n * Lists the provider's rotation slots with sidecar health overlaid. Stored\n * credentials own the lane when present; env slots participate only when\n * nothing is stored, preserving today's resolution precedence. An env slot's\n * persisted health applies only while its HMAC revision still matches the\n * current env value, so rotating a key in place clears its own stale block.\n */\nexport async function listRotationSlots(\n\tsources: RotationSources,\n\toptions: { acquireLeases?: boolean } = {},\n): Promise<RotationSlot[]> {\n\tconst acquireLeases = options.acquireLeases !== false;\n\tconst { providerId, credential, env, repository } = sources;\n\tconst policySlots = Object.entries(sources.policy?.slots ?? {}).flatMap(([name, ref]) => {\n\t\tconst envVarName = ref.env ?? `models.json:${name}`;\n\t\tconst key =\n\t\t\tref.env !== undefined ? env(ref.env) : ref.value !== undefined ? resolveConfigValue(ref.value, {}) : undefined;\n\t\tif (!key) return [];\n\t\treturn [{ name, envVarName, key, source: \"env\" as const }];\n\t});\n\tif (credential) {\n\t\tconst state = await repository.listSlots(providerId, \"stored\");\n\t\tconst slots: RotationSlot[] = [];\n\t\tfor (const slot of listCredentialSlots(credential)) {\n\t\t\tconst current = state[slot.name];\n\t\t\tif (\n\t\t\t\tacquireLeases &&\n\t\t\t\tcurrent?.blockedUntil !== undefined &&\n\t\t\t\tcurrent.blockedUntil <= (sources.now ?? Date.now)()\n\t\t\t) {\n\t\t\t\tconst lease = await acquireHalfOpenLease(repository, providerId, \"stored\", slot.name, {\n\t\t\t\t\tnow: (sources.now ?? Date.now)(),\n\t\t\t\t});\n\t\t\t\tif (!lease) continue;\n\t\t\t\tconst leased = await repository.listSlots(providerId, \"stored\");\n\t\t\t\tslots.push(\n\t\t\t\t\toverlayState(\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tname: slot.name,\n\t\t\t\t\t\t\tlane: \"stored\",\n\t\t\t\t\t\t\tpinned: (credential as PooledCredential).pinned === slot.name,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tleased[slot.name],\n\t\t\t\t\t),\n\t\t\t\t);\n\t\t\t\tcontinue;\n\t\t\t}\n\t\t\tslots.push(\n\t\t\t\toverlayState(\n\t\t\t\t\t{\n\t\t\t\t\t\tname: slot.name,\n\t\t\t\t\t\tlane: \"stored\",\n\t\t\t\t\t\tpinned: (credential as PooledCredential).pinned === slot.name,\n\t\t\t\t\t},\n\t\t\t\t\tcurrent,\n\t\t\t\t),\n\t\t\t);\n\t\t}\n\t\tif (policySlots.length === 0) return slots;\n\t\tconst namedSources: RotationSources = {\n\t\t\t...sources,\n\t\t\tcredential: undefined,\n\t\t\tpolicy: { ...sources.policy, slots: {} },\n\t\t};\n\t\tconst namedSlots = await listEnvRotationSlots(namedSources, policySlots, acquireLeases);\n\t\treturn [...slots, ...namedSlots];\n\t}\n\tconst envSlots = [...discoverEnvSlots(providerId, env), ...policySlots];\n\treturn listEnvRotationSlots(sources, envSlots, acquireLeases);\n}\n\nasync function listEnvRotationSlots(\n\tsources: RotationSources,\n\tenvSlots: readonly { name: string; envVarName: string; key: string }[],\n\tacquireLeases = true,\n): Promise<RotationSlot[]> {\n\tif (envSlots.length === 0) return [];\n\tconst { providerId, repository } = sources;\n\tconst state = await repository.listSlots(providerId, \"env\");\n\tconst slots: RotationSlot[] = [];\n\tfor (const slot of envSlots) {\n\t\tconst persisted = state[slot.name];\n\t\tconst revision = await repository.envCredentialRevision(slot.envVarName, slot.key);\n\t\tlet applicable = persisted?.credentialRevision === revision ? persisted : undefined;\n\t\tif (\n\t\t\tacquireLeases &&\n\t\t\tapplicable?.blockedUntil !== undefined &&\n\t\t\tapplicable.blockedUntil <= (sources.now ?? Date.now)()\n\t\t) {\n\t\t\tconst lease = await acquireHalfOpenLease(repository, providerId, \"env\", slot.name, {\n\t\t\t\tnow: (sources.now ?? Date.now)(),\n\t\t\t});\n\t\t\tif (!lease) continue;\n\t\t\tconst leased = await repository.listSlots(providerId, \"env\");\n\t\t\tapplicable = leased[slot.name];\n\t\t}\n\t\tslots.push(\n\t\t\toverlayState(\n\t\t\t\t{\n\t\t\t\t\tname: slot.name,\n\t\t\t\t\tlane: \"env\",\n\t\t\t\t\tenvKey: slot.key,\n\t\t\t\t\tenvVarName: slot.envVarName,\n\t\t\t\t},\n\t\t\t\tapplicable,\n\t\t\t),\n\t\t);\n\t}\n\treturn slots;\n}\n\nfunction blockPatch(\n\tblock: CredentialBlock,\n\tcurrent: CredentialSlotState | undefined,\n\tnow: number,\n\tcredentialRevision?: string,\n\tpolicy?: { cooldownBaseMs?: number; cooldownCapMs?: number },\n): Omit<CredentialSlotState, \"stateVersion\"> {\n\tconst failureCount = (current?.failureCount ?? 0) + 1;\n\tconst base = {\n\t\tfailureCount,\n\t\t...(credentialRevision === undefined ? {} : { credentialRevision }),\n\t\t...(current?.lastSuccessAt === undefined ? {} : { lastSuccessAt: current.lastSuccessAt }),\n\t};\n\tif (block.reason === \"rate_limit\") {\n\t\treturn {\n\t\t\t...base,\n\t\t\tblockedUntil: now + Math.min(policy?.cooldownCapMs ?? block.cooldownMs, block.cooldownMs),\n\t\t\tblockReason: \"rate_limit\",\n\t\t};\n\t}\n\treturn { ...base, blockReason: block.reason };\n}\n\nexport type CredentialRotationOptions = {\n\tsources: RotationSources;\n\t/** Stable session key keeps a session on its slot; absent, each request distributes. */\n\taffinityKey?: string;\n\thasher?: SlotHasher;\n\trunAttempt: (\n\t\tslot: RotationSlot,\n\t) => AsyncIterable<AssistantMessageEvent> | Promise<AsyncIterable<AssistantMessageEvent>>;\n};\n\nfunction errorFromEvent(event: AssistantMessageEvent): unknown {\n\tif (event.type !== \"error\") return undefined;\n\tconst message = event.error.errorMessage ?? \"provider stream error\";\n\treturn new Error(message);\n}\n\n/**\n * In-lane credential rotation for one provider request. Selection follows the\n * HRW order for the affinity key; only the `start` bookkeeping event counts as\n * pre-commit, so any delta bars silent rotation (default-DENY) and failures\n * after output carry the turn-retry suppression marker.\n */\nexport function streamWithCredentialRotation(\n\toptions: CredentialRotationOptions,\n): AsyncGenerator<AssistantMessageEvent> {\n\tconst { sources, runAttempt } = options;\n\tconst hasher = options.hasher ?? sha256SlotHasher;\n\tconst affinityKey = options.affinityKey ?? randomUUID();\n\tconst useAffinity = sources.policy?.affinity !== false;\n\tconst now = sources.now ?? Date.now;\n\n\treturn runCredentialFailover<AssistantMessageEvent, RotationSlot>({\n\t\tlistSlots: () => listRotationSlots(sources),\n\t\tselect: (candidates) => {\n\t\t\tconst pinned = candidates.find((candidate) => candidate.pinned === true);\n\t\t\tif (pinned) return pinned;\n\t\t\tconst ordered = useAffinity ? rendezvousOrder(affinityKey, candidates, hasher) : candidates;\n\n\t\t\tconst winner = ordered[0];\n\t\t\tif (!winner) throw new Error(\"credential rotation selected from an empty candidate set\");\n\t\t\treturn winner;\n\t\t},\n\t\trunAttempt,\n\t\tisCommittedOutput: (event) => event.type !== \"start\",\n\t\terrorFromEvent,\n\t\tclassify: (error, context) =>\n\t\t\tclassifyCredentialFailure(error, {\n\t\t\t\t...context,\n\t\t\t\tcooldownBaseMs: sources.policy?.cooldownBaseMs,\n\t\t\t\tcooldownCapMs: sources.policy?.cooldownCapMs,\n\t\t\t}),\n\t\tonSuccess: async (slot) => {\n\t\t\tawait sources.repository.mutateSlotState(sources.providerId, slot.lane, slot.name, (current) =>\n\t\t\t\tcurrent\n\t\t\t\t\t? {\n\t\t\t\t\t\t\t...current,\n\t\t\t\t\t\t\tlastSuccessAt: now(),\n\t\t\t\t\t\t\tlease: undefined,\n\t\t\t\t\t\t\tblockedUntil: undefined,\n\t\t\t\t\t\t\tblockReason: undefined,\n\t\t\t\t\t\t}\n\t\t\t\t\t: undefined,\n\t\t\t);\n\t\t},\n\t\tpersistBlock: async (slot, block) => {\n\t\t\tconst revision =\n\t\t\t\tslot.lane === \"env\" && slot.envVarName !== undefined && slot.envKey !== undefined\n\t\t\t\t\t? await sources.repository.envCredentialRevision(slot.envVarName, slot.envKey)\n\t\t\t\t\t: undefined;\n\t\t\tawait sources.repository.mutateSlotState(sources.providerId, slot.lane, slot.name, (current) =>\n\t\t\t\tblockPatch(block, current, now(), revision, sources.policy),\n\t\t\t);\n\t\t},\n\t\tnow,\n\t});\n}\n"]}

package/dist/core/extensions/builtin/compaction/context-pipeline.d.ts

@@ -0,0 +1,17 @@
+import type { ContextEvent, ExtensionContext } from "../../types.ts";
+import { type EmergencyPruneLatch } from "./speculative.ts";
+export declare function buildCompactionContext(input: {
+    event: ContextEvent;
+    ctx: ExtensionContext;
+    contextWindow: number;
+    /**
+     * Context window minus the model's output reserve. The emergency prune budgets against the
+     * space a request can actually occupy, so it must not see the full window.
+     */
+    promptContextWindow: number;
+    toolAdmissionEnabled: boolean;
+    breakerFallback: boolean;
+    laneOwnsCompaction: boolean;
+    emergencyPruneLatch: EmergencyPruneLatch;
+}): import("@earendil-works/pi-ai").Message[];
+//# sourceMappingURL=context-pipeline.d.ts.map

package/dist/core/extensions/builtin/compaction/context-pipeline.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"context-pipeline.d.ts","sourceRoot":"","sources":["../../../../../src/core/extensions/builtin/compaction/context-pipeline.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,YAAY,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAUrE,OAAO,EAAE,KAAK,mBAAmB,EAA2B,MAAM,kBAAkB,CAAC;AAErF,wBAAgB,sBAAsB,CAAC,KAAK,EAAE;IAC7C,KAAK,EAAE,YAAY,CAAC;IACpB,GAAG,EAAE,gBAAgB,CAAC;IACtB,aAAa,EAAE,MAAM,CAAC;IACtB;;;OAGG;IACH,mBAAmB,EAAE,MAAM,CAAC;IAC5B,oBAAoB,EAAE,OAAO,CAAC;IAC9B,eAAe,EAAE,OAAO,CAAC;IACzB,kBAAkB,EAAE,OAAO,CAAC;IAC5B,mBAAmB,EAAE,mBAAmB,CAAC;CACzC,6CAwBA"}

package/dist/core/extensions/builtin/compaction/context-pipeline.js

@@ -0,0 +1,28 @@
+import { convertToLlm } from "../../../messages.js";
+import { BUILTIN_CONTEXT_REDUCTION_OPTIONS, reduceContextMessages, shouldApplyContextReduction, } from "./context-reduction.js";
+import { markOpenAiRemoteReplayBoundary } from "./openai-remote.js";
+import { isOpenAiRemoteCompactionModel } from "./openai-remote-model.js";
+import { admitContextToolResults } from "./orchestration.js";
+import { repairOrphanedToolResults } from "./repair-tool-pairs.js";
+import { hardLimitEmergencyPrune } from "./speculative.js";
+export function buildCompactionContext(input) {
+    const admittedMessages = admitContextToolResults(input.event.messages, input.contextWindow, input.toolAdmissionEnabled);
+    const sourceMessages = !input.laneOwnsCompaction &&
+        (input.breakerFallback ||
+            shouldApplyContextReduction({
+                usageTokens: input.ctx.getContextUsage()?.tokens ?? null,
+                contextWindow: input.contextWindow,
+                isProviderNativeCompactionPath: isOpenAiRemoteCompactionModel(input.ctx.model) || input.laneOwnsCompaction,
+            }))
+        ? reduceContextMessages(admittedMessages, BUILTIN_CONTEXT_REDUCTION_OPTIONS).messages
+        : admittedMessages;
+    const emergency = input.laneOwnsCompaction
+        ? { messages: sourceMessages, needsAggressiveCompaction: false }
+        : hardLimitEmergencyPrune(sourceMessages, input.promptContextWindow, input.emergencyPruneLatch);
+    const marked = markOpenAiRemoteReplayBoundary(emergency.messages, {
+        model: input.ctx.model,
+        branchEntries: input.ctx.sessionManager.getBranch(),
+    });
+    return repairOrphanedToolResults(convertToLlm(marked));
+}
+//# sourceMappingURL=context-pipeline.js.map

package/dist/core/extensions/builtin/compaction/context-pipeline.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"context-pipeline.js","sourceRoot":"","sources":["../../../../../src/core/extensions/builtin/compaction/context-pipeline.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAC;AAEpD,OAAO,EACN,iCAAiC,EACjC,qBAAqB,EACrB,2BAA2B,GAC3B,MAAM,wBAAwB,CAAC;AAChC,OAAO,EAAE,8BAA8B,EAAE,MAAM,oBAAoB,CAAC;AACpE,OAAO,EAAE,6BAA6B,EAAE,MAAM,0BAA0B,CAAC;AACzE,OAAO,EAAE,uBAAuB,EAAE,MAAM,oBAAoB,CAAC;AAC7D,OAAO,EAAE,yBAAyB,EAAE,MAAM,wBAAwB,CAAC;AACnE,OAAO,EAA4B,uBAAuB,EAAE,MAAM,kBAAkB,CAAC;AAErF,MAAM,UAAU,sBAAsB,CAAC,KAatC;IACA,MAAM,gBAAgB,GAAG,uBAAuB,CAC/C,KAAK,CAAC,KAAK,CAAC,QAAQ,EACpB,KAAK,CAAC,aAAa,EACnB,KAAK,CAAC,oBAAoB,CAC1B,CAAC;IACF,MAAM,cAAc,GACnB,CAAC,KAAK,CAAC,kBAAkB;QACzB,CAAC,KAAK,CAAC,eAAe;YACrB,2BAA2B,CAAC;gBAC3B,WAAW,EAAE,KAAK,CAAC,GAAG,CAAC,eAAe,EAAE,EAAE,MAAM,IAAI,IAAI;gBACxD,aAAa,EAAE,KAAK,CAAC,aAAa;gBAClC,8BAA8B,EAAE,6BAA6B,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,kBAAkB;aAC1G,CAAC,CAAC;QACH,CAAC,CAAC,qBAAqB,CAAC,gBAAgB,EAAE,iCAAiC,CAAC,CAAC,QAAQ;QACrF,CAAC,CAAC,gBAAgB,CAAC;IACrB,MAAM,SAAS,GAAG,KAAK,CAAC,kBAAkB;QACzC,CAAC,CAAC,EAAE,QAAQ,EAAE,cAAc,EAAE,yBAAyB,EAAE,KAAK,EAAE;QAChE,CAAC,CAAC,uBAAuB,CAAC,cAAc,EAAE,KAAK,CAAC,mBAAmB,EAAE,KAAK,CAAC,mBAAmB,CAAC,CAAC;IACjG,MAAM,MAAM,GAAG,8BAA8B,CAAC,SAAS,CAAC,QAAQ,EAAE;QACjE,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,KAAK;QACtB,aAAa,EAAE,KAAK,CAAC,GAAG,CAAC,cAAc,CAAC,SAAS,EAAE;KACnD,CAAC,CAAC;IACH,OAAO,yBAAyB,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC;AACxD,CAAC","sourcesContent":["import { convertToLlm } from \"../../../messages.ts\";\nimport type { ContextEvent, ExtensionContext } from \"../../types.ts\";\nimport {\n\tBUILTIN_CONTEXT_REDUCTION_OPTIONS,\n\treduceContextMessages,\n\tshouldApplyContextReduction,\n} from \"./context-reduction.ts\";\nimport { markOpenAiRemoteReplayBoundary } from \"./openai-remote.ts\";\nimport { isOpenAiRemoteCompactionModel } from \"./openai-remote-model.ts\";\nimport { admitContextToolResults } from \"./orchestration.ts\";\nimport { repairOrphanedToolResults } from \"./repair-tool-pairs.ts\";\nimport { type EmergencyPruneLatch, hardLimitEmergencyPrune } from \"./speculative.ts\";\n\nexport function buildCompactionContext(input: {\n\tevent: ContextEvent;\n\tctx: ExtensionContext;\n\tcontextWindow: number;\n\t/**\n\t * Context window minus the model's output reserve. The emergency prune budgets against the\n\t * space a request can actually occupy, so it must not see the full window.\n\t */\n\tpromptContextWindow: number;\n\ttoolAdmissionEnabled: boolean;\n\tbreakerFallback: boolean;\n\tlaneOwnsCompaction: boolean;\n\temergencyPruneLatch: EmergencyPruneLatch;\n}) {\n\tconst admittedMessages = admitContextToolResults(\n\t\tinput.event.messages,\n\t\tinput.contextWindow,\n\t\tinput.toolAdmissionEnabled,\n\t);\n\tconst sourceMessages =\n\t\t!input.laneOwnsCompaction &&\n\t\t(input.breakerFallback ||\n\t\t\tshouldApplyContextReduction({\n\t\t\t\tusageTokens: input.ctx.getContextUsage()?.tokens ?? null,\n\t\t\t\tcontextWindow: input.contextWindow,\n\t\t\t\tisProviderNativeCompactionPath: isOpenAiRemoteCompactionModel(input.ctx.model) || input.laneOwnsCompaction,\n\t\t\t}))\n\t\t\t? reduceContextMessages(admittedMessages, BUILTIN_CONTEXT_REDUCTION_OPTIONS).messages\n\t\t\t: admittedMessages;\n\tconst emergency = input.laneOwnsCompaction\n\t\t? { messages: sourceMessages, needsAggressiveCompaction: false }\n\t\t: hardLimitEmergencyPrune(sourceMessages, input.promptContextWindow, input.emergencyPruneLatch);\n\tconst marked = markOpenAiRemoteReplayBoundary(emergency.messages, {\n\t\tmodel: input.ctx.model,\n\t\tbranchEntries: input.ctx.sessionManager.getBranch(),\n\t});\n\treturn repairOrphanedToolResults(convertToLlm(marked));\n}\n"]}

package/dist/core/extensions/builtin/compaction/deterministic-fallback.d.ts

@@ -6,14 +6,20 @@ interface RecoveryMetadata {
     todoSnapshot?: unknown;
     checkpoint?: unknown;
 }
+export type DeterministicFallbackRejectionReason = "missing-preparation-boundary" | "unsafe-retained-content" | "retained-token-budget-exceeded" | "atomic-tool-chain-cut" | "context-reconstruction-failed";
 interface DeterministicFallbackDetails {
     schema: "senpi.compaction.deterministic-fallback.v1";
     origin: "required-compaction-recovery";
     failureKind: RequiredCompactionFallbackFailure;
     taskIntent?: string;
-    retainedSuffix?: "prepared" | "latest-user-turn";
+    retainedSuffix?: "prepared" | "latest-user-turn" | "earlier-safe-boundary";
+}
+export interface DeterministicFallbackDiagnostic {
+    rejectionReason?: DeterministicFallbackRejectionReason;
+    candidatesChecked?: number;
+    budgetExceeded?: boolean;
 }
 export declare function classifyRequiredCompactionFallbackFailure(error: unknown): RequiredCompactionFallbackFailure | undefined;
-export declare function createRequiredCompactionFallback(preparation: CompactionPreparation, contextWindow: number, failureKind: RequiredCompactionFallbackFailure, metadata: RecoveryMetadata, branchEntries?: SessionEntry[]): CompactionResult<DeterministicFallbackDetails> | undefined;
+export declare function createRequiredCompactionFallback(preparation: CompactionPreparation, contextWindow: number, failureKind: RequiredCompactionFallbackFailure, metadata: RecoveryMetadata, branchEntries?: SessionEntry[], diagnostics?: DeterministicFallbackDiagnostic): CompactionResult<DeterministicFallbackDetails> | undefined;
 export {};
 //# sourceMappingURL=deterministic-fallback.d.ts.map

package/dist/core/extensions/builtin/compaction/deterministic-fallback.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"deterministic-fallback.d.ts","sourceRoot":"","sources":["../../../../../src/core/extensions/builtin/compaction/deterministic-fallback.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,qBAAqB,EAAE,KAAK,gBAAgB,EAAkB,MAAM,8BAA8B,CAAC;AAGjH,OAAO,EAA6C,KAAK,YAAY,EAAE,MAAM,6BAA6B,CAAC;AAM3G,MAAM,MAAM,iCAAiC,GAC1C,uBAAuB,GACvB,2BAA2B,GAC3B,kCAAkC,CAAC;AAEtC,UAAU,gBAAgB;IACzB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,YAAY,CAAC,EAAE,OAAO,CAAC;IACvB,UAAU,CAAC,EAAE,OAAO,CAAC;CACrB;AAED,UAAU,4BAA4B;IACrC,MAAM,EAAE,4CAA4C,CAAC;IACrD,MAAM,EAAE,8BAA8B,CAAC;IACvC,WAAW,EAAE,iCAAiC,CAAC;IAC/C,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,cAAc,CAAC,EAAE,UAAU,GAAG,kBAAkB,CAAC;CACjD;AAuED,wBAAgB,yCAAyC,CACxD,KAAK,EAAE,OAAO,GACZ,iCAAiC,GAAG,SAAS,CAW/C;AAED,wBAAgB,gCAAgC,CAC/C,WAAW,EAAE,qBAAqB,EAClC,aAAa,EAAE,MAAM,EACrB,WAAW,EAAE,iCAAiC,EAC9C,QAAQ,EAAE,gBAAgB,EAC1B,aAAa,GAAE,YAAY,EAAO,GAChC,gBAAgB,CAAC,4BAA4B,CAAC,GAAG,SAAS,CA8E5D"}
+{"version":3,"file":"deterministic-fallback.d.ts","sourceRoot":"","sources":["../../../../../src/core/extensions/builtin/compaction/deterministic-fallback.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,qBAAqB,EAAE,KAAK,gBAAgB,EAAkB,MAAM,8BAA8B,CAAC;AAGjH,OAAO,EAIN,KAAK,YAAY,EACjB,MAAM,6BAA6B,CAAC;AAOrC,MAAM,MAAM,iCAAiC,GAC1C,uBAAuB,GACvB,2BAA2B,GAC3B,kCAAkC,CAAC;AAEtC,UAAU,gBAAgB;IACzB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,YAAY,CAAC,EAAE,OAAO,CAAC;IACvB,UAAU,CAAC,EAAE,OAAO,CAAC;CACrB;AAED,MAAM,MAAM,oCAAoC,GAC7C,8BAA8B,GAC9B,yBAAyB,GACzB,gCAAgC,GAChC,uBAAuB,GACvB,+BAA+B,CAAC;AAEnC,UAAU,4BAA4B;IACrC,MAAM,EAAE,4CAA4C,CAAC;IACrD,MAAM,EAAE,8BAA8B,CAAC;IACvC,WAAW,EAAE,iCAAiC,CAAC;IAC/C,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,cAAc,CAAC,EAAE,UAAU,GAAG,kBAAkB,GAAG,uBAAuB,CAAC;CAC3E;AAED,MAAM,WAAW,+BAA+B;IAC/C,eAAe,CAAC,EAAE,oCAAoC,CAAC;IACvD,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,cAAc,CAAC,EAAE,OAAO,CAAC;CACzB;AA8CD,wBAAgB,yCAAyC,CACxD,KAAK,EAAE,OAAO,GACZ,iCAAiC,GAAG,SAAS,CAW/C;AAED,wBAAgB,gCAAgC,CAC/C,WAAW,EAAE,qBAAqB,EAClC,aAAa,EAAE,MAAM,EACrB,WAAW,EAAE,iCAAiC,EAC9C,QAAQ,EAAE,gBAAgB,EAC1B,aAAa,GAAE,YAAY,EAAO,EAClC,WAAW,CAAC,EAAE,+BAA+B,GAC3C,gBAAgB,CAAC,4BAA4B,CAAC,GAAG,SAAS,CAuN5D"}

package/dist/core/extensions/builtin/compaction/deterministic-fallback.js

@@ -1,8 +1,9 @@
 import { estimateTokens } from "../../../compaction/index.js";
 import { StreamDurationBudgetError, StreamIdleTimeoutError } from "../../../compaction/stream-watchdog.js";
 import { filterContextExcludedMessages } from "../../../messages.js";
-import { buildSessionContext } from "../../../session-manager.js";
+import { buildSessionContext, getSessionContextEntryId, } from "../../../session-manager.js";
 import { SummarizationOverflowExhaustedError } from "./overflow-retry.js";
+import { resolveEffectiveReserveTokens } from "./policy.js";
 import { hasUnsafeRetainedContent } from "./retained-message-safety.js";
 import { SummaryRequestError } from "./speculative.js";
 import { capUtf8Bytes } from "./task-intent.js";
@@ -59,31 +60,6 @@ function isSafeBoundedValue(value, seen = new Set(), depth = 0) {
     }
     return true;
 }
-/**
- * Conservative retained-context size. Fails closed (Infinity) on any unsafe or
- * un-serializable value, and exits early once `budget` is exceeded so an
- * oversized retained message cannot force an allocation-heavy full scan.
- */
-function estimateConservativeTokens(messages, budget) {
-    let total = 0;
-    for (const message of messages) {
-        if (!isSafeBoundedValue(message))
-            return Number.POSITIVE_INFINITY;
-        let serialized;
-        try {
-            serialized = JSON.stringify(message);
-        }
-        catch {
-            return Number.POSITIVE_INFINITY;
-        }
-        if (serialized === undefined)
-            return Number.POSITIVE_INFINITY;
-        total += Math.max(estimateTokens(message), Buffer.byteLength(serialized));
-        if (total > budget)
-            return total;
-    }
-    return total;
-}
 export function classifyRequiredCompactionFallbackFailure(error) {
     if (error instanceof StreamDurationBudgetError || error instanceof StreamIdleTimeoutError) {
         return "summarization-timeout";
@@ -96,9 +72,11 @@ export function classifyRequiredCompactionFallbackFailure(error) {
     }
     return undefined;
 }
-export function createRequiredCompactionFallback(preparation, contextWindow, failureKind, metadata, branchEntries = []) {
+export function createRequiredCompactionFallback(preparation, contextWindow, failureKind, metadata, branchEntries = [], diagnostics) {
     const preparedBoundaryIndex = branchEntries.findIndex((entry) => entry.id === preparation.firstKeptEntryId);
     if (!preparation.firstKeptEntryId || preparedBoundaryIndex === -1) {
+        if (diagnostics)
+            diagnostics.rejectionReason = "missing-preparation-boundary";
         return undefined;
     }
     const marker = [
@@ -125,7 +103,111 @@ export function createRequiredCompactionFallback(preparation, contextWindow, fai
         failureKind,
         ...(taskIntent ? { taskIntent } : {}),
     };
+    let candidateCount = 0;
+    const syntheticCompaction = {
+        type: "compaction",
+        id: "__senpi_deterministic_fallback_preview__",
+        parentId: branchEntries.at(-1)?.id ?? null,
+        timestamp: new Date(0).toISOString(),
+        summary,
+        firstKeptEntryId: branchEntries[0]?.id ?? preparation.firstKeptEntryId,
+        tokensBefore: preparation.tokensBefore,
+        details: { ...baseDetails, retainedSuffix: "prepared" },
+        fromHook: true,
+    };
+    let projectedMessages;
+    try {
+        projectedMessages = filterContextExcludedMessages(buildSessionContext([...branchEntries, syntheticCompaction]).messages);
+    }
+    catch {
+        if (diagnostics)
+            diagnostics.rejectionReason = "context-reconstruction-failed";
+        return undefined;
+    }
+    const messageIndexesByEntryId = new Map();
+    for (const [index, message] of projectedMessages.entries()) {
+        const entryId = getSessionContextEntryId(message);
+        if (entryId !== undefined)
+            messageIndexesByEntryId.set(entryId, index);
+    }
+    const summaryIndex = messageIndexesByEntryId.get(syntheticCompaction.id);
+    const unsafeSuffix = new Array(projectedMessages.length + 1).fill(false);
+    const tokenSuffix = new Array(projectedMessages.length + 1).fill(0);
+    const toolCalls = new Map();
+    const toolResults = new Map();
+    for (let index = projectedMessages.length - 1; index >= 0; index--) {
+        const message = projectedMessages[index];
+        tokenSuffix[index] = tokenSuffix[index + 1];
+        let messageUnsafe = false;
+        let serialized;
+        try {
+            messageUnsafe = hasUnsafeRetainedContent([message]);
+            serialized = isSafeBoundedValue(message) ? JSON.stringify(message) : undefined;
+            tokenSuffix[index] +=
+                serialized === undefined
+                    ? Number.POSITIVE_INFINITY
+                    : Math.max(estimateTokens(message), Buffer.byteLength(serialized));
+        }
+        catch {
+            messageUnsafe = true;
+            tokenSuffix[index] = Number.POSITIVE_INFINITY;
+        }
+        unsafeSuffix[index] = unsafeSuffix[index + 1] || messageUnsafe;
+        if (!isRecord(message))
+            continue;
+        if (message.role === "toolResult" && typeof message.toolCallId === "string") {
+            const indexes = toolResults.get(message.toolCallId) ?? [];
+            indexes.push(index);
+            toolResults.set(message.toolCallId, indexes);
+        }
+        else if (message.role === "assistant" && Array.isArray(message.content)) {
+            for (const block of message.content) {
+                if (!isRecord(block) || block.type !== "toolCall" || typeof block.id !== "string")
+                    continue;
+                const call = toolCalls.get(block.id) ?? { indexes: [], incomplete: false };
+                call.indexes.push(index);
+                call.incomplete ||= block.incomplete === true;
+                toolCalls.set(block.id, call);
+            }
+        }
+    }
+    const invalidToolChainDiff = new Int32Array(projectedMessages.length + 2);
+    const markInvalidRange = (start, endExclusive) => {
+        const boundedStart = Math.max(0, start);
+        const boundedEnd = Math.min(projectedMessages.length + 1, endExclusive);
+        if (boundedStart >= boundedEnd)
+            return;
+        invalidToolChainDiff[boundedStart]++;
+        invalidToolChainDiff[boundedEnd]--;
+    };
+    for (const [id, call] of toolCalls) {
+        const results = toolResults.get(id) ?? [];
+        if (call.indexes.length !== 1 || results.length !== 1) {
+            markInvalidRange(0, Math.max(call.indexes[0] ?? -1, results[0] ?? -1) + 1);
+            continue;
+        }
+        const callIndex = call.indexes[0];
+        const resultIndex = results[0];
+        if (call.incomplete)
+            markInvalidRange(0, Math.max(callIndex, resultIndex) + 1);
+        else if (callIndex < resultIndex)
+            markInvalidRange(callIndex + 1, resultIndex + 1);
+        else
+            markInvalidRange(0, Math.max(callIndex, resultIndex) + 1);
+    }
+    for (const [id, results] of toolResults) {
+        if (toolCalls.has(id))
+            continue;
+        markInvalidRange(0, results[0] + 1);
+    }
+    const toolChainValidAt = new Array(projectedMessages.length + 1).fill(true);
+    let invalidToolChainCount = 0;
+    for (let index = 0; index < toolChainValidAt.length; index++) {
+        invalidToolChainCount += invalidToolChainDiff[index];
+        toolChainValidAt[index] = invalidToolChainCount === 0;
+    }
     const projectCandidate = (firstKeptEntryId, retainedSuffix) => {
+        candidateCount++;
         const details = { ...baseDetails, retainedSuffix };
         const result = {
             summary,
@@ -133,41 +215,74 @@ export function createRequiredCompactionFallback(preparation, contextWindow, fai
             tokensBefore: preparation.tokensBefore,
             details,
         };
-        const syntheticCompaction = {
-            type: "compaction",
-            id: "__senpi_deterministic_fallback_preview__",
-            parentId: branchEntries.at(-1)?.id ?? null,
-            timestamp: new Date(0).toISOString(),
-            summary: result.summary,
-            firstKeptEntryId: result.firstKeptEntryId,
-            tokensBefore: result.tokensBefore,
-            details: result.details,
-            fromHook: true,
-        };
-        let retainedMessages;
-        try {
-            retainedMessages = filterContextExcludedMessages(buildSessionContext([...branchEntries, syntheticCompaction]).messages);
+        const startIndex = messageIndexesByEntryId.get(firstKeptEntryId);
+        if (startIndex === undefined) {
+            if (diagnostics)
+                diagnostics.rejectionReason = "context-reconstruction-failed";
+            return undefined;
         }
-        catch {
+        const retainedStart = Math.min(startIndex, projectedMessages.length);
+        if (unsafeSuffix[retainedStart]) {
+            if (diagnostics)
+                diagnostics.rejectionReason = "unsafe-retained-content";
             return undefined;
         }
-        if (hasUnsafeRetainedContent(retainedMessages))
+        if (!toolChainValidAt[retainedStart]) {
+            if (diagnostics)
+                diagnostics.rejectionReason = "atomic-tool-chain-cut";
             return undefined;
-        const budget = contextWindow - preparation.settings.reserveTokens;
-        const retainedTokens = estimateConservativeTokens(retainedMessages, budget);
-        if (retainedTokens > budget)
+        }
+        const budget = contextWindow -
+            resolveEffectiveReserveTokens(contextWindow, preparation.settings.reserveTokens, preparation.settings.reserveScalingEnabled !== false);
+        const summaryTokens = summaryIndex === undefined
+            ? Number.POSITIVE_INFINITY
+            : Math.max(estimateTokens(projectedMessages[summaryIndex]), Buffer.byteLength(JSON.stringify(projectedMessages[summaryIndex])));
+        const retainedTokens = (tokenSuffix[retainedStart] ?? Number.POSITIVE_INFINITY) + summaryTokens;
+        if (retainedTokens > budget) {
+            if (diagnostics) {
+                diagnostics.rejectionReason = "retained-token-budget-exceeded";
+                diagnostics.budgetExceeded = true;
+            }
             return undefined;
+        }
         return { ...result, estimatedTokensAfter: retainedTokens };
     };
+    // 1. Try prepared boundary
     const prepared = projectCandidate(preparation.firstKeptEntryId, "prepared");
-    if (prepared)
+    if (prepared) {
+        if (diagnostics)
+            diagnostics.candidatesChecked = candidateCount;
         return prepared;
+    }
+    // 2. If prepared boundary cut an atomic signed chain or was rejected, walk backward
+    // to the actual chain boundary. The chain length is determined by the session.
+    for (let index = preparedBoundaryIndex - 1; index >= 0; index--) {
+        const entry = branchEntries[index];
+        if (entry.type === "compaction")
+            break;
+        // Only consider message turn starts or assistant call starts
+        const earlier = projectCandidate(entry.id, "earlier-safe-boundary");
+        if (earlier) {
+            if (diagnostics)
+                diagnostics.candidatesChecked = candidateCount;
+            return earlier;
+        }
+    }
+    // 3. Try latest meaningful user turn
     for (let index = branchEntries.length - 1; index > preparedBoundaryIndex; index--) {
         const entry = branchEntries[index];
         if (!hasMeaningfulUserText(entry))
             continue;
-        return projectCandidate(entry.id, "latest-user-turn");
+        const latestUser = projectCandidate(entry.id, "latest-user-turn");
+        if (latestUser) {
+            if (diagnostics)
+                diagnostics.candidatesChecked = candidateCount;
+            return latestUser;
+        }
+        break;
     }
+    if (diagnostics)
+        diagnostics.candidatesChecked = candidateCount;
     return undefined;
 }
 //# sourceMappingURL=deterministic-fallback.js.map