@empiricalrun/test-gen 0.52.6 → 0.52.8

package/CHANGELOG.md

@@ -1,5 +1,22 @@
 # @empiricalrun/test-gen
 
+## 0.52.8
+
+### Patch Changes
+
+- 40b4619: fix: update browser agent prompt for todo prep
+- 7026a35: fix: missing package.json for cli launch
+
+## 0.52.7
+
+### Patch Changes
+
+- b7115ea: feat: add ascii art banner to cli
+- 3d634c9: feat: added disk persistence for gemini
+- Updated dependencies [cebf4dc]
+- Updated dependencies [3d634c9]
+  - @empiricalrun/llm@0.13.5
+
 ## 0.52.6
 
 ### Patch Changes

package/dist/agent/chat/index.js

@@ -37,7 +37,7 @@ function createChatModel(useDiskForChatState, selectedModel) {
         return new chat_1.ClaudeChatModel(useDiskForChatState);
     }
     if (selectedModel.startsWith("gemini")) {
-        return new chat_1.GeminiChatModel();
+        return new chat_1.GeminiChatModel(useDiskForChatState);
     }
     throw new Error(`Unsupported model: ${selectedModel}`);
 }

package/dist/bin/index.js

@@ -187,9 +187,9 @@ async function runAgentsWorkflow(testGenConfig, testGenToken) {
     }
     return agent;
 }
-(async function main() {
+async function main() {
     const removeListeners = setupProcessListeners(flushEvents);
-    console.log(`Running test-gen v${require("../../package.json").version} from ${__dirname}`);
+    (0, utils_2.printBanner)();
     const program = new commander_1.Command();
     program
         .option("--token <token>", "Test generation token")
@@ -256,4 +256,8 @@ async function runAgentsWorkflow(testGenConfig, testGenToken) {
     else {
         process.exit(0);
     }
-})();
+}
+main().catch((error) => {
+    console.error("Unhandled error in main function:", error);
+    process.exit(1);
+});

package/dist/bin/utils/index.d.ts

@@ -10,4 +10,5 @@ export interface CliOptions {
     chatModel?: "claude-3-7" | "3-7" | "claude-3-5" | "3-5" | "claude-3-7-sonnet-20250219" | "claude-3-5-sonnet-20241022" | "gemini-2.5-pro-preview-03-25";
 }
 export declare function validateAndCompleteCliOptions(options: CliOptions): Promise<CliOptions>;
+export declare function printBanner(): void;
 //# sourceMappingURL=index.d.ts.map

package/dist/bin/utils/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/bin/utils/index.ts"],"names":[],"mappings":"AAEA,MAAM,WAAW,UAAU;IACzB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,mBAAmB,CAAC,EAAE,OAAO,CAAC;IAC9B,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,SAAS,CAAC,EACN,YAAY,GACZ,KAAK,GACL,YAAY,GACZ,KAAK,GACL,4BAA4B,GAC5B,4BAA4B,GAC5B,8BAA8B,CAAC;CACpC;AAQD,wBAAsB,6BAA6B,CACjD,OAAO,EAAE,UAAU,GAClB,OAAO,CAAC,UAAU,CAAC,CAyDrB"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/bin/utils/index.ts"],"names":[],"mappings":"AAEA,MAAM,WAAW,UAAU;IACzB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,mBAAmB,CAAC,EAAE,OAAO,CAAC;IAC9B,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,SAAS,CAAC,EACN,YAAY,GACZ,KAAK,GACL,YAAY,GACZ,KAAK,GACL,4BAA4B,GAC5B,4BAA4B,GAC5B,8BAA8B,CAAC;CACpC;AAQD,wBAAsB,6BAA6B,CACjD,OAAO,EAAE,UAAU,GAClB,OAAO,CAAC,UAAU,CAAC,CAyDrB;AAED,wBAAgB,WAAW,SAgC1B"}

package/dist/bin/utils/index.js

