@earendil-works/pi-coding-agent 0.86.0 → 0.87.0

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

@@ -6,6 +6,7 @@ import * as crypto from "node:crypto";
 import * as fs from "node:fs";
 import * as os from "node:os";
 import * as path from "node:path";
+import { isRetryableAssistantError, } from "@earendil-works/pi-ai/compat";
 import * as TuiLayouts from "@earendil-works/pi-tui";
 import { CombinedAutocompleteProvider, Container, fuzzyFilter, getCapabilities, hyperlink, Markdown, matchesKey, Spacer, setCapabilityOverrides, setKeybindings, Text, TruncatedText, TuiAltScreen, TuiMainScreen, visibleWidth, } from "@earendil-works/pi-tui";
 import chalk from "chalk";
@@ -15,12 +16,12 @@ import { parseSkillBlock } from "../../core/agent-session.js";
 import { SessionImportFileNotFoundError } from "../../core/agent-session-runtime.js";
 import { CACHE_TTL_MS, collectCacheMisses, computeCacheWaste, detectCacheMiss, } from "../../core/cache-stats.js";
 import { formatCacheWarmingStatus, formatCacheWarmingUsage } from "../../core/cache-warmer.js";
-import { recordCrash, takeUnnotifiedCrash } from "../../core/crash-log.js";
+import { findExtensionStackMatches, recordCrash, takeUnnotifiedCrash } from "../../core/crash-log.js";
 import { DEFAULT_THINKING_LEVEL, THINKING_LEVEL_OPTIONS } from "../../core/defaults.js";
 import { FooterDataProvider } from "../../core/footer-data-provider.js";
 import { configureHttpDispatcher, formatHttpIdleTimeoutMs } from "../../core/http-dispatcher.js";
 import { KeybindingsManager } from "../../core/keybindings.js";
-import { createCompactionSummaryMessage } from "../../core/messages.js";
+import { createCompactionSummaryMessage, createCustomMessage } from "../../core/messages.js";
 import { defaultModelPerProvider, findExactModelReferenceMatch, resolveModelScopeFromModels, } from "../../core/model-resolver.js";
 import { CredentialSynchronizationError } from "../../core/model-runtime.js";
 import { DefaultPackageManager } from "../../core/package-manager.js";
@@ -120,6 +121,22 @@ function isDeadTerminalError(error) {
     const code = error.code;
     return code !== undefined && DEAD_TERMINAL_ERROR_CODES.has(code);
 }
