@asm-agent/coding-agent 0.8.8 → 0.9.0

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

@@ -7,9 +7,8 @@ import { BUILTIN_MCP_CATALOG } from "@asm-agent/ai/mcp";
 import { CombinedAutocompleteProvider, Container, Loader, Markdown, matchesKey, ProcessTerminal, Spacer, setKeybindings, Text, TruncatedText, TUI, truncateToWidth, visibleWidth, } from "@asm-agent/tui";
 import { spawn, spawnSync } from "child_process";
 import { buildDaemonUpdateRestartReport, launchDaemonUpdateRestartCoordinator, resolveDaemonUpdateRestartSocketPath, } from "../../cli/daemon-update-restart.js";
-import { APP_NAME, APP_TITLE, getAgentDir, getAgentTracesLogPath, getDebugLogPath, getLogsDir, getShareViewerUrl, SELF_UPDATE_INTERACTIVE_CHILD_ENV, SELF_UPDATE_NOT_ATTEMPTED_EXIT_CODE, VERSION, } from "../../config.js";
+import { APP_NAME, APP_TITLE, getAgentDir, getDebugLogPath, getLogsDir, getShareViewerUrl, SELF_UPDATE_INTERACTIVE_CHILD_ENV, SELF_UPDATE_NOT_ATTEMPTED_EXIT_CODE, VERSION, } from "../../config.js";
 import { AGENT_MESSAGE_RECEIVED_PREVIEW_LABEL, isAgentSessionMessage, startsAgentRun, } from "../../core/agent-messages.js";
-import { getAsmAgentTraceCredential, previewAgentTraceFile, uploadAgentTraceFile, uploadAllAgentTraces, } from "../../core/agent-traces.js";
 import { isNoModelsAvailableMessage } from "../../core/auth-guidance.js";
 import { DEFAULT_HEARTBEAT_DELIVERY_MODE, parseHeartbeatCommand, } from "../../core/cron-jobs.js";
 import { FooterDataProvider } from "../../core/footer-data-provider.js";
@@ -19,7 +18,6 @@ import { runMcpManagementCommand } from "../../core/mcp/mcp-command.js";
 import { bashOutputToText, COMPACTION_OUTCOME_CUSTOM_TYPE, createHeartbeatPromptMessage, HEARTBEAT_PROMPT_PREVIEW_LABEL, isCompactionOutcomeMessage, isRefinementOutcomeMessage, isSessionSlashCommandMessage, isSessionSlashCommandResultMessage, REFINEMENT_OUTCOME_CUSTOM_TYPE, SESSION_SLASH_COMMAND_CUSTOM_TYPE, SESSION_SLASH_COMMAND_RESULT_CUSTOM_TYPE, } from "../../core/messages.js";
 import { findExactModelReferenceMatch, resolveModelScopeFromModels } from "../../core/model-resolver.js";
 import { parseNewSessionCommand } from "../../core/new-session-command.js";
-import { resolveAsmAgentTracesBaseUrl } from "../../core/prime-inference-auth.js";
 import { resolvePrimeInferencePostLoginModelAction } from "../../core/prime-inference-model-selection.js";
 import { parseCommandArgs } from "../../core/prompt-templates.js";
 import { formatMissingSessionCwdPrompt, MissingSessionCwdError } from "../../core/session-cwd.js";
@@ -27,7 +25,6 @@ import { SessionImportFileNotFoundError } from "../../core/session-import-errors
 import { resolveSessionPath, SessionSelectorError, SessionSelectorNotFoundError } from "../../core/session-resolver.js";
 import { parseSkillBlock } from "../../core/skill-blocks.js";
 import { BUILTIN_SLASH_COMMANDS, builtinSlashCommandTakesArgument, isBuiltinSlashCommandName, parseSlashCommand, resolveBuiltinSlashCommandName, } from "../../core/slash-commands.js";
