vg-coder-cli 2.0.60 → 2.0.62

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "vg-coder-cli",
-  "version": "2.0.60",
+  "version": "2.0.62",
   "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/views/js/features/task-worker.js

@@ -133,12 +133,15 @@ async function fetchAsFile(url, name, mime) {
 async function waitForNewAssistantTurn(baselineCount, timeoutMs = 90_000) {
     const start = Date.now();
     while (Date.now() - start < timeoutMs) {
-        // Surface rate-limit / quota errors early — no point waiting for a turn
-        // that will never appear.
-        const rl = detectRateLimit();
-        if (rl) { const e = new Error(rl.message); e.code = rl.code; throw e; }
-
+        // Chỉ surface rate-limit/quota khi turn count đã vượt baseline. Trước
+        // đó turn cuối vẫn là turn cũ (có thể là error từ session restore) —
+        // detectRateLimit match phải khớp với turn MỚI.
         const turns = document.querySelectorAll('ms-chat-turn');
+        if (turns.length > baselineCount) {
+            // Turn mới đã render — bây giờ mới check (toast vẫn check global ok)
+            const rl = detectRateLimit();
+            if (rl) { const e = new Error(rl.message); e.code = rl.code; throw e; }
+        }
         if (turns.length >= baselineCount + 2) {
             const last = turns[turns.length - 1];
             const role = last.querySelector('[data-turn-role]')?.getAttribute('data-turn-role');
@@ -164,8 +167,15 @@ function detectRateLimit() {
         if (/failed to generate/i.test(t))   return { code: 'generation_failed', message: t.slice(0, 500) };
     }
 
-    // Inline model-error inside the last assistant turn
-    const modelErr = document.querySelector('.model-error');
+    // Inline model-error — CHỈ check trong turn CUỐI (response cho prompt vừa
+    // submit). AI Studio restore chat session từ cookies → turn cũ với
+    // .model-error rate-limit có thể vẫn trong DOM dù limit đã hết.
+    // Global querySelector('.model-error') match turn cũ → mark cooldown
+    // infinite loop. Scope theo last ms-chat-turn để chỉ trigger trên response
+    // mới.
+    const turns = document.querySelectorAll('ms-chat-turn');
+    const lastTurn = turns[turns.length - 1];
+    const modelErr = lastTurn?.querySelector('.model-error');
     if (modelErr) {
         const text = (modelErr.textContent || '').trim();
         if (/rate limit/i.test(text))       return { code: 'rate_limit_exceeded', message: text };
@@ -199,7 +209,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 +266,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`);