yiyan-browser-agent 1.0.1 → 1.0.3

package/src/agent.js

@@ -0,0 +1,257 @@
+// src/agent.js — The core agent loop that ties everything together
+'use strict';
+
+const fs                           = require('fs');
+const path                         = require('path');
+const { execSync }                 = require('child_process');
+const config                       = require('./config');
+const logger                       = require('./logger');
+const YiyanBrowser              = require('./browser');
+const { executeTool }              = require('./tools');
+const { parseResponse,
+        formatToolResult }         = require('./parser');
+const { ConversationManager }      = require('./prompt');
+
+// ─────────────────────────────────────────────
+//  Agent class
+// ─────────────────────────────────────────────
+
+class YiyanAgent {
+  constructor(options = {}) {
+    this.browser      = new YiyanBrowser();
+    this.conversation = new ConversationManager();
+    this.options      = options;
+    this._running     = false;
+  }
+
+  // ── Public API ──────────────────────────────────────────────────────────────
+
+  /** Boot the browser and load Yiyan */
+  async init() {
+    await this.browser.launch();
+    await this.browser.newChat();
+  }
+
+  /** Shut down cleanly */
+  async shutdown() {
+    await this.browser.close();
+  }
+
+  /**
+   * Run a task to completion.
+   * Returns JSON object with question, answer, duration.
+   */
+  async run(task) {
+    this._running   = true;
+    const maxIter   = config.MAX_ITERATIONS;
+    const startTime = Date.now();
+
+    // ── Send task directly ───────────────────────────────────
+    logger.info('Sending task to Yiyan...');
+    await this.browser.sendMessage(task);
+
+    // ── Agent loop ──────────────────────────────────────────────────────
+    let finalAnswer = '';
+    for (let iter = 1; iter <= maxIter; iter++) {
+      logger.iteration(iter, maxIter);
+
+      // Wait for response from Yiyan
+      const rawResponse = await this.browser.waitForResponse();
+
+      if (!rawResponse || rawResponse.trim().length === 0) {
+        logger.warn('Empty response received — retrying...');
+        await this.browser.sendMessage('Please continue.');
+        continue;
+      }
+
+      if (config.DEBUG) {
+        logger.dim(`--- Raw response (${rawResponse.length} chars) ---`);
+        logger.dim(rawResponse.slice(0, 400));
+      }
+
+      this.conversation.addAssistantMessage(rawResponse);
+      const parsed = parseResponse(rawResponse);
+
+      // ── Tool call ────────────────────────────────────────────────────
+      if (parsed.type === 'tool_call') {
+        logger.toolCall(parsed.name, parsed.args);
+        let result, isError = false;
+        try {
+          result = await executeTool(parsed.name, parsed.args);
+          logger.toolResult(result);
+        } catch (err) {
+          result = `Error: ${err.message}`;
+          isError = true;
+          logger.toolResult(result, true);
+        }
+        const feedbackMsg = this.conversation.addToolResult(parsed.name, result, isError);
+        await this.browser.sendMessage(feedbackMsg);
+        continue;
+      }
+
+      // ── Parse error ──────────────────────────────────────────────────
+      if (parsed.type === 'error') {
+        logger.warn(`Parse error: ${parsed.message}`);
+        const recovery = this.conversation.addToolResult('SYSTEM', `Parse error: ${parsed.message}`, true);
+        await this.browser.sendMessage(recovery);
+        continue;
+      }
+
+      // ── Final response ───────────────────────────────────────────────
+      if (parsed.type === 'final') {
+        finalAnswer = parsed.content;
+        if (this.options.saveLog) {
+          await this._saveConversationLog(task, parsed.content);
+        }
+        this._running = false;
+        break;
+      }
+    }
+
+    const duration = Date.now() - startTime;
+    let status = 'success';
+    if (!finalAnswer) {
+      finalAnswer = `Reached maximum iterations (${maxIter}).`;
+      status = 'incomplete';
+    }
+    this._running = false;
+
+    // Return JSON
+    return {
+      question: task,
+      answer: finalAnswer,
+      duration: duration,
+      status: status
+    };
+  }
+
+  // ── Interactive (REPL) Mode ────────────────────────────────────────────────
+
+  /**
+   * Run the agent in interactive mode — keeps the browser open
+   * and outputs JSON format.
+   */
+  async runInteractive() {
+    const readline = require('readline');
+
+    const rl = readline.createInterface({
+      input    : process.stdin,
+      output   : process.stdout,
+      terminal : true,
+    });
+
+    const ask = () => new Promise(resolve => rl.question('', resolve));
+
+    while (true) {
+      let task;
+      try {
+        task = (await ask()).trim();
+      } catch {
+        break;
+      }
+
+      if (!task) continue;
+
+      if (['exit', 'quit', 'q'].includes(task.toLowerCase())) {
+        break;
+      }
+
+      if (task.toLowerCase() === 'new') {
+        await this.browser.newChat();
+        this.conversation = new ConversationManager();
+        continue;
+      }
+
+      this.conversation = new ConversationManager();
+
+      try {
+        await this.browser.newChat();
+        const result = await this.run(task);
+        // Output JSON to stdout
+        console.log(JSON.stringify(result, null, 2));
+      } catch (err) {
+        console.log(JSON.stringify({
+          question: task,
+          answer: `Error: ${err.message}`,
+          duration: 0,
+          status: 'error'
+        }, null, 2));
+      }
+    }
+
+    rl.close();
+  }
+
+  // ── Helpers ────────────────────────────────────────────────────────────────
+
+  _getWorkingDirListing() {
+    try {
+      // Use Node.js fs for cross-platform compatibility (Windows compatible)
+      const fs = require('fs');
+      const pathModule = require('path');
+      const cwd = config.WORKING_DIR;
+
+      const excludeDirs = ['node_modules', '.git', 'dist', '.next', 'build', '__pycache__', '.idea', '.vscode'];
+      const excludeFiles = ['.lock', 'package-lock.json', 'yarn.lock', 'pnpm-lock.yaml'];
+
+      let results = [];
+
+      function walk(dir, depth) {
+        if (depth > 3) return;
+        try {
+          const entries = fs.readdirSync(dir, { withFileTypes: true });
+          for (const entry of entries) {
+            const fullPath = pathModule.join(dir, entry.name);
+            const relativePath = pathModule.relative(cwd, fullPath);
+
+            if (entry.isDirectory()) {
+              if (excludeDirs.includes(entry.name)) continue;
+              results.push(relativePath);
+              walk(fullPath, depth + 1);
+            } else if (entry.isFile()) {
+              if (excludeFiles.some(ext => entry.name.endsWith(ext))) continue;
+              results.push(relativePath);
+            }
+          }
+        } catch {}
+      }
+
+      walk(cwd, 1);
+      return results.slice(0, 80).join('\n') || '(empty directory)';
+    } catch {
+      return '(could not read directory)';
+    }
+  }
+
+  async _saveConversationLog(task, finalResponse) {
+    try {
+      const logsDir = path.join(os.homedir(), '.yiyan-agent', 'logs');
+      fs.mkdirSync(logsDir, { recursive: true });
+
+      const ts       = new Date().toISOString().replace(/[:.]/g, '-').slice(0, 19);
+      const logFile  = path.join(logsDir, `session-${ts}.txt`);
+      const content  = [
+        `Yiyan Agent — Session Log`,
+        `Date: ${new Date().toISOString()}`,
+        `Task: ${task}`,
+        `Working Dir: ${config.WORKING_DIR}`,
+        '═'.repeat(60),
+        this.conversation.exportLog(),
+        '',
+        '═'.repeat(60),
+        'FINAL RESPONSE:',
+        finalResponse,
+      ].join('\n');
+
+      fs.writeFileSync(logFile, content, 'utf8');
+      logger.dim(`Conversation saved: ${logFile}`);
+    } catch (err) {
+      logger.warn(`Could not save log: ${err.message}`);
+    }
+  }
+}
+
+// Pull os into scope for the log save helper
+const os = require('os');
+
+module.exports = YiyanAgent;