@empiricalrun/test-gen 0.69.5 → 0.69.7

package/CHANGELOG.md

@@ -1,5 +1,24 @@
 # @empiricalrun/test-gen
 
+## 0.69.7
+
+### Patch Changes
+
+- bca8c8b: fix: codegen timing issue
+- 49a3382: feat: add commands for environments and download build
+- Updated dependencies [16f3927]
+  - @empiricalrun/llm@0.19.3
+
+## 0.69.6
+
+### Patch Changes
+
+- c0c1a9a: fix: use gemini stable instead of preview
+- d67ca35: feat: add fetch-chat-session command, handle artifacts with same name
+- 9a923e2: chore: preview env resources are in ap-south-1
+- Updated dependencies [c0c1a9a]
+  - @empiricalrun/llm@0.19.2
+
 ## 0.69.5
 
 ### Patch Changes

package/dist/agent/chat/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/agent/chat/index.ts"],"names":[],"mappings":"AAEA,OAAO,EAGL,mBAAmB,EACpB,MAAM,4BAA4B,CAAC;AAyCpC,wBAAsB,yBAAyB,IAAI,OAAO,CACxD,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CACvB,CAwBA;AAED,wBAAsB,kBAAkB,CAAC,EACvC,mBAAmB,EACnB,aAAa,EACb,oBAAoB,GACrB,EAAE;IACD,aAAa,EAAE,mBAAmB,CAAC;IACnC,mBAAmB,EAAE,OAAO,CAAC;IAC7B,oBAAoB,EAAE,MAAM,GAAG,SAAS,CAAC;CAC1C,iBA+HA;AAuBD,wBAAsB,wBAAwB,CAAC,EAC7C,aAAa,EACb,aAAa,GACd,EAAE;IACD,aAAa,EAAE,mBAAmB,CAAC;IACnC,aAAa,EAAE,MAAM,CAAC;CACvB,iBA6DA"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/agent/chat/index.ts"],"names":[],"mappings":"AAEA,OAAO,EAGL,mBAAmB,EACpB,MAAM,4BAA4B,CAAC;AAyCpC,wBAAsB,yBAAyB,IAAI,OAAO,CACxD,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CACvB,CAwBA;AAED,wBAAsB,kBAAkB,CAAC,EACvC,mBAAmB,EACnB,aAAa,EACb,oBAAoB,GACrB,EAAE;IACD,aAAa,EAAE,mBAAmB,CAAC;IACnC,mBAAmB,EAAE,OAAO,CAAC;IAC7B,oBAAoB,EAAE,MAAM,GAAG,SAAS,CAAC;CAC1C,iBAkIA;AAuBD,wBAAsB,wBAAwB,CAAC,EAC7C,aAAa,EACb,aAAa,GACd,EAAE;IACD,aAAa,EAAE,mBAAmB,CAAC;IACnC,aAAa,EAAE,MAAM,CAAC;CACvB,iBA6DA"}

package/dist/agent/chat/index.js

@@ -118,7 +118,9 @@ async function runChatAgentForCLI({ useDiskForChatState, selectedModel, initialP
             }
             if (!stopCriteria(userPrompt)) {
                 const { text, attachments } = (0, utils_1.extractAttachments)(userPrompt);
-                const processed = await chatModel.processAttachments(attachments);
+                const processed = attachments.length > 0
+                    ? await chatModel.processAttachments(attachments)
+                    : [];
                 chatModel.pushUserMessage(text, processed);
             }
         }

