yiyan-browser-agent 1.6.4 → 1.6.6

package/package.json

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

package/src/browser.js

@@ -278,57 +278,62 @@ class YiyanBrowser {
    */
   async waitForResponse() {
     const timeout     = config.RESPONSE_TIMEOUT;
-    const stableDelay = config.STABLE_DELAY;
     const start       = Date.now();
 
-    // Phase 1: wait for a new message to appear
+    // Phase 1: wait for response container to appear
     const initialCount = await this._getMessageCount();
-    let   appeared     = false;
+    let appeared       = false;
 
     for (let i = 0; i < 40; i++) {
       const count = await this._getMessageCount();
-      if (count > initialCount) { appeared = true; break; }
+      if (count > initialCount) {
+        appeared = true;
+        break;
+      }
       await this.page.waitForTimeout(200);
     }
 
     if (!appeared) logger.warn('Response may have been delayed — continuing to wait...');
 
-    // Phase 2: wait for text to stabilise (with multiple checks)
-    let lastText    = '';
-    let stableStart = null;
-    let stableCount = 0;      // 连续稳定次数计数
+    // Phase 2: wait for completion marker (dialogCardBottom) to appear
     let dotCount    = 0;
+    let lastLength  = 0;
 
     while (Date.now() - start < timeout) {
       const text = await this._extractLastMessage();
+      lastLength = text.length;
 
-      if (text !== lastText) {
-        lastText    = text;
-        stableStart = null;
-        stableCount = 0;  // 内容变化，重置计数
-      } else if (text.length > 0) {
-        if (!stableStart) stableStart = Date.now();
-        else if (Date.now() - stableStart >= stableDelay) {
-          // 增加稳定性计数
-          stableCount++;
-
-          // 需要连续稳定检测 + _isGenerating 确认
-          if (stableCount >= 2) {  // 连续 2 次稳定检测
-            if (!await this._isGenerating()) break;
-            stableCount = 0;  // _isGenerating 说还在生成，重置
-            stableStart = null;
-          } else {
-            // 第一次稳定，继续等待下一次检测
-            stableStart = null;
+      // Progress indicator
+      dotCount = (dotCount + 1) % 4;
+      logger.thinking(`Receiving response${'.'.repeat(dotCount)}  (${lastLength} chars)`);
+
+      // 检测完成标记：dialogCardBottom 出现
+      const hasCompletion = await this.page.evaluate(() => {
+        const selectors = [
+          '.dialogCardBottom__qoXjps3z',
+          '[class*="dialogCardBottom"]',
+          '[class*="dialog-bottom"]',
+          '[class*="response-footer"]',
+        ];
+        for (const sel of selectors) {
+          const el = document.querySelector(sel);
+          if (el) {
+            const s = window.getComputedStyle(el);
+            if (s.display !== 'none' && s.visibility !== 'hidden') return true;
           }
         }
-      }
+        return false;
+      });
 
-      // Progress indicator
-      dotCount = (dotCount + 1) % 4;
-      logger.thinking(`Receiving response${'.'.repeat(dotCount)}  (${text.length} chars)`);
+      if (hasCompletion) {
+        // 完成标记出现，等待一小段时间确保内容完整
+        await this.page.waitForTimeout(500);
+        logger.clearLine();
+        logger.dim('Completion marker detected, finishing...');
+        break;
+      }
 
-      // 等待 200ms 给 DOM 时间更新，避免高频轮询读取不完整内容
+      // 等待 200ms 给 DOM 时间更新
       await this.page.waitForTimeout(200);
     }
 
@@ -498,7 +503,7 @@ class YiyanBrowser {
         '[class*="stop-gen"]',
         '[class*="stopGen"]',
         '[class*="generating"]',
-        '[class*=" Generating"]',  // Yiyan 可能的 class
+        '[class*=" Generating"]',
       ];
       for (const sel of stopSelectors) {
         const el = document.querySelector(sel);
@@ -519,7 +524,6 @@ class YiyanBrowser {
         '[class*="thinking"]',
         'svg[class*="loading"]',
         'svg[class*="spinner"]',
-        // Yiyan 特有的加载指示
         '.loading-indicator',
         '.generating-indicator',
       ];
@@ -531,39 +535,18 @@ class YiyanBrowser {
         }
       }
 
-      // ── 3. 检测输入框状态（生成时通常禁用）──
-      const inputSelectors = [
-        '[contenteditable="true"]',
-        'textarea',
-        '[role="textbox"]',
-        '.editable',
-      ];
-      for (const sel of inputSelectors) {
-        const input = document.querySelector(sel);
-        if (input) {
-          // 检查是否禁用
-          if (input.disabled) return true;
-          if (input.getAttribute('aria-disabled') === 'true') return true;
-          if (input.getAttribute('disabled') !== null) return true;
-
-          // 检查是否只读（有时会用 readonly 表示生成中）
-          if (input.readOnly || input.getAttribute('readonly') !== null) {
-            // 但要排除本来就是 readonly 的输入框
-            // 这里暂时不返回 true，需要更多测试
-          }
-        }
-      }
-
-      // ── 4. 检测是否缺少完成标记元素 ──
-      // 如果没有复制按钮/重新生成按钮，可能还在生成
+      // ── 3. 检测是否缺少完成标记元素 ──
+      // dialogCardBottom 是 Yiyan 响应完成后的底部操作区域
       const completionMarkers = [
+        '[class*="dialogCardBottom"]',        // Yiyan 对话卡片底部
+        '.dialogCardBottom__qoXjps3z',        // Yiyan 特定 class
         '[class*="copy-btn"]',
         '[class*="copyBtn"]',
         '[aria-label*="Copy" i]',
         '[aria-label*="复制"]',
         '[class*="regenerate"]',
         '[class*="retry"]',
-        '[class*="action-btn"]',  // Yiyan 的操作按钮区域
+        '[class*="action-btn"]',
       ];
       let hasCompletionMarker = false;
       for (const sel of completionMarkers) {
@@ -573,12 +556,10 @@ class YiyanBrowser {
         }
       }
 
-      // 如果有响应内容但没有完成标记 → 可能还在生成
-      // 这个检测放在后面，避免误判空响应
+      // 如果有响应内容但没有完成标记 → 还在生成
       const answerArea = document.querySelector('[class*="answer"], [class*="response"], [class*="markdown"]');
       if (answerArea && answerArea.innerText && answerArea.innerText.length > 50) {
         if (!hasCompletionMarker) {
-          // 有内容但没有完成按钮，可能还在生成
           return true;
         }
       }