@earendil-works/pi-coding-agent 0.84.2 → 0.84.3

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

@@ -9,11 +9,12 @@ import * as path from "node:path";
 import * as TuiLayouts from "@earendil-works/pi-tui";
 import { CombinedAutocompleteProvider, Container, fuzzyFilter, getCapabilities, hyperlink, Markdown, matchesKey, ProcessTerminal, Spacer, setKeybindings, Text, TruncatedText, TuiAltScreen, TuiMainScreen, visibleWidth, } from "@earendil-works/pi-tui";
 import chalk from "chalk";
-import { spawn, spawnSync } from "child_process";
-import { APP_NAME, APP_TITLE, CONFIG_DIR_NAME, getAgentDir, getAuthPath, getDebugLogPath, getDocsPath, getShareViewerUrl, VERSION, } from "../../config.js";
+import { spawn } from "child_process";
+import { APP_NAME, APP_TITLE, CONFIG_DIR_NAME, getAgentDir, getAuthPath, getDebugLogPath, getDocsPath, VERSION, } from "../../config.js";
 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 { 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";
@@ -35,12 +36,12 @@ import { openBrowser } from "../../utils/open-browser.js";
 import { getCwdRelativePath } from "../../utils/paths.js";
 import { getPiUserAgent } from "../../utils/pi-user-agent.js";
 import { killTrackedDetachedChildren } from "../../utils/shell.js";
+import { loadAllHighlightLanguages } from "../../utils/syntax-highlight.js";
 import { ensureTool } from "../../utils/tools-manager.js";
 import { checkForNewPiVersion } from "../../utils/version-check.js";
 import { ArminComponent } from "./components/armin.js";
 import { AssistantMessageComponent } from "./components/assistant-message.js";
 import { BashExecutionComponent } from "./components/bash-execution.js";
-import { BorderedLoader } from "./components/bordered-loader.js";
 import { BranchSummaryMessageComponent } from "./components/branch-summary-message.js";
 import { CompactionSummaryMessageComponent } from "./components/compaction-summary-message.js";
 import { CustomEditor } from "./components/custom-editor.js";
@@ -63,6 +64,7 @@ import { SessionSelectorComponent } from "./components/session-selector.js";
 import { SettingsSelectorComponent } from "./components/settings-selector.js";
 import { SkillInvocationMessageComponent } from "./components/skill-invocation-message.js";
 import { BranchSummaryStatusIndicator, CompactionStatusIndicator, IdleStatus, RetryStatusIndicator, WorkingStatusIndicator, } from "./components/status-indicator.js";
+import { ThinkingSelectorComponent } from "./components/thinking-selector.js";
 import { ToolExecutionComponent } from "./components/tool-execution.js";
 import { TreeSelectorComponent } from "./components/tree-selector.js";
 import { TrustSelectorComponent } from "./components/trust-selector.js";
@@ -71,6 +73,7 @@ import { UserMessageSelectorComponent } from "./components/user-message-selector
 import { editInExternalEditor } from "./external-editor.js";
 import { refreshModelCatalogs } from "./model-catalog-refresh.js";
 import { getModelSearchText } from "./model-search.js";
+import { shareSession } from "./session-share.js";
 import { getAvailableThemes, getAvailableThemesWithPaths, getEditorTheme, getMarkdownTheme, getThemeByName, onThemeChange, setRegisteredThemes, stopThemeWatcher, Theme, theme, } from "./theme/theme.js";
 import { InteractiveThemeController } from "./theme/theme-controller.js";
 function isExpandable(obj) {
@@ -91,6 +94,9 @@ class ExpandableText extends Text {
 function isCustomSessionEntry(item) {
     return "type" in item && item.type === "custom";
 }
+function isCompactionCostNotice(item) {
+    return "type" in item && item.type === "compaction_cost";
+}
 const DEAD_TERMINAL_ERROR_CODES = new Set(["EIO", "EPIPE", "ENOTCONN"]);
 function isDeadTerminalError(error) {
     if (!error || typeof error !== "object" || !("code" in error)) {
@@ -130,6 +136,11 @@ export function formatResumeCommand(sessionManager) {
 function hasDefaultModelProvider(providerId) {
     return providerId in defaultModelPerProvider;
 }
+function llamaCppPostLoginGuidance(actionLabel, loadedModelCount) {
+    return loadedModelCount === 0
+        ? `${actionLabel}. No llama.cpp models are loaded. Use /llama to load a model, then /model to select it.`
+        : `${actionLabel}. Use /model to select a loaded llama.cpp model, or /llama to manage models.`;
+}
 const AUTH_TYPE_ORDER = { oauth: 0, api_key: 1 };
 function createFuzzyAutocompleteItems(items, prefix, getSearchText, toAutocompleteItem) {
     const filtered = fuzzyFilter(items, prefix, getSearchText);
@@ -465,6 +476,15 @@ export class InteractiveMode {
                 }));
             };
         }
+        const thinkingCommand = slashCommands.find((command) => command.name === "thinking");
+        if (thinkingCommand) {
+            thinkingCommand.getArgumentCompletions = (prefix) => {
+                return createFuzzyAutocompleteItems(this.session.getAvailableThinkingLevels(), prefix, (level) => level, (level) => ({
+                    value: level,
+                    label: level,
+                }));
+            };
+        }
         const loginCommand = slashCommands.find((command) => command.name === "login");
         if (loginCommand) {
             loginCommand.getArgumentCompletions = (prefix) => {
@@ -749,6 +769,14 @@ export class InteractiveMode {
         });
         // Initialize available provider count for footer display
         await this.updateAvailableProviderCount();
+        // Flush the completed startup state before loading the remaining syntax grammars.
+        this.ui.renderNow();
+        void loadAllHighlightLanguages().then(() => {
+            if (!this.isInitialized)
+                return;
+            this.ui.invalidate();
+            this.ui.requestRender();
+        });
     }
     /**
      * Update terminal title with session name and cwd.
@@ -804,7 +832,18 @@ export class InteractiveMode {
             }
         });
         // Show startup warnings
-        const { migratedProviders, modelFallbackMessage, initialMessage, initialImages, initialMessages } = this.options;
+        const { migratedProviders, startupDiagnostics, modelFallbackMessage, initialMessage, initialImages, initialMessages, } = this.options;
+        for (const diagnostic of startupDiagnostics ?? []) {
+            if (diagnostic.type === "error") {
+                this.showError(diagnostic.message);
+            }
+            else if (diagnostic.type === "warning") {
+                this.showWarning(diagnostic.message);
+            }
+            else {
+                this.showStatus(diagnostic.message);
+            }
+        }
         if (migratedProviders && migratedProviders.length > 0) {
             this.showWarning(`Migrated credentials to auth.json: ${migratedProviders.join(", ")}`);
         }
@@ -2346,6 +2385,12 @@ export class InteractiveMode {
                 await this.handleModelCommand(searchTerm);
                 return;
             }
+            if (text === "/thinking" || text.startsWith("/thinking ")) {
+                const searchTerm = text.startsWith("/thinking ") ? text.slice(10).trim() : undefined;
+                this.editor.setText("");
+                this.handleThinkingCommand(searchTerm);
+                return;
+            }
             if (text === "/export" || text.startsWith("/export ")) {
                 await this.handleExportCommand(text);
                 this.editor.setText("");
@@ -2724,9 +2769,21 @@ export class InteractiveMode {
                     }
                 }
                 else if (event.result) {
+                    const entries = this.sessionManager.buildContextEntries();
+                    if (entries[0]?.type !== "compaction") {
+                        throw new Error("Completed compaction is missing from the session context");
+                    }
                     this.chatContainer.clear();
-                    this.rebuildChatFromMessages();
+                    // The latest compaction is prepended for model context; append it below at its chronological position.
+                    this.renderSessionEntries(entries.slice(1));
                     this.addMessageToChat(createCompactionSummaryMessage(event.result.summary, event.result.tokensBefore, new Date().toISOString()));
+                    if (event.result.usage) {
+                        this.addCompactionCostNotice({
+                            type: "compaction_cost",
+                            kind: "compaction",
+                            usage: event.result.usage,
+                        });
+                    }
                     this.footer.invalidate();
                 }
                 else if (event.errorMessage) {
@@ -2948,6 +3005,10 @@ export class InteractiveMode {
                 this.addCustomEntryToChat(item);
                 continue;
             }
+            if (isCompactionCostNotice(item)) {
+                this.addCompactionCostNotice(item);
+                continue;
+            }
             const message = item;
             // Assistant messages need special handling for tool calls
             if (message.role === "assistant") {
@@ -3015,10 +3076,28 @@ export class InteractiveMode {
             if (entry.type === "custom") {
                 return [entry];
             }
-            return sessionEntryToContextMessages(entry);
+            const messages = sessionEntryToContextMessages(entry);
+            if ((entry.type === "compaction" || entry.type === "branch_summary") && entry.usage && messages.length > 0) {
+                return [...messages, { type: "compaction_cost", kind: entry.type, usage: entry.usage }];
+            }
+            return messages;
         });
         this.renderSessionItems(items, options);
     }
+    /**
+     * Render billing usage for a compaction or branch summary. The notice is derived
+     * from persisted summary usage and is not stored as a separate session entry.
+     */
+    addCompactionCostNotice(notice) {
+        if (!this.settingsManager.getShowCacheMissNotices())
+            return;
+        const { usage } = notice;
+        const tokens = usage.input + usage.output + usage.cacheRead + usage.cacheWrite;
+        const cost = usage.cost.total >= 0.01 ? ` (~$${usage.cost.total.toFixed(2)})` : "";
+        const label = notice.kind === "compaction" ? "Compaction" : "Branch summary";
+        this.chatContainer.addChild(new Spacer(1));
+        this.chatContainer.addChild(new Text(theme.fg("warning", `${label}: ${formatTokens(tokens)} tokens billed${cost}`), 1, 0));
+    }
     /**
      * Show a transcript notice when a completed assistant message paid for a
      * significant cache miss. Only states observable facts: the miss itself,
@@ -3655,8 +3734,14 @@ export class InteractiveMode {
     showSettingsSelector() {
         this.showSelector((done) => {
             let selector;
+            const defaultProvider = this.settingsManager.getDefaultProvider();
+            const defaultModelId = this.settingsManager.getDefaultModel();
+            const defaultModel = defaultProvider && defaultModelId ? `${defaultProvider}/${defaultModelId}` : "not set";
             selector = new SettingsSelectorComponent({
                 autoCompact: this.session.autoCompactionEnabled,
+                defaultModel,
+                currentModel: this.session.model,
+                availableDefaultModels: this.session.modelRuntime.getAvailableSnapshot(),
                 showImages: this.settingsManager.getShowImages(),
                 imageWidthCells: this.settingsManager.getImageWidthCells(),
                 autoResizeImages: this.settingsManager.getImageAutoResize(),
@@ -3666,8 +3751,9 @@ export class InteractiveMode {
                 followUpMode: this.session.followUpMode,
                 transport: this.settingsManager.getTransport(),
                 httpIdleTimeoutMs: this.settingsManager.getHttpIdleTimeoutMs(),
-                thinkingLevel: this.session.thinkingLevel,
-                availableThinkingLevels: this.session.getAvailableThinkingLevels(),
+                thinkingLevel: this.settingsManager.getDefaultThinkingLevel() ?? DEFAULT_THINKING_LEVEL,
+                availableThinkingLevels: [...THINKING_LEVEL_OPTIONS],
+                modelThinkingLevels: this.settingsManager.getAllModelThinkingLevels(),
                 currentTheme: this.themeController.getThemeSelection() || "dark",
                 terminalTheme: this.themeController.getTerminalTheme(),
                 availableThemes: getAvailableThemes(),
@@ -3736,10 +3822,26 @@ export class InteractiveMode {
                     configureHttpDispatcher(timeoutMs);
                     this.showStatus(`HTTP idle timeout: ${formatHttpIdleTimeoutMs(timeoutMs)}`);
                 },
-                onThinkingLevelChange: (level) => {
-                    this.session.setThinkingLevel(level);
-                    this.footer.invalidate();
-                    this.updateEditorBorderColor();
+                onModelThinkingLevelChange: (provider, modelId, level) => {
+                    this.settingsManager.setModelThinkingLevel(provider, modelId, level);
+                    // If the override is for the current model, apply it to the session too
+                    const current = this.session.model;
+                    if (current && current.provider === provider && current.id === modelId) {
+                        this.session.setThinkingLevel(level);
+                        this.footer.invalidate();
+                        this.updateEditorBorderColor();
+                    }
+                },
+                onModelThinkingLevelRemove: (provider, modelId) => {
+                    this.settingsManager.removeModelThinkingLevel(provider, modelId);
+                    // If the override was for the current model, revert to global default
+                    const current = this.session.model;
+                    if (current && current.provider === provider && current.id === modelId) {
+                        const globalDefault = this.settingsManager.getDefaultThinkingLevel() ?? DEFAULT_THINKING_LEVEL;
+                        this.session.setThinkingLevel(globalDefault);
+                        this.footer.invalidate();
+                        this.updateEditorBorderColor();
+                    }
                 },
                 onThemeChange: (themeSetting) => {
                     this.settingsManager.setTheme(themeSetting);
@@ -3860,6 +3962,44 @@ export class InteractiveMode {
             return { component: selector, focus: selector.getSettingsList() };
         });
     }
+    handleThinkingCommand(searchTerm) {
+        const availableLevels = this.session.getAvailableThinkingLevels();
+        if (!searchTerm) {
+            this.showThinkingSelector();
+            return;
+        }
+        const normalized = searchTerm.trim().toLowerCase();
+        const level = availableLevels.find((candidate) => candidate.toLowerCase() === normalized);
+        if (!level) {
+            this.showError(`Unknown thinking level "${searchTerm}". Available levels: ${availableLevels.join(", ")}.`);
+            return;
+        }
+        this.selectThinkingLevel(level, false);
+    }
+    selectThinkingLevel(level, persist) {
+        try {
+            this.session.setThinkingLevel(level, { persist });
+            this.footer.invalidate();
+            this.updateEditorBorderColor();
+            this.showStatus(persist ? `Default thinking level: ${level}` : `Thinking level: ${level}`);
+        }
+        catch (error) {
+            this.showError(error instanceof Error ? error.message : String(error));
+        }
+    }
+    showThinkingSelector() {
+        this.showSelector((done) => {
+            const selectLevel = (level, persist) => {
+                this.selectThinkingLevel(level, persist);
+                done();
+            };
+            const selector = new ThinkingSelectorComponent(this.session.thinkingLevel ?? DEFAULT_THINKING_LEVEL, this.session.getAvailableThinkingLevels(), (level) => selectLevel(level, false), () => {
+                done();
+                this.ui.requestRender();
+            }, (level) => selectLevel(level, true), this.settingsManager.getDefaultThinkingLevel() ?? DEFAULT_THINKING_LEVEL);
+            return { component: selector, focus: selector };
+        });
+    }
     async handleModelCommand(searchTerm) {
         if (!searchTerm) {
             this.showModelSelector();
@@ -3868,7 +4008,7 @@ export class InteractiveMode {
         const model = await this.findExactModelMatch(searchTerm);
         if (model) {
             try {
-                await this.session.setModel(model);
+                await this.session.setModel(model, { persist: false });
                 this.footer.invalidate();
                 this.updateEditorBorderColor();
                 this.showStatus(`Model: ${model.id}`);
@@ -3997,13 +4137,13 @@ export class InteractiveMode {
     }
     showModelSelector(initialSearchInput) {
         this.showSelector((done) => {
-            const selector = new ModelSelectorComponent(this.ui, this.session.model, this.settingsManager, this.session.modelRuntime, this.session.scopedModels, async (model) => {
+            const selectModel = async (model, persist) => {
                 try {
-                    await this.session.setModel(model);
+                    await this.session.setModel(model, { persist });
                     this.footer.invalidate();
                     this.updateEditorBorderColor();
                     done();
-                    this.showStatus(`Model: ${model.id}`);
+                    this.showStatus(persist ? `Default model: ${model.provider}/${model.id}` : `Model: ${model.id}`);
                     void this.maybeWarnAboutAnthropicSubscriptionAuth(model);
                     this.checkDaxnutsEasterEgg(model);
                 }
@@ -4011,10 +4151,13 @@ export class InteractiveMode {
                     done();
                     this.showError(error instanceof Error ? error.message : String(error));
                 }
-            }, () => {
+            };
+            const defaultProvider = this.settingsManager.getDefaultProvider();
+            const defaultModel = this.settingsManager.getDefaultModel();
+            const selector = new ModelSelectorComponent(this.ui, this.session.model, this.session.modelRuntime, this.session.scopedModels, (model) => selectModel(model, false), () => {
                 done();
                 this.ui.requestRender();
-            }, initialSearchInput);
+            }, initialSearchInput, (model) => selectModel(model, true), defaultProvider && defaultModel ? { provider: defaultProvider, id: defaultModel } : undefined);
             return { component: selector, focus: selector, dispose: () => selector.dispose() };
         });
     }
@@ -4570,7 +4713,11 @@ export class InteractiveMode {
         if (isUnknownModel(previousModel)) {
             const availableModels = this.session.modelRuntime.getAvailableSnapshot();
             const providerModels = availableModels.filter((model) => model.provider === providerId);
-            if (!hasDefaultModelProvider(providerId)) {
+            // Matches LLAMA_PROVIDER_ID from extensions/llama/provider.ts; kept inline to avoid coupling interactive mode to the built-in extension.
+            if (providerId === "llama.cpp") {
+                selectionError = llamaCppPostLoginGuidance(actionLabel, providerModels.length);
+            }
+            else if (!hasDefaultModelProvider(providerId)) {
                 selectionError = `${actionLabel}, but no default model is configured for provider "${providerId}". Use /model to select a model.`;
             }
             else if (providerModels.length === 0) {
@@ -4584,7 +4731,7 @@ export class InteractiveMode {
                 }
                 else {
                     try {
-                        await this.session.setModel(selectedModel);
+                        await this.session.setModel(selectedModel, { persist: true });
                     }
                     catch (error) {
                         selectedModel = undefined;
@@ -4954,91 +5101,14 @@ export class InteractiveMode {
         }
     }
     async handleShareCommand() {
-        // Check if gh is available and logged in
-        try {
-            const authResult = spawnSync("gh", ["auth", "status"], { encoding: "utf-8" });
-            if (authResult.status !== 0) {
-                this.showError("GitHub CLI is not logged in. Run 'gh auth login' first.");
-                return;
-            }
-        }
-        catch {
-            this.showError("GitHub CLI (gh) is not installed. Install it from https://cli.github.com/");
-            return;
-        }
-        // Export to a temp file
-        const tmpFile = path.join(os.tmpdir(), "session.html");
-        try {
-            await this.session.exportToHtml(tmpFile, { themeName: theme.name });
-        }
-        catch (error) {
-            this.showError(`Failed to export session: ${error instanceof Error ? error.message : "Unknown error"}`);
-            return;
-        }
-        // Show cancellable loader, replacing the editor
-        const loader = new BorderedLoader(this.ui, theme, "Creating gist...");
-        this.editorContainer.clear();
-        this.editorContainer.addChild(loader);
-        this.ui.setFocus(loader);
-        this.ui.requestRender();
-        const restoreEditor = () => {
-            loader.dispose();
-            this.editorContainer.clear();
-            this.editorContainer.addChild(this.editor);
-            this.ui.setFocus(this.editor);
-            try {
-                fs.unlinkSync(tmpFile);
-            }
-            catch {
-                // Ignore cleanup errors
-            }
-        };
-        // Create a secret gist asynchronously
-        let proc = null;
-        loader.onAbort = () => {
-            proc?.kill();
-            restoreEditor();
-            this.showStatus("Share cancelled");
-        };
-        try {
-            const result = await new Promise((resolve) => {
-                proc = spawn("gh", ["gist", "create", "--public=false", tmpFile]);
-                let stdout = "";
-                let stderr = "";
-                proc.stdout?.on("data", (data) => {
-                    stdout += data.toString();
-                });
-                proc.stderr?.on("data", (data) => {
-                    stderr += data.toString();
-                });
-                proc.on("close", (code) => resolve({ stdout, stderr, code }));
-            });
-            if (loader.signal.aborted)
-                return;
-            restoreEditor();
-            if (result.code !== 0) {
-                const errorMsg = result.stderr?.trim() || "Unknown error";
-                this.showError(`Failed to create gist: ${errorMsg}`);
-                return;
-            }
-            // Extract gist ID from the URL returned by gh
-            // gh returns something like: https://gist.github.com/username/GIST_ID
-            const gistUrl = result.stdout?.trim();
-            const gistId = gistUrl?.split("/").pop();
-            if (!gistId) {
-                this.showError("Failed to parse gist ID from gh output");
-                return;
-            }
-            // Create the preview URL
-            const previewUrl = getShareViewerUrl(gistId);
-            this.showStatus(`Share URL: ${previewUrl}\nGist: ${gistUrl}`);
-        }
-        catch (error) {
-            if (!loader.signal.aborted) {
-                restoreEditor();
-                this.showError(`Failed to create gist: ${error instanceof Error ? error.message : "Unknown error"}`);
-            }
-        }
+        await shareSession({
+            session: this.session,
+            ui: this.ui,
+            editorContainer: this.editorContainer,
+            editor: this.editor,
+            showStatus: (message) => this.showStatus(message),
+            showError: (message) => this.showError(message),
+        });
     }
     async handleCopyCommand(options = {}) {
         const text = this.session.getLastAssistantText();