@getmegabrain/cli 0.1.0 → 0.1.2

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

@@ -7,27 +7,73 @@
  * same thing server-side for BrainClaw sandboxes) — same config shape, same
  * gateway, adapted to write to the local user's OpenCode config directory
  * instead of a pinned in-container path.
+ *
+ * The CLI rewrites this file on every launch (to pick up catalog and default
+ * changes), so it must MERGE into the existing config rather than replace it:
+ * only the keys the CLI owns (`provider.megabrain`, `model`) are overwritten,
+ * everything the user added themselves (lsp, theme, mcp, other providers…)
+ * is preserved.
  */
 import fs from 'node:fs';
 import { GATEWAY_BASE_URL, DEFAULT_MODEL_ID } from './constants.js';
 import { OPENCODE_CONFIG_DIR, OPENCODE_CONFIG_FILE } from './paths.js';
 const PROVIDER_ID = 'megabrain';
+/** Legacy auto-router default written by CLI ≤0.1.0; migrated to mb-auto/*. */
+const LEGACY_MODEL_PREFIX = `${PROVIDER_ID}/kilo-auto/`;
 /** OpenCode's `@ai-sdk/openai-compatible` provider appends `/chat/completions` to `baseURL`. */
 function toOpenCodeBaseUrl(gatewayBaseUrl) {
     return gatewayBaseUrl.replace(/\/+$/, '') + '/v1';
 }
-export function writeOpenCodeConfig(token, modelId = DEFAULT_MODEL_ID) {
+function readExistingConfig() {
+    try {
+        const parsed = JSON.parse(fs.readFileSync(OPENCODE_CONFIG_FILE, 'utf8'));
+        if (parsed !== null && typeof parsed === 'object' && !Array.isArray(parsed)) {
+            return parsed;
+        }
+    }
+    catch {
+        // Missing or unparsable — start from scratch.
+    }
+    return {};
+}
+/**
+ * @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 existing = readExistingConfig();
+    const existingProvider = existing.provider !== null && typeof existing.provider === 'object'
+        ? existing.provider
+        : {};
+    // Keep a user-chosen default model, but migrate the stale kilo-auto default
+    // that older CLI versions wrote.
+    const existingModel = typeof existing.model === 'string' ? existing.model : undefined;
+    const model = existingModel === undefined || existingModel.startsWith(LEGACY_MODEL_PREFIX)
+        ? `${PROVIDER_ID}/${DEFAULT_MODEL_ID}`
+        : existingModel;
     const config = {
+        ...existing,
         $schema: 'https://opencode.ai/config.json',
         provider: {
+            ...existingProvider,
             [PROVIDER_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,
+        // LSP diagnostics make the agent's edits far more reliable, but OpenCode
+        // ships with LSP off. Default it on; a user's explicit choice survives.
+        lsp: existing.lsp ?? true,
     };
     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.2",
   "description": "MegaBrain CLI — sign in once, then code from your terminal via OpenCode wired to the MegaBrain Gateway.",
   "license": "MIT",
   "type": "module",
@@ -10,6 +10,9 @@
   "files": [
     "dist"
   ],
+  "publishConfig": {
+    "access": "public"
+  },
   "engines": {
     "node": ">=18"
   },