testdriverai 6.1.3 → 6.1.4-canary.a7176c2.0

package/agent/index.js

@@ -892,7 +892,10 @@ commands:
   // it will generate files that contain only "prompts"
   // @todo revit the generate command
   async generate(count = 1, prompt = null) {
-    this.emitter.emit(events.log.debug, `generate called with count: ${count}`);
+    this.emitter.emit(
+      events.log.debug,
+      `generate called with count: ${count}, prompt: ${prompt}`,
+    );
 
     await this.runLifecycle("prerun");
 

package/interfaces/cli/lib/base.js

@@ -172,20 +172,27 @@ class BaseCommand extends Command {
     return file;
   }
 
-  async setupAgent(file, flags) {
+  async setupAgent(firstArg, flags) {
     // Load .env file into process.env for CLI usage
     require("dotenv").config();
 
     // Create the agent only when actually needed
     const TestDriverAgent = require("../../../agent/index.js");
 
-    // Use --path flag if provided, otherwise use the file argument
-    const filePath = this.id === "run" && flags.path ? flags.path : file;
+    let args;
+    if (this.id === "generate") {
+      // For generate command, the first parameter is a prompt, not a file
+      args = firstArg ? [firstArg] : [];
+    } else {
+      // For run and other commands, handle file path
+      const filePath = this.id === "run" && flags.path ? flags.path : firstArg;
+      args = filePath ? [filePath] : [];
+    }
 
     // Prepare CLI args for the agent with all derived options
     const cliArgs = {
       command: this.id,
-      args: filePath ? [filePath] : [], // Only pass file path if it exists
+      args,
       options: {
         ...flags,
         resultFile:

package/interfaces/cli/utils/factory.js

@@ -28,9 +28,18 @@ function createOclifCommand(commandName) {
           this.agent.readlineInterface = readlineInterface;
           await readlineInterface.start();
         } else {
-          // For run and sandbox commands, use the unified command system
-          const fileArg = args.file || args.action || null;
-          await this.setupAgent(fileArg, flags);
+          // For run and generate commands, use the unified command system
+          let commandArgs;
+          if (commandName === "generate") {
+            // Generate command: pass prompt as first argument
+            await this.setupAgent(args.prompt, flags);
+            commandArgs = [args.prompt];
+          } else {
+            // Run and other commands use file argument
+            const fileArg = args.file || args.action || null;
+            await this.setupAgent(fileArg, flags);
+            commandArgs = [fileArg];
+          }
 
           if (commandName === "run") {
             // Set error limit higher for run command
@@ -38,7 +47,11 @@ function createOclifCommand(commandName) {
           }
 
           // Execute through unified command system
-          await this.agent.executeUnifiedCommand(commandName, [fileArg], flags);
+          await this.agent.executeUnifiedCommand(
+            commandName,
+            commandArgs,
+            flags,
+          );
         }
       } catch (error) {
         console.error(`Error executing ${commandName} command:`, error);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "testdriverai",
-  "version": "6.1.3",
+  "version": "6.1.4-canary.a7176c2.0",
   "description": "Next generation autonomous AI agent for end-to-end testing of web & desktop",
   "main": "index.js",
   "bin": {