apex-code 0.0.1-alpha.8 → 0.0.1-alpha.9

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

@@ -14,6 +14,7 @@ import { APP_NAME, APP_TITLE, CONFIG_DIR_NAME, getAgentDir, getAuthPath, getDebu
 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 { getApexEnvironment } from "../../core/environment.js";
 import { FooterDataProvider } from "../../core/footer-data-provider.js";
 import { configureHttpDispatcher, formatHttpIdleTimeoutMs } from "../../core/http-dispatcher.js";
@@ -37,6 +38,7 @@ import { parseGitUrl } from "../../utils/git.js";
 import { openBrowser } from "../../utils/open-browser.js";
 import { getCwdRelativePath } from "../../utils/paths.js";
 import { killTrackedDetachedChildren } from "../../utils/shell.js";
+import { loadAllHighlightLanguages } from "../../utils/syntax-highlight.js";
 import { ensureTool } from "../../utils/tools-manager.js";
 import { checkForNewApexCodeVersion } from "../../utils/version-check.js";
 import { ArminComponent } from "./components/armin.js";
@@ -67,6 +69,7 @@ import { PERMISSION_MODE_OVERRIDE_HINTS, SettingsSelectorComponent } from "./com
 import { SkillInvocationMessageComponent } from "./components/skill-invocation-message.js";
 import { ApexSplashHeader } from "./components/splash-header.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";
@@ -95,6 +98,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)) {
@@ -134,6 +140,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);
@@ -496,6 +507,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) => {
@@ -792,6 +812,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.
@@ -847,7 +875,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(", ")}`);
         }
@@ -2492,6 +2531,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("");
@@ -2878,9 +2923,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) {
@@ -3102,6 +3159,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") {
@@ -3169,10 +3230,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,
@@ -3858,8 +3937,14 @@ export class InteractiveMode {
         const permissionMode = await this.session.getPermissionMode();
         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(),
@@ -3869,8 +3954,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(),
@@ -3940,10 +4026,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);
@@ -4067,6 +4169,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();
@@ -4075,7 +4215,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}`);
@@ -4204,13 +4344,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);
                 }
@@ -4218,10 +4358,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() };
         });
     }
@@ -4777,7 +4920,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) {
@@ -4791,7 +4938,7 @@ export class InteractiveMode {
                 }
                 else {
                     try {
-                        await this.session.setModel(selectedModel);
+                        await this.session.setModel(selectedModel, { persist: true });
                     }
                     catch (error) {
                         selectedModel = undefined;