yiyan-browser-agent 1.8.0 → 1.8.2

package/package.json

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

package/src/browser.js

@@ -297,12 +297,11 @@ class YiyanBrowser {
     if (!appeared) logger.warn('Response may have been delayed — continuing to wait...');
 
     // Phase 2: wait for completion
-    // 条件: dialogCardBottom 出现 + answer稳定 + 思考区域稳定
+    // 条件: 思考区域和答案区域都稳定不变
     let lastText            = '';
     let lastThinkingText    = '';
     let stableCount         = 0;
-    let thinkingStableCount = 0;
-    let lastCheckTime       = Date.now();
+    let lastStableTime      = Date.now();
     let dotCount            = 0;
     let hasCompletionMarker = false;
 
@@ -321,28 +320,24 @@ class YiyanBrowser {
         return { text, exists: true };
       });
 
-      // ── 思考区域稳定性检测 ──
-      if (thinkingInfo.text === lastThinkingText && thinkingInfo.text.length > 5) {
-        if (Date.now() - lastCheckTime >= stableDelay) {
-          thinkingStableCount++;
-          logger.dim(`Thinking stable: ${thinkingStableCount}/2 (${thinkingInfo.text.length} chars)`);
-        }
-      } else if (thinkingInfo.text !== lastThinkingText) {
-        lastThinkingText = thinkingInfo.text;
-        thinkingStableCount = 0;
-        lastCheckTime = Date.now();
-      }
+      // ── 统一稳定性检测：两者都不变才算稳定 ──
+      const thinkingStable = thinkingInfo.text === lastThinkingText;
+      const answerStable   = text === lastText;
+      const hasContent     = thinkingInfo.text.length > 5 && text.length > 50;
 
-      // ── answer 稳定性检测 ──
-      if (text === lastText && text.length > 50) {
-        if (Date.now() - lastCheckTime >= stableDelay) {
+      if (thinkingStable && answerStable && hasContent) {
+        // 两者都稳定且有内容，检查稳定时间
+        if (Date.now() - lastStableTime >= stableDelay) {
           stableCount++;
-          logger.dim(`Answer stable: ${stableCount}/2 (${text.length} chars)`);
+          lastStableTime = Date.now();
+          logger.dim(`Both stable: ${stableCount}/2 (thinking: ${thinkingInfo.text.length} chars, answer: ${text.length} chars)`);
         }
-      } else if (text !== lastText) {
+      } else {
+        // 任一变化，重置
+        lastThinkingText = thinkingInfo.text;
         lastText = text;
         stableCount = 0;
-        lastCheckTime = Date.now();
+        lastStableTime = Date.now();
       }
 
       // ── 完成标记检测 ──
@@ -367,9 +362,14 @@ class YiyanBrowser {
         hasCompletionMarker = true;
       }
 
-      // ── 完成判断：标记出现 + answer稳定 + 思考区域稳定 ──
-      if (hasCompletionMarker && stableCount >= 2 && thinkingStableCount >= 2) {
-        // 思考区域和答案都已稳定，检查 _isGenerating 作为最终确认
+      // ── 完成判断 ──
+      // 条件1: 标记出现 + 两者都稳定 2 次
+      // 条件2(备用): 两者都稳定 3 次（即使没有完成标记）
+      const condition1 = hasCompletionMarker && stableCount >= 2;
+      const condition2 = stableCount >= 3;
+
+      if (condition1 || condition2) {
+        // 检查 _isGenerating 作为最终确认
         if (!await this._isGenerating()) {
           await this.page.waitForTimeout(500);
           logger.clearLine();
@@ -378,8 +378,7 @@ class YiyanBrowser {
         }
         // _isGenerating 说还在生成，重置计数继续等待
         stableCount = 0;
-        thinkingStableCount = 0;
-        lastCheckTime = Date.now();
+        lastStableTime = Date.now();
       }
 
       // Progress indicator
@@ -488,26 +487,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 +504,7 @@ class YiyanBrowser {
         }
       }
 
-      // ── 3. 检测加载动画/typing指示器 ──
+      // ── 2. 检测加载动画/typing指示器 ──
       const loaderSelectors = [
         '[class*="typing"]',
         '[class*="loading"]',
@@ -546,7 +526,7 @@ class YiyanBrowser {
         }
       }
 
-      // ── 4. 检测是否缺少完成标记元素 ──
+      // ── 3. 检测是否缺少完成标记元素 ──
       const completionMarkers = [
         '[class*="dialogCardBottom"]',
         '.dialogCardBottom__qoXjps3z',