@empiricalrun/test-gen 0.71.1 → 0.72.0

package/dist/tools/definitions/index.d.ts

@@ -0,0 +1,22 @@
+import { PendingToolCall } from "@empiricalrun/llm/chat";
+import { SupportedChatModels, ToolDefinition } from "@empiricalrun/shared-types";
+export declare class ToolCaller {
+    tools: ToolDefinition[];
+    selectedModel: SupportedChatModels;
+    branchName: string;
+    chatSessionId: number | null;
+    workerEnv?: Record<string, string>;
+    constructor({ chatSessionId, selectedModel, branchName, workerEnv, }: {
+        chatSessionId: number | null;
+        selectedModel: SupportedChatModels;
+        branchName: string;
+        workerEnv?: Record<string, string>;
+    });
+    schemas(): {
+        name: string;
+        description: string;
+        parameters: import("zod").ZodType;
+    }[];
+    sendToQueue(requestId: string, toolCalls: PendingToolCall[]): Promise<void>;
+}
+//# sourceMappingURL=index.d.ts.map

package/dist/tools/definitions/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/tools/definitions/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAuB,eAAe,EAAE,MAAM,wBAAwB,CAAC;AAC9E,OAAO,EACL,mBAAmB,EACnB,cAAc,EACf,MAAM,4BAA4B,CAAC;AAcpC,qBAAa,UAAU;IACrB,KAAK,EAAE,cAAc,EAAE,CAAM;IAC7B,aAAa,EAAE,mBAAmB,CAAC;IACnC,UAAU,EAAE,MAAM,CAAC;IACnB,aAAa,EAAE,MAAM,GAAG,IAAI,CAAC;IAC7B,SAAS,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;gBACvB,EACV,aAAa,EACb,aAAa,EACb,UAAU,EACV,SAAS,GACV,EAAE;QACD,aAAa,EAAE,MAAM,GAAG,IAAI,CAAC;QAC7B,aAAa,EAAE,mBAAmB,CAAC;QACnC,UAAU,EAAE,MAAM,CAAC;QACnB,SAAS,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;KACpC;IAsBD,OAAO;;;;;IAID,WAAW,CACf,SAAS,EAAE,MAAM,EACjB,SAAS,EAAE,eAAe,EAAE,GAC3B,OAAO,CAAC,IAAI,CAAC;CAcjB"}

package/dist/tools/definitions/index.js

@@ -0,0 +1,61 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ToolCaller = void 0;
+const chat_1 = require("@empiricalrun/llm/chat");
+const commit_and_create_pr_1 = require("../commit-and-create-pr");
+const diagnosis_fetcher_1 = require("../diagnosis-fetcher");
+const download_build_1 = require("../download-build");
+const grep_1 = require("../grep");
+const list_environments_1 = require("../list-environments");
+const test_run_fetcher_1 = require("../test-run-fetcher");
+const upgrade_packages_1 = require("../upgrade-packages");
+const run_test_1 = require("./run-test");
+const str_replace_editor_1 = require("./str_replace_editor");
+const test_gen_browser_1 = require("./test-gen-browser");
+const queue_1 = require("./utils/queue");
+class ToolCaller {
+    tools = [];
+    selectedModel;
+    branchName;
+    chatSessionId;
+    workerEnv;
+    constructor({ chatSessionId, selectedModel, branchName, workerEnv, }) {
+        this.chatSessionId = chatSessionId;
+        this.selectedModel = selectedModel;
+        this.branchName = branchName;
+        this.workerEnv = workerEnv;
+        // TODO: Keep this list in sync with the ToolExecutor
+        this.tools = [
+            run_test_1.runTestTool,
+            test_gen_browser_1.generateTestWithBrowserAgent,
+            grep_1.grepTool,
+            test_run_fetcher_1.fetchTestRunReportTool,
+            diagnosis_fetcher_1.fetchDiagnosisReportTool,
+            commit_and_create_pr_1.createPullRequestTool,
+            list_environments_1.listEnvironmentsTool,
+            download_build_1.downloadBuildTool,
+            upgrade_packages_1.upgradePackagesTool,
+        ];
+        if ((0, chat_1.getProviderForModel)(this.selectedModel) !== "claude") {
+            this.tools.push(...Object.values(str_replace_editor_1.textEditorTools));
+        }
+    }
+    schemas() {
+        return this.tools.map((tool) => tool.schema);
+    }
+    async sendToQueue(requestId, toolCalls) {
+        const queueUrl = (0, queue_1.getQueueUrl)(toolCalls, this.tools, this.workerEnv);
+        if (!queueUrl) {
+            throw new Error(`queueUrl is required for remote execution.`);
+        }
+        await (0, queue_1.sendToolRequestToRemoteQueue)(queueUrl, {
+            toolCalls,
+            requestId,
+            chatSessionId: this.chatSessionId,
+            selectedModel: this.selectedModel,
+            branchName: this.branchName,
+            workerEnv: this.workerEnv,
+        });
+    }
+}
+exports.ToolCaller = ToolCaller;

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

