swarm-code 0.1.1 → 0.1.3

package/dist/interactive.js

@@ -8,6 +8,7 @@
  *   - Browse previous trajectories
  */
 import "./env.js";
+import { execFileSync, execSync, spawn as spawnChild } from "node:child_process";
 import * as fs from "node:fs";
 import * as os from "node:os";
 import * as path from "node:path";
@@ -306,6 +307,219 @@ function handleMultiLineAsContext(input) {
     return null;
 }
 // ── Banner ──────────────────────────────────────────────────────────────────
+// ── Ollama helpers ──────────────────────────────────────────────────────────
+function isOllamaInstalled() {
+    try {
+        execFileSync("ollama", ["--version"], { stdio: ["ignore", "pipe", "pipe"], timeout: 5000 });
+        return true;
+    }
+    catch {
+        return false;
+    }
+}
+function isOllamaModelAvailable(model) {
+    try {
+        const output = execFileSync("ollama", ["list"], {
+            encoding: "utf-8",
+            stdio: ["ignore", "pipe", "pipe"],
+            timeout: 10000,
+        });
+        return output.includes(model);
+    }
+    catch {
+        return false;
+    }
+}
+async function installOllama() {
+    console.log(`\n  ${c.bold}Installing Ollama...${c.reset}\n`);
+    if (process.platform === "darwin") {
+        // macOS: check for brew first, otherwise use the install script
+        try {
+            execFileSync("brew", ["--version"], { stdio: "ignore", timeout: 5000 });
+            console.log(`  ${c.dim}Using Homebrew...${c.reset}`);
+            try {
+                execSync("brew install ollama", { stdio: "inherit", timeout: 120000 });
+                return true;
+            }
+            catch {
+                console.log(`  ${c.dim}Homebrew install failed, trying curl installer...${c.reset}`);
+            }
+        }
+        catch {
+            // No brew — fall through to curl
+        }
+    }
+    // Linux / macOS fallback: official install script
+    if (process.platform === "linux" || process.platform === "darwin") {
+        try {
+            execSync("curl -fsSL https://ollama.com/install.sh | sh", { stdio: "inherit", timeout: 180000 });
+            return true;
+        }
+        catch {
+            return false;
+        }
+    }
+    // Windows: direct user to download page
+    console.log(`  ${c.dim}Download Ollama from: https://ollama.com/download${c.reset}`);
+    return false;
+}
+function isOllamaServing() {
+    try {
+        execFileSync("curl", ["-sf", "http://127.0.0.1:11434/api/tags"], {
+            stdio: ["ignore", "pipe", "pipe"],
+            timeout: 3000,
+        });
+        return true;
+    }
+    catch {
+        return false;
+    }
+}
+function startOllamaServe() {
+    // Start ollama serve in the background (detached)
+    const child = spawnChild("ollama", ["serve"], {
+        stdio: "ignore",
+        detached: true,
+    });
+    child.unref();
+}
+async function pullOllamaModel(model) {
+    console.log(`\n  ${c.bold}Pulling ${model}...${c.reset} ${c.dim}(this may take a few minutes)${c.reset}\n`);
+    return new Promise((resolve) => {
+        const child = spawnChild("ollama", ["pull", model], { stdio: "inherit" });
+        child.on("close", (code) => resolve(code === 0));
+        child.on("error", () => resolve(false));
+    });
+}
+async function ensureOllamaSetup(rl, model) {
+    const shortModel = model.replace("ollama/", "");
+    // 1. Check if Ollama is installed
+    if (!isOllamaInstalled()) {
+        console.log(`\n  ${c.dim}Ollama is not installed. It's needed to run open-source models locally.${c.reset}`);
+        const install = await questionWithEsc(rl, `  ${c.cyan}Install Ollama now? [Y/n]:${c.reset} `);
+        if (install !== null && install.toLowerCase() !== "n" && install.toLowerCase() !== "no") {
+            const ok = await installOllama();
+            if (!ok) {
+                console.log(`\n  ${c.red}Failed to install Ollama.${c.reset}`);
+                console.log(`  ${c.dim}Install manually from https://ollama.com/download${c.reset}\n`);
+                return false;
+            }
+            console.log(`  ${c.green}✓${c.reset} Ollama installed`);
+        }
+        else {
+            console.log(`\n  ${c.dim}Install later from https://ollama.com/download${c.reset}\n`);
+            return false;
+        }
+    }
+    else {
+        console.log(`  ${c.green}✓${c.reset} Ollama installed`);
+    }
+    // 2. Ensure ollama serve is running
+    if (!isOllamaServing()) {
+        console.log(`  ${c.dim}Starting Ollama server...${c.reset}`);
+        startOllamaServe();
+        // Give it a moment to start
+        let retries = 10;
+        while (retries > 0 && !isOllamaServing()) {
+            await new Promise((r) => setTimeout(r, 1000));
+            retries--;
+        }
+        if (isOllamaServing()) {
+            console.log(`  ${c.green}✓${c.reset} Ollama server running`);
+        }
+        else {
+            console.log(`  ${c.yellow}⚠${c.reset} Could not start Ollama server. Run ${c.bold}ollama serve${c.reset} manually.`);
+            return false;
+        }
+    }
+    else {
+        console.log(`  ${c.green}✓${c.reset} Ollama server running`);
+    }
+    // 3. Check if model is already pulled
+    if (isOllamaModelAvailable(shortModel)) {
+        console.log(`  ${c.green}✓${c.reset} Model ${c.bold}${shortModel}${c.reset} ready\n`);
+        return true;
+    }
+    // 4. Pull the model
+    console.log(`  ${c.dim}Model ${shortModel} not found locally.${c.reset}`);
+    const pull = await questionWithEsc(rl, `\n  ${c.cyan}Pull ${shortModel} now? [Y/n]:${c.reset} `);
+    if (pull !== null && pull.toLowerCase() !== "n" && pull.toLowerCase() !== "no") {
+        const ok = await pullOllamaModel(shortModel);
+        if (!ok) {
+            console.log(`\n  ${c.red}Failed to pull ${shortModel}.${c.reset}`);
+            console.log(`  ${c.dim}Try manually: ollama pull ${shortModel}${c.reset}\n`);
+            return false;
+        }
+        console.log(`  ${c.green}✓${c.reset} Model ${c.bold}${shortModel}${c.reset} ready\n`);
+        return true;
+    }
+    console.log(`\n  ${c.dim}Run later: ollama pull ${shortModel}${c.reset}\n`);
+    return false;
+}
+/**
+ * Interactive checkbox selector — arrow keys to navigate, space to toggle, enter to confirm.
+ * Returns indices of selected items.
+ */
+function checkboxSelect(items) {
+    return new Promise((resolve) => {
+        let cursor = 0;
+        const render = () => {
+            // Move cursor up to overwrite previous render (except first time)
+            if (rendered) {
+                process.stdout.write(`\x1b[${items.length}A`);
+            }
+            for (let i = 0; i < items.length; i++) {
+                const item = items[i];
+                const pointer = i === cursor ? `${c.cyan}❯${c.reset}` : " ";
+                const box = item.checked ? `${c.green}◼${c.reset}` : `${c.dim}◻${c.reset}`;
+                const label = i === cursor ? `${c.bold}${item.label}${c.reset}` : item.label;
+                const desc = `${c.dim}${item.desc}${c.reset}`;
+                process.stdout.write(`\x1b[2K  ${pointer} ${box} ${label}  ${desc}\n`);
+            }
+        };
+        let rendered = false;
+        render();
+        rendered = true;
+        const wasRaw = stdin.isRaw;
+        if (stdin.isTTY)
+            stdin.setRawMode(true);
+        const onData = (data) => {
+            const key = data.toString();
+            if (key === "\x1b[A" || key === "k") {
+                // Up
+                cursor = (cursor - 1 + items.length) % items.length;
+                render();
+            }
+            else if (key === "\x1b[B" || key === "j") {
+                // Down
+                cursor = (cursor + 1) % items.length;
+                render();
+            }
+            else if (key === " ") {
+                // Toggle
+                items[cursor].checked = !items[cursor].checked;
+                render();
+            }
+            else if (key === "\r" || key === "\n") {
+                // Confirm
+                stdin.removeListener("data", onData);
+                if (stdin.isTTY)
+                    stdin.setRawMode(wasRaw ?? false);
+                const selected = items.map((item, i) => (item.checked ? i : -1)).filter((i) => i >= 0);
+                resolve(selected);
+            }
+            else if (key === "\x1b" || key === "\x03") {
+                // Escape or Ctrl+C
+                stdin.removeListener("data", onData);
+                if (stdin.isTTY)
+                    stdin.setRawMode(wasRaw ?? false);
+                resolve([]);
+            }
+        };
+        stdin.on("data", onData);
+    });
+}
+// ── Banner ──────────────────────────────────────────────────────────────────
 function printBanner() {
     console.log(`
 ${c.cyan}${c.bold}
@@ -316,7 +530,7 @@ ${c.cyan}${c.bold}
                ███████║╚███╔███╔╝██║  ██║██║  ██║██║ ╚═╝ ██║
                ╚══════╝ ╚══╝╚══╝ ╚═╝  ╚═╝╚═╝  ╚═╝╚═╝     ╚═╝
 ${c.reset}
-${c.dim}            Swarm-native coding agent orchestrator${c.reset}
+${c.dim}                Swarm-native coding agent orchestrator${c.reset}
 `);
 }
 // ── Status line ─────────────────────────────────────────────────────────────
