archbyte 0.2.7 → 0.2.9

package/README.md

@@ -14,7 +14,7 @@ archbyte login
 # Configure your AI provider (BYOK)
 archbyte init
 
-# Or, if you use Claude Code / Codex — skip BYOK entirely:
+# Using Claude Code / Codex? Install the MCP server too:
 archbyte mcp install
 ```
 
@@ -22,7 +22,7 @@ Your API keys stay on your machine. ArchByte never stores or transmits your prov
 
 ### Setup with Claude Code
 
-**Option A: MCP (recommended)** — use ArchByte tools directly in Claude Code, no API key needed:
+**Option A: MCP** — use ArchByte tools directly in Claude Code. You still need to sign in and configure a provider first:
 
 ```bash
 archbyte mcp install

package/dist/cli/setup.js

@@ -11,7 +11,7 @@ const __filename = fileURLToPath(import.meta.url);
 const __dirname = path.dirname(__filename);
 const PROVIDERS = [
     { name: "anthropic", label: "Anthropic", hint: "Claude Sonnet / Opus" },
-    { name: "openai", label: "OpenAI", hint: "Codex / GPT-5.2" },
+    { name: "openai", label: "OpenAI", hint: "GPT-5.2 / o4-mini" },
     { name: "google", label: "Google", hint: "Gemini Flash / Pro" },
 ];
 const PROVIDER_MODELS = {
@@ -22,10 +22,9 @@ const PROVIDER_MODELS = {
         { id: "claude-opus-4-6", label: "Claude Opus 4.6", hint: "Most capable" },
     ],
     openai: [
-        { id: "", label: "Default (recommended)", hint: "Codex for fast agents, GPT-5.2 for connections" },
-        { id: "gpt-5.2-codex", label: "GPT-5.2 Codex", hint: "Fast, coding-optimized" },
-        { id: "gpt-5.2", label: "GPT-5.2", hint: "General purpose, thinking model" },
-        { id: "o3", label: "o3", hint: "Reasoning model" },
+        { id: "", label: "Default (recommended)", hint: "GPT-5.2 for all agents" },
+        { id: "gpt-5.2", label: "GPT-5.2", hint: "Latest flagship model" },
+        { id: "o4-mini", label: "o4-mini", hint: "Fast reasoning model" },
     ],
     google: [
         { id: "", label: "Default (recommended)", hint: "Flash for fast agents, Pro for connections" },
@@ -93,10 +92,47 @@ function askHidden(prompt) {
         stdin.on("data", onData);
     });
 }
