@wanghuimvp/axon 0.4.0 → 0.4.2

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 ?? {} }
   };
@@ -820,6 +834,21 @@ ${truncate2(text, 2e3)}`);
   return parts.join("\n\n");
 }
 
+// src/core/systemPrompt.ts
+function buildSystemPrompt(opts) {
+  const identity = `You are the coding agent inside Axon, a command-line coding tool. Axon is the CLI program \u2014 it is NOT a language model. You are powered by the model "${opts.model}" via the ${opts.provider} provider. If the user asks which model or LLM you are, answer honestly: "${opts.model}" (via ${opts.provider}), running inside the Axon CLI. Never claim that "Axon" is a model.`;
+  const toolLine = opts.tools === "all" ? `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.` : `Use the read-only tools to inspect the project: read_file, list_dir, glob, grep.`;
+  const closing = `Explain briefly what you are doing. When the task is done, stop calling tools.`;
+  let prompt = `${identity}
+
+${toolLine} ${closing}`;
+  if (opts.context) prompt += `
+
+Project context:
+${opts.context}`;
+  return prompt;
+}
+
 // src/ui/permissionController.ts
 function createPermissionController() {
   const sessionAllow = /* @__PURE__ */ new Set();
@@ -1046,36 +1075,54 @@ function App({ engine, controller, provider, model, yolo }) {
 
 // src/ui/Setup.tsx
 import { useState as useState2 } from "react";
-import { Box as Box5, Text as Text5 } from "ink";
+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({
-  provider,
-  info,
+  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(k);
+    if (k) onSubmit(provider, k);
   };
   return /* @__PURE__ */ jsxs5(Box5, { flexDirection: "column", children: [
-    /* @__PURE__ */ jsxs5(Text5, { color: "yellow", children: [
-      'No API key found for "',
-      provider,
-      '".'
-    ] }),
+    /* @__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: [
-      "Set ",
+      "Active: ",
+      /* @__PURE__ */ jsx5(Text5, { bold: true, children: provider }),
+      " \u2014 set ",
       /* @__PURE__ */ jsx5(Text5, { bold: true, children: info.envVar }),
-      " in your environment, or paste a key below to save it to ~/.axon/config.json."
+      ", or paste a key below to save it to ~/.axon/config.json."
     ] }),
     info.url ? /* @__PURE__ */ jsxs5(Text5, { color: "gray", children: [
-      "Get a key at: ",
+      "Get a ",
+      provider,
+      " key at: ",
       info.url
     ] }) : null,
     /* @__PURE__ */ jsxs5(Box5, { children: [
@@ -1087,29 +1134,33 @@ function Setup({
 
 // src/ui/runTui.tsx
 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 Root({ deps }) {
-  const { cfg, controller, yolo, welcome, buildEngine, persistKey } = 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) {
-    const info = keyProviderInfo(cfg.provider);
     return /* @__PURE__ */ jsx6(
       Setup,
       {
-        provider: cfg.provider,
-        info,
-        onSubmit: (key) => {
-          persistKey(cfg.provider, key);
-          cfg.providers[cfg.provider] = { ...cfg.providers[cfg.provider] ?? {}, apiKey: key };
+        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__ */ jsx6(Text6, { color: "green", children: welcome }),
+    /* @__PURE__ */ jsxs6(Text6, { color: "green", children: [
+      "Axon \xB7 ",
+      cfg.provider,
+      "/",
+      resolveModel(cfg),
+      " \xB7 type a request, Ctrl+C to quit"
+    ] }),
     /* @__PURE__ */ jsx6(
       App,
       {
@@ -1131,10 +1182,10 @@ function runTui(opts) {
     cfg,
     controller,
     yolo: Boolean(opts.yolo),
-    welcome: `Axon \xB7 ${cfg.provider}/${resolveModel(cfg)} \xB7 type a request, Ctrl+C to quit`,
     persistKey: (provider, key) => {
       try {
         setApiKey(provider, key);
+        setConfigValue("provider", provider);
       } catch {
       }
     },
@@ -1143,10 +1194,7 @@ function runTui(opts) {
       const tools = buildAllTools();
       const gate = opts.yolo ? allowAllGate : controller.gate;
       const context = loadProjectContext(process.cwd());
-      const system = TUI_SYSTEM + (context ? `
-
-Project context:
-${context}` : "");
+      const system = buildSystemPrompt({ provider: cfg.provider, model: resolveModel(cfg), tools: "all", context });
       return new Engine({ provider, tools, system, cwd: process.cwd(), gate });
     }
   };
@@ -1158,8 +1206,6 @@ ${context}` : "");
 }
 
 // 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;
@@ -1205,12 +1251,13 @@ async function main(opts) {
   const provider = createProvider(cfg);
   const tools = opts.yolo ? buildAllTools() : buildReadOnlyTools();
   const gate = opts.yolo ? allowAllGate : denyGate;
-  const baseSystem = opts.yolo ? YOLO_SYSTEM : READONLY_SYSTEM;
   const context = loadProjectContext(process.cwd());
-  const system = baseSystem + (context ? `
-
-Project context:
-${context}` : "");
+  const system = buildSystemPrompt({
+    provider: cfg.provider,
+    model: resolveModel(cfg),
+    tools: opts.yolo ? "all" : "readonly",
+    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.4.0",
+  "version": "0.4.2",
   "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": {