@pasko70/pibo 1.9.7 → 1.9.9

package/dist/core/routed-session.js

@@ -1,1179 +0,0 @@
-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");
-}