vg-coder-cli 2.0.52 → 2.0.53

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "vg-coder-cli",
-  "version": "2.0.52",
+  "version": "2.0.53",
   "description": "🚀 CLI tool to analyze projects, concatenate source files, count tokens, and export HTML with syntax highlighting and copy functionality",
   "main": "src/index.js",
   "bin": {

package/src/server/task-queue.js

@@ -415,7 +415,10 @@ class TaskQueue {
       return;
     }
     task.status = 'done';
-    task.result = { chatId: payload?.chatId || null };
+    task.result = {
+      chatId: payload?.chatId || null,
+      actualModel: payload?.actualModel || null,
+    };
     task.timing.finishedAt = Date.now();
     task.timing.durationMs = task.timing.finishedAt - (task.timing.startedAt || task.timing.createdAt);
     await store.writeResult(task.workingDir, task.id, payload?.markdown || '');

package/src/server/views/js/features/task-worker.js

@@ -198,6 +198,39 @@ function throwIfRateLimited(stage = '') {
     }
 }
 
+// Đọc model ID thật từ AI Studio sidebar — bắt được redirect AI Studio làm khi
+// account thiếu access tới preview model (vd request gemini-3-pro-preview, run
+// thật trên gemini-3-flash-preview). URL param không reliable vì AI Studio
+// không update URL khi fallback. Sidebar `<ms-model-selector>` mới là source-
+// of-truth.
+//
+// Strategy: thử đọc DOM sẵn trước (zero side-effect). Nếu panel chưa render —
+// click toggle "Run settings" để mở, đọc, rồi click lại để khôi phục state ban
+// đầu. Không throw — fail thì return null, caller chấp nhận `actualModel: null`.
+async function readActualModel() {
+    try {
+        // Fast path: panel đã mở (worker tab đã chạy task → AI Studio thường giữ panel open)
+        let selector = document.querySelector('ms-model-selector');
+        if (selector) {
+            const name = selector.querySelector('[data-test-id="model-name"]')?.innerText?.trim();
+            if (name) return name;
+        }
+        // Slow path: toggle panel mở. Lưu state để restore sau khi đọc xong.
+        const toggle = document.querySelector('[aria-label="Toggle run settings panel"]');
+        if (!toggle) return null;
+        toggle.click();
+        await new Promise(r => setTimeout(r, 800));
+        selector = document.querySelector('ms-model-selector');
+        const name = selector?.querySelector('[data-test-id="model-name"]')?.innerText?.trim() || null;
+        // Đóng panel lại để không thay đổi UX (panel mở/đóng visible qua noVNC).
+        toggle.click();
+        return name;
+    } catch (err) {
+        console.warn('[TaskWorker] readActualModel failed:', err?.message);
+        return null;
+    }
+}
+
 async function handleTaskExecute(payload) {
     const { taskId, prompt, files = [] } = payload || {};
     if (!taskId) return;
@@ -251,6 +284,7 @@ async function handleTaskExecute(payload) {
 
         const markdown = await window.AIChat.copyLastTurnAsMarkdown();
         const chatId = window.AIChat.getChatIdFromUrl?.() || null;
+        const actualModel = await readActualModel();
 
         if (cancelFlags.has(taskId)) {
             cancelFlags.delete(taskId);
@@ -258,8 +292,8 @@ async function handleTaskExecute(payload) {
             return;
         }
 
-        socket.emit('task:complete', { taskId, markdown, chatId });
-        console.log(`[TaskWorker] Completed ${taskId}`);
+        socket.emit('task:complete', { taskId, markdown, chatId, actualModel });
+        console.log(`[TaskWorker] Completed ${taskId} (model=${actualModel || 'unknown'})`);
     } catch (err) {
         console.error(`[TaskWorker] Failed ${taskId}:`, err);
         socket.emit('task:failed', {