-async function validateProviderSilent(providerName, apiKey) {
+function askText(prompt) {
+    return new Promise((resolve) => {
+        process.stdout.write(prompt);
+        const stdin = process.stdin;
+        const wasRaw = stdin.isRaw;
+        stdin.setRawMode(true);
+        stdin.resume();
+        stdin.setEncoding("utf8");
+        let input = "";
+        const onData = (data) => {
+            for (const ch of data) {
+                if (ch === "\n" || ch === "\r" || ch === "\u0004") {
+                    stdin.setRawMode(wasRaw ?? false);
+                    stdin.pause();
+                    stdin.removeListener("data", onData);
+                    process.stdout.write("\n");
+                    resolve(input);
+                    return;
+                }
+                else if (ch === "\u0003") {
+                    process.stdout.write("\n");
+                    process.exit(0);
+                }
+                else if (ch === "\u007F" || ch === "\b") {
+                    if (input.length > 0) {
+                        input = input.slice(0, -1);
+                        process.stdout.write("\b \b");
+                    }
+                }
+                else {
+                    input += ch;
+                    process.stdout.write(ch);
+                }
+            }
+        };
+        stdin.on("data", onData);
+    });
+}
+async function validateProviderSilent(providerName, apiKey, model) {
     try {
         const provider = createProvider({ provider: providerName, apiKey });
-        const model = resolveModel(providerName, "fast");
         await provider.chat({
             model,
             system: "Reply with 'ok'.",
@@ -105,7 +141,11 @@ async function validateProviderSilent(providerName, apiKey) {
         });
         return true;
     }
-    catch {
+    catch (err) {
+        if (process.env.ARCHBYTE_DEBUG) {
+            const msg = err instanceof Error ? err.message : String(err);
+            console.log(chalk.red(`\n  Validation error: ${msg}`));
+        }
         return false;
     }
 }
@@ -132,26 +172,7 @@ export async function handleSetup() {
     if (hasClaude || hasCodex) {
         const tools = [hasClaude && "Claude Code", hasCodex && "Codex CLI"].filter(Boolean).join(" and ");
         console.log(chalk.cyan(`  Detected ${tools} on this machine.`));
-        console.log(chalk.white(`  You can use ArchByte directly through MCP. No API key needed.`));
-        console.log(chalk.white(`  Your AI tool already provides the model, so you skip the BYOK step.`));
-        console.log();
-        console.log(chalk.white(`  Run: `) + chalk.bold.cyan(`archbyte mcp install`));
-        console.log();
-        const continueIdx = await select("Continue with BYOK setup anyway?", [
-            `Skip ${chalk.gray("(use MCP instead, recommended)")}`,
-            `Continue ${chalk.gray("(set up your own API key)")}`,
-        ]);
-        if (continueIdx === 0) {
-            console.log();
-            console.log(chalk.gray("  Run `archbyte mcp install` to configure MCP for your AI tool."));
-            console.log();
-            console.log(chalk.gray("  Then open your AI tool and try:"));
-            console.log(chalk.cyan('    "Analyze the architecture of this project"'));
-            console.log(chalk.cyan('    "Export the architecture as a Mermaid diagram"'));
-            console.log(chalk.cyan('    "Show me the architecture stats"'));
-            console.log();
-            return;
-        }
+        console.log(chalk.white(`  After setup, run `) + chalk.bold.cyan(`archbyte mcp install`) + chalk.white(` to use ArchByte from your AI tool.`));
         console.log();
     }
     const config = loadConfig();
@@ -182,7 +203,8 @@ export async function handleSetup() {
         for (const p of configured) {
             const active = p === config.provider ? chalk.green(" (active)") : "";
             const model = profiles[p].model ? chalk.gray(` model: ${profiles[p].model}`) : "";
-            console.log(chalk.gray(`  ${p}: ${maskKey(profiles[p].apiKey)}${model}${active}`));
+            const email = profiles[p].email ? chalk.gray(` (${profiles[p].email})`) : "";
+            console.log(chalk.gray(`  ${p}: ${maskKey(profiles[p].apiKey)}${email}${model}${active}`));
         }
         console.log();
     }
@@ -232,6 +254,30 @@ export async function handleSetup() {
     if (!profiles[provider])
         profiles[provider] = { apiKey: "" };
     profiles[provider].apiKey = apiKey;
+    // Step 2b: Email for this provider account (optional)
+    const existingEmail = profiles[provider].email;
+    if (existingEmail) {
+        console.log(chalk.gray(`\n  Account email: ${existingEmail}`));
+        const emailIdx = await select("  Update email?", [
+            `Keep existing ${chalk.gray("(" + existingEmail + ")")}`,
+            "Enter new email",
+            "Skip",
+        ]);
+        if (emailIdx === 1) {
+            const newEmail = await askText(chalk.bold("  Email: "));
+            if (newEmail) {
+                profiles[provider].email = newEmail;
+                console.log(chalk.green(`  ✓ Email: ${newEmail}`));
+            }
+        }
+    }
+    else {
+        const email = await askText(chalk.bold(`  ${selected.label} account email ${chalk.gray("(optional, Enter to skip)")}: `));
+        if (email) {
+            profiles[provider].email = email;
+            console.log(chalk.green(`  ✓ Email: ${email}`));
+        }
+    }
     // Step 3: Model selection
     const models = PROVIDER_MODELS[provider];
     if (models) {
@@ -251,9 +297,11 @@ export async function handleSetup() {
         }
     }
     config.profiles = profiles;
+    // Validate with the model the user actually chose
+    const validationModel = profiles[provider].model ?? resolveModel(provider, "fast");
     // Validate provider with spinner
-    const s = spinner("Validating credentials");
-    let isValid = await validateProviderSilent(provider, apiKey);
+    const s = spinner(`Validating credentials with ${validationModel}`);
+    let isValid = await validateProviderSilent(provider, apiKey, validationModel);
     s.stop(isValid ? "valid" : "invalid", isValid ? "green" : "red");
     // Retry loop on failure
     if (!isValid) {
@@ -267,8 +315,8 @@ export async function handleSetup() {
                 profiles[provider].apiKey = newKey;
                 console.log(chalk.green(`  ✓ API key: ${maskKey(newKey)}`));
             }
-            const s2 = spinner("Validating credentials");
-            isValid = await validateProviderSilent(provider, profiles[provider].apiKey);
+            const s2 = spinner(`Validating credentials with ${validationModel}`);
+            isValid = await validateProviderSilent(provider, profiles[provider].apiKey, validationModel);
             s2.stop(isValid ? "valid" : "invalid", isValid ? "green" : "red");
         }
         if (!isValid) {
@@ -319,6 +367,10 @@ export async function handleSetup() {
     console.log();
     console.log(chalk.bold("  Next: ") + chalk.cyan("archbyte run") + chalk.bold(" to analyze your codebase."));
     console.log(chalk.gray("        Run from your project root, or use ") + chalk.cyan("archbyte run -d /path/to/project"));
+    if (hasClaude || hasCodex) {
+        console.log();
+        console.log(chalk.bold("  MCP:  ") + chalk.cyan("archbyte mcp install") + chalk.bold(" to use ArchByte from your AI tool."));
+    }
     console.log();
 }
 function writeArchbyteReadme(archbyteDir) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "archbyte",
-  "version": "0.2.7",
+  "version": "0.2.9",
   "description": "ArchByte - See what agents build. As they build it.",
   "type": "module",
   "bin": {