yiyan-browser-agent 1.7.3 → 1.7.5

package/package.json

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

package/src/browser.js

@@ -411,11 +411,17 @@ class YiyanBrowser {
           if (node.nodeType !== Node.ELEMENT_NODE) return;
           const tag = node.tagName.toLowerCase();
 
+          // 排除思考过程区域
+          const cls = node.className || '';
+          if (cls.includes('container__SPpahQHm') || cls.includes('thinking') || cls.includes('Thinking')) {
+            return;  // 跳过思考区域
+          }
+
           if (tag === 'pre') {
             const codeEl = node.querySelector('code');
             if (codeEl) {
-              const cls  = codeEl.className || '';
-              const lang = (cls.match(/language-(\S+)/) || [])[1] || '';
+              const cls2  = codeEl.className || '';
+              const lang = (cls2.match(/language-(\S+)/) || [])[1] || '';
               const body = codeEl.textContent || '';
               result += '\n```' + lang + '\n' + body + '\n```\n';
             } else {
@@ -444,7 +450,7 @@ class YiyanBrowser {
         return result.trim();
       }
 
-      // ── 优先：仅从 answer_text_id 提取内容 ──
+      // ── 优先：仅从 answer_text_id 提取内容（排除思考区域）──
       const answerEl = document.querySelector('#answer_text_id');
       if (answerEl) {
         const t = getFullText(answerEl);
@@ -528,7 +534,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*="停止"]',
@@ -545,7 +570,7 @@ class YiyanBrowser {
         }
       }
 
-      // ── 2. 检测加载动画/typing指示器 ──
+      // ── 3. 检测加载动画/typing指示器 ──
       const loaderSelectors = [
         '[class*="typing"]',
         '[class*="loading"]',
@@ -567,11 +592,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]',
@@ -605,7 +629,26 @@ class YiyanBrowser {
   _cleanText(text) {
     if (!text) return '';
 
-    // Remove everything before "准备输出结果" (Yiyan's thinking process)
+    // Remove thinking process markers (Yiyan's 思考过程)
+    const thinkingMarkers = [
+      '正在思考中',
+      '正在思考',
+      '思考过程',
+      '我来',
+      '我需要',
+      '根据搜索结果',
+      '参考',
+      'picaole需要',
+    ];
+
+    // Remove lines that contain thinking markers
+    for (const marker of thinkingMarkers) {
+      // 如果整行包含思考标记，移除该行
+      const lines = text.split('\n');
+      text = lines.filter(line => !line.includes(marker)).join('\n');
+    }
+
+    // Remove everything before "准备输出结果" (Yiyan's thinking process end marker)
     const outputMarker = '准备输出结果';
     const markerIndex = text.indexOf(outputMarker);
     if (markerIndex !== -1) {
@@ -624,6 +667,9 @@ class YiyanBrowser {
 
     return text
       // Strip any remaining AI thinking blocks
+      .replace(/阶段性总结[\s\S]*?(?=```|$)/gi, '')
+      .replace(/根据搜索结果[\s\S]*?(?=```|$)/gi, '')
+      // Strip think tags
       .replace(/<think>[\s\S]*?<\/think>\n?/gi, '')
       // Strip copy-code button artifacts
       .replace(/^\d+(?:Copy|Run|Insert|Edit)\b.*$/gm, '')