@taco-ai/pi-harness 0.0.0-stage → 0.1.0

package/dist/harness/runtime/drive/response.js

@@ -0,0 +1,395 @@
+import { isContextOverflow, isRecoverableLength, isRetryableAssistantError, retryDelayMs, } from "@earendil-works/pi-ai";
+import { AssistantMessageFrameEncoder } from "../../../shims/assistant-message-frame.js";
+import { insertEntry, insertUsage } from "../../session/commit.js";
+import { SessionInvariantError } from "../../session/session.js";
+import { operationScopeOf, } from "../../session/types.js";
+import { branchTip, deleteList, operationPreparation, pendingAssistantFrames, setValue } from "../../session/values.js";
+import { openFrameProgress } from "../progress.js";
+import { retryNotBefore } from "./retry.js";
+import { prepareOverflowCompaction } from "./structural.js";
+import { operationCleanupWrites, operationResultRecord } from "./terminal.js";
+export function openAssistantResponse(lane, drive, responseEntryId, recovery = false) {
+    const progress = openFrameProgress(lane, drive, responseEntryId);
+    const frameEncoder = new AssistantMessageFrameEncoder();
+    const eventContext = {
+        lane: lane.name,
+        runId: drive.operationId,
+        ...(recovery ? { recovery: true } : {}),
+    };
+    const close = async () => {
+        progress.seal();
+        await progress.drain();
+    };
+    return {
+        observer: {
+            start(message, event, context) {
+                const frame = frameEncoder.encode(event);
+                if (frame !== undefined)
+                    progress.write(frame);
+                return lane.emitBatch([{ type: "message_start", ...eventContext, message }], context);
+            },
+            update(message, event, context) {
+                const frame = frameEncoder.encode(event);
+                if (frame !== undefined)
+                    progress.write(frame);
+                return lane.emitBatch([{ type: "message_update", ...eventContext, message, event, ...(frame === undefined ? {} : { frame }) }], context);
+            },
+            end(message, context) {
+                return lane.emitBatch([{ type: "message_end", ...eventContext, message, entryId: responseEntryId }], context);
+            },
+        },
+        async afterResponse(message, metadata, context) {
+            await close();
+            const result = await lane.hooks.runWithGate("after_response", { lane: lane.name, runId: drive.operationId, ...metadata, message }, drive.gate, context);
+            return result?.message ?? message;
+        },
+        close,
+    };
+}
+/** Publish a non-retryable request-configuration failure before reserving response ids. */
+export async function publishConfigurationFailure(lane, drive, capability, error) {
+    const result = await lane.continueOperation(capability, async (state, current, meta, reader) => {
+        if (state.tipId === null)
+            throw new SessionInvariantError("Failed run has no Branch tip");
+        const record = operationResultRecord(meta, "failed", state.tipId, error);
+        const cleanup = await operationCleanupWrites(reader, drive.operationId, current, drive.context);
+        return {
+            kind: "finish",
+            writes: cleanup,
+            record,
+            materialize: () => ({ kind: "settled", outcome: record }),
+            events: () => [
+                {
+                    type: "run_end",
+                    lane: lane.name,
+                    runId: drive.operationId,
+                    status: "failed",
+                    error,
+                    fromTipId: meta.sourceTipId,
+                    tipId: state.tipId,
+                    endedAt: record.endedAt,
+                },
+            ],
+        };
+    }, drive.context);
+    return result.kind === "cancel_requested" ? { kind: "continue" } : result.value;
+}
+function uuidV7Timestamp(id) {
+    const timestamp = Number.parseInt(id.slice(0, 8) + id.slice(9, 13), 16);
+    if (!Number.isSafeInteger(timestamp))
+        throw new SessionInvariantError(`Invalid reserved UUIDv7 ${id}`);
+    return timestamp;
+}
+function providerError(source, message) {
+    return {
+        code: "assistant_error",
+        message: message.errorMessage ??
+            `${source === "assistant" ? "Assistant" : "Deferred"} request ended with ${message.stopReason}`,
+    };
+}
+function normalizeError(message, errorMessage) {
+    return { ...message, stopReason: "error", errorMessage };
+}
+function normalizeAborted(source, message) {
+    return {
+        ...message,
+        stopReason: "aborted",
+        errorMessage: message.errorMessage ?? `${source === "assistant" ? "Assistant" : "Deferred"} request was cancelled`,
+    };
+}
+function deferredHandleIsValid(message, generation) {
+    const handle = message.deferred;
+    const identity = generation.generationContext.configuration.model;
+    return (message.stopReason === "deferred" &&
+        handle !== undefined &&
+        handle.id.length !== 0 &&
+        handle.provider === identity.provider &&
+        handle.modelId === identity.modelId &&
+        handle.api === message.api);
+}
+/** Classify and atomically settle one assistant-generation or deferred-poll response. */
+export async function publishResponse(lane, drive, intent, response, options = {}) {
+    const overflow = intent.at === "assistant.effect_pending" &&
+        (isContextOverflow(response, intent.contextWindow) || isRecoverableLength(response, intent.intendedOutputLimit));
+    const overflowPreparation = overflow && !intent.generationContext.overflowRecoveryUsed
+        ? await prepareOverflowCompaction(lane, drive, intent)
+        : undefined;
+    return lane.settleOperation(intent, async (state, current, meta, reader) => {
+        const source = current.at === "assistant.effect_pending" ? "assistant" : "deferred";
+        const responseEntryId = intent.responseEntryId;
+        const configuration = current.at === "assistant.effect_pending" ? current.generationContext.configuration : current.configuration;
+        const turnId = current.at === "assistant.effect_pending"
+            ? current.generationContext.stepId
+            : `${current.stepId}:poll:${current.poll}`;
+        const scope = { ...operationScopeOf(current), latestAssistantEntryId: responseEntryId };
+        let committed = response;
+        let settled;
+        let failure;
+        if (current.control.status === "cancel_requested") {
+            committed = normalizeAborted(source, response);
+            settled = {
+                ...scope,
+                at: "checkpoint",
+                continuation: { kind: "may_finish", includeFinalAssistant: true },
+                triggerEntryId: responseEntryId,
+            };
+        }
+        else if (response.stopReason === "aborted") {
+            throw new SessionInvariantError(`${source === "assistant" ? "Assistant" : "Deferred"} response is aborted while durable control is running`);
+        }
+        else if (current.at === "assistant.effect_pending" && overflow) {
+            committed = normalizeError(response, response.errorMessage ?? "Assistant request exceeded the context window");
+            if (current.generationContext.overflowRecoveryUsed || overflowPreparation === undefined) {
+                failure = providerError(source, committed);
+            }
+            else {
+                const structural = {
+                    ...scope,
+                    at: "summary.deciding",
+                    task: {
+                        taskId: overflowPreparation.taskId,
+                        reason: "overflow",
+                        boundary: {
+                            kind: "resume_checkpoint",
+                            resumeAfter: {
+                                continuation: { kind: "need_assistant", overflowRecoveryUsed: true },
+                                triggerEntryId: current.generationContext.triggerEntryId,
+                            },
+                        },
+                    },
+                };
+                settled = structural;
+            }
+        }
+        else if (response.stopReason === "deferred") {
+            if (current.at === "assistant.effect_pending") {
+                if (deferredHandleIsValid(response, current)) {
+                    settled = {
+                        ...scope,
+                        at: "deferred.suspended",
+                        stepId: current.generationContext.stepId,
+                        sourceEntryId: responseEntryId,
+                        poll: 0,
+                        configuration,
+                        streamOptions: current.generationContext.streamOptions,
+                    };
+                }
+                else {
+                    committed = normalizeError(response, "Provider returned an invalid deferred handle");
+                    failure = providerError(source, committed);
+                }
+            }
+            else {
+                settled = {
+                    ...scope,
+                    at: "deferred.suspended",
+                    stepId: current.stepId,
+                    sourceEntryId: responseEntryId,
+                    poll: current.poll,
+                    configuration,
+                    streamOptions: current.streamOptions,
+                };
+            }
+        }
+        else if (response.stopReason === "error") {
+            if (current.at === "assistant.effect_pending" &&
+                (options.recovery === true || isRetryableAssistantError(response)) &&
+                current.attempt < current.generationContext.retryPolicy.maxAttempts) {
+                settled = {
+                    ...scope,
+                    at: "assistant.retry_wait",
+                    generationContext: current.generationContext,
+                    nextAttempt: current.attempt + 1,
+                    notBefore: retryNotBefore(current.generationContext.retryPolicy, current.attempt),
+                    errorMessage: response.errorMessage ?? "Assistant request failed",
+                };
+            }
+            else {
+                failure = providerError(source, response);
+            }
+        }
+        else {
+            const calls = response.content.flatMap((content, sourceIndex) => content.type === "toolCall" ? [{ sourceIndex }] : []);
+            if (calls.length !== 0) {
+                const timestamp = uuidV7Timestamp(responseEntryId);
+                const planned = calls.map(({ sourceIndex }) => ({
+                    status: "planned",
+                    sourceIndex,
+                    resultEntryId: lane.session.idGenerator.next(timestamp),
+                }));
+                settled = {
+                    ...scope,
+                    at: "tools",
+                    batch: { assistantEntryId: responseEntryId, configuration, turnId, calls: planned },
+                };
+            }
+            else if (response.stopReason === "toolUse") {
+                committed = normalizeError(response, "Provider reported tool use without any tool calls");
+                failure = providerError(source, committed);
+            }
+            else {
+                settled = {
+                    ...scope,
+                    at: "checkpoint",
+                    continuation: { kind: "may_finish", includeFinalAssistant: true },
+                    triggerEntryId: responseEntryId,
+                };
+            }
+        }
+        const responseEntry = {
+            id: responseEntryId,
+            parentId: state.tipId,
+            type: "message",
+            message: committed,
+        };
+        const usageRow = {
+            id: intent.usageId,
+            usage: committed.usage,
+            entryId: responseEntryId,
+            adjustment: false,
+        };
+        if (settled === undefined && failure === undefined) {
+            throw new SessionInvariantError("Response settlement has no durable disposition");
+        }
+        const record = failure === undefined ? undefined : operationResultRecord(meta, "failed", responseEntryId, failure);
+        const cleanup = record === undefined ? [] : await operationCleanupWrites(reader, drive.operationId, current, drive.context);
+        const writes = [
+            insertEntry(responseEntry),
+            insertUsage(usageRow),
+            setValue(branchTip(lane.name), responseEntryId),
+            ...(record === undefined
+                ? [deleteList(pendingAssistantFrames(drive.operationId, responseEntryId))]
+                : cleanup),
+            ...(settled?.at === "summary.deciding" && overflowPreparation !== undefined
+                ? [
+                    setValue(operationPreparation(drive.operationId, overflowPreparation.taskId), overflowPreparation.preparation),
+                ]
+                : []),
+        ];
+        const materializeEntry = (commit) => ({
+            ...responseEntry,
+            seq: commit.seqs[0],
+            timestamp: commit.timestamp,
+        });
+        const events = (commit) => {
+            const entry = materializeEntry(commit);
+            const batch = [
+                { type: "entry_added", lane: lane.name, entry, ...options },
+                {
+                    type: "usage",
+                    lane: lane.name,
+                    row: { ...usageRow, seq: commit.seqs[1] },
+                    totals: commit.stats.usage,
+                },
+            ];
+            if (current.at === "assistant.effect_pending") {
+                if (options.recovery !== true && current.attempt > 1 && settled?.at !== "assistant.retry_wait") {
+                    const success = committed.stopReason !== "error" && committed.stopReason !== "aborted";
+                    batch.push({
+                        type: "retry_end",
+                        lane: lane.name,
+                        runId: drive.operationId,
+                        step: turnId,
+                        attempt: current.attempt,
+                        success,
+                        ...(success
+                            ? {}
+                            : {
+                                finalError: committed.errorMessage ?? `Assistant request ended with ${committed.stopReason}`,
+                            }),
+                    });
+                }
+                if (options.recovery !== true && settled?.at === "assistant.retry_wait") {
+                    batch.push({
+                        type: "retry_scheduled",
+                        lane: lane.name,
+                        runId: drive.operationId,
+                        step: turnId,
+                        attempt: settled.nextAttempt,
+                        maxAttempts: settled.generationContext.retryPolicy.maxAttempts,
+                        delayMs: retryDelayMs(current.generationContext.retryPolicy, current.attempt),
+                        notBefore: settled.notBefore,
+                        errorMessage: settled.errorMessage,
+                    });
+                }
+                if (options.recovery !== true && settled?.at !== "tools" && settled?.at !== "assistant.retry_wait") {
+                    batch.push({
+                        type: "turn_end",
+                        lane: lane.name,
+                        runId: drive.operationId,
+                        turnId,
+                        message: committed,
+                        toolResults: [],
+                    });
+                }
+                if (settled?.at === "summary.deciding") {
+                    batch.push({
+                        type: "compaction_start",
+                        lane: lane.name,
+                        runId: drive.operationId,
+                        reason: "overflow",
+                        startedAt: commit.timestamp,
+                    });
+                }
+            }
+            else if (settled?.at !== "tools") {
+                batch.push({
+                    type: "turn_end",
+                    lane: lane.name,
+                    runId: drive.operationId,
+                    turnId,
+                    message: committed,
+                    toolResults: [],
+                    ...options,
+                });
+            }
+            if ((current.at === "deferred.effect_pending" || options.recovery !== true) &&
+                settled?.at === "deferred.suspended" &&
+                committed.deferred !== undefined) {
+                batch.push({
+                    type: "run_suspend",
+                    lane: lane.name,
+                    runId: drive.operationId,
+                    reason: "deferred",
+                    deferred: committed.deferred,
+                    poll: settled.poll,
+                    ...(current.at === "deferred.effect_pending" ? options : {}),
+                });
+            }
+            if (record !== undefined && failure !== undefined) {
+                batch.push({
+                    type: "run_end",
+                    lane: lane.name,
+                    runId: drive.operationId,
+                    status: "failed",
+                    error: failure,
+                    fromTipId: meta.sourceTipId,
+                    tipId: responseEntryId,
+                    endedAt: record.endedAt,
+                });
+            }
+            return batch;
+        };
+        if (record !== undefined) {
+            return {
+                kind: "finish",
+                writes,
+                record,
+                lane: { tipId: responseEntryId },
+                materialize: () => ({ kind: "settled", outcome: record }),
+                events,
+            };
+        }
+        if (settled === undefined)
+            throw new SessionInvariantError("Response settlement is missing its next state");
+        return {
+            kind: "commit",
+            writes,
+            operationState: settled,
+            lane: { tipId: responseEntryId },
+            materialize: () => ({ kind: "continue" }),
+            events,
+        };
+    }, drive.context);
+}
+//# sourceMappingURL=response.js.map

