aether-code 0.6.1 → 0.8.0

package/src/api.js

@@ -1,236 +1,234 @@
-// API client.
-
-import { getConfig } from "./config.js";
-
-/**
- * Free balance + plan check via /api/v1/me. Doesn't charge credits.
- */
-export async function fetchBalance() {
-  const { apiKey, baseUrl } = getConfig();
-  if (!apiKey) {
-    throw new AetherError(
-      "No API key. Set AETHER_API_KEY or run `aether config set <key>`.",
-      "NO_API_KEY",
-      0,
-    );
-  }
-  let res;
-  try {
-    res = await fetch(`${baseUrl}/api/v1/me`, {
-      method: "GET",
-      headers: {
-        Accept: "application/json",
-        Authorization: `Bearer ${apiKey}`,
-        "User-Agent": "aether-code/0.2.0",
-      },
-    });
-  } catch (e) {
-    throw new AetherError(`Network error: ${e.message}`, "NETWORK", 0);
-  }
-  let data = null;
-  try { data = await res.json(); } catch { /* non-JSON */ }
-  if (!res.ok) {
-    const code = data?.code || `HTTP_${res.status}`;
-    const msg = data?.error || `${res.status} ${res.statusText}`;
-    throw new AetherError(msg, code, res.status, data);
-  }
-  return data; // { plan, role, planCredits, topupCredits, balance, isSuspended, rate }
-}
-
-export class AetherError extends Error {
-  constructor(message, code, status, data) {
-    super(message);
-    this.name = "AetherError";
-    this.code = code;
-    this.status = status;
-    this.data = data;
-  }
-}
-
-/**
- * Streaming agent turn. Calls /api/v1/agent/stream and invokes the supplied
- * callbacks as events arrive. Returns the assembled assistant message + final
- * usage/credit info once the stream ends.
- *
- * Event handlers (all optional):
- *   onDelta(text)            — text fragment from the model
- *   onToolCallDelta(part)    — partial tool call: { index, id?, name?, args_delta? }
- *   onFinish(reason)         — per-choice finish reason ("stop" | "tool_calls" | "length")
- *
- * Returns:
- *   { message: { role:"assistant", content, tool_calls }, finish_reason,
- *     creditsCharged, balanceAfter, usage }
- */
-export async function agentTurnStream({
-  messages,
-  tools,
-  maxTokens,
-  temperature,
-  toolChoice,
-  onDelta,
-  onToolCallDelta,
-  onFinish,
-}) {
-  const { apiKey, baseUrl } = getConfig();
-  if (!apiKey) {
-    throw new AetherError(
-      "No API key. Set AETHER_API_KEY or run `aether config set <key>`.",
-      "NO_API_KEY",
-      0,
-    );
-  }
-
-  let res;
-  try {
-    res = await fetch(`${baseUrl}/api/v1/agent/stream`, {
-      method: "POST",
-      headers: {
-        "Content-Type": "application/json",
-        Accept: "text/event-stream",
-        Authorization: `Bearer ${apiKey}`,
-        "User-Agent": "aether-code/0.1.0",
-      },
-      body: JSON.stringify({
-        messages,
-        tools,
-        max_tokens: maxTokens,
-        temperature,
-        ...(toolChoice ? { tool_choice: toolChoice } : {}),
-      }),
-    });
-  } catch (e) {
-    throw new AetherError(`Network error: ${e.message}`, "NETWORK", 0);
-  }
-
-  // Non-streaming JSON error response (auth, credit, validation failures)
-  if (!res.ok) {
-    let data = null;
-    try { data = await res.json(); } catch { /* non-JSON */ }
-    const code = data?.code || `HTTP_${res.status}`;
-    const msg = data?.error || `${res.status} ${res.statusText}`;
-    throw new AetherError(msg, code, res.status, data);
-  }
-  if (!res.body) throw new AetherError("Response has no body", "NO_BODY", res.status);
-
-  // Accumulate streaming state
-  let textBuf = "";
-  const toolBuf = new Map(); // index -> { id, name, args }
-  let finishReason = "stop";
-  let creditsCharged = 0;
-  let balanceAfter = null;
-  let usage = { prompt_tokens: 0, completion_tokens: 0 };
-  let streamError = null;
-
-  const reader = res.body.getReader();
-  const decoder = new TextDecoder();
-  let buffer = "";
-  while (true) {
-    const { value, done } = await reader.read();
-    if (done) break;
-    buffer += decoder.decode(value, { stream: true });
-    const lines = buffer.split("\n");
-    buffer = lines.pop() ?? "";
-    for (const line of lines) {
-      if (!line.startsWith("data:")) continue;
-      const json = line.slice(5).trim();
-      if (!json) continue;
-      let evt;
-      try { evt = JSON.parse(json); } catch { continue; }
-      if (evt.kind === "delta") {
-        textBuf += evt.content;
-        if (onDelta) onDelta(evt.content);
-      } else if (evt.kind === "tool_call_delta") {
-        const slot = toolBuf.get(evt.index) || { id: undefined, name: undefined, args: "" };
-        if (evt.id) slot.id = evt.id;
-        if (evt.name) slot.name = evt.name;
-        if (evt.args_delta) slot.args += evt.args_delta;
-        toolBuf.set(evt.index, slot);
-        if (onToolCallDelta) onToolCallDelta(evt);
-      } else if (evt.kind === "finish") {
-        finishReason = evt.reason;
-        if (onFinish) onFinish(evt.reason);
-      } else if (evt.kind === "done") {
-        creditsCharged = evt.creditsCharged ?? 0;
-        balanceAfter = evt.balanceAfter ?? null;
-        usage = evt.usage ?? usage;
-      } else if (evt.kind === "error") {
-        streamError = evt.error;
-      }
-    }
-  }
-
-  if (streamError) {
-    throw new AetherError(streamError, "STREAM_ERROR", 0);
-  }
-
-  // Assemble final tool_calls in stable index order
-  const tool_calls = [...toolBuf.entries()]
-    .sort((a, b) => a[0] - b[0])
-    .map(([, slot]) => ({
-      id: slot.id || `call_${Math.random().toString(36).slice(2, 10)}`,
-      type: "function",
-      function: { name: slot.name || "", arguments: slot.args || "" },
-    }));
-
-  return {
-    message: {
-      role: "assistant",
-      content: textBuf || null,
-      tool_calls: tool_calls.length > 0 ? tool_calls : undefined,
-    },
-    finish_reason: finishReason,
-    creditsCharged,
-    balanceAfter,
-    usage,
-  };
-}
-
-/**
- * Non-streaming variant — kept as a fallback for environments where SSE is
- * problematic (corporate proxies, weird client setups). The CLI defaults to
- * streaming.
- */
-export async function agentTurn({ messages, tools, maxTokens, temperature }) {
-  const { apiKey, baseUrl } = getConfig();
-  if (!apiKey) {
-    throw new AetherError(
-      "No API key. Set AETHER_API_KEY env var or run `aether-cli config set <key>`.",
-      "NO_API_KEY",
-      0,
-    );
-  }
-
-  let res;
-  try {
-    res = await fetch(`${baseUrl}/api/v1/agent`, {
-      method: "POST",
-      headers: {
-        "Content-Type": "application/json",
-        Authorization: `Bearer ${apiKey}`,
-        "User-Agent": "aether-code/0.1.0",
-      },
-      body: JSON.stringify({
-        messages,
-        tools,
-        max_tokens: maxTokens,
-        temperature,
-      }),
-    });
-  } catch (e) {
-    throw new AetherError(`Network error: ${e.message}`, "NETWORK", 0);
-  }
-
-  let data = null;
-  try {
-    data = await res.json();
-  } catch {
-    /* non-JSON */
-  }
-
-  if (!res.ok) {
-    const code = (data && data.code) || `HTTP_${res.status}`;
-    const msg = (data && data.error) || `${res.status} ${res.statusText}`;
-    throw new AetherError(msg, code, res.status, data);
-  }
-  return data;
-}
+// API client.
+
+import { getConfig } from "./config.js";
+
+/**
+ * Free balance + plan check via /api/v1/me. Doesn't charge credits.
+ */
+export async function fetchBalance() {
+  const { apiKey, baseUrl } = getConfig();
+  if (!apiKey) {
+    throw new AetherError(
+      "No API key. Set AETHER_API_KEY or run `aether config set <key>`.",
+      "NO_API_KEY",
+      0,
+    );
+  }
+  let res;
+  try {
+    res = await fetch(`${baseUrl}/api/v1/me`, {
+      method: "GET",
+      headers: {
+        Accept: "application/json",
+        Authorization: `Bearer ${apiKey}`,
+        "User-Agent": "aether-code/0.2.0",
+      },
+    });
+  } catch (e) {
+    throw new AetherError(`Network error: ${e.message}`, "NETWORK", 0);
+  }
+  let data = null;
+  try { data = await res.json(); } catch { /* non-JSON */ }
+  if (!res.ok) {
+    const code = data?.code || `HTTP_${res.status}`;
+    const msg = data?.error || `${res.status} ${res.statusText}`;
+    throw new AetherError(msg, code, res.status, data);
+  }
+  return data; // { plan, role, planCredits, topupCredits, balance, isSuspended, rate }
+}
+
+export class AetherError extends Error {
+  constructor(message, code, status, data) {
+    super(message);
+    this.name = "AetherError";
+    this.code = code;
+    this.status = status;
+    this.data = data;
+  }
+}
+
+/**
+ * Streaming agent turn. Calls /api/v1/agent/stream and invokes the supplied
+ * callbacks as events arrive. Returns the assembled assistant message + final
+ * usage/credit info once the stream ends.
+ *
+ * Event handlers (all optional):
+ *   onDelta(text)            — text fragment from the model
+ *   onToolCallDelta(part)    — partial tool call: { index, id?, name?, args_delta? }
+ *   onFinish(reason)         — per-choice finish reason ("stop" | "tool_calls" | "length")
+ *
+ * Returns:
+ *   { message: { role:"assistant", content, tool_calls }, finish_reason,
+ *     creditsCharged, balanceAfter, usage }
+ */
+export async function agentTurnStream({
+  messages,
+  tools,
+  maxTokens,
+  temperature,
+  onDelta,
+  onToolCallDelta,
+  onFinish,
+}) {
+  const { apiKey, baseUrl } = getConfig();
+  if (!apiKey) {
+    throw new AetherError(
+      "No API key. Set AETHER_API_KEY or run `aether config set <key>`.",
+      "NO_API_KEY",
+      0,
+    );
+  }
+
+  let res;
+  try {
+    res = await fetch(`${baseUrl}/api/v1/agent/stream`, {
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json",
+        Accept: "text/event-stream",
+        Authorization: `Bearer ${apiKey}`,
+        "User-Agent": "aether-code/0.1.0",
+      },
+      body: JSON.stringify({
+        messages,
+        tools,
+        max_tokens: maxTokens,
+        temperature,
+      }),
+    });
+  } catch (e) {
+    throw new AetherError(`Network error: ${e.message}`, "NETWORK", 0);
+  }
+
+  // Non-streaming JSON error response (auth, credit, validation failures)
+  if (!res.ok) {
+    let data = null;
+    try { data = await res.json(); } catch { /* non-JSON */ }
+    const code = data?.code || `HTTP_${res.status}`;
+    const msg = data?.error || `${res.status} ${res.statusText}`;
+    throw new AetherError(msg, code, res.status, data);
+  }
+  if (!res.body) throw new AetherError("Response has no body", "NO_BODY", res.status);
+
+  // Accumulate streaming state
+  let textBuf = "";
+  const toolBuf = new Map(); // index -> { id, name, args }
+  let finishReason = "stop";
+  let creditsCharged = 0;
+  let balanceAfter = null;
+  let usage = { prompt_tokens: 0, completion_tokens: 0 };
+  let streamError = null;
+
+  const reader = res.body.getReader();
+  const decoder = new TextDecoder();
+  let buffer = "";
+  while (true) {
+    const { value, done } = await reader.read();
+    if (done) break;
+    buffer += decoder.decode(value, { stream: true });
+    const lines = buffer.split("\n");
+    buffer = lines.pop() ?? "";
+    for (const line of lines) {
+      if (!line.startsWith("data:")) continue;
+      const json = line.slice(5).trim();
+      if (!json) continue;
+      let evt;
+      try { evt = JSON.parse(json); } catch { continue; }
+      if (evt.kind === "delta") {
+        textBuf += evt.content;
+        if (onDelta) onDelta(evt.content);
+      } else if (evt.kind === "tool_call_delta") {
+        const slot = toolBuf.get(evt.index) || { id: undefined, name: undefined, args: "" };
+        if (evt.id) slot.id = evt.id;
+        if (evt.name) slot.name = evt.name;
+        if (evt.args_delta) slot.args += evt.args_delta;
+        toolBuf.set(evt.index, slot);
+        if (onToolCallDelta) onToolCallDelta(evt);
+      } else if (evt.kind === "finish") {
+        finishReason = evt.reason;
+        if (onFinish) onFinish(evt.reason);
+      } else if (evt.kind === "done") {
+        creditsCharged = evt.creditsCharged ?? 0;
+        balanceAfter = evt.balanceAfter ?? null;
+        usage = evt.usage ?? usage;
+      } else if (evt.kind === "error") {
+        streamError = evt.error;
+      }
+    }
+  }
+
+  if (streamError) {
+    throw new AetherError(streamError, "STREAM_ERROR", 0);
+  }
+
+  // Assemble final tool_calls in stable index order
+  const tool_calls = [...toolBuf.entries()]
+    .sort((a, b) => a[0] - b[0])
+    .map(([, slot]) => ({
+      id: slot.id || `call_${Math.random().toString(36).slice(2, 10)}`,
+      type: "function",
+      function: { name: slot.name || "", arguments: slot.args || "" },
+    }));
+
+  return {
+    message: {
+      role: "assistant",
+      content: textBuf || null,
+      tool_calls: tool_calls.length > 0 ? tool_calls : undefined,
+    },
+    finish_reason: finishReason,
+    creditsCharged,
+    balanceAfter,
+    usage,
+  };
+}
+
+/**
+ * Non-streaming variant — kept as a fallback for environments where SSE is
+ * problematic (corporate proxies, weird client setups). The CLI defaults to
+ * streaming.
+ */
+export async function agentTurn({ messages, tools, maxTokens, temperature }) {
+  const { apiKey, baseUrl } = getConfig();
+  if (!apiKey) {
+    throw new AetherError(
+      "No API key. Set AETHER_API_KEY env var or run `aether-cli config set <key>`.",
+      "NO_API_KEY",
+      0,
+    );
+  }
+
+  let res;
+  try {
+    res = await fetch(`${baseUrl}/api/v1/agent`, {
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json",
+        Authorization: `Bearer ${apiKey}`,
+        "User-Agent": "aether-code/0.1.0",
+      },
+      body: JSON.stringify({
+        messages,
+        tools,
+        max_tokens: maxTokens,
+        temperature,
+      }),
+    });
+  } catch (e) {
+    throw new AetherError(`Network error: ${e.message}`, "NETWORK", 0);
+  }
+
+  let data = null;
+  try {
+    data = await res.json();
+  } catch {
+    /* non-JSON */
+  }
+
+  if (!res.ok) {
+    const code = (data && data.code) || `HTTP_${res.status}`;
+    const msg = (data && data.error) || `${res.status} ${res.statusText}`;
+    throw new AetherError(msg, code, res.status, data);
+  }
+  return data;
+}

