@empiricalrun/test-gen 0.71.2 → 0.73.0

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/tools/delete-file.d.ts

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

package/dist/tools/delete-file.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"delete-file.d.ts","sourceRoot":"","sources":["../../src/tools/delete-file.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAEV,IAAI,EAEL,MAAM,4BAA4B,CAAC;AAepC,eAAO,MAAM,cAAc,EAAE,IAwE5B,CAAC"}

package/dist/tools/delete-file.js

@@ -0,0 +1,83 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.deleteFileTool = void 0;
+const promises_1 = require("fs/promises");
+const path_1 = __importDefault(require("path"));
+const zod_1 = require("zod");
+const git_1 = require("./executor/utils/git");
+const DeleteFileInputSchema = zod_1.z.object({
+    path: zod_1.z
+        .string()
+        .describe("The path to the file to delete (relative to the repository root). For example, tests/foo.spec.ts"),
+});
+exports.deleteFileTool = {
+    schema: {
+        name: "deleteFile",
+        description: `Delete a file from the filesystem.
+This tool will permanently remove the specified file.
+Use with caution as deleted files cannot be recovered.`,
+        parameters: DeleteFileInputSchema,
+    },
+    needsBrowser: false,
+    execute: async ({ input, repoPath, collectArtifacts, }) => {
+        try {
+            const filePath = path_1.default.join(repoPath, input.path);
+            // Validate that the file exists and is actually a file (not a directory)
+            try {
+                const stats = await (0, promises_1.stat)(filePath);
+                if (!stats.isFile()) {
+                    return {
+                        isError: true,
+                        result: `Error: '${input.path}' is not a file. Use a different tool to delete directories.`,
+                    };
+                }
+            }
+            catch (error) {
+                return {
+                    isError: true,
+                    result: `Error: File '${input.path}' does not exist.`,
+                };
+            }
+            const resolvedPath = path_1.default.resolve(filePath);
+            const resolvedRepoPath = path_1.default.resolve(repoPath);
+            if (!resolvedPath.startsWith(resolvedRepoPath)) {
+                return {
+                    isError: true,
+                    result: `Error: Cannot delete files outside the repository root.`,
+                };
+            }
+            await (0, promises_1.unlink)(filePath);
+            // Collect git patch artifact after deletion
+            if (collectArtifacts) {
+                try {
+                    const gitPatch = (0, git_1.getGitDiff)(input.path, repoPath);
+                    if (gitPatch.trim()) {
+                        const patchArtifact = {
+                            name: `${path_1.default.basename(input.path, path_1.default.extname(input.path))}_delete.patch`,
+                            contentType: "text/plain",
+                            data: Buffer.from(gitPatch, "utf-8"),
+                        };
+                        collectArtifacts([patchArtifact]);
+                    }
+                }
+                catch (error) {
+                    console.warn("Failed to create git patch artifact:", error);
+                }
+            }
+            return {
+                isError: false,
+                result: `Successfully deleted file: ${input.path}`,
+            };
+        }
+        catch (error) {
+            console.error("Error deleting file", error);
+            return {
+                isError: true,
+                result: error instanceof Error ? error.message : String(error),
+            };
+        }
+    },
+};

