yiyan-browser-agent 1.6.3 → 1.6.4

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "yiyan-browser-agent",
-  "version": "1.6.3",
+  "version": "1.6.4",
   "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"]',  // Yiyan 可能的 class
       ];
       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,12 @@ class YiyanBrowser {
         '[class*="blink"]',
         '[class*="cursor"]',
         '[class*="pulsing"]',
+        '[class*="thinking"]',
         'svg[class*="loading"]',
         'svg[class*="spinner"]',
+        // Yiyan 特有的加载指示
+        '.loading-indicator',
+        '.generating-indicator',
       ];
       for (const sel of loaderSelectors) {
         const el = document.querySelector(sel);
@@ -511,6 +531,58 @@ class YiyanBrowser {
         }
       }
 
+      // ── 3. 检测输入框状态（生成时通常禁用）──
+      const inputSelectors = [
+        '[contenteditable="true"]',
+        'textarea',
+        '[role="textbox"]',
+        '.editable',
+      ];
+      for (const sel of inputSelectors) {
+        const input = document.querySelector(sel);
+        if (input) {
+          // 检查是否禁用
+          if (input.disabled) return true;
+          if (input.getAttribute('aria-disabled') === 'true') return true;
+          if (input.getAttribute('disabled') !== null) return true;
+
+          // 检查是否只读（有时会用 readonly 表示生成中）
+          if (input.readOnly || input.getAttribute('readonly') !== null) {
+            // 但要排除本来就是 readonly 的输入框
+            // 这里暂时不返回 true，需要更多测试
+          }
+        }
+      }
+
+      // ── 4. 检测是否缺少完成标记元素 ──
+      // 如果没有复制按钮/重新生成按钮，可能还在生成
+      const completionMarkers = [
+        '[class*="copy-btn"]',
+        '[class*="copyBtn"]',
+        '[aria-label*="Copy" i]',
+        '[aria-label*="复制"]',
+        '[class*="regenerate"]',
+        '[class*="retry"]',
+        '[class*="action-btn"]',  // Yiyan 的操作按钮区域
+      ];
+      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;
     });
   }