-import { captureAgentCommandUsed, captureOnboardingCompleted, } from "../../core/telemetry.js";
 import { truncateTail } from "../../core/tools/truncate.js";
 import { ASM_LOGO } from "../../themes/asm-logo.js";
 import { getChangelogPath, parseChangelog } from "../../utils/changelog.js";
@@ -71,6 +68,7 @@ import { FooterComponent } from "./components/footer.js";
 import { HeartbeatManagerComponent } from "./components/heartbeat-manager.js";
 import { InjectedPromptMessageComponent, isInjectedPromptMessage } from "./components/injected-prompt-message.js";
 import { formatKeyText, keyHint, keyText, rawKeyHint } from "./components/keybinding-hints.js";
+import { createMermaidMarkdownTransform } from "./components/mermaid.js";
 import { OnboardingSplashComponent } from "./components/onboarding-splash.js";
 import { MalformedRefinementOutcomeMessageComponent, RefinementOutcomeMessageComponent, } from "./components/refinement-outcome-message.js";
 import { ScopedModelsSelectorComponent } from "./components/scoped-models-selector.js";
@@ -88,7 +86,7 @@ import { UserMessageSelectorComponent } from "./components/user-message-selector
 import { FeatureHintDeck } from "./feature-hints.js";
 import { scopeHeartbeatsToSession } from "./heartbeat-scope.js";
 import { collectMarkedImages, evictImagesToBudget, formatImageMarker, imageMarkerIds, remapImageMarkers, } from "./image-markers.js";
-import { isOnboardingModelReady, shouldRunOnboarding, shouldRunPrimeCliOnboardingSplash, } from "./onboarding.js";
+import { shouldRunOnboarding, shouldRunPrimeCliOnboardingSplash } from "./onboarding.js";
 import { QueueSelection } from "./queue-selection.js";
 import { formatResumeHint } from "./resume-hint.js";
 import { getAvailableThemes, getAvailableThemesWithPaths, getEditorTheme, getMarkdownTheme, getThemeByName, initTheme, onThemeChange, setRegisteredThemes, setTheme, setThemeInstance, stopThemeWatcher, Theme, theme, } from "./theme/theme.js";
