yiyan-browser-agent 1.7.6 → 1.7.8

package/package.json

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

package/src/browser.js

@@ -344,29 +344,30 @@ class YiyanBrowser {
         hasCompletionMarker = true;
       }
 
-      // ── 完成判断：标记出现 + 连续两次稳定 + _isGenerating 确认 + 思考区域检查 ──
+      // ── 完成判断：标记出现 + 连续两次稳定 + 思考区域已消失 ──
       if (hasCompletionMarker && stableCount >= 2) {
-        // 检查思考区域是否有内容
-        const hasThinkingContent = await this.page.evaluate(() => {
+        // 检查思考区域是否存在且有内容
+        const thinkingStatus = 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 (!thinkingEl) return { exists: false, hasContent: false };
+
+          const s = window.getComputedStyle(thinkingEl);
+          const isVisible = s.display !== 'none' && s.visibility !== 'hidden';
+          const text = thinkingEl.innerText || '';
+          const hasContent = text.length > 5;
+
+          return { exists: true, isVisible, hasContent };
         });
 
-        if (hasThinkingContent) {
-          // 思考区域有内容，还没完成
+        // 思考区域存在且可见且有内容 → 还在思考，继续等待
+        if (thinkingStatus.exists && thinkingStatus.isVisible && thinkingStatus.hasContent) {
           stableCount = 0;
           lastCheckTime = Date.now();
           logger.dim('Thinking in progress, continue waiting...');
           continue;
         }
 
+        // 思考区域已消失/不可见/无内容 → 检查 _isGenerating
         if (!await this._isGenerating()) {
           await this.page.waitForTimeout(500);
           logger.clearLine();
@@ -471,84 +472,13 @@ class YiyanBrowser {
         return result.trim();
       }
 
-      // ── 优先：仅从 answer_text_id 提取内容（排除思考区域）──
+      // ── 仅从 answer_text_id 提取内容，不使用任何 fallback ──
       const answerEl = document.querySelector('#answer_text_id');
       if (answerEl) {
-        const t = getFullText(answerEl);
-        if (t.length > 0) return t;
+        return getFullText(answerEl);
       }
 
-      // ── 备用：其他选择器（仅当 answer_text_id 不存在时）──
-      const backupSelectors = [
-        '[class*="answer"]',
-        '[class*="response"]',
-        '[class*="message"][class*="content"]',
-        '.ds-markdown',
-        '[class*="assistant"] [class*="markdown"]',
-        '[class*="assistant"] [class*="content"]',
-        '[data-role="assistant"] [class*="content"]',
-        '[class*="ai-message"] [class*="content"]',
-        '[class*="bot-message"] [class*="content"]',
-        '[class*="response-content"]',
-        '[class*="message-content"]:last-child',
-      ];
-
-      for (const sel of backupSelectors) {
-        try {
-          const els = document.querySelectorAll(sel);
-          if (els.length > 0) {
-            const t = getFullText(els[els.length - 1]);
-            if (t.length > 5) return t;
-          }
-        } catch {}
-      }
-
-      try {
-        const markdownEls = document.querySelectorAll(
-          '[class*="markdown"], [class*="prose"], [class*="rendered"], [class*="content"]'
-        );
-        if (markdownEls.length > 0) {
-          const t = getFullText(markdownEls[markdownEls.length - 1]);
-          if (t.length > 10) return t;
-        }
-      } catch {}
-
-      try {
-        const allBlocks = Array.from(
-          document.querySelectorAll('[class*="message"], [class*="chat-item"], [class*="turn"], [class*="answer"], [class*="content"]')
-        );
-        const candidates = allBlocks.filter(el => {
-          const cls = el.className || '';
-          const id = el.id || '';
-          return (
-            !cls.toLowerCase().includes('input') &&
-            !cls.toLowerCase().includes('user') &&
-            !cls.toLowerCase().includes('editable') &&
-            !id.toLowerCase().includes('input') &&
-            !el.querySelector('textarea, input[type="text"], [contenteditable="true"]') &&
-            (el.innerText || '').length > 20
-          );
-        });
-
-        if (candidates.length > 0) {
-          return getFullText(candidates[candidates.length - 1]);
-        }
-      } catch {}
-
-      try {
-        const allDivs = Array.from(document.querySelectorAll('div, section'));
-        const textBlocks = allDivs.filter(el => {
-          const text = el.innerText || '';
-          const cls = el.className || '';
-          if (cls.includes('input') || cls.includes('editable') || cls.includes('user')) return false;
-          return text.length > 50 && !el.querySelector('textarea, [contenteditable]');
-        });
-        if (textBlocks.length > 0) {
-          textBlocks.sort((a, b) => (b.innerText || '').length - (a.innerText || '').length);
-          return getFullText(textBlocks[0]);
-        }
-      } catch {}
-
+      // answer_text_id 不存在，返回空字符串
       return '';
     });
   }