yiyan-browser-agent 1.6.6 → 1.6.7

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "yiyan-browser-agent",
-  "version": "1.6.6",
+  "version": "1.6.7",
   "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,19 +296,43 @@ 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
+    // 主要：dialogCardBottom 出现
+    // 备用：内容稳定 + _isGenerating() 确认
+    let lastText      = '';
+    let stableStart   = null;
+    let stableCount   = 0;
+    let dotCount      = 0;
 
     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++;
+
+          // 连续稳定检测 + _isGenerating 确认
+          if (stableCount >= 2) {
+            if (!await this._isGenerating()) {
+              logger.clearLine();
+              logger.dim('Stable completion detected');
+              break;
+            }
+            stableCount = 0;
+            stableStart = null;
+          } else {
+            stableStart = null;
+          }
+        }
+      }
 
-      // 检测完成标记：dialogCardBottom 出现
+      // ── 完成标记检测（主要退出路径）──
       const hasCompletion = await this.page.evaluate(() => {
         const selectors = [
           '.dialogCardBottom__qoXjps3z',
@@ -326,13 +351,16 @@ class YiyanBrowser {
       });
 
       if (hasCompletion) {
-        // 完成标记出现，等待一小段时间确保内容完整
         await this.page.waitForTimeout(500);
         logger.clearLine();
-        logger.dim('Completion marker detected, finishing...');
+        logger.dim('Completion marker detected');
         break;
       }
 
+      // Progress indicator
+      dotCount = (dotCount + 1) % 4;
+      logger.thinking(`Receiving response${'.'.repeat(dotCount)}  (${text.length} chars)`);
+
       // 等待 200ms 给 DOM 时间更新
       await this.page.waitForTimeout(200);
     }