@@ -3,7 +3,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.validateAndCompleteCliOptions = void 0;
+exports.printBanner = exports.validateAndCompleteCliOptions = void 0;
 const inquirer_1 = __importDefault(require("inquirer"));
 async function validateAndCompleteCliOptions(options) {
     // For existing flow between dashboard <> test-gen (via ci-worker)
@@ -61,3 +61,39 @@ async function validateAndCompleteCliOptions(options) {
     return options;
 }
 exports.validateAndCompleteCliOptions = validateAndCompleteCliOptions;
+function printBanner() {
+    const gray = "\x1b[90m";
+    const reset = "\x1b[0m";
+    const asciiArtRaw = `
+  ,_,
+ {o,o}
+ /)__)
+--"-"-`;
+    const version = require("../../../package.json").version;
+    const logLine1 = `Running test-gen v${version}`;
+    const logLine2 = `from ${__dirname}`;
+    // Process ASCII art
+    const asciiLines = asciiArtRaw
+        .split("\n")
+        .filter((line) => line.trim() !== "");
+    const maxWidth = Math.max(...asciiLines.map((line) => line.length));
+    const middleIndex = Math.floor(asciiLines.length / 2);
+    const spacing = "    "; // 4 spaces for padding
+    // Print formatted output (art default, logs gray)
+    asciiLines.forEach((line, index) => {
+        const paddedLine = line.padEnd(maxWidth, " ");
+        if (index === middleIndex) {
+            // Print art line (default color) + first log line (gray)
+            console.log(`${paddedLine}${spacing}${gray}${logLine1}${reset}`);
+        }
+        else if (index === middleIndex + 1) {
+            // Print corresponding art line (default color) + second log line (gray)
+            console.log(`${paddedLine}${spacing}${gray}${logLine2}${reset}`);
+        }
+        else {
+            // Print other art lines (default color)
+            console.log(paddedLine);
+        }
+    });
+}
+exports.printBanner = printBanner;

package/dist/reporter/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/reporter/index.ts"],"names":[],"mappings":"AAYA,OAAO,EAAkB,QAAQ,EAAE,MAAM,OAAO,CAAC;AAEjD,KAAK,kBAAkB,GAAG;IACxB,aAAa,EAAE,MAAM,CAAC;IACtB,YAAY,EAAE,MAAM,CAAC;IACrB,eAAe,EAAE,MAAM,CAAC;CACzB,CAAC;AAKF,wBAAgB,WAAW,IAAI,QAAQ,GAAG,SAAS,CAUlD;AAED;;;GAGG;AACH,wBAAgB,iBAAiB,CAAC,MAAM,EAAE,kBAAkB,GAAG,IAAI,CAGlE;AAED,qBAAa,sBAAsB;IACjC,OAAO,CAAC,OAAO,CAAS;;IAKlB,eAAe,CAAC,EACpB,eAAe,EACf,QAAQ,GACT,EAAE;QACD,eAAe,EAAE,MAAM,CAAC;QACxB,QAAQ,EAAE,MAAM,CAAC;KAClB;IAgDK,eAAe,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IA8C9C,WAAW,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAY3C,UAAU,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAY1C,iBAAiB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;CAWxD"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/reporter/index.ts"],"names":[],"mappings":"AAYA,OAAO,EAAkB,QAAQ,EAAE,MAAM,OAAO,CAAC;AAEjD,KAAK,kBAAkB,GAAG;IACxB,aAAa,EAAE,MAAM,CAAC;IACtB,YAAY,EAAE,MAAM,CAAC;IACrB,eAAe,EAAE,MAAM,CAAC;CACzB,CAAC;AAKF,wBAAgB,WAAW,IAAI,QAAQ,GAAG,SAAS,CAUlD;AAED;;;GAGG;AACH,wBAAgB,iBAAiB,CAAC,MAAM,EAAE,kBAAkB,GAAG,IAAI,CAElE;AAED,qBAAa,sBAAsB;IACjC,OAAO,CAAC,OAAO,CAAS;;IAKlB,eAAe,CAAC,EACpB,eAAe,EACf,QAAQ,GACT,EAAE;QACD,eAAe,EAAE,MAAM,CAAC;QACxB,QAAQ,EAAE,MAAM,CAAC;KAClB;IAgDK,eAAe,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IA8C9C,WAAW,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAY3C,UAAU,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAY1C,iBAAiB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;CAWxD"}

package/dist/reporter/index.js

@@ -29,7 +29,6 @@ exports.getReporter = getReporter;
  * method won't throw error if it fails to report
  */
 function setReporterConfig(config) {
-    console.info("initialised reporter config");
     reporterConfig = config;
 }
 exports.setReporterConfig = setReporterConfig;

package/dist/tools/test-gen-browser.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"test-gen-browser.d.ts","sourceRoot":"","sources":["../../src/tools/test-gen-browser.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,wBAAwB,CAAC;AA0DnD,eAAO,MAAM,4BAA4B,EAAE,IA0E1C,CAAC"}
+{"version":3,"file":"test-gen-browser.d.ts","sourceRoot":"","sources":["../../src/tools/test-gen-browser.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,wBAAwB,CAAC;AAwEnD,eAAO,MAAM,4BAA4B,EAAE,IA0E1C,CAAC"}

package/dist/tools/test-gen-browser.js

@@ -46,8 +46,22 @@ For the above file, the browser environment will execute the steps before the TO
 to the browser agent. The agent will do the actions described in the TODO comment and then resume control back to the
 test code.
 
-The browser agent will return a summary of actions that it took, and the generated Playwright code for them. You can
-then use the text editor tool to replace the TODO comment with the generated Playwright code.
+Note that you CANNOT add TODO comments in the middle of a multi-line statement. This will break the tool.
+
+For example, this is invalid:
+
+\`\`\`
+await extPage
+  .getByTestId("virtuoso-item-list")
+  // TODO(agent): Click on the STARS button
+  .getByText("STARS", { exact: true })
+  .click();
+// This is invalid, because the TODO is in the middle of a multi-line statement
+// Instead, add the TODO before or after the multi-line statement
+\`\`\`
+
+After execution, the browser agent will return a summary of actions that it took, and the generated Playwright code for them.
+You can then use the text editor tool to replace the TODO comment with the generated Playwright code.
 `;
 exports.generateTestWithBrowserAgent = {
     schema: {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@empiricalrun/test-gen",
-  "version": "0.52.6",
+  "version": "0.52.8",
   "publishConfig": {
     "registry": "https://registry.npmjs.org/",
     "access": "public"
@@ -68,7 +68,7 @@
     "tsx": "^4.16.2",
     "typescript": "^5.3.3",
     "zod": "^3.23.8",
-    "@empiricalrun/llm": "^0.13.4",
+    "@empiricalrun/llm": "^0.13.5",
     "@empiricalrun/r2-uploader": "^0.3.8",
     "@empiricalrun/test-run": "^0.7.6"
   },