@empiricalrun/test-gen 0.57.2 → 0.59.0

package/dist/index.d.ts

@@ -1,4 +1,5 @@
 import { FrameLocator, Page } from "playwright";
 import { ScopeVars } from "./types";
+export { downloadBuild } from "./test-build";
 export declare function createTest(task: string, pageRef: Page | FrameLocator, scope?: ScopeVars): Promise<void>;
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAQhD,OAAO,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AAoBpC,wBAAsB,UAAU,CAC9B,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,IAAI,GAAG,YAAY,EAC5B,KAAK,CAAC,EAAE,SAAS,iBA0ElB"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAQhD,OAAO,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AACpC,OAAO,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAoB7C,wBAAsB,UAAU,CAC9B,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,IAAI,GAAG,YAAY,EAC5B,KAAK,CAAC,EAAE,SAAS,iBA0ElB"}

package/dist/index.js

@@ -3,7 +3,8 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.createTest = void 0;
+exports.downloadBuild = void 0;
+exports.createTest = createTest;
 const llm_1 = require("@empiricalrun/llm");
 const cua_1 = require("./agent/cua");
 const run_1 = require("./agent/master/run");
@@ -11,6 +12,8 @@ const scenarios_1 = require("./bin/utils/scenarios");
 const client_1 = __importDefault(require("./file/client"));
 const reporter_1 = require("./reporter");
 const session_1 = require("./session");
+var test_build_1 = require("./test-build");
+Object.defineProperty(exports, "downloadBuild", { enumerable: true, get: function () { return test_build_1.downloadBuild; } });
 const flushEvents = async () => {
     await (0, llm_1.flushAllTraces)();
 };
@@ -95,4 +98,3 @@ async function createTest(task, pageRef, scope) {
         await flushEvents();
     }
 }
-exports.createTest = createTest;

package/dist/prompts/lib/ts-transformer.js

@@ -15,13 +15,23 @@ var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (
 }) : function(o, v) {
     o["default"] = v;
 });
-var __importStar = (this && this.__importStar) || function (mod) {
-    if (mod && mod.__esModule) return mod;
-    var result = {};
-    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
-    __setModuleDefault(result, mod);
-    return result;
-};
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
 Object.defineProperty(exports, "__esModule", { value: true });
 const fs = __importStar(require("fs"));
 const path = __importStar(require("path"));

package/dist/reporter/index.js

@@ -3,7 +3,9 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.TestGenUpdatesReporter = exports.setReporterConfig = exports.getReporter = void 0;
+exports.TestGenUpdatesReporter = void 0;
+exports.getReporter = getReporter;
+exports.setReporterConfig = setReporterConfig;
 const r2_uploader_1 = require("@empiricalrun/r2-uploader");
 const fs_1 = __importDefault(require("fs"));
 const path_1 = __importDefault(require("path"));
@@ -23,7 +25,6 @@ function getReporter() {
     }
     return reporterInstance;
 }
