yiyan-browser-agent 1.8.0 → 1.8.1

package/package.json

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

package/src/browser.js

@@ -367,9 +367,14 @@ class YiyanBrowser {
         hasCompletionMarker = true;
       }
 
-      // ── 完成判断：标记出现 + answer稳定 + 思考区域稳定 ──
-      if (hasCompletionMarker && stableCount >= 2 && thinkingStableCount >= 2) {
-        // 思考区域和答案都已稳定，检查 _isGenerating 作为最终确认
+      // ── 完成判断 ──
+      // 条件1: 标记出现 + answer稳定 + 思考区域稳定
+      // 条件2(备用): 两者都稳定达到 4 次（即使没有完成标记）
+      const condition1 = hasCompletionMarker && stableCount >= 2 && thinkingStableCount >= 2;
+      const condition2 = stableCount >= 4 && thinkingStableCount >= 4;
+
+      if (condition1 || condition2) {
+        // 检查 _isGenerating 作为最终确认
         if (!await this._isGenerating()) {
           await this.page.waitForTimeout(500);
           logger.clearLine();
@@ -488,26 +493,7 @@ class YiyanBrowser {
 
   async _isGenerating() {
     return await this.page.evaluate(() => {
-      // ── 1. 检测思考过程区域（最重要）──
-      const thinkingSelectors = [
-        '.container__SPpahQHm',        // Yiyan 思考过程区域
-        '[class*="container__SPpah"]',
-        '[class*="thinking"]',
-        '[class*="Thinking"]',
-        '[class*="thought"]',
-      ];
-      for (const sel of thinkingSelectors) {
-        const el = document.querySelector(sel);
-        if (el) {
-          const s = window.getComputedStyle(el);
-          // 元素可见且有内容
-          if (s.display !== 'none' && s.visibility !== 'hidden' && el.innerText && el.innerText.length > 5) {
-            return true;  // 思考区域有内容，还在生成
-          }
-        }
-      }
-
-      // ── 2. 检测停止按钮/生成状态 UI ──
+      // ── 1. 检测停止按钮/生成状态 UI ──
       const stopSelectors = [
         'button[aria-label*="Stop" i]',
         'button[aria-label*="停止"]',
@@ -524,7 +510,7 @@ class YiyanBrowser {
         }
       }
 
-      // ── 3. 检测加载动画/typing指示器 ──
+      // ── 2. 检测加载动画/typing指示器 ──
       const loaderSelectors = [
         '[class*="typing"]',
         '[class*="loading"]',
@@ -546,7 +532,7 @@ class YiyanBrowser {
         }
       }
 
-      // ── 4. 检测是否缺少完成标记元素 ──
+      // ── 3. 检测是否缺少完成标记元素 ──
       const completionMarkers = [
         '[class*="dialogCardBottom"]',
         '.dialogCardBottom__qoXjps3z',