package/dist/agent/cua/pw-codegen/pw-pause/for-recorder.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"for-recorder.d.ts","sourceRoot":"","sources":["../../../../../src/agent/cua/pw-codegen/pw-pause/for-recorder.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAIvC,OAAO,EAAE,cAAc,EAAE,MAAM,SAAS,CAAC;AAEzC,qBAAa,iCAAiC;IAO1C,OAAO,CAAC,eAAe;IANzB,OAAO,CAAC,IAAI,CAAa;IACzB,OAAO,CAAC,IAAI,CAAmB;IAC/B,OAAO,CAAC,MAAM,CAA4C;IAC1D,OAAO,CAAC,iBAAiB,CAAqB;gBAGpC,eAAe,EAAE,CAAC,IAAI,EAAE,cAAc,EAAE,KAAK,OAAO,CAAC,IAAI,CAAC;YAKtD,QAAQ;IAWhB,UAAU,CAAC,IAAI,EAAE,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;IAkBrC,sBAAsB,CAAC,IAAI,EAAE,IAAI;CA4CxC"}
+{"version":3,"file":"for-recorder.d.ts","sourceRoot":"","sources":["../../../../../src/agent/cua/pw-codegen/pw-pause/for-recorder.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAIvC,OAAO,EAAE,cAAc,EAAE,MAAM,SAAS,CAAC;AAGzC,qBAAa,iCAAiC;IAO1C,OAAO,CAAC,eAAe;IANzB,OAAO,CAAC,IAAI,CAAa;IACzB,OAAO,CAAC,IAAI,CAAmB;IAC/B,OAAO,CAAC,MAAM,CAA4C;IAC1D,OAAO,CAAC,iBAAiB,CAAqB;gBAGpC,eAAe,EAAE,CAAC,IAAI,EAAE,cAAc,EAAE,KAAK,OAAO,CAAC,IAAI,CAAC;YAKtD,QAAQ;IAWhB,UAAU,CAAC,IAAI,EAAE,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;IAkBrC,sBAAsB,CAAC,IAAI,EAAE,IAAI;CAyCxC"}

package/dist/agent/cua/pw-codegen/pw-pause/for-recorder.js

@@ -7,6 +7,7 @@ exports.PlaywrightPauseCodegenForRecorder = void 0;
 const express_1 = __importDefault(require("express"));
 const logger_1 = require("../../../../logger");
 const ipc_1 = require("./ipc");
+const utils_1 = require("./utils");
 class PlaywrightPauseCodegenForRecorder {
     sourcesCallback;
     port = 0;
@@ -45,29 +46,27 @@ class PlaywrightPauseCodegenForRecorder {
         await this.startPlaywrightCodegen(page);
     }
     async startPlaywrightCodegen(page) {
-        // Similar to the same name method in the main codegen
-        /**
-         * x-pw-glass
-         *   # inside shadow DOM (closed)
-         *   |----> x-pw-overlay (controls)
-         *   |----> x-pw-highlight (red highlight)
-         *   |----> x-pw-tooltip
-         */
-        const timerPromise = new Promise((resolve) => {
-            setTimeout(resolve, 1000);
-        });
+        // TODO: Merge this with the other page.pause oriented codegen
         const pausePromise = page.pause();
-        await timerPromise;
+        await (0, utils_1.sleep)(5_000);
         const evaluatePromise = page.evaluate(() => {
             // @ts-ignore
             console.log(window["__pw_recorderSetMode"]("recording"));
             // Remove playwright's glass pane
+            /**
+             * x-pw-glass
+             *   # inside shadow DOM (closed)
+             *   |----> x-pw-overlay (controls)
+             *   |----> x-pw-highlight (red highlight)
+             *   |----> x-pw-tooltip
+             */
             const glassPane = document.querySelector("x-pw-glass");
             if (glassPane) {
                 const styles = glassPane.getAttribute("style") || "";
                 glassPane.setAttribute("style", styles + "; display: none;");
             }
         });
+        await Promise.all([pausePromise, evaluatePromise]);
         await page.addInitScript(() => {
             let displayNoneSet = false;
             function setDisplayNone() {
@@ -86,7 +85,6 @@ class PlaywrightPauseCodegenForRecorder {
                 setDisplayNone();
             }, 100);
         });
-        await Promise.all([pausePromise, evaluatePromise]);
     }
 }
 exports.PlaywrightPauseCodegenForRecorder = PlaywrightPauseCodegenForRecorder;

package/dist/agent/cua/pw-codegen/pw-pause/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../../src/agent/cua/pw-codegen/pw-pause/index.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAGvC,OAAO,EAAE,qBAAqB,EAAE,MAAM,UAAU,CAAC;AAKjD,OAAO,EAAE,2BAA2B,EAAE,sBAAsB,EAAE,MAAM,SAAS,CAAC;AAE9E,wBAAsB,kBAAkB,CAAC,OAAO,EAAE,MAAM,oBAyCvD;AAED,qBAAa,sBAAuB,YAAW,qBAAqB;IAClE,OAAO,CAAC,IAAI,CAAa;IACzB,OAAO,CAAC,IAAI,CAAmB;IAC/B,OAAO,CAAC,MAAM,CAA4C;IAC1D,OAAO,CAAC,iBAAiB,CAAqB;;YAMhC,QAAQ;IAUhB,UAAU,CAAC,IAAI,EAAE,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;IAkBrC,sBAAsB,CAAC,IAAI,EAAE,IAAI;IAyBjC,YAAY,IAAI,OAAO,CAAC,IAAI,CAAC;IAI7B,oBAAoB,IAAI,OAAO,CAAC,MAAM,CAAC;CAU9C"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../../src/agent/cua/pw-codegen/pw-pause/index.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAGvC,OAAO,EAAE,qBAAqB,EAAE,MAAM,UAAU,CAAC;AAKjD,OAAO,EAAE,2BAA2B,EAAE,sBAAsB,EAAE,MAAM,SAAS,CAAC;AAE9E,wBAAsB,kBAAkB,CAAC,OAAO,EAAE,MAAM,oBAyCvD;AAED,qBAAa,sBAAuB,YAAW,qBAAqB;IAClE,OAAO,CAAC,IAAI,CAAa;IACzB,OAAO,CAAC,IAAI,CAAmB;IAC/B,OAAO,CAAC,MAAM,CAA4C;IAC1D,OAAO,CAAC,iBAAiB,CAAqB;;YAMhC,QAAQ;IAUhB,UAAU,CAAC,IAAI,EAAE,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;IAkBrC,sBAAsB,CAAC,IAAI,EAAE,IAAI;IAqBjC,YAAY,IAAI,OAAO,CAAC,IAAI,CAAC;IAI7B,oBAAoB,IAAI,OAAO,CAAC,MAAM,CAAC;CAU9C"}

package/dist/agent/cua/pw-codegen/pw-pause/index.js

@@ -9,6 +9,7 @@ const express_1 = __importDefault(require("express"));
 const logger_1 = require("../../../../logger");
 const ipc_1 = require("./ipc");
 const patch_1 = require("./patch");
