atom-lite 0.6.2 → 0.6.3

package/dist/cli/app.js

@@ -9,14 +9,15 @@ import { execSync, spawn } from "node:child_process";
 import { defaultShell, terminateProcessTree } from "../security/process.js";
 import { openSearchablePicker } from "./picker.js";
 import { openCommandPalette } from "./palette.js";
-import { bareCommandName, createCommandRegistry } from "./commands.js";
+import { bareCommandName, createCommandRegistry, } from "./commands.js";
 import { resolveTheme, themeLabel, themeNames, markdownThemeFor, } from "./theme.js";
 import { SecretInput } from "./secret-input.js";
 import { StatusLine } from "./status-line.js";
 import { Activity } from "./activity.js";
 import { ToolBox } from "./tool-box.js";
+import { ApprovalPrompt } from "./approval-prompt.js";
 import { Banner } from "./banner.js";
-import { formatContextUsage, knownContextWindow, } from "../providers/context-window.js";
+import { formatContextUsage, knownContextWindow, knownVisionModel, } from "../providers/context-window.js";
 import { StreamText } from "./stream-text.js";
 import { Transcript } from "./transcript.js";
 import { AgentRuntime } from "../agent/loop.js";
@@ -28,8 +29,8 @@ import { ProjectTrustPrompt } from "./trust-prompt.js";
 import { formatSessionList, formatSessionSearch, searchSessions, sessionTitleFromEvents, } from "../sessions/search.js";
 import { providerSwitchNotice, sessionHistoryLines, sessionSelectionFromEvents, } from "../sessions/history.js";
 import { blockAtLine, createMouseCapture, handleMouseEvent, parseSgrMouse, shouldEnableMouse, } from "./mouse.js";
-import { readTool, writeTool, editTool, bashTool, webfetchTool, websearchTool, grepTool, globTool } from "../tools/index.js";
-import { McpManager, formatMcpStatus, formatMcpDoctor, createMcpTools } from "../mcp/index.js";
+import { readTool, writeTool, editTool, bashTool, webfetchTool, websearchTool, grepTool, globTool, taskTool, } from "../tools/index.js";
+import { McpManager, formatMcpStatus, formatMcpDoctor, createMcpTools, } from "../mcp/index.js";
 import { createProvider } from "../providers/factory.js";
 import { availableProviderNames, getPreset, listModelsForProvider, materializeProviderConfig, probeApiKey, providerLabel, PROVIDER_PRESETS, } from "../providers/catalog.js";
 import { ContextManager } from "../context/manager.js";
