yiyan-browser-agent 1.7.2 → 1.7.4

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "yiyan-browser-agent",
-  "version": "1.7.2",
+  "version": "1.7.4",
   "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,11 +450,15 @@ class YiyanBrowser {
         return result.trim();
       }
 
-      const directSelectors = [
-        // Yiyan specific: answer text container (highest priority)
-        '#answer_text_id',
-        '[id="answer_text_id"]',
-        // Generic selectors
+      // ── 优先：仅从 answer_text_id 提取内容（排除思考区域）──
+      const answerEl = document.querySelector('#answer_text_id');
+      if (answerEl) {
+        const t = getFullText(answerEl);
+        if (t.length > 0) return t;
+      }
+
+      // ── 备用：其他选择器（仅当 answer_text_id 不存在时）──
+      const backupSelectors = [
         '[class*="answer"]',
         '[class*="response"]',
         '[class*="message"][class*="content"]',
@@ -462,12 +472,12 @@ class YiyanBrowser {
         '[class*="message-content"]:last-child',
       ];
 
-      for (const sel of directSelectors) {
+      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 > 10) return t;
+            if (t.length > 5) return t;
           }
         } catch {}
       }
@@ -601,7 +611,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) {
@@ -620,6 +649,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, '')