yiyan-browser-agent 1.0.7 → 1.0.9

package/package.json

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

package/src/browser.js

@@ -242,7 +242,8 @@ class YiyanBrowser {
     await this.page.waitForTimeout(100);
 
     // Type text character by character (stable, works reliably)
-    await this.page.keyboard.type(text, { delay: 30 });
+    // delay: 15ms for faster typing while staying stable
+    await this.page.keyboard.type(text, { delay: 15 });
 
     // Press Enter to send
     await this.page.keyboard.press('Enter');
@@ -267,72 +268,52 @@ class YiyanBrowser {
     );
   }
 
-  // ── Waiting for Response (optimized: MutationObserver) ────────────────────
+  // ── Waiting for Response (stable polling) ────────────────────────────────────
 
+  /**
+   * Wait until Yiyan finishes generating and return the response text.
+   */
   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 < 40; i++) {
+      const count = await this._getMessageCount();
+      if (count > initialCount) { appeared = true; break; }
+      await this.page.waitForTimeout(200);
+    }
 
-      // 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;
+          stableStart = null;
+        }
       }
 
       // 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;
-      }
-    });
+      await this.page.waitForTimeout(200);
 
     logger.clearLine();
 

package/src/config.js

@@ -12,10 +12,10 @@ const defaults = {
   SESSION_DIR    : path.join(os.homedir(), '.yiyan-agent', 'session'),
   HEADLESS       : false,
 
-  // Timing (optimized for faster response detection)
+  // Timing (configurable for speed vs stability tradeoff)
   RESPONSE_TIMEOUT : 180_000,
-  STABLE_DELAY     : 1_500,  // Reduced from 2500ms (MutationObserver is faster)
-  SEND_DELAY       : 100,    // Reduced from 400ms (paste is instant)
+  STABLE_DELAY     : 1_000,  // 1s stability check (faster, may cut off long responses)
+  SEND_DELAY       : 100,    // Reduced for faster operation
 
   // Agent
   MAX_ITERATIONS   : 40,