yiyan-browser-agent 1.6.6 → 1.6.8

package/package.json

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

package/src/browser.js

@@ -278,6 +278,7 @@ class YiyanBrowser {
    */
   async waitForResponse() {
     const timeout     = config.RESPONSE_TIMEOUT;
+    const stableDelay = config.STABLE_DELAY;
     const start       = Date.now();
 
     // Phase 1: wait for response container to appear
@@ -295,20 +296,47 @@ class YiyanBrowser {
 
     if (!appeared) logger.warn('Response may have been delayed — continuing to wait...');
 
-    // Phase 2: wait for completion marker (dialogCardBottom) to appear
-    let dotCount    = 0;
-    let lastLength  = 0;
+    // Phase 2: wait for completion (双重条件都必须满足)
+    // 条件1: dialogCardBottom 出现
+    // 条件2: 内容稳定 + _isGenerating() 确认
+    let lastText      = '';
+    let stableStart   = null;
+    let stableCount   = 0;
+    let dotCount      = 0;
+    let hasCompletionMarker = false;
 
     while (Date.now() - start < timeout) {
       const text = await this._extractLastMessage();
-      lastLength = text.length;
 
-      // Progress indicator
-      dotCount = (dotCount + 1) % 4;
-      logger.thinking(`Receiving response${'.'.repeat(dotCount)}  (${lastLength} chars)`);
+      // ── 稳定性检测 ──
+      if (text !== lastText) {
+        lastText    = text;
+        stableStart = null;
+        stableCount = 0;
+      } else if (text.length > 50) {
+        if (!stableStart) stableStart = Date.now();
+        else if (Date.now() - stableStart >= stableDelay) {
+          stableCount++;
+
+          if (stableCount >= 2) {
+            if (!await this._isGenerating()) {
+              // 稳定性检测通过，检查是否同时有完成标记
+              if (hasCompletionMarker) {
+                logger.clearLine();
+                logger.dim('Completion confirmed (stable + marker)');
+                break;
+              }
+            }
+            stableCount = 0;
+            stableStart = null;
+          } else {
+            stableStart = null;
+          }
+        }
+      }
 
-      // 检测完成标记：dialogCardBottom 出现
-      const hasCompletion = await this.page.evaluate(() => {
+      // ── 完成标记检测 ──
+      const detected = await this.page.evaluate(() => {
         const selectors = [
           '.dialogCardBottom__qoXjps3z',
           '[class*="dialogCardBottom"]',
@@ -325,14 +353,21 @@ class YiyanBrowser {
         return false;
       });
 
-      if (hasCompletion) {
-        // 完成标记出现，等待一小段时间确保内容完整
-        await this.page.waitForTimeout(500);
-        logger.clearLine();
-        logger.dim('Completion marker detected, finishing...');
-        break;
+      if (detected) {
+        hasCompletionMarker = true;
+        // 有标记，检查稳定性是否也通过
+        if (stableCount >= 2 && lastText.length > 50) {
+          await this.page.waitForTimeout(500);
+          logger.clearLine();
+          logger.dim('Completion confirmed (marker + stable)');
+          break;
+        }
       }
 
+      // Progress indicator
+      dotCount = (dotCount + 1) % 4;
+      logger.thinking(`Receiving response${'.'.repeat(dotCount)}  (${text.length} chars)`);
+
       // 等待 200ms 给 DOM 时间更新
       await this.page.waitForTimeout(200);
     }