@@ -0,0 +1,23 @@
+import { ToolDefinition } from "@empiricalrun/shared-types";
+import { z } from "zod";
+export declare const RunTestSchema: z.ZodObject<{
+    testName: z.ZodString;
+    suites: z.ZodArray<z.ZodString, "many">;
+    filePath: z.ZodString;
+    project: z.ZodString;
+    buildUrl: z.ZodOptional<z.ZodString>;
+}, "strip", z.ZodTypeAny, {
+    suites: string[];
+    testName: string;
+    filePath: string;
+    project: string;
+    buildUrl?: string | undefined;
+}, {
+    suites: string[];
+    testName: string;
+    filePath: string;
+    project: string;
+    buildUrl?: string | undefined;
+}>;
+export declare const runTestTool: ToolDefinition;
+//# sourceMappingURL=run-test.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"run-test.d.ts","sourceRoot":"","sources":["../../../src/tools/definitions/run-test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;AAC5D,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,eAAO,MAAM,aAAa;;;;;;;;;;;;;;;;;;EAmBxB,CAAC;AAEH,eAAO,MAAM,WAAW,EAAE,cAOzB,CAAC"}

package/dist/tools/definitions/run-test.js

@@ -0,0 +1,28 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.runTestTool = exports.RunTestSchema = void 0;
+const zod_1 = require("zod");
+exports.RunTestSchema = zod_1.z.object({
+    testName: zod_1.z.string().describe("The name of the test to run"),
+    suites: zod_1.z
+        .array(zod_1.z.string())
+        .describe("The suites (describe blocks) where the test is located."),
+    filePath: zod_1.z
+        .string()
+        .describe("Path of the file where the test is located. Path must be relative to the root of the repository. File name must end with .spec.ts. For example: tests/lesson.spec.ts"),
+    project: zod_1.z.string().describe("The project to run the test on"),
+    buildUrl: zod_1.z
+        .string()
+        .optional()
+        .describe(`The URL of the build to run the test on.
+This is ONLY useful for preview deployments that are dynamically created on every pull request, like Vercel preview deployments.
+For all other cases, the build url does not need to be provided, since it is already hard coded in the test repo.`),
+});
+exports.runTestTool = {
+    schema: {
+        name: "runTest",
+        description: "Run a test",
+        parameters: exports.RunTestSchema,
+    },
+    needsBrowser: true,
+};

package/dist/tools/definitions/str_replace_editor.d.ts

@@ -0,0 +1,3 @@
+import { ToolDefinition } from "@empiricalrun/shared-types";
+export declare const textEditorTools: Record<string, ToolDefinition>;
+//# sourceMappingURL=str_replace_editor.d.ts.map

package/dist/tools/definitions/str_replace_editor.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"str_replace_editor.d.ts","sourceRoot":"","sources":["../../../src/tools/definitions/str_replace_editor.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;AA0E5D,eAAO,MAAM,eAAe,EAAE,MAAM,CAAC,MAAM,EAAE,cAAc,CAK1D,CAAC"}

package/dist/tools/definitions/str_replace_editor.js

