codemaxxing 0.3.1 → 0.4.1

package/dist/index.js

@@ -1,11 +1,11 @@
 #!/usr/bin/env node
-import { jsx as _jsx, jsxs as _jsxs } from "react/jsx-runtime";
+import { jsx as _jsx, jsxs as _jsxs, Fragment as _Fragment } from "react/jsx-runtime";
 import React, { useState, useEffect, useCallback } from "react";
 import { render, Box, Text, useInput, useApp, useStdout } from "ink";
 import { EventEmitter } from "events";
 import TextInput from "ink-text-input";
 import { CodingAgent } from "./agent.js";
-import { loadConfig, detectLocalProvider, parseCLIArgs, applyOverrides, listModels } from "./config.js";
+import { loadConfig, saveConfig, detectLocalProvider, parseCLIArgs, applyOverrides, listModels } from "./config.js";
 import { listSessions, getSession, loadMessages, deleteSession } from "./utils/sessions.js";
 import { execSync } from "child_process";
 import { isGitRepo, getBranch, getStatus, getDiff, undoLastCommit } from "./utils/git.js";
@@ -13,6 +13,9 @@ import { getTheme, listThemes, THEMES, DEFAULT_THEME } from "./themes.js";
 import { PROVIDERS, getCredentials, openRouterOAuth, anthropicSetupToken, importCodexToken, importQwenToken, copilotDeviceFlow } from "./utils/auth.js";
 import { listInstalledSkills, installSkill, removeSkill, getRegistrySkills, searchRegistry, createSkillScaffold, getActiveSkills, getActiveSkillCount } from "./utils/skills.js";
 import { listServers, addServer, removeServer, getConnectedServers } from "./utils/mcp.js";
