@wanghuimvp/axon 0.3.0 → 0.4.1

package/README.md

@@ -18,6 +18,10 @@ Run `axon` with no arguments to open the interactive chat:
 axon
 ```
 
+On first run without an API key, `axon` opens a short setup screen — paste a key for the active provider and it is saved to `~/.axon/config.json` (with restricted file permissions) so later runs start straight in the chat. You can also set the provider's environment variable instead (e.g. `ANTHROPIC_API_KEY`), which always takes precedence.
+
+Axon reads your project on startup — `package.json` and the first of `AGENTS.md` / `CLAUDE.md` / `README.md` — and gives the model that context.
+
 Type a request and press Enter. Axon streams its reasoning and tool calls. When it wants to write a file or run a command, you get an inline prompt:
 
 ```

package/dist/cli.js

@@ -23,6 +23,20 @@ var DEFAULTS = {
     gemini: { apiKey: "env:GEMINI_API_KEY" }
   }
 };
+var ENV_VARS = {
+  anthropic: "ANTHROPIC_API_KEY",
+  openai: "OPENAI_API_KEY",
+  gemini: "GEMINI_API_KEY"
+};
+function detectProvider(fileCfg) {
+  for (const name of ["anthropic", "openai", "gemini"]) {
+    const literal = fileCfg.providers?.[name]?.apiKey;
+    const hasLiteral = typeof literal === "string" && !literal.startsWith("env:") && literal.trim().length > 0;
+    const hasEnv = (process.env[ENV_VARS[name]] ?? "").trim().length > 0;
+    if (hasLiteral || hasEnv) return name;
+  }
+  return "anthropic";
+}
 function resolveModel(cfg) {
   const model = cfg.model ?? cfg.providers[cfg.provider]?.model ?? DEFAULT_MODELS[cfg.provider];
   if (!model) {
@@ -46,7 +60,7 @@ function loadConfig() {
   } catch {
   }
   const merged = {
-    provider: fileCfg.provider ?? DEFAULTS.provider,
+    provider: fileCfg.provider ?? detectProvider(fileCfg),
     model: fileCfg.model,
     providers: { ...DEFAULTS.providers, ...fileCfg.providers ?? {} }
   };
@@ -54,7 +68,7 @@ function loadConfig() {
 }
 
 // src/config/configFile.ts
-import { readFileSync as readFileSync2, writeFileSync, mkdirSync } from "node:fs";
+import { readFileSync as readFileSync2, writeFileSync, mkdirSync, chmodSync } from "node:fs";
 import { homedir as homedir2 } from "node:os";
 import { join as join2, dirname } from "node:path";
 function configPath() {
@@ -86,6 +100,18 @@ function setConfigValue(key, value) {
   mkdirSync(dirname(path), { recursive: true });
   writeFileSync(path, JSON.stringify(cfg, null, 2) + "\n");
 }
+function setApiKey(provider, key) {
+  const cfg = readConfigFile();
+  const providers = cfg.providers ??= {};
+  (providers[provider] ??= {}).apiKey = key;
+  const path = configPath();
+  mkdirSync(dirname(path), { recursive: true });
+  writeFileSync(path, JSON.stringify(cfg, null, 2) + "\n", { mode: 384 });
+  try {
+    chmodSync(path, 384);
+  } catch {
+  }
+}
 
 // src/providers/registry.ts
 import Anthropic from "@anthropic-ai/sdk";
@@ -774,7 +800,39 @@ function truncate(s, max = 500) {
 }
 
 // src/ui/runTui.tsx
-import { render } from "ink";
+import { useState as useState3, useRef } from "react";
+import { Box as Box6, Text as Text6, render } from "ink";
+
+// src/core/projectContext.ts
+import { readFileSync as readFileSync3 } from "node:fs";
+import { join as join4 } from "node:path";
+function truncate2(s, max) {
+  return s.length > max ? `${s.slice(0, max)}\u2026` : s;
+}
+function loadProjectContext(cwd) {
+  const parts = [];
+  try {
+    const pkg = JSON.parse(readFileSync3(join4(cwd, "package.json"), "utf8"));
+    const bits = [];
+    if (pkg.name) bits.push(`name: ${pkg.name}`);
+    if (pkg.description) bits.push(`description: ${pkg.description}`);
+    if (pkg.scripts && typeof pkg.scripts === "object") bits.push(`scripts: ${Object.keys(pkg.scripts).join(", ")}`);
+    if (bits.length) parts.push(`package.json \u2014 ${bits.join("; ")}`);
+  } catch {
+  }
+  for (const name of ["AGENTS.md", "CLAUDE.md", "README.md"]) {
+    try {
+      const text = readFileSync3(join4(cwd, name), "utf8").trim();
+      if (text) {
+        parts.push(`${name}:
+${truncate2(text, 2e3)}`);
+        break;
+      }
+    } catch {
+    }
+  }
+  return parts.join("\n\n");
+}
 
 // src/ui/permissionController.ts
 function createPermissionController() {
@@ -823,6 +881,20 @@ function createPermissionController() {
   return { gate, subscribe, getPending, resolve: resolve3 };
 }
 
+// src/config/credentials.ts
+function hasUsableKey(cfg) {
+  const k = cfg.providers[cfg.provider]?.apiKey;
+  return typeof k === "string" && k.trim().length > 0;
+}
+var INFO = {
+  anthropic: { envVar: "ANTHROPIC_API_KEY", url: "https://console.anthropic.com/settings/keys" },
+  openai: { envVar: "OPENAI_API_KEY", url: "https://platform.openai.com/api-keys" },
+  gemini: { envVar: "GEMINI_API_KEY", url: "https://aistudio.google.com/apikey" }
+};
+function keyProviderInfo(provider) {
+  return INFO[provider] ?? { envVar: `${provider.toUpperCase()}_API_KEY`, url: "" };
+}
+
 // src/ui/app.tsx
 import { useEffect, useState, useSyncExternalStore, useCallback } from "react";
 import { Box as Box4, Text as Text4, useInput as useInput2 } from "ink";
@@ -835,7 +907,7 @@ function toolIcon(status) {
   if (status === "running") return "\u23F3";
   return status === "ok" ? "\u2705" : "\u274C";
 }
-function truncate2(s, max = 300) {
+function truncate3(s, max = 300) {
   return s.length > max ? `${s.slice(0, max)}\u2026` : s;
 }
 function MessageView({ items }) {
@@ -870,9 +942,9 @@ function MessageView({ items }) {
         " ",
         it.name,
         "(",
-        truncate2(JSON.stringify(it.args ?? {}), 60),
+        truncate3(JSON.stringify(it.args ?? {}), 60),
         ")",
-        it.output ? ` \u2014 ${truncate2(it.output)}` : ""
+        it.output ? ` \u2014 ${truncate3(it.output)}` : ""
       ] }, i);
     }
     return null;
@@ -986,40 +1058,159 @@ function App({ engine, controller, provider, model, yolo }) {
   ] });
 }
 