+export function formatCrashExtensionHint(extensionMatches) {
+    const matches = Array.isArray(extensionMatches)
+        ? extensionMatches.filter((match) => typeof match === "string" && match.length > 0)
+        : [];
+    if (matches.length === 0)
+        return undefined;
+    const quoted = matches.map((match) => `\`${match}\``);
+    const labels = quoted.length === 1
+        ? quoted[0]
+        : quoted.length === 2
+            ? quoted.join(" and ")
+            : `${quoted.slice(0, -1).join(", ")}, and ${quoted[quoted.length - 1]}`;
+    const noun = matches.length === 1 ? "extension" : "extensions";
+    const pronoun = matches.length === 1 ? "it" : "them";
+    return `A stack frame came from loaded ${noun} ${labels}, which may be involved. Try disabling ${pronoun} with \`${APP_NAME} config\`, or run \`${APP_NAME} -ne\` to confirm.`;
+}
 const ANTHROPIC_SUBSCRIPTION_AUTH_WARNING = "Anthropic subscription auth is active. Third-party harness usage draws from extra usage and is billed per token, not your Claude plan limits. Manage extra usage at https://claude.ai/settings/usage. Disable this warning in /settings.";
 function isAnthropicSubscriptionAuthKey(apiKey) {
     return typeof apiKey === "string" && apiKey.startsWith("sk-ant-oat");
@@ -242,6 +259,7 @@ export class InteractiveMode {
     managedToolStatusStarted = false;
     // Streaming message tracking
     streamingComponent = undefined;
+    entriesRenderedByBoundaryCompaction = new Set();
     streamingMessage = undefined;
     // Tool execution tracking: toolCallId -> component
     pendingTools = new Map();
@@ -1539,6 +1557,10 @@ export class InteractiveMode {
     async handleFatalRuntimeError(prefix, error) {
         const message = error instanceof Error ? error.message : String(error);
         this.showError(`${prefix}: ${message}`);
+        const extensionHint = this.getCrashExtensionHint(error);
+        if (extensionHint) {
+            this.chatContainer.addChild(new Text(theme.fg("warning", extensionHint), this.outputPad, 0));
+        }
         if (this.recordCrash("fatal_error", error)) {
             this.chatContainer.addChild(new Text(theme.fg("muted", this.crashReportInstructions()), this.outputPad, 0));
         }
@@ -1546,6 +1568,14 @@ export class InteractiveMode {
         this.stop("transcript");
         process.exit(1);
     }
+    getCrashExtensionHint(error) {
+        try {
+            return formatCrashExtensionHint(findExtensionStackMatches(error instanceof Error ? error.stack : undefined, this.session.resourceLoader.getExtensions().extensions));
+        }
+        catch {
+            return undefined;
+        }
+    }
     /** Persist a crash so the next start can point the user at `/bug`. Returns false when nothing was written. */
     recordCrash(kind, error) {
         try {
@@ -1571,6 +1601,13 @@ export class InteractiveMode {
         this.chatContainer.addChild(new Text(theme.fg("muted", `If this looks like a ${APP_NAME} bug, /bug sends a report to the developers.`), this.outputPad, 0));
         this.ui.requestRender();
     }
+    maybeSuggestBugReport(message) {
+        if (message.stopReason !== "error" || isRetryableAssistantError(message))
+            return;
+        if (/\b(?:abort(?:ed)?|cancel(?:l?ed)?)\b/i.test(message.errorMessage ?? ""))
+            return;
+        this.suggestBugReport();
+    }
     renderCurrentSessionState() {
         this.loadedResourcesContainer.clear();
         this.chatContainer.clear();
@@ -2636,6 +2673,8 @@ export class InteractiveMode {
                 this.ui.requestRender();
                 break;
             case "entry_appended":
+                if (this.entriesRenderedByBoundaryCompaction.delete(event.entry.id))
+                    break;
                 if (event.entry.type === "custom") {
                     this.addCustomEntryToChat(event.entry);
                     this.ui.requestRender();
@@ -2644,6 +2683,34 @@ export class InteractiveMode {
                     this.addCacheWarmingUsage(event.entry);
                     this.ui.requestRender();
                 }
+                else if (event.entry.type === "custom_message" && event.entry.display) {
+                    this.addMessageToChat(createCustomMessage(event.entry.customType, event.entry.content, event.entry.display, event.entry.details, event.entry.timestamp));
+                    this.ui.requestRender();
+                }
+                else if (event.entry.type === "compaction") {
+                    const entries = this.sessionManager.buildContextEntries();
+                    if (entries[0]?.id !== event.entry.id)
+                        break;
+                    this.chatContainer.clear();
+                    const branch = this.sessionManager.getBranch();
+                    const compactionIndex = branch.findIndex((entry) => entry.id === event.entry.id);
+                    const entriesAfterCompaction = new Set(branch.slice(compactionIndex + 1).map((entry) => entry.id));
+                    const retainedEntries = entries.slice(1);
+                    this.renderSessionEntries(retainedEntries.filter((entry) => !entriesAfterCompaction.has(entry.id)));
+                    this.addMessageToChat(createCompactionSummaryMessage(event.entry.summary, event.entry.tokensBefore, event.entry.timestamp));
+                    if (event.entry.usage) {
+                        this.addCompactionCostNotice({
+                            type: "compaction_cost",
+                            kind: "compaction",
+                            usage: event.entry.usage,
+                        });
+                    }
+                    this.renderSessionEntries(retainedEntries.filter((entry) => entriesAfterCompaction.has(entry.id)));
+                    for (const entryId of entriesAfterCompaction)
+                        this.entriesRenderedByBoundaryCompaction.add(entryId);
+                    this.footer.invalidate();
+                    this.ui.requestRender();
+                }
                 break;
             case "session_info_changed":
                 this.updateTerminalTitle();
@@ -2724,8 +2791,7 @@ export class InteractiveMode {
                             });
                         }
                         this.pendingTools.clear();
-                        if (this.streamingMessage.stopReason === "error")
-                            this.suggestBugReport();
+                        this.maybeSuggestBugReport(this.streamingMessage);
                     }
                     else {
                         // Args are now complete - trigger diff computation for edit tools
@@ -2873,7 +2939,6 @@ export class InteractiveMode {
                 // Show error only on final failure (success shows normal response)
                 if (!event.success) {
                     this.showError(`Retry failed after ${event.attempt} attempts: ${event.finalError || "Unknown error"}`);
-                    this.suggestBugReport();
                 }
                 this.ui.requestRender();
                 break;
@@ -3368,6 +3433,9 @@ export class InteractiveMode {
         catch { }
         console.error(`${APP_NAME} exiting due to uncaughtException:`);
         console.error(error);
+        const extensionHint = this.getCrashExtensionHint(error);
+        if (extensionHint)
+            console.error(`\n${extensionHint}`);
         if (this.recordCrash("uncaught_exception", error)) {
             console.error(`\n${this.crashReportInstructions()}`);
         }
@@ -5258,6 +5326,7 @@ export class InteractiveMode {
             ui: this.ui,
             editorContainer: this.editorContainer,
             editor: this.editor,
+            keybindings: this.keybindings,
             showStatus: (message) => this.showStatus(message),
             showError: (message) => this.showError(message),
         }, hint);