vg-coder-cli 2.0.59 → 2.0.60

package/package.json

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

@@ -347,12 +347,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 +376,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);
                 }