+// src/ui/Setup.tsx
+import { useState as useState2 } from "react";
+import { Box as Box5, Text as Text5, useInput as useInput3 } from "ink";
+import TextInput2 from "ink-text-input";
+import { jsx as jsx5, jsxs as jsxs5 } from "react/jsx-runtime";
+var SETUP_PROVIDERS = ["anthropic", "openai", "gemini"];
+function trimmedKey(v) {
+  const t = v.trim();
+  return t ? t : null;
+}
+function providerForDigit(input) {
+  const idx = Number(input) - 1;
+  return Number.isInteger(idx) && idx >= 0 && idx < SETUP_PROVIDERS.length ? SETUP_PROVIDERS[idx] : null;
+}
+function Setup({
+  initialProvider,
+  onSubmit
+}) {
+  const start = SETUP_PROVIDERS.includes(initialProvider) ? initialProvider : "anthropic";
+  const [provider, setProvider] = useState2(start);
+  const [value, setValue] = useState2("");
+  useInput3((input) => {
+    const picked = providerForDigit(input);
+    if (picked) setProvider(picked);
+  });
+  const info = keyProviderInfo(provider);
+  const submit = (v) => {
+    const k = trimmedKey(v);
+    if (k) onSubmit(provider, k);
+  };
+  return /* @__PURE__ */ jsxs5(Box5, { flexDirection: "column", children: [
+    /* @__PURE__ */ jsx5(Text5, { color: "yellow", children: "Welcome to Axon. Pick a provider, then paste its API key." }),
+    SETUP_PROVIDERS.map((p, i) => /* @__PURE__ */ jsxs5(Text5, { color: p === provider ? "cyan" : "gray", children: [
+      p === provider ? "\u276F" : " ",
+      " [",
+      i + 1,
+      "] ",
+      p
+    ] }, p)),
+    /* @__PURE__ */ jsxs5(Text5, { children: [
+      "Active: ",
+      /* @__PURE__ */ jsx5(Text5, { bold: true, children: provider }),
+      " \u2014 set ",
+      /* @__PURE__ */ jsx5(Text5, { bold: true, children: info.envVar }),
+      ", or paste a key below to save it to ~/.axon/config.json."
+    ] }),
+    info.url ? /* @__PURE__ */ jsxs5(Text5, { color: "gray", children: [
+      "Get a ",
+      provider,
+      " key at: ",
+      info.url
+    ] }) : null,
+    /* @__PURE__ */ jsxs5(Box5, { children: [
+      /* @__PURE__ */ jsx5(Text5, { color: "cyan", children: "key \u203A " }),
+      /* @__PURE__ */ jsx5(TextInput2, { value, onChange: setValue, onSubmit: submit, mask: "*" })
+    ] })
+  ] });
+}
+
 // src/ui/runTui.tsx
