yiyan-browser-agent 1.6.3 → 1.6.5

package/package.json

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

package/src/browser.js

@@ -293,9 +293,10 @@ class YiyanBrowser {
 
     if (!appeared) logger.warn('Response may have been delayed — continuing to wait...');
 
-    // Phase 2: wait for text to stabilise
+    // Phase 2: wait for text to stabilise (with multiple checks)
     let lastText    = '';
     let stableStart = null;
+    let stableCount = 0;      // 连续稳定次数计数
     let dotCount    = 0;
 
     while (Date.now() - start < timeout) {
@@ -304,11 +305,22 @@ class YiyanBrowser {
       if (text !== lastText) {
         lastText    = text;
         stableStart = null;
+        stableCount = 0;  // 内容变化，重置计数
       } else if (text.length > 0) {
         if (!stableStart) stableStart = Date.now();
         else if (Date.now() - stableStart >= stableDelay) {
-          if (!await this._isGenerating()) break;
-          stableStart = null;
+          // 增加稳定性计数
+          stableCount++;
+
+          // 需要连续稳定检测 + _isGenerating 确认
+          if (stableCount >= 2) {  // 连续 2 次稳定检测
+            if (!await this._isGenerating()) break;
+            stableCount = 0;  // _isGenerating 说还在生成，重置
+            stableStart = null;
+          } else {
+            // 第一次稳定，继续等待下一次检测
+            stableStart = null;
+          }
         }
       }
 
@@ -479,11 +491,14 @@ class YiyanBrowser {
 
   async _isGenerating() {
     return await this.page.evaluate(() => {
+      // ── 1. 检测停止按钮/生成状态 UI ──
       const stopSelectors = [
         'button[aria-label*="Stop" i]',
+        'button[aria-label*="停止"]',
         '[class*="stop-gen"]',
         '[class*="stopGen"]',
         '[class*="generating"]',
+        '[class*=" Generating"]',
       ];
       for (const sel of stopSelectors) {
         const el = document.querySelector(sel);
@@ -493,6 +508,7 @@ class YiyanBrowser {
         }
       }
 
+      // ── 2. 检测加载动画/typing指示器 ──
       const loaderSelectors = [
         '[class*="typing"]',
         '[class*="loading"]',
@@ -500,8 +516,11 @@ class YiyanBrowser {
         '[class*="blink"]',
         '[class*="cursor"]',
         '[class*="pulsing"]',
+        '[class*="thinking"]',
         'svg[class*="loading"]',
         'svg[class*="spinner"]',
+        '.loading-indicator',
+        '.generating-indicator',
       ];
       for (const sel of loaderSelectors) {
         const el = document.querySelector(sel);
@@ -511,6 +530,35 @@ class YiyanBrowser {
         }
       }
 
+      // ── 3. 检测是否缺少完成标记元素 ──
+      // dialogCardBottom 是 Yiyan 响应完成后的底部操作区域
+      const completionMarkers = [
+        '[class*="dialogCardBottom"]',        // Yiyan 对话卡片底部
+        '.dialogCardBottom__qoXjps3z',        // Yiyan 特定 class
+        '[class*="copy-btn"]',
+        '[class*="copyBtn"]',
+        '[aria-label*="Copy" i]',
+        '[aria-label*="复制"]',
+        '[class*="regenerate"]',
+        '[class*="retry"]',
+        '[class*="action-btn"]',
+      ];
+      let hasCompletionMarker = false;
+      for (const sel of completionMarkers) {
+        if (document.querySelector(sel)) {
+          hasCompletionMarker = true;
+          break;
+        }
+      }
+
+      // 如果有响应内容但没有完成标记 → 还在生成
+      const answerArea = document.querySelector('[class*="answer"], [class*="response"], [class*="markdown"]');
+      if (answerArea && answerArea.innerText && answerArea.innerText.length > 50) {
+        if (!hasCompletionMarker) {
+          return true;
+        }
+      }
+
       return false;
     });
   }