aether-code 0.4.0 → 0.5.0

package/bin/aether-code.js

@@ -14,9 +14,10 @@ import { runRepl } from "../src/repl.js";
 import { runSetup } from "../src/setup.js";
 import { fetchBalance, AetherError } from "../src/api.js";
 import { writeConfigFile, getConfig, CONFIG_PATH } from "../src/config.js";
+import { generateAndApprovePlan } from "../src/plan.js";
 import { c, errorLine, divider } from "../src/render.js";
 
-const VERSION = "0.4.0";
+const VERSION = "0.5.0";
 
 const HELP = `${c.bold("aether")} — uncensored AI coding agent
 
@@ -40,6 +41,7 @@ ${c.bold("EXAMPLES")}
   aether --cwd ./my-project "fix the failing tests"
 
 ${c.bold("FLAGS")}
+  --plan             Show a numbered plan first; approve / refine / cancel before any tools fire.
   --yes              Auto-approve all writes and shell commands. Use with care.
   --cwd <path>       Working directory for the agent (default: current dir).
   --max-turns <n>    Maximum turns before stopping (default: 25).
@@ -65,6 +67,7 @@ function parseArgs(argv) {
   for (let i = 0; i < argv.length; i++) {
     const a = argv[i];
     if (a === "--yes") { args.flags.yes = true; }
+    else if (a === "--plan") { args.flags.plan = true; }
     else if (a === "--unsafe-paths") { args.flags.unsafePaths = true; }
     else if (a === "--help" || a === "-h") { args.flags.help = true; }
     else if (a === "--version" || a === "-v") { args.flags.version = true; }
@@ -140,12 +143,45 @@ async function main() {
   }
 
   console.log(divider());
-  console.log(c.magenta(c.bold("aether-code")) + c.gray(` · cwd ${cwd}${autoYes ? " · auto-yes" : ""}${unsafePaths ? " · unsafe-paths" : ""}`));
+  console.log(c.magenta(c.bold("aether-code")) + c.gray(` · cwd ${cwd}${autoYes ? " · auto-yes" : ""}${unsafePaths ? " · unsafe-paths" : ""}${args.flags.plan ? " · plan-mode" : ""}`));
   console.log(c.gray(`task: `) + prompt);
   console.log(divider());
 
+  // Plan mode: generate plan first, get approval, then execute with the plan
+  // injected into the conversation as context.
+  let priorMessages = undefined;
+  if (args.flags.plan) {
+    let currentTask = prompt;
+    // Up to 3 plan/refine cycles before bailing
+    for (let attempt = 0; attempt < 3; attempt++) {
+      const planResult = await generateAndApprovePlan({ initialPrompt: currentTask, cwd });
+      if (planResult.cancelled) {
+        console.log(c.gray("\nCancelled. No tools were run."));
+        return;
+      }
+      if (planResult.approved) {
+        // Inject the plan as prior context so the agent remembers it as it executes
+        priorMessages = [
+          { role: "user", content: currentTask },
+          { role: "assistant", content: `Plan:\n${planResult.planText}` },
+        ];
+        break;
+      }
+      if (planResult.refinement) {
+        console.log(c.gray(`Refining: ${planResult.refinement}`));
+        currentTask = `${currentTask}\n\nRefinement: ${planResult.refinement}`;
+      }
+    }
+    if (!priorMessages) {
+      console.log(c.yellow("\nGave up after 3 refinement cycles. Try a clearer task."));
+      return;
+    }
+    console.log("");
+  }
+
   const result = await runAgent({
     initialPrompt: prompt,
+    priorMessages,
     cwd,
     autoYes,
     unsafePaths,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "aether-code",
-  "version": "0.4.0",
+  "version": "0.5.0",
   "description": "Uncensored AI coding agent for your terminal. Type `aether` to launch the interactive REPL — like Claude Code, with no refusal layer.",
   "homepage": "https://trynoguard.com",
   "repository": {
@@ -22,7 +22,7 @@
     "node": ">=18"
   },
   "scripts": {
-    "lint": "node --check bin/aether-code.js src/agent.js src/api.js src/config.js src/render.js src/tools.js src/diff.js src/repl.js"
+    "lint": "node --check bin/aether-code.js src/agent.js src/api.js src/config.js src/render.js src/tools.js src/diff.js src/repl.js src/plan.js"
   },
   "keywords": [
     "aether",

package/src/api.js

@@ -66,6 +66,7 @@ export async function agentTurnStream({
   tools,
   maxTokens,
   temperature,
+  toolChoice,
   onDelta,
   onToolCallDelta,
   onFinish,
@@ -94,6 +95,7 @@ export async function agentTurnStream({
         tools,
         max_tokens: maxTokens,
         temperature,
+        ...(toolChoice ? { tool_choice: toolChoice } : {}),
       }),
     });
   } catch (e) {

package/src/plan.js

@@ -0,0 +1,133 @@
+// Plan mode — get a structured multi-step plan from the model BEFORE any
+// tools fire, show it to the user, let them approve or refine.
+//
+// Flow:
+//   1. Send a preflight request to /api/v1/agent with tool_choice: "none"
+//      and a planning system prompt. Model returns a numbered plan as text.
+//   2. Render the plan with borders. Prompt user: approve / refine / cancel.
+//   3. If approved, the plan text is appended as a tool-style "plan" entry
+//      in the message history (so the model remembers it as it executes).
+//   4. Normal agent loop continues with tools enabled.
+
+import readline from "node:readline";
+import { agentTurnStream, AetherError } from "./api.js";
+import { TOOL_DEFINITIONS } from "./tools.js";
+import { c, errorLine } from "./render.js";
+
+const PLAN_SYSTEM = `You are in PLAN MODE. The user wants to see what you'll do BEFORE you do it.
+
+Output a NUMBERED PLAN of the concrete steps you'll take. Each step should be:
+- One short line
+- Reference a specific tool you'll call (read_file, write_file, edit_file, run_shell, list_dir, search_files)
+- Mention the file path or command involved
+
+DO NOT call any tools. DO NOT include code blocks. Just the numbered plan.
+
+End with one line: "Approve this plan? (y / n / refine)"
+
+Example output:
+
+1. read_file: src/auth.ts  — inspect the existing handler
+2. edit_file: src/auth.ts  — add input validation before the password check
+3. write_file: src/auth.test.ts  — add a test for invalid input
+4. run_shell: npm test  — verify the test passes
+
+Approve this plan? (y / n / refine)`;
+
+function ask(question) {
+  return new Promise((resolve) => {
+    const rl = readline.createInterface({ input: process.stdin, output: process.stdout });
+    rl.question(question, (answer) => {
+      rl.close();
+      resolve(answer.trim());
+    });
+  });
+}
+
+function renderPlanBox(plan) {
+  const w = Math.min(process.stdout.columns || 80, 90);
+  const top = c.gray("╭" + "─".repeat(w - 2) + "╮");
+  const bot = c.gray("╰" + "─".repeat(w - 2) + "╯");
+  const blank = c.gray("│") + " ".repeat(w - 2) + c.gray("│");
+  const lines = [top, blank, c.gray("│ ") + c.bold(c.cyan("📋 Plan")) + " ".repeat(w - 11) + c.gray("│"), blank];
+  for (const line of plan.split("\n")) {
+    // Wrap long lines
+    const wrapWidth = w - 4;
+    let rest = line;
+    while (rest.length > wrapWidth) {
+      const slice = rest.slice(0, wrapWidth);
+      lines.push(c.gray("│ ") + slice + " ".repeat(w - 3 - slice.length) + c.gray("│"));
+      rest = rest.slice(wrapWidth);
+    }
+    lines.push(c.gray("│ ") + rest + " ".repeat(Math.max(0, w - 3 - rest.length)) + c.gray("│"));
+  }
+  lines.push(blank, bot);
+  return lines.join("\n");
+}
+
+/**
+ * Generate a plan, show it, get approval. Returns:
+ *   { approved: true, planText: "..." }       — user said yes
+ *   { approved: false, cancelled: true }       — user said no
+ *   { approved: false, refinement: "..." }     — user wants to refine
+ */
+export async function generateAndApprovePlan({ initialPrompt, cwd, priorMessages }) {
+  process.stdout.write(c.dim("Generating plan...\n"));
+
+  const messages = priorMessages
+    ? [...priorMessages, { role: "user", content: initialPrompt }]
+    : [
+        { role: "system", content: PLAN_SYSTEM },
+        { role: "user", content: initialPrompt },
+      ];
+
+  let planText = "";
+  let creditsCharged = 0;
+  let inputTokens = 0;
+  let outputTokens = 0;
+
+  try {
+    const res = await agentTurnStream({
+      messages,
+      tools: TOOL_DEFINITIONS,
+      // Override server heuristic — we explicitly want NO tools here.
+      toolChoice: "none",
+      maxTokens: 1200,
+    });
+    planText = (res.message.content || "").trim();
+    creditsCharged = res.creditsCharged ?? 0;
+    inputTokens = res.usage?.prompt_tokens ?? 0;
+    outputTokens = res.usage?.completion_tokens ?? 0;
+  } catch (err) {
+    if (err instanceof AetherError) {
+      console.log(errorLine(err.message));
+      return { approved: false, cancelled: true };
+    }
+    throw err;
+  }
+
+  if (!planText) {
+    console.log(errorLine("Got empty plan from model. Try again or run without --plan."));
+    return { approved: false, cancelled: true };
+  }
+
+  // Render the plan in a box
+  console.log("");
+  console.log(renderPlanBox(planText));
+  console.log(c.dim(`  ${creditsCharged} cr · ${inputTokens}→${outputTokens} tokens (planning)\n`));
+
+  // Prompt for approval
+  const answer = await ask(
+    c.yellow("  ? ") + c.bold("Run this plan?") + c.gray(" [y / n / refine]: "),
+  );
+  const lower = answer.toLowerCase();
+
+  if (lower === "y" || lower === "yes") {
+    return { approved: true, planText, creditsCharged, inputTokens, outputTokens };
+  }
+  if (lower === "n" || lower === "no" || lower === "" || lower === "exit") {
+    return { approved: false, cancelled: true };
+  }
+  // Anything else is treated as refinement text
+  return { approved: false, refinement: answer || lower };
+}

package/src/repl.js

@@ -15,7 +15,7 @@ import { fetchBalance, AetherError } from "./api.js";
 import { runSetup } from "./setup.js";
 import { c, errorLine, banner, statusLine } from "./render.js";
 
-const VERSION = "0.4.0";
+const VERSION = "0.5.0";
 const MODEL_NAME = "Aether Core";
 
 const SHORTCUTS = `