-import { jsx as jsx5 } from "react/jsx-runtime";
+import { jsx as jsx6, jsxs as jsxs6 } from "react/jsx-runtime";
 var TUI_SYSTEM = `You are Axon, an interactive agentic coding assistant. Use the tools to inspect and modify the project: read_file, list_dir, glob, grep (read-only) and write_file, edit_file, shell (these change the workspace; the user is prompted to approve each). Prefer edit_file for surgical changes. Explain briefly what you are doing.`;
-function runTui(opts) {
-  const cfg = loadConfig();
-  if (opts.provider) cfg.provider = opts.provider;
-  if (opts.model) cfg.model = opts.model;
-  const provider = createProvider(cfg);
-  const tools = buildAllTools();
-  const controller = createPermissionController();
-  const gate = opts.yolo ? allowAllGate : controller.gate;
-  const engine = new Engine({ provider, tools, system: TUI_SYSTEM, cwd: process.cwd(), gate });
-  render(
-    /* @__PURE__ */ jsx5(
+function Root({ deps }) {
+  const { cfg, controller, yolo, buildEngine, persistKey } = deps;
+  const [ready, setReady] = useState3(hasUsableKey(cfg));
+  const engineRef = useRef(null);
+  if (ready && !engineRef.current) engineRef.current = buildEngine();
+  if (!ready || !engineRef.current) {
+    return /* @__PURE__ */ jsx6(
+      Setup,
+      {
+        initialProvider: cfg.provider,
+        onSubmit: (provider, key) => {
+          persistKey(provider, key);
+          cfg.provider = provider;
+          cfg.providers[provider] = { ...cfg.providers[provider] ?? {}, apiKey: key };
+          setReady(true);
+        }
+      }
+    );
+  }
+  return /* @__PURE__ */ jsxs6(Box6, { flexDirection: "column", children: [
+    /* @__PURE__ */ jsxs6(Text6, { color: "green", children: [
+      "Axon \xB7 ",
+      cfg.provider,
+      "/",
+      resolveModel(cfg),
+      " \xB7 type a request, Ctrl+C to quit"
+    ] }),
+    /* @__PURE__ */ jsx6(
       App,
       {
-        engine,
+        engine: engineRef.current,
         controller,
         provider: cfg.provider,
         model: resolveModel(cfg),
-        yolo: Boolean(opts.yolo)
+        yolo
       }
     )
-  );
+  ] });
+}
+function runTui(opts) {
+  const cfg = loadConfig();
+  if (opts.provider) cfg.provider = opts.provider;
+  if (opts.model) cfg.model = opts.model;
+  const controller = createPermissionController();
+  const deps = {
+    cfg,
+    controller,
+    yolo: Boolean(opts.yolo),
+    persistKey: (provider, key) => {
+      try {
+        setApiKey(provider, key);
+        setConfigValue("provider", provider);
+      } catch {
+      }
+    },
+    buildEngine: () => {
+      const provider = createProvider(cfg);
+      const tools = buildAllTools();
+      const gate = opts.yolo ? allowAllGate : controller.gate;
+      const context = loadProjectContext(process.cwd());
+      const system = TUI_SYSTEM + (context ? `
+
+Project context:
+${context}` : "");
+      return new Engine({ provider, tools, system, cwd: process.cwd(), gate });
+    }
+  };
+  if (!process.stdin.isTTY) {
+    process.stderr.write('axon: the interactive chat needs a terminal. For non-interactive use, run: axon -p "your prompt"\n');
+    process.exit(1);
+  }
+  render(/* @__PURE__ */ jsx6(Root, { deps }));
 }
 
 // src/cli.ts
 var READONLY_SYSTEM = `You are Axon, an agentic coding assistant. Use the read-only tools \u2014 read_file, list_dir, glob, grep \u2014 to inspect the project and answer precisely. When done, stop calling tools.`;
 var YOLO_SYSTEM = `You are Axon, an agentic coding assistant. Use the provided tools to inspect AND modify the project: read_file, list_dir, glob, grep (read-only), and write_file, edit_file, shell (these change the workspace). Prefer edit_file for surgical changes. When done, stop calling tools.`;
+function redactKeys(cfg) {
+  const providers = cfg.providers;
+  if (!providers || typeof providers !== "object") return cfg;
+  const masked = {};
+  for (const [name, p] of Object.entries(providers)) {
+    masked[name] = p && typeof p === "object" && "apiKey" in p && p.apiKey ? { ...p, apiKey: "***redacted***" } : p;
+  }
+  return { ...cfg, providers: masked };
+}
 var program = new Command();
 program.name("axon").version(VERSION).option("-p, --print <prompt>", "run one prompt non-interactively and stream the result").option("--provider <name>", "override the provider for this run (anthropic | openai | gemini)").option("--model <name>", "override the model for this run").option("--yolo", "allow write/edit/shell tools without prompting (non-interactive)");
 program.command("config").argument("<action>", "get | set").argument("[key]", "config key (provider | model | <provider>.<baseUrl|model>)").argument("[value]", "value to set").action((action, key, value) => {
   if (action === "get") {
-    process.stdout.write(JSON.stringify(readConfigFile(), null, 2) + "\n");
+    process.stdout.write(JSON.stringify(redactKeys(readConfigFile()), null, 2) + "\n");
     return;
   }
   if (action === "set") {
@@ -1051,7 +1242,12 @@ async function main(opts) {
   const provider = createProvider(cfg);
   const tools = opts.yolo ? buildAllTools() : buildReadOnlyTools();
   const gate = opts.yolo ? allowAllGate : denyGate;
-  const system = opts.yolo ? YOLO_SYSTEM : READONLY_SYSTEM;
+  const baseSystem = opts.yolo ? YOLO_SYSTEM : READONLY_SYSTEM;
+  const context = loadProjectContext(process.cwd());
+  const system = baseSystem + (context ? `
+
+Project context:
+${context}` : "");
   const engine = new Engine({ provider, tools, system, cwd: process.cwd(), gate });
   printRunner(engine, (s) => process.stdout.write(s));
   process.stderr.write(`[axon: ${cfg.provider} / ${resolveModel(cfg)}${opts.yolo ? " / yolo" : ""}]

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@wanghuimvp/axon",
-  "version": "0.3.0",
+  "version": "0.4.1",
   "description": "Axon — a multi-provider agentic coding CLI (Anthropic, OpenAI + OpenAI-compatible endpoints, Gemini). Runs a multi-step tool loop over your codebase.",
   "type": "module",
   "bin": {