package/dist/harness/runtime/drive/response.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"response.js","sourceRoot":"","sources":["../../../../src/harness/runtime/drive/response.ts"],"names":[],"mappings":"AAAA,OAAO,EAEN,iBAAiB,EACjB,mBAAmB,EACnB,yBAAyB,EACzB,YAAY,GACZ,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EAAE,4BAA4B,EAAE,MAAM,2CAA2C,CAAC;AAIzF,OAAO,EAAE,WAAW,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AACnE,OAAO,EAAE,qBAAqB,EAAE,MAAM,0BAA0B,CAAC;AACjE,OAAO,EAQN,gBAAgB,GAKhB,MAAM,wBAAwB,CAAC;AAChC,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,oBAAoB,EAAE,sBAAsB,EAAE,QAAQ,EAAE,MAAM,yBAAyB,CAAC;AAExH,OAAO,EAAE,iBAAiB,EAAE,MAAM,gBAAgB,CAAC;AAEnD,OAAO,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAC5C,OAAO,EAAE,yBAAyB,EAAE,MAAM,iBAAiB,CAAC;AAC5D,OAAO,EAAE,sBAAsB,EAAE,qBAAqB,EAAE,MAAM,eAAe,CAAC;AAY9E,MAAM,UAAU,qBAAqB,CACpC,IAAoB,EACpB,KAAY,EACZ,eAAuB,EACvB,QAAQ,GAAG,KAAK;IAEhB,MAAM,QAAQ,GAAG,iBAAiB,CAAC,IAAI,EAAE,KAAK,EAAE,eAAe,CAAC,CAAC;IACjE,MAAM,YAAY,GAAG,IAAI,4BAA4B,EAAE,CAAC;IACxD,MAAM,YAAY,GAAG;QACpB,IAAI,EAAE,IAAI,CAAC,IAAI;QACf,KAAK,EAAE,KAAK,CAAC,WAAW;QACxB,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,IAAa,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAChD,CAAC;IACF,MAAM,KAAK,GAAG,KAAK,IAAI,EAAE;QACxB,QAAQ,CAAC,IAAI,EAAE,CAAC;QAChB,MAAM,QAAQ,CAAC,KAAK,EAAE,CAAC;IACxB,CAAC,CAAC;IACF,OAAO;QACN,QAAQ,EAAE;YACT,KAAK,CAAC,OAAO,EAAE,KAAK,EAAE,OAAO;gBAC5B,MAAM,KAAK,GAAG,YAAY,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBACzC,IAAI,KAAK,KAAK,SAAS;oBAAE,QAAQ,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;gBAC/C,OAAO,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,GAAG,YAAY,EAAE,OAAO,EAAE,CAAC,EAAE,OAAO,CAAC,CAAC;YACvF,CAAC;YACD,MAAM,CAAC,OAAO,EAAE,KAA4B,EAAE,OAAO;gBACpD,MAAM,KAAK,GAAG,YAAY,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBACzC,IAAI,KAAK,KAAK,SAAS;oBAAE,QAAQ,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;gBAC/C,OAAO,IAAI,CAAC,SAAS,CACpB,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,GAAG,YAAY,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,EACxG,OAAO,CACP,CAAC;YACH,CAAC;YACD,GAAG,CAAC,OAAO,EAAE,OAAO;gBACnB,OAAO,IAAI,CAAC,SAAS,CACpB,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,GAAG,YAAY,EAAE,OAAO,EAAE,OAAO,EAAE,eAAe,EAAE,CAAC,EAC7E,OAAO,CACP,CAAC;YACH,CAAC;SACD;QACD,KAAK,CAAC,aAAa,CAAC,OAAO,EAAE,QAAQ,EAAE,OAAO;YAC7C,MAAM,KAAK,EAAE,CAAC;YACd,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,WAAW,CAC1C,gBAAgB,EAChB,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,KAAK,CAAC,WAAW,EAAE,GAAG,QAAQ,EAAE,OAAO,EAAE,EACnE,KAAK,CAAC,IAAI,EACV,OAAO,CACP,CAAC;YACF,OAAO,MAAM,EAAE,OAAO,IAAI,OAAO,CAAC;QACnC,CAAC;QACD,KAAK;KACL,CAAC;AACH,CAAC;AAQD,2FAA2F;AAC3F,MAAM,CAAC,KAAK,UAAU,2BAA2B,CAG/C,IAAoB,EAAE,KAAY,EAAE,UAAkB,EAAE,KAAqB;IAC9E,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAC1C,UAAU,EACV,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE;QACtC,IAAI,KAAK,CAAC,KAAK,KAAK,IAAI;YAAE,MAAM,IAAI,qBAAqB,CAAC,8BAA8B,CAAC,CAAC;QAC1F,MAAM,MAAM,GAAG,qBAAqB,CAAC,IAAI,EAAE,QAAQ,EAAE,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;QACzE,MAAM,OAAO,GAAG,MAAM,sBAAsB,CAAC,MAAM,EAAE,KAAK,CAAC,WAAW,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;QAChG,OAAO;YACN,IAAI,EAAE,QAAQ;YACd,MAAM,EAAE,OAAO;YACf,MAAM;YACN,WAAW,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,EAAE,CAAU;YAClE,MAAM,EAAE,GAAG,EAAE,CAAC;gBACb;oBACC,IAAI,EAAE,SAAS;oBACf,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,KAAK,EAAE,KAAK,CAAC,WAAW;oBACxB,MAAM,EAAE,QAAQ;oBAChB,KAAK;oBACL,SAAS,EAAE,IAAI,CAAC,WAAW;oBAC3B,KAAK,EAAE,KAAK,CAAC,KAAK;oBAClB,OAAO,EAAE,MAAM,CAAC,OAAO;iBACvB;aACD;SACD,CAAC;IACH,CAAC,EACD,KAAK,CAAC,OAAO,CACb,CAAC;IACF,OAAO,MAAM,CAAC,IAAI,KAAK,kBAAkB,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;AACjF,CAAC;AAED,SAAS,eAAe,CAAC,EAAU;IAClC,MAAM,SAAS,GAAG,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;IACxE,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,SAAS,CAAC;QAAE,MAAM,IAAI,qBAAqB,CAAC,2BAA2B,EAAE,EAAE,CAAC,CAAC;IACvG,OAAO,SAAS,CAAC;AAClB,CAAC;AAED,SAAS,aAAa,CAAC,MAAgC,EAAE,OAAgC;IACxF,OAAO;QACN,IAAI,EAAE,iBAAiB;QACvB,OAAO,EACN,OAAO,CAAC,YAAY;YACpB,GAAG,MAAM,KAAK,WAAW,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,UAAU,uBAAuB,OAAO,CAAC,UAAU,EAAE;KAChG,CAAC;AACH,CAAC;AAED,SAAS,cAAc,CAAC,OAAgC,EAAE,YAAoB;IAC7E,OAAO,EAAE,GAAG,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC;AAC1D,CAAC;AAED,SAAS,gBAAgB,CAAC,MAAgC,EAAE,OAAgC;IAC3F,OAAO;QACN,GAAG,OAAO;QACV,UAAU,EAAE,SAAS;QACrB,YAAY,EACX,OAAO,CAAC,YAAY,IAAI,GAAG,MAAM,KAAK,WAAW,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,UAAU,wBAAwB;KACrG,CAAC;AACH,CAAC;AAED,SAAS,qBAAqB,CAAC,OAAgC,EAAE,UAA2C;IAC3G,MAAM,MAAM,GAAG,OAAO,CAAC,QAAQ,CAAC;IAChC,MAAM,QAAQ,GAAG,UAAU,CAAC,iBAAiB,CAAC,aAAa,CAAC,KAAK,CAAC;IAClE,OAAO,CACN,OAAO,CAAC,UAAU,KAAK,UAAU;QACjC,MAAM,KAAK,SAAS;QACpB,MAAM,CAAC,EAAE,CAAC,MAAM,KAAK,CAAC;QACtB,MAAM,CAAC,QAAQ,KAAK,QAAQ,CAAC,QAAQ;QACrC,MAAM,CAAC,OAAO,KAAK,QAAQ,CAAC,OAAO;QACnC,MAAM,CAAC,GAAG,KAAK,OAAO,CAAC,GAAG,CAC1B,CAAC;AACH,CAAC;AAED,yFAAyF;AACzF,MAAM,CAAC,KAAK,UAAU,eAAe,CACpC,IAAoB,EACpB,KAAY,EACZ,MAAsB,EACtB,QAAiC,EACjC,OAAO,GAAwB,EAAE;IAEjC,MAAM,QAAQ,GACb,MAAM,CAAC,EAAE,KAAK,0BAA0B;QACxC,CAAC,iBAAiB,CAAC,QAAQ,EAAE,MAAM,CAAC,aAAa,CAAC,IAAI,mBAAmB,CAAC,QAAQ,EAAE,MAAM,CAAC,mBAAmB,CAAC,CAAC,CAAC;IAClH,MAAM,mBAAmB,GACxB,QAAQ,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,oBAAoB;QACzD,CAAC,CAAC,MAAM,yBAAyB,CAAC,IAAI,EAAE,KAAK,EAAE,MAAM,CAAC;QACtD,CAAC,CAAC,SAAS,CAAC;IACd,OAAO,IAAI,CAAC,eAAe,CAC1B,MAAM,EACN,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE;QACtC,MAAM,MAAM,GAAG,OAAO,CAAC,EAAE,KAAK,0BAA0B,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,UAAU,CAAC;QACpF,MAAM,eAAe,GAAG,MAAM,CAAC,eAAe,CAAC;QAC/C,MAAM,aAAa,GAClB,OAAO,CAAC,EAAE,KAAK,0BAA0B,CAAC,CAAC,CAAC,OAAO,CAAC,iBAAiB,CAAC,aAAa,CAAC,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC;QAC7G,MAAM,MAAM,GACX,OAAO,CAAC,EAAE,KAAK,0BAA0B;YACxC,CAAC,CAAC,OAAO,CAAC,iBAAiB,CAAC,MAAM;YAClC,CAAC,CAAC,GAAG,OAAO,CAAC,MAAM,SAAS,OAAO,CAAC,IAAI,EAAE,CAAC;QAC7C,MAAM,KAAK,GAAG,EAAE,GAAG,gBAAgB,CAAC,OAAO,CAAC,EAAE,sBAAsB,EAAE,eAAe,EAAE,CAAC;QACxF,IAAI,SAAS,GAAG,QAAQ,CAAC;QACzB,IAAI,OAAmC,CAAC;QACxC,IAAI,OAAmC,CAAC;QAExC,IAAI,OAAO,CAAC,OAAO,CAAC,MAAM,KAAK,kBAAkB,EAAE,CAAC;YACnD,SAAS,GAAG,gBAAgB,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;YAC/C,OAAO,GAAG;gBACT,GAAG,KAAK;gBACR,EAAE,EAAE,YAAY;gBAChB,YAAY,EAAE,EAAE,IAAI,EAAE,YAAY,EAAE,qBAAqB,EAAE,IAAI,EAAE;gBACjE,cAAc,EAAE,eAAe;aAC/B,CAAC;QACH,CAAC;aAAM,IAAI,QAAQ,CAAC,UAAU,KAAK,SAAS,EAAE,CAAC;YAC9C,MAAM,IAAI,qBAAqB,CAC9B,GAAG,MAAM,KAAK,WAAW,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,UAAU,uDAAuD,CAC3G,CAAC;QACH,CAAC;aAAM,IAAI,OAAO,CAAC,EAAE,KAAK,0BAA0B,IAAI,QAAQ,EAAE,CAAC;YAClE,SAAS,GAAG,cAAc,CACzB,QAAQ,EACR,QAAQ,CAAC,YAAY,IAAI,+CAA+C,CACxE,CAAC;YACF,IAAI,OAAO,CAAC,iBAAiB,CAAC,oBAAoB,IAAI,mBAAmB,KAAK,SAAS,EAAE,CAAC;gBACzF,OAAO,GAAG,aAAa,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;YAC5C,CAAC;iBAAM,CAAC;gBACP,MAAM,UAAU,GAA6B;oBAC5C,GAAG,KAAK;oBACR,EAAE,EAAE,kBAAkB;oBACtB,IAAI,EAAE;wBACL,MAAM,EAAE,mBAAmB,CAAC,MAAM;wBAClC,MAAM,EAAE,UAAU;wBAClB,QAAQ,EAAE;4BACT,IAAI,EAAE,mBAAmB;4BACzB,WAAW,EAAE;gCACZ,YAAY,EAAE,EAAE,IAAI,EAAE,gBAAgB,EAAE,oBAAoB,EAAE,IAAI,EAAE;gCACpE,cAAc,EAAE,OAAO,CAAC,iBAAiB,CAAC,cAAc;6BACxD;yBACD;qBACD;iBACD,CAAC;gBACF,OAAO,GAAG,UAAU,CAAC;YACtB,CAAC;QACF,CAAC;aAAM,IAAI,QAAQ,CAAC,UAAU,KAAK,UAAU,EAAE,CAAC;YAC/C,IAAI,OAAO,CAAC,EAAE,KAAK,0BAA0B,EAAE,CAAC;gBAC/C,IAAI,qBAAqB,CAAC,QAAQ,EAAE,OAAO,CAAC,EAAE,CAAC;oBAC9C,OAAO,GAAG;wBACT,GAAG,KAAK;wBACR,EAAE,EAAE,oBAAoB;wBACxB,MAAM,EAAE,OAAO,CAAC,iBAAiB,CAAC,MAAM;wBACxC,aAAa,EAAE,eAAe;wBAC9B,IAAI,EAAE,CAAC;wBACP,aAAa;wBACb,aAAa,EAAE,OAAO,CAAC,iBAAiB,CAAC,aAAa;qBACtD,CAAC;gBACH,CAAC;qBAAM,CAAC;oBACP,SAAS,GAAG,cAAc,CAAC,QAAQ,EAAE,8CAA8C,CAAC,CAAC;oBACrF,OAAO,GAAG,aAAa,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;gBAC5C,CAAC;YACF,CAAC;iBAAM,CAAC;gBACP,OAAO,GAAG;oBACT,GAAG,KAAK;oBACR,EAAE,EAAE,oBAAoB;oBACxB,MAAM,EAAE,OAAO,CAAC,MAAM;oBACtB,aAAa,EAAE,eAAe;oBAC9B,IAAI,EAAE,OAAO,CAAC,IAAI;oBAClB,aAAa;oBACb,aAAa,EAAE,OAAO,CAAC,aAAa;iBACpC,CAAC;YACH,CAAC;QACF,CAAC;aAAM,IAAI,QAAQ,CAAC,UAAU,KAAK,OAAO,EAAE,CAAC;YAC5C,IACC,OAAO,CAAC,EAAE,KAAK,0BAA0B;gBACzC,CAAC,OAAO,CAAC,QAAQ,KAAK,IAAI,IAAI,yBAAyB,CAAC,QAAQ,CAAC,CAAC;gBAClE,OAAO,CAAC,OAAO,GAAG,OAAO,CAAC,iBAAiB,CAAC,WAAW,CAAC,WAAW,EAClE,CAAC;gBACF,OAAO,GAAG;oBACT,GAAG,KAAK;oBACR,EAAE,EAAE,sBAAsB;oBAC1B,iBAAiB,EAAE,OAAO,CAAC,iBAAiB;oBAC5C,WAAW,EAAE,OAAO,CAAC,OAAO,GAAG,CAAC;oBAChC,SAAS,EAAE,cAAc,CAAC,OAAO,CAAC,iBAAiB,CAAC,WAAW,EAAE,OAAO,CAAC,OAAO,CAAC;oBACjF,YAAY,EAAE,QAAQ,CAAC,YAAY,IAAI,0BAA0B;iBACjE,CAAC;YACH,CAAC;iBAAM,CAAC;gBACP,OAAO,GAAG,aAAa,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;YAC3C,CAAC;QACF,CAAC;aAAM,CAAC;YACP,MAAM,KAAK,GAAG,QAAQ,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,WAAW,EAAE,EAAE,CAC/D,OAAO,CAAC,IAAI,KAAK,UAAU,CAAC,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CACpD,CAAC;YACF,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACxB,MAAM,SAAS,GAAG,eAAe,CAAC,eAAe,CAAC,CAAC;gBACnD,MAAM,OAAO,GAAe,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC,CAAC;oBAC3D,MAAM,EAAE,SAAS;oBACjB,WAAW;oBACX,aAAa,EAAE,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC;iBACvD,CAAC,CAAC,CAAC;gBACJ,OAAO,GAAG;oBACT,GAAG,KAAK;oBACR,EAAE,EAAE,OAAO;oBACX,KAAK,EAAE,EAAE,gBAAgB,EAAE,eAAe,EAAE,aAAa,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE;iBACnF,CAAC;YACH,CAAC;iBAAM,IAAI,QAAQ,CAAC,UAAU,KAAK,SAAS,EAAE,CAAC;gBAC9C,SAAS,GAAG,cAAc,CAAC,QAAQ,EAAE,mDAAmD,CAAC,CAAC;gBAC1F,OAAO,GAAG,aAAa,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;YAC5C,CAAC;iBAAM,CAAC;gBACP,OAAO,GAAG;oBACT,GAAG,KAAK;oBACR,EAAE,EAAE,YAAY;oBAChB,YAAY,EAAE,EAAE,IAAI,EAAE,YAAY,EAAE,qBAAqB,EAAE,IAAI,EAAE;oBACjE,cAAc,EAAE,eAAe;iBAC/B,CAAC;YACH,CAAC;QACF,CAAC;QAED,MAAM,aAAa,GAA2B;YAC7C,EAAE,EAAE,eAAe;YACnB,QAAQ,EAAE,KAAK,CAAC,KAAK;YACrB,IAAI,EAAE,SAAS;YACf,OAAO,EAAE,SAAS;SAClB,CAAC;QACF,MAAM,QAAQ,GAA0B;YACvC,EAAE,EAAE,MAAM,CAAC,OAAO;YAClB,KAAK,EAAE,SAAS,CAAC,KAAK;YACtB,OAAO,EAAE,eAAe;YACxB,UAAU,EAAE,KAAK;SACjB,CAAC;QACF,IAAI,OAAO,KAAK,SAAS,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;YACpD,MAAM,IAAI,qBAAqB,CAAC,gDAAgD,CAAC,CAAC;QACnF,CAAC;QACD,MAAM,MAAM,GACX,OAAO,KAAK,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,qBAAqB,CAAC,IAAI,EAAE,QAAQ,EAAE,eAAe,EAAE,OAAO,CAAC,CAAC;QACrG,MAAM,OAAO,GACZ,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,sBAAsB,CAAC,MAAM,EAAE,KAAK,CAAC,WAAW,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;QAC7G,MAAM,MAAM,GAAG;YACd,WAAW,CAAC,aAAa,CAAC;YAC1B,WAAW,CAAC,QAAQ,CAAC;YACrB,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,eAAe,CAAC;YAC/C,GAAG,CAAC,MAAM,KAAK,SAAS;gBACvB,CAAC,CAAC,CAAC,UAAU,CAAC,sBAAsB,CAAC,KAAK,CAAC,WAAW,EAAE,eAAe,CAAC,CAAC,CAAC;gBAC1E,CAAC,CAAC,OAAO,CAAC;YACX,GAAG,CAAC,OAAO,EAAE,EAAE,KAAK,kBAAkB,IAAI,mBAAmB,KAAK,SAAS;gBAC1E,CAAC,CAAC;oBACA,QAAQ,CACP,oBAAoB,CAAC,KAAK,CAAC,WAAW,EAAE,mBAAmB,CAAC,MAAM,CAAC,EACnE,mBAAmB,CAAC,WAAW,CAC/B;iBACD;gBACF,CAAC,CAAC,EAAE,CAAC;SACN,CAAC;QACF,MAAM,gBAAgB,GAAG,CAAC,MAAoB,EAAgB,EAAE,CAAC,CAAC;YACjE,GAAG,aAAa;YAChB,GAAG,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAE;YACpB,SAAS,EAAE,MAAM,CAAC,SAAS;SAC3B,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,CAAC,MAAoB,EAAkB,EAAE;YACvD,MAAM,KAAK,GAAG,gBAAgB,CAAC,MAAM,CAAC,CAAC;YACvC,MAAM,KAAK,GAAmB;gBAC7B,EAAE,IAAI,EAAE,aAAa,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,GAAG,OAAO,EAAE;gBAC3D;oBACC,IAAI,EAAE,OAAO;oBACb,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,GAAG,EAAE,EAAE,GAAG,QAAQ,EAAE,GAAG,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAE,EAAE;oBAC1C,MAAM,EAAE,MAAM,CAAC,KAAK,CAAC,KAAK;iBAC1B;aACD,CAAC;YACF,IAAI,OAAO,CAAC,EAAE,KAAK,0BAA0B,EAAE,CAAC;gBAC/C,IAAI,OAAO,CAAC,QAAQ,KAAK,IAAI,IAAI,OAAO,CAAC,OAAO,GAAG,CAAC,IAAI,OAAO,EAAE,EAAE,KAAK,sBAAsB,EAAE,CAAC;oBAChG,MAAM,OAAO,GAAG,SAAS,CAAC,UAAU,KAAK,OAAO,IAAI,SAAS,CAAC,UAAU,KAAK,SAAS,CAAC;oBACvF,KAAK,CAAC,IAAI,CAAC;wBACV,IAAI,EAAE,WAAW;wBACjB,IAAI,EAAE,IAAI,CAAC,IAAI;wBACf,KAAK,EAAE,KAAK,CAAC,WAAW;wBACxB,IAAI,EAAE,MAAM;wBACZ,OAAO,EAAE,OAAO,CAAC,OAAO;wBACxB,OAAO;wBACP,GAAG,CAAC,OAAO;4BACV,CAAC,CAAC,EAAE;4BACJ,CAAC,CAAC;gCACA,UAAU,EACT,SAAS,CAAC,YAAY,IAAI,gCAAgC,SAAS,CAAC,UAAU,EAAE;6BACjF,CAAC;qBACJ,CAAC,CAAC;gBACJ,CAAC;gBACD,IAAI,OAAO,CAAC,QAAQ,KAAK,IAAI,IAAI,OAAO,EAAE,EAAE,KAAK,sBAAsB,EAAE,CAAC;oBACzE,KAAK,CAAC,IAAI,CAAC;wBACV,IAAI,EAAE,iBAAiB;wBACvB,IAAI,EAAE,IAAI,CAAC,IAAI;wBACf,KAAK,EAAE,KAAK,CAAC,WAAW;wBACxB,IAAI,EAAE,MAAM;wBACZ,OAAO,EAAE,OAAO,CAAC,WAAW;wBAC5B,WAAW,EAAE,OAAO,CAAC,iBAAiB,CAAC,WAAW,CAAC,WAAW;wBAC9D,OAAO,EAAE,YAAY,CAAC,OAAO,CAAC,iBAAiB,CAAC,WAAW,EAAE,OAAO,CAAC,OAAO,CAAC;wBAC7E,SAAS,EAAE,OAAO,CAAC,SAAS;wBAC5B,YAAY,EAAE,OAAO,CAAC,YAAY;qBAClC,CAAC,CAAC;gBACJ,CAAC;gBACD,IAAI,OAAO,CAAC,QAAQ,KAAK,IAAI,IAAI,OAAO,EAAE,EAAE,KAAK,OAAO,IAAI,OAAO,EAAE,EAAE,KAAK,sBAAsB,EAAE,CAAC;oBACpG,KAAK,CAAC,IAAI,CAAC;wBACV,IAAI,EAAE,UAAU;wBAChB,IAAI,EAAE,IAAI,CAAC,IAAI;wBACf,KAAK,EAAE,KAAK,CAAC,WAAW;wBACxB,MAAM;wBACN,OAAO,EAAE,SAAS;wBAClB,WAAW,EAAE,EAAE;qBACf,CAAC,CAAC;gBACJ,CAAC;gBACD,IAAI,OAAO,EAAE,EAAE,KAAK,kBAAkB,EAAE,CAAC;oBACxC,KAAK,CAAC,IAAI,CAAC;wBACV,IAAI,EAAE,kBAAkB;wBACxB,IAAI,EAAE,IAAI,CAAC,IAAI;wBACf,KAAK,EAAE,KAAK,CAAC,WAAW;wBACxB,MAAM,EAAE,UAAU;wBAClB,SAAS,EAAE,MAAM,CAAC,SAAS;qBAC3B,CAAC,CAAC;gBACJ,CAAC;YACF,CAAC;iBAAM,IAAI,OAAO,EAAE,EAAE,KAAK,OAAO,EAAE,CAAC;gBACpC,KAAK,CAAC,IAAI,CAAC;oBACV,IAAI,EAAE,UAAU;oBAChB,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,KAAK,EAAE,KAAK,CAAC,WAAW;oBACxB,MAAM;oBACN,OAAO,EAAE,SAAS;oBAClB,WAAW,EAAE,EAAE;oBACf,GAAG,OAAO;iBACV,CAAC,CAAC;YACJ,CAAC;YACD,IACC,CAAC,OAAO,CAAC,EAAE,KAAK,yBAAyB,IAAI,OAAO,CAAC,QAAQ,KAAK,IAAI,CAAC;gBACvE,OAAO,EAAE,EAAE,KAAK,oBAAoB;gBACpC,SAAS,CAAC,QAAQ,KAAK,SAAS,EAC/B,CAAC;gBACF,KAAK,CAAC,IAAI,CAAC;oBACV,IAAI,EAAE,aAAa;oBACnB,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,KAAK,EAAE,KAAK,CAAC,WAAW;oBACxB,MAAM,EAAE,UAAU;oBAClB,QAAQ,EAAE,SAAS,CAAC,QAAQ;oBAC5B,IAAI,EAAE,OAAO,CAAC,IAAI;oBAClB,GAAG,CAAC,OAAO,CAAC,EAAE,KAAK,yBAAyB,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;iBAC5D,CAAC,CAAC;YACJ,CAAC;YACD,IAAI,MAAM,KAAK,SAAS,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;gBACnD,KAAK,CAAC,IAAI,CAAC;oBACV,IAAI,EAAE,SAAS;oBACf,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,KAAK,EAAE,KAAK,CAAC,WAAW;oBACxB,MAAM,EAAE,QAAQ;oBAChB,KAAK,EAAE,OAAO;oBACd,SAAS,EAAE,IAAI,CAAC,WAAW;oBAC3B,KAAK,EAAE,eAAe;oBACtB,OAAO,EAAE,MAAM,CAAC,OAAO;iBACvB,CAAC,CAAC;YACJ,CAAC;YACD,OAAO,KAAK,CAAC;QACd,CAAC,CAAC;QACF,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;YAC1B,OAAO;gBACN,IAAI,EAAE,QAAQ;gBACd,MAAM;gBACN,MAAM;gBACN,IAAI,EAAE,EAAE,KAAK,EAAE,eAAe,EAAE;gBAChC,WAAW,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,EAAE,CAAU;gBAClE,MAAM;aACN,CAAC;QACH,CAAC;QACD,IAAI,OAAO,KAAK,SAAS;YAAE,MAAM,IAAI,qBAAqB,CAAC,+CAA+C,CAAC,CAAC;QAC5G,OAAO;YACN,IAAI,EAAE,QAAQ;YACd,MAAM;YACN,cAAc,EAAE,OAAO;YACvB,IAAI,EAAE,EAAE,KAAK,EAAE,eAAe,EAAE;YAChC,WAAW,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,CAAU;YAClD,MAAM;SACN,CAAC;IACH,CAAC,EACD,KAAK,CAAC,OAAO,CACb,CAAC;AACH,CAAC","sourcesContent":["import {\n\ttype AssistantMessageEvent,\n\tisContextOverflow,\n\tisRecoverableLength,\n\tisRetryableAssistantError,\n\tretryDelayMs,\n} from \"@earendil-works/pi-ai\";\nimport { AssistantMessageFrameEncoder } from \"../../../shims/assistant-message-frame.ts\";\nimport type { HarnessEvent } from \"../../agent-harness.ts\";\nimport type { Context } from \"../../context.ts\";\nimport type { AssistantResponseMetadata, AssistantStreamObserver } from \"../../execution/assistant.ts\";\nimport { insertEntry, insertUsage } from \"../../session/commit.ts\";\nimport { SessionInvariantError } from \"../../session/session.ts\";\nimport {\n\ttype AssistantEffectPendingOperation,\n\ttype CommitResult,\n\ttype DeferredEffectPendingOperation,\n\ttype MessageEntry,\n\ttype NewEntry,\n\ttype OperationError,\n\ttype OperationState,\n\toperationScopeOf,\n\ttype SettledAssistantMessage,\n\ttype SummaryDecidingOperation,\n\ttype ToolCall,\n\ttype UsageRow,\n} from \"../../session/types.ts\";\nimport { branchTip, deleteList, operationPreparation, pendingAssistantFrames, setValue } from \"../../session/values.ts\";\nimport type { Lane } from \"../lane.ts\";\nimport { openFrameProgress } from \"../progress.ts\";\nimport type { Drive, ProcedureResult } from \"../types.ts\";\nimport { retryNotBefore } from \"./retry.ts\";\nimport { prepareOverflowCompaction } from \"./structural.ts\";\nimport { operationCleanupWrites, operationResultRecord } from \"./terminal.ts\";\n\nexport type AssistantResponseLifecycle = {\n\tobserver: AssistantStreamObserver;\n\tafterResponse(\n\t\tmessage: SettledAssistantMessage,\n\t\tmetadata: AssistantResponseMetadata,\n\t\tcontext: Context,\n\t): Promise<SettledAssistantMessage>;\n\tclose(): Promise<void>;\n};\n\nexport function openAssistantResponse<TContext extends object | undefined>(\n\tlane: Lane<TContext>,\n\tdrive: Drive,\n\tresponseEntryId: string,\n\trecovery = false,\n): AssistantResponseLifecycle {\n\tconst progress = openFrameProgress(lane, drive, responseEntryId);\n\tconst frameEncoder = new AssistantMessageFrameEncoder();\n\tconst eventContext = {\n\t\tlane: lane.name,\n\t\trunId: drive.operationId,\n\t\t...(recovery ? { recovery: true as const } : {}),\n\t};\n\tconst close = async () => {\n\t\tprogress.seal();\n\t\tawait progress.drain();\n\t};\n\treturn {\n\t\tobserver: {\n\t\t\tstart(message, event, context) {\n\t\t\t\tconst frame = frameEncoder.encode(event);\n\t\t\t\tif (frame !== undefined) progress.write(frame);\n\t\t\t\treturn lane.emitBatch([{ type: \"message_start\", ...eventContext, message }], context);\n\t\t\t},\n\t\t\tupdate(message, event: AssistantMessageEvent, context) {\n\t\t\t\tconst frame = frameEncoder.encode(event);\n\t\t\t\tif (frame !== undefined) progress.write(frame);\n\t\t\t\treturn lane.emitBatch(\n\t\t\t\t\t[{ type: \"message_update\", ...eventContext, message, event, ...(frame === undefined ? {} : { frame }) }],\n\t\t\t\t\tcontext,\n\t\t\t\t);\n\t\t\t},\n\t\t\tend(message, context) {\n\t\t\t\treturn lane.emitBatch(\n\t\t\t\t\t[{ type: \"message_end\", ...eventContext, message, entryId: responseEntryId }],\n\t\t\t\t\tcontext,\n\t\t\t\t);\n\t\t\t},\n\t\t},\n\t\tasync afterResponse(message, metadata, context) {\n\t\t\tawait close();\n\t\t\tconst result = await lane.hooks.runWithGate(\n\t\t\t\t\"after_response\",\n\t\t\t\t{ lane: lane.name, runId: drive.operationId, ...metadata, message },\n\t\t\t\tdrive.gate,\n\t\t\t\tcontext,\n\t\t\t);\n\t\t\treturn result?.message ?? message;\n\t\t},\n\t\tclose,\n\t};\n}\n\ntype ResponseIntent = AssistantEffectPendingOperation | DeferredEffectPendingOperation;\ntype ConfigurationFailureState = Extract<\n\tOperationState,\n\t{ at: \"assistant.ready\" | \"assistant.retry_wait\" | \"deferred.suspended\" | \"deferred.effect_pending\" }\n>;\n\n/** Publish a non-retryable request-configuration failure before reserving response ids. */\nexport async function publishConfigurationFailure<\n\tTContext extends object | undefined,\n\tTState extends ConfigurationFailureState,\n>(lane: Lane<TContext>, drive: Drive, capability: TState, error: OperationError): Promise<ProcedureResult> {\n\tconst result = await lane.continueOperation(\n\t\tcapability,\n\t\tasync (state, current, meta, reader) => {\n\t\t\tif (state.tipId === null) throw new SessionInvariantError(\"Failed run has no Branch tip\");\n\t\t\tconst record = operationResultRecord(meta, \"failed\", state.tipId, error);\n\t\t\tconst cleanup = await operationCleanupWrites(reader, drive.operationId, current, drive.context);\n\t\t\treturn {\n\t\t\t\tkind: \"finish\",\n\t\t\t\twrites: cleanup,\n\t\t\t\trecord,\n\t\t\t\tmaterialize: () => ({ kind: \"settled\", outcome: record }) as const,\n\t\t\t\tevents: () => [\n\t\t\t\t\t{\n\t\t\t\t\t\ttype: \"run_end\",\n\t\t\t\t\t\tlane: lane.name,\n\t\t\t\t\t\trunId: drive.operationId,\n\t\t\t\t\t\tstatus: \"failed\",\n\t\t\t\t\t\terror,\n\t\t\t\t\t\tfromTipId: meta.sourceTipId,\n\t\t\t\t\t\ttipId: state.tipId,\n\t\t\t\t\t\tendedAt: record.endedAt,\n\t\t\t\t\t},\n\t\t\t\t],\n\t\t\t};\n\t\t},\n\t\tdrive.context,\n\t);\n\treturn result.kind === \"cancel_requested\" ? { kind: \"continue\" } : result.value;\n}\n\nfunction uuidV7Timestamp(id: string): number {\n\tconst timestamp = Number.parseInt(id.slice(0, 8) + id.slice(9, 13), 16);\n\tif (!Number.isSafeInteger(timestamp)) throw new SessionInvariantError(`Invalid reserved UUIDv7 ${id}`);\n\treturn timestamp;\n}\n\nfunction providerError(source: \"assistant\" | \"deferred\", message: SettledAssistantMessage): OperationError {\n\treturn {\n\t\tcode: \"assistant_error\",\n\t\tmessage:\n\t\t\tmessage.errorMessage ??\n\t\t\t`${source === \"assistant\" ? \"Assistant\" : \"Deferred\"} request ended with ${message.stopReason}`,\n\t};\n}\n\nfunction normalizeError(message: SettledAssistantMessage, errorMessage: string): SettledAssistantMessage {\n\treturn { ...message, stopReason: \"error\", errorMessage };\n}\n\nfunction normalizeAborted(source: \"assistant\" | \"deferred\", message: SettledAssistantMessage): SettledAssistantMessage {\n\treturn {\n\t\t...message,\n\t\tstopReason: \"aborted\",\n\t\terrorMessage:\n\t\t\tmessage.errorMessage ?? `${source === \"assistant\" ? \"Assistant\" : \"Deferred\"} request was cancelled`,\n\t};\n}\n\nfunction deferredHandleIsValid(message: SettledAssistantMessage, generation: AssistantEffectPendingOperation): boolean {\n\tconst handle = message.deferred;\n\tconst identity = generation.generationContext.configuration.model;\n\treturn (\n\t\tmessage.stopReason === \"deferred\" &&\n\t\thandle !== undefined &&\n\t\thandle.id.length !== 0 &&\n\t\thandle.provider === identity.provider &&\n\t\thandle.modelId === identity.modelId &&\n\t\thandle.api === message.api\n\t);\n}\n\n/** Classify and atomically settle one assistant-generation or deferred-poll response. */\nexport async function publishResponse<TContext extends object | undefined>(\n\tlane: Lane<TContext>,\n\tdrive: Drive,\n\tintent: ResponseIntent,\n\tresponse: SettledAssistantMessage,\n\toptions: { recovery?: true } = {},\n): Promise<ProcedureResult> {\n\tconst overflow =\n\t\tintent.at === \"assistant.effect_pending\" &&\n\t\t(isContextOverflow(response, intent.contextWindow) || isRecoverableLength(response, intent.intendedOutputLimit));\n\tconst overflowPreparation =\n\t\toverflow && !intent.generationContext.overflowRecoveryUsed\n\t\t\t? await prepareOverflowCompaction(lane, drive, intent)\n\t\t\t: undefined;\n\treturn lane.settleOperation<ResponseIntent, ProcedureResult>(\n\t\tintent,\n\t\tasync (state, current, meta, reader) => {\n\t\t\tconst source = current.at === \"assistant.effect_pending\" ? \"assistant\" : \"deferred\";\n\t\t\tconst responseEntryId = intent.responseEntryId;\n\t\t\tconst configuration =\n\t\t\t\tcurrent.at === \"assistant.effect_pending\" ? current.generationContext.configuration : current.configuration;\n\t\t\tconst turnId =\n\t\t\t\tcurrent.at === \"assistant.effect_pending\"\n\t\t\t\t\t? current.generationContext.stepId\n\t\t\t\t\t: `${current.stepId}:poll:${current.poll}`;\n\t\t\tconst scope = { ...operationScopeOf(current), latestAssistantEntryId: responseEntryId };\n\t\t\tlet committed = response;\n\t\t\tlet settled: OperationState | undefined;\n\t\t\tlet failure: OperationError | undefined;\n\n\t\t\tif (current.control.status === \"cancel_requested\") {\n\t\t\t\tcommitted = normalizeAborted(source, response);\n\t\t\t\tsettled = {\n\t\t\t\t\t...scope,\n\t\t\t\t\tat: \"checkpoint\",\n\t\t\t\t\tcontinuation: { kind: \"may_finish\", includeFinalAssistant: true },\n\t\t\t\t\ttriggerEntryId: responseEntryId,\n\t\t\t\t};\n\t\t\t} else if (response.stopReason === \"aborted\") {\n\t\t\t\tthrow new SessionInvariantError(\n\t\t\t\t\t`${source === \"assistant\" ? \"Assistant\" : \"Deferred\"} response is aborted while durable control is running`,\n\t\t\t\t);\n\t\t\t} else if (current.at === \"assistant.effect_pending\" && overflow) {\n\t\t\t\tcommitted = normalizeError(\n\t\t\t\t\tresponse,\n\t\t\t\t\tresponse.errorMessage ?? \"Assistant request exceeded the context window\",\n\t\t\t\t);\n\t\t\t\tif (current.generationContext.overflowRecoveryUsed || overflowPreparation === undefined) {\n\t\t\t\t\tfailure = providerError(source, committed);\n\t\t\t\t} else {\n\t\t\t\t\tconst structural: SummaryDecidingOperation = {\n\t\t\t\t\t\t...scope,\n\t\t\t\t\t\tat: \"summary.deciding\",\n\t\t\t\t\t\ttask: {\n\t\t\t\t\t\t\ttaskId: overflowPreparation.taskId,\n\t\t\t\t\t\t\treason: \"overflow\",\n\t\t\t\t\t\t\tboundary: {\n\t\t\t\t\t\t\t\tkind: \"resume_checkpoint\",\n\t\t\t\t\t\t\t\tresumeAfter: {\n\t\t\t\t\t\t\t\t\tcontinuation: { kind: \"need_assistant\", overflowRecoveryUsed: true },\n\t\t\t\t\t\t\t\t\ttriggerEntryId: current.generationContext.triggerEntryId,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t};\n\t\t\t\t\tsettled = structural;\n\t\t\t\t}\n\t\t\t} else if (response.stopReason === \"deferred\") {\n\t\t\t\tif (current.at === \"assistant.effect_pending\") {\n\t\t\t\t\tif (deferredHandleIsValid(response, current)) {\n\t\t\t\t\t\tsettled = {\n\t\t\t\t\t\t\t...scope,\n\t\t\t\t\t\t\tat: \"deferred.suspended\",\n\t\t\t\t\t\t\tstepId: current.generationContext.stepId,\n\t\t\t\t\t\t\tsourceEntryId: responseEntryId,\n\t\t\t\t\t\t\tpoll: 0,\n\t\t\t\t\t\t\tconfiguration,\n\t\t\t\t\t\t\tstreamOptions: current.generationContext.streamOptions,\n\t\t\t\t\t\t};\n\t\t\t\t\t} else {\n\t\t\t\t\t\tcommitted = normalizeError(response, \"Provider returned an invalid deferred handle\");\n\t\t\t\t\t\tfailure = providerError(source, committed);\n\t\t\t\t\t}\n\t\t\t\t} else {\n\t\t\t\t\tsettled = {\n\t\t\t\t\t\t...scope,\n\t\t\t\t\t\tat: \"deferred.suspended\",\n\t\t\t\t\t\tstepId: current.stepId,\n\t\t\t\t\t\tsourceEntryId: responseEntryId,\n\t\t\t\t\t\tpoll: current.poll,\n\t\t\t\t\t\tconfiguration,\n\t\t\t\t\t\tstreamOptions: current.streamOptions,\n\t\t\t\t\t};\n\t\t\t\t}\n\t\t\t} else if (response.stopReason === \"error\") {\n\t\t\t\tif (\n\t\t\t\t\tcurrent.at === \"assistant.effect_pending\" &&\n\t\t\t\t\t(options.recovery === true || isRetryableAssistantError(response)) &&\n\t\t\t\t\tcurrent.attempt < current.generationContext.retryPolicy.maxAttempts\n\t\t\t\t) {\n\t\t\t\t\tsettled = {\n\t\t\t\t\t\t...scope,\n\t\t\t\t\t\tat: \"assistant.retry_wait\",\n\t\t\t\t\t\tgenerationContext: current.generationContext,\n\t\t\t\t\t\tnextAttempt: current.attempt + 1,\n\t\t\t\t\t\tnotBefore: retryNotBefore(current.generationContext.retryPolicy, current.attempt),\n\t\t\t\t\t\terrorMessage: response.errorMessage ?? \"Assistant request failed\",\n\t\t\t\t\t};\n\t\t\t\t} else {\n\t\t\t\t\tfailure = providerError(source, response);\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tconst calls = response.content.flatMap((content, sourceIndex) =>\n\t\t\t\t\tcontent.type === \"toolCall\" ? [{ sourceIndex }] : [],\n\t\t\t\t);\n\t\t\t\tif (calls.length !== 0) {\n\t\t\t\t\tconst timestamp = uuidV7Timestamp(responseEntryId);\n\t\t\t\t\tconst planned: ToolCall[] = calls.map(({ sourceIndex }) => ({\n\t\t\t\t\t\tstatus: \"planned\",\n\t\t\t\t\t\tsourceIndex,\n\t\t\t\t\t\tresultEntryId: lane.session.idGenerator.next(timestamp),\n\t\t\t\t\t}));\n\t\t\t\t\tsettled = {\n\t\t\t\t\t\t...scope,\n\t\t\t\t\t\tat: \"tools\",\n\t\t\t\t\t\tbatch: { assistantEntryId: responseEntryId, configuration, turnId, calls: planned },\n\t\t\t\t\t};\n\t\t\t\t} else if (response.stopReason === \"toolUse\") {\n\t\t\t\t\tcommitted = normalizeError(response, \"Provider reported tool use without any tool calls\");\n\t\t\t\t\tfailure = providerError(source, committed);\n\t\t\t\t} else {\n\t\t\t\t\tsettled = {\n\t\t\t\t\t\t...scope,\n\t\t\t\t\t\tat: \"checkpoint\",\n\t\t\t\t\t\tcontinuation: { kind: \"may_finish\", includeFinalAssistant: true },\n\t\t\t\t\t\ttriggerEntryId: responseEntryId,\n\t\t\t\t\t};\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tconst responseEntry: NewEntry<MessageEntry> = {\n\t\t\t\tid: responseEntryId,\n\t\t\t\tparentId: state.tipId,\n\t\t\t\ttype: \"message\",\n\t\t\t\tmessage: committed,\n\t\t\t};\n\t\t\tconst usageRow: Omit<UsageRow, \"seq\"> = {\n\t\t\t\tid: intent.usageId,\n\t\t\t\tusage: committed.usage,\n\t\t\t\tentryId: responseEntryId,\n\t\t\t\tadjustment: false,\n\t\t\t};\n\t\t\tif (settled === undefined && failure === undefined) {\n\t\t\t\tthrow new SessionInvariantError(\"Response settlement has no durable disposition\");\n\t\t\t}\n\t\t\tconst record =\n\t\t\t\tfailure === undefined ? undefined : operationResultRecord(meta, \"failed\", responseEntryId, failure);\n\t\t\tconst cleanup =\n\t\t\t\trecord === undefined ? [] : await operationCleanupWrites(reader, drive.operationId, current, drive.context);\n\t\t\tconst writes = [\n\t\t\t\tinsertEntry(responseEntry),\n\t\t\t\tinsertUsage(usageRow),\n\t\t\t\tsetValue(branchTip(lane.name), responseEntryId),\n\t\t\t\t...(record === undefined\n\t\t\t\t\t? [deleteList(pendingAssistantFrames(drive.operationId, responseEntryId))]\n\t\t\t\t\t: cleanup),\n\t\t\t\t...(settled?.at === \"summary.deciding\" && overflowPreparation !== undefined\n\t\t\t\t\t? [\n\t\t\t\t\t\t\tsetValue(\n\t\t\t\t\t\t\t\toperationPreparation(drive.operationId, overflowPreparation.taskId),\n\t\t\t\t\t\t\t\toverflowPreparation.preparation,\n\t\t\t\t\t\t\t),\n\t\t\t\t\t\t]\n\t\t\t\t\t: []),\n\t\t\t];\n\t\t\tconst materializeEntry = (commit: CommitResult): MessageEntry => ({\n\t\t\t\t...responseEntry,\n\t\t\t\tseq: commit.seqs[0]!,\n\t\t\t\ttimestamp: commit.timestamp,\n\t\t\t});\n\t\t\tconst events = (commit: CommitResult): HarnessEvent[] => {\n\t\t\t\tconst entry = materializeEntry(commit);\n\t\t\t\tconst batch: HarnessEvent[] = [\n\t\t\t\t\t{ type: \"entry_added\", lane: lane.name, entry, ...options },\n\t\t\t\t\t{\n\t\t\t\t\t\ttype: \"usage\",\n\t\t\t\t\t\tlane: lane.name,\n\t\t\t\t\t\trow: { ...usageRow, seq: commit.seqs[1]! },\n\t\t\t\t\t\ttotals: commit.stats.usage,\n\t\t\t\t\t},\n\t\t\t\t];\n\t\t\t\tif (current.at === \"assistant.effect_pending\") {\n\t\t\t\t\tif (options.recovery !== true && current.attempt > 1 && settled?.at !== \"assistant.retry_wait\") {\n\t\t\t\t\t\tconst success = committed.stopReason !== \"error\" && committed.stopReason !== \"aborted\";\n\t\t\t\t\t\tbatch.push({\n\t\t\t\t\t\t\ttype: \"retry_end\",\n\t\t\t\t\t\t\tlane: lane.name,\n\t\t\t\t\t\t\trunId: drive.operationId,\n\t\t\t\t\t\t\tstep: turnId,\n\t\t\t\t\t\t\tattempt: current.attempt,\n\t\t\t\t\t\t\tsuccess,\n\t\t\t\t\t\t\t...(success\n\t\t\t\t\t\t\t\t? {}\n\t\t\t\t\t\t\t\t: {\n\t\t\t\t\t\t\t\t\t\tfinalError:\n\t\t\t\t\t\t\t\t\t\t\tcommitted.errorMessage ?? `Assistant request ended with ${committed.stopReason}`,\n\t\t\t\t\t\t\t\t\t}),\n\t\t\t\t\t\t});\n\t\t\t\t\t}\n\t\t\t\t\tif (options.recovery !== true && settled?.at === \"assistant.retry_wait\") {\n\t\t\t\t\t\tbatch.push({\n\t\t\t\t\t\t\ttype: \"retry_scheduled\",\n\t\t\t\t\t\t\tlane: lane.name,\n\t\t\t\t\t\t\trunId: drive.operationId,\n\t\t\t\t\t\t\tstep: turnId,\n\t\t\t\t\t\t\tattempt: settled.nextAttempt,\n\t\t\t\t\t\t\tmaxAttempts: settled.generationContext.retryPolicy.maxAttempts,\n\t\t\t\t\t\t\tdelayMs: retryDelayMs(current.generationContext.retryPolicy, current.attempt),\n\t\t\t\t\t\t\tnotBefore: settled.notBefore,\n\t\t\t\t\t\t\terrorMessage: settled.errorMessage,\n\t\t\t\t\t\t});\n\t\t\t\t\t}\n\t\t\t\t\tif (options.recovery !== true && settled?.at !== \"tools\" && settled?.at !== \"assistant.retry_wait\") {\n\t\t\t\t\t\tbatch.push({\n\t\t\t\t\t\t\ttype: \"turn_end\",\n\t\t\t\t\t\t\tlane: lane.name,\n\t\t\t\t\t\t\trunId: drive.operationId,\n\t\t\t\t\t\t\tturnId,\n\t\t\t\t\t\t\tmessage: committed,\n\t\t\t\t\t\t\ttoolResults: [],\n\t\t\t\t\t\t});\n\t\t\t\t\t}\n\t\t\t\t\tif (settled?.at === \"summary.deciding\") {\n\t\t\t\t\t\tbatch.push({\n\t\t\t\t\t\t\ttype: \"compaction_start\",\n\t\t\t\t\t\t\tlane: lane.name,\n\t\t\t\t\t\t\trunId: drive.operationId,\n\t\t\t\t\t\t\treason: \"overflow\",\n\t\t\t\t\t\t\tstartedAt: commit.timestamp,\n\t\t\t\t\t\t});\n\t\t\t\t\t}\n\t\t\t\t} else if (settled?.at !== \"tools\") {\n\t\t\t\t\tbatch.push({\n\t\t\t\t\t\ttype: \"turn_end\",\n\t\t\t\t\t\tlane: lane.name,\n\t\t\t\t\t\trunId: drive.operationId,\n\t\t\t\t\t\tturnId,\n\t\t\t\t\t\tmessage: committed,\n\t\t\t\t\t\ttoolResults: [],\n\t\t\t\t\t\t...options,\n\t\t\t\t\t});\n\t\t\t\t}\n\t\t\t\tif (\n\t\t\t\t\t(current.at === \"deferred.effect_pending\" || options.recovery !== true) &&\n\t\t\t\t\tsettled?.at === \"deferred.suspended\" &&\n\t\t\t\t\tcommitted.deferred !== undefined\n\t\t\t\t) {\n\t\t\t\t\tbatch.push({\n\t\t\t\t\t\ttype: \"run_suspend\",\n\t\t\t\t\t\tlane: lane.name,\n\t\t\t\t\t\trunId: drive.operationId,\n\t\t\t\t\t\treason: \"deferred\",\n\t\t\t\t\t\tdeferred: committed.deferred,\n\t\t\t\t\t\tpoll: settled.poll,\n\t\t\t\t\t\t...(current.at === \"deferred.effect_pending\" ? options : {}),\n\t\t\t\t\t});\n\t\t\t\t}\n\t\t\t\tif (record !== undefined && failure !== undefined) {\n\t\t\t\t\tbatch.push({\n\t\t\t\t\t\ttype: \"run_end\",\n\t\t\t\t\t\tlane: lane.name,\n\t\t\t\t\t\trunId: drive.operationId,\n\t\t\t\t\t\tstatus: \"failed\",\n\t\t\t\t\t\terror: failure,\n\t\t\t\t\t\tfromTipId: meta.sourceTipId,\n\t\t\t\t\t\ttipId: responseEntryId,\n\t\t\t\t\t\tendedAt: record.endedAt,\n\t\t\t\t\t});\n\t\t\t\t}\n\t\t\t\treturn batch;\n\t\t\t};\n\t\t\tif (record !== undefined) {\n\t\t\t\treturn {\n\t\t\t\t\tkind: \"finish\",\n\t\t\t\t\twrites,\n\t\t\t\t\trecord,\n\t\t\t\t\tlane: { tipId: responseEntryId },\n\t\t\t\t\tmaterialize: () => ({ kind: \"settled\", outcome: record }) as const,\n\t\t\t\t\tevents,\n\t\t\t\t};\n\t\t\t}\n\t\t\tif (settled === undefined) throw new SessionInvariantError(\"Response settlement is missing its next state\");\n\t\t\treturn {\n\t\t\t\tkind: \"commit\",\n\t\t\t\twrites,\n\t\t\t\toperationState: settled,\n\t\t\t\tlane: { tipId: responseEntryId },\n\t\t\t\tmaterialize: () => ({ kind: \"continue\" }) as const,\n\t\t\t\tevents,\n\t\t\t};\n\t\t},\n\t\tdrive.context,\n\t);\n}\n"]}