@@ -611,6 +609,10 @@ export class InteractiveMode {
     agentMessagesExpanded = false;
     editDiffsExpanded = false;
     hideThinkingBlock = false;
+    mermaidMarkdownTransform = createMermaidMarkdownTransform({
+        getMode: () => this.settingsManager.getMermaidRenderingMode(),
+        theme,
+    });
     skillCommands = new Map();
     connectionCommands = [];
     connectionModels = [];
@@ -642,7 +644,6 @@ export class InteractiveMode {
     refineLoader = undefined;
     retryLoader = undefined;
     retryCountdown = undefined;
-    traceUploadAllAbortController = undefined;
     // Session-owned queued messages mirrored from connection events.
     connectionQueue = { steering: [], followUp: [] };
     queueSelection = new QueueSelection();
@@ -1253,34 +1254,11 @@ export class InteractiveMode {
         if (!this.shouldRunOnboarding()) {
             return false;
         }
-        const startedAt = Date.now();
         const showPrimeCliSplash = this.shouldRunPrimeCliOnboardingSplash();
-        let outcome = "aborted";
-        try {
-            this.markOnboardingShown();
-            await this.settingsManager.flush();
-            await this.runOnboardingFlow(showPrimeCliSplash);
-            outcome = isOnboardingModelReady(this.getOnboardingState()) ? "success" : "aborted";
-            return true;
-        }
-        catch (error) {
-            outcome = "error";
-            throw error;
-        }
-        finally {
-            const model = this.getCurrentModel();
-            const authStatus = model ? this.modelRegistry.getProviderAuthStatus(model.provider) : undefined;
-            const storedCredential = model ? this.modelRegistry.authStorage.get(model.provider) : undefined;
-            void captureOnboardingCompleted({
-                agentDir: getAgentDir(),
-                settingsManager: this.settingsManager,
-                durationMs: Date.now() - startedAt,
-                outcome,
-                provider: model?.provider,
-                authSource: authStatus?.source,
-                storedCredentialType: storedCredential?.type,
-            }).catch(() => { });
-        }
+        this.markOnboardingShown();
+        await this.settingsManager.flush();
+        await this.runOnboardingFlow(showPrimeCliSplash);
+        return true;
     }
     async showOnboardingModelSelection(splash) {
         try {
@@ -2067,7 +2045,6 @@ export class InteractiveMode {
             this.isBashRunning() ||
             this.getRetryAttempt() > 0 ||
             this.connectionState?.sessionActions.active !== undefined ||
-            this.traceUploadAllAbortController !== undefined ||
             this.sideQuestionEvent?.status === "running");
     }
     getRetryAttempt() {
@@ -3733,13 +3710,6 @@ export class InteractiveMode {
                     this.ui.requestRender();
                     return;
                 }
-                if (commandName) {
-                    void captureAgentCommandUsed({
-                        agentDir: getAgentDir(),
-                        settingsManager: this.settingsManager,
-                        commandName,
-                    }).catch(() => { });
-                }
                 if (commandName === "btw") {
                     this.editor.setText("");
                     await this.handleSideQuestion(commandArgs);
@@ -3818,11 +3788,6 @@ export class InteractiveMode {
                     this.editor.setText("");
                     return;
                 }
-                if (commandName === "traces") {
-                    await this.handleTracesCommand(canonicalCommandText);
-                    this.editor.setText("");
-                    return;
-                }
                 if (commandName === "context" && !commandArgs) {
                     this.echoLocalCommand(text);
                     await this.handleContextCommand();
@@ -4553,7 +4518,7 @@ export class InteractiveMode {
             case "message_update":
                 if (event.message.role === "assistant") {
                     this.streamingMessage = event.message;
-                    this.ensureAssistantStreamingComponent(event.message).updateContent(this.streamingMessage);
+                    this.ensureAssistantStreamingComponent(event.message).updateContent(this.streamingMessage, true);
                     for (const content of this.streamingMessage.content) {
                         if (content.type === "toolCall") {
                             await this.getOrCreatePendingToolComponent(content);
@@ -4579,7 +4544,7 @@ export class InteractiveMode {
                                 : `Operation aborted${elapsedSuffix}`;
                         this.streamingMessage.errorMessage = errorMessage;
                     }
-                    this.ensureAssistantStreamingComponent(event.message).updateContent(this.streamingMessage);
+                    this.ensureAssistantStreamingComponent(event.message).updateContent(this.streamingMessage, false);
                     if (this.streamingMessage.stopReason === "aborted" || this.streamingMessage.stopReason === "error") {
                         if (!errorMessage) {
                             errorMessage = this.streamingMessage.errorMessage || "Error";
@@ -4660,7 +4625,7 @@ export class InteractiveMode {
                 this.syncWorkingLoader();
                 if (this.streamingComponent) {
                     if (this.streamingMessage) {
-                        this.streamingComponent.updateContent(this.streamingMessage);
+                        this.streamingComponent.updateContent(this.streamingMessage, false);
                     }
                     else {
                         this.chatContainer.removeChild(this.streamingComponent);
@@ -4775,10 +4740,11 @@ export class InteractiveMode {
             expanded: this.toolOutputExpanded,
             precededByToolActivity: this.chatContainer.children.at(-1) instanceof ToolExecutionComponent ||
                 this.chatContainer.children.at(-1) instanceof AgentMessageComponent,
+            mermaidTransform: this.mermaidMarkdownTransform,
         });
         this.streamingMessage = message;
         this.chatContainer.addChild(this.streamingComponent);
-        this.streamingComponent.updateContent(this.streamingMessage);
+        this.streamingComponent.updateContent(this.streamingMessage, true);
     }
     ensureAssistantStreamingComponent(message) {
         let component = this.streamingComponent;
@@ -5339,6 +5305,7 @@ export class InteractiveMode {
                     expanded: this.toolOutputExpanded,
                     precededByToolActivity: this.chatContainer.children.at(-1) instanceof ToolExecutionComponent ||
                         this.chatContainer.children.at(-1) instanceof AgentMessageComponent,
+                    mermaidTransform: this.mermaidMarkdownTransform,
                 });
                 this.chatContainer.addChild(assistantComponent);
                 break;
@@ -5594,7 +5561,6 @@ export class InteractiveMode {
         this.showCtrlCExitHint();
     }
     interruptOrClearInput() {
-        this.traceUploadAllAbortController?.abort(new Error("Trace upload cancelled"));
         if (this.sideQuestionEvent?.status === "running") {
             this.abortSideQuestion(this.sideQuestionEvent.id, true);
         }
@@ -6358,6 +6324,7 @@ export class InteractiveMode {
                 currentTheme: this.settingsManager.getTheme() || "asm",
                 availableThemes: getAvailableThemes(),
                 hideThinkingBlock: this.hideThinkingBlock,
+                mermaidRenderingMode: this.settingsManager.getMermaidRenderingMode(),
                 treeFilterMode: this.settingsManager.getTreeFilterMode(),
                 showHardwareCursor: this.settingsManager.getShowHardwareCursor(),
                 editorPaddingX: this.settingsManager.getEditorPaddingX(),
@@ -6456,6 +6423,11 @@ export class InteractiveMode {
                         this.showError(error instanceof Error ? error.message : String(error));
                     });
                 },
+                onMermaidRenderingModeChange: (mode) => {
+                    this.settingsManager.setMermaidRenderingMode(mode);
+                    this.chatContainer.invalidate();
+                    this.ui.requestRender();
+                },
                 onQuietStartupChange: (enabled) => {
                     this.settingsManager.setQuietStartup(enabled);
                 },
@@ -7906,234 +7878,6 @@ export class InteractiveMode {
         this.chatContainer.addChild(new Text(prompt, 1, 0));
         this.ui.requestRender();
     }
-    formatTraceUploadResult(result) {
-        switch (result.status) {
-            case "uploaded":
-                return `Trace uploaded (${result.bytesStored.toLocaleString()} bytes).`;
-            case "disabled":
-                return "Trace sharing is disabled.";
-            case "missing_credentials":
-                return "Trace sharing needs a Prime API key. Run /traces login.";
-            case "no_session_file":
-                return "Current session has no persisted trace yet.";
-            case "empty_session":
-                return "Current session trace is empty.";
-            case "invalid_session":
-                return `Trace upload skipped: ${result.message}.`;
-            case "too_large":
-                return `Trace upload skipped: session file is ${result.size.toLocaleString()} bytes; limit is ${result.maxBytes.toLocaleString()} bytes.`;
-            case "failed":
-                if (result.statusCode === 404) {
-                    return "Trace upload endpoint was not found. The platform API may not be deployed yet, or ASM_AGENT_TRACES_BASE_URL points at the wrong API.";
-                }
-                return `Trace upload failed: ${result.statusCode ? `HTTP ${result.statusCode}: ` : ""}${result.message}. See ${getAgentTracesLogPath()} for details.`;
-        }
-    }
-    async uploadCurrentTraceOnce() {
-        const state = await this.agentConnection.getState();
-        return uploadAgentTraceFile({
-            sessionFile: state.sessionFile,
-            authStorage: this.modelRegistry.authStorage,
-            settingsManager: this.settingsManager,
-            requireEnabled: false,
-            reloadConfig: false,
-        });
-    }
-    async previewCurrentTrace() {
-        const state = await this.agentConnection.getState();
-        const result = await previewAgentTraceFile({ sessionFile: state.sessionFile });
-        let info;
-        switch (result.status) {
-            case "no_session_file":
-                info = "Trace preview is unavailable until the current session has a persisted assistant response.";
-                break;
-            case "empty_session":
-                info = "The current trace is empty.";
-                break;
-            case "invalid_session":
-            case "failed":
-                info = `Trace preview failed: ${result.message}.`;
-                break;
-            case "ready":
-                info = this.formatTracePreview(result);
-                break;
-        }
-        this.chatContainer.addChild(new Spacer(1));
-        this.chatContainer.addChild(new Text(info, 1, 0));
-        this.ui.requestRender();
-    }
-    formatTracePreview(result) {
-        const lines = [
-            theme.bold("Trace Preview"),
-            theme.fg("dim", "Nothing has been uploaded by this command."),
-            "",
-            `${theme.fg("dim", "File:")} ${result.sessionFile}`,
-            `${theme.fg("dim", "Size:")} ${result.size.toLocaleString()} bytes`,
-            `${theme.fg("dim", "Uploadable:")} ${result.uploadable ? "Yes" : `No (limit ${result.maxBytes.toLocaleString()} bytes)`}`,
-            `${theme.fg("dim", "Endpoint:")} ${result.endpoint}`,
-            `${theme.fg("dim", "Session ID:")} ${result.sessionId}`,
-            `${theme.fg("dim", "Trace ID:")} ${result.traceId}`,
-        ];
-        if (result.parentSessionId) {
-            lines.push(`${theme.fg("dim", "Parent session:")} ${result.parentSessionId}`);
-        }
-        if (result.gitRepo) {
-            lines.push(`${theme.fg("dim", "Git repository:")} ${result.gitRepo}`);
-        }
-        if (result.gitCommit) {
-            lines.push(`${theme.fg("dim", "Git commit:")} ${result.gitCommit}`);
-        }
-        lines.push("", theme.bold("Raw JSONL payload preview"));
-        if (result.contentPreview) {
-            lines.push(result.contentPreview);
-            if (result.truncated) {
-                lines.push("", theme.fg("dim", "Preview truncated; upload sends the complete file."));
-            }
-        }
-        else {
-            lines.push(theme.fg("dim", "Payload omitted because the trace exceeds the upload limit."));
-        }
-        return lines.join("\n");
-    }
-    async uploadAllTraces(sessionDir, signal) {
-        return uploadAllAgentTraces({
-            authStorage: this.modelRegistry.authStorage,
-            settingsManager: this.settingsManager,
-            sessionDir,
-            requireEnabled: false,
-            reloadConfig: false,
-            signal,
-            onProgress: ({ completed, total }) => {
-                if (total > 0 && (completed === 0 || completed === total || completed % 10 === 0)) {
-                    this.showStatus(`Uploading traces: ${completed.toLocaleString()}/${total.toLocaleString()} (${keyText("app.clear")} to cancel)`);
-                }
-            },
-        });
-    }
-    async handleTracesCommand(text) {
-        const command = text
-            .replace(/^\/traces\b/, "")
-            .trim()
-            .toLowerCase() || "status";
-        if (command === "status") {
-            await this.settingsManager.reload().catch(() => undefined);
-            const credential = await getAsmAgentTraceCredential(this.modelRegistry.authStorage);
-            const state = await this.agentConnection.getState();
-            const info = [
-                theme.bold("Trace Sharing"),
-                "",
-                `${theme.fg("dim", "Automatic uploads:")} ${this.settingsManager.getAgentTracesEnabled() ? "Enabled" : "Disabled"}`,
-                `${theme.fg("dim", "Credential:")} ${credential?.label ?? "Not configured"}`,
-                `${theme.fg("dim", "Endpoint:")} ${resolveAsmAgentTracesBaseUrl()}`,
-                `${theme.fg("dim", "Session file:")} ${state.sessionFile ?? "In-memory"}`,
-                "",
-                theme.fg("dim", "Commands: /traces on, /traces off, /traces preview, /traces upload-current, /traces upload-all, /traces login"),
-            ].join("\n");
-            this.chatContainer.addChild(new Spacer(1));
-            this.chatContainer.addChild(new Text(info, 1, 0));
-            this.ui.requestRender();
-            return;
-        }
-        if (command === "off" || command === "disable") {
-            this.settingsManager.setAgentTracesEnabled(false);
-            await this.settingsManager.flush();
-            this.showStatus("Trace sharing disabled.");
-            return;
-        }
-        if (command === "login") {
-            await this.createAuthFlows().runAsmAgentTracesLogin();
-            return;
-        }
-        if (command === "preview") {
-            await this.previewCurrentTrace();
-            return;
-        }
-        if (command === "on" || command === "enable") {
-            let credential = await getAsmAgentTraceCredential(this.modelRegistry.authStorage);
-            if (!credential) {
-                const authResult = await this.createAuthFlows().runAsmAgentTracesLogin();
-                if (authResult.status !== "success") {
-                    return;
-                }
-                credential = await getAsmAgentTraceCredential(this.modelRegistry.authStorage);
-            }
-            if (!credential) {
-                this.showError("Trace sharing needs a Prime API key.");
-                return;
-            }
-            this.settingsManager.setAgentTracesEnabled(true);
-            await this.settingsManager.flush();
-            const uploadResult = await this.uploadCurrentTraceOnce();
-            const uploadMessage = uploadResult.status === "no_session_file" || uploadResult.status === "empty_session"
-                ? "Current session will upload after the first assistant response."
-                : this.formatTraceUploadResult(uploadResult);
-            this.showStatus(`Trace sharing enabled. ${uploadMessage}`);
-            return;
-        }
-        if (command === "upload" || command === "upload-current") {
-            const credential = await getAsmAgentTraceCredential(this.modelRegistry.authStorage);
-            if (!credential) {
-                this.showError("Trace sharing needs a Prime API key. Run /traces login.");
-                return;
-            }
-            const uploadResult = await this.uploadCurrentTraceOnce();
-            const message = this.formatTraceUploadResult(uploadResult);
-            if (uploadResult.status === "failed") {
-                this.showError(message);
-            }
-            else {
-                this.showStatus(message);
-            }
-            return;
-        }
-        if (command === "upload-all") {
-            const credential = await getAsmAgentTraceCredential(this.modelRegistry.authStorage);
-            if (!credential) {
-                this.showError("Trace sharing needs a Prime API key. Run /traces login.");
-                return;
-            }
-            if (this.traceUploadAllAbortController) {
-                this.showWarning("A trace upload is already running. Cancel it before starting another.");
-                return;
-            }
-            const state = await this.agentConnection.getState();
-            const abortController = new AbortController();
-            this.traceUploadAllAbortController = abortController;
-            let result;
-            try {
-                result = await this.uploadAllTraces(state.sessionDir, abortController.signal);
-            }
-            finally {
-                if (this.traceUploadAllAbortController === abortController) {
-                    this.traceUploadAllAbortController = undefined;
-                }
-            }
-            if (abortController.signal.aborted) {
-                this.showStatus("Trace upload cancelled.");
-                return;
-            }
-            if (result.total === 0) {
-                this.showStatus("No persisted traces were found.");
-                return;
-            }
-            const summary = [
-                `Uploaded ${result.uploaded.toLocaleString()} of ${result.total.toLocaleString()} traces`,
-                result.skipped > 0 ? `${result.skipped.toLocaleString()} skipped` : undefined,
-                result.failed > 0 ? `${result.failed.toLocaleString()} failed` : undefined,
-                `${result.bytesStored.toLocaleString()} bytes stored`,
-            ]
-                .filter((part) => part !== undefined)
-                .join("; ");
-            if (result.failed > 0) {
-                this.showWarning(`${summary}. See ${getAgentTracesLogPath()} for details.`);
-            }
-            else {
-                this.showStatus(`${summary}.`);
-            }
-            return;
-        }
-        this.showWarning("Usage: /traces [status|on|off|preview|upload|upload-current|upload-all|login]");
-    }
     async handleContextCommand() {
         let info;
         try {