+import { detectHardware, formatBytes } from "./utils/hardware.js";
+import { getRecommendationsWithLlmfit, getFitIcon, isLlmfitAvailable } from "./utils/models.js";
+import { isOllamaInstalled, isOllamaRunning, getOllamaInstallCommand, startOllama, stopOllama, pullModel, listInstalledModelsDetailed, deleteModel, getGPUMemoryUsage } from "./utils/ollama.js";
 const VERSION = "0.1.9";
 // ── Helpers ──
 function formatTimeAgo(date) {
@@ -64,6 +67,12 @@ const SLASH_COMMANDS = [
     { cmd: "/mcp add", desc: "add MCP server" },
     { cmd: "/mcp remove", desc: "remove MCP server" },
     { cmd: "/mcp reconnect", desc: "reconnect MCP servers" },
+    { cmd: "/ollama", desc: "Ollama status & models" },
+    { cmd: "/ollama list", desc: "list installed models" },
+    { cmd: "/ollama start", desc: "start Ollama server" },
+    { cmd: "/ollama stop", desc: "stop Ollama server" },
+    { cmd: "/ollama pull", desc: "download a model" },
+    { cmd: "/ollama delete", desc: "delete a model" },
     { cmd: "/quit", desc: "exit" },
 ];
 const SPINNER_FRAMES = ["⣾", "⣽", "⣻", "⢿", "⡿", "⣟", "⣯", "⣷"];
@@ -137,6 +146,17 @@ function App() {
     const [skillsPickerIndex, setSkillsPickerIndex] = useState(0);
     const [sessionDisabledSkills, setSessionDisabledSkills] = useState(new Set());
     const [approval, setApproval] = useState(null);
+    // ── Ollama Management State ──
+    const [ollamaDeleteConfirm, setOllamaDeleteConfirm] = useState(null);
+    const [ollamaPulling, setOllamaPulling] = useState(null);
+    const [ollamaExitPrompt, setOllamaExitPrompt] = useState(false);
+    const [wizardScreen, setWizardScreen] = useState(null);
+    const [wizardIndex, setWizardIndex] = useState(0);
+    const [wizardHardware, setWizardHardware] = useState(null);
+    const [wizardModels, setWizardModels] = useState([]);
+    const [wizardPullProgress, setWizardPullProgress] = useState(null);
+    const [wizardPullError, setWizardPullError] = useState(null);
+    const [wizardSelectedModel, setWizardSelectedModel] = useState(null);
     // Listen for paste events from stdin interceptor
     useEffect(() => {
         const handler = ({ content, lines }) => {
@@ -174,10 +194,11 @@ function App() {
             }
             else {
                 info.push("✗ No local LLM server found.");
-                info.push("  /connect  — retry after starting LM Studio or Ollama");
-                info.push("  /login    — authenticate with a cloud provider");
                 setConnectionInfo([...info]);
                 setReady(true);
+                // Show the setup wizard on first run
+                setWizardScreen("connection");
+                setWizardIndex(0);
                 return;
             }
         }
@@ -318,7 +339,8 @@ function App() {
                 // Commands that need args (like /commit, /model) — fill input instead of executing
                 if (selected.cmd === "/commit" || selected.cmd === "/model" || selected.cmd === "/session delete" ||
                     selected.cmd === "/skills install" || selected.cmd === "/skills remove" || selected.cmd === "/skills search" ||
-                    selected.cmd === "/skills on" || selected.cmd === "/skills off" || selected.cmd === "/architect") {
+                    selected.cmd === "/skills on" || selected.cmd === "/skills off" || selected.cmd === "/architect" ||
+                    selected.cmd === "/ollama pull" || selected.cmd === "/ollama delete") {
                     setInput(selected.cmd + " ");
                     setCmdIndex(0);
                     setInputKey((k) => k + 1);
@@ -344,7 +366,22 @@ function App() {
             return;
         addMsg("user", trimmed);
         if (trimmed === "/quit" || trimmed === "/exit") {
-            exit();
+            // Check if Ollama is running and offer to stop it
+            const ollamaUp = await isOllamaRunning();
+            if (ollamaUp) {
+                const config = loadConfig();
+                if (config.defaults.stopOllamaOnExit) {
+                    addMsg("info", "Stopping Ollama...");
+                    await stopOllama();
+                    exit();
+                }
+                else {
+                    setOllamaExitPrompt(true);
+                }
+            }
+            else {
+                exit();
+            }
             return;
         }
         if (trimmed === "/login" || trimmed === "/auth") {
@@ -387,6 +424,12 @@ function App() {
                 "  /mcp add   — add MCP server to global config",
                 "  /mcp remove — remove MCP server",
                 "  /mcp reconnect — reconnect all MCP servers",
+                "  /ollama    — Ollama status, models & GPU usage",
+                "  /ollama list — list installed models with sizes",
+                "  /ollama start — start Ollama server",
+                "  /ollama stop — stop Ollama server (frees GPU RAM)",
+                "  /ollama pull <model> — download a model",
+                "  /ollama delete <model> — delete a model from disk",
                 "  /quit      — exit",
             ].join("\n"));
             return;
@@ -554,6 +597,142 @@ function App() {
             addMsg("info", "🔍 Auto-lint OFF");
             return;
         }
+        // ── Ollama commands (work without agent) ──
+        if (trimmed === "/ollama" || trimmed === "/ollama status") {
+            const running = await isOllamaRunning();
+            const lines = [`Ollama: ${running ? "running" : "stopped"}`];
+            if (running) {
+                const models = await listInstalledModelsDetailed();
+                if (models.length > 0) {
+                    lines.push(`Installed models (${models.length}):`);
+                    for (const m of models) {
+                        const sizeGB = (m.size / (1024 * 1024 * 1024)).toFixed(1);
+                        lines.push(`  ${m.name}  (${sizeGB} GB)`);
+                    }
+                }
+                else {
+                    lines.push("No models installed.");
+                }
+                const gpuMem = getGPUMemoryUsage();
+                if (gpuMem)
+                    lines.push(`GPU: ${gpuMem}`);
+            }
+            else {
+                lines.push("Start with: /ollama start");
+            }
+            addMsg("info", lines.join("\n"));
+            return;
+        }
+        if (trimmed === "/ollama list") {
+            const running = await isOllamaRunning();
+            if (!running) {
+                addMsg("info", "Ollama is not running. Start with /ollama start");
+                return;
+            }
+            const models = await listInstalledModelsDetailed();
+            if (models.length === 0) {
+                addMsg("info", "No models installed. Pull one with /ollama pull <model>");
+            }
+            else {
+                const lines = models.map((m) => {
+                    const sizeGB = (m.size / (1024 * 1024 * 1024)).toFixed(1);
+                    return `  ${m.name}  (${sizeGB} GB)`;
+                });
+                addMsg("info", `Installed models:\n${lines.join("\n")}`);
+            }
+            return;
+        }
+        if (trimmed === "/ollama start") {
+            const running = await isOllamaRunning();
+            if (running) {
+                addMsg("info", "Ollama is already running.");
+                return;
+            }
+            if (!isOllamaInstalled()) {
+                addMsg("error", `Ollama is not installed. Install with: ${getOllamaInstallCommand(process.platform === "darwin" ? "macos" : process.platform === "win32" ? "windows" : "linux")}`);
+                return;
+            }
+            startOllama();
+            addMsg("info", "Starting Ollama server...");
+            // Wait for it to come up
+            for (let i = 0; i < 10; i++) {
+                await new Promise(r => setTimeout(r, 1000));
+                if (await isOllamaRunning()) {
+                    addMsg("info", "Ollama is running.");
+                    return;
+                }
+            }
+            addMsg("error", "Ollama did not start in time. Try running 'ollama serve' manually.");
+            return;
+        }
+        if (trimmed === "/ollama stop") {
+            const running = await isOllamaRunning();
+            if (!running) {
+                addMsg("info", "Ollama is not running.");
+                return;
+            }
+            addMsg("info", "Stopping Ollama...");
+            const result = await stopOllama();
+            addMsg(result.ok ? "info" : "error", result.ok ? `\u2705 ${result.message}` : `\u274C ${result.message}`);
+            return;
+        }
+        if (trimmed.startsWith("/ollama pull ")) {
+            const modelId = trimmed.replace("/ollama pull ", "").trim();
+            if (!modelId) {
+                addMsg("info", "Usage: /ollama pull <model>\n  Example: /ollama pull qwen2.5-coder:7b");
+                return;
+            }
+            if (!isOllamaInstalled()) {
+                addMsg("error", "Ollama is not installed.");
+                return;
+            }
+            // Ensure ollama is running
+            let running = await isOllamaRunning();
+            if (!running) {
+                startOllama();
+                addMsg("info", "Starting Ollama server...");
+                for (let i = 0; i < 10; i++) {
+                    await new Promise(r => setTimeout(r, 1000));
+                    if (await isOllamaRunning()) {
+                        running = true;
+                        break;
+                    }
+                }
+                if (!running) {
+                    addMsg("error", "Could not start Ollama. Run 'ollama serve' manually.");
+                    return;
+                }
+            }
+            setOllamaPulling({ model: modelId, progress: { status: "starting", percent: 0 } });
+            try {
+                await pullModel(modelId, (p) => {
+                    setOllamaPulling({ model: modelId, progress: p });
+                });
+                setOllamaPulling(null);
+                addMsg("info", `\u2705 Downloaded ${modelId}`);
+            }
+            catch (err) {
+                setOllamaPulling(null);
+                addMsg("error", `Failed to pull ${modelId}: ${err.message}`);
+            }
+            return;
+        }
+        if (trimmed.startsWith("/ollama delete ")) {
+            const modelId = trimmed.replace("/ollama delete ", "").trim();
+            if (!modelId) {
+                addMsg("info", "Usage: /ollama delete <model>");
+                return;
+            }
+            // Look up size for confirmation
+            const models = await listInstalledModelsDetailed();
+            const found = models.find((m) => m.name === modelId || m.name.startsWith(modelId));
+            if (!found) {
+                addMsg("error", `Model "${modelId}" not found. Use /ollama list to see installed models.`);
+                return;
+            }
+            setOllamaDeleteConfirm({ model: found.name, size: found.size });
+            return;
+        }
         // ── MCP commands (partially work without agent) ──
         if (trimmed === "/mcp" || trimmed === "/mcp list") {
             const servers = listServers(process.cwd());
@@ -1114,6 +1293,264 @@ function App() {
             }
             return;
         }
+        // ── Ollama delete confirmation ──
+        if (ollamaDeleteConfirm) {
+            if (inputChar === "y" || inputChar === "Y") {
+                const model = ollamaDeleteConfirm.model;
+                setOllamaDeleteConfirm(null);
+                const result = deleteModel(model);
+                addMsg(result.ok ? "info" : "error", result.ok ? `\u2705 ${result.message}` : `\u274C ${result.message}`);
+                return;
+            }
+            if (inputChar === "n" || inputChar === "N" || key.escape) {
+                setOllamaDeleteConfirm(null);
+                addMsg("info", "Delete cancelled.");
+                return;
+            }
+            return;
+        }
+        // ── Ollama exit prompt ──
+        if (ollamaExitPrompt) {
+            if (inputChar === "y" || inputChar === "Y") {
+                setOllamaExitPrompt(false);
+                stopOllama().then(() => exit());
+                return;
+            }
+            if (inputChar === "n" || inputChar === "N") {
+                setOllamaExitPrompt(false);
+                exit();
+                return;
+            }
+            if (inputChar === "a" || inputChar === "A") {
+                setOllamaExitPrompt(false);
+                saveConfig({ defaults: { ...loadConfig().defaults, stopOllamaOnExit: true } });
+                addMsg("info", "Saved preference: always stop Ollama on exit.");
+                stopOllama().then(() => exit());
+                return;
+            }
+            if (key.escape) {
+                setOllamaExitPrompt(false);
+                return;
+            }
+            return;
+        }
+        // ── Setup Wizard Navigation ──
+        if (wizardScreen) {
+            if (wizardScreen === "connection") {
+                const items = ["local", "openrouter", "apikey", "existing"];
+                if (key.upArrow) {
+                    setWizardIndex((prev) => (prev - 1 + items.length) % items.length);
+                    return;
+                }
+                if (key.downArrow) {
+                    setWizardIndex((prev) => (prev + 1) % items.length);
+                    return;
+                }
+                if (key.escape) {
+                    setWizardScreen(null);
+                    return;
+                }
+                if (key.return) {
+                    const selected = items[wizardIndex];
+                    if (selected === "local") {
+                        // Scan hardware and show model picker (use llmfit if available)
+                        const hw = detectHardware();
+                        setWizardHardware(hw);
+                        const { models: recs } = getRecommendationsWithLlmfit(hw);
+                        setWizardModels(recs.filter(m => m.fit !== "skip"));
+                        setWizardScreen("models");
+                        setWizardIndex(0);
+                    }
+                    else if (selected === "openrouter") {
+                        setWizardScreen(null);
+                        addMsg("info", "Starting OpenRouter OAuth — opening browser...");
+                        setLoading(true);
+                        setSpinnerMsg("Waiting for authorization...");
+                        openRouterOAuth((msg) => addMsg("info", msg))
+                            .then(() => {
+                            addMsg("info", "✅ OpenRouter authenticated! Use /connect to connect.");
+                            setLoading(false);
+                        })
+                            .catch((err) => { addMsg("error", `OAuth failed: ${err.message}`); setLoading(false); });
+                    }
+                    else if (selected === "apikey") {
+                        setWizardScreen(null);
+                        setLoginPicker(true);
+                        setLoginPickerIndex(0);
+                    }
+                    else if (selected === "existing") {
+                        setWizardScreen(null);
+                        addMsg("info", "Start your LLM server, then type /connect to retry.");
+                    }
+                    return;
+                }
+                return;
+            }
+            if (wizardScreen === "models") {
+                const models = wizardModels;
+                if (key.upArrow) {
+                    setWizardIndex((prev) => (prev - 1 + models.length) % models.length);
+                    return;
+                }
+                if (key.downArrow) {
+                    setWizardIndex((prev) => (prev + 1) % models.length);
+                    return;
+                }
+                if (key.escape) {
+                    setWizardScreen("connection");
+                    setWizardIndex(0);
+                    return;
+                }
+                if (key.return) {
+                    const selected = models[wizardIndex];
+                    if (selected) {
+                        setWizardSelectedModel(selected);
+                        // Check if Ollama is installed
+                        if (!isOllamaInstalled()) {
+                            setWizardScreen("install-ollama");
+                        }
+                        else {
+                            // Start pulling the model
+                            setWizardScreen("pulling");
+                            setWizardPullProgress({ status: "starting", percent: 0 });
+                            setWizardPullError(null);
+                            (async () => {
+                                try {
+                                    // Ensure ollama is running
+                                    const running = await isOllamaRunning();
+                                    if (!running) {
+                                        setWizardPullProgress({ status: "Starting Ollama server...", percent: 0 });
+                                        startOllama();
+                                        // Wait for it to come up
+                                        for (let i = 0; i < 15; i++) {
+                                            await new Promise(r => setTimeout(r, 1000));
+                                            if (await isOllamaRunning())
+                                                break;
+                                        }
+                                        if (!(await isOllamaRunning())) {
+                                            setWizardPullError("Could not start Ollama server. Run 'ollama serve' manually, then press Enter.");
+                                            return;
+                                        }
+                                    }
+                                    await pullModel(selected.ollamaId, (p) => {
+                                        setWizardPullProgress(p);
+                                    });
+                                    setWizardPullProgress({ status: "success", percent: 100 });
+                                    // Wait briefly then connect
+                                    await new Promise(r => setTimeout(r, 500));
+                                    setWizardScreen(null);
+                                    setWizardPullProgress(null);
+                                    setWizardSelectedModel(null);
+                                    addMsg("info", `✅ ${selected.name} installed! Connecting...`);
+                                    await connectToProvider(true);
+                                }
+                                catch (err) {
+                                    setWizardPullError(err.message);
+                                }
+                            })();
+                        }
+                    }
+                    return;
+                }
+                return;
+            }
+            if (wizardScreen === "install-ollama") {
+                if (key.escape) {
+                    setWizardScreen("models");
+                    setWizardIndex(0);
+                    return;
+                }
+                if (key.return) {
+                    // User says they installed it — check and proceed
+                    if (isOllamaInstalled()) {
+                        const selected = wizardSelectedModel;
+                        if (selected) {
+                            setWizardScreen("pulling");
+                            setWizardPullProgress({ status: "starting", percent: 0 });
+                            setWizardPullError(null);
+                            (async () => {
+                                try {
+                                    const running = await isOllamaRunning();
+                                    if (!running) {
+                                        setWizardPullProgress({ status: "Starting Ollama server...", percent: 0 });
+                                        startOllama();
+                                        for (let i = 0; i < 15; i++) {
+                                            await new Promise(r => setTimeout(r, 1000));
+                                            if (await isOllamaRunning())
+                                                break;
+                                        }
+                                        if (!(await isOllamaRunning())) {
+                                            setWizardPullError("Could not start Ollama server. Run 'ollama serve' manually, then press Enter.");
+                                            return;
+                                        }
+                                    }
+                                    await pullModel(selected.ollamaId, (p) => setWizardPullProgress(p));
+                                    setWizardPullProgress({ status: "success", percent: 100 });
+                                    await new Promise(r => setTimeout(r, 500));
+                                    setWizardScreen(null);
+                                    setWizardPullProgress(null);
+                                    setWizardSelectedModel(null);
+                                    addMsg("info", `✅ ${selected.name} installed! Connecting...`);
+                                    await connectToProvider(true);
+                                }
+                                catch (err) {
+                                    setWizardPullError(err.message);
+                                }
+                            })();
+                        }
+                    }
+                    else {
+                        addMsg("info", "Ollama not found yet. Install it and press Enter again.");
+                    }
+                    return;
+                }
+                return;
+            }
+            if (wizardScreen === "pulling") {
+                // Allow retry on error
+                if (wizardPullError && key.return) {
+                    const selected = wizardSelectedModel;
+                    if (selected) {
+                        setWizardPullError(null);
+                        setWizardPullProgress({ status: "retrying", percent: 0 });
+                        (async () => {
+                            try {
+                                const running = await isOllamaRunning();
+                                if (!running) {
+                                    startOllama();
+                                    for (let i = 0; i < 15; i++) {
+                                        await new Promise(r => setTimeout(r, 1000));
+                                        if (await isOllamaRunning())
+                                            break;
+                                    }
+                                }
+                                await pullModel(selected.ollamaId, (p) => setWizardPullProgress(p));
+                                setWizardPullProgress({ status: "success", percent: 100 });
+                                await new Promise(r => setTimeout(r, 500));
+                                setWizardScreen(null);
+                                setWizardPullProgress(null);
+                                setWizardSelectedModel(null);
+                                addMsg("info", `✅ ${selected.name} installed! Connecting...`);
+                                await connectToProvider(true);
+                            }
+                            catch (err) {
+                                setWizardPullError(err.message);
+                            }
+                        })();
+                    }
+                    return;
+                }
+                if (wizardPullError && key.escape) {
+                    setWizardScreen("models");
+                    setWizardIndex(0);
+                    setWizardPullError(null);
+                    setWizardPullProgress(null);
+                    return;
+                }
+                return; // Ignore keys while pulling
+            }
+            return;
+        }
         // Theme picker navigation
         if (themePicker) {
             const themeKeys = listThemes();
@@ -1266,7 +1703,14 @@ function App() {
         }
         if (key.ctrl && inputChar === "c") {
             if (ctrlCPressed) {
-                exit();
+                // Force quit on second Ctrl+C — don't block
+                const config = loadConfig();
+                if (config.defaults.stopOllamaOnExit) {
+                    stopOllama().finally(() => exit());
+                }
+                else {
+                    exit();
+                }
             }
             else {
                 setCtrlCPressed(true);
@@ -1344,7 +1788,12 @@ function App() {
             })(), skillsPicker === "remove" && (() => {
                 const installed = listInstalledSkills();
                 return (_jsxs(Box, { flexDirection: "column", borderStyle: "single", borderColor: theme.colors.error, paddingX: 1, marginBottom: 0, children: [_jsx(Text, { bold: true, color: theme.colors.error, children: "Remove a skill:" }), installed.map((s, i) => (_jsxs(Text, { children: [i === skillsPickerIndex ? _jsx(Text, { color: theme.colors.suggestion, bold: true, children: "▸ " }) : _jsx(Text, { children: "  " }), _jsxs(Text, { color: i === skillsPickerIndex ? theme.colors.suggestion : theme.colors.muted, children: [s.name, " \u2014 ", s.description] })] }, s.name))), _jsx(Text, { dimColor: true, children: "  ↑↓ navigate · Enter remove · Esc back" })] }));
-            })(), themePicker && (_jsxs(Box, { flexDirection: "column", borderStyle: "single", borderColor: theme.colors.border, paddingX: 1, marginBottom: 0, children: [_jsx(Text, { bold: true, color: theme.colors.secondary, children: "Choose a theme:" }), listThemes().map((key, i) => (_jsxs(Text, { children: [i === themePickerIndex ? _jsx(Text, { color: theme.colors.suggestion, bold: true, children: "▸ " }) : _jsx(Text, { children: "  " }), _jsx(Text, { color: i === themePickerIndex ? theme.colors.suggestion : theme.colors.primary, bold: true, children: key }), _jsxs(Text, { color: theme.colors.muted, children: [" — ", THEMES[key].description] }), key === theme.name.toLowerCase() ? _jsx(Text, { color: theme.colors.muted, children: " (current)" }) : null] }, key))), _jsx(Text, { dimColor: true, children: "  ↑↓ navigate · Enter select · Esc cancel" })] })), sessionPicker && (_jsxs(Box, { flexDirection: "column", borderStyle: "single", borderColor: theme.colors.secondary, paddingX: 1, marginBottom: 0, children: [_jsx(Text, { bold: true, color: theme.colors.secondary, children: "Resume a session:" }), sessionPicker.map((s, i) => (_jsxs(Text, { children: [i === sessionPickerIndex ? _jsx(Text, { color: theme.colors.suggestion, bold: true, children: "▸ " }) : _jsx(Text, { children: "  " }), _jsx(Text, { color: i === sessionPickerIndex ? theme.colors.suggestion : theme.colors.muted, children: s.display })] }, s.id))), _jsx(Text, { dimColor: true, children: "  ↑↓ navigate · Enter select · Esc cancel" })] })), deleteSessionPicker && (_jsxs(Box, { flexDirection: "column", borderStyle: "single", borderColor: theme.colors.error, paddingX: 1, marginBottom: 0, children: [_jsx(Text, { bold: true, color: theme.colors.error, children: "Delete a session:" }), deleteSessionPicker.map((s, i) => (_jsxs(Text, { children: [i === deleteSessionPickerIndex ? _jsx(Text, { color: theme.colors.suggestion, bold: true, children: "▸ " }) : _jsx(Text, { children: "  " }), _jsx(Text, { color: i === deleteSessionPickerIndex ? theme.colors.suggestion : theme.colors.muted, children: s.display })] }, s.id))), _jsx(Text, { dimColor: true, children: "  ↑↓ navigate · Enter select · Esc cancel" })] })), deleteSessionConfirm && (_jsxs(Box, { flexDirection: "column", borderStyle: "single", borderColor: theme.colors.warning, paddingX: 1, marginBottom: 0, children: [_jsxs(Text, { bold: true, color: theme.colors.warning, children: ["Delete session ", deleteSessionConfirm.id, "?"] }), _jsxs(Text, { color: theme.colors.muted, children: ["  ", deleteSessionConfirm.display] }), _jsxs(Text, { children: [_jsx(Text, { color: theme.colors.error, bold: true, children: " [y]" }), _jsx(Text, { children: "es  " }), _jsx(Text, { color: theme.colors.success, bold: true, children: "[n]" }), _jsx(Text, { children: "o" })] })] })), showSuggestions && (_jsxs(Box, { flexDirection: "column", borderStyle: "single", borderColor: theme.colors.muted, paddingX: 1, marginBottom: 0, children: [cmdMatches.slice(0, 6).map((c, i) => (_jsxs(Text, { children: [i === cmdIndex ? _jsx(Text, { color: theme.colors.suggestion, bold: true, children: "▸ " }) : _jsx(Text, { children: "  " }), _jsx(Text, { color: i === cmdIndex ? theme.colors.suggestion : theme.colors.primary, bold: true, children: c.cmd }), _jsxs(Text, { color: theme.colors.muted, children: [" — ", c.desc] })] }, i))), _jsx(Text, { dimColor: true, children: "  ↑↓ navigate · Tab select" })] })), _jsxs(Box, { borderStyle: "single", borderColor: approval ? theme.colors.warning : theme.colors.border, paddingX: 1, children: [_jsx(Text, { color: theme.colors.secondary, bold: true, children: "> " }), approval ? (_jsx(Text, { color: theme.colors.warning, children: "waiting for approval..." })) : ready && !loading ? (_jsxs(Box, { children: [pastedChunks.map((p) => (_jsxs(Text, { color: theme.colors.muted, children: ["[Pasted text #", p.id, " +", p.lines, " lines]"] }, p.id))), _jsx(TextInput, { value: input, onChange: (v) => { setInput(v); setCmdIndex(0); }, onSubmit: handleSubmit }, inputKey)] })) : (_jsx(Text, { dimColor: true, children: loading ? "waiting for response..." : "initializing..." }))] }), agent && (_jsx(Box, { paddingX: 2, children: _jsxs(Text, { dimColor: true, children: ["💬 ", agent.getContextLength(), " messages · ~", (() => {
+            })(), themePicker && (_jsxs(Box, { flexDirection: "column", borderStyle: "single", borderColor: theme.colors.border, paddingX: 1, marginBottom: 0, children: [_jsx(Text, { bold: true, color: theme.colors.secondary, children: "Choose a theme:" }), listThemes().map((key, i) => (_jsxs(Text, { children: [i === themePickerIndex ? _jsx(Text, { color: theme.colors.suggestion, bold: true, children: "▸ " }) : _jsx(Text, { children: "  " }), _jsx(Text, { color: i === themePickerIndex ? theme.colors.suggestion : theme.colors.primary, bold: true, children: key }), _jsxs(Text, { color: theme.colors.muted, children: [" — ", THEMES[key].description] }), key === theme.name.toLowerCase() ? _jsx(Text, { color: theme.colors.muted, children: " (current)" }) : null] }, key))), _jsx(Text, { dimColor: true, children: "  ↑↓ navigate · Enter select · Esc cancel" })] })), sessionPicker && (_jsxs(Box, { flexDirection: "column", borderStyle: "single", borderColor: theme.colors.secondary, paddingX: 1, marginBottom: 0, children: [_jsx(Text, { bold: true, color: theme.colors.secondary, children: "Resume a session:" }), sessionPicker.map((s, i) => (_jsxs(Text, { children: [i === sessionPickerIndex ? _jsx(Text, { color: theme.colors.suggestion, bold: true, children: "▸ " }) : _jsx(Text, { children: "  " }), _jsx(Text, { color: i === sessionPickerIndex ? theme.colors.suggestion : theme.colors.muted, children: s.display })] }, s.id))), _jsx(Text, { dimColor: true, children: "  ↑↓ navigate · Enter select · Esc cancel" })] })), deleteSessionPicker && (_jsxs(Box, { flexDirection: "column", borderStyle: "single", borderColor: theme.colors.error, paddingX: 1, marginBottom: 0, children: [_jsx(Text, { bold: true, color: theme.colors.error, children: "Delete a session:" }), deleteSessionPicker.map((s, i) => (_jsxs(Text, { children: [i === deleteSessionPickerIndex ? _jsx(Text, { color: theme.colors.suggestion, bold: true, children: "▸ " }) : _jsx(Text, { children: "  " }), _jsx(Text, { color: i === deleteSessionPickerIndex ? theme.colors.suggestion : theme.colors.muted, children: s.display })] }, s.id))), _jsx(Text, { dimColor: true, children: "  ↑↓ navigate · Enter select · Esc cancel" })] })), deleteSessionConfirm && (_jsxs(Box, { flexDirection: "column", borderStyle: "single", borderColor: theme.colors.warning, paddingX: 1, marginBottom: 0, children: [_jsxs(Text, { bold: true, color: theme.colors.warning, children: ["Delete session ", deleteSessionConfirm.id, "?"] }), _jsxs(Text, { color: theme.colors.muted, children: ["  ", deleteSessionConfirm.display] }), _jsxs(Text, { children: [_jsx(Text, { color: theme.colors.error, bold: true, children: " [y]" }), _jsx(Text, { children: "es  " }), _jsx(Text, { color: theme.colors.success, bold: true, children: "[n]" }), _jsx(Text, { children: "o" })] })] })), ollamaDeleteConfirm && (_jsxs(Box, { flexDirection: "column", borderStyle: "single", borderColor: theme.colors.warning, paddingX: 1, marginBottom: 0, children: [_jsxs(Text, { bold: true, color: theme.colors.warning, children: ["Delete ", ollamaDeleteConfirm.model, " (", (ollamaDeleteConfirm.size / (1024 * 1024 * 1024)).toFixed(1), " GB)?"] }), _jsxs(Text, { children: [_jsx(Text, { color: theme.colors.error, bold: true, children: " [y]" }), _jsx(Text, { children: "es  " }), _jsx(Text, { color: theme.colors.success, bold: true, children: "[n]" }), _jsx(Text, { children: "o" })] })] })), ollamaPulling && (_jsxs(Box, { flexDirection: "column", borderStyle: "single", borderColor: theme.colors.border, paddingX: 1, marginBottom: 0, children: [_jsxs(Text, { bold: true, color: theme.colors.secondary, children: ["  Downloading ", ollamaPulling.model, "..."] }), ollamaPulling.progress.status === "downloading" || ollamaPulling.progress.percent > 0 ? (_jsxs(Text, { children: ["  ", _jsxs(Text, { color: theme.colors.primary, children: ["\u2588".repeat(Math.floor(ollamaPulling.progress.percent / 5)), "\u2591".repeat(20 - Math.floor(ollamaPulling.progress.percent / 5))] }), "  ", _jsxs(Text, { bold: true, children: [ollamaPulling.progress.percent, "%"] }), ollamaPulling.progress.completed != null && ollamaPulling.progress.total != null ? (_jsxs(Text, { color: theme.colors.muted, children: [" \u00B7 ", formatBytes(ollamaPulling.progress.completed), " / ", formatBytes(ollamaPulling.progress.total)] })) : null] })) : (_jsxs(Text, { color: theme.colors.muted, children: ["  ", ollamaPulling.progress.status, "..."] }))] })), ollamaExitPrompt && (_jsxs(Box, { flexDirection: "column", borderStyle: "single", borderColor: theme.colors.warning, paddingX: 1, marginBottom: 0, children: [_jsx(Text, { bold: true, color: theme.colors.warning, children: "Ollama is still running." }), _jsx(Text, { color: theme.colors.muted, children: "Stop it to free GPU memory?" }), _jsxs(Text, { children: [_jsx(Text, { color: theme.colors.success, bold: true, children: " [y]" }), _jsx(Text, { children: "es  " }), _jsx(Text, { color: theme.colors.error, bold: true, children: "[n]" }), _jsx(Text, { children: "o  " }), _jsx(Text, { color: theme.colors.primary, bold: true, children: "[a]" }), _jsx(Text, { children: "lways" })] })] })), wizardScreen === "connection" && (_jsxs(Box, { flexDirection: "column", borderStyle: "single", borderColor: theme.colors.border, paddingX: 1, marginBottom: 0, children: [_jsx(Text, { bold: true, color: theme.colors.secondary, children: "No LLM detected. How do you want to connect?" }), _jsx(Text, { children: "" }), [
+                        { key: "local", icon: "\uD83D\uDDA5\uFE0F", label: "Set up a local model", desc: "free, runs on your machine" },
+                        { key: "openrouter", icon: "\uD83C\uDF10", label: "OpenRouter", desc: "200+ cloud models, browser login" },
+                        { key: "apikey", icon: "\uD83D\uDD11", label: "Enter API key manually", desc: "" },
+                        { key: "existing", icon: "\u2699\uFE0F", label: "I already have a server running", desc: "" },
+                    ].map((item, i) => (_jsxs(Text, { children: [i === wizardIndex ? _jsx(Text, { color: theme.colors.suggestion, bold: true, children: "  \u25B8 " }) : _jsx(Text, { children: "    " }), _jsxs(Text, { color: i === wizardIndex ? theme.colors.suggestion : theme.colors.primary, bold: true, children: [item.icon, "  ", item.label] }), item.desc ? _jsxs(Text, { color: theme.colors.muted, children: [" (", item.desc, ")"] }) : null] }, item.key))), _jsx(Text, { children: "" }), _jsx(Text, { dimColor: true, children: "  \u2191\u2193 navigate \u00B7 Enter to select" })] })), wizardScreen === "models" && wizardHardware && (_jsxs(Box, { flexDirection: "column", borderStyle: "single", borderColor: theme.colors.border, paddingX: 1, marginBottom: 0, children: [_jsx(Text, { bold: true, color: theme.colors.secondary, children: "Your hardware:" }), _jsxs(Text, { color: theme.colors.muted, children: ["  CPU: ", wizardHardware.cpu.name, " (", wizardHardware.cpu.cores, " cores)"] }), _jsxs(Text, { color: theme.colors.muted, children: ["  RAM: ", formatBytes(wizardHardware.ram)] }), wizardHardware.gpu ? (_jsxs(Text, { color: theme.colors.muted, children: ["  GPU: ", wizardHardware.gpu.name, wizardHardware.gpu.vram > 0 ? ` (${formatBytes(wizardHardware.gpu.vram)})` : ""] })) : (_jsx(Text, { color: theme.colors.muted, children: "  GPU: none detected" })), !isLlmfitAvailable() && (_jsx(Text, { dimColor: true, children: "  Tip: Install llmfit for smarter recommendations: brew install llmfit" })), _jsx(Text, { children: "" }), _jsx(Text, { bold: true, color: theme.colors.secondary, children: "Recommended models:" }), _jsx(Text, { children: "" }), wizardModels.map((m, i) => (_jsxs(Text, { children: [i === wizardIndex ? _jsx(Text, { color: theme.colors.suggestion, bold: true, children: "  \u25B8 " }) : _jsx(Text, { children: "    " }), _jsxs(Text, { children: [getFitIcon(m.fit), " "] }), _jsx(Text, { color: i === wizardIndex ? theme.colors.suggestion : theme.colors.primary, bold: true, children: m.name }), _jsxs(Text, { color: theme.colors.muted, children: ["     ~", m.size, " GB \u00B7 ", m.quality === "best" ? "Best" : m.quality === "great" ? "Great" : "Good", " quality \u00B7 ", m.speed] })] }, m.ollamaId))), wizardModels.length === 0 && (_jsx(Text, { color: theme.colors.error, children: "  No suitable models found for your hardware." })), _jsx(Text, { children: "" }), _jsx(Text, { dimColor: true, children: "  \u2191\u2193 navigate \u00B7 Enter to install \u00B7 Esc back" })] })), wizardScreen === "install-ollama" && (_jsxs(Box, { flexDirection: "column", borderStyle: "single", borderColor: theme.colors.warning, paddingX: 1, marginBottom: 0, children: [_jsx(Text, { bold: true, color: theme.colors.warning, children: "Ollama is required for local models." }), _jsx(Text, { children: "" }), _jsxs(Text, { color: theme.colors.primary, children: ["  Install with: ", _jsx(Text, { bold: true, children: getOllamaInstallCommand(wizardHardware?.os ?? "linux") })] }), _jsx(Text, { children: "" }), _jsx(Text, { dimColor: true, children: "  Run the command above, then press Enter to continue..." }), _jsx(Text, { dimColor: true, children: "  Esc to go back" })] })), wizardScreen === "pulling" && wizardSelectedModel && (_jsx(Box, { flexDirection: "column", borderStyle: "single", borderColor: theme.colors.border, paddingX: 1, marginBottom: 0, children: wizardPullError ? (_jsxs(_Fragment, { children: [_jsxs(Text, { color: theme.colors.error, bold: true, children: ["  \u274C Error: ", wizardPullError] }), _jsx(Text, { children: "" }), _jsx(Text, { dimColor: true, children: "  Press Enter to retry \u00B7 Esc to go back" })] })) : wizardPullProgress ? (_jsxs(_Fragment, { children: [_jsxs(Text, { bold: true, color: theme.colors.secondary, children: ["  Downloading ", wizardSelectedModel.name, "..."] }), wizardPullProgress.status === "downloading" || wizardPullProgress.percent > 0 ? (_jsx(_Fragment, { children: _jsxs(Text, { children: ["  ", _jsxs(Text, { color: theme.colors.primary, children: ["\u2588".repeat(Math.floor(wizardPullProgress.percent / 5)), "\u2591".repeat(20 - Math.floor(wizardPullProgress.percent / 5))] }), "  ", _jsxs(Text, { bold: true, children: [wizardPullProgress.percent, "%"] }), wizardPullProgress.completed != null && wizardPullProgress.total != null ? (_jsxs(Text, { color: theme.colors.muted, children: [" \u00B7 ", formatBytes(wizardPullProgress.completed), " / ", formatBytes(wizardPullProgress.total)] })) : null] }) })) : (_jsxs(Text, { color: theme.colors.muted, children: ["  ", wizardPullProgress.status, "..."] }))] })) : null })), showSuggestions && (_jsxs(Box, { flexDirection: "column", borderStyle: "single", borderColor: theme.colors.muted, paddingX: 1, marginBottom: 0, children: [cmdMatches.slice(0, 6).map((c, i) => (_jsxs(Text, { children: [i === cmdIndex ? _jsx(Text, { color: theme.colors.suggestion, bold: true, children: "▸ " }) : _jsx(Text, { children: "  " }), _jsx(Text, { color: i === cmdIndex ? theme.colors.suggestion : theme.colors.primary, bold: true, children: c.cmd }), _jsxs(Text, { color: theme.colors.muted, children: [" — ", c.desc] })] }, i))), _jsx(Text, { dimColor: true, children: "  ↑↓ navigate · Tab select" })] })), _jsxs(Box, { borderStyle: "single", borderColor: approval ? theme.colors.warning : theme.colors.border, paddingX: 1, children: [_jsx(Text, { color: theme.colors.secondary, bold: true, children: "> " }), approval ? (_jsx(Text, { color: theme.colors.warning, children: "waiting for approval..." })) : ready && !loading ? (_jsxs(Box, { children: [pastedChunks.map((p) => (_jsxs(Text, { color: theme.colors.muted, children: ["[Pasted text #", p.id, " +", p.lines, " lines]"] }, p.id))), _jsx(TextInput, { value: input, onChange: (v) => { setInput(v); setCmdIndex(0); }, onSubmit: handleSubmit }, inputKey)] })) : (_jsx(Text, { dimColor: true, children: loading ? "waiting for response..." : "initializing..." }))] }), agent && (_jsx(Box, { paddingX: 2, children: _jsxs(Text, { dimColor: true, children: ["💬 ", agent.getContextLength(), " messages · ~", (() => {
                             const tokens = agent.estimateTokens();
                             return tokens >= 1000 ? `${(tokens / 1000).toFixed(1)}k` : String(tokens);
                         })(), " tokens", (() => {

package/dist/utils/hardware.d.ts

@@ -0,0 +1,17 @@
+export interface HardwareInfo {
+    cpu: {
+        name: string;
+        cores: number;
+        speed: number;
+    };
+    ram: number;
+    gpu: {
+        name: string;
+        vram: number;
+    } | null;
+    os: "macos" | "linux" | "windows";
+    appleSilicon: boolean;
+}
+export declare function detectHardware(): HardwareInfo;
+/** Format bytes to human-readable string */
+export declare function formatBytes(bytes: number): string;