testdriverai 6.1.3-canary.3479900.0 → 6.1.3-canary.4aa7788.0

package/agent/index.js

@@ -891,7 +891,7 @@ commands:
   // based on the current state of the system (primarily the current screenshot)
   // it will generate files that contain only "prompts"
   // @todo revit the generate command
-  async generate(count = 1) {
+  async generate(count = 1, prompt = null) {
     this.emitter.emit(events.log.debug, `generate called with count: ${count}`);
 
     await this.runLifecycle("prerun");
@@ -909,7 +909,7 @@ commands:
     let message = await this.sdk.req(
       "generate",
       {
-        prompt: "make sure to do a spellcheck",
+        prompt: prompt || "make sure to do a spellcheck",
         image,
         mousePosition: mouse,
         activeWindow: activeWindow,

package/agent/interface.js

@@ -200,8 +200,8 @@ function createCommandDefinitions(agent) {
     generate: {
       description: "Generate test files based on current screen state",
       args: {
-        file: Args.string({
-          description: "Base test file to run before generating (optional)",
+        prompt: Args.string({
+          description: "Multi-line text prompt describing what to generate",
           required: false,
         }),
       },
@@ -227,9 +227,8 @@ function createCommandDefinitions(agent) {
         }),
       },
       handler: async (args, flags) => {
-        // The file argument is already handled by thisFile in the agent constructor
-        // Just call generate with the count
-        await agent.generate(flags.count || 3);
+        // Call generate with the count and prompt
+        await agent.generate(flags.count || 3, args.prompt);
       },
     },
   };

package/package.json

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