@pasko70/pibo 1.9.9 → 1.9.11

package/dist/core/routed-session.js

@@ -0,0 +1,1179 @@
+import { SessionManager, shouldCompact } from "@earendil-works/pi-coding-agent";
+import { getOpenAiCodexProviderUsageForActiveModel } from "../auth/openai-codex-usage.js";
+import { normalizeSessionErrorDetails, runtimeSessionErrorDetails } from "./session-errors.js";
+import { expandInlineSkills } from "./skill-expansion.js";
+import { PIBO_CONTEXT_GUARD_RESUME_MESSAGE_TYPE, PIBO_CONTEXT_GUARD_RESUME_PROMPT, cancelPiboAssistantContextGuardRecovery, claimPiboAssistantContextGuardRecovery, waitForPiboAssistantContextGuardRecovery, } from "./context-guard.js";
+import { PIBO_PROVIDER_RECOVERY_MESSAGE_TYPE, PIBO_PROVIDER_RECOVERY_PROMPT, PiboProviderRecoveryCancelledError, isRetryablePiboAssistantError, isRetryablePiboProviderError, resolvePiboProviderRecoverySettings, waitForPiboProviderRecovery, } from "./provider-recovery.js";
+const FAST_SERVICE_TIER = "priority";
+function modelSupportsFastServiceTier(model) {
+    if (!model)
+        return false;
+    if (model.api === "openai-codex-responses")
+        return true;
+    return model.api === "openai-responses" && (model.provider === "openai" || model.provider === "openai-codex");
+}
+function withFastServiceTier(payload) {
+    if (!payload || typeof payload !== "object" || Array.isArray(payload))
+        return payload;
+    return { ...payload, service_tier: FAST_SERVICE_TIER };
+}
+function withFastServiceTierOption(options) {
+    return { ...(options ?? {}), serviceTier: FAST_SERVICE_TIER };
+}
+function errorMessage(error) {
+    return error instanceof Error ? error.message : String(error);
+}
+function isAssistantMessage(message) {
+    return Boolean(message && typeof message === "object" && message.role === "assistant");
+}
+function numberValue(value) {
+    return typeof value === "number" && Number.isFinite(value) ? value : undefined;
+}
+function stringValue(value) {
+    return typeof value === "string" && value.length > 0 ? value : undefined;
+}
+function parseEmbeddedProviderError(message) {
+    const jsonStart = message.indexOf("{");
+    if (jsonStart === -1)
+        return undefined;
+    try {
+        const parsed = JSON.parse(message.slice(jsonStart));
+        const error = parsed.error;
+        if (!error || typeof error !== "object")
+            return undefined;
+        return {
+            providerType: stringValue(error.type),
+            providerCode: stringValue(error.code),
+            providerParam: stringValue(error.param),
+            providerMessage: stringValue(error.message),
+        };
+    }
+    catch {
+        return undefined;
+    }
+}
+function assistantErrorDetails(message, context) {
+    const rawError = typeof message.errorMessage === "string" ? message.errorMessage : "";
+    const providerError = parseEmbeddedProviderError(rawError);
+    const usage = message.usage && typeof message.usage === "object" ? message.usage : undefined;
+    const contextTokens = numberValue(usage?.totalTokens);
+    const details = normalizeSessionErrorDetails(rawError, {
+        ...(providerError?.providerCode === "context_length_exceeded" ? { category: "context_overflow", errorClass: "provider_context", code: "context_length_exceeded" } : {}),
+        ...providerError,
+        api: stringValue(message.api),
+        provider: stringValue(message.provider),
+        model: stringValue(message.model),
+        contextWindow: context?.contextWindow,
+        contextTokens,
+    });
+    return Object.values(details).some((value) => value !== undefined) ? details : undefined;
+}
+function formatAssistantError(message, context) {
+    const rawError = typeof message.errorMessage === "string" && message.errorMessage.length > 0 ? message.errorMessage : "Assistant message failed.";
+    const details = assistantErrorDetails(message, context);
+    if (details?.category !== "context_overflow")
+        return rawError;
+    const lines = ["Context window exceeded while sending the model request."];
+    if (details.providerMessage)
+        lines.push(`Provider message: ${details.providerMessage}`);
+    if (details.providerCode)
+        lines.push(`Provider code: ${details.providerCode}`);
+    if (details.providerParam)
+        lines.push(`Provider param: ${details.providerParam}`);
+    if (details.api || details.provider || details.model) {
+        lines.push(`Model: ${[details.api, details.provider, details.model].filter(Boolean).join(" / ")}`);
+    }
+    if (details.contextTokens !== undefined || details.contextWindow !== undefined) {
+        const usage = details.contextWindow
+            ? `${details.contextTokens ?? "unknown"} / ${details.contextWindow} tokens`
+            : `${details.contextTokens} tokens`;
+        lines.push(`Context usage: ${usage}`);
+    }
+    lines.push("Automatic overflow compaction will be attempted if auto-compaction is enabled.");
+    return lines.join("\n");
+}
+function messageContentIndex(candidate) {
+    return typeof candidate.assistantMessageEvent?.contentIndex === "number"
+        ? candidate.assistantMessageEvent.contentIndex
+        : undefined;
+}
+function promptSource(source) {
+    return source === "user" || source === "ui" ? "interactive" : "rpc";
+}
+function lastTextPartFromMessage(message) {
+    if (!message || typeof message !== "object")
+        return undefined;
+    const content = message.content;
+    if (!Array.isArray(content))
+        return undefined;
+    for (let index = content.length - 1; index >= 0; index -= 1) {
+        const part = content[index];
+        if (!part || typeof part !== "object")
+            continue;
+        const candidate = part;
+        if (candidate.type === "text" && typeof candidate.text === "string" && candidate.text.length > 0) {
+            return { text: candidate.text, contentIndex: index };
+        }
+    }
+    return undefined;
+}
+function toolCallFromMessage(message, contentIndex) {
+    if (!message || typeof message !== "object" || typeof contentIndex !== "number")
+        return undefined;
+    const content = message.content;
+    if (!Array.isArray(content))
+        return undefined;
+    const candidate = content[contentIndex];
+    if (!candidate || typeof candidate !== "object")
+        return undefined;
+    const toolCall = candidate;
+    if (toolCall.type !== "toolCall" || typeof toolCall.id !== "string" || typeof toolCall.name !== "string") {
+        return undefined;
+    }
+    return { id: toolCall.id, name: toolCall.name, args: toolCall.arguments ?? {} };
+}
+function toolCallFromAssistantEvent(candidate) {
+    const eventToolCall = candidate.assistantMessageEvent?.toolCall;
+    if (eventToolCall && typeof eventToolCall.id === "string" && typeof eventToolCall.name === "string") {
+        return { id: eventToolCall.id, name: eventToolCall.name, args: eventToolCall.arguments ?? {} };
+    }
+    return toolCallFromMessage(candidate.message, candidate.assistantMessageEvent?.contentIndex);
+}
+function normalizeToolCallEvent(piboSessionId, candidate) {
+    if (candidate.type === "message_update" &&
+        (candidate.assistantMessageEvent?.type === "toolcall_start" ||
+            candidate.assistantMessageEvent?.type === "toolcall_end")) {
+        const toolCall = toolCallFromAssistantEvent(candidate);
+        if (!toolCall)
+            return undefined;
+        return {
+            type: "tool_call",
+            piboSessionId,
+            toolCallId: toolCall.id,
+            toolName: toolCall.name,
+            args: toolCall.args,
+            argsComplete: candidate.assistantMessageEvent.type === "toolcall_end",
+        };
+    }
+    return undefined;
+}
+function normalizeToolExecutionEvent(piboSessionId, candidate) {
+    if (typeof candidate.toolCallId !== "string" || typeof candidate.toolName !== "string") {
+        return undefined;
+    }
+    if (candidate.type === "tool_execution_start") {
+        return {
+            type: "tool_execution_started",
+            piboSessionId,
+            toolCallId: candidate.toolCallId,
+            toolName: candidate.toolName,
+            args: candidate.args,
+        };
+    }
+    if (candidate.type === "tool_execution_update") {
+        return {
+            type: "tool_execution_updated",
+            piboSessionId,
+            toolCallId: candidate.toolCallId,
+            toolName: candidate.toolName,
+            args: candidate.args,
+            partialResult: candidate.partialResult,
+        };
+    }
+    if (candidate.type === "tool_execution_end") {
+        return {
+            type: "tool_execution_finished",
+            piboSessionId,
+            toolCallId: candidate.toolCallId,
+            toolName: candidate.toolName,
+            result: candidate.result,
+            isError: candidate.isError === true,
+        };
+    }
+    return undefined;
+}
+function normalizeProviderWebSearchEvent(piboSessionId, candidate) {
+    const item = recordValue(candidate.item) ?? recordValue(candidate.assistantMessageEvent?.item);
+    const itemType = stringValue(item?.type);
+    const rawEventType = stringValue(candidate.type) ?? stringValue(candidate.assistantMessageEvent?.type);
+    const rawType = rawEventType?.toLowerCase().includes("web_search") ? rawEventType : itemType;
+    if (!rawType || !rawType.toLowerCase().includes("web_search"))
+        return undefined;
+    const action = recordValue(candidate.action) ?? recordValue(candidate.assistantMessageEvent?.action) ?? recordValue(item?.action);
+    const status = providerWebSearchStatus(rawEventType ?? rawType, candidate.status ?? candidate.assistantMessageEvent?.status ?? item?.status);
+    if (!status)
+        return undefined;
+    const rawId = stringValue(candidate.toolCallId) ??
+        stringValue(candidate.item_id) ??
+        stringValue(item?.id) ??
+        stringValue(candidate.assistantMessageEvent?.toolCall?.id) ??
+        (typeof candidate.assistantMessageEvent?.contentIndex === "number"
+            ? `content-${candidate.assistantMessageEvent.contentIndex}`
+            : undefined) ??
+        "active";
+    const toolCallId = rawId.startsWith("provider:web_search:") ? rawId : `provider:web_search:${rawId}`;
+    const query = stringValue(candidate.query) ??
+        stringValue(candidate.assistantMessageEvent?.query) ??
+        stringValue(action?.query) ??
+        stringValue(item?.query) ??
+        stringValue(item?.search_query);
+    const sources = firstArray(candidate.sources, candidate.assistantMessageEvent?.sources, item?.sources, item?.results, item?.citations);
+    const args = { providerTool: "web_search", ...(query ? { query } : {}) };
+    if (status === "running") {
+        return {
+            type: "tool_execution_started",
+            piboSessionId,
+            toolCallId,
+            toolName: "web_search",
+            args,
+        };
+    }
+    const error = candidate.error ?? candidate.assistantMessageEvent?.error ?? item?.error;
+    return {
+        type: "tool_execution_finished",
+        piboSessionId,
+        toolCallId,
+        toolName: "web_search",
+        result: status === "error"
+            ? (error ?? "Web search failed")
+            : {
+                ...(query ? { query } : {}),
+                ...(sources ? { sources, sourceCount: sources.length } : {}),
+            },
+        isError: status === "error",
+    };
+}
+function providerWebSearchStatus(rawType, rawStatus) {
+    const type = rawType.toLowerCase();
+    const status = stringValue(rawStatus)?.toLowerCase();
+    if (type.includes("failed") || type.includes("error") || status === "failed" || status === "error")
+        return "error";
+    if (type.includes("completed") || type.includes("done") || type.includes("end") || status === "completed" || status === "done")
+        return "done";
+    if (type.includes("started") || type.includes("added") || type.includes("in_progress") || type.includes("searching") || status === "in_progress" || status === "running" || status === "searching")
+        return "running";
+    return undefined;
+}
+function recordValue(value) {
+    return value && typeof value === "object" && !Array.isArray(value) ? value : undefined;
+}
+function firstArray(...values) {
+    for (const value of values) {
+        if (Array.isArray(value))
+            return value;
+    }
+    return undefined;
+}
+export function normalizePiEvent(piboSessionId, event, context) {
+    if (!event || typeof event !== "object")
+        return undefined;
+    const candidate = event;
+    const providerToolEvent = normalizeProviderWebSearchEvent(piboSessionId, candidate);
+    if (providerToolEvent)
+        return providerToolEvent;
+    if (candidate.type === "message_update" &&
+        candidate.assistantMessageEvent?.type === "text_delta" &&
+        typeof candidate.assistantMessageEvent.delta === "string") {
+        return {
+            type: "assistant_delta",
+            piboSessionId,
+            contentIndex: messageContentIndex(candidate),
+            text: candidate.assistantMessageEvent.delta,
+        };
+    }
+    if (candidate.type === "message_update" &&
+        candidate.assistantMessageEvent?.type === "thinking_start") {
+        return { type: "thinking_started", piboSessionId, contentIndex: messageContentIndex(candidate) };
+    }
+    if (candidate.type === "message_update" &&
+        candidate.assistantMessageEvent?.type === "thinking_delta" &&
+        typeof candidate.assistantMessageEvent.delta === "string") {
+        return { type: "thinking_delta", piboSessionId, contentIndex: messageContentIndex(candidate), text: candidate.assistantMessageEvent.delta };
+    }
+    if (candidate.type === "message_update" && candidate.assistantMessageEvent?.type === "thinking_end") {
+        const text = typeof candidate.assistantMessageEvent.content === "string" ? candidate.assistantMessageEvent.content : undefined;
+        return text === undefined
+            ? { type: "thinking_finished", piboSessionId, contentIndex: messageContentIndex(candidate) }
+            : { type: "thinking_finished", piboSessionId, contentIndex: messageContentIndex(candidate), text };
+    }
+    const toolCallEvent = normalizeToolCallEvent(piboSessionId, candidate);
+    if (toolCallEvent)
+        return toolCallEvent;
+    const toolExecutionEvent = normalizeToolExecutionEvent(piboSessionId, candidate);
+    if (toolExecutionEvent)
+        return toolExecutionEvent;
+    if (candidate.type === "message_end") {
+        const message = candidate.message;
+        const role = message?.role;
+        if (role === "assistant") {
+            if (message?.stopReason === "error" || typeof message?.errorMessage === "string") {
+                return {
+                    type: "session_error",
+                    piboSessionId,
+                    error: formatAssistantError(message, context),
+                    errorDetails: assistantErrorDetails(message, context),
+                };
+            }
+            const textPart = lastTextPartFromMessage(candidate.message);
+            if (textPart) {
+                return {
+                    type: "assistant_message",
+                    piboSessionId,
+                    contentIndex: textPart.contentIndex,
+                    text: textPart.text,
+                };
+            }
+        }
+    }
+    return undefined;
+}
+function estimateContextTokens(messages) {
+    let chars = 0;
+    for (const msg of messages) {
+        if (!msg || typeof msg !== "object")
+            continue;
+        const message = msg;
+        if (message.role === "user") {
+            const content = message.content;
+            if (typeof content === "string") {
+                chars += content.length;
+            }
+            else if (Array.isArray(content)) {
+                for (const block of content) {
+                    if (block && typeof block === "object" && "type" in block && block.type === "text" && "text" in block && typeof block.text === "string") {
+                        chars += block.text.length;
+                    }
+                }
+            }
+        }
+        else if (message.role === "assistant") {
+            const content = message.content;
+            if (Array.isArray(content)) {
+                for (const block of content) {
+                    if (!block || typeof block !== "object")
+                        continue;
+                    if ("type" in block) {
+                        if (block.type === "text" && "text" in block && typeof block.text === "string")
+                            chars += block.text.length;
+                        if (block.type === "thinking" && "thinking" in block && typeof block.thinking === "string")
+                            chars += block.thinking.length;
+                        if (block.type === "toolCall" && "name" in block && "arguments" in block) {
+                            chars += String(block.name).length + JSON.stringify(block.arguments).length;
+                        }
+                    }
+                }
+            }
+        }
+        else if (message.role === "toolResult" || message.role === "custom") {
+            const content = message.content;
+            if (typeof content === "string") {
+                chars += content.length;
+            }
+            else if (Array.isArray(content)) {
+                for (const block of content) {
+                    if (block && typeof block === "object" && "type" in block && block.type === "text" && "text" in block && typeof block.text === "string") {
+                        chars += block.text.length;
+                    }
+                    if (block && typeof block === "object" && "type" in block && block.type === "image") {
+                        chars += 4800; // approximate image size
+                    }
+                }
+            }
+        }
+    }
+    return Math.ceil(chars / 4);
+}
+export class RoutedSession {
+    piboSessionId;
+    runtime;
+    emit;
+    pluginRegistry;
+    forwardPiEvents;
+    onPiEventTelemetry;
+    onSessionOperation;
+    onKillChildren;
+    onStateChange;
+    onMessagesInterrupted;
+    queue = [];
+    processing = false;
+    disposed = false;
+    fastMode = false;
+    fastModePatchedAgents = new WeakSet();
+    activeMessage;
+    activeAssistantIndex;
+    nextAssistantIndex = 0;
+    activeThinkingIndex;
+    nextThinkingIndex = 0;
+    pendingAssistantError;
+    pendingAssistantErrorRetryable = false;
+    activeMessageFailed = false;
+    providerRecoveryCancelled = false;
+    providerRecoveryAbortController;
+    unsubscribe;
+    recoverySession;
+    isContinuePatched = false;
+    constructor(piboSessionId, runtime, emit, pluginRegistry, forwardPiEvents, onPiEventTelemetry, initialFastMode, onSessionOperation, onKillChildren, onStateChange, onMessagesInterrupted) {
+        this.piboSessionId = piboSessionId;
+        this.runtime = runtime;
+        this.emit = emit;
+        this.pluginRegistry = pluginRegistry;
+        this.forwardPiEvents = forwardPiEvents;
+        this.onPiEventTelemetry = onPiEventTelemetry;
+        this.onSessionOperation = onSessionOperation;
+        this.onKillChildren = onKillChildren;
+        this.onStateChange = onStateChange;
+        this.onMessagesInterrupted = onMessagesInterrupted;
+        this.fastMode = initialFastMode && this.fastModeSupported();
+        this.bindRuntimeSession();
+        this.patchFastModeProviderRequest();
+        this.patchAgentContinue();
+        this.runtime.setRebindSession(async () => {
+            this.bindRuntimeSession();
+            this.patchFastModeProviderRequest();
+            this.patchAgentContinue();
+        });
+    }
+    patchFastModeProviderRequest() {
+        const agent = this.runtime.session.agent;
+        if (!agent || typeof agent !== "object")
+            return;
+        if (this.fastModePatchedAgents.has(agent))
+            return;
+        this.fastModePatchedAgents.add(agent);
+        const originalStreamFn = typeof agent.streamFn === "function" ? agent.streamFn.bind(agent) : undefined;
+        if (originalStreamFn) {
+            agent.streamFn = (model, context, options) => {
+                const nextOptions = this.shouldUseFastServiceTier(model) ? withFastServiceTierOption(options) : options;
+                return originalStreamFn(model, context, nextOptions);
+            };
+        }
+        const originalOnPayload = agent.onPayload?.bind(agent);
+        agent.onPayload = async (payload, model) => {
+            const useFastTier = this.shouldUseFastServiceTier(model);
+            const payloadForHooks = useFastTier ? withFastServiceTier(payload) : payload;
+            const nextPayload = originalOnPayload ? await originalOnPayload(payloadForHooks, model) : payloadForHooks;
+            return useFastTier ? withFastServiceTier(nextPayload) : nextPayload;
+        };
+    }
+    shouldUseFastServiceTier(model) {
+        if (this.getFastModeResult().mode !== "fast")
+            return false;
+        return modelSupportsFastServiceTier(model) || modelSupportsFastServiceTier(this.runtime.session.model);
+    }
+    patchAgentContinue() {
+        if (this.isContinuePatched)
+            return;
+        this.isContinuePatched = true;
+        const agent = this.runtime.session.agent;
+        if (!agent)
+            return;
+        const originalContinue = agent.continue.bind(agent);
+        const session = this.runtime.session;
+        agent.continue = async () => {
+            const model = session.model;
+            if (!model)
+                return originalContinue();
+            const contextWindow = model.contextWindow ?? 0;
+            const settings = session.settingsManager.getCompactionSettings();
+            if (!settings.enabled)
+                return originalContinue();
+            let contextTokens = null;
+            const contextUsage = session.getContextUsage();
+            if (contextUsage && contextUsage.tokens !== null) {
+                contextTokens = contextUsage.tokens;
+            }
+            else {
+                // Fallback: estimate tokens from current messages when no post-compaction
+                // assistant usage is available yet. This prevents context overflow between
+                // tool calls after a compaction.
+                contextTokens = estimateContextTokens(session.messages);
+            }
+            if (contextTokens === null)
+                return originalContinue();
+            if (shouldCompact(contextTokens, contextWindow, settings)) {
+                await session.compact();
+            }
+            return originalContinue();
+        };
+    }
+    bindRuntimeSession() {
+        this.unsubscribe?.();
+        const session = this.runtime.session;
+        if (this.recoverySession && this.recoverySession !== session) {
+            this.cancelProviderRecovery();
+            cancelPiboAssistantContextGuardRecovery(this.recoverySession, new Error("Context guard recovery cancelled because the Pi session changed"));
+        }
+        this.recoverySession = session;
+        claimPiboAssistantContextGuardRecovery(session);
+        this.unsubscribe = session.subscribe((event) => {
+            this.onPiEventTelemetry?.(this.piboSessionId, event, { status: this.getStatus(), activeEventId: this.activeMessage?.id });
+            const model = this.runtime.session.model;
+            const normalized = normalizePiEvent(this.piboSessionId, event, { contextWindow: numberValue(model?.contextWindow) });
+            const candidate = event && typeof event === "object" ? event : undefined;
+            const assistantMessageEnded = candidate?.type === "message_end" && isAssistantMessage(candidate.message);
+            // Pi gets the first chance to recover through its short retry/compaction loop.
+            // Keep the final error pending so the routed turn can continue durable recovery.
+            if (assistantMessageEnded && normalized?.type === "session_error") {
+                this.pendingAssistantError = this.withActiveMessage(normalized);
+                this.pendingAssistantErrorRetryable = isRetryablePiboAssistantError(candidate.message);
+            }
+            else {
+                if (assistantMessageEnded) {
+                    this.pendingAssistantError = undefined;
+                    this.pendingAssistantErrorRetryable = false;
+                }
+                if (normalized)
+                    this.emit(this.withActiveMessage(normalized));
+            }
+            if (this.forwardPiEvents) {
+                this.emit({ type: "pi_event", piboSessionId: this.piboSessionId, event });
+            }
+            this.handleCompactionEvent(event);
+        });
+    }
+    flushPendingAssistantError() {
+        if (!this.pendingAssistantError)
+            return;
+        this.activeMessageFailed = true;
+        this.emit(this.pendingAssistantError);
+        this.pendingAssistantError = undefined;
+        this.pendingAssistantErrorRetryable = false;
+    }
+    cancelProviderRecovery() {
+        this.providerRecoveryCancelled = true;
+        this.providerRecoveryAbortController?.abort();
+        this.providerRecoveryAbortController = undefined;
+    }
+    async resumeContextGuardRecovery(session) {
+        while (await waitForPiboAssistantContextGuardRecovery(session)) {
+            try {
+                await session.sendCustomMessage({
+                    customType: PIBO_CONTEXT_GUARD_RESUME_MESSAGE_TYPE,
+                    content: [{ type: "text", text: PIBO_CONTEXT_GUARD_RESUME_PROMPT }],
+                    display: false,
+                }, { triggerTurn: true });
+            }
+            catch (error) {
+                const resumeError = error instanceof Error ? error : new Error(String(error));
+                cancelPiboAssistantContextGuardRecovery(session, resumeError);
+                throw resumeError;
+            }
+        }
+    }
+    async recoverTransientProviderErrors(session) {
+        let attempt = 0;
+        while (this.pendingAssistantError && this.pendingAssistantErrorRetryable) {
+            if (this.providerRecoveryCancelled)
+                throw new PiboProviderRecoveryCancelledError();
+            const settings = resolvePiboProviderRecoverySettings(session.settingsManager);
+            if (!settings.enabled)
+                return;
+            attempt += 1;
+            this.pendingAssistantError = undefined;
+            this.pendingAssistantErrorRetryable = false;
+            const controller = new AbortController();
+            this.providerRecoveryAbortController = controller;
+            try {
+                await waitForPiboProviderRecovery(attempt, settings, controller.signal);
+            }
+            finally {
+                if (this.providerRecoveryAbortController === controller) {
+                    this.providerRecoveryAbortController = undefined;
+                }
+            }
+            if (this.providerRecoveryCancelled || this.disposed || this.runtime.session !== session || !this.activeMessage) {
+                throw new PiboProviderRecoveryCancelledError();
+            }
+            try {
+                await session.sendCustomMessage({
+                    customType: PIBO_PROVIDER_RECOVERY_MESSAGE_TYPE,
+                    content: [{ type: "text", text: PIBO_PROVIDER_RECOVERY_PROMPT }],
+                    display: false,
+                    details: { attempt },
+                }, { triggerTurn: true });
+                await this.resumeContextGuardRecovery(session);
+                if (this.providerRecoveryCancelled)
+                    throw new PiboProviderRecoveryCancelledError();
+            }
+            catch (error) {
+                if (error instanceof PiboProviderRecoveryCancelledError)
+                    throw error;
+                if (!isRetryablePiboProviderError(error))
+                    throw error;
+                const message = errorMessage(error);
+                this.pendingAssistantError = {
+                    type: "session_error",
+                    piboSessionId: this.piboSessionId,
+                    eventId: this.activeMessage.id,
+                    error: message,
+                    errorDetails: runtimeSessionErrorDetails(message),
+                };
+                this.pendingAssistantErrorRetryable = true;
+            }
+        }
+    }
+    handleCompactionEvent(event) {
+        if (!event || typeof event !== "object")
+            return;
+        const candidate = event;
+        if (candidate.type === "compaction_start") {
+            this.emit({
+                type: "compaction_start",
+                piboSessionId: this.piboSessionId,
+                reason: typeof candidate.reason === "string" ? candidate.reason : "unknown",
+            });
+        }
+        if (candidate.type === "compaction_end") {
+            if (candidate.result && candidate.aborted !== true) {
+                // Reset assistant message indices so the next assistant response starts
+                // fresh after compaction, matching the reduced agent context.
+                this.activeAssistantIndex = undefined;
+                this.nextAssistantIndex = 0;
+                this.activeThinkingIndex = undefined;
+                this.nextThinkingIndex = 0;
+            }
+            this.emit({
+                type: "compaction_end",
+                piboSessionId: this.piboSessionId,
+                reason: typeof candidate.reason === "string" ? candidate.reason : "unknown",
+                result: candidate.result,
+                aborted: candidate.aborted === true,
+                errorMessage: typeof candidate.errorMessage === "string" ? candidate.errorMessage : undefined,
+            });
+        }
+    }
+    enqueueMessage(event) {
+        this.assertActive();
+        this.queue.push({ kind: "message", event });
+        const output = {
+            type: "message_queued",
+            piboSessionId: this.piboSessionId,
+            eventId: event.id,
+            queuedMessages: this.queue.length,
+            text: event.text,
+            source: event.source,
+        };
+        this.emit(output);
+        this.onStateChange?.({ processing: this.processing, queuedMessages: this.queue.length, disposed: this.disposed });
+        void this.drain();
+        return output;
+    }
+    async executeAction(event) {
+        this.assertActive();
+        if (event.action === "compact") {
+            return this.enqueueCompactAction(event);
+        }
+        const result = await this.runAction(event);
+        if (isSessionOperationResult(result))
+            await this.onSessionOperation?.(result, event);
+        const output = {
+            type: "execution_result",
+            piboSessionId: this.piboSessionId,
+            eventId: event.id,
+            action: event.action,
+            result,
+        };
+        this.emit(output);
+        return output;
+    }
+    getStatus() {
+        const enabledTools = this.runtime.session.getActiveToolNames();
+        const thinkingLevel = this.runtime.session.thinkingLevel;
+        const settingsManager = this.runtime.session.settingsManager;
+        return {
+            piboSessionId: this.piboSessionId,
+            queuedMessages: this.queue.length,
+            processing: this.processing,
+            streaming: this.runtime.session.isStreaming,
+            activeTools: enabledTools,
+            enabledTools,
+            cwd: this.runtime.cwd,
+            disposed: this.disposed,
+            thinkingLevel,
+            fastMode: this.getFastModeResult().mode === "fast",
+            ...(settingsManager ? {
+                retry: {
+                    ...settingsManager.getRetrySettings(),
+                    provider: settingsManager.getProviderRetrySettings(),
+                },
+            } : {}),
+        };
+    }
+    getContextUsage() {
+        return this.runtime.session.getContextUsage();
+    }
+    getActiveModel() {
+        const model = this.runtime.session.model;
+        return model ? { provider: model.provider, id: model.id } : undefined;
+    }
+    async getProviderUsage() {
+        try {
+            return await getOpenAiCodexProviderUsageForActiveModel(this.getActiveModel());
+        }
+        catch {
+            return undefined;
+        }
+    }
+    removeQueuedMessages(predicate) {
+        this.assertActive();
+        const removedMessages = [];
+        for (let index = this.queue.length - 1; index >= 0; index -= 1) {
+            const item = this.queue[index];
+            if (item.kind !== "message" || !predicate(item.event))
+                continue;
+            this.queue.splice(index, 1);
+            removedMessages.push(item.event);
+        }
+        removedMessages.reverse();
+        this.notifyMessagesInterrupted(removedMessages, "queued message removed");
+        return removedMessages.length;
+    }
+    getCurrentSession() {
+        return this.createSessionSnapshot();
+    }
+    async listSessions() {
+        const manager = this.runtime.session.sessionManager;
+        const sessions = await SessionManager.list(this.runtime.cwd, manager.getSessionDir());
+        return sessions.map((session) => ({
+            path: session.path,
+            id: session.id,
+            cwd: session.cwd,
+            name: session.name,
+            parentSessionPath: session.parentSessionPath,
+            created: session.created.toISOString(),
+            modified: session.modified.toISOString(),
+            messageCount: session.messageCount,
+            firstMessage: session.firstMessage,
+        }));
+    }
+    getForkCandidates() {
+        return this.runtime.session.getUserMessagesForForking();
+    }
+    async forkSession(entryId) {
+        this.assertActive();
+        const previous = this.createSessionSnapshot();
+        const result = await this.runtime.fork(entryId);
+        return {
+            piboSessionId: this.piboSessionId,
+            previous,
+            current: this.createSessionSnapshot(),
+            cancelled: result.cancelled,
+            selectedText: result.selectedText,
+        };
+    }
+    async cloneSession() {
+        this.assertActive();
+        const leafId = this.runtime.session.sessionManager.getLeafId();
+        if (!leafId) {
+            throw new Error("Cannot clone session: no current entry selected");
+        }
+        const previous = this.createSessionSnapshot();
+        const result = await this.runtime.fork(leafId, { position: "at" });
+        return {
+            piboSessionId: this.piboSessionId,
+            previous,
+            current: this.createSessionSnapshot(),
+            cancelled: result.cancelled,
+        };
+    }
+    getSessionTree() {
+        this.assertActive();
+        return {
+            current: this.createSessionSnapshot(),
+            tree: normalizeSessionTree(this.runtime.session.sessionManager.getTree()),
+        };
+    }
+    async navigateSessionTree(params) {
+        this.assertActive();
+        const previous = this.createSessionSnapshot();
+        const result = await this.runtime.session.navigateTree(params.entryId, {
+            summarize: params.summarize,
+            customInstructions: params.customInstructions,
+            replaceInstructions: params.replaceInstructions,
+            label: params.label,
+        });
+        return {
+            piboSessionId: this.piboSessionId,
+            previous,
+            current: this.createSessionSnapshot(),
+            cancelled: result.cancelled,
+            editorText: result.editorText,
+            summaryEntryId: result.summaryEntry?.id,
+        };
+    }
+    async switchSession(params) {
+        this.assertActive();
+        const previous = this.createSessionSnapshot();
+        const result = await this.runtime.switchSession(params.sessionFile, { cwdOverride: params.cwdOverride });
+        return {
+            piboSessionId: this.piboSessionId,
+            previous,
+            current: this.createSessionSnapshot(),
+            cancelled: result.cancelled,
+        };
+    }
+    async setModel(model) {
+        this.assertActive();
+        const resolved = this.runtime.session.modelRegistry.find(model.provider, model.id);
+        if (!resolved)
+            throw new Error(`Unknown model ${model.provider}/${model.id}`);
+        await this.runtime.session.setModel(resolved);
+        return { provider: resolved.provider, id: resolved.id };
+    }
+    setThinkingLevel(level) {
+        this.assertActive();
+        this.runtime.session.setThinkingLevel(level);
+        return this.getThinkingResult();
+    }
+    cycleThinkingLevel() {
+        this.assertActive();
+        this.runtime.session.cycleThinkingLevel();
+        return this.getThinkingResult();
+    }
+    getFastMode() {
+        this.assertActive();
+        return this.getFastModeResult();
+    }
+    setFastMode(enabled) {
+        this.assertActive();
+        const before = this.getFastModeResult().mode;
+        if (this.fastModeSupported())
+            this.fastMode = enabled;
+        const current = this.getFastModeResult();
+        return { ...current, changed: before !== current.mode };
+    }
+    async compact(customInstructions) {
+        this.assertActive();
+        return await this.runtime.session.compact(customInstructions);
+    }
+    async dispose() {
+        if (this.disposed)
+            return;
+        this.notifyMessagesInterrupted(this.activeAndQueuedMessages(), "session disposed");
+        this.cancelProviderRecovery();
+        this.queue.length = 0;
+        this.onStateChange?.({ processing: this.processing, queuedMessages: this.queue.length, disposed: true });
+        this.unsubscribe?.();
+        this.unsubscribe = undefined;
+        if (this.recoverySession) {
+            this.cancelContextGuardRecovery("Context guard recovery cancelled because the routed session was disposed");
+            this.recoverySession = undefined;
+        }
+        this.disposed = true;
+        await this.runtime.dispose();
+    }
+    async kill() {
+        this.notifyMessagesInterrupted(this.activeAndQueuedMessages(), "session killed");
+        this.cancelProviderRecovery();
+        this.queue.length = 0;
+        this.onStateChange?.({ processing: this.processing, queuedMessages: this.queue.length, disposed: this.disposed });
+        this.cancelContextGuardRecovery("Context guard recovery cancelled because the routed session was killed");
+        await this.runtime.session.abort();
+        return this.piboSessionId;
+    }
+    async cancelMessage(eventId) {
+        this.assertActive();
+        const queuedIndex = this.queue.findIndex((item) => item.event.id === eventId);
+        if (queuedIndex >= 0) {
+            const [removed] = this.queue.splice(queuedIndex, 1);
+            if (removed?.kind === "message")
+                this.notifyMessagesInterrupted([removed.event], "message cancelled");
+            this.onStateChange?.({ processing: this.processing, queuedMessages: this.queue.length, disposed: this.disposed });
+            return true;
+        }
+        if (this.activeMessage?.id === eventId) {
+            this.notifyMessagesInterrupted([this.activeMessage], "message cancelled");
+            this.cancelProviderRecovery();
+            this.cancelContextGuardRecovery("Context guard recovery cancelled with the active message");
+            await this.runtime.session.abort();
+            return true;
+        }
+        return false;
+    }
+    async drain() {
+        if (this.processing || this.disposed)
+            return;
+        this.processing = true;
+        this.onStateChange?.({ processing: this.processing, queuedMessages: this.queue.length, disposed: this.disposed });
+        try {
+            while (this.queue.length > 0 && !this.disposed) {
+                const item = this.queue.shift();
+                this.onStateChange?.({ processing: this.processing, queuedMessages: this.queue.length, disposed: this.disposed });
+                if (item.kind === "compact") {
+                    await this.processQueuedCompact(item.event);
+                }
+                else {
+                    await this.processQueuedMessage(item.event);
+                }
+            }
+        }
+        finally {
+            this.processing = false;
+            this.onStateChange?.({ processing: this.processing, queuedMessages: this.queue.length, disposed: this.disposed });
+        }
+    }
+    async processQueuedMessage(event) {
+        this.emit({
+            type: "message_started",
+            piboSessionId: this.piboSessionId,
+            eventId: event.id,
+            text: event.text,
+            source: event.source,
+        });
+        try {
+            this.activeMessage = event;
+            this.providerRecoveryCancelled = false;
+            this.pendingAssistantError = undefined;
+            this.pendingAssistantErrorRetryable = false;
+            this.activeMessageFailed = false;
+            this.activeAssistantIndex = undefined;
+            this.nextAssistantIndex = 0;
+            this.activeThinkingIndex = undefined;
+            this.nextThinkingIndex = 0;
+            const session = this.runtime.session;
+            const expandedText = expandInlineSkills(event.text, session.resourceLoader.getSkills().skills);
+            await session.prompt(expandedText, { source: promptSource(event.source) });
+            await this.resumeContextGuardRecovery(session);
+            await this.recoverTransientProviderErrors(session);
+            this.flushPendingAssistantError();
+            if (!this.activeMessageFailed) {
+                this.emit({
+                    type: "message_finished",
+                    piboSessionId: this.piboSessionId,
+                    eventId: event.id,
+                    source: event.source,
+                });
+            }
+        }
+        catch (error) {
+            if (error instanceof PiboProviderRecoveryCancelledError)
+                return;
+            const message = errorMessage(error);
+            this.emit({
+                type: "session_error",
+                piboSessionId: this.piboSessionId,
+                eventId: event.id,
+                error: message,
+                errorDetails: runtimeSessionErrorDetails(message),
+            });
+        }
+        finally {
+            this.activeMessage = undefined;
+            this.providerRecoveryCancelled = false;
+            this.pendingAssistantError = undefined;
+            this.pendingAssistantErrorRetryable = false;
+            this.activeMessageFailed = false;
+            this.activeAssistantIndex = undefined;
+            this.nextAssistantIndex = 0;
+            this.activeThinkingIndex = undefined;
+            this.nextThinkingIndex = 0;
+        }
+    }
+    async processQueuedCompact(event) {
+        try {
+            const result = await this.runAction(event);
+            this.emit({
+                type: "execution_result",
+                piboSessionId: this.piboSessionId,
+                eventId: event.id,
+                action: event.action,
+                result,
+            });
+        }
+        catch (error) {
+            const message = errorMessage(error);
+            this.emit({
+                type: "session_error",
+                piboSessionId: this.piboSessionId,
+                eventId: event.id,
+                error: message,
+                errorDetails: runtimeSessionErrorDetails(message),
+            });
+        }
+    }
+    enqueueCompactAction(event) {
+        this.queue.push({ kind: "compact", event });
+        const output = {
+            type: "execution_result",
+            piboSessionId: this.piboSessionId,
+            eventId: event.id,
+            action: event.action,
+            result: { queued: true, queuedMessages: this.queue.length },
+        };
+        this.emit(output);
+        this.onStateChange?.({ processing: this.processing, queuedMessages: this.queue.length, disposed: this.disposed });
+        void this.drain();
+        return output;
+    }
+    async runAction(event) {
+        const action = event.action;
+        const gatewayAction = this.pluginRegistry.getGatewayAction(action);
+        if (!gatewayAction) {
+            throw new Error(`Unknown execution action "${action}"`);
+        }
+        return await gatewayAction.execute({
+            piboSessionId: this.piboSessionId,
+            getStatus: () => this.getStatus(),
+            getContextUsage: () => this.getContextUsage(),
+            getActiveModel: () => this.getActiveModel(),
+            getProviderUsage: () => this.getProviderUsage(),
+            clearQueue: () => this.clearQueue(),
+            abort: async () => {
+                if (this.activeMessage)
+                    this.notifyMessagesInterrupted([this.activeMessage], "abort requested");
+                this.cancelContextGuardRecovery("Context guard recovery cancelled by abort");
+                await this.runtime.session.abort();
+            },
+            dispose: () => this.dispose(),
+            getCurrentSession: () => this.getCurrentSession(),
+            listSessions: () => this.listSessions(),
+            getForkCandidates: () => this.getForkCandidates(),
+            forkSession: (entryId) => this.forkSession(entryId),
+            cloneSession: () => this.cloneSession(),
+            getSessionTree: () => this.getSessionTree(),
+            navigateSessionTree: (params) => this.navigateSessionTree(params),
+            switchSession: (params) => this.switchSession(params),
+            getThinkingLevel: () => this.getThinkingResult(),
+            setThinkingLevel: (level) => this.setThinkingLevel(level),
+            cycleThinkingLevel: () => this.cycleThinkingLevel(),
+            getFastMode: () => this.getFastMode(),
+            setFastMode: (enabled) => this.setFastMode(enabled),
+            setModel: (model) => this.setModel(model),
+            compact: (customInstructions) => this.compact(customInstructions),
+            kill: async () => {
+                const killed = [await this.kill()];
+                let cancelledRuns = [];
+                if (this.onKillChildren) {
+                    const children = await this.onKillChildren(this.piboSessionId);
+                    killed.push(...children.killed);
+                    cancelledRuns = children.cancelledRuns;
+                }
+                return { killed, cancelledRuns };
+            },
+            killAll: async () => {
+                const killed = [await this.kill()];
+                let cancelledRuns = [];
+                if (this.onKillChildren) {
+                    const children = await this.onKillChildren(this.piboSessionId, { includeRuns: true });
+                    killed.push(...children.killed);
+                    cancelledRuns = children.cancelledRuns;
+                }
+                return { killed, cancelledRuns };
+            },
+        }, event);
+    }
+    cancelContextGuardRecovery(message) {
+        const session = this.recoverySession ?? this.runtime.session;
+        cancelPiboAssistantContextGuardRecovery(session, new Error(message));
+        session.abortCompaction?.();
+    }
+    assertActive() {
+        if (this.disposed) {
+            throw new Error(`Session "${this.piboSessionId}" has been disposed`);
+        }
+    }
+    getThinkingResult() {
+        return {
+            level: this.runtime.session.thinkingLevel,
+            availableLevels: this.runtime.session.getAvailableThinkingLevels(),
+            supported: this.runtime.session.supportsThinking(),
+        };
+    }
+    getFastModeResult() {
+        const supported = this.fastModeSupported();
+        return { mode: supported && this.fastMode ? "fast" : "normal", supported };
+    }
+    fastModeSupported() {
+        return this.runtime.session.supportsThinking() && modelSupportsFastServiceTier(this.runtime.session.model);
+    }
+    clearQueue() {
+        const cleared = this.queue.length;
+        const removedMessages = this.queue.flatMap((item) => item.kind === "message" ? [item.event] : []);
+        this.queue.length = 0;
+        this.notifyMessagesInterrupted(removedMessages, "queue cleared");
+        this.onStateChange?.({ processing: this.processing, queuedMessages: this.queue.length, disposed: this.disposed });
+        return cleared;
+    }
+    activeAndQueuedMessages() {
+        const messages = this.queue.flatMap((item) => item.kind === "message" ? [item.event] : []);
+        return this.activeMessage ? [this.activeMessage, ...messages] : messages;
+    }
+    notifyMessagesInterrupted(messages, reason) {
+        if (messages.length === 0)
+            return;
+        this.onMessagesInterrupted?.(messages, reason);
+    }
+    createSessionSnapshot() {
+        const session = this.runtime.session;
+        const manager = session.sessionManager;
+        return {
+            piSessionId: session.sessionId,
+            sessionFile: session.sessionFile,
+            leafId: manager.getLeafId(),
+            cwd: this.runtime.cwd,
+            sessionName: session.sessionName,
+            parentSessionFile: manager.getHeader()?.parentSession,
+        };
+    }
+    withActiveMessage(event) {
+        if (this.activeMessage?.id && event.type === "assistant_delta") {
+            const assistantIndex = this.activeAssistantIndex ?? this.nextAssistantIndex;
+            if (this.activeAssistantIndex === undefined) {
+                this.nextAssistantIndex += 1;
+                this.activeAssistantIndex = assistantIndex;
+            }
+            return { ...event, eventId: this.activeMessage.id, assistantIndex };
+        }
+        if (this.activeMessage?.id && event.type === "assistant_message") {
+            const assistantIndex = this.activeAssistantIndex ?? this.nextAssistantIndex;
+            if (this.activeAssistantIndex === undefined) {
+                this.nextAssistantIndex += 1;
+            }
+            this.activeAssistantIndex = undefined;
+            return { ...event, eventId: this.activeMessage.id, assistantIndex };
+        }
+        if (this.activeMessage?.id && event.type === "thinking_started") {
+            const thinkingIndex = this.nextThinkingIndex;
+            this.nextThinkingIndex += 1;
+            this.activeThinkingIndex = thinkingIndex;
+            return { ...event, eventId: this.activeMessage.id, thinkingIndex };
+        }
+        if (this.activeMessage?.id && (event.type === "thinking_delta" || event.type === "thinking_finished")) {
+            const thinkingIndex = this.activeThinkingIndex ?? this.nextThinkingIndex;
+            if (this.activeThinkingIndex === undefined) {
+                this.nextThinkingIndex += 1;
+                this.activeThinkingIndex = thinkingIndex;
+            }
+            const output = { ...event, eventId: this.activeMessage.id, thinkingIndex };
+            if (event.type === "thinking_finished")
+                this.activeThinkingIndex = undefined;
+            return output;
+        }
+        if (this.activeMessage?.id &&
+            (event.type === "tool_call" ||
+                event.type === "tool_execution_started" ||
+                event.type === "tool_execution_updated" ||
+                event.type === "tool_execution_finished" ||
+                event.type === "session_error" ||
+                event.type === "execution_result")) {
+            return { ...event, eventId: this.activeMessage.id };
+        }
+        return event;
+    }
+}
+function normalizeSessionTree(nodes) {
+    return nodes.map((node) => ({
+        entry: JSON.parse(JSON.stringify(node.entry)),
+        children: normalizeSessionTree(node.children),
+        label: node.label,
+        labelTimestamp: node.labelTimestamp,
+    }));
+}
+function isSessionOperationResult(value) {
+    if (!value || typeof value !== "object")
+        return false;
+    const candidate = value;
+    return (typeof candidate.piboSessionId === "string" &&
+        Boolean(candidate.current) &&
+        typeof candidate.current?.piSessionId === "string");
+}