package/dist/tools/diagnosis-fetcher.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"diagnosis-fetcher.d.ts","sourceRoot":"","sources":["../../src/tools/diagnosis-fetcher.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAe,MAAM,4BAA4B,CAAC;AAepE,eAAO,MAAM,wBAAwB,EAAE,IAsFtC,CAAC"}
+{"version":3,"file":"diagnosis-fetcher.d.ts","sourceRoot":"","sources":["../../src/tools/diagnosis-fetcher.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,4BAA4B,CAAC;AAavD,eAAO,MAAM,wBAAwB,EAAE,IAiFtC,CAAC"}

package/dist/tools/diagnosis-fetcher.js

@@ -7,7 +7,6 @@ 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 DiagnosisSchema = zod_1.z.object({
     diagnosisUrl: zod_1.z
         .string()
@@ -20,9 +19,7 @@ exports.fetchDiagnosisReportTool = {
         parameters: DiagnosisSchema,
     },
     needsBrowser: false,
-    execute: async (params) => {
-        const { input } = params;
-        const { repoPath, apiKey } = params;
+    execute: async ({ input, repoPath, apiClient }) => {
         const { diagnosisUrl } = input;
         // Extract the slug from the URL - it's the part after the last '--'
         const slug = diagnosisUrl.split("--").pop();
@@ -34,9 +31,8 @@ exports.fetchDiagnosisReportTool = {
         }
         let data = null;
         try {
-            data = await (0, utils_1.makeDashboardRequest)({
-                path: `/api/diagnosis/${slug}/detailed`,
-                apiKey,
+            data = await apiClient.request(`/api/diagnosis/${slug}/detailed`, {
+                method: "GET",
             });
         }
         catch (error) {

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

@@ -0,0 +1,24 @@
+import { TraceClient } from "@empiricalrun/llm";
+import { PendingToolCall, ToolResult } from "@empiricalrun/llm/chat";
+import { ChatSessionInfo, IDashboardAPIClient, ToolExecute } from "@empiricalrun/shared-types";
+export declare class ToolExecutor {
+    toolExecutors: Record<string, ToolExecute>;
+    chatSession: ChatSessionInfo | null;
+    branchName: string;
+    repoPath: string;
+    apiClient: IDashboardAPIClient;
+    trace?: TraceClient;
+    featureFlags: string[];
+    environmentOverrides: Record<string, string>;
+    constructor({ chatSession, branchName, repoPath, apiClient, trace, featureFlags, environmentOverrides, }: {
+        chatSession: ChatSessionInfo | null;
+        branchName: string;
+        repoPath: string;
+        apiClient: IDashboardAPIClient;
+        trace?: TraceClient;
+        featureFlags: string[];
+        environmentOverrides?: Record<string, string>;
+    });
+    execute(toolCalls: PendingToolCall[]): Promise<ToolResult[]>;
+}
+//# 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,EAAE,eAAe,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AACrE,OAAO,EAEL,eAAe,EACf,mBAAmB,EACnB,WAAW,EACZ,MAAM,4BAA4B,CAAC;AAqBpC,qBAAa,YAAY;IACvB,aAAa,EAAE,MAAM,CAAC,MAAM,EAAE,WAAW,CAAC,CAAM;IAChD,WAAW,EAAE,eAAe,GAAG,IAAI,CAAC;IACpC,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,EAAE,mBAAmB,CAAC;IAC/B,KAAK,CAAC,EAAE,WAAW,CAAC;IACpB,YAAY,EAAE,MAAM,EAAE,CAAC;IACvB,oBAAoB,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;gBAEjC,EACV,WAAW,EACX,UAAU,EACV,QAAQ,EACR,SAAS,EACT,KAAK,EACL,YAAY,EACZ,oBAAyB,GAC1B,EAAE;QACD,WAAW,EAAE,eAAe,GAAG,IAAI,CAAC;QACpC,UAAU,EAAE,MAAM,CAAC;QACnB,QAAQ,EAAE,MAAM,CAAC;QACjB,SAAS,EAAE,mBAAmB,CAAC;QAC/B,KAAK,CAAC,EAAE,WAAW,CAAC;QACpB,YAAY,EAAE,MAAM,EAAE,CAAC;QACvB,oBAAoB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;KAC/C;IAkCK,OAAO,CAAC,SAAS,EAAE,eAAe,EAAE,GAAG,OAAO,CAAC,UAAU,EAAE,CAAC;CAqEnE"}

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

@@ -1,78 +1,62 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.ToolCallService = 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 run_test_1 = require("../tools/run-test");
-const str_replace_editor_1 = require("../tools/str_replace_editor");
-const test_gen_browser_1 = require("../tools/test-gen-browser");
-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 {
-    tools = [];
+exports.ToolExecutor = void 0;
+const artifacts_1 = require("../../artifacts");
+const commit_and_create_pr_1 = require("../commit-and-create-pr");
+const delete_file_1 = require("../delete-file");
+const diagnosis_fetcher_1 = require("../diagnosis-fetcher");
+const download_build_1 = require("../download-build");
+const fetch_image_1 = require("../fetch-image");
+const grep_1 = require("../grep");
+const list_environments_1 = require("../list-environments");
+const merge_conflicts_1 = require("../merge-conflicts");
+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 {
     toolExecutors = {};
-    chatSessionId;
-    selectedModel;
+    chatSession;
     branchName;
     repoPath;
-    apiKey;
+    apiClient;
     trace;
     featureFlags;
     environmentOverrides;
-    constructor({ chatSessionId, selectedModel, branchName, repoPath, apiKey, trace, featureFlags, environmentOverrides = {}, }) {
-        this.chatSessionId = chatSessionId;
-        this.selectedModel = selectedModel;
+    constructor({ chatSession, branchName, repoPath, apiClient, trace, featureFlags, environmentOverrides = {}, }) {
+        this.chatSession = chatSession;
         this.branchName = branchName;
         this.trace = trace;
         this.repoPath = repoPath;
-        this.apiKey = apiKey;
+        this.apiClient = apiClient;
         this.featureFlags = featureFlags;
         this.environmentOverrides = environmentOverrides;
-        this.tools = [
+        const tools = [
             grep_1.grepTool,
             run_test_1.runTestTool,
             test_run_fetcher_1.fetchTestRunReportTool,
             diagnosis_fetcher_1.fetchDiagnosisReportTool,
             test_gen_browser_1.generateTestWithBrowserAgent,
             commit_and_create_pr_1.createPullRequestTool,
+            merge_conflicts_1.mergeConflictsTool,
             list_environments_1.listEnvironmentsTool,
             download_build_1.downloadBuildTool,
             upgrade_packages_1.upgradePackagesTool,
+            fetch_image_1.fetchImageTool,
+            delete_file_1.deleteFileTool,
+            ...str_replace_editor_1.textEditorTools,
         ];
-        if ((0, chat_1.getProviderForModel)(this.selectedModel) !== "claude") {
-            this.tools.push(...str_replace_editor_1.textEditorTools);
-        }
-        this.tools.forEach((tool) => {
+        tools.forEach((tool) => {
             this.toolExecutors[tool.schema.name] = tool.execute;
         });
-        if ((0, chat_1.getProviderForModel)(this.selectedModel) === "claude") {
-            this.toolExecutors = {
-                ...this.toolExecutors,
-                // Support for Claude 3x and 4: They use different tool names
-                str_replace_editor: str_replace_editor_1.strReplaceEditorExecutor,
-                str_replace_based_edit_tool: str_replace_editor_1.strReplaceEditorExecutor,
-            };
-        }
-    }
-    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,
-        });
+        this.toolExecutors = {
+            ...this.toolExecutors,
+            // Support for Claude 3x and 4: They use different tool names
+            str_replace_editor: str_replace_editor_1.strReplaceEditorExecutor,
+            str_replace_based_edit_tool: str_replace_editor_1.strReplaceEditorExecutor,
+        };
     }
     async execute(toolCalls) {
         const executeSpan = this.trace?.span({
@@ -90,7 +74,6 @@ class ToolCallService {
                 uploadArtifactsQueue.addTask(artifactsInput).catch((error) => {
                     console.error("Error collecting artifacts:", error);
                 });
-                return;
             };
             const toolExecutor = this.toolExecutors[toolCall.name];
             if (!toolExecutor) {
@@ -103,16 +86,13 @@ class ToolCallService {
                 span?.end({ output: errorResult });
                 continue;
             }
-            if (!this.apiKey) {
-                throw new Error("API key is required for tool execution");
-            }
             try {
                 const result = await toolExecutor({
                     input: toolCall.input,
                     repoPath: this.repoPath,
-                    apiKey: this.apiKey,
+                    apiClient: this.apiClient,
                     trace: this.trace,
-                    chatSessionId: this.chatSessionId,
+                    chatSession: this.chatSession,
                     collectArtifacts: collectArtifactsFn,
                     environmentOverrides: this.environmentOverrides,
                     featureFlags: this.featureFlags,
@@ -131,7 +111,7 @@ class ToolCallService {
                 span?.end({ output: errorResult });
             }
         }
-        await (0, checkpoint_1.createCheckpoint)({
+        (0, checkpoint_1.createCommitCheckpoint)({
             toolCalls,
             branchName: this.branchName,
             repoPath: this.repoPath,
@@ -140,4 +120,4 @@ class ToolCallService {
         return toolResults;
     }
 }
-exports.ToolCallService = ToolCallService;
+exports.ToolExecutor = ToolExecutor;

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

@@ -0,0 +1,7 @@
+import { PendingToolCall } from "@empiricalrun/llm/chat";
+export declare function createCommitCheckpoint({ toolCalls, branchName, repoPath, }: {
+    toolCalls: PendingToolCall[];
+    branchName: string;
+    repoPath: string;
+}): void;
+//# sourceMappingURL=checkpoint.d.ts.map

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;AAKzD,wBAAgB,sBAAsB,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,QA0BA"}

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

@@ -1,9 +1,10 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.createCheckpoint = createCheckpoint;
+exports.createCommitCheckpoint = createCommitCheckpoint;
+const child_process_1 = require("child_process");
 const git_1 = require("./git");
-async function createCheckpoint({ toolCalls, branchName, repoPath, }) {
-    const filesChanged = await (0, git_1.getFilesChanged)(repoPath);
+function createCommitCheckpoint({ toolCalls, branchName, repoPath, }) {
+    const filesChanged = (0, git_1.getFilesChanged)(repoPath);
     const toolsWithUpdatedFiles = toolCalls
         .filter((tc) => "path" in tc.input &&
         tc.input.path &&
@@ -15,11 +16,11 @@ async function createCheckpoint({ toolCalls, branchName, repoPath, }) {
     const filesToCommit = toolsWithUpdatedFiles.map((tool) => tool.path);
     if (toolsWithUpdatedFiles.length > 0) {
         let commitMessage = `${toolsWithUpdatedFiles.map((tool) => `${tool.name} on ${tool.path}`).join(", ")}`;
-        await (0, git_1.commitFilesAndPushBranch)({
+        (0, git_1.commitFilesAsBotUser)({
             commitMessage,
-            branchName,
             files: filesToCommit,
             repoPath,
         });
+        (0, child_process_1.execSync)(`git push origin ${branchName}`, { cwd: repoPath });
     }
 }

package/dist/{utils → tools/executor/utils}/git.d.ts

@@ -1,24 +1,19 @@
+import { IDashboardAPIClient } from "@empiricalrun/shared-types";
 export declare function getGitDiff(filepath: string, cwd: string): string;
 export declare function getGitDiffForNewFile(filepath: string, cwd: string): string;
 export declare function checkoutBranch(branchName: string, cwd: string): Promise<void>;
-export declare function commitAsBotUser(commitMessage: string, cwd: string): Promise<boolean>;
+export declare function commitAsBotUser(commitMessage: string, cwd: string): boolean;
 export declare function getCurrentBranchName(repoPath: string): Promise<string>;
 export declare function pullBranch(branchName: string, cwd: string): Promise<void>;
-export declare function commitFilesAndPushBranch({ commitMessage, branchName, files, repoPath, }: {
+export declare function commitFilesAsBotUser({ commitMessage, files, repoPath, }: {
     commitMessage: string;
-    branchName: string;
     files: string[];
     repoPath: string;
-}): Promise<void>;
-export declare function getFilesChanged(cwd: string): Promise<string[]>;
-export declare function mergePullRequest({ repoName, prNumber, apiKey, }: {
+}): void;
+export declare function getFilesChanged(cwd: string): string[];
+export declare function mergePullRequest({ repoName, prNumber, apiClient, }: {
     repoName: string;
     prNumber: number;
-    apiKey: string;
+    apiClient: IDashboardAPIClient;
 }): Promise<boolean>;
-export declare function deleteBranch({ repoName, branchName, apiKey, }: {
-    repoName: string;
-    branchName: string;
-    apiKey: string;
-}): Promise<void>;
 //# sourceMappingURL=git.d.ts.map

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":"AAAA,OAAO,EAAE,mBAAmB,EAAE,MAAM,4BAA4B,CAAC;AAMjE,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,iBAMnE;AAED,wBAAgB,eAAe,CAAC,aAAa,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,WAcjE;AAED,wBAAsB,oBAAoB,CAAC,QAAQ,EAAE,MAAM,mBAO1D;AAED,wBAAsB,UAAU,CAAC,UAAU,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,iBAE/D;AAED,wBAAgB,oBAAoB,CAAC,EACnC,aAAa,EACb,KAAK,EACL,QAAQ,GACT,EAAE;IACD,aAAa,EAAE,MAAM,CAAC;IACtB,KAAK,EAAE,MAAM,EAAE,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC;CAClB,QAGA;AAED,wBAAgB,eAAe,CAAC,GAAG,EAAE,MAAM,YAU1C;AAED,wBAAsB,gBAAgB,CAAC,EACrC,QAAQ,EACR,QAAQ,EACR,SAAS,GACV,EAAE;IACD,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,EAAE,mBAAmB,CAAC;CAChC,oBA8BA"}