package/src/config.js

@@ -1,38 +1,38 @@
-// Config — reads ~/.aetherrc (shared format with aether-cli) and env vars.
-// Same precedence: env wins over file.
-
-import fs from "node:fs";
-import path from "node:path";
-import os from "node:os";
-
-const FILE = path.join(os.homedir(), ".aetherrc");
-const DEFAULT_BASE = "https://trynoguard.com";
-
-export function readConfigFile() {
-  try {
-    const raw = fs.readFileSync(FILE, "utf8");
-    return JSON.parse(raw);
-  } catch {
-    return {};
-  }
-}
-
-export function writeConfigFile(patch) {
-  const current = readConfigFile();
-  const next = { ...current, ...patch };
-  // 0600 — readable only by the user. Mirrors what ssh, gnupg, etc. enforce
-  // on credential files. Treat the API key like an SSH key.
-  fs.writeFileSync(FILE, JSON.stringify(next, null, 2), { mode: 0o600 });
-  try { fs.chmodSync(FILE, 0o600); } catch { /* windows ignores */ }
-}
-
-export function getConfig() {
-  const file = readConfigFile();
-  return {
-    apiKey: process.env.AETHER_API_KEY || file.apiKey || "",
-    baseUrl: (process.env.AETHER_BASE_URL || file.baseUrl || DEFAULT_BASE).replace(/\/+$/, ""),
-    configPath: FILE,
-  };
-}
-
-export const CONFIG_PATH = FILE;
+// Config — reads ~/.aetherrc (shared format with aether-cli) and env vars.
+// Same precedence: env wins over file.
+
+import fs from "node:fs";
+import path from "node:path";
+import os from "node:os";
+
+const FILE = path.join(os.homedir(), ".aetherrc");
+const DEFAULT_BASE = "https://trynoguard.com";
+
+export function readConfigFile() {
+  try {
+    const raw = fs.readFileSync(FILE, "utf8");
+    return JSON.parse(raw);
+  } catch {
+    return {};
+  }
+}
+
+export function writeConfigFile(patch) {
+  const current = readConfigFile();
+  const next = { ...current, ...patch };
+  // 0600 — readable only by the user. Mirrors what ssh, gnupg, etc. enforce
+  // on credential files. Treat the API key like an SSH key.
+  fs.writeFileSync(FILE, JSON.stringify(next, null, 2), { mode: 0o600 });
+  try { fs.chmodSync(FILE, 0o600); } catch { /* windows ignores */ }
+}
+
+export function getConfig() {
+  const file = readConfigFile();
+  return {
+    apiKey: process.env.AETHER_API_KEY || file.apiKey || "",
+    baseUrl: (process.env.AETHER_BASE_URL || file.baseUrl || DEFAULT_BASE).replace(/\/+$/, ""),
+    configPath: FILE,
+  };
+}
+
+export const CONFIG_PATH = FILE;

