yiyan-browser-agent 1.7.4 → 1.7.6

package/package.json

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

package/src/browser.js

@@ -344,8 +344,29 @@ class YiyanBrowser {
         hasCompletionMarker = true;
       }
 
-      // ── 完成判断：标记出现 + 连续两次稳定 + _isGenerating 确认 ──
+      // ── 完成判断：标记出现 + 连续两次稳定 + _isGenerating 确认 + 思考区域检查 ──
       if (hasCompletionMarker && stableCount >= 2) {
+        // 检查思考区域是否有内容
+        const hasThinkingContent = await this.page.evaluate(() => {
+          const thinkingEl = document.querySelector('.container__SPpahQHm, [class*="container__SPpah"]');
+          if (thinkingEl) {
+            const s = window.getComputedStyle(thinkingEl);
+            if (s.display !== 'none' && s.visibility !== 'hidden') {
+              const text = thinkingEl.innerText || '';
+              return text.length > 5;  // 思考区域有内容
+            }
+          }
+          return false;
+        });
+
+        if (hasThinkingContent) {
+          // 思考区域有内容，还没完成
+          stableCount = 0;
+          lastCheckTime = Date.now();
+          logger.dim('Thinking in progress, continue waiting...');
+          continue;
+        }
+
         if (!await this._isGenerating()) {
           await this.page.waitForTimeout(500);
           logger.clearLine();
@@ -534,7 +555,26 @@ class YiyanBrowser {
 
   async _isGenerating() {
     return await this.page.evaluate(() => {
-      // ── 1. 检测停止按钮/生成状态 UI ──
+      // ── 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 ──
       const stopSelectors = [
         'button[aria-label*="Stop" i]',
         'button[aria-label*="停止"]',
@@ -551,7 +591,7 @@ class YiyanBrowser {
         }
       }
 
-      // ── 2. 检测加载动画/typing指示器 ──
+      // ── 3. 检测加载动画/typing指示器 ──
       const loaderSelectors = [
         '[class*="typing"]',
         '[class*="loading"]',
@@ -573,11 +613,10 @@ class YiyanBrowser {
         }
       }
 
-      // ── 3. 检测是否缺少完成标记元素 ──
-      // dialogCardBottom 是 Yiyan 响应完成后的底部操作区域
+      // ── 4. 检测是否缺少完成标记元素 ──
       const completionMarkers = [
-        '[class*="dialogCardBottom"]',        // Yiyan 对话卡片底部
-        '.dialogCardBottom__qoXjps3z',        // Yiyan 特定 class
+        '[class*="dialogCardBottom"]',
+        '.dialogCardBottom__qoXjps3z',
         '[class*="copy-btn"]',
         '[class*="copyBtn"]',
         '[aria-label*="Copy" i]',