package/dist/harness/runtime/drive/retry.d.ts

@@ -0,0 +1,4 @@
+import { type RetryPolicy } from "@earendil-works/pi-ai";
+export declare function retryNotBefore(policy: Pick<RetryPolicy, "baseDelayMs" | "maxAgentDelayMs">, attempt: number, now?: number): number;
+export declare function waitUntil(notBefore: number, signal: AbortSignal): Promise<void>;
+//# sourceMappingURL=retry.d.ts.map

package/dist/harness/runtime/drive/retry.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"retry.d.ts","sourceRoot":"","sources":["../../../../src/harness/runtime/drive/retry.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,WAAW,EAAgB,MAAM,uBAAuB,CAAC;AAEvE,wBAAgB,cAAc,CAC7B,MAAM,EAAE,IAAI,CAAC,WAAW,EAAE,aAAa,GAAG,iBAAiB,CAAC,EAC5D,OAAO,EAAE,MAAM,EACf,GAAG,SAAa,GACd,MAAM,CAGR;AAED,wBAAgB,SAAS,CAAC,SAAS,EAAE,MAAM,EAAE,MAAM,EAAE,WAAW,GAAG,OAAO,CAAC,IAAI,CAAC,CAwB/E"}

package/dist/harness/runtime/drive/retry.js