package/src/diff.js

@@ -1,48 +1,48 @@
-// Tiny line-by-line diff for write_file confirmation prompts.
-// Not a "real" diff — just a side-by-side highlight of what's changing.
-// Good enough for confirmation prompts, deliberately not pretending to be `git diff`.
-
-import { c } from "./render.js";
-
-export function unifiedDiff(oldText, newText, filename) {
-  const oldLines = (oldText || "").split("\n");
-  const newLines = (newText || "").split("\n");
-  const max = Math.max(oldLines.length, newLines.length);
-
-  // Find common prefix and suffix to keep the diff focused
-  let prefix = 0;
-  while (prefix < max && oldLines[prefix] === newLines[prefix]) prefix++;
-  let suffix = 0;
-  while (
-    suffix < max - prefix &&
-    oldLines[oldLines.length - 1 - suffix] === newLines[newLines.length - 1 - suffix]
-  ) {
-    suffix++;
-  }
-
-  const changedOld = oldLines.slice(prefix, oldLines.length - suffix);
-  const changedNew = newLines.slice(prefix, newLines.length - suffix);
-
-  const lines = [];
-  lines.push(c.bold(c.cyan(`@@ ${filename} @@`)));
-  if (prefix > 0) lines.push(c.gray(`  …${prefix} unchanged line${prefix === 1 ? "" : "s"} above…`));
-  for (const l of changedOld) lines.push(c.red(`- ${l}`));
-  for (const l of changedNew) lines.push(c.green(`+ ${l}`));
-  if (suffix > 0) lines.push(c.gray(`  …${suffix} unchanged line${suffix === 1 ? "" : "s"} below…`));
-
-  // Cap output so massive writes don't flood the terminal
-  if (lines.length > 60) {
-    return [...lines.slice(0, 30), c.gray(`  …${lines.length - 60} more lines hidden…`), ...lines.slice(-30)].join(
-      "\n",
-    );
-  }
-  return lines.join("\n");
-}
-
-export function summarizeWrite(oldText, newText, filename) {
-  const oldLines = (oldText || "").split("\n").length;
-  const newLines = (newText || "").split("\n").length;
-  const isCreate = oldText === null || oldText === undefined;
-  const verb = isCreate ? "create" : "rewrite";
-  return c.dim(`${verb} ${filename} (${oldLines} → ${newLines} lines, ${(newText || "").length} bytes)`);
-}
+// Tiny line-by-line diff for write_file confirmation prompts.
+// Not a "real" diff — just a side-by-side highlight of what's changing.
+// Good enough for confirmation prompts, deliberately not pretending to be `git diff`.
+
+import { c } from "./render.js";
+
+export function unifiedDiff(oldText, newText, filename) {
+  const oldLines = (oldText || "").split("\n");
+  const newLines = (newText || "").split("\n");
+  const max = Math.max(oldLines.length, newLines.length);
+
+  // Find common prefix and suffix to keep the diff focused
+  let prefix = 0;
+  while (prefix < max && oldLines[prefix] === newLines[prefix]) prefix++;
+  let suffix = 0;
+  while (
+    suffix < max - prefix &&
+    oldLines[oldLines.length - 1 - suffix] === newLines[newLines.length - 1 - suffix]
+  ) {
+    suffix++;
+  }
+
+  const changedOld = oldLines.slice(prefix, oldLines.length - suffix);
+  const changedNew = newLines.slice(prefix, newLines.length - suffix);
+
+  const lines = [];
+  lines.push(c.bold(c.cyan(`@@ ${filename} @@`)));
+  if (prefix > 0) lines.push(c.gray(`  …${prefix} unchanged line${prefix === 1 ? "" : "s"} above…`));
+  for (const l of changedOld) lines.push(c.red(`- ${l}`));
+  for (const l of changedNew) lines.push(c.green(`+ ${l}`));
+  if (suffix > 0) lines.push(c.gray(`  …${suffix} unchanged line${suffix === 1 ? "" : "s"} below…`));
+
+  // Cap output so massive writes don't flood the terminal
+  if (lines.length > 60) {
+    return [...lines.slice(0, 30), c.gray(`  …${lines.length - 60} more lines hidden…`), ...lines.slice(-30)].join(
+      "\n",
+    );
+  }
+  return lines.join("\n");
+}
+
+export function summarizeWrite(oldText, newText, filename) {
+  const oldLines = (oldText || "").split("\n").length;
+  const newLines = (newText || "").split("\n").length;
+  const isCreate = oldText === null || oldText === undefined;
+  const verb = isCreate ? "create" : "rewrite";
+  return c.dim(`${verb} ${filename} (${oldLines} → ${newLines} lines, ${(newText || "").length} bytes)`);
+}