yiyan-browser-agent 1.0.24 → 1.0.26

package/package.json

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

package/src/browser-manager.js

@@ -39,8 +39,8 @@ class BrowserManager {
           userAgent: 'Mozilla/5.0 AppleWebKit/537.36 Chrome/124.0.0.0 Safari/537.36',
         });
         const page = await context.newPage();
-        await page.goto(config.YIYAN_URL, { waitUntil: 'domcontentloaded', timeout: 15000 });
-        await page.waitForTimeout(500);
+        await page.goto(config.YIYAN_URL, { waitUntil: 'networkidle', timeout: 20000 });
+        await page.waitForTimeout(1500);  // Extra wait for UI to render
 
         logger.success('Connected! New tab opened.');
         return { browser, context, page, isNew: false };
@@ -61,7 +61,6 @@ class BrowserManager {
     logger.info('Launching new browser server...');
 
     // Use launchServer to create independent browser process
-    // Browser will NOT close when Node.js process exits
     const browserServer = await chromium.launchServer({
       headless: false,
       args: [
@@ -73,14 +72,14 @@ class BrowserManager {
 
     // Get the WebSocket endpoint
     const wsEndpoint = browserServer.wsEndpoint();
-    const port = wsEndpoint.match(/:(\d+)/)?.[1] || CDP_PORT;
+    logger.dim('WebSocket endpoint: ' + wsEndpoint);
 
-    // Save endpoint info
+    // Save endpoint info IMMEDIATELY before any other operations
     fs.writeFileSync(CDP_PORT_FILE, JSON.stringify({
       wsEndpoint: wsEndpoint,
-      port: port,
       launchedAt: Date.now()
     }));
+    logger.dim('Endpoint saved to: ' + CDP_PORT_FILE);
 
     logger.success('Browser server started!');
 
@@ -98,7 +97,10 @@ class BrowserManager {
     await page.waitForTimeout(800);
 
     logger.success('Tab ready.');
-    return { browser, context, page, isNew: true, browserServer };
+
+    // Return without browserServer - let it run independently
+    // Do NOT close browserServer on process exit
+    return { browser, context, page, isNew: true };
   }
 
   /**

package/src/index.js

@@ -151,24 +151,25 @@ async function main() {
   const agent = new YiyanAgent({ saveLog: opts.saveLog });
 
   // ── Graceful shutdown handler ──────────────────────────────────────────────
-  // Only close browser on explicit exit (Ctrl+C, error)
-  const shutdown = async (code = 0) => {
-    logger.info('\nShutting down...');
-    try { await BrowserManager.forceClose(); } catch {}
+  // Only delete endpoint file on Ctrl+C (explicit exit)
+  const shutdown = async (code = 0, keepBrowser = true) => {
+    if (!keepBrowser) {
+      try { BrowserManager.forceClose(); } catch {}
+    }
     process.exit(code);
   };
 
-  process.on('SIGINT',  () => shutdown(0));
-  process.on('SIGTERM', () => shutdown(0));
+  process.on('SIGINT',  () => shutdown(0, false));  // Ctrl+C: close browser
+  process.on('SIGTERM', () => shutdown(0, false));  // Kill: close browser
   process.on('uncaughtException', async err => {
     logger.error(`Uncaught error: ${err.message}`);
     if (config.DEBUG) console.error(err.stack);
-    await shutdown(1);
+    await shutdown(1, false);  // Error: close browser
   });
   process.on('unhandledRejection', async reason => {
     logger.error(`Unhandled rejection: ${reason}`);
     if (config.DEBUG) console.error(reason);
-    await shutdown(1);
+    await shutdown(1, false);  // Error: close browser
   });
 
   // ── Calibrate mode ─────────────────────────────────────────────────────────

package/src/page-agent.js

@@ -27,17 +27,24 @@ class PageAgent {
    * Send message to Yiyan
    */
   async sendMessage(text) {
+    // Wait for page to be fully loaded
+    await this.page.waitForLoadState('networkidle', { timeout: 10000 }).catch(() => {});
+    await this.page.waitForTimeout(1000);
+
     // Find input
     let inputEl = null;
     for (const sel of SEL.chatInput) {
       try {
-        inputEl = await this.page.waitForSelector(sel, { timeout: 5000, state: 'visible' });
-        if (inputEl) break;
+        inputEl = await this.page.waitForSelector(sel, { timeout: 8000, state: 'visible' });
+        if (inputEl) {
+          logger.dim('Found input: ' + sel);
+          break;
+        }
       } catch {}
     }
 
     if (!inputEl) {
-      throw new Error('Cannot find input box');
+      throw new Error('Cannot find input box. Make sure Yiyan page is loaded.');
     }
 
     // Focus and clear