@@ -0,0 +1,34 @@
+import { retryDelayMs } from "@earendil-works/pi-ai";
+export function retryNotBefore(policy, attempt, now = Date.now()) {
+    const sum = now + retryDelayMs(policy, attempt);
+    return Number.isSafeInteger(sum) ? sum : Number.MAX_SAFE_INTEGER;
+}
+export function waitUntil(notBefore, signal) {
+    return new Promise((resolve, reject) => {
+        let timer;
+        const cleanup = () => {
+            if (timer !== undefined)
+                clearTimeout(timer);
+            signal.removeEventListener("abort", onAbort);
+        };
+        const onAbort = () => {
+            cleanup();
+            reject(signal.reason);
+        };
+        const check = () => {
+            const remaining = notBefore - Date.now();
+            if (remaining <= 0) {
+                cleanup();
+                resolve();
+                return;
+            }
+            timer = setTimeout(check, Math.min(remaining, 2_147_483_647));
+        };
+        signal.addEventListener("abort", onAbort, { once: true });
+        if (signal.aborted)
+            onAbort();
+        else
+            check();
+    });
+}
+//# sourceMappingURL=retry.js.map

package/dist/harness/runtime/drive/retry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"retry.js","sourceRoot":"","sources":["../../../../src/harness/runtime/drive/retry.ts"],"names":[],"mappings":"AAAA,OAAO,EAAoB,YAAY,EAAE,MAAM,uBAAuB,CAAC;AAEvE,MAAM,UAAU,cAAc,CAC7B,MAA4D,EAC5D,OAAe,EACf,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE;IAEhB,MAAM,GAAG,GAAG,GAAG,GAAG,YAAY,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAChD,OAAO,MAAM,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,CAAC,gBAAgB,CAAC;AAClE,CAAC;AAED,MAAM,UAAU,SAAS,CAAC,SAAiB,EAAE,MAAmB;IAC/D,OAAO,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QAC5C,IAAI,KAAgD,CAAC;QACrD,MAAM,OAAO,GAAG,GAAG,EAAE;YACpB,IAAI,KAAK,KAAK,SAAS;gBAAE,YAAY,CAAC,KAAK,CAAC,CAAC;YAC7C,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QAC9C,CAAC,CAAC;QACF,MAAM,OAAO,GAAG,GAAG,EAAE;YACpB,OAAO,EAAE,CAAC;YACV,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QACvB,CAAC,CAAC;QACF,MAAM,KAAK,GAAG,GAAG,EAAE;YAClB,MAAM,SAAS,GAAG,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YACzC,IAAI,SAAS,IAAI,CAAC,EAAE,CAAC;gBACpB,OAAO,EAAE,CAAC;gBACV,OAAO,EAAE,CAAC;gBACV,OAAO;YACR,CAAC;YACD,KAAK,GAAG,UAAU,CAAC,KAAK,EAAE,IAAI,CAAC,GAAG,CAAC,SAAS,EAAE,aAAa,CAAC,CAAC,CAAC;QAC/D,CAAC,CAAC;QACF,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QAC1D,IAAI,MAAM,CAAC,OAAO;YAAE,OAAO,EAAE,CAAC;;YACzB,KAAK,EAAE,CAAC;IACd,CAAC,CAAC,CAAC;AACJ,CAAC","sourcesContent":["import { type RetryPolicy, retryDelayMs } from \"@earendil-works/pi-ai\";\n\nexport function retryNotBefore(\n\tpolicy: Pick<RetryPolicy, \"baseDelayMs\" | \"maxAgentDelayMs\">,\n\tattempt: number,\n\tnow = Date.now(),\n): number {\n\tconst sum = now + retryDelayMs(policy, attempt);\n\treturn Number.isSafeInteger(sum) ? sum : Number.MAX_SAFE_INTEGER;\n}\n\nexport function waitUntil(notBefore: number, signal: AbortSignal): Promise<void> {\n\treturn new Promise<void>((resolve, reject) => {\n\t\tlet timer: ReturnType<typeof setTimeout> | undefined;\n\t\tconst cleanup = () => {\n\t\t\tif (timer !== undefined) clearTimeout(timer);\n\t\t\tsignal.removeEventListener(\"abort\", onAbort);\n\t\t};\n\t\tconst onAbort = () => {\n\t\t\tcleanup();\n\t\t\treject(signal.reason);\n\t\t};\n\t\tconst check = () => {\n\t\t\tconst remaining = notBefore - Date.now();\n\t\t\tif (remaining <= 0) {\n\t\t\t\tcleanup();\n\t\t\t\tresolve();\n\t\t\t\treturn;\n\t\t\t}\n\t\t\ttimer = setTimeout(check, Math.min(remaining, 2_147_483_647));\n\t\t};\n\t\tsignal.addEventListener(\"abort\", onAbort, { once: true });\n\t\tif (signal.aborted) onAbort();\n\t\telse check();\n\t});\n}\n"]}

