yiyan-browser-agent 1.0.17 → 1.0.19

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "yiyan-browser-agent",
-  "version": "1.0.17",
+  "version": "1.0.19",
   "description": "AI coding agent powered by Yiyan (文心一言) via browser automation — no API key needed",
   "main": "src/index.js",
   "bin": {
@@ -39,7 +39,7 @@
   "license": "MIT",
   "repository": {
     "type": "git",
-    "url": "https://github.com/readfor/yiyan-browser-agent"
+    "url": "git+https://github.com/readfor/yiyan-browser-agent.git"
   },
   "homepage": "https://github.com/readfor/yiyan-browser-agent#readme"
 }

package/src/index.js

@@ -6,7 +6,8 @@ const path         = require('path');
 const fs           = require('fs');
 const config       = require('./config');
 const logger       = require('./logger');
-const YiyanAgent = require('./agent');
+const YiyanAgent   = require('./agent');
+const BrowserManager = require('./browser-manager');
 
 // ─────────────────────────────────────────────
 //  Parse CLI arguments
@@ -150,23 +151,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 +201,40 @@ async function main() {
   try {
     if (opts.interactive) {
       await agent.runInteractive();
+      await shutdown(0, true);
     } else {
       const result = await agent.run(opts.task);
       console.log(JSON.stringify(result, null, 2));
+
+      // Success: keep process alive for next task (read from stdin)
+      if (result.status === 'success') {
+        logger.info('Browser kept open. Enter next task or Ctrl+C to exit...');
+        const readline = require('readline');
+        const rl = readline.createInterface({
+          input  : process.stdin,
+          output : process.stdout,
+        });
+
+        // Wait for next task
+        while (true) {
+          const nextTask = await new Promise(resolve => rl.question('', resolve));
+          const trimmed = nextTask.trim();
+
+          if (!trimmed) continue;
+          if (['exit', 'quit', 'q'].includes(trimmed.toLowerCase())) break;
+
+          // Run next task with same browser
+          agent.conversation = new ConversationManager();
+          await BrowserManager.newChat();
+          const nextResult = await agent.run(trimmed);
+          console.log(JSON.stringify(nextResult, null, 2));
+        }
+
+        rl.close();
+        await shutdown(0, true);
+      } else {
+        await shutdown(0, true);  // Error: close browser
+      }
     }
   } catch (err) {
     console.log(JSON.stringify({
@@ -209,10 +243,8 @@ async function main() {
       duration: 0,
       status: 'error'
     }, null, 2));
-    await shutdown(1);
+    await shutdown(1, true);  // Error: close browser
   }
-
-  await shutdown(0);
 }
 
 main();