agy-mcp-bridge 0.1.3 → 0.1.4

package/README.md

@@ -370,7 +370,9 @@ Khi `npm start` đang chạy, mở `http://127.0.0.1:3000/ui` (đúng cổng đ
 
 **Thay đổi trong workspace:** mỗi task có mục "Thay đổi trong workspace" (mở ra mới đọc, có nút Làm mới) liệt kê file đã sửa, xóa, file mới chưa theo dõi và số dòng thêm/xóa so với commit HEAD, kể cả thay đổi có từ trước khi task chạy. Chỉ hiện tên file và số dòng, không hiện nội dung, và chỉ cho workspace vẫn còn trong danh sách được phép. Để đọc an toàn, bridge **không bao giờ nạp cấu hình của repo**: `git status`/`git diff` thông thường chạy cả những lệnh do chính repo khai báo (`core.fsmonitor`, `filter.*.clean`, `diff.*.textconv`), mà Antigravity có quyền ghi file nên có thể cài sẵn vào `.git/config`; chỉ tắt trình diff ngoài là chưa đủ (đã thử: lệnh `filter clean` vẫn chạy). Thay vào đó bridge so sánh trong một thư mục git tạm với cấu hình rỗng, chỉ mượn kho đối tượng của repo, tắt cấu hình hệ thống/người dùng, và chỉ sao chép vài thiết lập xuống dòng/quyền file đã kiểm tra giá trị (`core.autocrlf`, `eol`, `filemode`, `symlinks`, `ignorecase`). Có test dựng repo "nhiễm độc" để chứng minh không lệnh nào bị chạy. Giới hạn: cần `git` trong PATH, chỉ tính khác biệt nội dung (file chỉ khác kiểu xuống dòng sau chuyển đổi không tính), tối đa 500 mục mỗi nhóm, không hiện submodule, và không thay thế việc tự xem diff đầy đủ.
 
-Lưu ý bảo mật: nội dung yêu cầu và phản hồi được hiển thị dạng văn bản thuần, trang dùng CSP không cho script/style nội tuyến hay tải từ nơi khác, nên output của model không thể thực thi. Yêu cầu của Tech Lead được lưu cùng bản ghi task trong `.bridge-state/tasks` (đã qua lọc secret nhận diện được) để xem lại sau khi restart; đừng đưa secret vào task. Giao diện chỉ đọc, ngoại trừ hai thao tác cần phiên đăng nhập: hủy task và bật/tắt giao việc. Mã nguồn giao diện nằm ở `ui/` (HTML/CSS/JS thuần, không cần build), phần máy chủ ở `src/ui/`.
+Lưu ý bảo mật: nội dung yêu cầu và phản hồi được hiển thị dạng văn bản thuần, trang dùng CSP không cho script/style nội tuyến hay tải từ nơi khác, nên output của model không thể thực thi. Yêu cầu của Tech Lead được lưu cùng bản ghi task trong `.bridge-state/tasks` (đã qua lọc secret nhận diện được) để xem lại sau khi restart; đừng đưa secret vào task. Giao diện chỉ đọc, ngoại trừ các thao tác cần phiên đăng nhập: hủy task, bật/tắt giao việc và chọn model. Mã nguồn giao diện nằm ở `ui/` (HTML/CSS/JS thuần, không cần build), phần máy chủ ở `src/ui/`.
+
+**Chọn model:** mục Cài đặt → Model liệt kê các model lấy từ `agy models` (chỉ chạy khi mở trang Cài đặt). Chọn một model thì bridge truyền `--model <id>` cho mọi task mới; chọn "Mặc định của Antigravity" thì không truyền cờ nào. Có hiệu lực ngay, không cần khởi động lại; task đang chạy hoặc đang chờ giữ model đã chốt lúc nhận, và mỗi task ghi lại model đã dùng (hiện trong chi tiết task). Chỉ nhận id nằm trong danh sách của `agy models`; nếu không đọc được danh sách thì không cho đổi. Lựa chọn lưu ở `.bridge-state/model.json`. Codex/Claude không chọn được model theo từng task (không thêm tham số vào tool MCP nên không tốn thêm token).
 
 ## 11. Ghi chú thiết kế và việc chưa làm
 

package/dist/src/antigravity/invocation.js

@@ -1,5 +1,5 @@
-export function invocationFor(prompt, timeout, transport) {
-    const common = ['--print-timeout', `${timeout}s`, '--disable-slash-commands'];
+export function invocationFor(prompt, timeout, transport, model = null) {
+    const common = ['--print-timeout', `${timeout}s`, '--disable-slash-commands', ...(model ? ['--model', model] : [])];
     return transport === 'argv'
         ? { args: ['--print', prompt, '--output-format', 'json', ...common], stdin: undefined }
         : { args: ['--input-format', 'stream-json', '--output-format', 'stream-json', ...common], stdin: JSON.stringify({ event: 'user', message: { content: prompt } }) + '\n' };

package/dist/src/models.js

@@ -0,0 +1,63 @@
+import { mkdir, readFile, rename, rm, writeFile } from 'node:fs/promises';
+import path from 'node:path';
+import { startProcess, terminateOwned } from './antigravity/process.js';
+export const MODEL_FILE = 'model.json';
+// Ids go straight into the agy command line, so only plain identifier characters are ever accepted (and never a leading dash).
+const MODEL_ID = /^[A-Za-z0-9][A-Za-z0-9._:-]{0,99}$/;
+export const validModelId = (value) => typeof value === 'string' && MODEL_ID.test(value);
+/** Parses `agy models`: one model per line, the id first, then the display name after two or more spaces. */
+export function parseModels(text) {
+    const models = [];
+    for (const line of text.split(/\r?\n/)) {
+        const match = line.match(/^(\S+)\s{2,}(.+?)\s*$/);
+        if (match && validModelId(match[1]) && !models.some(model => model.id === match[1]))
+            models.push({ id: match[1], label: match[2].slice(0, 100) });
+    }
+    return models;
+}
+/** Runs `agy models`; rejects when the command fails, times out or prints nothing recognizable. */
+export function listModels(executable, cwd, extraEnvPassthrough = []) {
+    return new Promise((resolve, reject) => {
+        const child = startProcess(executable, ['models'], cwd, { extraEnvPassthrough });
+        let text = '';
+        let timedOut = false;
+        const timer = setTimeout(() => { timedOut = true; void terminateOwned(child).catch(() => { }); }, 10000);
+        const collect = (chunk) => { if (text.length < 32768)
+            text += chunk.toString().slice(0, 32768 - text.length); };
+        child.stdout.on('data', collect);
+        child.once('error', error => { clearTimeout(timer); reject(error); });
+        child.once('close', code => {
+            clearTimeout(timer);
+            const models = parseModels(text);
+            if (timedOut)
+                reject(new Error('`agy models` timed out'));
+            else if (code !== 0 || !models.length)
+                reject(new Error('`agy models` did not return a model list'));
+            else
+                resolve(models);
+        });
+    });
+}
+/** The model chosen in the web UI, or null to leave the choice to Antigravity. Read on every task, so a change needs no restart. */
+export async function readModel(stateDirectory) {
+    try {
+        const value = JSON.parse(await readFile(path.join(stateDirectory, MODEL_FILE), 'utf8')).model;
+        return validModelId(value) ? value : null;
+    }
+    catch {
+        return null;
+    }
+}
+export async function writeModel(stateDirectory, model) {
+    const file = path.join(stateDirectory, MODEL_FILE);
+    if (model === null) {
+        await rm(file, { force: true });
+        return;
+    }
+    if (!validModelId(model))
+        throw new Error('Invalid model id');
+    await mkdir(stateDirectory, { recursive: true });
+    await writeFile(`${file}.tmp`, JSON.stringify({ model }) + '\n');
+    await rename(`${file}.tmp`, file);
+}
+//# sourceMappingURL=models.js.map

package/dist/src/tasks/manager.js

@@ -8,6 +8,8 @@ import { redact } from '../security/redact.js';
 import { delegationState } from '../delegation.js';
 import { startProcess, terminateOwned } from '../antigravity/process.js';
 import { invocationFor, finalEnvelope } from '../antigravity/invocation.js';
+import { packageRoot } from '../paths.js';
+import { listModels, readModel } from '../models.js';
 import { pidAlive, recoverExecutionLock } from './execution-lock.js';
 const TASK_FILE = /^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}\.json$/;
 export class TaskManager {
@@ -69,7 +71,8 @@ export class TaskManager {
             request: {
                 task: redact(input.task), acceptance_criteria: input.acceptance_criteria.map(redact),
                 context: input.context === undefined ? null : redact(input.context), timeout_seconds: timeout,
-                client: meta.client ? redact(meta.client).slice(0, 100) : null
+                client: meta.client ? redact(meta.client).slice(0, 100) : null,
+                model: await readModel(this.runtime.stateDirectory)
             }
         };
         this.entries.set(task.task_id, entry);
@@ -140,7 +143,7 @@ export class TaskManager {
             task.started_at = new Date().toISOString();
             await this.persist(task);
             checkStarting();
-            const invocation = invocationFor(promptFor({ ...input, workspace }, entry.request?.client), timeout, this.config.prompt_transport);
+            const invocation = invocationFor(promptFor({ ...input, workspace }, entry.request?.client), timeout, this.config.prompt_transport, entry.request?.model ?? null);
             const options = { extraEnvPassthrough: this.config.extra_env_passthrough, stdin: invocation.stdin };
             const child = this.runtime.launch ? this.runtime.launch(invocation.args, workspace, options) : startProcess(this.runtime.executable, invocation.args, workspace, options);
             entry.child = child;
@@ -314,6 +317,11 @@ export class TaskManager {
             .map(([id, entry]) => ({ ...this.summary(id), title: entry.request?.task.split('\n').find(line => line.trim())?.trim().slice(0, 140) ?? null }))
             .sort((a, b) => Date.parse(b.queued_at ?? b.started_at) - Date.parse(a.queued_at ?? a.started_at));
     }
+    /** Models offered by the installed Antigravity CLI (read on demand, never at startup). */
+    availableModels() {
+        return this.runtime.models ? this.runtime.models() : listModels(this.runtime.executable, packageRoot, this.config.extra_env_passthrough);
+    }
+    currentModel() { return readModel(this.runtime.stateDirectory); }
     /** Full record for the web UI: the request, the exact prompt shape sent to the CLI (redacted), and the task with logs. */
     detail(id) {
         const task = this.get(id);

package/dist/src/ui/routes.js

@@ -6,6 +6,7 @@ import { redact } from '../security/redact.js';
 import { paths } from '../paths.js';
 import { knownUpdate, readPackageInfo } from '../update.js';
 import { workspaceChanges } from '../changes.js';
+import { validModelId, writeModel } from '../models.js';
 import { inside } from '../security/workspace.js';
 const assets = {
     '/ui': { file: 'index.html', type: 'text/html; charset=utf-8' },
@@ -51,6 +52,7 @@ async function overview(ctx) {
         },
         limits: env.limits ?? {}, allowed_workspaces: env.allowed_workspaces ?? [], session_workspaces: env.session_workspaces ?? [], limitations: env.limitations ?? [],
         delegation: { ...state, controllable: delegationControllable() },
+        model: await ctx.manager.currentModel(),
         queue: ctx.manager.overview().queue,
         warnings: [await ctx.manager.lockDiagnostic()].filter(Boolean),
         update: knownUpdate() && { ...knownUpdate(), package: readPackageInfo().name, command: 'agy-mcp-bridge update' },
@@ -162,6 +164,38 @@ export async function handleUi(req, res, ctx) {
             json(res, 200, { ...delegationState(ctx.manager.stateDirectory), controllable: true });
             return true;
         }
+        if (pathname === '/ui/api/models' && method === 'GET') {
+            const current = await ctx.manager.currentModel();
+            try {
+                json(res, 200, { models: await ctx.manager.availableModels(), current });
+            }
+            catch (error) {
+                json(res, 200, { models: [], current, error: error instanceof Error ? redact(error.message) : 'Could not list models' });
+            }
+            return true;
+        }
+        if (pathname === '/ui/api/model' && method === 'POST') {
+            const body = await readJson(req);
+            if (body.model !== null && !validModelId(body.model)) {
+                json(res, 400, { error: 'model must be a model id or null' });
+                return true;
+            }
+            // Only ids that `agy models` lists are accepted; if the list cannot be read, the choice cannot be verified, so it is refused.
+            if (body.model !== null) {
+                const listed = await ctx.manager.availableModels().catch(() => null);
+                if (!listed) {
+                    json(res, 409, { error: 'Cannot verify the model: `agy models` is not available' });
+                    return true;
+                }
+                if (!listed.some(model => model.id === body.model)) {
+                    json(res, 400, { error: 'Unknown model; choose one listed by `agy models`' });
+                    return true;
+                }
+            }
+            await writeModel(ctx.manager.stateDirectory, body.model);
+            json(res, 200, { current: body.model });
+            return true;
+        }
         json(res, 404, { error: 'Not found' });
     }
     catch (error) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "agy-mcp-bridge",
-  "version": "0.1.3",
+  "version": "0.1.4",
   "description": "Local authenticated HTTP MCP bridge that lets Codex and Claude Code delegate bounded coding tasks to the Antigravity CLI (agy)",
   "license": "MIT",
   "keywords": [

package/ui/app.css

@@ -203,3 +203,7 @@ pre.log { margin: 0; padding: 12px; border-radius: 10px; background: #0b0e14; co
   .login-card { padding: 24px; }
 }
 @media (prefers-reduced-motion: reduce) { * { animation: none !important; transition: none !important; } }
+
+.select { max-width: 100%; padding: 9px 12px; border: 1px solid var(--border); border-radius: 10px; background: var(--surface-2); font: inherit; color: inherit; }
+.select:focus { border-color: var(--accent); }
+.select:disabled { opacity: .55; cursor: not-allowed; }

package/ui/app.js

@@ -135,6 +135,15 @@ async function setDelegation(enabled) {
   try { await api('delegation', {method: 'POST', body: {enabled}}); notify(enabled ? 'Đã bật giao việc cho Antigravity' : 'Đã tắt giao việc cho Antigravity'); await refresh(); }
   catch (error) { notify(error.message); }
 }
+async function loadModels() {
+  state.modelsLoading = true;
+  try { state.models = await api('models'); } catch (error) { state.models = {models: [], current: null, error: error.message}; }
+  state.modelsLoading = false; patch();
+}
+async function setModel(model) {
+  try { await api('model', {method: 'POST', body: {model}}); notify(model ? 'Đã đổi model cho các task mới' : 'Đã về model mặc định của Antigravity'); state.models = null; await refresh(); }
+  catch (error) { notify(error.message); await refresh(); }
+}
 async function copy(text) {
   try { await navigator.clipboard.writeText(text); notify('Đã sao chép'); } catch { notify('Không thể sao chép tự động'); }
 }
@@ -248,7 +257,7 @@ function detailView() {
   const facts = h('dl', {class: 'kv'},
     ...[['Gửi lúc', clock(task.queued_at || task.started_at)], ['Bắt đầu chạy', queued ? 'Chưa chạy' : clock(task.started_at)], ['Kết thúc', clock(task.ended_at)],
       ['Thời gian chạy', running ? elapsedEl(task.started_at) : queued ? '—' : duration(task.duration_ms)],
-      ['Giới hạn', request ? `${request.timeout_seconds}s` : '—'], ['Exit code', task.exit_code ?? '—'], ['PID', task.pid ?? '—'],
+      ['Giới hạn', request ? `${request.timeout_seconds}s` : '—'], ['Model', request && 'model' in request ? request.model || 'mặc định' : '—'], ['Exit code', task.exit_code ?? '—'], ['PID', task.pid ?? '—'],
       ['CLI status', task.cli_status ?? '—']].map(([k, v]) => h('div', {}, h('dt', {text: k}), h('dd', {}, typeof v === 'string' || typeof v === 'number' ? String(v) : v))));
 
   const asker = request?.client ? `Tech Lead · ${request.client}` : 'Tech Lead (Codex / Claude)';
@@ -301,6 +310,20 @@ function patchLogs() {
 }
 
 function row(label, value) { return h('div', {}, h('dt', {text: label}), h('dd', {text: String(value)})); }
+function modelCard(o) {
+  const loaded = state.models;
+  const known = (loaded?.models || []).some(m => m.id === o.model);
+  const select = h('select', {class: 'select', 'aria-label': 'Model cho task mới', disabled: !loaded || Boolean(loaded.error), onchange: event => setModel(event.target.value || null)},
+    h('option', {value: '', text: 'Mặc định của Antigravity'}),
+    ...(loaded?.models || []).map(m => h('option', {value: m.id, text: `${m.label} (${m.id})`})),
+    o.model && !known ? h('option', {value: o.model, text: `${o.model} (không còn trong danh sách)`}) : null);
+  select.value = o.model || '';
+  return h('section', {class: 'card panel wide'}, h('h2', {text: 'Model'}),
+    h('div', {class: 'switch-row'}, h('div', {}, h('strong', {text: o.model || 'Mặc định của Antigravity'}),
+      h('span', {class: 'muted', text: o.model ? 'Bridge truyền --model này cho mọi task mới.' : 'Bridge không truyền --model; Antigravity tự dùng model đã cấu hình.'})), select),
+    loaded?.error ? h('div', {class: 'callout warn', text: `Không lấy được danh sách model: ${loaded.error}`}) : null,
+    h('p', {text: 'Áp dụng cho task mới, không cần khởi động lại. Task đang chạy hoặc đang chờ giữ model đã chốt lúc nhận. Danh sách lấy từ `agy models`; model được ghi lại trong từng task.'}));
+}
 function settingsView(o) {
   const delegation = o.delegation;
   return [h('div', {class: 'page-head'}, h('div', {}, h('h1', {text: 'Cài đặt'}), h('p', {text: 'Trạng thái và giới hạn của bridge. Phần lớn chỉ đọc; chỉnh bằng file cấu hình rồi khởi động lại.'}))),
@@ -311,6 +334,7 @@ function settingsView(o) {
           switchControl(delegation.enabled, delegation.controllable, 'Cho phép giao việc cho Antigravity')),
         delegation.controllable ? h('p', {text: 'Tắt để buộc Codex / Claude tự làm. Task đang chạy hoặc đang chờ không bị dừng, nhưng task đang chờ sẽ bị hủy khi đến lượt. Có hiệu lực ngay, không cần khởi động lại.'})
           : h('div', {class: 'callout warn', text: 'Đang bị khóa bởi biến môi trường BRIDGE_DELEGATION. Đổi biến rồi khởi động lại server để mở.'})),
+      modelCard(o),
       h('section', {class: 'card panel'}, h('h2', {text: 'Kết nối'}),
         h('div', {class: 'copy-row'}, h('code', {text: o.server.mcp_endpoint}), h('button', {class: 'btn small', onclick: () => copy(o.server.mcp_endpoint), text: 'Sao chép'})),
         h('dl', {class: 'rows'}, row('Giao thức', 'MCP Streamable HTTP'), row('Xác thực', 'Bearer token dùng chung'), row('Phiên bản bridge', o.server.version),
@@ -388,7 +412,8 @@ function patch() {
     setRegion('detail', detailKey, () => detailView());
     if (!rebuilt) patchLogs();
   } else {
-    setRegion('settings', JSON.stringify([o.cli, o.delegation, o.limits, o.allowed_workspaces, o.session_workspaces, o.server.mcp_endpoint]), () => settingsView(o));
+    if (!state.models && !state.modelsLoading) loadModels();
+    setRegion('settings', JSON.stringify([state.models, o.model, o.cli, o.delegation, o.limits, o.allowed_workspaces, o.session_workspaces, o.server.mcp_endpoint]), () => settingsView(o));
   }
   tick();
 }