@@ -1323,90 +1537,78 @@ async function interactive() {
                 requiresKey: true,
             },
         ];
+        // ── Step 1: Checkbox agent selection ─────────────────────────
+        console.log(`  ${c.bold}Select your coding agent(s):${c.reset}`);
+        console.log(`  ${c.dim}↑/↓ navigate  ·  space toggle  ·  enter confirm${c.reset}\n`);
+        const checkboxItems = AGENT_CHOICES.map((a) => ({
+            label: a.name,
+            desc: a.desc,
+            checked: false,
+        }));
+        const selectedIndices = await checkboxSelect(checkboxItems);
+        if (selectedIndices.length === 0) {
+            console.log(`\n  ${c.dim}No agents selected. Exiting.${c.reset}\n`);
+            process.exit(0);
+        }
+        const selectedAgents = selectedIndices.map((i) => AGENT_CHOICES[i]);
+        console.log();
+        for (const agent of selectedAgents) {
+            console.log(`  ${c.green}✓${c.reset} ${c.bold}${agent.name}${c.reset}`);
+        }
+        console.log();
+        // ── Step 2: Configure each agent ─────────────────────────────
         const setupRl = readline.createInterface({ input: stdin, output: stdout, terminal: true });
-        let setupDone = false;
-        while (!setupDone) {
-            console.log(`  ${c.bold}Select your coding agent:${c.reset}\n`);
-            for (let i = 0; i < AGENT_CHOICES.length; i++) {
-                const a = AGENT_CHOICES[i];
-                console.log(`  ${c.dim}${i + 1}${c.reset}  ${a.name}  ${c.dim}${a.desc}${c.reset}`);
-            }
-            console.log();
-            const choice = await questionWithEsc(setupRl, `  ${c.cyan}Agent [1-${AGENT_CHOICES.length}]:${c.reset} `);
-            if (choice === null) {
-                console.log(`\n  ${c.dim}Exiting.${c.reset}\n`);
-                setupRl.close();
-                process.exit(0);
-            }
-            const idx = parseInt(choice, 10) - 1;
-            if (Number.isNaN(idx) || idx < 0 || idx >= AGENT_CHOICES.length) {
-                console.log(`\n  ${c.dim}Invalid choice.${c.reset}\n`);
-                continue;
-            }
-            const agent = AGENT_CHOICES[idx];
-            console.log(`\n  ${c.green}✓${c.reset} Agent: ${c.bold}${agent.name}${c.reset}\n`);
-            // Prompt for API key(s) one by one based on the selected agent
-            let gotAnyKey = false;
+        const configuredProviders = new Set();
+        let usesOllama = false;
+        for (const agent of selectedAgents) {
             if (agent.requiresKey) {
-                // Agent requires at least one key — prompt until we get one
-                console.log(`  ${c.dim}${agent.name} requires an API key. Configure one now:${c.reset}\n`);
+                // Agent needs at least one API key
+                console.log(`  ${c.bold}${agent.name}${c.reset} ${c.dim}— configure API key:${c.reset}\n`);
                 for (const provider of agent.keys) {
+                    if (configuredProviders.has(provider.env) || process.env[provider.env]) {
+                        console.log(`  ${c.green}✓${c.reset} ${provider.name} ${c.dim}(already configured)${c.reset}`);
+                        continue;
+                    }
                     const gotKey = await promptForProviderKey(setupRl, provider);
                     if (gotKey === null)
-                        break; // ESC — skip remaining
+                        break;
                     if (gotKey) {
-                        gotAnyKey = true;
-                        break; // Got one — that's enough
+                        configuredProviders.add(provider.env);
+                        break;
                     }
                 }
-                if (!gotAnyKey) {
-                    console.log(`\n  ${c.dim}No key provided. Try another agent or set keys in .env${c.reset}\n`);
-                    continue;
+                const agentHasKey = agent.keys.some((p) => configuredProviders.has(p.env) || process.env[p.env]);
+                if (!agentHasKey) {
+                    console.log(`  ${c.dim}No key for ${agent.name} — set one in .env later${c.reset}`);
                 }
+                console.log();
             }
             else {
-                // Agent works without keys (e.g. OpenCode with open-source models)
-                // Offer to configure a provider key optionally
-                console.log(`  ${c.dim}${agent.name} works with open-source models (no API key needed).${c.reset}`);
-                console.log(`  ${c.dim}Optionally configure a provider for cloud models:${c.reset}\n`);
-                for (const provider of agent.keys) {
-                    console.log(`    ${c.dim}-${c.reset} ${provider.name} ${c.dim}(${provider.env})${c.reset}`);
-                }
+                // Agent works without keys (e.g. OpenCode) — set up Ollama directly
+                console.log(`  ${c.bold}${agent.name}${c.reset} ${c.dim}— setting up Ollama for local models:${c.reset}\n`);
+                const ok = await ensureOllamaSetup(setupRl, "ollama/deepseek-coder-v2");
+                if (ok)
+                    usesOllama = true;
                 console.log();
-                const addKey = await questionWithEsc(setupRl, `  ${c.cyan}Add an API key? [y/N]:${c.reset} `);
-                if (addKey !== null && (addKey.toLowerCase() === "y" || addKey.toLowerCase() === "yes")) {
-                    for (const provider of agent.keys) {
-                        const gotKey = await promptForProviderKey(setupRl, provider);
-                        if (gotKey === null)
-                            break;
-                        if (gotKey) {
-                            gotAnyKey = true;
-                            break;
-                        }
-                    }
-                }
             }
-            // Set default model
-            const activeProvider = Object.keys(PROVIDER_KEYS).find((p) => process.env[providerEnvKey(p)]);
-            if (activeProvider) {
-                // Has an API key — use that provider's default model
-                currentProviderName = activeProvider;
-                const defaultModel = getDefaultModelForProvider(activeProvider);
-                if (defaultModel) {
-                    currentModelId = defaultModel;
-                    saveModelPreference(currentModelId);
-                    console.log(`  ${c.green}✓${c.reset} Default model: ${c.bold}${currentModelId}${c.reset}`);
-                }
-            }
-            else {
-                // No API key — default to open-source model via OpenCode
-                currentModelId = "ollama/deepseek-coder-v2";
+        }
+        // ── Step 3: Set default model ────────────────────────────────
+        const activeProvider = Object.keys(PROVIDER_KEYS).find((p) => process.env[providerEnvKey(p)]);
+        if (activeProvider) {
+            currentProviderName = activeProvider;
+            const defaultModel = getDefaultModelForProvider(activeProvider);
+            if (defaultModel) {
+                currentModelId = defaultModel;
                 saveModelPreference(currentModelId);
-                console.log(`  ${c.green}✓${c.reset} Default model: ${c.bold}${currentModelId}${c.reset} ${c.dim}(open-source, requires Ollama)${c.reset}`);
+                console.log(`  ${c.green}✓${c.reset} Default model: ${c.bold}${currentModelId}${c.reset}`);
             }
-            console.log();
-            setupDone = true;
         }
+        else if (usesOllama) {
+            currentModelId = "ollama/deepseek-coder-v2";
+            saveModelPreference(currentModelId);
+            console.log(`  ${c.green}✓${c.reset} Default model: ${c.bold}${currentModelId}${c.reset} ${c.dim}(local)${c.reset}`);
+        }
+        console.log();
         setupRl.close();
     }
     // Resolve model — ensure the resolved provider actually has an API key
@@ -1437,8 +1639,19 @@ async function interactive() {
         }
     }
     if (!currentModel) {
+        if (currentModelId.startsWith("ollama/")) {
+            // Ollama model selected — this interactive REPL mode needs a cloud API.
+            // Redirect to swarm mode which works with OpenCode + Ollama.
+            console.log(`\n  ${c.green}✓${c.reset} Ollama model selected: ${c.bold}${currentModelId}${c.reset}`);
+            console.log(`\n  ${c.dim}This interactive REPL uses direct LLM API calls.${c.reset}`);
+            console.log(`  ${c.dim}To use Ollama models with OpenCode, run:${c.reset}\n`);
+            console.log(`  ${c.bold}swarm --dir ./your-project "your task"${c.reset}\n`);
+            process.exit(0);
+        }
         console.log(`\n  ${c.red}Model "${currentModelId}" not found.${c.reset}`);
-        console.log(`  Check ${c.bold}RLM_MODEL${c.reset} in your .env file.\n`);
+        console.log(`  Set ${c.bold}ANTHROPIC_API_KEY${c.reset}, ${c.bold}OPENAI_API_KEY${c.reset}, or ${c.bold}GEMINI_API_KEY${c.reset} in your .env file.`);
+        console.log(`\n  ${c.dim}For agent-based mode (works with open-source models):${c.reset}`);
+        console.log(`  ${c.bold}swarm --dir ./your-project "your task"${c.reset}\n`);
         process.exit(1);
     }
     // Auto-load cwd context so the LLM knows the project structure

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "swarm-code",
-  "version": "0.1.1",
+  "version": "0.1.3",
   "description": "Open-source swarm-native coding agent orchestrator — spawns parallel coding agents in isolated git worktrees, built on RLM (arXiv:2512.24601)",
   "type": "module",
   "bin": {