@averyyy/pi-coding-agent 0.84.1-piclient.2 → 0.84.1-piclient.3

package/dist/modes/interactive/interactive-mode.js

@@ -72,6 +72,7 @@ import { editInExternalEditor } from "./external-editor.js";
 import { getModelSearchText } from "./model-search.js";
 import { getAvailableThemes, getAvailableThemesWithPaths, getEditorTheme, getMarkdownTheme, getThemeByName, onThemeChange, setRegisteredThemes, stopThemeWatcher, Theme, theme, } from "./theme/theme.js";
 import { InteractiveThemeController } from "./theme/theme-controller.js";
+const TRANSCRIPT_LAZY_LOAD_BATCH_SIZE = 20;
 function isExpandable(obj) {
     return typeof obj === "object" && obj !== null && "setExpanded" in obj && typeof obj.setExpanded === "function";
 }
@@ -216,6 +217,11 @@ export class InteractiveMode {
     chatContainer;
     documentContainer;
     transcriptScrollView;
+    transcriptLazyEntries = [];
+    loadedTranscriptEntryIds = new Set();
+    transcriptWindowInitialized = false;
+    transcriptLatestCompactionId;
+    loadingEarlierTranscript = false;
     fullscreenLayoutRoot;
     pendingMessagesContainer;
     statusContainer;
@@ -625,6 +631,7 @@ export class InteractiveMode {
             follow: "end",
             primary: true,
             overscroll: "chain",
+            onScroll: (scrollTop) => this.handleTranscriptScroll(scrollTop),
             scrollbar: this.settingsManager.getFullscreenScrollbar(),
             scrollbarStyle: (text) => theme.bg("scrollbarThumb", text),
         });
@@ -2249,6 +2256,7 @@ export class InteractiveMode {
         this.defaultEditor.onAction("app.session.tree", () => this.showTreeSelector());
         this.defaultEditor.onAction("app.session.fork", () => this.showUserMessageSelector());
         this.defaultEditor.onAction("app.session.resume", () => this.showSessionSelector());
+        this.defaultEditor.onAction("app.session.loadEarlier", () => this.loadEarlierTranscript());
         this.defaultEditor.onChange = (text) => {
             const wasBashMode = this.isBashMode;
             this.isBashMode = text.trimStart().startsWith("!");
@@ -3019,9 +3027,66 @@ export class InteractiveMode {
         this.chatContainer.addChild(new Spacer(1));
         this.chatContainer.addChild(new Text(text, 1, 0));
     }
+    getTranscriptEntriesForRender(reset = false) {
+        const branch = this.sessionManager.getBranch();
+        const contextEntries = this.sessionManager.buildContextEntries();
+        let latestCompactionId;
+        for (let index = branch.length - 1; index >= 0; index--) {
+            const entry = branch[index];
+            if (entry?.type === "compaction") {
+                latestCompactionId = entry.id;
+                break;
+            }
+        }
+        if (reset || (this.transcriptWindowInitialized && this.transcriptLatestCompactionId !== latestCompactionId)) {
+            this.loadedTranscriptEntryIds.clear();
+        }
+        this.transcriptWindowInitialized = true;
+        this.transcriptLatestCompactionId = latestCompactionId;
+        const contextEntryIds = new Set(contextEntries.map((entry) => entry.id));
+        const earlierEntries = branch.filter((entry) => !contextEntryIds.has(entry.id));
+        const loadedEntries = earlierEntries.filter((entry) => this.loadedTranscriptEntryIds.has(entry.id));
+        this.transcriptLazyEntries = earlierEntries.filter((entry) => !this.loadedTranscriptEntryIds.has(entry.id));
+        return [...loadedEntries, ...contextEntries];
+    }
+    renderTranscriptEntries(entries, options = {}) {
+        if (this.transcriptLazyEntries.length > 0) {
+            const count = this.transcriptLazyEntries.length.toLocaleString();
+            this.chatContainer.addChild(new Text(theme.fg("dim", `${count} earlier transcript entries hidden (${keyText("app.session.loadEarlier")} to load)`), 1, 0));
+            this.chatContainer.addChild(new Spacer(1));
+        }
+        this.renderSessionEntries(entries, options);
+    }
+    handleTranscriptScroll(scrollTop) {
+        if (scrollTop === 0) {
+            this.loadEarlierTranscript();
+        }
+    }
+    loadEarlierTranscript() {
+        if (this.loadingEarlierTranscript || this.session.isStreaming || this.transcriptLazyEntries.length === 0) {
+            return;
+        }
+        this.loadingEarlierTranscript = true;
+        try {
+            const firstLoadedIndex = Math.max(0, this.transcriptLazyEntries.length - TRANSCRIPT_LAZY_LOAD_BATCH_SIZE);
+            const entriesToLoad = this.transcriptLazyEntries.slice(firstLoadedIndex);
+            for (const entry of entriesToLoad) {
+                this.loadedTranscriptEntryIds.add(entry.id);
+            }
+            this.rebuildChatFromMessages();
+            const remaining = this.transcriptLazyEntries.length;
+            this.showStatus(remaining > 0
+                ? `Loaded ${entriesToLoad.length} earlier transcript entries (${remaining} remaining)`
+                : "Loaded the beginning of the transcript");
+            this.ui.requestRender();
+        }
+        finally {
+            this.loadingEarlierTranscript = false;
+        }
+    }
     renderInitialMessages() {
-        const entries = this.sessionManager.getBranch();
-        this.renderSessionEntries(entries, {
+        const entries = this.getTranscriptEntriesForRender(true);
+        this.renderTranscriptEntries(entries, {
             updateFooter: true,
             populateHistory: true,
         });
@@ -3057,7 +3122,7 @@ export class InteractiveMode {
     }
     rebuildChatFromMessages() {
         this.chatContainer.clear();
-        this.renderSessionEntries(this.sessionManager.getBranch());
+        this.renderTranscriptEntries(this.getTranscriptEntriesForRender());
     }
     // =========================================================================
     // Key handlers