@wanghuimvp/axon 0.4.0 → 0.4.1

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 ?? {} }
   };
@@ -1046,36 +1060,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: [
@@ -1089,27 +1121,32 @@ function Setup({
 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 +1168,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 {
       }
     },

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@wanghuimvp/axon",
-  "version": "0.4.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": {