@timurproko/a1 0.1.8-dev.332 → 0.1.8-dev.368

package/dist/integrations/pi/session-ui/session-shell-root.js

@@ -1,6 +1,7 @@
-import { PROMPT_GLYPH, backgroundSgrSpan, caretCell, composeSubmittedPromptRows, displayWidth, faint, formatSubmittedPromptTime, heldNativeHyperlinkStyle, hyperlinkSgrSpan, nativeHyperlinkStyle, overlaySpan, progressStatusText, submittedPromptLayout, stripAnsi, } from "../../../ui/components/index.js";
+import { acceptsTranscriptUpdate } from "../../../contracts/owned-ui/index.js";
+import { backgroundSgrSpan, caretCell, composeSubmittedPromptRows, PromptInput, promptArrow, displayWidth, faint, formatSubmittedPromptTime, heldNativeHyperlinkStyle, hyperlinkSgrSpan, nativeHyperlinkStyle, overlaySpan, progressStatusText, submittedPromptLayout, stripAnsi, } from "../../../ui/components/index.js";
 import { PINNED_PI_HIDDEN_COMMAND_NAMES, PINNED_PI_WORKFLOW_COMMAND_NAMES, } from "../engine/index.js";
-import { createPiExtensionUiBridge, createPiQueuedInputStatus, createPiShellArmin, createPiShellAuthProviderSelector, createPiShellChangelog, createPiShellCollapsedChangelog, createPiShellDaxnuts, createPiShellDialog, createPiShellEarendilAnnouncement, createPiShellEditor, createPiShellExtensionSelector, createPiShellFooter, createPiShellHeader, createPiShellHotkeys, createPiShellLoadedResources, createPiShellLoginDialog, createPiShellModelSelector, createPiShellOperationLoader, createPiShellReloadBox, createPiShellScopedModelsSelector, createPiShellSelector, createPiShellSessionInfo, createPiShellSessionSelector, createPiShellSettingsSelector, createPiShellStatus, createPiShellTranscriptComponent, createPiShellTreeSelector, createPiShellTrustSelector, createPiShellUserMessageSelector, onPiThemeChange, piShellVisibleWidth, piTheme, renderPiShellCommandMessage, renderPiShellPackageUpdateNotice, renderPiShellStartupDiagnostic, renderPiShellStatusText, renderPiShellTranscriptBlock, } from "../components/index.js";
+import { createPiExtensionUiBridge, createPiQueuedInputStatus, createPiShellArmin, createPiShellAuthProviderSelector, createPiShellChangelog, createPiShellCollapsedChangelog, createPiShellDaxnuts, createPiShellDialog, createPiShellEarendilAnnouncement, createPiShellEditor, createPiShellExtensionSelector, createPiShellFooter, createPiShellHeader, createPiShellHotkeys, createPiShellLoadedResources, createPiShellLoginDialog, createPiShellModelSelector, createPiShellOperationLoader, createPiShellReloadBox, createPiShellScopedModelsSelector, createPiShellSelector, createPiShellSessionInfo, createPiShellSessionSelector, createPiShellSettingsSelector, createPiShellStatus, createPiShellTranscriptComponent, createPiShellTreeSelector, createPiShellTrustSelector, createPiShellUserMessageSelector, onPiThemeChange, isPiPromptStyleCompaction, piShellVisibleWidth, piShellTruncateToWidth, piTheme, renderPiShellCommandMessage, renderPiShellPackageUpdateNotice, renderPiShellStartupDiagnostic, renderPiShellStatusText, renderPiShellTranscriptBlock, } from "../components/index.js";
 import { PiTuiRuntimeAdapter, classifyPiTuiInput, } from "../tui-runtime/index.js";
 import { PromptChipStore } from "./prompt-chips.js";
 import { SessionViewportController } from "./session-viewport-controller.js";