@@ -0,0 +1,74 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.textEditorTools = void 0;
+const zod_1 = require("zod");
+const fileViewTool = {
+    schema: {
+        name: "fileViewTool",
+        description: `A tool to view the content of a file or directory. If the path points 
+to a directory, the tool will return a list of files in the directory, separated by line breaks.
+If the path points to a file, the tool will return the content of the file.
+
+File contents are returned with line numbers, starting from 1.
+
+1: line 1
+2: line 2
+...`,
+        parameters: zod_1.z.object({
+            // Does not support view_range for now
+            path: zod_1.z.string().describe("The path to the file or directory to view."),
+        }),
+    },
+    needsBrowser: false,
+};
+const fileCreateTool = {
+    schema: {
+        name: "fileCreateTool",
+        description: `A tool to create a new file with given contents.
+This tool will also create parent directories that do not exist.
+For example, for path "tests/example/foo.spec.ts", the tool will create
+directories "tests", "tests/example", and "tests/example/foo.spec.ts".`,
+        parameters: zod_1.z.object({
+            path: zod_1.z.string().describe("The path to the new file."),
+            file_text: zod_1.z.string().describe("The contents of the new file."),
+        }),
+    },
+    needsBrowser: false,
+};
+const stringReplaceTool = {
+    schema: {
+        name: "stringReplaceTool",
+        description: `A tool to replace a string in a file. This tool requires old_str to be unique
+in the file. If old_str is not unique, the tool will return an error.`,
+        parameters: zod_1.z.object({
+            path: zod_1.z.string().describe("The path to the file."),
+            old_str: zod_1.z.string().describe("The string to be replaced."),
+            new_str: zod_1.z.string().describe("The string to replace old_str with."),
+        }),
+    },
+    needsBrowser: false,
+};
+const stringInsertTool = {
+    schema: {
+        name: "stringInsertTool",
+        description: "A tool to insert a string at a specific line in a file. Line numbers are 1-indexed, just like the file view tool.",
+        parameters: zod_1.z.object({
+            path: zod_1.z.string().describe("The path to the file."),
+            insert_line: zod_1.z
+                .number()
+                .int()
+                .min(1)
+                .describe(`The line number on which to insert the text (1 for beginning of file).
+To insert a string at the beginning of the file, you should use insert_line = 1.
+To insert a string at the end of the file, you should use insert_line = (total lines + 1).`),
+            new_str: zod_1.z.string().describe("The string to insert."),
+        }),
+    },
+    needsBrowser: false,
+};
+exports.textEditorTools = {
+    fileViewTool: fileViewTool,
+    fileCreateTool: fileCreateTool,
+    stringReplaceTool: stringReplaceTool,
+    stringInsertTool: stringInsertTool,
+};

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

@@ -0,0 +1,26 @@
+import { ToolDefinition } from "@empiricalrun/shared-types";
+import { z } from "zod";
+export declare const BrowserAgentSchema: z.ZodObject<{
+    testName: z.ZodString;
+    testSuites: z.ZodArray<z.ZodString, "many">;
+    filePath: z.ZodString;
+    project: z.ZodString;
+    buildUrl: z.ZodOptional<z.ZodString>;
+    changeToMake: z.ZodString;
+}, "strip", z.ZodTypeAny, {
+    testName: string;
+    testSuites: string[];
+    filePath: string;
+    project: string;
+    changeToMake: string;
+    buildUrl?: string | undefined;
+}, {
+    testName: string;
+    testSuites: string[];
+    filePath: string;
+    project: string;
+    changeToMake: string;
+    buildUrl?: string | undefined;
+}>;
+export declare const generateTestWithBrowserAgent: ToolDefinition;
+//# sourceMappingURL=test-gen-browser.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"test-gen-browser.d.ts","sourceRoot":"","sources":["../../../src/tools/definitions/test-gen-browser.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;AAC5D,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,eAAO,MAAM,kBAAkB;;;;;;;;;;;;;;;;;;;;;EAwB7B,CAAC;AA4DH,eAAO,MAAM,4BAA4B,EAAE,cAO1C,CAAC"}

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

