yiyan-browser-agent 1.6.5 → 1.6.6

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "yiyan-browser-agent",
-  "version": "1.6.5",
+  "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);
     }