archbyte 0.3.1 → 0.3.3

package/bin/archbyte.js

@@ -36,9 +36,10 @@ program
   .version(PKG_VERSION)
   .addHelpText('after', `
   Quick start:
-    $ archbyte login       Sign in
-    $ archbyte init        Configure your model provider
-    $ archbyte run         Analyze → generate → serve
+    1. Sign up at https://archbyte.heartbyte.io
+    2. $ archbyte login       Sign in from the CLI
+    3. $ archbyte init        Configure your model provider
+    4. $ archbyte run         Analyze → generate → serve
 
   https://archbyte.heartbyte.io
   Support: archbyte@heartbyte.io

package/dist/agents/providers/google.d.ts

@@ -5,6 +5,5 @@ export declare class GoogleProvider implements LLMProvider {
     constructor(apiKey: string);
     chat(params: ChatParams): Promise<LLMResponse>;
     stream(params: ChatParams): AsyncIterable<LLMChunk>;
-    private toGeminiContents;
-    private toGeminiFunctionDecl;
+    private toContents;
 }

package/dist/agents/providers/google.js

@@ -1,43 +1,47 @@
-import { GoogleGenerativeAI, SchemaType, } from "@google/generative-ai";
+import { GoogleGenAI } from "@google/genai";
 export class GoogleProvider {
     name = "google";
     client;
     constructor(apiKey) {
-        this.client = new GoogleGenerativeAI(apiKey);
+        this.client = new GoogleGenAI({ apiKey });
     }
     async chat(params) {
-        const model = this.client.getGenerativeModel({
-            model: params.model ?? "gemini-2.5-pro",
-            systemInstruction: params.system,
-            ...(params.tools?.length
-                ? {
-                    tools: [
-                        {
-                            functionDeclarations: params.tools.map((t) => this.toGeminiFunctionDecl(t)),
-                        },
-                    ],
-                }
-                : {}),
+        const model = params.model ?? "gemini-2.5-pro";
+        const response = await this.client.models.generateContent({
+            model,
+            contents: this.toContents(params.messages),
+            config: {
+                systemInstruction: params.system,
+                maxOutputTokens: params.maxTokens ?? 8192,
+                ...(params.tools?.length
+                    ? {
+                        tools: [
+                            {
+                                functionDeclarations: params.tools.map((t) => ({
+                                    name: t.name,
+                                    description: t.description,
+                                    parameters: t.input_schema,
+                                })),
+                            },
+                        ],
+                    }
+                    : {}),
+            },
         });
-        const contents = this.toGeminiContents(params.messages);
-        const result = await model.generateContent({ contents });
-        const response = result.response;
         const content = [];
         let hasToolCalls = false;
-        for (const candidate of response.candidates ?? []) {
-            for (const part of candidate.content?.parts ?? []) {
-                if (part.text) {
-                    content.push({ type: "text", text: part.text });
-                }
-                if (part.functionCall) {
-                    hasToolCalls = true;
-                    content.push({
-                        type: "tool_use",
-                        id: `call_${Date.now()}_${Math.random().toString(36).slice(2, 8)}`,
-                        name: part.functionCall.name,
-                        input: part.functionCall.args ?? {},
-                    });
-                }
+        if (response.text) {
+            content.push({ type: "text", text: response.text });
+        }
+        if (response.functionCalls) {
+            for (const fc of response.functionCalls) {
+                hasToolCalls = true;
+                content.push({
+                    type: "tool_use",
+                    id: fc.id ?? `call_${Date.now()}_${Math.random().toString(36).slice(2, 8)}`,
+                    name: fc.name ?? "",
+                    input: fc.args ?? {},
+                });
             }
         }
         return {
@@ -50,44 +54,49 @@ export class GoogleProvider {
         };
     }
     async *stream(params) {
-        const model = this.client.getGenerativeModel({
-            model: params.model ?? "gemini-2.5-pro",
-            systemInstruction: params.system,
-            ...(params.tools?.length
-                ? {
-                    tools: [
-                        {
-                            functionDeclarations: params.tools.map((t) => this.toGeminiFunctionDecl(t)),
-                        },
-                    ],
-                }
-                : {}),
-        });
-        const contents = this.toGeminiContents(params.messages);
-        const result = await model.generateContentStream({ contents });
-        for await (const chunk of result.stream) {
-            for (const candidate of chunk.candidates ?? []) {
-                for (const part of candidate.content?.parts ?? []) {
-                    if (part.text) {
-                        yield { type: "text", text: part.text };
-                    }
-                    if (part.functionCall) {
-                        yield {
-                            type: "tool_use_start",
-                            toolName: part.functionCall.name,
-                            toolId: `call_${Date.now()}`,
-                        };
-                        yield {
-                            type: "tool_use_input",
-                            input: JSON.stringify(part.functionCall.args ?? {}),
-                        };
+        const model = params.model ?? "gemini-2.5-pro";
+        const stream = await this.client.models.generateContentStream({
+            model,
+            contents: this.toContents(params.messages),
+            config: {
+                systemInstruction: params.system,
+                maxOutputTokens: params.maxTokens ?? 8192,
+                ...(params.tools?.length
+                    ? {
+                        tools: [
+                            {
+                                functionDeclarations: params.tools.map((t) => ({
+                                    name: t.name,
+                                    description: t.description,
+                                    parameters: t.input_schema,
+                                })),
+                            },
+                        ],
                     }
+                    : {}),
+            },
+        });
+        for await (const chunk of stream) {
+            if (chunk.text) {
+                yield { type: "text", text: chunk.text };
+            }
+            if (chunk.functionCalls) {
+                for (const fc of chunk.functionCalls) {
+                    yield {
+                        type: "tool_use_start",
+                        toolName: fc.name ?? "",
+                        toolId: fc.id ?? `call_${Date.now()}`,
+                    };
+                    yield {
+                        type: "tool_use_input",
+                        input: JSON.stringify(fc.args ?? {}),
+                    };
                 }
             }
         }
         yield { type: "done" };
     }
-    toGeminiContents(messages) {
+    toContents(messages) {
         return messages.map((m) => {
             const parts = [];
             if (typeof m.content === "string") {
@@ -122,15 +131,4 @@ export class GoogleProvider {
             };
         });
     }
-    toGeminiFunctionDecl(tool) {
-        return {
-            name: tool.name,
-            description: tool.description,
-            parameters: {
-                type: SchemaType.OBJECT,
-                properties: (tool.input_schema.properties ?? {}),
-                required: tool.input_schema.required ?? [],
-            },
-        };
-    }
 }

package/dist/cli/analyze.d.ts

@@ -9,6 +9,8 @@ interface AnalyzeOptions {
     dryRun?: boolean;
     force?: boolean;
     dir?: string;
+    /** When true, skip "archbyte serve" in the Next steps (e.g. called from `archbyte run`) */
+    skipServeHint?: boolean;
 }
 export declare function handleAnalyze(options: AnalyzeOptions): Promise<void>;
 /**

package/dist/cli/analyze.js

@@ -63,7 +63,7 @@ export async function handleAnalyze(options) {
         progress.update(2, "Generating diagram...");
         await autoGenerate(rootDir, options);
         progress.done("Analysis complete");
-        printSummary(analysis, duration, "static");
+        printSummary(analysis, duration, "static", { skipServeHint: options.skipServeHint });
         return;
     }
     // ─── Default mode: static context → LLM pipeline ───
@@ -248,7 +248,7 @@ export async function handleAnalyze(options) {
             writeSpec(rootDir, fbSpec);
             writeScanMetadata(rootDir, duration, "static-fallback");
             await autoGenerate(rootDir, options);
-            printSummary(analysis, duration, "static");
+            printSummary(analysis, duration, "static", { skipServeHint: options.skipServeHint });
         }
         catch (fallbackErr) {
             console.error(chalk.red(`  Static fallback also failed: ${fallbackErr instanceof Error ? fallbackErr.message : String(fallbackErr)}`));
@@ -292,7 +292,7 @@ export async function handleAnalyze(options) {
         agentCount: 5,
         durationMs: duration,
     });
-    printSummary(analysis, duration, "pipeline");
+    printSummary(analysis, duration, "pipeline", { skipServeHint: options.skipServeHint });
 }
 // ─── Shared helpers ───
 function writeAnalysisStatus(rootDir, status, error) {
@@ -352,7 +352,7 @@ async function autoGenerate(rootDir, options) {
         console.error(chalk.gray("You can run 'archbyte generate' manually."));
     }
 }
-function printSummary(analysis, durationMs, mode) {
+function printSummary(analysis, durationMs, mode, options) {
     const components = analysis.components ?? [];
     const connections = analysis.connections ?? [];
     console.log();
@@ -363,7 +363,9 @@ function printSummary(analysis, durationMs, mode) {
     console.log(chalk.gray(`  Connections: ${connections.length}`));
     console.log();
     console.log(chalk.bold("Next"));
-    console.log(`  ${chalk.cyan("archbyte serve")}      Open the interactive diagram`);
+    if (!options?.skipServeHint) {
+        console.log(`  ${chalk.cyan("archbyte serve")}      Open the interactive diagram`);
+    }
     console.log(`  ${chalk.cyan("archbyte validate")}   Check architecture fitness rules ${chalk.yellow("[Pro]")}`);
     console.log(`  ${chalk.cyan("archbyte patrol")}     Continuous architecture monitoring ${chalk.yellow("[Pro]")}`);
     console.log();

package/dist/cli/auth.js

@@ -7,7 +7,9 @@ import { CONFIG_DIR, CREDENTIALS_PATH, API_BASE, CLI_CALLBACK_PORT, OAUTH_TIMEOU
 export async function handleLogin(provider) {
     console.log();
     console.log(chalk.bold.cyan("ArchByte Login"));
-    console.log(chalk.gray("Sign in or create a free account to get started."));
+    console.log();
+    console.log(chalk.gray("  Don't have an account yet? Sign up first:"));
+    console.log(`  ${chalk.bold.cyan("https://archbyte.heartbyte.io")}`);
     console.log();
     const existing = loadCredentials();
     if (existing && !isExpired(existing)) {
@@ -35,7 +37,6 @@ export async function handleLogin(provider) {
     }
     const selectedProvider = provider ?? "github";
     console.log(chalk.gray(`Opening browser for ${selectedProvider} sign-in...`));
-    console.log(chalk.gray("Don't have an account? One will be created automatically."));
     console.log();
     try {
         const credentials = await startOAuthFlow(selectedProvider);

package/dist/cli/run.js

@@ -13,6 +13,7 @@ export async function handleRun(options) {
         force: options.force,
         dryRun: options.dryRun,
         dir: options.dir,
+        skipServeHint: true,
     });
     if (options.dryRun)
         return;

package/dist/cli/serve.js

@@ -75,7 +75,9 @@ export async function handleServe(options) {
     // Try to start, handle port conflicts interactively
     while (true) {
         try {
-            console.log(chalk.green(`UI: http://localhost:${port}`));
+            const url = `http://localhost:${port}`;
+            console.log();
+            console.log(chalk.bold.green(`  ➜  ${chalk.underline(url)}`));
             console.log();
             await startServer({
                 name: projectName,

package/dist/cli/setup.js

@@ -143,10 +143,14 @@ async function validateProviderSilent(providerName, apiKey, model) {
         return true;
     }
     catch (err) {
+        const msg = err instanceof Error ? err.message : String(err);
         if (process.env.ARCHBYTE_DEBUG) {
-            const msg = err instanceof Error ? err.message : String(err);
             console.log(chalk.red(`\n  Validation error: ${msg}`));
         }
+        // Quota/rate limit = key is valid, just exhausted
+        if (msg.includes("429") || msg.includes("RESOURCE_EXHAUSTED") || msg.includes("rate_limit") || msg.includes("quota")) {
+            return "quota";
+        }
         return false;
     }
 }
@@ -308,12 +312,17 @@ export async function handleSetup() {
     const validationModel = profiles[provider].model ?? resolveModel(provider, "fast");
     // Validate provider with spinner
     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) {
+    let result = await validateProviderSilent(provider, apiKey, validationModel);
+    if (result === "quota") {
+        s.stop("valid (quota exceeded — key accepted)", "yellow");
+    }
+    else {
+        s.stop(result ? "valid" : "invalid", result ? "green" : "red");
+    }
+    // Retry loop on failure (not for quota — that means the key works)
+    if (result === false) {
         let retries = 0;
-        while (!isValid && retries < 2) {
+        while (result === false && retries < 2) {
             if (!await confirm("  Retry?"))
                 break;
             retries++;
@@ -323,10 +332,15 @@ export async function handleSetup() {
                 console.log(chalk.green(`  ✓ API key: ${maskKey(newKey)}`));
             }
             const s2 = spinner(`Validating credentials with ${validationModel}`);
-            isValid = await validateProviderSilent(provider, profiles[provider].apiKey, validationModel);
-            s2.stop(isValid ? "valid" : "invalid", isValid ? "green" : "red");
+            result = await validateProviderSilent(provider, profiles[provider].apiKey, validationModel);
+            if (result === "quota") {
+                s2.stop("valid (quota exceeded — key accepted)", "yellow");
+            }
+            else {
+                s2.stop(result ? "valid" : "invalid", result ? "green" : "red");
+            }
         }
-        if (!isValid) {
+        if (result === false) {
             console.log(chalk.yellow("  Warning: credentials unverified. Saving anyway."));
         }
     }
@@ -393,7 +407,10 @@ export async function handleSetup() {
     if (hasClaude || hasCodex) {
         console.log("  " + chalk.cyan("archbyte mcp install") + "    Use from your AI tool");
     }
-    console.log();
+    if (result === false) {
+        console.log(chalk.yellow("  Warning: credentials unverified. Saving anyway."));
+        console.log();
+    }
 }
 function writeArchbyteReadme(archbyteDir) {
     const readmePath = path.join(archbyteDir, "README.md");

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "archbyte",
-  "version": "0.3.1",
+  "version": "0.3.3",
   "description": "ArchByte - See what agents build. As they build it.",
   "type": "module",
   "bin": {
@@ -36,7 +36,7 @@
   },
   "dependencies": {
     "@anthropic-ai/sdk": "^0.74.0",
-    "@google/generative-ai": "^0.24.1",
+    "@google/genai": "^1.41.0",
     "@modelcontextprotocol/sdk": "^1.26.0",
     "chalk": "^5.3.0",
     "chokidar": "^3.5.3",