yiyan-browser-agent 1.0.16 → 1.0.18

package/package.json

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

package/src/agent.js

@@ -6,7 +6,7 @@ const path                         = require('path');
 const { execSync }                 = require('child_process');
 const config                       = require('./config');
 const logger                       = require('./logger');
-const YiyanBrowser              = require('./browser');
+const BrowserManager               = require('./browser-manager');
 const { executeTool }              = require('./tools');
 const { parseResponse,
         formatToolResult }         = require('./parser');
@@ -18,7 +18,7 @@ const { ConversationManager }      = require('./prompt');
 
 class YiyanAgent {
   constructor(options = {}) {
-    this.browser      = new YiyanBrowser();
+    this.browser      = null;  // Will be obtained from BrowserManager
     this.conversation = new ConversationManager();
     this.options      = options;
     this._running     = false;
@@ -26,15 +26,15 @@ class YiyanAgent {
 
   // ── Public API ──────────────────────────────────────────────────────────────
 
-  /** Boot the browser and load Yiyan */
+  /** Boot the browser (reuse if available, restart if error) */
   async init() {
-    await this.browser.launch();
-    await this.browser.newChat();
+    this.browser = await BrowserManager.getInstance();
+    await BrowserManager.newChat();
   }
 
-  /** Shut down cleanly */
+  /** Shut down cleanly (only on explicit exit, not after each task) */
   async shutdown() {
-    await this.browser.close();
+    await BrowserManager.close();
   }
 
   /**
@@ -48,7 +48,13 @@ class YiyanAgent {
 
     // ── Send task directly ───────────────────────────────────
     logger.info('Sending task to Yiyan...');
-    await this.browser.sendMessage(task);
+    try {
+      await this.browser.sendMessage(task);
+    } catch (err) {
+      // Input error - mark for restart
+      BrowserManager.markError();
+      throw err;
+    }
 
     // ── Agent loop ──────────────────────────────────────────────────────
     let finalAnswer = '';
@@ -114,6 +120,14 @@ class YiyanAgent {
       finalAnswer = `Reached maximum iterations (${maxIter}).`;
       status = 'incomplete';
     }
+
+    // Mark browser status for reuse
+    if (status === 'success') {
+      BrowserManager.markSuccess();  // Keep browser open for next task
+    } else {
+      BrowserManager.markError();    // Restart browser next time
+    }
+
     this._running = false;
 
     // Return JSON

package/src/browser-manager.js

@@ -0,0 +1,85 @@
+// src/browser-manager.js — Singleton browser instance manager for reuse
+'use strict';
+
+const YiyanBrowser = require('./browser');
+const logger       = require('./logger');
+
+// Global singleton instance
+let _instance = null;
+let _initialized = false;
+let _hasError = false;
+
+class BrowserManager {
+  /**
+   * Get or create browser instance (singleton)
+   * If previous run had error, will restart the browser
+   */
+  static async getInstance() {
+    // If there was an error, restart the browser
+    if (_hasError && _instance) {
+      logger.info('Previous error detected, restarting browser...');
+      try {
+        await _instance.close();
+      } catch {}
+      _instance = null;
+      _initialized = false;
+      _hasError = false;
+    }
+
+    // Create new instance if needed
+    if (!_instance) {
+      _instance = new YiyanBrowser();
+      await _instance.launch();
+      _initialized = true;
+      logger.success('Browser initialized (will be reused)');
+    }
+
+    return _instance;
+  }
+
+  /**
+   * Mark that an error occurred (browser should be restarted next time)
+   */
+  static markError() {
+    _hasError = true;
+  }
+
+  /**
+   * Mark that everything is working fine (keep browser open)
+   */
+  static markSuccess() {
+    _hasError = false;
+  }
+
+  /**
+   * Check if browser is currently initialized
+   */
+  static isInitialized() {
+    return _initialized && _instance && !_instance._closed;
+  }
+
+  /**
+   * Force close the browser (for shutdown)
+   */
+  static async close() {
+    if (_instance) {
+      try {
+        await _instance.close();
+      } catch {}
+      _instance = null;
+      _initialized = false;
+      _hasError = false;
+    }
+  }
+
+  /**
+   * Reset for new chat session
+   */
+  static async newChat() {
+    if (_instance) {
+      await _instance.newChat();
+    }
+  }
+}
+
+module.exports = BrowserManager;

package/src/index.js

@@ -150,23 +150,25 @@ async function main() {
   const agent = new YiyanAgent({ saveLog: opts.saveLog });
 
   // ── Graceful shutdown handler ──────────────────────────────────────────────
-  const shutdown = async (code = 0) => {
-    logger.info('\nShutting down...');
-    try { await agent.shutdown(); } catch {}
+  const shutdown = async (code = 0, closeBrowser = true) => {
+    if (closeBrowser) {
+      logger.info('\nShutting down...');
+      try { await agent.shutdown(); } catch {}
+    }
     process.exit(code);
   };
 
-  process.on('SIGINT',  () => shutdown(0));
-  process.on('SIGTERM', () => shutdown(0));
+  process.on('SIGINT',  () => shutdown(0, true));
+  process.on('SIGTERM', () => shutdown(0, true));
   process.on('uncaughtException', async err => {
     logger.error(`Uncaught error: ${err.message}`);
     if (config.DEBUG) console.error(err.stack);
-    await shutdown(1);
+    await shutdown(1, true);  // 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, true);  // Error: close browser
   });
 
   // ── Calibrate mode ─────────────────────────────────────────────────────────
@@ -198,9 +200,12 @@ async function main() {
   try {
     if (opts.interactive) {
       await agent.runInteractive();
+      await shutdown(0, true);  // Interactive mode exit: close browser
     } else {
       const result = await agent.run(opts.task);
       console.log(JSON.stringify(result, null, 2));
+      // Single task mode: keep browser open if success, close on error
+      await shutdown(0, result.status === 'error');
     }
   } catch (err) {
     console.log(JSON.stringify({
@@ -209,10 +214,8 @@ async function main() {
       duration: 0,
       status: 'error'
     }, null, 2));
-    await shutdown(1);
+    await shutdown(1, true);  // Error: close browser
   }
-
-  await shutdown(0);
 }
 
 main();