vg-coder-cli 2.0.59 → 2.0.61

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "vg-coder-cli",
-  "version": "2.0.59",
+  "version": "2.0.61",
   "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

@@ -163,7 +163,13 @@ class TaskQueue {
       if (wasUnknown && email !== existing.email) {
         // shouldn't happen — kept for clarity (email already assigned above)
       }
-      console.log(chalk.green(`[TaskQueue] Worker re-register: ${socket.id} (${email})`));
+      // Re-establish pin từ URL hiện tại của tab worker. Khắc phục race khi
+      // open-tab handler set _pinnedModelByEmail nhưng launcher email lúc đó
+      // null (chưa scrape) → pin bị mất → recycle reopen với default.
+      if (meta?.pinnedModel && !email.startsWith('unknown:')) {
+        this._pinnedModelByEmail.set(email, meta.pinnedModel);
+      }
+      console.log(chalk.green(`[TaskQueue] Worker re-register: ${socket.id} (${email})${meta?.pinnedModel ? ` pinned=${meta.pinnedModel}` : ''}`));
       setImmediate(() => this._drain());
       return true;
     }
@@ -184,7 +190,11 @@ class TaskQueue {
     if (meta?.chromeId && email && !email.startsWith('unknown:')) {
       this._bindChromeIdToEmail(meta.chromeId, email);
     }
-    console.log(chalk.green(`[TaskQueue] Worker registered: ${socket.id} (${email})`));
+    // Re-establish pin từ URL hiện tại (xem comment ở re-register branch).
+    if (meta?.pinnedModel && !email.startsWith('unknown:')) {
+      this._pinnedModelByEmail.set(email, meta.pinnedModel);
+    }
+    console.log(chalk.green(`[TaskQueue] Worker registered: ${socket.id} (${email})${meta?.pinnedModel ? ` pinned=${meta.pinnedModel}` : ''}`));
     setImmediate(() => this._drain());
     return true;
   }

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

@@ -199,7 +199,7 @@ 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
+// account thiếu access tới preview model (vd request gemini-3.1-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.
@@ -256,6 +256,18 @@ async function handleTaskExecute(payload) {
         }
         await new Promise(r => setTimeout(r, 600)); // settle after navigation
 
+        // 1b. Pre-flight model check: AI Studio có thể fallback sang paid agent
+        // (vd "deep-research-preview-04-2026") khi pin model không có access.
+        // Submit task lên Deep Research → rate limit ngay (paid quota), URL không
+        // có ?model= để recycle pin lại → loop fail. Fail-fast nếu detect agent
+        // không phải gemini.
+        const preFlightModel = await readActualModel();
+        if (preFlightModel && !/^gemini[-.]/i.test(preFlightModel)) {
+            const e = new Error(`Worker tab loaded non-Gemini model: ${preFlightModel}. Pin model unavailable for this account.`);
+            e.code = 'model_unavailable';
+            throw e;
+        }
+
         if (cancelFlags.has(taskId)) {
             cancelFlags.delete(taskId);
             console.log(`[TaskWorker] ${taskId} canceled before send — skipping`);
@@ -347,12 +359,20 @@ function connect() {
         // Register with whatever email we have right now (may be null on cold load).
         const initialEmail = extractEmail();
         const chromeId = (window.vetgo && window.vetgo.chromeId) || null;
+        // pinnedModel: lấy từ URL ?model=X — server dùng để re-establish pin
+        // sau khi worker register với email thật (không lúc open-tab vì lúc đó
+        // launcher có thể chưa scrape email).
+        const pinnedModel = (() => {
+            try { return new URLSearchParams(location.search).get('model') || null; }
+            catch (_) { return null; }
+        })();
         socket.emit('worker:register', {
             domain: location.hostname,
             chatId: window.AIChat?.getChatIdFromUrl?.() || null,
             userAgent: navigator.userAgent,
             email: initialEmail,
-            chromeId
+            chromeId,
+            pinnedModel
         });
         console.log(`[TaskWorker] Initial email: ${initialEmail || '(pending)'}, chromeId: ${chromeId || '(none)'}`);
 
@@ -368,14 +388,19 @@ function connect() {
                 try {
                     const resolved = await resolveEmail(3000, emailAbort?.signal);
                     if (!resolved || !socket.connected) return;
+                    const pm = (() => {
+                        try { return new URLSearchParams(location.search).get('model') || null; }
+                        catch (_) { return null; }
+                    })();
                     socket.emit('worker:register', {
                         domain: location.hostname,
                         chatId: window.AIChat?.getChatIdFromUrl?.() || null,
                         userAgent: navigator.userAgent,
                         email: resolved,
-                        chromeId: (window.vetgo && window.vetgo.chromeId) || null
+                        chromeId: (window.vetgo && window.vetgo.chromeId) || null,
+                        pinnedModel: pm
                     });
-                    console.log(`[TaskWorker] Re-registered with email: ${resolved}`);
+                    console.log(`[TaskWorker] Re-registered with email: ${resolved} (pinnedModel=${pm || 'none'})`);
                 } catch (err) {
                     console.error('[TaskWorker] Email retry failed:', err);
                 }