@@ -66,7 +67,7 @@ export class OwnedUiSessionShellRoot {
         this.#submittedPromptComposer = this.#customViewport
             ? {
                 layout: submittedPromptLayout,
-                compose: (rows, width, source, style) => composeSubmittedPromptRows(rows, width, source, style)
+                compose: (rows, width, source, style) => composeSubmittedPromptRows(rows, width, source, { ...style, prefix: text => promptArrow(text, piTheme()) })
                     .map(row => nativeHyperlinkStyle(row, nativeTranscriptLinkColor)),
             }
             : undefined;
@@ -77,10 +78,13 @@ export class OwnedUiSessionShellRoot {
         this.#onViewportFrame = handlers.onViewportFrame;
         this.#onInputSurfaceChanged = handlers.onInputSurfaceChanged;
         this.#dockInputReuseEnabled = handlers.enableDockInputReuse ?? true;
-        this.header = createPiShellHeader(startup);
+        this.header = createPiShellHeader({
+            ...startup,
+            ...(this.#customViewport ? { getKeybindings: () => this.editor.keybindingConfig() } : {}),
+        });
         this.resources = createPiShellLoadedResources(startup.resources ?? [], startup.expanded ?? false);
         this.#status = createPiShellStatus(view, progressStatusText, handlers);
-        this.#footer = createPiShellFooter(this.#viewWithExtensionStatuses(view), cwd);
+        this.#footer = createPiShellFooter(this.#viewWithExtensionStatuses(view), cwd, this.#customViewport ? "a1" : "pi");
         this.#queued = createPiQueuedInputStatus(view.editor.queuedSubmissions, this.#customViewport ? "custom-viewport" : "pinned");
         this.editor = createPiShellEditor({
             ...handlers,
@@ -122,7 +126,10 @@ export class OwnedUiSessionShellRoot {
             onToolsExpand: () => this.#setToolsExpanded(!this.#toolsExpanded),
             ...(this.#customViewport ? {
                 promptPresentation: {
-                    prefix: PROMPT_GLYPH,
+                    input: new PromptInput({ fg: (token, text) => piTheme().fg(token, text) }, {
+                        measure: piShellVisibleWidth,
+                        truncate: piShellTruncateToWidth,
+                    }),
                     styleSuggestion: faint,
                     styleSuggestionCaret: caretCell,
                 },
@@ -209,6 +216,12 @@ export class OwnedUiSessionShellRoot {
     prepareHistoryText(text) {
         return this.#promptChips.prepareHistoryText(text);
     }
+    rehydrateHistoryText(text, resolveImage) {
+        return this.#promptChips.rehydrateHistoryText(text, resolveImage);
+    }
+    imageChipAttachmentsFromEditor(text) {
+        return this.#promptChips.imageChipAttachments(text);
+    }
     preparePromptSubmission(text) {
         return this.#promptChips.prepareSubmission(text);
     }
@@ -224,17 +237,26 @@ export class OwnedUiSessionShellRoot {
         this.resetPendingPastes();
         return this.#promptChips.dispose();
     }
-    canPreparePromptSuggestion() {
-        return this.usesDefaultInputSurface()
-            && this.#view.dialog === null
-            && this.#view.overlay === null
-            && this.editor.getText().length === 0;
-    }
-    canPresentPromptSuggestion() {
-        return this.canPreparePromptSuggestion()
-            && this.#view.lifecycle === "ready"
-            && this.editor.canPresentPromptSuggestion();
-    }
+    promptSuggestionPrepareBlockReason() {
+        if (!this.usesDefaultInputSurface())
+            return "replacement-input";
+        if (this.#view.dialog !== null || this.#view.overlay !== null)
+            return "modal";
+        if (this.editor.getText().length > 0)
+            return "draft";
+        return null;
+    }
+    canPreparePromptSuggestion() { return this.promptSuggestionPrepareBlockReason() === null; }
+    promptSuggestionPresentationBlockReason() {
+        const reason = this.promptSuggestionPrepareBlockReason();
+        if (reason !== null)
+            return reason;
+        if (this.#view.lifecycle !== "ready")
+            return "not-ready";
+        return this.editor.promptSuggestionBlockReason?.()
+            ?? (this.editor.canPresentPromptSuggestion() ? null : "presentation-unavailable");
+    }
+    canPresentPromptSuggestion() { return this.promptSuggestionPresentationBlockReason() === null; }
     setPromptSuggestion(text) {
         this.editor.setPromptSuggestion(text);
     }
@@ -250,7 +272,8 @@ export class OwnedUiSessionShellRoot {
         this.#syncTranscript(view.transcript);
         this.editor.setSubmitEnabled(view.lifecycle !== "stopping" && view.lifecycle !== "stopped" && view.lifecycle !== "failed");
         this.editor.setThinkingLevel(view.thinkingLevel);
-        this.invalidate();
+        // Performance: semantic updates already invalidate affected transcript blocks. Chrome is a separate authority.
+        this.#invalidateChrome();
     }
     /**
      * Applies one block: its component is created or updated in place and the order grows
@@ -260,13 +283,12 @@ export class OwnedUiSessionShellRoot {
     applyTranscriptBlock(block) {
         const current = this.#blocksById.get(block.id);
         // Invariant: a full-view/final presentation may preempt queued partials. Never revive an older revision.
-        if (current !== undefined && (current.revision > block.revision || current.status === "finalized" && block.status === "live"))
+        if (current !== undefined && !acceptsTranscriptUpdate(current, block))
             return;
         this.#blocksById.set(block.id, block);
         const component = this.#transcript.get(block.id);
         if (component === undefined) {
-            const created = createPiShellTranscriptComponent(block, this.#cwd, this.#extensionRenderers, this.#submittedPromptComposer, this.#outputPad, !this.#thinkingVisible, this.#mermaidRenderingMode, this.#showImages, this.#imageWidthCells, this.#imageAssets);
-            created.setExpanded(this.#toolsExpanded);
+            const created = this.#mountTranscript(block);
             this.#transcript.set(block.id, created);
             this.#transcriptOrder.push(block.id);
         }
@@ -278,6 +300,20 @@ export class OwnedUiSessionShellRoot {
         this.#renderedRows.delete(block.id);
         this.#documentLayouts.clear();
     }
+    #mountTranscript(block) {
+        const created = createPiShellTranscriptComponent(block, this.#cwd, this.#extensionRenderers, this.#submittedPromptComposer, this.#outputPad, !this.#thinkingVisible, this.#mermaidRenderingMode, this.#showImages, this.#imageWidthCells, this.#imageAssets, { ...this.#componentRuntime, changed: () => {
+                if (this.#transcript.get(block.id) !== created)
+                    return;
+                this.#renderedRows.delete(block.id);
+                this.#documentLayouts.clear();
+                this.#visibleViewportSnapshot = undefined;
+                this.#dockInputCandidate = false;
+                this.#dockInputRevision = undefined;
+                this.#componentRuntime.requestRender();
+            } });
+        created.setExpanded(this.#toolsExpanded);
+        return created;
+    }
     render(width) {
         if (!this.#customViewport) {
             this.#viewportController.setEditorPointerFrame(undefined);
@@ -297,20 +333,27 @@ export class OwnedUiSessionShellRoot {
         const dockInputCandidate = this.#dockInputCandidate;
         // Concurrency: input may change after a keyboard receipt but before its paint.
         // Capture only the live inputs this composition will actually represent.
-        const viewportRevision = this.#viewportController.presentationRevision;
-        const selectionRevision = this.#viewportController.selectionRevision;
-        const pointerPosition = this.#viewportController.pointerPosition;
         const inputSurface = this.#inputSurface;
         const dockRows = dock.rows;
         const selectableDocumentRowCount = document.rows.length;
         const dockStartRow = height - dockRows.length + 1;
         const editorOffset = dock.editorOffset;
+        this.#viewportController.setInputSurfaceFrame(this.usesDefaultInputSurface() ? undefined : {
+            component: this.#inputSurface,
+            columnStart: 1,
+            columnEnd: width,
+            rowStart: Math.max(1, dockStartRow + editorOffset),
+            rowEnd: Math.min(height, dockStartRow + editorOffset + dock.inputRows - 1),
+        });
         this.#viewportController.setEditorPointerFrame(this.usesDefaultInputSurface()
             ? {
                 rowStart: dockStartRow + editorOffset,
                 rowEnd: dockStartRow + editorOffset + dock.inputRows - 1,
             }
             : undefined);
+        const viewportRevision = this.#viewportController.presentationRevision;
+        const selectionRevision = this.#viewportController.selectionRevision;
+        const pointerPosition = this.#viewportController.pointerPosition;
         this.#dockInputCandidate = false;
         let frame = dockInputCandidate
             && snapshot !== undefined
@@ -407,14 +450,31 @@ export class OwnedUiSessionShellRoot {
     noteCompletedAssistantMessage() {
         this.#viewportController.noteCompletedAssistantMessage();
     }
+    /** A new session binding owns new mounts even when it reuses semantic invocation ids. */
+    resetTranscript() {
+        for (const component of this.#transcript.values())
+            component.dispose?.();
+        this.#transcript.clear();
+        this.#blocksById.clear();
+        this.#transcriptOrder = [];
+        this.#renderedRows.clear();
+        this.#documentLayouts.clear();
+    }
     resetViewport() {
         this.#viewportController.reset();
     }
     clearViewportPointerState() {
         this.#viewportController.clearPointerState();
     }
-    handleViewportPreInput(data, allowWheel = true, now = Date.now()) {
-        return this.#viewportController.handlePreInput(data, allowWheel, now);
+    setViewportOverlaySurfaces(surfaces) {
+        this.#viewportController.setOverlaySurfaces(surfaces);
+    }
+    viewportInputGeometryReady(width, height) {
+        const frame = this.#viewportController.frame;
+        return this.#viewportController.inputGeometryReady && frame?.descriptor.width === width && frame.descriptor.height === height;
+    }
+    handleViewportPreInput(data, allowWheel = true, now = Date.now(), editorActive = this.usesDefaultInputSurface()) {
+        return this.#viewportController.handlePreInput(data, allowWheel, now, editorActive);
     }
     #renderDock(width) {
         return this.#renderDockLayout(width).rows;
@@ -425,13 +485,15 @@ export class OwnedUiSessionShellRoot {
         const transientRows = [...queued, ...statusRows];
         const aboveWidgets = this.#renderWidgets("aboveEditor", width);
         const input = this.#inputSurface.render(width);
+        // Invariant: pointer rows describe the body, not the autocomplete block now preceding it.
+        const body = this.usesDefaultInputSurface() ? this.editor.bodyGeometry?.() : undefined;
         const belowWidgets = this.#renderWidgets("belowEditor", width);
         const footer = this.#renderFooter(width);
         const rowsWithoutTransient = [...aboveWidgets, ...input, ...belowWidgets, ...footer];
         return {
             rows: [...transientRows, ...rowsWithoutTransient],
-            editorOffset: transientRows.length + aboveWidgets.length,
-            inputRows: input.length,
+            editorOffset: transientRows.length + aboveWidgets.length + (body?.rowOffset ?? 0),
+            inputRows: body?.rowCount ?? input.length,
         };
     }
     #renderQueued(width) {
@@ -520,14 +582,16 @@ export class OwnedUiSessionShellRoot {
             const block = this.#blocksById.get(id);
             if (!this.#thinkingVisible && block?.kind === "thinking")
                 continue;
+            const promptLike = block !== undefined
+                && (block.kind === "user" || this.#customViewport && isPiPromptStyleCompaction(block));
             const blockWidth = this.#customViewport
-                && block?.kind === "user"
+                && promptLike
                 && this.#viewportController.config.scrollbarAppearance !== "hidden"
                 && width > 1
                 ? width - 1
                 : width;
             const blockRows = this.#blockRows(id, block, blockWidth);
-            if (block?.kind === "user") {
+            if (promptLike) {
                 // Compatibility: the first natural prompt gets one breathing row at the document top.
                 // Once scrolling advances, the prompt itself reaches row zero and then
                 // becomes sticky there, so the spacer is never pinned with it.
@@ -541,7 +605,7 @@ export class OwnedUiSessionShellRoot {
                 liveTailStartRow = firstRow;
             }
             rows.push(...blockRows);
-            if (block?.kind === "user" && blockRows[0] !== undefined) {
+            if (promptLike && blockRows[0] !== undefined) {
                 promptAnchors.push({
                     id: block.id,
                     firstRow,
@@ -596,14 +660,15 @@ export class OwnedUiSessionShellRoot {
             return render();
         let byWidth = this.#renderedRows.get(id);
         const cached = byWidth?.get(width);
-        if (cached?.revision === block.revision)
+        const presentationRevision = component.presentationRevision ?? 0;
+        if (cached?.revision === block.revision && cached.presentationRevision === presentationRevision)
             return cached.rows;
         const rows = render();
         if (byWidth === undefined) {
             byWidth = new Map();
             this.#renderedRows.set(id, byWidth);
         }
-        byWidth.set(width, { revision: block.revision, rows });
+        byWidth.set(width, { revision: block.revision, presentationRevision, rows });
         // Invariant: bare A1 probes full width before reserving the overflowing rail column;
         // retain both widths without turning resize history into an unbounded cache.
         while (byWidth.size > 2)
@@ -619,7 +684,7 @@ export class OwnedUiSessionShellRoot {
             const block = this.#blocksById.get(id);
             if (block === undefined || (!this.#thinkingVisible && block.kind === "thinking"))
                 continue;
-            if (rows.length > 0 && block.kind === "user")
+            if (rows.length > 0 && (block.kind === "user" || this.#customViewport && isPiPromptStyleCompaction(block)))
                 rows.push("");
             rows.push(...this.#blockRows(id, block, width).map(sanitizeExitTranscriptRow));
         }
@@ -830,6 +895,7 @@ export class OwnedUiSessionShellRoot {
             return;
         }
         this.#dockInputCandidate = false;
+        this.#viewportController.invalidateInputSurface();
         this.#inputSurface.setFocused?.(false);
         if (disposePrevious && this.#inputSurface !== this.editor)
             this.#inputSurface.dispose?.();
@@ -852,10 +918,15 @@ export class OwnedUiSessionShellRoot {
         }
     }
     invalidate() {
-        this.header.invalidate();
-        this.resources.invalidate();
+        this.#renderedRows.clear();
+        this.#documentLayouts.clear();
         for (const component of this.#transcript.values())
             component.invalidate();
+        this.#invalidateChrome();
+    }
+    #invalidateChrome() {
+        this.header.invalidate();
+        this.resources.invalidate();
         this.editor.invalidate();
         if (this.#inputSurface !== this.editor)
             this.#inputSurface.invalidate();
@@ -919,8 +990,7 @@ export class OwnedUiSessionShellRoot {
         for (const block of blocks) {
             const component = this.#transcript.get(block.id);
             if (component === undefined) {
-                const created = createPiShellTranscriptComponent(block, this.#cwd, this.#extensionRenderers, this.#submittedPromptComposer, this.#outputPad, !this.#thinkingVisible, this.#mermaidRenderingMode, this.#showImages, this.#imageWidthCells, this.#imageAssets);
-                created.setExpanded(this.#toolsExpanded);
+                const created = this.#mountTranscript(block);
                 this.#transcript.set(block.id, created);
             }
             else if (component.revision !== block.revision) {

package/dist/integrations/pi/session-ui/session-shell.js

@@ -23,6 +23,7 @@ export class OwnedUiSessionShell {
     #unsubscribePromptSuggestions;
     #promptSuggestions;
     #promptHistory = null;
+    #promptHistoryImageSidecar;
     #extensionBridge;
     #stopped;
     #resolveStopped;
@@ -102,7 +103,6 @@ export class OwnedUiSessionShell {
             onEditorChange: () => { this.#editorRevision++; promptSuggestionController?.invalidate(); },
             onPromptSuggestionAccepted: () => promptSuggestionController?.accept(),
             onInputSurfaceChanged: () => {
-                this.root.clearViewportPointerState();
                 promptSuggestionController?.invalidate();
                 this.#promptHistory?.synchronize();
             },
@@ -154,6 +154,7 @@ export class OwnedUiSessionShell {
             root: this.root,
             mode: tuiMode,
             ...(this.#customViewport ? {
+                onOverlayGeometry: surfaces => this.root.setViewportOverlaySurfaces(surfaces),
                 decorateTerminal: (terminal) => {
                     damageTerminal = new DamageAwareTerminalAdapter(terminal, {
                         regionalScroll: process.env.TERM !== "dumb",
@@ -194,8 +195,10 @@ export class OwnedUiSessionShell {
         promptSuggestionController = promptSuggestionOptions === undefined ? null : new ContextualPromptSuggestionController({
             generator: promptSuggestionOptions.generator,
             enabled: promptSuggestionOptions.enabled(),
+            ...(promptSuggestionOptions.diagnostics === undefined ? {} : { diagnostics: promptSuggestionOptions.diagnostics }),
             surface: {
-                canPresent: identity => this.#canPresentPromptSuggestion(identity),
+                canPresent: identity => this.#promptSuggestionPresentationBlockReason(identity) === null,
+                presentationBlockReason: identity => this.#promptSuggestionPresentationBlockReason(identity),
                 present: text => {
                     if (!this.root.canPresentPromptSuggestion())
                         return false;
@@ -251,16 +254,13 @@ export class OwnedUiSessionShell {
                     this.runtime.requestRender();
                     return { consume: true };
                 }
-                // Invariant: an overlay or editor-replacement screen owns its entire pointer
-                // surface. Letting the transcript pre-router inspect those reports
-                // steals settings value menus and numeric +/- controls before the
-                // settings app can receive them.
-                if (this.runtime.hasOverlay() || !this.root.usesDefaultInputSurface()) {
-                    if (data.includes("\u001b[<"))
-                        this.root.clearViewportPointerState();
-                    return undefined;
+                // Invariant: geometry must belong to the painted frame, including newly opened/nested
+                // surfaces. Steady pointer input does not trigger a synchronous composition.
+                const viewport = this.runtime.viewport();
+                if (data.includes("\u001b[<") && !this.root.viewportInputGeometryReady(viewport.columns, viewport.rows)) {
+                    this.runtime.renderNow();
                 }
-                const routed = this.root.handleViewportPreInput(data, true);
+                const routed = this.root.handleViewportPreInput(data, true, Date.now(), this.root.usesDefaultInputSurface() && !this.runtime.hasFocusedOverlay());
                 if (routed.copyText !== undefined) {
                     this.runtime.writeControl(`\u001b]52;c;${Buffer.from(routed.copyText, "utf8").toString("base64")}\u0007`);
                 }
@@ -329,6 +329,8 @@ export class OwnedUiSessionShell {
                 } },
         });
         if (this.#customViewport && options.promptHistory !== undefined) {
+            this.#promptHistoryImageSidecar = options.promptHistory.imageSidecar;
+            const sidecar = this.#promptHistoryImageSidecar;
             this.#promptHistory = new PromptHistoryController({
                 editor: this.root.editor,
                 store: options.promptHistory.store,
@@ -336,6 +338,7 @@ export class OwnedUiSessionShell {
                 fallback: this.view().transcript.flatMap(block => block.kind === "user" ? [block.text] : []),
                 active: () => this.root.usesDefaultInputSurface(),
                 render: () => this.runtime.requestRender(),
+                rehydrate: value => this.root.rehydrateHistoryText(value, id => sidecar?.readAttachment(id) ?? null),
             });
         }
         this.#extensionBridge = createPiExtensionUiBridge({
@@ -385,19 +388,21 @@ export class OwnedUiSessionShell {
             }
             if (event.type === "assistant-message-completed") {
                 this.root.noteCompletedAssistantMessage();
-                if (event.model !== null) {
-                    this.#promptSuggestions?.consider({
-                        sessionId: event.sessionId,
-                        sessionGeneration: event.sessionGeneration,
-                        runSequence: event.runSequence,
-                        responseSequence: event.responseSequence,
-                        model: event.model,
-                    }, event.successful
-                        && event.stopReason === "stop"
-                        && !event.toolContinuation
-                        && event.assistantMessageCount >= 2
-                        && this.root.canPreparePromptSuggestion());
-                }
+                const identity = {
+                    sessionId: event.sessionId,
+                    sessionGeneration: event.sessionGeneration,
+                    runSequence: event.runSequence,
+                    responseSequence: event.responseSequence,
+                    model: event.model,
+                };
+                if (event.model === null)
+                    this.#promptSuggestions?.skip(identity, "no-model");
+                else
+                    this.#promptSuggestions?.consider({ ...identity, model: event.model }, !event.successful ? "failed-response"
+                        : event.toolContinuation || event.stopReason === "toolUse" ? "tool-continuation"
+                            : event.stopReason !== "stop" ? "incomplete-response"
+                                : event.assistantMessageCount < 2 ? "early-conversation"
+                                    : this.root.promptSuggestionPrepareBlockReason());
             }
             const semanticOnly = event.type === "agent-run-started" || event.type === "assistant-message-completed";
             const view = event.type === "transcript-block" && this.#sessionGeneration === this.backend.sessionGeneration
@@ -432,13 +437,14 @@ export class OwnedUiSessionShell {
     view() {
         return this.backend.view();
     }
-    #canPresentPromptSuggestion(identity) {
+    #promptSuggestionPresentationBlockReason(identity) {
         const view = this.view();
-        return !this.#disposed
-            && identity.sessionId === view.sessionId
-            && identity.sessionGeneration === this.backend.sessionGeneration
-            && modelKey(view) === `${identity.model.providerId}/${identity.model.modelId}`
-            && this.root.canPresentPromptSuggestion();
+        if (this.#disposed)
+            return "disposed";
+        if (identity.sessionId !== view.sessionId || identity.sessionGeneration !== this.backend.sessionGeneration
+            || modelKey(view) !== `${identity.model.providerId}/${identity.model.modelId}`)
+            return "stale-identity";
+        return this.root.promptSuggestionPresentationBlockReason();
     }
     damagePresentationDecision() {
         return this.#damageTerminal?.lastDecision ?? null;
@@ -508,8 +514,21 @@ export class OwnedUiSessionShell {
             const reusable = this.root.prepareHistoryText(text);
             if (reusable.length === 0)
                 this.#promptHistory.rememberRecovery(text);
-            else
+            else {
+                this.#persistImageSidecars(text);
                 this.#promptHistory.capture(reusable, kind, this.#cwd, this.backend.sessionId);
+            }
+        }
+    }
+    // Rationale: write the image chip payloads referenced by an editor draft to the sidecar
+    // directory before their history row commits. Missing sidecar dependency = silent skip; the
+    // recall path treats an absent sidecar the same as a corrupt one.
+    #persistImageSidecars(text) {
+        const sidecar = this.#promptHistoryImageSidecar;
+        if (sidecar === undefined)
+            return;
+        for (const record of this.root.imageChipAttachmentsFromEditor(text)) {
+            sidecar.write(record.id, { tag: record.tag, data: record.image.data, mimeType: record.image.mimeType, savedAt: new Date().toISOString() });
         }
     }
     async #submit(text) {
@@ -1304,6 +1323,7 @@ export class OwnedUiSessionShell {
             // Invariant: delivery recovery invalidates callbacks, not same-session local recall or its draft.
             if (this.#sessionBindingGeneration !== this.backend.sessionBindingGeneration) {
                 this.#sessionBindingGeneration = this.backend.sessionBindingGeneration;
+                this.root.resetTranscript();
                 this.#promptHistory?.reset(view.transcript.flatMap(block => block.kind === "user" ? [block.text] : []));
             }
             this.#activeLoginDialog = undefined;

package/dist/integrations/pi/session-ui/session-viewport-controller.d.ts

@@ -1,3 +1,4 @@
+import { type PiTuiPointerSurface } from "../tui-runtime/index.js";
 import type { OwnedUiViewportSettings } from "../../../contracts/owned-ui/index.js";
 import { type TranscriptViewportFrame, type TranscriptViewportFrameInput } from "../../../ui/components/index.js";
 import type { PiShellEditorPort } from "../components/index.js";
@@ -38,6 +39,11 @@ export declare class SessionViewportController {
         readonly rowStart: number;
         readonly rowEnd: number;
     } | undefined): void;
+    /** Replacement input changed before its new bounds have been painted. */
+    invalidateInputSurface(): void;
+    setInputSurfaceFrame(surface: PiTuiPointerSurface | undefined): void;
+    setOverlaySurfaces(surfaces: readonly PiTuiPointerSurface[] | null): void;
+    get inputGeometryReady(): boolean;
     compose(input: Omit<TranscriptViewportFrameInput, "pointerPosition">): TranscriptViewportFrame;
     composeDockOnly(dockRows: readonly string[], width: number, height: number): TranscriptViewportFrame | null;
     /** Returns true when document wrapping may have changed. */
@@ -46,5 +52,5 @@ export declare class SessionViewportController {
     noteCompletedAssistantMessage(): void;
     reset(): void;
     clearPointerState(): void;
-    handlePreInput(data: string, allowWheel?: boolean, now?: number): SessionViewportInputResult;
+    handlePreInput(data: string, allowWheel?: boolean, now?: number, editorActive?: boolean): SessionViewportInputResult;
 }