@empiricalrun/test-gen 0.52.6 → 0.52.7

package/CHANGELOG.md

@@ -1,5 +1,15 @@
 # @empiricalrun/test-gen
 
+## 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/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@empiricalrun/test-gen",
-  "version": "0.52.6",
+  "version": "0.52.7",
   "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"
   },