pi-ui-extend 1.0.42 → 1.0.45

package/dist/app/session/lazy-session-manager.js

@@ -22,6 +22,7 @@ class LazySessionManager {
     labelsById = new Map();
     leafId = null;
     hydrated;
+    contextManager;
     tailEntryCount;
     tailStartOffset = 0;
     constructor(sessionPath, options = {}) {
@@ -74,6 +75,7 @@ class LazySessionManager {
         this.byId.clear();
         this.labelsById.clear();
         this.leafId = null;
+        this.contextManager = undefined;
         mkdirSync(this.sessionDirPath, { recursive: true });
         this.sessionFilePath = join(this.sessionDirPath, `${timestamp.replace(/[:.]/g, "-")}_${sessionId}.jsonl`);
         writeFileSync(this.sessionFilePath, `${JSON.stringify(header)}\n`, "utf8");
@@ -103,6 +105,11 @@ class LazySessionManager {
     getEntries() {
         return this.hydrated?.getEntries() ?? [...this.entries];
     }
+    // Before hydration this is a presentation/file tail, NOT selected ancestry:
+    // it can omit old ancestors and contain abandoned side-branch entries.
+    // Lineage-sensitive callers must use readFullBranchEntries(); retry proofs
+    // must follow parentId, never compare positions in this cache. See
+    // specs/dcp-statistics.md#lazy-tail-retry-invariant and its regression tests.
     getBranch(fromId) {
         if (this.hydrated)
             return this.hydrated.getBranch(fromId);
@@ -161,6 +168,8 @@ class LazySessionManager {
     buildContextEntries() {
         if (this.hydrated)
             return this.hydrated.buildContextEntries();
+        if (this.tailStartOffset > 0)
+            return this.completeContextManager().buildContextEntries();
         const entries = this.contextEntries();
         const byId = new Map(entries.map((entry) => [entry.id, entry]));
         return buildSdkContextEntries(entries, entries.at(-1)?.id ?? null, byId);
@@ -168,6 +177,8 @@ class LazySessionManager {
     buildSessionContext() {
         if (this.hydrated)
             return this.hydrated.buildSessionContext();
+        if (this.tailStartOffset > 0)
+            return this.completeContextManager().buildSessionContext();
         const entries = this.contextEntries();
         const byId = new Map(entries.map((entry) => [entry.id, entry]));
         return buildSessionContext(entries, entries.at(-1)?.id ?? null, byId);
@@ -212,6 +223,7 @@ class LazySessionManager {
             return;
         }
         this.leafId = branchFromId;
+        this.contextManager = undefined;
     }
     resetLeaf() {
         if (this.hydrated) {
@@ -219,6 +231,7 @@ class LazySessionManager {
             return;
         }
         this.leafId = null;
+        this.contextManager = undefined;
     }
     createBranchedSession(leafId) {
         return this.hydrate().createBranchedSession(leafId);
@@ -292,9 +305,29 @@ class LazySessionManager {
     hydrate() {
         if (!this.hydrated) {
             this.hydrated = SessionManager.open(this.sessionFilePath, this.sessionDirPath, this.cwdPath);
+            this.contextManager = undefined;
         }
         return this.hydrated;
     }
+    /**
+     * The tail cache is a presentation/history optimisation, not a valid model
+     * context boundary. If older entries exist, reconstruct provider context from
+     * the complete persisted session while leaving the facade itself lazy so the
+     * UI can continue paging old history on demand.
+     */
+    completeContextManager() {
+        if (!this.contextManager) {
+            const manager = SessionManager.open(this.sessionFilePath, this.sessionDirPath, this.cwdPath);
+            if (this.leafId === null) {
+                manager.resetLeaf();
+            }
+            else if (manager.getLeafId() !== this.leafId && manager.getEntry(this.leafId)) {
+                manager.branch(this.leafId);
+            }
+            this.contextManager = manager;
+        }
+        return this.contextManager;
+    }
     async loadHeaderAsync(cwdOverride) {
         const existingHeader = await readSessionHeaderFast(this.sessionFilePath);
         if (existingHeader)
@@ -335,6 +368,7 @@ class LazySessionManager {
         this.entries.push(entry);
         this.byId.set(entry.id, entry);
         this.leafId = entry.id;
+        this.contextManager = undefined;
         return entry.id;
     }
     newEntry(type, payload) {

package/dist/app/session/session-event-controller.d.ts

@@ -91,6 +91,7 @@ export declare class AppSessionEventController {
     private currentThinkingEntryStartedAt;
     private thinkingElapsedRenderTimer;
     private assistantMessageClosed;
+    private currentAssistantTerminalErrorRendered;
     private assistantTextBuffer;
     constructor(host: AppSessionEventControllerHost);
     snapshotState(): AppSessionEventControllerState;

package/dist/app/session/session-event-controller.js

@@ -1,7 +1,7 @@
 import { isRetryableAssistantError } from "@earendil-works/pi-ai";
 import { createId } from "../id.js";
 import { extractImageContents, renderContent, renderUserMessageContent, stringifyUnknown } from "../rendering/message-content.js";
-import { customMessageEntry, extensionSessionEntry, loadSessionHistoryEntries, loadSessionHistoryEntriesAsync } from "./session-history.js";
+import { assistantTerminalErrorText, customMessageEntry, extensionSessionEntry, loadSessionHistoryEntries, loadSessionHistoryEntriesAsync } from "./session-history.js";
 import { sessionHistoryDisplayMessages, sessionHistoryDisplayMessagesFromEntries, sessionHistoryFullBranchEntries, sessionHistoryOlderMessagesReader } from "./pix-system-message.js";
 import { THINKING_TOOL_NAME } from "../constants.js";
 import { isRecord } from "../guards.js";
@@ -34,6 +34,7 @@ export class AppSessionEventController {
     currentThinkingEntryStartedAt;
     thinkingElapsedRenderTimer;
     assistantMessageClosed = false;
+    currentAssistantTerminalErrorRendered = false;
     assistantTextBuffer = "";
     constructor(host) {
         this.host = host;
@@ -586,6 +587,7 @@ export class AppSessionEventController {
         }
         if (isRecord(message) && message.role === "assistant") {
             this.assistantMessageClosed = false;
+            this.currentAssistantTerminalErrorRendered = false;
             this.clearCurrentAssistantState();
         }
     }
@@ -601,6 +603,12 @@ export class AppSessionEventController {
             this.renderAssistantToolCallsFromMessage(message);
             this.finishCurrentThinkingEntry();
             this.flushAssistantTextBuffer(true);
+            const terminalError = assistantTerminalErrorText(message);
+            if (terminalError && !this.currentAssistantTerminalErrorRendered) {
+                this.addEntry({ id: createId("error"), kind: "error", text: terminalError });
+                this.currentAssistantTerminalErrorRendered = true;
+                this.showManualRetryOnTerminalError(message);
+            }
             this.clearCurrentAssistantState();
             this.assistantMessageClosed = true;
         }
@@ -726,6 +734,7 @@ export class AppSessionEventController {
                 this.host.setSessionActivity(this.host.runtime()?.session.isStreaming ? "running" : "idle");
                 const errorText = assistantEvent.error.errorMessage ?? assistantEvent.reason;
                 this.addEntry({ id: createId("error"), kind: "error", text: errorText });
+                this.currentAssistantTerminalErrorRendered = true;
                 this.showManualRetryOnTerminalError(assistantEvent.error);
                 break;
             }

package/dist/app/session/session-history.d.ts

@@ -41,4 +41,5 @@ export declare function customMessageEntry(message: Record<string, unknown>): En
 export declare function extensionSessionEntry(value: unknown): Extract<Entry, {
     kind: "extension-entry";
 }> | undefined;
+export declare function assistantTerminalErrorText(message: Record<string, unknown>): string | undefined;
 export {};

package/dist/app/session/session-history.js

@@ -4,6 +4,7 @@ import { isOnlyHiddenMetadata } from "../../markdown-format.js";
 import { extractImageContents, renderContent, renderUserMessageContent, stringifyUnknown } from "../rendering/message-content.js";
 import { THINKING_TOOL_NAME } from "../constants.js";
 import { PIX_EXTENSION_ENTRY_ROLE, PIX_SESSION_ENTRY_ID_FIELD, PIX_SYSTEM_DISPLAY_ENTRY_CUSTOM_TYPE, PIX_SYSTEM_MESSAGE_CUSTOM_TYPE, PIX_THINKING_LEVEL_FIELD } from "./pix-system-message.js";
+import { WORKSPACE_MUTATION_ENTRY_TYPE } from "../workspace/workspace-undo.js";
 const HISTORICAL_SUBAGENTS_OBSERVATION = { showSnapshot: false };
 const DEFAULT_HISTORY_CHUNK_SIZE = 50;
 const DEFAULT_HISTORY_TAIL_MESSAGE_COUNT = 80;
@@ -189,7 +190,7 @@ export function customMessageEntry(message) {
 export function extensionSessionEntry(value) {
     if (!isRecord(value) || value.type !== "custom" || typeof value.id !== "string" || typeof value.customType !== "string")
         return undefined;
-    if (value.customType === PIX_SYSTEM_DISPLAY_ENTRY_CUSTOM_TYPE)
+    if (value.customType === PIX_SYSTEM_DISPLAY_ENTRY_CUSTOM_TYPE || value.customType === WORKSPACE_MUTATION_ENTRY_TYPE)
         return undefined;
     return {
         id: `extension-entry-${value.id}`,
@@ -198,6 +199,12 @@ export function extensionSessionEntry(value) {
         expanded: false,
     };
 }
+export function assistantTerminalErrorText(message) {
+    if (message.stopReason !== "error")
+        return undefined;
+    const errorMessage = typeof message.errorMessage === "string" ? message.errorMessage.trim() : "";
+    return errorMessage || "Request failed before the assistant produced a response.";
+}
 function renderAssistantHistoryMessage(message, toolResults, options) {
     const content = message.content;
     if (!Array.isArray(content))
@@ -273,4 +280,8 @@ function renderAssistantHistoryMessage(message, toolResults, options) {
     if (assistantText && !isOnlyHiddenMetadata(assistantText)) {
         options.addEntry({ id: createId("assistant"), kind: "assistant", text: assistantText });
     }
+    const terminalError = assistantTerminalErrorText(message);
+    if (terminalError) {
+        options.addEntry({ id: createId("error"), kind: "error", text: terminalError });
+    }
 }

package/dist/app/session/session-lifecycle-controller.d.ts

@@ -3,7 +3,12 @@ import type { InputEditor } from "../../input-editor.js";
 import type { AppOptions, Entry, PixExtensionUIContext, SessionActivity } from "../types.js";
 export type AppSessionLifecycleHost = {
     options: AppOptions;
-    createRuntime(): Promise<AgentSessionRuntime>;
+    prepareStartupRuntime?(): Promise<{
+        sessionPath?: string;
+    } | null>;
+    createRuntime(options?: {
+        sessionPath?: string;
+    }): Promise<AgentSessionRuntime>;
     entries: Entry[];
     runtime(): AgentSessionRuntime | undefined;
     setRuntime(runtime: AgentSessionRuntime | undefined): void;

package/dist/app/session/session-lifecycle-controller.js

@@ -32,7 +32,16 @@ export class AppSessionLifecycleController {
         });
         try {
             await this.host.loadStartupConfig();
-            const runtime = await this.host.createRuntime();
+            const startupRuntime = await this.host.prepareStartupRuntime?.();
+            if (startupRuntime === null) {
+                this.host.setRuntime(undefined);
+                this.host.setSessionStatus(undefined);
+                this.host.setSessionActivity("idle");
+                await this.host.restoreTabsAfterStartup();
+                this.host.render();
+                return;
+            }
+            const runtime = await this.host.createRuntime(startupRuntime ?? undefined);
             if (!this.host.isRunning()) {
                 await this.host.disposeRuntimeForQuit(runtime);
                 return;

package/dist/app/session/tabs-controller.d.ts

@@ -16,8 +16,11 @@ export type AppTabsControllerHost = {
     readonly maxProjectSessions?: number | (() => number | undefined);
     readonly blinkController: AppBlinkController;
     runtime(): AgentSessionRuntime | undefined;
-    createRuntimeForNewSession(): Promise<AgentSessionRuntime>;
+    createRuntimeForNewSession(modelRef?: string): Promise<AgentSessionRuntime>;
     createRuntimeForSession(sessionPath: string): Promise<AgentSessionRuntime>;
+    draftModelOverrideRef?(): string | undefined;
+    resetDraftModelSelection?(): void;
+    deactivateRuntimeForDraft?(): void;
     awaitCurrentSessionExtensions?(runtime?: AgentSessionRuntime): Promise<void>;
     activateRuntime(runtime: AgentSessionRuntime, options?: BindCurrentSessionOptions): Promise<void>;
     disposeRuntime(runtime: AgentSessionRuntime): Promise<void>;
@@ -43,6 +46,8 @@ export type AppTabsControllerHost = {
     restoreAutoUserMessages?(messages: readonly SubmittedUserMessage[]): void;
     captureDeferredUserMessages?(): readonly SubmittedUserMessage[];
     restoreDeferredUserMessages?(messages: readonly SubmittedUserMessage[]): void;
+    openDraftSessionSelector?(): void;
+    closeDraftSessionSelector?(): void;
     contextInventoryText?(runtime: AgentSessionRuntime, heading: string): string;
     addEntry(entry: Entry): void;
     showToast(message: string, kind: "success" | "error" | "warning" | "info"): void;
@@ -64,6 +69,7 @@ export declare class AppTabsController {
     private readonly tabIdsNeedingHistoryReload;
     private readonly historyInvalidationGenerationByTabId;
     private readonly runtimeOwnershipGenerationByTabId;
+    private readonly touchedDraftTabIds;
     private readonly orphanRuntimeDisposals;
     private readonly lifecycleMutationQueue;
     private activeTabId;
@@ -74,6 +80,9 @@ export declare class AppTabsController {
     private lifecycleGeneration;
     private lifecycleMutationRunning;
     private saveTabsWriteTail;
+    private startupSavedTabs;
+    private startupPrepared;
+    private startupRuntimePlan;
     private restored;
     private retentionCleanupRunning;
     private retentionCleanupScheduled;
@@ -88,6 +97,9 @@ export declare class AppTabsController {
     markTerminalBellAttention(sessionPath: string | undefined): void;
     tabPanelRows(terminalRows: number): number;
     activeInputTabId(): string | undefined;
+    isDraftTabActive(): boolean;
+    openSessionPaths(): readonly string[];
+    noteActiveDraftInputChanged(): void;
     inputStateForTab(tabId: string | undefined): TabInputState | undefined;
     setInputStateForTab(tabId: string | undefined, state: TabInputState): Promise<void>;
     disposeInactiveRuntimes(disposeRuntime?: (runtime: AgentSessionRuntime) => Promise<void>): Promise<void>;
@@ -99,18 +111,28 @@ export declare class AppTabsController {
         save?: boolean;
         force?: boolean;
     }): void;
+    prepareStartupRuntime(): Promise<{
+        sessionPath?: string;
+    } | null>;
     restoreAfterStartup(): Promise<void>;
     private restoreAfterStartupMutation;
     openNewTab(): Promise<void>;
     private openNewTabMutation;
+    materializeActiveDraftTab(): Promise<AgentSessionRuntime | undefined>;
+    private materializeActiveDraftTabMutation;
+    openSessionInActiveDraftTab(sessionPath: string): Promise<boolean>;
+    private openSessionInActiveDraftTabMutation;
+    private ownsDraftLifecycle;
     openSessionInNewTab(sessionPath: string): Promise<boolean>;
     private openSessionInNewTabMutation;
     forkSessionEntryInNewTab(entryId: string): Promise<boolean>;
     private forkSessionEntryInNewTabMutation;
     switchToTab(tabId: string): Promise<void>;
     private switchToTabMutation;
+    private switchDraftToPersistedTab;
     closeTab(tabId: string): Promise<void>;
     private closeTabMutation;
+    private closeDraftTabMutation;
     private replaceLastTabWithNewSession;
     private loadActiveSessionHistory;
     private cancelHistoryLoad;
@@ -118,6 +140,8 @@ export declare class AppTabsController {
     private runtimeForCommand;
     private idleRuntime;
     private activeTab;
+    private createDraftTab;
+    private activateDraftTab;
     private clearStartupTabPlaceholders;
     private storeActiveRuntime;
     private storeActiveSessionView;