yiyan-browser-agent 1.6.5 → 1.6.7

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "yiyan-browser-agent",
-  "version": "1.6.5",
+  "version": "1.6.7",
   "description": "AI coding agent powered by Yiyan (文心一言) via browser automation — no API key needed",
   "main": "src/index.js",
   "bin": {

package/src/browser.js

@@ -281,54 +281,87 @@ class YiyanBrowser {
     const stableDelay = config.STABLE_DELAY;
     const start       = Date.now();
 
-    // Phase 1: wait for a new message to appear
+    // Phase 1: wait for response container to appear
     const initialCount = await this._getMessageCount();
-    let   appeared     = false;
+    let appeared       = false;
 
     for (let i = 0; i < 40; i++) {
       const count = await this._getMessageCount();
-      if (count > initialCount) { appeared = true; break; }
+      if (count > initialCount) {
+        appeared = true;
+        break;
+      }
       await this.page.waitForTimeout(200);
     }
 
     if (!appeared) logger.warn('Response may have been delayed — continuing to wait...');
 
-    // Phase 2: wait for text to stabilise (with multiple checks)
-    let lastText    = '';
-    let stableStart = null;
-    let stableCount = 0;      // 连续稳定次数计数
-    let dotCount    = 0;
+    // Phase 2: wait for completion
+    // 主要：dialogCardBottom 出现
+    // 备用：内容稳定 + _isGenerating() 确认
+    let lastText      = '';
+    let stableStart   = null;
+    let stableCount   = 0;
+    let dotCount      = 0;
 
     while (Date.now() - start < timeout) {
       const text = await this._extractLastMessage();
 
+      // ── 稳定性检测（备用退出路径）──
       if (text !== lastText) {
         lastText    = text;
         stableStart = null;
-        stableCount = 0;  // 内容变化，重置计数
-      } else if (text.length > 0) {
+        stableCount = 0;
+      } else if (text.length > 50) {  // 内容足够长才进入稳定检测
         if (!stableStart) stableStart = Date.now();
         else if (Date.now() - stableStart >= stableDelay) {
-          // 增加稳定性计数
           stableCount++;
 
-          // 需要连续稳定检测 + _isGenerating 确认
-          if (stableCount >= 2) {  // 连续 2 次稳定检测
-            if (!await this._isGenerating()) break;
-            stableCount = 0;  // _isGenerating 说还在生成，重置
+          // 连续稳定检测 + _isGenerating 确认
+          if (stableCount >= 2) {
+            if (!await this._isGenerating()) {
+              logger.clearLine();
+              logger.dim('Stable completion detected');
+              break;
+            }
+            stableCount = 0;
             stableStart = null;
           } else {
-            // 第一次稳定，继续等待下一次检测
             stableStart = null;
           }
         }
       }
 
+      // ── 完成标记检测（主要退出路径）──
+      const hasCompletion = await this.page.evaluate(() => {
+        const selectors = [
+          '.dialogCardBottom__qoXjps3z',
+          '[class*="dialogCardBottom"]',
+          '[class*="dialog-bottom"]',
+          '[class*="response-footer"]',
+        ];
+        for (const sel of selectors) {
+          const el = document.querySelector(sel);
+          if (el) {
+            const s = window.getComputedStyle(el);
+            if (s.display !== 'none' && s.visibility !== 'hidden') return true;
+          }
+        }
+        return false;
+      });
+
+      if (hasCompletion) {
+        await this.page.waitForTimeout(500);
+        logger.clearLine();
+        logger.dim('Completion marker detected');
+        break;
+      }
+
       // Progress indicator
       dotCount = (dotCount + 1) % 4;
       logger.thinking(`Receiving response${'.'.repeat(dotCount)}  (${text.length} chars)`);
 
-      // 等待 200ms 给 DOM 时间更新，避免高频轮询读取不完整内容
+      // 等待 200ms 给 DOM 时间更新
       await this.page.waitForTimeout(200);
     }