yiyan-browser-agent 1.0.6 → 1.0.8

package/package.json

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

@@ -228,51 +228,21 @@ class YiyanBrowser {
     });
   }
 
-  // ── Sending Messages (optimized: paste instead of type) ────────────────────
+  // ── Sending Messages (stable: keyboard.type) ────────────────────────────────
 
   async sendMessage(text) {
     const { el } = await this._findInput();
 
-    // Focus the element
-    await el.focus();
+    // Focus and select all existing content
+    await el.click({ clickCount: 3, force: true });
+    await this.page.waitForTimeout(200);
 
-    // Clear existing content
-    await el.evaluate(e => {
-      e.textContent = '';
-      if (e.innerText) e.innerText = '';
-    });
-
-    // Use clipboard paste for instant input (much faster than typing)
-    await this.page.evaluate(async (text) => {
-      // Try modern clipboard API first
-      try {
-        await navigator.clipboard.writeText(text);
-        document.execCommand('paste');
-        return;
-      } catch {}
-
-      // Fallback: direct DOM manipulation with proper event dispatch
-      const activeEl = document.activeElement;
-      if (activeEl && activeEl.isContentEditable) {
-        activeEl.textContent = text;
-
-        // Dispatch input event so Yiyan recognizes the content
-        const inputEvent = new InputEvent('input', {
-          bubbles: true,
-          cancelable: true,
-          inputType: 'insertText',
-          data: text,
-        });
-        activeEl.dispatchEvent(inputEvent);
-
-        // Also dispatch change event for good measure
-        const changeEvent = new Event('change', { bubbles: true });
-        activeEl.dispatchEvent(changeEvent);
-      }
-    }, text);
+    // Clear by pressing Delete
+    await this.page.keyboard.press('Delete');
+    await this.page.waitForTimeout(100);
 
-    // Small delay to let UI process the input
-    await this.page.waitForTimeout(50);
+    // Type text character by character (stable, works reliably)
+    await this.page.keyboard.type(text, { delay: 30 });
 
     // Press Enter to send
     await this.page.keyboard.press('Enter');
@@ -297,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();