package/dist/harness/runtime/drive/structural.d.ts

@@ -0,0 +1,32 @@
+import type { BranchPreparation } from "../../compaction/branch-summarization.ts";
+import type { CompactionPreparation } from "../../compaction/compaction.ts";
+import { type AssistantEffectPendingOperation, type CheckpointOperation, type DurableStructuralPreparation, type NavigationReadyToCommitOperation, type SummaryDecidingOperation, type SummaryEffectPendingOperation, type SummaryReadyOperation, type SummaryRetryWaitOperation } from "../../session/types.ts";
+import type { Lane } from "../lane.ts";
+import type { ContinueOperationResult, Drive, ProcedureResult } from "../types.ts";
+export declare function durableCompactionPreparation(preparation: CompactionPreparation): Extract<DurableStructuralPreparation, {
+    kind: "compaction";
+}>;
+export declare function durableBranchPreparation(preparation: BranchPreparation): Extract<DurableStructuralPreparation, {
+    kind: "branch_summary";
+}>;
+/** Consume one durable structural preparation and decision hook. */
+export declare function runStructuralDecision<TContext extends object | undefined>(lane: Lane<TContext>, drive: Drive, deciding: SummaryDecidingOperation): Promise<ProcedureResult>;
+/** Execute one ready structural generation attempt. */
+export declare function runStructuralGeneration<TContext extends object | undefined>(lane: Lane<TContext>, drive: Drive, ready: SummaryReadyOperation): Promise<ProcedureResult>;
+/** Consume one structural retry wait without starting a provider effect. */
+export declare function runStructuralRetryWait<TContext extends object | undefined>(lane: Lane<TContext>, drive: Drive, retry: SummaryRetryWaitOperation): Promise<ProcedureResult>;
+/** Convert an orphaned structural attempt into a fresh numbered attempt or terminal failure. */
+export declare function recoverStructuralGeneration<TContext extends object | undefined>(lane: Lane<TContext>, drive: Drive, effect: SummaryEffectPendingOperation): Promise<ProcedureResult>;
+/** Prepare threshold compaction only when no newer compaction already guards this trigger. */
+export declare function prepareCompactionThreshold<TContext extends object | undefined>(lane: Lane<TContext>, drive: Drive, checkpoint: CheckpointOperation): Promise<ContinueOperationResult<{
+    taskId: string;
+    preparation: DurableStructuralPreparation;
+} | undefined>>;
+/** Prepare one overflow compaction before the response settlement transaction. */
+export declare function prepareOverflowCompaction<TContext extends object | undefined>(lane: Lane<TContext>, drive: Drive, generation: AssistantEffectPendingOperation): Promise<{
+    taskId: string;
+    preparation: DurableStructuralPreparation;
+} | undefined>;
+/** Atomically move an unsummarized navigation and finish its operation. */
+export declare function commitNavigation<TContext extends object | undefined>(lane: Lane<TContext>, drive: Drive, navigation: NavigationReadyToCommitOperation): Promise<ProcedureResult>;
+//# sourceMappingURL=structural.d.ts.map

