yiyan-browser-agent 1.0.17 → 1.0.18

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "yiyan-browser-agent",
-  "version": "1.0.17",
+  "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/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();