yiyan-browser-agent 1.0.7 → 1.0.8

package/package.json

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

package/src/browser.js

@@ -267,73 +267,59 @@ class YiyanBrowser {
     );
   }
 
-  // ── Waiting for Response (optimized: MutationObserver) ────────────────────
-
+  // ── Waiting for Response (stable polling) ────────────────────────────────────
+
+  /**
+   * Wait until Yiyan finishes generating and return the response text.
+   *
+   * Algorithm:
+   *  1. Record how many assistant messages are on the page right now.
+   *  2. Wait until a new message appears (count goes up).
+   *  3. Poll the last message text every 300 ms.
+   *  4. When the text has not changed for STABLE_DELAY ms AND
+   *     no stop/loading indicator is visible → done.
+   */
   async waitForResponse() {
     const timeout     = config.RESPONSE_TIMEOUT;
     const stableDelay = config.STABLE_DELAY;
     const start       = Date.now();
 
-    // Use MutationObserver for efficient change detection
-    const responseId = await this.page.evaluate(() => {
-      let lastText    = '';
-      let stableStart = null;
-      let done        = false;
-
-      // Find response container
-      const container = document.querySelector('[class*="answer"], [class*="response"], [class*="markdown"], .ds-markdown, [class*="content"]');
-      if (!container) return null;
-
-      const observer = new MutationObserver(() => {
-        const currentText = container.textContent || '';
-
-        if (currentText !== lastText) {
-          lastText    = currentText;
-          stableStart = Date.now();
-        } else if (stableStart && Date.now() - stableStart >= 2500) {
-          // Text stable for 2.5s = done
-          done = true;
-        }
-      });
+    // Phase 1: wait for a new message to appear
+    const initialCount = await this._getMessageCount();
+    let   appeared     = false;
 
-      observer.observe(container, { childList: true, subtree: true, characterData: true });
+    for (let i = 0; i < 30; i++) {
+      const count = await this._getMessageCount();
+      if (count > initialCount) { appeared = true; break; }
+      await this.page.waitForTimeout(300);
+    }
 
-      // Store for cleanup
-      window.__yiyanObserver = observer;
-      window.__yiyanDone = () => done;
-      window.__yiyanText = () => lastText;
+    if (!appeared) logger.warn('Response may have been delayed — continuing to wait...');
 
-      return container.className || 'response-container';
-    });
+    // Phase 2: wait for text to stabilise
+    let lastText    = '';
+    let stableStart = null;
+    let dotCount    = 0;
 
-    // Wait for completion signal
-    let dotCount = 0;
     while (Date.now() - start < timeout) {
-      const state = await this.page.evaluate(() => ({
-        done  : window.__yiyanDone ? window.__yiyanDone() : false,
-        text  : window.__yiyanText ? window.__yiyanText() : '',
-        length: (window.__yiyanText ? window.__yiyanText() : '').length,
-      }));
-
-      if (state.done && state.length > 10 && !await this._isGenerating()) {
-        break;
+      const text = await this._extractLastMessage();
+
+      if (text !== lastText) {
+        lastText    = text;
+        stableStart = null;
+      } else if (text.length > 0) {
+        if (!stableStart) stableStart = Date.now();
+        else if (Date.now() - stableStart >= stableDelay) {
+          if (!await this._isGenerating()) break;  // confirmed done
+          stableStart = null;                       // still generating, reset
+        }
       }
 
       // Progress indicator
       dotCount = (dotCount + 1) % 4;
-      logger.thinking(`Receiving response${'.'.repeat(dotCount)}  (${state.length} chars)`);
+      logger.thinking(`Receiving response${'.'.repeat(dotCount)}  (${text.length} chars)`);
     }
 
-    // Cleanup observer
-    await this.page.evaluate(() => {
-      if (window.__yiyanObserver) {
-        window.__yiyanObserver.disconnect();
-        delete window.__yiyanObserver;
-        delete window.__yiyanDone;
-        delete window.__yiyanText;
-      }
-    });
-
     logger.clearLine();
 
     const final = await this._extractLastMessage();