@@ -90,7 +91,8 @@ export async function run(args) {
         }
     }
     if (showVersion) {
-        process.stdout.write("0.3.1\n");
+        const { VERSION } = await import("../version.js");
+        process.stdout.write(VERSION + "\n");
         return;
     }
     if (listSessions || searchSessionsQuery !== undefined) {
@@ -125,12 +127,23 @@ export async function run(args) {
         modelByProvider.set(provider.id, modelOverride);
     const useTools = !positional.includes("__notools__");
     const builtInTools = useTools
-        ? [readTool, writeTool, editTool, bashTool, webfetchTool, websearchTool, grepTool, globTool]
+        ? [
+            readTool,
+            writeTool,
+            editTool,
+            bashTool,
+            webfetchTool,
+            websearchTool,
+            grepTool,
+            globTool,
+            taskTool,
+        ]
         : [];
     const interactive = process.stdin.isTTY === true && !jsonMode;
     // one-shot task from positional or stdin
     let stdinTask = "";
-    if (!process.stdin.isTTY && positional.filter((p) => !p.startsWith("__")).length === 0) {
+    if (!process.stdin.isTTY &&
+        positional.filter((p) => !p.startsWith("__")).length === 0) {
         const chunks = [];
         for await (const chunk of process.stdin)
             chunks.push(chunk);
@@ -140,9 +153,11 @@ export async function run(args) {
     const oneShot = (taskArgs.join(" ").trim() || stdinTask).trim();
     const store = new JsonlSessionStore();
     let session = sessionOverride
-        ? store.resume(sessionOverride) ?? store.create(process.cwd(), provider.id)
+        ? (store.resume(sessionOverride) ??
+            store.create(process.cwd(), provider.id))
         : positional.includes("__continue__")
-            ? store.resumeLatest(process.cwd()) ?? store.create(process.cwd(), provider.id)
+            ? (store.resumeLatest(process.cwd()) ??
+                store.create(process.cwd(), provider.id))
             : store.create(process.cwd(), provider.id);
     const mcpServers = (config.mcpServers ?? {});
     const mcpManager = new McpManager({
@@ -165,7 +180,9 @@ export async function run(args) {
             try {
                 const { execFileSync } = require("node:child_process");
                 const parts = value.slice(1).trim().split(/\s+/);
-                return execFileSync(parts[0], parts.slice(1), { encoding: "utf8" }).trim();
+                return execFileSync(parts[0], parts.slice(1), {
+                    encoding: "utf8",
+                }).trim();
             }
             catch {
                 return undefined;
@@ -203,7 +220,7 @@ export async function run(args) {
     // Bridge Pi-style extension providers into Atom-lite's config so they
     // appear in /provider, /model, and the command palette.
     for (const loaded of extensionLoader.list()) {
-        for (const [name, provider] of (loaded.registries?.providers ?? new Map())) {
+        for (const [name, provider] of loaded.registries?.providers ?? new Map()) {
             const cfg = provider?.config ?? {};
             const firstModel = Array.isArray(cfg.models) && cfg.models.length > 0
                 ? (cfg.models[0].id ?? cfg.models[0].name ?? "default")
@@ -278,9 +295,14 @@ export async function run(args) {
         contextManager,
         extensionDispatch,
         bashConcurrency: config.runtime.bashConcurrency,
+        checkpointingEnabled: interactive || Boolean(config.runtime.checkpointing),
     });
     mcpManager.onToolsChanged = () => {
-        runtime.setTools([...builtInTools, ...extensionTools, ...createMcpTools(mcpManager)]);
+        runtime.setTools([
+            ...builtInTools,
+            ...extensionTools,
+            ...createMcpTools(mcpManager),
+        ]);
     };
     if (oneShot || !interactive) {
         // non-interactive / one-shot path
@@ -290,7 +312,11 @@ export async function run(args) {
         try {
             if (useTools) {
                 await mcpManager.syncTools();
-                runtime.setTools([...builtInTools, ...extensionTools, ...createMcpTools(mcpManager)]);
+                runtime.setTools([
+                    ...builtInTools,
+                    ...extensionTools,
+                    ...createMcpTools(mcpManager),
+                ]);
             }
             const result = await runtime.runAgentTurn(oneShot || " ", session, controller.signal);
             if (jsonMode) {
@@ -302,7 +328,9 @@ export async function run(args) {
         }
         catch (err) {
             if (jsonMode)
-                process.stderr.write(JSON.stringify({ error: err instanceof Error ? err.message : String(err) }) + "\n");
+                process.stderr.write(JSON.stringify({
+                    error: err instanceof Error ? err.message : String(err),
+                }) + "\n");
             else
                 process.stderr.write(`atom-lite: ${err instanceof Error ? err.message : String(err)}\n`);
             process.exit(1);
@@ -314,14 +342,16 @@ export async function run(args) {
         return;
     }
     // ---- Interactive TUI (first-class) ----
-    const { TuiAltScreen, Text, Input, Editor, Container, Spacer, ScrollView, VStack, Loader, Markdown, Box } = await import("@earendil-works/pi-tui");
+    const { TuiAltScreen, Text, Input, Editor, Container, Spacer, ScrollView, VStack, Loader, Markdown, Box, } = await import("@earendil-works/pi-tui");
     const mouseEnabled = shouldEnableMouse({
         interactive,
         configured: config.ui.mouse,
         stdoutIsTTY: process.stdout.isTTY === true,
     });
     const terminal = new ProcessTerminal();
-    const tui = new TuiAltScreen(terminal, false, undefined, { mouse: mouseEnabled });
+    const tui = new TuiAltScreen(terminal, false, undefined, {
+        mouse: mouseEnabled,
+    });
     const mouseCapture = createMouseCapture((data) => terminal.write(data));
     let effortLevel = "medium";
     runtime.setEffort(effortLevel);
@@ -349,9 +379,17 @@ export async function run(args) {
             displayPath = "~" + cwd.slice(home.length);
         let branch = "";
         try {
-            branch = execSync("git rev-parse --abbrev-ref HEAD", { cwd, encoding: "utf8", stdio: ["ignore", "pipe", "ignore"] }).trim();
+            branch = execSync("git rev-parse --abbrev-ref HEAD", {
+                cwd,
+                encoding: "utf8",
+                stdio: ["ignore", "pipe", "ignore"],
+            }).trim();
             if (branch === "HEAD") {
-                branch = execSync("git rev-parse --short HEAD", { cwd, encoding: "utf8", stdio: ["ignore", "pipe", "ignore"] }).trim();
+                branch = execSync("git rev-parse --short HEAD", {
+                    cwd,
+                    encoding: "utf8",
+                    stdio: ["ignore", "pipe", "ignore"],
+                }).trim();
             }
         }
         catch {
@@ -401,7 +439,9 @@ export async function run(args) {
         const sig = skillCacheSignature(process.cwd());
         if (sig === registrySkillSignature)
             return;
-        commandRegistry = createCommandRegistry({ skills: cachedSkillRegistry().list() });
+        commandRegistry = createCommandRegistry({
+            skills: cachedSkillRegistry().list(),
+        });
         registrySkillSignature = sig;
         input.setAutocompleteProvider(new CombinedAutocompleteProvider(commandRegistry.autocomplete(), process.cwd(), fdPath));
     }
@@ -433,7 +473,9 @@ export async function run(args) {
                 input.autocompleteList) {
                 const selected = input.autocompleteList.getSelectedItem?.();
                 const name = selected?.value ?? selected?.name ?? selected?.label;
-                const cmd = name ? commandRegistry.get(String(name).replace(/^\//, "")) : undefined;
+                const cmd = name
+                    ? commandRegistry.get(String(name).replace(/^\//, ""))
+                    : undefined;
                 if (cmd?.bareInvokable) {
                     originalHandleInput("\t");
                     return;
@@ -595,7 +637,11 @@ export async function run(args) {
         try {
             if (useTools) {
                 await mcpManager.syncTools();
-                runtime.setTools([...builtInTools, ...extensionTools, ...createMcpTools(mcpManager)]);
+                runtime.setTools([
+                    ...builtInTools,
+                    ...extensionTools,
+                    ...createMcpTools(mcpManager),
+                ]);
             }
             const turnResult = await runtime.runAgentTurn(prompt, session, controller.signal);
             if (turnResult === "TOKEN_BUDGET_REACHED") {
@@ -746,11 +792,21 @@ export async function run(args) {
     function refreshContextWindow() {
         const profile = provider.id;
         const model = currentModelLabel();
-        const configured = profileConfig(profile)?.contextWindow;
-        const window = typeof configured === "number" && configured > 0
-            ? configured
+        const pcfg = profileConfig(profile);
+        const configuredWindow = pcfg?.contextWindow;
+        const window = typeof configuredWindow === "number" && configuredWindow > 0
+            ? configuredWindow
             : knownContextWindow(model);
         runtime.setContextWindow(window);
+        // Config override for vision takes precedence over provider metadata.
+        if (typeof pcfg?.supportsVision === "boolean") {
+            runtime.setSupportsVision(pcfg.supportsVision);
+        }
+        else {
+            const guessed = knownVisionModel(model);
+            if (typeof guessed === "boolean")
+                runtime.setSupportsVision(guessed);
+        }
         updateStatus();
         const getModelInfo = provider.getModelInfo;
         if (typeof getModelInfo !== "function")
@@ -761,10 +817,16 @@ export async function run(args) {
             if (provider.id !== profile || currentModelLabel() !== model)
                 return;
             const resolved = info?.contextWindow;
-            if (typeof resolved === "number" && resolved > 0) {
+            if (typeof resolved === "number" &&
+                resolved > 0 &&
+                typeof configuredWindow !== "number") {
                 runtime.setContextWindow(resolved);
                 updateStatus();
             }
+            if (typeof pcfg?.supportsVision !== "boolean" &&
+                typeof info?.supportsVision === "boolean") {
+                runtime.setSupportsVision(info.supportsVision);
+            }
         })
             .catch(() => {
             // Keep the static/config fallback.
@@ -775,7 +837,7 @@ export async function run(args) {
     const MODEL_CACHE_TTL_MS = 5 * 60 * 1000;
     const modelCache = new Map();
     for (const loaded of extensionLoader.list()) {
-        for (const [name, provider] of (loaded.registries?.providers ?? new Map())) {
+        for (const [name, provider] of loaded.registries?.providers ?? new Map()) {
             const cfg = provider?.config ?? {};
             const models = Array.isArray(cfg.models)
                 ? cfg.models.map((m) => m.id ?? m.name).filter(Boolean)
@@ -1011,7 +1073,11 @@ export async function run(args) {
         provider = createProvider(config, preset.id, chosen);
         runtime.switchModel(provider, chosen);
         if (typeof session.appendEvent === "function") {
-            session.appendEvent({ type: "provider_selected", provider: preset.id, model: chosen });
+            session.appendEvent({
+                type: "provider_selected",
+                provider: preset.id,
+                model: chosen,
+            });
         }
         refreshContextWindow();
         appendLine(`provider: ${preset.label} (${preset.id}) · model: ${chosen}`);
@@ -1262,7 +1328,9 @@ export async function run(args) {
                 return "No extensions discovered.";
             return report
                 .map((ext) => {
-                const origin = ext.origin ? `${ext.source}:${ext.origin}` : ext.source;
+                const origin = ext.origin
+                    ? `${ext.source}:${ext.origin}`
+                    : ext.source;
                 const version = ext.version ? `v${ext.version}` : "-";
                 const counts = `${ext.toolCount} tools / ${ext.commandCount} commands`;
                 const error = ext.error ? `\t${ext.error}` : "";
@@ -1280,7 +1348,9 @@ export async function run(args) {
             return showThinking;
         },
         setTheme: setThemeTarget,
-        listSkills: () => cachedSkillRegistry().list().map((skill) => ({
+        listSkills: () => cachedSkillRegistry()
+            .list()
+            .map((skill) => ({
             name: skill.name,
             description: skill.description,
         })),
@@ -1347,9 +1417,7 @@ export async function run(args) {
             const summaries = scope === "all" ? store.list() : store.list(process.cwd());
             const toggle = {
                 value: "__scope__",
-                label: scope === "all"
-                    ? "Show this workspace only"
-                    : "Show all workspaces",
+                label: scope === "all" ? "Show this workspace only" : "Show all workspaces",
                 description: `scope: ${scope === "all" ? "all workspaces" : process.cwd()}`,
             };
             return [
@@ -1373,7 +1441,9 @@ export async function run(args) {
                 if (item.value === "__scope__") {
                     scope = scope === "all" ? "workspace" : "all";
                     picker.updateItems(buildItems());
-                    picker.setStatus(scope === "all" ? "showing all workspaces" : "showing this workspace");
+                    picker.setStatus(scope === "all"
+                        ? "showing all workspaces"
+                        : "showing this workspace");
                     return;
                 }
                 picker.close();
@@ -1530,7 +1600,8 @@ export async function run(args) {
         }
         extensionProviderNames.clear();
         for (const loaded of extensionLoader.list()) {
-            for (const [name, provider] of (loaded.registries?.providers ?? new Map())) {
+            for (const [name, provider] of loaded.registries?.providers ??
+                new Map()) {
                 const cfg = provider?.config ?? {};
                 const firstModel = Array.isArray(cfg.models) && cfg.models.length > 0
                     ? (cfg.models[0].id ?? cfg.models[0].name ?? "default")
@@ -1715,6 +1786,22 @@ export async function run(args) {
         else if (e.type === "context_usage") {
             updateStatus();
         }
+        else if (e.type === "background_task_complete") {
+            appendLine(`background task ${e.id}: ${e.status} (exit ${e.exitCode ?? "null"})`, theme.tokens.dim);
+            tui.requestRender();
+        }
+        else if (e.type === "checkpoint_created") {
+            appendLine(`checkpoint ${e.id} created`, theme.tokens.dim);
+            tui.requestRender();
+        }
+        else if (e.type === "checkpoint_failed") {
+            appendLine(`checkpoint failed: ${e.reason}`, theme.tokens.warning);
+            tui.requestRender();
+        }
+        else if (e.type === "checkpoint_unavailable") {
+            appendLine(`checkpoint unavailable: ${e.reason ?? "not a git repository"}`, theme.tokens.warning);
+            tui.requestRender();
+        }
         else if (e.type === "budget_warning") {
             appendLine(`budget warning (${e.scope}): ${e.used}/${e.limit} tokens`, theme.tokens.warning);
             tui.requestRender();
@@ -1722,7 +1809,11 @@ export async function run(args) {
         else if (e.type === "model_switched") {
             appendLine(`switched model to ${e.provider}/${e.model}`);
             if (typeof session.appendEvent === "function") {
-                session.appendEvent({ type: "model_switched", provider: e.provider, model: e.model });
+                session.appendEvent({
+                    type: "model_switched",
+                    provider: e.provider,
+                    model: e.model,
+                });
             }
             updateStatus();
         }
@@ -1768,7 +1859,11 @@ export async function run(args) {
             lines.push(`  config: FAIL ${e instanceof Error ? e.message : String(e)}`);
         }
         try {
-            const r = await provider.generate({ model: modelOverride ?? "default", messages: [{ role: "user", content: [{ type: "text", text: "ping" }] }], tools: [] });
+            const r = await provider.generate({
+                model: modelOverride ?? "default",
+                messages: [{ role: "user", content: [{ type: "text", text: "ping" }] }],
+                tools: [],
+            });
             lines.push(`  provider generate: ok (${r.finishReason})`);
         }
         catch (e) {
@@ -1793,7 +1888,11 @@ export async function run(args) {
             }
             if (useTools) {
                 await mcpManager.syncTools();
-                runtime.setTools([...builtInTools, ...extensionTools, ...createMcpTools(mcpManager)]);
+                runtime.setTools([
+                    ...builtInTools,
+                    ...extensionTools,
+                    ...createMcpTools(mcpManager),
+                ]);
             }
             appendLine(formatMcpStatus(mcpManager.status()));
             return;
@@ -1901,7 +2000,9 @@ export async function run(args) {
         for (const message of session.messages) {
             if (message.role === "user") {
                 turn++;
-                const snippet = redactText(messageText(message)).split("\\n")[0].slice(0, 80);
+                const snippet = redactText(messageText(message))
+                    .split("\\n")[0]
+                    .slice(0, 80);
                 lines.push(`${turn}. ${snippet || "(empty)"}`);
             }
         }
@@ -2030,8 +2131,27 @@ export async function run(args) {
         appendLine(`> ${trimmed}`, theme.tokens.user);
         await runTurn(trimmed);
     };
-    // Approval prompts are superseded by plan/yolo modes.
-    approvals.handler = undefined;
+    approvals.handler = interactive
+        ? (req) => new Promise((resolve) => {
+            uiModalOpen = true;
+            closeOverlay();
+            dropdownContainer.clear();
+            activity.setPhase("waiting");
+            const prompt = new ApprovalPrompt(req, (decision) => {
+                uiModalOpen = false;
+                dropdownContainer.clear();
+                tui.setFocus(input);
+                activity.setPhase("working");
+                if (decision === "trust-session") {
+                    appendLine("trusted this session: outside-workspace operations will run without approval", theme.tokens.dim);
+                }
+                resolve(decision);
+            });
+            dropdownContainer.addChild(prompt);
+            tui.setFocus(prompt);
+            tui.requestRender();
+        })
+        : undefined;
     process.on("SIGTERM", () => {
         void mcpManager.close().finally(() => {
             tui.stop();