@getmegabrain/cli 0.1.0 → 0.1.1

package/dist/constants.js

@@ -5,5 +5,5 @@
 export const APP_BASE_URL = (process.env.MEGABRAIN_APP_URL ?? 'https://getmegabrain.com').replace(/\/+$/, '');
 /** OpenAI-compatible gateway root; OpenCode's provider appends `/chat/completions`. */
 export const GATEWAY_BASE_URL = `${APP_BASE_URL}/api/gateway/`;
-/** Gateway model id used when the user hasn't picked one. */
-export const DEFAULT_MODEL_ID = 'kilo-auto/balanced';
+/** Gateway model id used when the user hasn't picked one (MegaBrain auto-router). */
+export const DEFAULT_MODEL_ID = 'mb-auto/balanced';

package/dist/index.js

@@ -3,20 +3,27 @@ import { spawn } from 'node:child_process';
 import { login } from './device-auth.js';
 import { readCredentials, writeCredentials, clearCredentials } from './credentials.js';
 import { writeOpenCodeConfig } from './opencode-config.js';
-async function ensureSignedIn() {
-    if (readCredentials())
-        return;
+import { fetchGatewayModelIds } from './models.js';
+async function signIn() {
     console.log('Signing in to MegaBrain…');
     const { token, userId, userEmail } = await login();
     writeCredentials({ token, userId, userEmail });
-    writeOpenCodeConfig(token);
+    writeOpenCodeConfig(token, await fetchGatewayModelIds());
+    return userEmail;
+}
+async function ensureSignedIn() {
+    const credentials = readCredentials();
+    if (credentials) {
+        // Refresh OpenCode's config on every launch so CLI upgrades (new default
+        // model, catalog changes) take effect without an explicit re-login.
+        writeOpenCodeConfig(credentials.token, await fetchGatewayModelIds());
+        return;
+    }
+    const userEmail = await signIn();
     console.log(`\nSigned in as ${userEmail}.\n`);
 }
 async function runLogin() {
-    console.log('Signing in to MegaBrain…');
-    const { token, userId, userEmail } = await login();
-    writeCredentials({ token, userId, userEmail });
-    writeOpenCodeConfig(token);
+    const userEmail = await signIn();
     console.log(`\nSigned in as ${userEmail}. Run \`megabrain\` to start coding.\n`);
 }
 function runLogout() {

package/dist/models.js

@@ -0,0 +1,25 @@
+import { APP_BASE_URL } from './constants.js';
+/**
+ * Fetches the MegaBrain Gateway's public model catalog so OpenCode's model
+ * picker can list every model the gateway serves (including BYOK-eligible
+ * ones — BYOK routing is transparent gateway-side, the ids are the same).
+ *
+ * Best-effort: returns [] on any failure so login never blocks on this —
+ * the caller falls back to registering just the default auto-router model.
+ */
+export async function fetchGatewayModelIds() {
+    try {
+        const res = await fetch(`${APP_BASE_URL}/api/gateway/models`);
+        if (!res.ok)
+            return [];
+        const body = (await res.json());
+        if (!Array.isArray(body.data))
+            return [];
+        return body.data
+            .map(model => model.id)
+            .filter((id) => typeof id === 'string' && id.length > 0);
+    }
+    catch {
+        return [];
+    }
+}

package/dist/opencode-config.js

@@ -16,7 +16,18 @@ const PROVIDER_ID = 'megabrain';
 function toOpenCodeBaseUrl(gatewayBaseUrl) {
     return gatewayBaseUrl.replace(/\/+$/, '') + '/v1';
 }
-export function writeOpenCodeConfig(token, modelId = DEFAULT_MODEL_ID) {
+/**
+ * @param token Gateway token from `megabrain login`.
+ * @param modelIds Full catalog to register in OpenCode's model picker; the
+ *   default auto-router model is always included and stays the default.
+ */
+export function writeOpenCodeConfig(token, modelIds = []) {
+    const models = {
+        [DEFAULT_MODEL_ID]: { name: DEFAULT_MODEL_ID },
+    };
+    for (const id of modelIds) {
+        models[id] = { name: id };
+    }
     const config = {
         $schema: 'https://opencode.ai/config.json',
         provider: {
@@ -24,10 +35,10 @@ export function writeOpenCodeConfig(token, modelId = DEFAULT_MODEL_ID) {
                 npm: '@ai-sdk/openai-compatible',
                 name: 'MegaBrain Gateway',
                 options: { baseURL: toOpenCodeBaseUrl(GATEWAY_BASE_URL), apiKey: token },
-                models: { [modelId]: { name: modelId } },
+                models,
             },
         },
-        model: `${PROVIDER_ID}/${modelId}`,
+        model: `${PROVIDER_ID}/${DEFAULT_MODEL_ID}`,
     };
     fs.mkdirSync(OPENCODE_CONFIG_DIR, { recursive: true });
     fs.writeFileSync(OPENCODE_CONFIG_FILE, JSON.stringify(config, null, 2), { mode: 0o600 });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@getmegabrain/cli",
-  "version": "0.1.0",
+  "version": "0.1.1",
   "description": "MegaBrain CLI — sign in once, then code from your terminal via OpenCode wired to the MegaBrain Gateway.",
   "license": "MIT",
   "type": "module",