package/dist/harness/runtime/drive/structural.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"structural.d.ts","sourceRoot":"","sources":["../../../../src/harness/runtime/drive/structural.ts"],"names":[],"mappings":"AASA,OAAO,KAAK,EAAE,iBAAiB,EAAuB,MAAM,0CAA0C,CAAC;AAEvG,OAAO,KAAK,EAAE,qBAAqB,EAAiC,MAAM,gCAAgC,CAAC;AAO3G,OAAO,EACN,KAAK,+BAA+B,EAGpC,KAAK,mBAAmB,EAIxB,KAAK,4BAA4B,EAGjC,KAAK,gCAAgC,EAMrC,KAAK,wBAAwB,EAC7B,KAAK,6BAA6B,EAClC,KAAK,qBAAqB,EAC1B,KAAK,yBAAyB,EAI9B,MAAM,wBAAwB,CAAC;AAGhC,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAEvC,OAAO,KAAK,EAAE,uBAAuB,EAAE,KAAK,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AA2BnF,wBAAgB,4BAA4B,CAC3C,WAAW,EAAE,qBAAqB,GAChC,OAAO,CAAC,4BAA4B,EAAE;IAAE,IAAI,EAAE,YAAY,CAAA;CAAE,CAAC,CAY/D;AAED,wBAAgB,wBAAwB,CACvC,WAAW,EAAE,iBAAiB,GAC5B,OAAO,CAAC,4BAA4B,EAAE;IAAE,IAAI,EAAE,gBAAgB,CAAA;CAAE,CAAC,CAOnE;AA2fD,oEAAoE;AACpE,wBAAsB,qBAAqB,CAAC,QAAQ,SAAS,MAAM,GAAG,SAAS,EAC9E,IAAI,EAAE,IAAI,CAAC,QAAQ,CAAC,EACpB,KAAK,EAAE,KAAK,EACZ,QAAQ,EAAE,wBAAwB,GAChC,OAAO,CAAC,eAAe,CAAC,CA4D1B;AAiVD,uDAAuD;AACvD,wBAAsB,uBAAuB,CAAC,QAAQ,SAAS,MAAM,GAAG,SAAS,EAChF,IAAI,EAAE,IAAI,CAAC,QAAQ,CAAC,EACpB,KAAK,EAAE,KAAK,EACZ,KAAK,EAAE,qBAAqB,GAC1B,OAAO,CAAC,eAAe,CAAC,CAe1B;AAED,4EAA4E;AAC5E,wBAAsB,sBAAsB,CAAC,QAAQ,SAAS,MAAM,GAAG,SAAS,EAC/E,IAAI,EAAE,IAAI,CAAC,QAAQ,CAAC,EACpB,KAAK,EAAE,KAAK,EACZ,KAAK,EAAE,yBAAyB,GAC9B,OAAO,CAAC,eAAe,CAAC,CAsC1B;AAED,gGAAgG;AAChG,wBAAsB,2BAA2B,CAAC,QAAQ,SAAS,MAAM,GAAG,SAAS,EACpF,IAAI,EAAE,IAAI,CAAC,QAAQ,CAAC,EACpB,KAAK,EAAE,KAAK,EACZ,MAAM,EAAE,6BAA6B,GACnC,OAAO,CAAC,eAAe,CAAC,CAkC1B;AAED,8FAA8F;AAC9F,wBAAsB,0BAA0B,CAAC,QAAQ,SAAS,MAAM,GAAG,SAAS,EACnF,IAAI,EAAE,IAAI,CAAC,QAAQ,CAAC,EACpB,KAAK,EAAE,KAAK,EACZ,UAAU,EAAE,mBAAmB,GAC7B,OAAO,CAAC,uBAAuB,CAAC;IAAE,MAAM,EAAE,MAAM,CAAC;IAAC,WAAW,EAAE,4BAA4B,CAAA;CAAE,GAAG,SAAS,CAAC,CAAC,CA8B7G;AAED,kFAAkF;AAClF,wBAAsB,yBAAyB,CAAC,QAAQ,SAAS,MAAM,GAAG,SAAS,EAClF,IAAI,EAAE,IAAI,CAAC,QAAQ,CAAC,EACpB,KAAK,EAAE,KAAK,EACZ,UAAU,EAAE,+BAA+B,GACzC,OAAO,CAAC;IAAE,MAAM,EAAE,MAAM,CAAC;IAAC,WAAW,EAAE,4BAA4B,CAAA;CAAE,GAAG,SAAS,CAAC,CAWpF;AAED,2EAA2E;AAC3E,wBAAgB,gBAAgB,CAAC,QAAQ,SAAS,MAAM,GAAG,SAAS,EACnE,IAAI,EAAE,IAAI,CAAC,QAAQ,CAAC,EACpB,KAAK,EAAE,KAAK,EACZ,UAAU,EAAE,gCAAgC,GAC1C,OAAO,CAAC,eAAe,CAAC,CA6C1B"}