+const utils_1 = require("./utils");
 var patch_2 = require("./patch");
 Object.defineProperty(exports, "preparePlaywrightForCodegen", { enumerable: true, get: function () { return patch_2.preparePlaywrightForCodegen; } });
 Object.defineProperty(exports, "revertToOriginalPwCode", { enumerable: true, get: function () { return patch_2.revertToOriginalPwCode; } });
@@ -77,14 +78,10 @@ class PlaywrightPauseCodegen {
         await this.startPlaywrightCodegen(page);
     }
     async startPlaywrightCodegen(page) {
-        // TODO: Glass pane needs to be removed on every page load
         // We use bindings that Playwright exposes to the page
         // Ref: https://github.com/microsoft/playwright/blob/e1c8e0f6b33923c95cc4b9416aefa6977b1d3c55/packages/playwright-core/src/server/recorder.ts#L191
-        const timerPromise = new Promise((resolve) => {
-            setTimeout(resolve, 1000);
-        });
         const pausePromise = page.pause();
-        await timerPromise;
+        await (0, utils_1.sleep)(5_000);
         const evaluatePromise = page.evaluate(() => {
             // First, we start recording
             // @ts-ignore

package/dist/agent/cua/pw-codegen/pw-pause/utils.d.ts

@@ -0,0 +1,2 @@
+export declare function sleep(ms: number): Promise<void>;
+//# sourceMappingURL=utils.d.ts.map

package/dist/agent/cua/pw-codegen/pw-pause/utils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../../../src/agent/cua/pw-codegen/pw-pause/utils.ts"],"names":[],"mappings":"AAAA,wBAAsB,KAAK,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAErD"}

package/dist/agent/cua/pw-codegen/pw-pause/utils.js

@@ -0,0 +1,6 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.sleep = sleep;
+async function sleep(ms) {
+    return new Promise((resolve) => setTimeout(resolve, ms));
+}

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

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/artifacts/index.ts"],"names":[],"mappings":"AAIA,OAAO,EACL,QAAQ,EACR,aAAa,EAGd,MAAM,4BAA4B,CAAC;AAkBpC,wBAAgB,2BAA2B,uBAM1C;AA6HD;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAsCG;AACH,wBAAsB,gBAAgB,CACpC,MAAM,EAAE,aAAa,EAAE,EACvB,OAAO,EAAE,MAAM,EACf,UAAU,EAAE,MAAM,GACjB,OAAO,CAAC,QAAQ,EAAE,CAAC,CA6DrB;AAED,qBAAa,oBAAoB;IAC/B,OAAO,CAAC,UAAU,CAAS;IAC3B,OAAO,CAAC,YAAY,CAAS;IAC7B,OAAO,CAAC,eAAe,CAAkB;IACzC,OAAO,CAAC,aAAa,CAA8B;gBAEvC,YAAY,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM;IAKvC,OAAO,CAAC,SAAS,EAAE,aAAa,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAelD,iBAAiB,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;CAMtD"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/artifacts/index.ts"],"names":[],"mappings":"AAIA,OAAO,EACL,QAAQ,EACR,aAAa,EAGd,MAAM,4BAA4B,CAAC;AAkBpC,wBAAgB,2BAA2B,uBAM1C;AA6HD;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAsCG;AACH,wBAAsB,gBAAgB,CACpC,MAAM,EAAE,aAAa,EAAE,EACvB,OAAO,EAAE,MAAM,EACf,UAAU,EAAE,MAAM,GACjB,OAAO,CAAC,QAAQ,EAAE,CAAC,CA8DrB;AAED,qBAAa,oBAAoB;IAC/B,OAAO,CAAC,UAAU,CAAS;IAC3B,OAAO,CAAC,YAAY,CAAS;IAC7B,OAAO,CAAC,eAAe,CAAkB;IACzC,OAAO,CAAC,aAAa,CAA8B;gBAEvC,YAAY,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM;IAKvC,OAAO,CAAC,SAAS,EAAE,aAAa,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAelD,iBAAiB,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;CAMtD"}

package/dist/artifacts/index.js

@@ -167,8 +167,9 @@ async function collectArtifacts(inputs, repoDir, toolCallId) {
     try {
         const dirs = { destinationDir, sourceDir: repoDir };
         const dataArtifactsWithFileName = dataArtifacts.map((a) => {
+            const randomSuffix = Math.random().toString(36).substring(2, 15);
             const fileExt = mime_1.default.getExtension(a.contentType);
-            const fileName = `${(0, slug_1.slugify)(a.name)}-${Date.now().toString()}.${fileExt}`;
+            const fileName = `${(0, slug_1.slugify)(a.name)}-${randomSuffix}.${fileExt}`;
             return { ...a, fileName };
         });
         await Promise.all([

package/dist/bin/index.js

@@ -7,6 +7,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 const llm_1 = require("@empiricalrun/llm");
 const commander_1 = require("commander");
 const dotenv_1 = __importDefault(require("dotenv"));
+const fs_1 = __importDefault(require("fs"));
 const run_1 = require("../agent/browsing/run");
 const utils_1 = require("../agent/browsing/utils");
 const chat_1 = require("../agent/chat");
@@ -19,6 +20,7 @@ const run_3 = require("../agent/planner/run");
 const auth_1 = require("../auth");
 const api_client_1 = require("../auth/api-client");
 const recorder_1 = require("../recorder");
+const validation_1 = require("../recorder/validation");
 const reporter_1 = require("../reporter");
 const session_1 = require("../session");
 const test_build_1 = require("../test-build");
@@ -285,6 +287,63 @@ async function main() {
         }
         process.exit(0);
     });
+    program
+        .command("environments")
+        .description("List environments and their latest builds")
+        .option("--repo-name <name>", "Repository name to fetch environments for")
+        .action(async (opts) => {
+        try {
+            let repoName;
+            if (opts.repoName) {
+                repoName = opts.repoName;
+            }
+            else {
+                // Auto-detect repo name from current directory
+                repoName = await (0, validation_1.validatePackageJson)(process.cwd());
+            }
+            const response = await api_client_1.apiClient.request(`/api/environments/list?project_repo_name=${encodeURIComponent(repoName)}`);
+            if (!response.ok) {
+                console.log(await response.text());
+                console.error(`❌ Failed to fetch environments for repo ${repoName}:`, response.statusText);
+                process.exit(1);
+            }
+            const result = await response.json();
+            if (result.data.environments && result.data.environments.length > 0) {
+                result.data.environments.forEach((env) => {
+                    console.log(`  Environment: ${env.name}`);
+                    if (env.latest_build) {
+                        console.log(`    Latest build: ${env.latest_build.url}`);
+                    }
+                });
+            }
+            else {
+                console.log(`No environments found for repository: ${repoName}`);
+            }
+        }
+        catch (error) {
+            console.error("❌ Error fetching environments:", error.message);
+            process.exit(1);
+        }
+        process.exit(0);
+    });
+    program
+        .command("download-build")
+        .description("Download a build from a URL")
+        .requiredOption("--url <url>", "Build URL to download")
+        .action(async (opts) => {
+        try {
+            await (0, test_build_1.downloadBuild)({
+                buildUrl: opts.url,
+                repoPath: process.cwd(),
+            });
+            console.log("✅ Build downloaded successfully");
+        }
+        catch (error) {
+            console.error("❌ Error downloading build:", error.message);
+            process.exit(1);
+        }
+        process.exit(0);
+    });
     program
         .command("chat-agent")
         .description("Run the chat agent")
@@ -310,6 +369,29 @@ async function main() {
         await (0, setup_1.runSetup)({ repoName: options.repoName });
         process.exit(0);
     });
+    program
+        .command("fetch-chat-state")
+        .description("Fetch chat session state from the dashboard")
+        .requiredOption("--id <id>", "Chat session ID")
+        .action(async (opts) => {
+        try {
+            const response = await api_client_1.apiClient.request(`/api/chat-sessions/${opts.id}`);
+            if (!response.ok) {
+                console.error("❌ Failed to fetch chat session:", response.statusText);
+                process.exit(1);
+            }
+            const responseData = await response.json();
+            const chatState = responseData.data.chat_session.chat_state;
+            const filename = `chat-session-${opts.id}.json`;
+            fs_1.default.writeFileSync(filename, JSON.stringify(chatState, null, 2));
+            console.log(`✅ Chat state saved to ${filename}`);
+        }
+        catch (error) {
+            console.error("❌ Error fetching chat session:", error.message);
+            process.exit(1);
+        }
+        process.exit(0);
+    });
     program
         .command("legacy")
         .description("Run the legacy workflows")
@@ -338,7 +420,6 @@ async function main() {
             await (0, test_build_1.downloadBuild)({
                 buildUrl: testGenConfig.build.url,
                 repoPath: process.cwd(),
-                apiKey: process.env.EMPIRICALRUN_API_KEY,
             });
         }
         let agentUsed;

package/dist/bin/utils/index.js

@@ -14,9 +14,9 @@ exports.ARGS_TO_MODEL_MAP = {
     "claude-4": "claude-sonnet-4-20250514",
     "claude-sonnet-4": "claude-sonnet-4-20250514",
     "claude-opus-4": "claude-opus-4-20250514",
-    "gemini-2.5": "gemini-2.5-pro-preview-06-05",
-    "gemini-2.5-pro": "gemini-2.5-pro-preview-06-05",
-    "gemini-2.5-pro-preview-03-25": "gemini-2.5-pro-preview-06-05",
+    gemini: "gemini-2.5-pro",
+    "gemini-2.5": "gemini-2.5-pro",
+    "gemini-2.5-pro": "gemini-2.5-pro",
     o3: "o3-2025-04-16",
     "o3-2025-04-16": "o3-2025-04-16",
     "o4-mini": "o4-mini-2025-04-16",

package/dist/test-build/index.d.ts

@@ -1,7 +1,6 @@
 export declare function hasDownloadScript(repoPath: string): Promise<boolean>;
-export declare function downloadBuild({ buildUrl, repoPath, apiKey, }: {
+export declare function downloadBuild({ buildUrl, repoPath, }: {
     buildUrl: string;
     repoPath: string;
-    apiKey: string | undefined;
 }): Promise<void>;
 //# sourceMappingURL=index.d.ts.map

package/dist/test-build/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/test-build/index.ts"],"names":[],"mappings":"AAgBA,wBAAsB,iBAAiB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAG1E;AAED,wBAAsB,aAAa,CAAC,EAClC,QAAQ,EACR,QAAQ,EACR,MAAM,GACP,EAAE;IACD,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC;CAC5B,GAAG,OAAO,CAAC,IAAI,CAAC,CAWhB"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/test-build/index.ts"],"names":[],"mappings":"AAgBA,wBAAsB,iBAAiB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAG1E;AAED,wBAAsB,aAAa,CAAC,EAClC,QAAQ,EACR,QAAQ,GACT,EAAE;IACD,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;CAClB,GAAG,OAAO,CAAC,IAAI,CAAC,CAWhB"}

package/dist/test-build/index.js

@@ -18,7 +18,7 @@ async function hasDownloadScript(repoPath) {
     const packageJSON = await getPackageJSON(repoPath);
     return !!packageJSON.scripts["download"];
 }
-async function downloadBuild({ buildUrl, repoPath, apiKey, }) {
+async function downloadBuild({ buildUrl, repoPath, }) {
     const logger = new logger_1.CustomLogger({ useReporter: false });
     const packageJSON = await getPackageJSON(repoPath);
     const buildDownloadScript = packageJSON.scripts["download"];
@@ -26,7 +26,7 @@ async function downloadBuild({ buildUrl, repoPath, apiKey, }) {
         logger.log(`Downloading build from ${buildUrl}`);
         await (0, exec_1.cmd)(`npm`, ["run", "download", buildUrl], {
             cwd: repoPath,
-            envOverrides: { ...Object(process.env), EMPIRICALRUN_API_KEY: apiKey },
+            envOverrides: { ...Object(process.env) },
         });
     }
 }

package/dist/tool-call-service/utils.d.ts

@@ -1,5 +1,6 @@
 import { PendingToolCall, ServicePayload, Tool } from "@empiricalrun/shared-types";
 export declare function appendBranchNameToQueueUrl(queueUrl: string, branchName: string): string;
+export declare function replaceRegionInUrl(url: string, newRegion: string): string;
 export declare function getQueueUrl(toolCalls: PendingToolCall[], tools: Tool[]): string;
 export declare function sendToolRequestToRemoteQueue(queueUrl: string, payload: ServicePayload): Promise<void>;
 //# sourceMappingURL=utils.d.ts.map

package/dist/tool-call-service/utils.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../src/tool-call-service/utils.ts"],"names":[],"mappings":"AACA,OAAO,EACL,eAAe,EACf,cAAc,EACd,IAAI,EACL,MAAM,4BAA4B,CAAC;AAEpC,wBAAgB,0BAA0B,CACxC,QAAQ,EAAE,MAAM,EAChB,UAAU,EAAE,MAAM,GACjB,MAAM,CAOR;AAYD,wBAAgB,WAAW,CACzB,SAAS,EAAE,eAAe,EAAE,EAC5B,KAAK,EAAE,IAAI,EAAE,GACZ,MAAM,CAeR;AAED,wBAAsB,4BAA4B,CAChD,QAAQ,EAAE,MAAM,EAChB,OAAO,EAAE,cAAc,iBAiBxB"}
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../src/tool-call-service/utils.ts"],"names":[],"mappings":"AACA,OAAO,EACL,eAAe,EACf,cAAc,EACd,IAAI,EACL,MAAM,4BAA4B,CAAC;AAEpC,wBAAgB,0BAA0B,CACxC,QAAQ,EAAE,MAAM,EAChB,UAAU,EAAE,MAAM,GACjB,MAAM,CAOR;AAED,wBAAgB,kBAAkB,CAAC,GAAG,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,MAAM,CAOzE;AAgBD,wBAAgB,WAAW,CACzB,SAAS,EAAE,eAAe,EAAE,EAC5B,KAAK,EAAE,IAAI,EAAE,GACZ,MAAM,CAeR;AAED,wBAAsB,4BAA4B,CAChD,QAAQ,EAAE,MAAM,EAChB,OAAO,EAAE,cAAc,iBAiBxB"}

package/dist/tool-call-service/utils.js

@@ -1,6 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.appendBranchNameToQueueUrl = appendBranchNameToQueueUrl;
+exports.replaceRegionInUrl = replaceRegionInUrl;
 exports.getQueueUrl = getQueueUrl;
 exports.sendToolRequestToRemoteQueue = sendToolRequestToRemoteQueue;
 const client_sqs_1 = require("@aws-sdk/client-sqs");
@@ -12,12 +13,18 @@ function appendBranchNameToQueueUrl(queueUrl, branchName) {
     const queueNameWithoutFifo = queueName?.replace(/\.fifo$/, "");
     return `${parts.join("/")}/${queueNameWithoutFifo}-${suffix}.fifo`;
 }
+function replaceRegionInUrl(url, newRegion) {
+    const parsedUrl = new URL(url);
+    parsedUrl.hostname = parsedUrl.hostname.replace(/\.([a-z0-9-]+)\.amazonaws\.com$/, `.${newRegion}.amazonaws.com`);
+    return parsedUrl.toString();
+}
 function buildQueueUrlsForPreview(baseQueueUrl) {
     const branch = process.env.VERCEL_GIT_COMMIT_REF;
     if (!branch) {
         throw new Error("VERCEL_GIT_COMMIT_REF is not set");
     }
-    const queueUrl = appendBranchNameToQueueUrl(baseQueueUrl, branch);
+    const updatedRegionForQueue = replaceRegionInUrl(baseQueueUrl, process.env.AWS_REGION);
+    const queueUrl = appendBranchNameToQueueUrl(updatedRegionForQueue, branch);
     console.log("queueUrl for preview", queueUrl);
     return queueUrl;
 }

package/dist/tools/download-build.js

@@ -17,7 +17,7 @@ Environment details will include the build URL.`,
     needsBrowser: false,
     execute: async (params) => {
         const { input } = params;
-        const { repoPath, apiKey } = params;
+        const { repoPath } = params;
         if (!(await (0, test_build_1.hasDownloadScript)(repoPath))) {
             return {
                 isError: true,
@@ -27,7 +27,7 @@ You probably don't need to worry about this, since it means this repo does not h
         }
         const { buildUrl } = input;
         try {
-            await (0, test_build_1.downloadBuild)({ buildUrl, repoPath, apiKey });
+            await (0, test_build_1.downloadBuild)({ buildUrl, repoPath });
             return {
                 isError: false,
                 result: "Build downloaded successfully",

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":"AACA,OAAO,KAAK,EACV,IAAI,EAGL,MAAM,4BAA4B,CAAC;AAoGpC,eAAO,MAAM,4BAA4B,EAAE,IAoM1C,CAAC"}
+{"version":3,"file":"test-gen-browser.d.ts","sourceRoot":"","sources":["../../src/tools/test-gen-browser.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACV,IAAI,EAGL,MAAM,4BAA4B,CAAC;AAoGpC,eAAO,MAAM,4BAA4B,EAAE,IA+L1C,CAAC"}

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

@@ -153,21 +153,17 @@ exports.generateTestWithBrowserAgent = {
         catch (err) {
             console.warn("[generateTestWithBrowserAgent] Error preparing playwright for codegen", err);
         }
-        let envOverrides = { ...environmentOverrides };
+        let envOverrides = {
+            ...environmentOverrides,
+            // Running browser agent in headed implies using page.pause method for codegen
+            RUN_PLAYWRIGHT_HEADED: "true",
+        };
         if (input.buildUrl) {
             envOverrides = {
                 ...envOverrides,
                 BUILD_URL: input.buildUrl,
             };
         }
-        if (featureFlags.includes("headed_codegen_in_browser_agent")) {
-            // Running browser agent in headed implies using page.pause method for codegen
-            // which improves codegen quality when script injection fails (iframes, etc.)
-            envOverrides = {
-                ...envOverrides,
-                RUN_PLAYWRIGHT_HEADED: "true",
-            };
-        }
         const testGenToken = (0, scenarios_1.buildTokenFromOptions)({ name: testName, file: filePath, prompt: input.changeToMake }, { useComputerUseAgent: true });
         console.log("[generateTestWithBrowserAgent] Validations passed, starting agent");
         const agentResult = await (0, run_1.runBrowsingAgent)({

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@empiricalrun/test-gen",
-  "version": "0.69.5",
+  "version": "0.69.7",
   "publishConfig": {
     "registry": "https://registry.npmjs.org/",
     "access": "public"
@@ -64,7 +64,7 @@
     "tsx": "^4.16.2",
     "typescript": "^5.3.3",
     "zod": "^3.23.8",
-    "@empiricalrun/llm": "^0.19.1",
+    "@empiricalrun/llm": "^0.19.3",
     "@empiricalrun/r2-uploader": "^0.3.9",
     "@empiricalrun/test-run": "^0.10.6"
   },
@@ -82,7 +82,7 @@
     "playwright": "1.53.0",
     "serve-handler": "^6.1.6",
     "ts-patch": "^3.3.0",
-    "@empiricalrun/shared-types": "0.6.0"
+    "@empiricalrun/shared-types": "0.6.1"
   },
   "scripts": {
     "dev": "tspc --build --watch",

package/tsconfig.tsbuildinfo

@@ -1 +1 @@
-{"root":["./src/index.ts","./src/logger.ts","./src/actions/assert.ts","./src/actions/click.ts","./src/actions/done.ts","./src/actions/fill.ts","./src/actions/goto.ts","./src/actions/hover.ts","./src/actions/index.ts","./src/actions/next-task.ts","./src/actions/press.ts","./src/actions/skill.ts","./src/actions/text-content.ts","./src/actions/constants/index.ts","./src/actions/utils/index.ts","./src/agent/browsing/index.ts","./src/agent/browsing/run.ts","./src/agent/browsing/utils.ts","./src/agent/chat/agent-loop.ts","./src/agent/chat/exports.ts","./src/agent/chat/index.ts","./src/agent/chat/models.ts","./src/agent/chat/state.ts","./src/agent/chat/types.ts","./src/agent/chat/utils.ts","./src/agent/chat/prompt/index.ts","./src/agent/chat/prompt/pw-utils-docs.ts","./src/agent/chat/prompt/repo.ts","./src/agent/codegen/create-test-block.ts","./src/agent/codegen/fix-ts-errors.ts","./src/agent/codegen/generate-code-apply-changes.ts","./src/agent/codegen/lexical-scoped-vars.ts","./src/agent/codegen/repo-edit.ts","./src/agent/codegen/run.ts","./src/agent/codegen/skills-retriever.ts","./src/agent/codegen/test-update-feedback.ts","./src/agent/codegen/types.ts","./src/agent/codegen/update-flow.ts","./src/agent/codegen/use-skill.ts","./src/agent/codegen/utils.ts","./src/agent/cua/computer.ts","./src/agent/cua/index.ts","./src/agent/cua/model.ts","./src/agent/cua/pw-codegen/element-from-point.ts","./src/agent/cua/pw-codegen/types.ts","./src/agent/cua/pw-codegen/pw-pause/for-recorder.ts","./src/agent/cua/pw-codegen/pw-pause/index.ts","./src/agent/cua/pw-codegen/pw-pause/ipc.ts","./src/agent/cua/pw-codegen/pw-pause/patch.ts","./src/agent/cua/pw-codegen/pw-pause/types.ts","./src/agent/diagnosis-agent/index.ts","./src/agent/diagnosis-agent/strict-mode-violation.ts","./src/agent/enrich-prompt/index.ts","./src/agent/enrich-prompt/utils.ts","./src/agent/infer-agent/index.ts","./src/agent/master/action-tool-calls.ts","./src/agent/master/element-annotation.ts","./src/agent/master/execute-browser-action.ts","./src/agent/master/execute-skill-action.ts","./src/agent/master/next-action.ts","./src/agent/master/planner.ts","./src/agent/master/run.ts","./src/agent/master/scroller.ts","./src/agent/master/with-hints.ts","./src/agent/master/browser-tests/cua.spec.ts","./src/agent/master/browser-tests/fixtures.ts","./src/agent/master/browser-tests/index.spec.ts","./src/agent/master/browser-tests/skills.spec.ts","./src/agent/master/icon-descriptor/index.ts","./src/agent/master/icon-descriptor/normalize-svg.ts","./src/agent/planner/run-time-planner.ts","./src/agent/planner/run.ts","./src/artifacts/index.ts","./src/artifacts/utils.ts","./src/auth/api-client.ts","./src/auth/cli-auth.ts","./src/auth/index.ts","./src/auth/token-store.ts","./src/bin/index.ts","./src/bin/setup.ts","./src/bin/logger/index.ts","./src/bin/utils/context.ts","./src/bin/utils/index.ts","./src/bin/utils/fs/index.ts","./src/bin/utils/platform/web/index.ts","./src/bin/utils/platform/web/test-files/ts-path-import-validate.ts","./src/bin/utils/scenarios/index.ts","./src/browser-injected-scripts/annotate-elements.spec.ts","./src/constants/index.ts","./src/errors/index.ts","./src/evals/add-scenario-agent.evals.ts","./src/evals/append-create-test-agent.evals.ts","./src/evals/fetch-pom-skills-agent.evals.ts","./src/evals/infer-master-or-code-agent.evals.ts","./src/evals/master-agent.evals.ts","./src/evals/type.ts","./src/evals/update-scenario-agent.evals.ts","./src/file/client.ts","./src/file/server.ts","./src/human-in-the-loop/cli.ts","./src/human-in-the-loop/index.ts","./src/human-in-the-loop/ipc.ts","./src/page/index.ts","./src/prompts/lib/ts-transformer.ts","./src/recorder/env-variables.ts","./src/recorder/glob.ts","./src/recorder/index.ts","./src/recorder/request.ts","./src/recorder/temp-files.ts","./src/recorder/upload.ts","./src/recorder/validation.ts","./src/reporter/index.ts","./src/reporter/lib.ts","./src/session/index.ts","./src/test-build/index.ts","./src/tool-call-service/index.ts","./src/tool-call-service/utils.ts","./src/tools/commit-and-create-pr.ts","./src/tools/diagnosis-fetcher.ts","./src/tools/download-build.ts","./src/tools/list-environments.ts","./src/tools/str_replace_editor.ts","./src/tools/test-gen-browser.ts","./src/tools/test-run.ts","./src/tools/grep/index.ts","./src/tools/grep/ripgrep/index.ts","./src/tools/grep/ripgrep/types.ts","./src/tools/test-run-fetcher/index.ts","./src/tools/test-run-fetcher/types.ts","./src/tools/upgrade-packages/index.ts","./src/tools/upgrade-packages/utils.ts","./src/tools/utils/index.ts","./src/types/handlebars.d.ts","./src/types/index.ts","./src/uploader/index.ts","./src/uploader/utils.ts","./src/utils/checkpoint.ts","./src/utils/env.ts","./src/utils/exec.ts","./src/utils/file-tree.ts","./src/utils/file.ts","./src/utils/git.ts","./src/utils/html.ts","./src/utils/index.ts","./src/utils/json.ts","./src/utils/repo-tree.ts","./src/utils/slug.ts","./src/utils/string.ts","./src/utils/stripAnsi.ts"],"version":"5.8.3"}
+{"root":["./src/index.ts","./src/logger.ts","./src/actions/assert.ts","./src/actions/click.ts","./src/actions/done.ts","./src/actions/fill.ts","./src/actions/goto.ts","./src/actions/hover.ts","./src/actions/index.ts","./src/actions/next-task.ts","./src/actions/press.ts","./src/actions/skill.ts","./src/actions/text-content.ts","./src/actions/constants/index.ts","./src/actions/utils/index.ts","./src/agent/browsing/index.ts","./src/agent/browsing/run.ts","./src/agent/browsing/utils.ts","./src/agent/chat/agent-loop.ts","./src/agent/chat/exports.ts","./src/agent/chat/index.ts","./src/agent/chat/models.ts","./src/agent/chat/state.ts","./src/agent/chat/types.ts","./src/agent/chat/utils.ts","./src/agent/chat/prompt/index.ts","./src/agent/chat/prompt/pw-utils-docs.ts","./src/agent/chat/prompt/repo.ts","./src/agent/codegen/create-test-block.ts","./src/agent/codegen/fix-ts-errors.ts","./src/agent/codegen/generate-code-apply-changes.ts","./src/agent/codegen/lexical-scoped-vars.ts","./src/agent/codegen/repo-edit.ts","./src/agent/codegen/run.ts","./src/agent/codegen/skills-retriever.ts","./src/agent/codegen/test-update-feedback.ts","./src/agent/codegen/types.ts","./src/agent/codegen/update-flow.ts","./src/agent/codegen/use-skill.ts","./src/agent/codegen/utils.ts","./src/agent/cua/computer.ts","./src/agent/cua/index.ts","./src/agent/cua/model.ts","./src/agent/cua/pw-codegen/element-from-point.ts","./src/agent/cua/pw-codegen/types.ts","./src/agent/cua/pw-codegen/pw-pause/for-recorder.ts","./src/agent/cua/pw-codegen/pw-pause/index.ts","./src/agent/cua/pw-codegen/pw-pause/ipc.ts","./src/agent/cua/pw-codegen/pw-pause/patch.ts","./src/agent/cua/pw-codegen/pw-pause/types.ts","./src/agent/cua/pw-codegen/pw-pause/utils.ts","./src/agent/diagnosis-agent/index.ts","./src/agent/diagnosis-agent/strict-mode-violation.ts","./src/agent/enrich-prompt/index.ts","./src/agent/enrich-prompt/utils.ts","./src/agent/infer-agent/index.ts","./src/agent/master/action-tool-calls.ts","./src/agent/master/element-annotation.ts","./src/agent/master/execute-browser-action.ts","./src/agent/master/execute-skill-action.ts","./src/agent/master/next-action.ts","./src/agent/master/planner.ts","./src/agent/master/run.ts","./src/agent/master/scroller.ts","./src/agent/master/with-hints.ts","./src/agent/master/browser-tests/cua.spec.ts","./src/agent/master/browser-tests/fixtures.ts","./src/agent/master/browser-tests/index.spec.ts","./src/agent/master/browser-tests/skills.spec.ts","./src/agent/master/icon-descriptor/index.ts","./src/agent/master/icon-descriptor/normalize-svg.ts","./src/agent/planner/run-time-planner.ts","./src/agent/planner/run.ts","./src/artifacts/index.ts","./src/artifacts/utils.ts","./src/auth/api-client.ts","./src/auth/cli-auth.ts","./src/auth/index.ts","./src/auth/token-store.ts","./src/bin/index.ts","./src/bin/setup.ts","./src/bin/logger/index.ts","./src/bin/utils/context.ts","./src/bin/utils/index.ts","./src/bin/utils/fs/index.ts","./src/bin/utils/platform/web/index.ts","./src/bin/utils/platform/web/test-files/ts-path-import-validate.ts","./src/bin/utils/scenarios/index.ts","./src/browser-injected-scripts/annotate-elements.spec.ts","./src/constants/index.ts","./src/errors/index.ts","./src/evals/add-scenario-agent.evals.ts","./src/evals/append-create-test-agent.evals.ts","./src/evals/fetch-pom-skills-agent.evals.ts","./src/evals/infer-master-or-code-agent.evals.ts","./src/evals/master-agent.evals.ts","./src/evals/type.ts","./src/evals/update-scenario-agent.evals.ts","./src/file/client.ts","./src/file/server.ts","./src/human-in-the-loop/cli.ts","./src/human-in-the-loop/index.ts","./src/human-in-the-loop/ipc.ts","./src/page/index.ts","./src/prompts/lib/ts-transformer.ts","./src/recorder/env-variables.ts","./src/recorder/glob.ts","./src/recorder/index.ts","./src/recorder/request.ts","./src/recorder/temp-files.ts","./src/recorder/upload.ts","./src/recorder/validation.ts","./src/reporter/index.ts","./src/reporter/lib.ts","./src/session/index.ts","./src/test-build/index.ts","./src/tool-call-service/index.ts","./src/tool-call-service/utils.ts","./src/tools/commit-and-create-pr.ts","./src/tools/diagnosis-fetcher.ts","./src/tools/download-build.ts","./src/tools/list-environments.ts","./src/tools/str_replace_editor.ts","./src/tools/test-gen-browser.ts","./src/tools/test-run.ts","./src/tools/grep/index.ts","./src/tools/grep/ripgrep/index.ts","./src/tools/grep/ripgrep/types.ts","./src/tools/test-run-fetcher/index.ts","./src/tools/test-run-fetcher/types.ts","./src/tools/upgrade-packages/index.ts","./src/tools/upgrade-packages/utils.ts","./src/tools/utils/index.ts","./src/types/handlebars.d.ts","./src/types/index.ts","./src/uploader/index.ts","./src/uploader/utils.ts","./src/utils/checkpoint.ts","./src/utils/env.ts","./src/utils/exec.ts","./src/utils/file-tree.ts","./src/utils/file.ts","./src/utils/git.ts","./src/utils/html.ts","./src/utils/index.ts","./src/utils/json.ts","./src/utils/repo-tree.ts","./src/utils/slug.ts","./src/utils/string.ts","./src/utils/stripAnsi.ts"],"version":"5.8.3"}