rlm-cli 0.2.0 → 0.2.2

package/README.md

@@ -13,34 +13,38 @@ CLI for **Recursive Language Models** — based on the [RLM paper](https://arxiv
 
 Instead of dumping a huge context into a single LLM call, RLM lets the model write Python code to process it — slicing, chunking, running sub-queries on pieces, and building up an answer across multiple iterations.
 
-## Quick Start
+## Install
+
+```bash
+npm install -g rlm-cli
+```
+
+Set your API key:
+
+```bash
+export ANTHROPIC_API_KEY=sk-ant-...
+# or
+export OPENAI_API_KEY=sk-...
+```
+
+That's it. Run `rlm` and you're in.
+
+### From Source
 
 ```bash
 git clone https://github.com/viplismism/rlm-cli.git
 cd rlm-cli
 npm install
 npm run build
-npm link   # makes `rlm` available globally
+npm link
 ```
 
-Then create a `.env` file with your API key:
+Create a `.env` file in the project root with your API key:
 
 ```bash
 cp .env.example .env
 ```
 
-```bash
-# .env
-ANTHROPIC_API_KEY=sk-ant-...
-# or
-OPENAI_API_KEY=sk-...
-
-# Optional: override default model
-# RLM_MODEL=claude-sonnet-4-5-20250929
-```
-
-That's it. Run `rlm` and you're in.
-
 ## Usage
 
 ### Interactive Terminal

package/dist/interactive.js

@@ -153,19 +153,23 @@ function printWelcome() {
 function printCommandHelp() {
     console.log(`
 ${c.bold}Context${c.reset}
-  ${c.yellow}/file${c.reset} <path>         Load file as context
-  ${c.yellow}/url${c.reset} <url>           Fetch URL as context
-  ${c.yellow}/paste${c.reset}               Multi-line paste mode (EOF to finish)
-  ${c.yellow}/context${c.reset}             Show loaded context info
-  ${c.yellow}/clear-context${c.reset}       Unload context
+  ${c.cyan}/file${c.reset} <path>         Load file as context
+  ${c.cyan}/url${c.reset} <url>           Fetch URL as context
+  ${c.cyan}/paste${c.reset}               Multi-line paste mode (EOF to finish)
+  ${c.cyan}/context${c.reset}             Show loaded context info
+  ${c.cyan}/clear-context${c.reset}       Unload context
+
+${c.bold}Model${c.reset}
+  ${c.cyan}/model${c.reset}               List available models
+  ${c.cyan}/model${c.reset} <#|id>         Switch model by number or ID
 
 ${c.bold}Tools${c.reset}
-  ${c.yellow}/trajectories${c.reset}        List saved runs
+  ${c.cyan}/trajectories${c.reset}        List saved runs
 
 ${c.bold}General${c.reset}
-  ${c.yellow}/clear${c.reset}               Clear screen
-  ${c.yellow}/help${c.reset}                Show this help
-  ${c.yellow}/quit${c.reset}                Exit
+  ${c.cyan}/clear${c.reset}               Clear screen
+  ${c.cyan}/help${c.reset}                Show this help
+  ${c.cyan}/quit${c.reset}                Exit
 
   ${c.dim}Or just paste a URL or 4+ lines of code, then type your query.${c.reset}
 `);
@@ -353,6 +357,20 @@ function displaySubQueryResult(info) {
     }
     console.log(`    ${c.magenta}└─${c.reset} ${c.dim}${elapsed}s · ${formatSize(info.resultLength)} received${c.reset}`);
 }
+// ── Available models list ────────────────────────────────────────────────────
+/** Collect models from providers that have an API key set. */
+function getAvailableModels() {
+    const items = [];
+    for (const provider of getProviders()) {
+        const providerKey = `${provider.toUpperCase().replace(/-/g, "_")}_API_KEY`;
+        if (!process.env[providerKey] && provider !== detectProvider())
+            continue;
+        for (const m of getModels(provider)) {
+            items.push({ id: m.id, provider });
+        }
+    }
+    return items;
+}
 // ── Truncate helper ─────────────────────────────────────────────────────────
 function truncateStr(text, max) {
     return text.length <= max ? text : text.slice(0, max - 3) + "...";
@@ -595,9 +613,13 @@ async function interactive() {
     // Validate env
     const hasApiKey = process.env.ANTHROPIC_API_KEY || process.env.OPENAI_API_KEY;
     if (!hasApiKey) {
-        console.log(`\n  ${c.red}No API key found.${c.reset}`);
-        console.log(`  Set ${c.bold}ANTHROPIC_API_KEY${c.reset} or ${c.bold}OPENAI_API_KEY${c.reset} in your .env file.\n`);
-        process.exit(1);
+        printBanner();
+        console.log(`  ${c.red}No API key found.${c.reset}\n`);
+        console.log(`  Set one of these environment variables:\n`);
+        console.log(`    ${c.yellow}export ANTHROPIC_API_KEY=sk-ant-...${c.reset}    ${c.dim}# Anthropic (Claude)${c.reset}`);
+        console.log(`    ${c.yellow}export OPENAI_API_KEY=sk-...${c.reset}           ${c.dim}# OpenAI (GPT)${c.reset}\n`);
+        console.log(`  ${c.dim}Add to your shell profile (~/.zshrc or ~/.bashrc) to persist across sessions.${c.reset}\n`);
+        process.exit(0);
     }
     // Resolve model
     currentModel = resolveModel(currentModelId);
@@ -613,6 +635,21 @@ async function interactive() {
         prompt: `${c.cyan}>${c.reset} `,
         terminal: true,
     });
+    // Color slash commands cyan as the user types
+    const rlAny = rl;
+    const promptStr = rl.getPrompt();
+    rlAny._writeToOutput = function (str) {
+        if (!rlAny.line?.startsWith("/")) {
+            rlAny.output.write(str);
+            return;
+        }
+        if (str.startsWith(promptStr)) {
+            rlAny.output.write(promptStr + c.cyan + str.slice(promptStr.length) + c.reset);
+        }
+        else {
+            rlAny.output.write(c.cyan + str + c.reset);
+        }
+    };
     rl.prompt();
     rl.on("line", async (rawLine) => {
         if (isRunning)
@@ -675,6 +712,40 @@ async function interactive() {
                     contextSource = "";
                     console.log(`  ${c.green}✓${c.reset} Context cleared.`);
                     break;
+                case "model":
+                case "m": {
+                    const available = getAvailableModels();
+                    if (arg) {
+                        // Accept a number or a model ID
+                        const pick = /^\d+$/.test(arg) ? available[parseInt(arg, 10) - 1]?.id : arg;
+                        const newModel = pick ? resolveModel(pick) : undefined;
+                        if (newModel && pick) {
+                            currentModelId = pick;
+                            currentModel = newModel;
+                            console.log(`  ${c.green}✓${c.reset} Switched to ${c.bold}${currentModelId}${c.reset}`);
+                            console.log();
+                            printStatusLine();
+                        }
+                        else {
+                            console.log(`  ${c.red}Model "${arg}" not found.${c.reset} Use ${c.cyan}/model${c.reset} to list available models.`);
+                        }
+                    }
+                    else {
+                        console.log(`\n  ${c.bold}Current model:${c.reset} ${c.cyan}${currentModelId}${c.reset}\n`);
+                        const pad = String(available.length).length;
+                        for (let i = 0; i < available.length; i++) {
+                            const m = available[i];
+                            const num = String(i + 1).padStart(pad);
+                            const dot = m.id === currentModelId ? `${c.green}●${c.reset}` : ` `;
+                            const label = m.id === currentModelId
+                                ? `${c.cyan}${m.id}${c.reset}`
+                                : `${c.dim}${m.id}${c.reset}`;
+                            console.log(`  ${c.dim}${num}${c.reset} ${dot} ${label}`);
+                        }
+                        console.log(`\n  ${c.dim}Type${c.reset} ${c.cyan}/model <number>${c.reset} ${c.dim}or${c.reset} ${c.cyan}/model <id>${c.reset} ${c.dim}to switch.${c.reset}`);
+                    }
+                    break;
+                }
                 case "trajectories":
                 case "traj":
                     handleTrajectories();
@@ -689,7 +760,7 @@ async function interactive() {
                     process.exit(0);
                     break;
                 default:
-                    console.log(`  ${c.red}Unknown command: /${cmd}${c.reset}. Type ${c.yellow}/help${c.reset} for commands.`);
+                    console.log(`  ${c.red}Unknown command: /${cmd}${c.reset}. Type ${c.cyan}/help${c.reset} for commands.`);
             }
             rl.prompt();
             return;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "rlm-cli",
-  "version": "0.2.0",
+  "version": "0.2.2",
   "description": "Standalone CLI for Recursive Language Models (RLMs) — implements Algorithm 1 from arXiv:2512.24601",
   "type": "module",
   "bin": {
@@ -32,7 +32,7 @@
   ],
   "license": "MIT",
   "dependencies": {
-    "@mariozechner/pi-ai": "^0.53.0"
+    "@mariozechner/pi-ai": "^0.55.1"
   },
   "devDependencies": {
     "esbuild": "^0.27.3",