-exports.getReporter = getReporter;
 /**
  * function will upload videos and json summary of test results to r2 and report them to reporter.
  * method won't throw error if it fails to report
@@ -31,7 +32,6 @@ exports.getReporter = getReporter;
 function setReporterConfig(config) {
     reporterConfig = config;
 }
-exports.setReporterConfig = setReporterConfig;
 class TestGenUpdatesReporter {
     repoDir;
     constructor() {

package/dist/reporter/lib.js

@@ -58,7 +58,7 @@ class Reporter {
             }, {
                 retries: 3,
                 minTimeout: 1000,
-                maxTimeout: 60000,
+                maxTimeout: 60_000,
                 factor: 3,
             });
         }

package/dist/session/index.js

@@ -3,7 +3,12 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.updateSessionStatus = exports.getSessionDetails = exports.endSession = exports.getSessionState = exports.shouldStopSession = exports.setSessionDetails = void 0;
+exports.setSessionDetails = setSessionDetails;
+exports.shouldStopSession = shouldStopSession;
+exports.getSessionState = getSessionState;
+exports.endSession = endSession;
+exports.getSessionDetails = getSessionDetails;
+exports.updateSessionStatus = updateSessionStatus;
 const package_json_1 = __importDefault(require("../../package.json"));
 const env_1 = require("../utils/env");
 const sessionDetails = {
@@ -22,19 +27,16 @@ function getSessionDetails() {
         version: package_json_1.default.version,
     };
 }
-exports.getSessionDetails = getSessionDetails;
 function setSessionDetails({ sessionId, generationId, testCaseId, projectRepoName, }) {
     sessionDetails.sessionId = sessionId;
     sessionDetails.generationId = generationId;
     sessionDetails.testUrl = `${process.env.DASHBOARD_DOMAIN ?? ""}/${projectRepoName}/test-cases/${testCaseId}`;
 }
-exports.setSessionDetails = setSessionDetails;
 async function shouldStopSession() {
     const sessionState = await getSessionState();
     const shouldStopSession = sessionState !== "started";
     return shouldStopSession;
 }
-exports.shouldStopSession = shouldStopSession;
 async function getSessionState() {
     // if there is no dashboard domain to connect to or session details missing, then continue the generation session.
     if (!DASHBOARD_DOMAIN ||
@@ -54,7 +56,6 @@ async function getSessionState() {
     // TODO: fix this heirarchy
     return generationStateData.data.state.state;
 }
-exports.getSessionState = getSessionState;
 async function endSession() {
     if (!DASHBOARD_DOMAIN) {
         return;
@@ -82,7 +83,6 @@ async function endSession() {
         console.error("Failed to end session:", error);
     }
 }
-exports.endSession = endSession;
 async function updateSessionStatus(sessionId, payload) {
     if (!DASHBOARD_DOMAIN) {
         return;
@@ -102,4 +102,3 @@ async function updateSessionStatus(sessionId, payload) {
         console.error("Failed to update session status:", error);
     }
 }
-exports.updateSessionStatus = updateSessionStatus;

package/dist/test-build/index.js

@@ -3,7 +3,8 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.downloadBuild = exports.hasDownloadScript = void 0;
+exports.hasDownloadScript = hasDownloadScript;
+exports.downloadBuild = downloadBuild;
 const fs_1 = __importDefault(require("fs"));
 const logger_1 = require("../bin/logger");
 const exec_1 = require("../utils/exec");
@@ -16,16 +17,14 @@ async function hasDownloadScript() {
     const packageJSON = await getPackageJSON();
     return !!packageJSON.scripts["download"];
 }
-exports.hasDownloadScript = hasDownloadScript;
 async function downloadBuild(buildUrl) {
     const logger = new logger_1.CustomLogger({ useReporter: false });
     const packageJSON = await getPackageJSON();
     const buildDownloadScript = packageJSON.scripts["download"];
     if (buildDownloadScript && buildUrl) {
         logger.log(`Downloading build from ${buildUrl}`);
-        await (0, exec_1.cmd)(`npm run download ${buildUrl}`.split(" "), {
+        await (0, exec_1.cmd)(`npm`, ["run", "download", buildUrl], {
             env: { ...Object(process.env) },
         });
     }
 }
-exports.downloadBuild = downloadBuild;

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

@@ -1,6 +1,5 @@
 import { TraceClient } from "@empiricalrun/llm";
-import { PendingToolCall, Tool, ToolResult } from "@empiricalrun/llm/chat";
-import { SupportedChatModels } from "../agent/chat/types";
+import { PendingToolCall, SupportedChatModels, Tool, ToolResult } from "@empiricalrun/llm/chat";
 export type { SupportedChatModels };
 type ToolExecutors = {
     [key: string]: (input: any, trace?: TraceClient) => Promise<ToolResult>;
@@ -15,6 +14,7 @@ export declare class ToolCallService {
     getTools(): Promise<{
         tools: Tool[];
     }>;
-    execute(toolCalls: PendingToolCall[], isRemote: boolean, trace?: TraceClient): Promise<ToolResult[]>;
+    sendToQueue(toolCalls: PendingToolCall[]): Promise<void>;
+    execute(toolCalls: PendingToolCall[], trace?: TraceClient): Promise<ToolResult[]>;
 }
 //# sourceMappingURL=index.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/tool-call-service/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAE,eAAe,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AAE3E,OAAO,EAAE,mBAAmB,EAAE,MAAM,qBAAqB,CAAC;AAe1D,YAAY,EAAE,mBAAmB,EAAE,CAAC;AAEpC,KAAK,aAAa,GAAG;IACnB,CAAC,GAAG,EAAE,MAAM,GAAG,CAAC,KAAK,EAAE,GAAG,EAAE,KAAK,CAAC,EAAE,WAAW,KAAK,OAAO,CAAC,UAAU,CAAC,CAAC;CACzE,CAAC;AA6BF,qBAAa,eAAe;IAC1B,KAAK,EAAE,IAAI,EAAE,CAAM;IACnB,aAAa,EAAE,aAAa,CAAM;IAClC,aAAa,EAAE,MAAM,GAAG,IAAI,CAAC;IAC7B,aAAa,EAAE,mBAAmB,CAAC;IACnC,UAAU,EAAE,MAAM,CAAC;gBAEjB,aAAa,EAAE,MAAM,GAAG,IAAI,EAC5B,aAAa,EAAE,mBAAmB,EAClC,UAAU,EAAE,MAAM;IAiBd,QAAQ;;;IAaR,OAAO,CACX,SAAS,EAAE,eAAe,EAAE,EAC5B,QAAQ,EAAE,OAAO,EACjB,KAAK,CAAC,EAAE,WAAW,GAClB,OAAO,CAAC,UAAU,EAAE,CAAC;CAyDzB"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/tool-call-service/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAEL,eAAe,EACf,mBAAmB,EACnB,IAAI,EACJ,UAAU,EACX,MAAM,wBAAwB,CAAC;AAiBhC,YAAY,EAAE,mBAAmB,EAAE,CAAC;AAEpC,KAAK,aAAa,GAAG;IACnB,CAAC,GAAG,EAAE,MAAM,GAAG,CAAC,KAAK,EAAE,GAAG,EAAE,KAAK,CAAC,EAAE,WAAW,KAAK,OAAO,CAAC,UAAU,CAAC,CAAC;CACzE,CAAC;AAEF,qBAAa,eAAe;IAC1B,KAAK,EAAE,IAAI,EAAE,CAAM;IACnB,aAAa,EAAE,aAAa,CAAM;IAClC,aAAa,EAAE,MAAM,GAAG,IAAI,CAAC;IAC7B,aAAa,EAAE,mBAAmB,CAAC;IACnC,UAAU,EAAE,MAAM,CAAC;gBAEjB,aAAa,EAAE,MAAM,GAAG,IAAI,EAC5B,aAAa,EAAE,mBAAmB,EAClC,UAAU,EAAE,MAAM;IAiBd,QAAQ;;;IAaR,WAAW,CAAC,SAAS,EAAE,eAAe,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAcxD,OAAO,CACX,SAAS,EAAE,eAAe,EAAE,EAC5B,KAAK,CAAC,EAAE,WAAW,GAClB,OAAO,CAAC,UAAU,EAAE,CAAC;CA2CzB"}

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

@@ -1,7 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.ToolCallService = void 0;
-const client_sqs_1 = require("@aws-sdk/client-sqs");
+const chat_1 = require("@empiricalrun/llm/chat");
 const commit_and_create_pr_1 = require("../tools/commit-and-create-pr");
 const diagnosis_fetcher_1 = require("../tools/diagnosis-fetcher");
 const download_build_1 = require("../tools/download-build");
@@ -12,25 +12,7 @@ const test_gen_browser_1 = require("../tools/test-gen-browser");
 const test_run_1 = require("../tools/test-run");
 const test_run_fetcher_1 = require("../tools/test-run-fetcher");
 const checkpoint_1 = require("../utils/checkpoint");
-async function sendToolRequestToRemoteQueue(payload) {
-    const sqs = new client_sqs_1.SQSClient({
-        region: process.env.AWS_REGION,
-        credentials: {
-            accessKeyId: process.env.AWS_ACCESS_KEY_ID,
-            secretAccessKey: process.env.AWS_SECRET_ACCESS_KEY,
-        },
-    });
-    const queueUrl = process.env.TOOL_EXECUTION_SQS_URL;
-    if (!queueUrl) {
-        throw new Error("TOOL_EXECUTION_SQS_URL is required for remote execution.");
-    }
-    await sqs.send(new client_sqs_1.SendMessageCommand({
-        QueueUrl: queueUrl,
-        MessageBody: JSON.stringify(payload),
-        MessageGroupId: payload.requestId,
-        MessageDeduplicationId: payload.requestId, // unique id for the tool request
-    }));
-}
+const utils_1 = require("./utils");
 class ToolCallService {
     tools = [];
     toolExecutors = {};
@@ -53,70 +35,68 @@ class ToolCallService {
         ];
     }
     async getTools() {
-        if (!this.selectedModel.startsWith("claude")) {
+        if ((0, chat_1.getProviderForModel)(this.selectedModel) !== "claude") {
             this.tools.push(...str_replace_editor_1.textEditorTools);
         }
         this.tools.forEach((tool) => {
             this.toolExecutors[tool.schema.name] = tool.execute;
         });
-        if (this.selectedModel.startsWith("claude")) {
+        if ((0, chat_1.getProviderForModel)(this.selectedModel) === "claude") {
             this.toolExecutors["str_replace_editor"] = str_replace_editor_1.strReplaceEditorExecutor;
         }
         return { tools: this.tools };
     }
-    async execute(toolCalls, isRemote, trace) {
-        if (isRemote && this.chatSessionId) {
-            await sendToolRequestToRemoteQueue({
-                toolCalls,
-                requestId: crypto.randomUUID(),
-                chatSessionId: this.chatSessionId,
-                selectedModel: this.selectedModel,
-                branchName: this.branchName,
-            });
-            return toolCalls.map(() => ({
-                isError: false,
-                result: `Tool request sent to remote queue to execute.`,
-            }));
+    async sendToQueue(toolCalls) {
+        const requestId = toolCalls[0]?.id;
+        if (!requestId) {
+            throw new Error("Could not find an id for the tool call.");
         }
-        else {
-            const executeSpan = trace?.span({
-                name: "execute_tools",
-                input: { toolCalls: toolCalls.map((tc) => ({ name: tc.name })) },
+        await (0, utils_1.sendToolRequestToRemoteQueue)({
+            toolCalls,
+            requestId,
+            chatSessionId: this.chatSessionId,
+            selectedModel: this.selectedModel,
+            branchName: this.branchName,
+        });
+    }
+    async execute(toolCalls, trace) {
+        const executeSpan = trace?.span({
+            name: "execute_tools",
+            input: { toolCalls: toolCalls.map((tc) => ({ name: tc.name })) },
+        });
+        const toolResults = [];
+        for (const toolCall of toolCalls) {
+            const span = executeSpan?.span({
+                name: `tool: ${toolCall.name}`,
+                input: toolCall.input,
             });
-            const toolResults = [];
-            for (const toolCall of toolCalls) {
-                const span = executeSpan?.span({
-                    name: `tool: ${toolCall.name}`,
-                    input: toolCall.input,
-                });
-                const toolExecutor = this.toolExecutors[toolCall.name];
-                if (!toolExecutor) {
-                    const errorResult = {
-                        isError: true,
-                        result: `Invalid function/tool call: invalid_tool_call not found`,
-                    };
-                    toolResults.push(errorResult);
-                    span?.end({ output: errorResult });
-                    continue;
-                }
-                try {
-                    const result = await toolExecutor(toolCall.input, trace);
-                    toolResults.push(result);
-                    span?.end({ output: result });
-                }
-                catch (error) {
-                    const errorResult = {
-                        isError: true,
-                        result: error instanceof Error ? error.message : String(error),
-                    };
-                    toolResults.push(errorResult);
-                    span?.end({ output: errorResult });
-                }
+            const toolExecutor = this.toolExecutors[toolCall.name];
+            if (!toolExecutor) {
+                const errorResult = {
+                    isError: true,
+                    result: `Invalid function/tool call: ${toolCall.name} not found`,
+                };
+                toolResults.push(errorResult);
+                span?.end({ output: errorResult });
+                continue;
+            }
+            try {
+                const result = await toolExecutor(toolCall.input, trace);
+                toolResults.push(result);
+                span?.end({ output: result });
+            }
+            catch (error) {
+                const errorResult = {
+                    isError: true,
+                    result: error instanceof Error ? error.message : String(error),
+                };
+                toolResults.push(errorResult);
+                span?.end({ output: errorResult });
             }
-            await (0, checkpoint_1.createCheckpoint)(toolCalls, this.branchName);
-            executeSpan?.end({ output: { toolResults } });
-            return toolResults;
         }
+        await (0, checkpoint_1.createCheckpoint)(toolCalls, this.branchName);
+        executeSpan?.end({ output: { toolResults } });
+        return toolResults;
     }
 }
 exports.ToolCallService = ToolCallService;

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

@@ -0,0 +1,10 @@
+import { SupportedChatModels } from "@empiricalrun/llm/chat";
+import { PendingToolCall } from "@empiricalrun/shared-types";
+export declare function sendToolRequestToRemoteQueue(payload: {
+    toolCalls: PendingToolCall[];
+    requestId: string;
+    chatSessionId: number;
+    selectedModel: SupportedChatModels;
+    branchName: string;
+}): Promise<void>;
+//# sourceMappingURL=utils.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../src/tool-call-service/utils.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,mBAAmB,EAAE,MAAM,wBAAwB,CAAC;AAC7D,OAAO,EAAE,eAAe,EAAE,MAAM,4BAA4B,CAAC;AAE7D,wBAAsB,4BAA4B,CAAC,OAAO,EAAE;IAC1D,SAAS,EAAE,eAAe,EAAE,CAAC;IAC7B,SAAS,EAAE,MAAM,CAAC;IAClB,aAAa,EAAE,MAAM,CAAC;IACtB,aAAa,EAAE,mBAAmB,CAAC;IACnC,UAAU,EAAE,MAAM,CAAC;CACpB,iBAoBA"}

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

@@ -0,0 +1,23 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.sendToolRequestToRemoteQueue = sendToolRequestToRemoteQueue;
+const client_sqs_1 = require("@aws-sdk/client-sqs");
+async function sendToolRequestToRemoteQueue(payload) {
+    const sqs = new client_sqs_1.SQSClient({
+        region: process.env.AWS_REGION,
+        credentials: {
+            accessKeyId: process.env.AWS_ACCESS_KEY_ID,
+            secretAccessKey: process.env.AWS_SECRET_ACCESS_KEY,
+        },
+    });
+    const queueUrl = process.env.TOOL_EXECUTION_SQS_URL;
+    if (!queueUrl) {
+        throw new Error("TOOL_EXECUTION_SQS_URL is required for remote execution.");
+    }
+    await sqs.send(new client_sqs_1.SendMessageCommand({
+        QueueUrl: queueUrl,
+        MessageBody: JSON.stringify(payload),
+        MessageGroupId: payload.requestId,
+        MessageDeduplicationId: payload.requestId, // unique id for the tool request
+    }));
+}

package/dist/tools/commit-and-create-pr.js

@@ -1,6 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.createPullRequestTool = exports.parseGitHubUrl = void 0;
+exports.createPullRequestTool = void 0;
+exports.parseGitHubUrl = parseGitHubUrl;
 const child_process_1 = require("child_process");
 const zod_1 = require("zod");
 const git_1 = require("../utils/git");
@@ -19,7 +20,6 @@ function parseGitHubUrl(url) {
     }
     return { owner, repo };
 }
-exports.parseGitHubUrl = parseGitHubUrl;
 const createPullRequestSchema = zod_1.z.object({
     pullRequestTitle: zod_1.z
         .string()

package/dist/tools/download-build.d.ts

@@ -1,3 +1,12 @@
 import type { Tool } from "@empiricalrun/llm/chat";
+import { z } from "zod";
+export declare const downloadBuildToolSchema: z.ZodObject<{
+    buildUrl: z.ZodString;
+}, "strip", z.ZodTypeAny, {
+    buildUrl: string;
+}, {
+    buildUrl: string;
+}>;
+export type DownloadBuildToolInput = z.infer<typeof downloadBuildToolSchema>;
 export declare const downloadBuildTool: Tool;
 //# sourceMappingURL=download-build.d.ts.map

package/dist/tools/download-build.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"download-build.d.ts","sourceRoot":"","sources":["../../src/tools/download-build.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,wBAAwB,CAAC;AAKnD,eAAO,MAAM,iBAAiB,EAAE,IAkC/B,CAAC"}
+{"version":3,"file":"download-build.d.ts","sourceRoot":"","sources":["../../src/tools/download-build.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,wBAAwB,CAAC;AACnD,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAIxB,eAAO,MAAM,uBAAuB;;;;;;EAElC,CAAC;AAEH,MAAM,MAAM,sBAAsB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,uBAAuB,CAAC,CAAC;AAE7E,eAAO,MAAM,iBAAiB,EAAE,IA8B/B,CAAC"}

package/dist/tools/download-build.js

@@ -1,17 +1,18 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.downloadBuildTool = void 0;
+exports.downloadBuildTool = exports.downloadBuildToolSchema = void 0;
 const zod_1 = require("zod");
 const test_build_1 = require("../test-build");
+exports.downloadBuildToolSchema = zod_1.z.object({
+    buildUrl: zod_1.z.string().describe("The URL of the build to download"),
+});
 exports.downloadBuildTool = {
     schema: {
         name: "downloadBuild",
         description: `Download a build from a build URL. If you do not have
 have a build URL, you can try getting the environment details with the getEnvironment tool.
 Environment details will include the build URL.`,
-        parameters: zod_1.z.object({
-            buildUrl: zod_1.z.string().describe("The URL of the build to download"),
-        }),
+        parameters: exports.downloadBuildToolSchema,
     },
     execute: async (input) => {
         if (!(await (0, test_build_1.hasDownloadScript)())) {

package/dist/tools/grep/ripgrep/index.js

@@ -14,7 +14,8 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
     for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.ripgrep = exports.isRgAvailable = void 0;
+exports.isRgAvailable = isRgAvailable;
+exports.ripgrep = ripgrep;
 const child_process_1 = require("child_process");
 const types_1 = require("./types");
 __exportStar(require("./types"), exports);
@@ -40,7 +41,6 @@ function isRgAvailable() {
         return false;
     }
 }
-exports.isRgAvailable = isRgAvailable;
 function ripgrep(cwd, options) {
     if (!cwd) {
         throw new Error("No `cwd` provided");
@@ -87,4 +87,3 @@ function ripgrep(cwd, options) {
         throw new types_1.RipGrepError(error, stderr);
     }
 }
-exports.ripgrep = ripgrep;

package/dist/tools/grep/ripgrep/types.d.ts

@@ -1,5 +1,3 @@
-/// <reference types=".pnpm/@types+node@20.14.11/node_modules/@types/node/child_process" />
-/// <reference types=".pnpm/@types+node@20.16.10/node_modules/@types/node/child_process" />
 import { ExecException } from "child_process";
 type StringSearchOptions = {
     string: string;

package/dist/tools/grep/ripgrep/types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../../src/tools/grep/ripgrep/types.ts"],"names":[],"mappings":";;AAAA,OAAO,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AAE9C,KAAK,mBAAmB,GAAG;IACzB,MAAM,EAAE,MAAM,CAAC;CAChB,CAAC;AAEF,KAAK,kBAAkB,GAAG;IACxB,KAAK,EAAE,MAAM,CAAC;CACf,CAAC;AAEF,KAAK,cAAc,GAAG,mBAAmB,GAAG,kBAAkB,CAAC;AAE/D,MAAM,MAAM,OAAO,GAAG,cAAc,GAAG;IACrC,KAAK,CAAC,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;IACtB,QAAQ,CAAC,EAAE,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC;IAClC,SAAS,CAAC,EAAE,OAAO,CAAC;CACrB,CAAC;AAEF,MAAM,MAAM,mBAAmB,GAAG;IAChC,KAAK,EAAE;QAAE,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC;IACxB,KAAK,EAAE,MAAM,CAAC;IACd,GAAG,EAAE,MAAM,CAAC;CACb,CAAC;AAEF,MAAM,MAAM,gBAAgB,GAAG;IAC7B,IAAI,EAAE,OAAO,CAAC;IACd,IAAI,EAAE;QACJ,IAAI,EAAE;YACJ,IAAI,EAAE,MAAM,CAAC;SACd,CAAC;QACF,KAAK,EAAE;YACL,IAAI,EAAE,MAAM,CAAC;SACd,CAAC;QACF,WAAW,EAAE,MAAM,CAAC;QACpB,eAAe,EAAE,MAAM,CAAC;QACxB,UAAU,EAAE,KAAK,CAAC,mBAAmB,CAAC,CAAC;KACxC,CAAC;CACH,CAAC;AAEF,MAAM,MAAM,KAAK,GAAG,gBAAgB,CAAC,MAAM,CAAC,CAAC;AAE7C,qBAAa,YAAY;IACvB,KAAK,EAAE,aAAa,CAAC;IACrB,MAAM,EAAE,MAAM,CAAC;gBAEH,KAAK,EAAE,aAAa,EAAE,MAAM,EAAE,MAAM;IAKhD,IAAI,OAAO,IAAI,MAAM,CAEpB;CACF"}
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../../src/tools/grep/ripgrep/types.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AAE9C,KAAK,mBAAmB,GAAG;IACzB,MAAM,EAAE,MAAM,CAAC;CAChB,CAAC;AAEF,KAAK,kBAAkB,GAAG;IACxB,KAAK,EAAE,MAAM,CAAC;CACf,CAAC;AAEF,KAAK,cAAc,GAAG,mBAAmB,GAAG,kBAAkB,CAAC;AAE/D,MAAM,MAAM,OAAO,GAAG,cAAc,GAAG;IACrC,KAAK,CAAC,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;IACtB,QAAQ,CAAC,EAAE,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC;IAClC,SAAS,CAAC,EAAE,OAAO,CAAC;CACrB,CAAC;AAEF,MAAM,MAAM,mBAAmB,GAAG;IAChC,KAAK,EAAE;QAAE,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC;IACxB,KAAK,EAAE,MAAM,CAAC;IACd,GAAG,EAAE,MAAM,CAAC;CACb,CAAC;AAEF,MAAM,MAAM,gBAAgB,GAAG;IAC7B,IAAI,EAAE,OAAO,CAAC;IACd,IAAI,EAAE;QACJ,IAAI,EAAE;YACJ,IAAI,EAAE,MAAM,CAAC;SACd,CAAC;QACF,KAAK,EAAE;YACL,IAAI,EAAE,MAAM,CAAC;SACd,CAAC;QACF,WAAW,EAAE,MAAM,CAAC;QACpB,eAAe,EAAE,MAAM,CAAC;QACxB,UAAU,EAAE,KAAK,CAAC,mBAAmB,CAAC,CAAC;KACxC,CAAC;CACH,CAAC;AAEF,MAAM,MAAM,KAAK,GAAG,gBAAgB,CAAC,MAAM,CAAC,CAAC;AAE7C,qBAAa,YAAY;IACvB,KAAK,EAAE,aAAa,CAAC;IACrB,MAAM,EAAE,MAAM,CAAC;gBAEH,KAAK,EAAE,aAAa,EAAE,MAAM,EAAE,MAAM;IAKhD,IAAI,OAAO,IAAI,MAAM,CAEpB;CACF"}

package/dist/tools/str_replace_editor.js

@@ -3,7 +3,9 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.textEditorTools = exports.strReplaceEditorExecutor = exports.cleanupBackupFiles = void 0;
+exports.textEditorTools = void 0;
+exports.cleanupBackupFiles = cleanupBackupFiles;
+exports.strReplaceEditorExecutor = strReplaceEditorExecutor;
 const fs_1 = __importDefault(require("fs"));
 const path_1 = __importDefault(require("path"));
 const zod_1 = require("zod");
@@ -130,7 +132,6 @@ function cleanupBackupFiles(repoDir) {
     }
     return cleanedCount;
 }
-exports.cleanupBackupFiles = cleanupBackupFiles;
 function escapeRegExp(text) {
     return text.replace(/[-[\]{}()*+?.,\\^$|#\s]/g, "\\$&");
 }
@@ -291,7 +292,6 @@ async function strReplaceEditorExecutor(input) {
         };
     }
 }
-exports.strReplaceEditorExecutor = strReplaceEditorExecutor;
 const fileViewTool = {
     schema: {
         name: "fileViewTool",

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

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

@@ -104,34 +104,29 @@ exports.generateTestWithBrowserAgent = {
         }
         const fileBackup = await promises_1.default.readFile(fileName, "utf-8");
         try {
-            await (0, utils_1.replaceTodoWithCreateTest)({
-                testCaseName: testName,
-                testCaseSuites: testSuites,
-                testFilePath: fileName,
-            });
+            await (0, utils_1.replaceTodoWithCreateTest)(fileName);
         }
         catch (error) {
+            // Undo the TODO -> createTest and test.only changes
+            await promises_1.default.writeFile(fileName, fileBackup, "utf-8");
             return {
                 isError: true,
                 result: `Error running tool: ${error}`,
             };
         }
+        const testGenToken = (0, scenarios_1.buildTokenFromOptions)({ name: testName, file: fileName, prompt: changeToMake }, { useComputerUseAgent: true });
         const toolResult = await (0, run_1.generateTestsUsingMasterAgent)({
+            testCaseName: testName,
+            testCaseSuites: testSuites,
             testFilePath: fileName,
             filePathToUpdate: fileName,
             pwProjectsFilter: [project],
             traceId: trace?.id,
-            testGenToken: (0, scenarios_1.buildTokenFromOptions)({
-                name: testName,
-                file: fileName,
-                prompt: changeToMake,
-            }, {
-                useComputerUseAgent: true,
-            }),
+            testGenToken,
             repoDir: process.cwd(),
             editFileWithGeneratedCode: false,
         });
-        // Undo the TODO -> createTest change
+        // Undo the TODO -> createTest changes
         await promises_1.default.writeFile(fileName, fileBackup, "utf-8");
         const { isError, error, actionsSummary } = toolResult;
         if (!isError) {

package/dist/tools/test-run-fetcher/index.js

@@ -1,6 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.fetchTestRunReportTool = exports.extractPathAfterSourceRepo = void 0;
+exports.fetchTestRunReportTool = void 0;
+exports.extractPathAfterSourceRepo = extractPathAfterSourceRepo;
 const zod_1 = require("zod");
 const utils_1 = require("../utils");
 const TestRunSchema = zod_1.z.object({
@@ -17,7 +18,6 @@ function extractPathAfterSourceRepo(fullPath) {
         return parts[1].replace(/^\/|\/$/g, "");
     }
 }
-exports.extractPathAfterSourceRepo = extractPathAfterSourceRepo;
 exports.fetchTestRunReportTool = {
     schema: {
         name: "fetchTestRunDetails",

package/dist/tools/test-run.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"test-run.d.ts","sourceRoot":"","sources":["../../src/tools/test-run.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,wBAAwB,CAAC;AAgDnD,eAAO,MAAM,WAAW,EAAE,IAuDzB,CAAC"}
+{"version":3,"file":"test-run.d.ts","sourceRoot":"","sources":["../../src/tools/test-run.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,wBAAwB,CAAC;AAgDnD,eAAO,MAAM,WAAW,EAAE,IAiDzB,CAAC"}