vg-coder-cli 2.0.61 → 2.0.62

package/package.json

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