@@ -0,0 +1,88 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.generateTestWithBrowserAgent = exports.BrowserAgentSchema = void 0;
+const zod_1 = require("zod");
+exports.BrowserAgentSchema = zod_1.z.object({
+    testName: zod_1.z.string().describe("The name of the test to create or modify"),
+    testSuites: zod_1.z
+        .array(zod_1.z.string())
+        .describe("The suites (describe blocks) where the test is located"),
+    filePath: zod_1.z
+        .string()
+        .describe("Path of the file where the test is located. Path must be relative to the root of the repository. File name must end with .spec.ts. For example: tests/lesson.spec.ts"),
+    project: zod_1.z
+        .string()
+        .describe("The Playwright project to run tests against (e.g. 'chromium' or 'firefox')"),
+    buildUrl: zod_1.z
+        .string()
+        .optional()
+        .describe(`The URL of the build to run the test on.
+This is ONLY useful for preview deployments that are dynamically created on every pull request, like Vercel preview deployments.
+For all other cases, the build url does not need to be provided, since it is already hard coded in the test repo.`),
+    changeToMake: zod_1.z.string().describe("The change to make to the test"),
+});
+const BROWSER_AGENT_DESCRIPTION = `
+Create or modify a test case with browser agent. The browser agent can take user interactions in a web browser
+and generate Playwright code for that actions. This is a useful tool when the modifications require knowing the 
+locator/selector for an element on the page.
+
+IMPORTANT: Before you invoke this tool, you need to ensure that the test code is correctly prepared for this
+agent. Preparation involves adding a TODO comment that describes the change that needs to be made, and the page 
+variable name where the actions must be performed. The content of the TODO comment calls out the element and browser 
+interactions it must take. The TODO comment also has (agent on pageName) next to it, to clearly label that the change
+is for the agent to make on the given page (pageName in this case).
+
+To choose the page variable name, go through the test code and find available page variables. If you are replacing some 
+existing test code, use the same page variable name as in the existing test code. If you are adding steps to the test,
+use the page variable name that is appropriate for the new steps. The page variable represents the browser page (or tab) that
+the agent is supposed to interact with.
+
+For example, this is a good TODO comment:
+
+\`\`\`
+test("Example test code", async ({ page }) => {
+  await page.goto("https://example.com");
+  // TODO(agent on page): Click on the login button
+});
+\`\`\`
+
+For the above file, the browser environment will execute the steps before the TODO comment and hand-over the control
+to the browser agent. The agent will do the actions described in the TODO comment and then resume control back to the
+test 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 on extPage): 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
+\`\`\`
+
+The TODO comment must be inside a test block, not outside of it. For example, this is invalid:
+
+\`\`\`
+// --- BEGIN INVALID EXAMPLE ---
+test("Example test code", async ({ page }) => {
+  await page.goto("https://example.com");
+});
+// TODO(agent on page): Click on the login button
+// --- END INVALID EXAMPLE ---
+\`\`\`
+
+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: {
+        name: "generateTestWithBrowserAgent",
+        description: BROWSER_AGENT_DESCRIPTION,
+        parameters: exports.BrowserAgentSchema,
+    },
+    needsBrowser: true,
+};

package/dist/{tool-call-service/utils.d.ts → tools/definitions/utils/queue.d.ts}

@@ -1,6 +1,8 @@
-import { PendingToolCall, ServicePayload, Tool } from "@empiricalrun/shared-types";
+import { PendingToolCall, ServicePayload, ToolDefinition } 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
+export declare function getQueueUrl(toolCalls: PendingToolCall[], tools: ToolDefinition[], workerEnv?: Record<string, string>): string;
+export declare function sendToolRequestToRemoteQueue(queueUrl: string, payload: ServicePayload & {
+    workerEnv?: Record<string, string>;
+}): Promise<void>;
+//# sourceMappingURL=queue.d.ts.map

package/dist/tools/definitions/utils/queue.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"queue.d.ts","sourceRoot":"","sources":["../../../../src/tools/definitions/utils/queue.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,eAAe,EACf,cAAc,EACd,cAAc,EACf,MAAM,4BAA4B,CAAC;AAIpC,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,cAAc,EAAE,EACvB,SAAS,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GACjC,MAAM,CAyBR;AAED,wBAAsB,4BAA4B,CAChD,QAAQ,EAAE,MAAM,EAChB,OAAO,EAAE,cAAc,GAAG;IAAE,SAAS,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;CAAE,iBAiBjE"}

package/dist/{tool-call-service/utils.js → tools/definitions/utils/queue.js}

@@ -4,7 +4,7 @@ exports.appendBranchNameToQueueUrl = appendBranchNameToQueueUrl;
 exports.replaceRegionInUrl = replaceRegionInUrl;
 exports.getQueueUrl = getQueueUrl;
 exports.sendToolRequestToRemoteQueue = sendToolRequestToRemoteQueue;
-const client_sqs_1 = require("@aws-sdk/client-sqs");
+const SQSClient_1 = require("../../../utils/SQSClient");
 function appendBranchNameToQueueUrl(queueUrl, branchName) {
     const cleanBranchName = branchName.replace(/[/.]/g, "-");
     const suffix = cleanBranchName.slice(0, 50);
@@ -28,12 +28,17 @@ function buildQueueUrlsForPreview(baseQueueUrl) {
     console.log("queueUrl for preview", queueUrl);
     return queueUrl;
 }
-function getQueueUrl(toolCalls, tools) {
+function getQueueUrl(toolCalls, tools, workerEnv) {
     const needsBrowser = toolCalls.some((toolCall) => tools.find((t) => t.schema.name === toolCall.name)?.needsBrowser);
-    const baseQueueUrl = needsBrowser
-        ? process.env.TOOL_EXECUTION_SQS_URL_BROWSER
-        : process.env.TOOL_EXECUTION_SQS_URL;
-    if (process.env.VERCEL_ENV !== "preview") {
+    const browserSwsQueueUrl = workerEnv?.TOOL_EXECUTION_SQS_URL_BROWSER ||
+        process.env.TOOL_EXECUTION_SQS_URL_BROWSER;
+    const baseSqsQueueUrl = workerEnv?.TOOL_EXECUTION_SQS_URL || process.env.TOOL_EXECUTION_SQS_URL;
+    const baseQueueUrl = needsBrowser ? browserSwsQueueUrl : baseSqsQueueUrl;
+    if (!baseQueueUrl) {
+        throw new Error("No queue URL found");
+    }
+    const vercelEnv = workerEnv?.VERCEL_ENV || process.env.VERCEL_ENV || "";
+    if (vercelEnv !== "preview") {
         return baseQueueUrl;
     }
     else {
@@ -41,17 +46,13 @@ function getQueueUrl(toolCalls, tools) {
     }
 }
 async function sendToolRequestToRemoteQueue(queueUrl, 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,
-        },
-    });
-    await sqs.send(new client_sqs_1.SendMessageCommand({
-        QueueUrl: queueUrl,
-        MessageGroupId: payload.requestId,
-        MessageDeduplicationId: payload.requestId,
-        MessageBody: JSON.stringify(payload),
-    }));
+    const { workerEnv } = payload;
+    const region = workerEnv?.AWS_REGION || process.env.AWS_REGION;
+    const accessKeyId = workerEnv?.AWS_ACCESS_KEY_ID || process.env.AWS_ACCESS_KEY_ID;
+    const secretAccessKey = workerEnv?.AWS_SECRET_ACCESS_KEY || process.env.AWS_SECRET_ACCESS_KEY;
+    if (!region || !accessKeyId || !secretAccessKey) {
+        throw new Error("AWS_REGION, AWS_ACCESS_KEY_ID, and AWS_SECRET_ACCESS_KEY must be set");
+    }
+    const sqs = new SQSClient_1.SQSClient(region, accessKeyId, secretAccessKey);
+    await sqs.sendMessage(queueUrl, payload);
 }

package/dist/tools/diagnosis-fetcher.js

@@ -7,7 +7,7 @@ exports.fetchDiagnosisReportTool = void 0;
 const promises_1 = __importDefault(require("fs/promises"));
 const path_1 = __importDefault(require("path"));
 const zod_1 = require("zod");
-const utils_1 = require("./utils");
+const utils_1 = require("./executor/utils");
 const DiagnosisSchema = zod_1.z.object({
     diagnosisUrl: zod_1.z
         .string()

package/dist/{tool-call-service → tools/executor}/index.d.ts

@@ -1,12 +1,9 @@
 import { TraceClient } from "@empiricalrun/llm";
 import { PendingToolCall, ToolResult } from "@empiricalrun/llm/chat";
 import { SupportedChatModels, Tool, ToolExecute } from "@empiricalrun/shared-types";
-type ToolExecutors = {
-    [key: string]: ToolExecute;
-};
-export declare class ToolCallService {
+export declare class ToolExecutor {
     tools: Tool[];
-    toolExecutors: ToolExecutors;
+    toolExecutors: Record<string, ToolExecute>;
     chatSessionId: number | null;
     selectedModel: SupportedChatModels;
     branchName: string;
@@ -25,8 +22,6 @@ export declare class ToolCallService {
         featureFlags: string[];
         environmentOverrides?: Record<string, string>;
     });
-    sendToQueue(requestId: string, toolCalls: PendingToolCall[]): Promise<void>;
     execute(toolCalls: PendingToolCall[]): Promise<ToolResult[]>;
 }
-export {};
 //# sourceMappingURL=index.d.ts.map

package/dist/tools/executor/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/tools/executor/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAEL,eAAe,EACf,UAAU,EACX,MAAM,wBAAwB,CAAC;AAChC,OAAO,EAEL,mBAAmB,EACnB,IAAI,EACJ,WAAW,EACZ,MAAM,4BAA4B,CAAC;AAkBpC,qBAAa,YAAY;IACvB,KAAK,EAAE,IAAI,EAAE,CAAM;IACnB,aAAa,EAAE,MAAM,CAAC,MAAM,EAAE,WAAW,CAAC,CAAM;IAChD,aAAa,EAAE,MAAM,GAAG,IAAI,CAAC;IAC7B,aAAa,EAAE,mBAAmB,CAAC;IACnC,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC;IAC3B,KAAK,CAAC,EAAE,WAAW,CAAC;IACpB,YAAY,EAAE,MAAM,EAAE,CAAC;IACvB,oBAAoB,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;gBAEjC,EACV,aAAa,EACb,aAAa,EACb,UAAU,EACV,QAAQ,EACR,MAAM,EACN,KAAK,EACL,YAAY,EACZ,oBAAyB,GAC1B,EAAE;QACD,aAAa,EAAE,MAAM,GAAG,IAAI,CAAC;QAC7B,aAAa,EAAE,mBAAmB,CAAC;QACnC,UAAU,EAAE,MAAM,CAAC;QACnB,QAAQ,EAAE,MAAM,CAAC;QACjB,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC;QAC3B,KAAK,CAAC,EAAE,WAAW,CAAC;QACpB,YAAY,EAAE,MAAM,EAAE,CAAC;QACvB,oBAAoB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;KAC/C;IAoCK,OAAO,CAAC,SAAS,EAAE,eAAe,EAAE,GAAG,OAAO,CAAC,UAAU,EAAE,CAAC;CAyEnE"}

package/dist/{tool-call-service → tools/executor}/index.js

@@ -1,21 +1,20 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.ToolCallService = void 0;
+exports.ToolExecutor = void 0;
 const chat_1 = require("@empiricalrun/llm/chat");
-const artifacts_1 = require("../artifacts");
-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");
-const grep_1 = require("../tools/grep");
-const list_environments_1 = require("../tools/list-environments");
-const str_replace_editor_1 = require("../tools/str_replace_editor");
-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 upgrade_packages_1 = require("../tools/upgrade-packages");
-const checkpoint_1 = require("../utils/checkpoint");
-const utils_1 = require("./utils");
-class ToolCallService {
+const artifacts_1 = require("../../artifacts");
+const commit_and_create_pr_1 = require("../commit-and-create-pr");
+const diagnosis_fetcher_1 = require("../diagnosis-fetcher");
+const download_build_1 = require("../download-build");
+const grep_1 = require("../grep");
+const list_environments_1 = require("../list-environments");
+const run_test_1 = require("../run-test");
+const str_replace_editor_1 = require("../str_replace_editor");
+const test_gen_browser_1 = require("../test-gen-browser");
+const test_run_fetcher_1 = require("../test-run-fetcher");
+const upgrade_packages_1 = require("../upgrade-packages");
+const checkpoint_1 = require("./utils/checkpoint");
+class ToolExecutor {
     tools = [];
     toolExecutors = {};
     chatSessionId;
@@ -37,7 +36,7 @@ class ToolCallService {
         this.environmentOverrides = environmentOverrides;
         this.tools = [
             grep_1.grepTool,
-            test_run_1.runTestTool,
+            run_test_1.runTestTool,
             test_run_fetcher_1.fetchTestRunReportTool,
             diagnosis_fetcher_1.fetchDiagnosisReportTool,
             test_gen_browser_1.generateTestWithBrowserAgent,
@@ -61,19 +60,6 @@ class ToolCallService {
             };
         }
     }
-    async sendToQueue(requestId, toolCalls) {
-        const queueUrl = (0, utils_1.getQueueUrl)(toolCalls, this.tools);
-        if (!queueUrl) {
-            throw new Error(`queueUrl is required for remote execution.`);
-        }
-        await (0, utils_1.sendToolRequestToRemoteQueue)(queueUrl, {
-            toolCalls,
-            requestId,
-            chatSessionId: this.chatSessionId,
-            selectedModel: this.selectedModel,
-            branchName: this.branchName,
-        });
-    }
     async execute(toolCalls) {
         const executeSpan = this.trace?.span({
             name: "execute_tools",
@@ -140,4 +126,4 @@ class ToolCallService {
         return toolResults;
     }
 }
-exports.ToolCallService = ToolCallService;
+exports.ToolExecutor = ToolExecutor;

package/dist/tools/executor/utils/checkpoint.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"checkpoint.d.ts","sourceRoot":"","sources":["../../../../src/tools/executor/utils/checkpoint.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC;AAIzD,wBAAsB,gBAAgB,CAAC,EACrC,SAAS,EACT,UAAU,EACV,QAAQ,GACT,EAAE;IACD,SAAS,EAAE,eAAe,EAAE,CAAC;IAC7B,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;CAClB,iBAyBA"}

package/dist/tools/executor/utils/git.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"git.d.ts","sourceRoot":"","sources":["../../../../src/tools/executor/utils/git.ts"],"names":[],"mappings":"AAOA,wBAAgB,UAAU,CAAC,QAAQ,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,GAAG,MAAM,CAMhE;AAED,wBAAgB,oBAAoB,CAAC,QAAQ,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,GAAG,MAAM,CAe1E;AAED,wBAAsB,cAAc,CAAC,UAAU,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,iBAQnE;AAED,wBAAsB,eAAe,CAAC,aAAa,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,oBAcvE;AAED,wBAAsB,oBAAoB,CAAC,QAAQ,EAAE,MAAM,mBAO1D;AAED,wBAAsB,UAAU,CAAC,UAAU,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,iBAE/D;AAED,wBAAsB,wBAAwB,CAAC,EAC7C,aAAa,EACb,UAAU,EACV,KAAK,EACL,QAAQ,GACT,EAAE;IACD,aAAa,EAAE,MAAM,CAAC;IACtB,UAAU,EAAE,MAAM,CAAC;IACnB,KAAK,EAAE,MAAM,EAAE,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC;CAClB,iBAIA;AAED,wBAAsB,eAAe,CAAC,GAAG,EAAE,MAAM,qBAUhD;AAED,wBAAsB,gBAAgB,CAAC,EACrC,QAAQ,EACR,QAAQ,EACR,MAAM,GACP,EAAE;IACD,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;CAChB,oBA+BA;AAED,wBAAsB,YAAY,CAAC,EACjC,QAAQ,EACR,UAAU,EACV,MAAM,GACP,EAAE;IACD,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;IACnB,MAAM,EAAE,MAAM,CAAC;CAChB,iBAoBA"}

package/dist/{utils → tools/executor/utils}/git.js

@@ -11,7 +11,7 @@ exports.getFilesChanged = getFilesChanged;
 exports.mergePullRequest = mergePullRequest;
 exports.deleteBranch = deleteBranch;
 const child_process_1 = require("child_process");
-const utils_1 = require("../tools/utils");
+const index_1 = require("./index");
 const GIT_USER_NAME = "empiricalrun[bot]";
 const GIT_USER_EMAIL = "180257021+empiricalrun[bot]@users.noreply.github.com";
 function getGitDiff(filepath, cwd) {
@@ -88,12 +88,12 @@ async function getFilesChanged(cwd) {
     return filesChanged;
 }
 async function mergePullRequest({ repoName, prNumber, apiKey, }) {
-    const url = `${utils_1.GITHUB_API_BASE}/${repoName}/pulls/${prNumber}/merge`;
+    const url = `${index_1.GITHUB_API_BASE}/${repoName}/pulls/${prNumber}/merge`;
     const mergeBody = {
         merge_method: "squash",
     };
     try {
-        const response = await (0, utils_1.callGitHubProxy)({
+        const response = await (0, index_1.callGitHubProxy)({
             method: "PUT",
             url: url,
             body: mergeBody,
@@ -112,9 +112,9 @@ async function mergePullRequest({ repoName, prNumber, apiKey, }) {
     }
 }
 async function deleteBranch({ repoName, branchName, apiKey, }) {
-    const url = `${utils_1.GITHUB_API_BASE}/${repoName}/git/refs/heads/${branchName}`;
+    const url = `${index_1.GITHUB_API_BASE}/${repoName}/git/refs/heads/${branchName}`;
     try {
-        await (0, utils_1.callGitHubProxy)({
+        await (0, index_1.callGitHubProxy)({
             method: "DELETE",
             url: url,
             apiKey,

package/dist/tools/executor/utils/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/tools/executor/utils/index.ts"],"names":[],"mappings":"AAAA,wBAAsB,oBAAoB,CAAC,CAAC,EAAE,EAC5C,IAAI,EACJ,MAAc,EACd,IAAI,EACJ,MAAM,GACP,EAAE;IACD,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,IAAI,CAAC,EAAE,GAAG,CAAC;IACX,MAAM,EAAE,MAAM,CAAC;CAChB,GAAG,OAAO,CAAC,CAAC,CAAC,CAmBb;AAED,wBAAsB,eAAe,CAAC,CAAC,EAAE,EACvC,MAAM,EACN,GAAG,EACH,IAAI,EACJ,MAAM,GACP,EAAE;IACD,MAAM,EAAE,MAAM,CAAC;IACf,GAAG,EAAE,MAAM,CAAC;IACZ,IAAI,CAAC,EAAE,GAAG,CAAC;IACX,MAAM,EAAE,MAAM,CAAC;CAChB,GAAG,OAAO,CAAC,CAAC,CAAC,CAYb;AAGD,eAAO,MAAM,eAAe,+CAA+C,CAAC;AAG5E,MAAM,WAAW,cAAc;IAC7B,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;CACd;AAED,wBAAgB,cAAc,CAAC,GAAG,EAAE,MAAM,GAAG,cAAc,CAiB1D;AAGD,MAAM,WAAW,eAAe;IAC9B,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE;QAAE,GAAG,EAAE,MAAM,CAAA;KAAE,CAAC;CACvB;AAED,wBAAsB,cAAc,CAAC,EACnC,KAAK,EACL,IAAI,EACJ,UAAU,EACV,MAAM,GACP,EAAE;IACD,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,EAAE,MAAM,CAAC;IACnB,MAAM,EAAE,MAAM,CAAC;CAChB,GAAG,OAAO,CAAC,eAAe,GAAG,IAAI,CAAC,CAYlC;AAED,wBAAsB,iBAAiB,CAAC,EACtC,KAAK,EACL,IAAI,EACJ,KAAK,EACL,IAAI,EACJ,IAAa,EACb,IAAI,EACJ,MAAM,EACN,MAAM,GACP,EAAE;IACD,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC;IAClB,MAAM,EAAE,MAAM,CAAC;CAChB,GAAG,OAAO,CAAC,eAAe,CAAC,CAqB3B;AAED,wBAAsB,iBAAiB,CAAC,EACtC,KAAK,EACL,IAAI,EACJ,QAAQ,EACR,IAAI,EACJ,MAAM,GACP,EAAE;IACD,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,MAAM,CAAC;CAChB,GAAG,OAAO,CAAC,eAAe,CAAC,CAU3B;AAED,wBAAsB,uBAAuB,CAAC,EAC5C,KAAK,EACL,IAAI,EACJ,UAAU,EACV,KAAK,EACL,IAAI,EACJ,MAAM,EACN,MAAM,GACP,EAAE;IACD,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,EAAE,MAAM,CAAC;IACnB,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC;IAClB,MAAM,EAAE,MAAM,CAAC;CAChB,GAAG,OAAO,CAAC;IAAE,EAAE,EAAE,eAAe,CAAC;IAAC,KAAK,EAAE,OAAO,CAAA;CAAE,CAAC,CAyBnD"}

package/dist/tools/list-environments.js

@@ -2,7 +2,7 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.listEnvironmentsTool = void 0;
 const zod_1 = require("zod");
-const utils_1 = require("./utils");
+const utils_1 = require("./executor/utils");
 exports.listEnvironmentsTool = {
     schema: {
         name: "listEnvironments",

package/dist/tools/{test-run.d.ts → run-test.d.ts}

@@ -1,3 +1,3 @@
 import type { Tool } from "@empiricalrun/shared-types";
 export declare const runTestTool: Tool;
-//# sourceMappingURL=test-run.d.ts.map
+//# sourceMappingURL=run-test.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"run-test.d.ts","sourceRoot":"","sources":["../../src/tools/run-test.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAc,MAAM,4BAA4B,CAAC;AA0CnE,eAAO,MAAM,WAAW,EAAE,IAgFzB,CAAC"}