@empiricalrun/test-gen 0.72.0 → 0.73.0

package/dist/tools/executor/utils/index.js

@@ -1,46 +1,12 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.GITHUB_API_BASE = void 0;
-exports.makeDashboardRequest = makeDashboardRequest;
-exports.callGitHubProxy = callGitHubProxy;
 exports.parseGitHubUrl = parseGitHubUrl;
-exports.findExistingPR = findExistingPR;
+exports.getExistingPR = getExistingPR;
+exports.getMergeableState = getMergeableState;
+exports.getMergeableStateDescription = getMergeableStateDescription;
 exports.createPullRequest = createPullRequest;
 exports.updatePullRequest = updatePullRequest;
 exports.findOrCreatePullRequest = findOrCreatePullRequest;
-async function makeDashboardRequest({ path, method = "GET", body, apiKey, }) {
-    const requestHeaders = {
-        "Content-Type": "application/json",
-        Authorization: `Bearer ${apiKey}`,
-        "User-Agent": "empiricalrun/test-gen",
-    };
-    const baseUrl = process.env.DASHBOARD_DOMAIN || "https://dash.empirical.run";
-    const response = await fetch(`${baseUrl}${path}`, {
-        method,
-        headers: requestHeaders,
-        ...(body && { body: JSON.stringify(body) }),
-    });
-    if (!response.ok) {
-        const errorBody = await response.text();
-        throw new Error(`API request failed for ${method} ${path} (Status: ${response.status}). Body: ${errorBody}`);
-    }
-    return await response.json();
-}
-async function callGitHubProxy({ method, url, body, apiKey, }) {
-    const githubApiPath = url.replace("https://api.github.com", "");
-    return makeDashboardRequest({
-        path: "/api/github/proxy",
-        method: "POST",
-        body: {
-            method,
-            url: githubApiPath,
-            body,
-        },
-        apiKey,
-    });
-}
-// Common GitHub constants
-exports.GITHUB_API_BASE = "https://api.github.com/repos/empirical-run";
 function parseGitHubUrl(url) {
     const githubIndex = url.indexOf("github.com");
     if (githubIndex === -1) {
@@ -55,19 +21,56 @@ function parseGitHubUrl(url) {
     }
     return { owner, repo };
 }
-async function findExistingPR({ owner, repo, branchName, apiKey, }) {
-    const prs = (await callGitHubProxy({
+async function getExistingPR({ owner, repo, branchName, apiClient, }) {
+    const prs = await apiClient.callGitHubProxy({
         method: "GET",
-        url: `https://api.github.com/repos/${owner}/${repo}/pulls`,
+        url: `/repos/${owner}/${repo}/pulls`,
         body: {
             head: `${owner}:${branchName}`,
             state: "open",
         },
-        apiKey,
-    }));
+    });
     return prs?.find((pr) => pr.head.ref === branchName) || null;
 }
-async function createPullRequest({ owner, repo, title, head, base = "main", body, labels, apiKey, }) {
+async function getMergeableState({ owner, repo, pullRequest, apiClient, }) {
+    if (pullRequest.mergeable_state !== "unknown") {
+        return pullRequest.mergeable_state;
+    }
+    const maxAttempts = 10;
+    let attempts = 0;
+    while (attempts < maxAttempts) {
+        await new Promise((resolve) => setTimeout(resolve, 5_000));
+        attempts++;
+        try {
+            const updatedPR = await apiClient.callGitHubProxy({
+                method: "GET",
+                url: `/repos/${owner}/${repo}/pulls/${pullRequest.number}`,
+            });
+            if (updatedPR.mergeable_state !== "unknown") {
+                return updatedPR.mergeable_state;
+            }
+        }
+        catch (error) {
+            console.warn(`Failed to fetch PR mergeable state (attempt ${attempts}):`, error);
+        }
+    }
+    return "unknown";
+}
+function getMergeableStateDescription(state) {
+    switch (state) {
+        case "clean":
+            return "The pull request is mergeable without conflicts.";
+        case "dirty":
+            return "The pull request has merge conflicts that must be resolved.";
+        case "unstable":
+            return "The pull request has failing or pending required status checks.";
+        case "unknown":
+            return "The mergeability of the pull request is still being calculated.";
+        default:
+            return `The pull request has mergeable state: ${state}.`;
+    }
+}
+async function createPullRequest({ owner, repo, title, head, base = "main", body, labels, apiClient, }) {
     const createPrBody = {
         title,
         head,
@@ -75,40 +78,36 @@ async function createPullRequest({ owner, repo, title, head, base = "main", body
         body,
         ...(labels && { labels }),
     };
-    const pr = (await callGitHubProxy({
+    const pr = await apiClient.callGitHubProxy({
         method: "POST",
-        url: `https://api.github.com/repos/${owner}/${repo}/pulls`,
+        url: `/repos/${owner}/${repo}/pulls`,
         body: createPrBody,
-        apiKey,
-    }));
+    });
     if (!pr || !("number" in pr)) {
         throw new Error(`Failed to create PR or PR number missing in response.`);
     }
     return pr;
 }
-async function updatePullRequest({ owner, repo, prNumber, body, apiKey, }) {
+async function updatePullRequest({ owner, repo, prNumber, body, apiClient, }) {
     const updateBody = {};
     if (body)
         updateBody.body = body;
-    return (await callGitHubProxy({
+    return await apiClient.callGitHubProxy({
         method: "PATCH",
-        url: `https://api.github.com/repos/${owner}/${repo}/pulls/${prNumber}`,
+        url: `/repos/${owner}/${repo}/pulls/${prNumber}`,
         body: updateBody,
-        apiKey,
-    }));
+    });
 }
-async function findOrCreatePullRequest({ owner, repo, branchName, title, body, labels, apiKey, }) {
-    // Check for existing PR
-    const existingPR = await findExistingPR({
+async function findOrCreatePullRequest({ owner, repo, branchName, title, body, labels, apiClient, }) {
+    const existingPR = await getExistingPR({
         owner,
         repo,
         branchName,
-        apiKey,
+        apiClient,
     });
     if (existingPR) {
         return { pr: existingPR, isNew: false };
     }
-    // Create new PR
     const newPR = await createPullRequest({
         owner,
         repo,
@@ -116,7 +115,7 @@ async function findOrCreatePullRequest({ owner, repo, branchName, title, body, l
         head: branchName,
         body,
         labels,
-        apiKey,
+        apiClient,
     });
     return { pr: newPR, isNew: true };
 }

package/dist/tools/executor/utils/pr-description.d.ts

@@ -0,0 +1,4 @@
+import type { ChatSessionInfo } from "@empiricalrun/shared-types";
+export declare function generateSessionMetadata(chatSession: ChatSessionInfo | null): string;
+export declare function addMetadataToPRDescription(description: string, chatSession: ChatSessionInfo | null): string;
+//# sourceMappingURL=pr-description.d.ts.map

package/dist/tools/executor/utils/pr-description.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"pr-description.d.ts","sourceRoot":"","sources":["../../../../src/tools/executor/utils/pr-description.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,4BAA4B,CAAC;AAElE,wBAAgB,uBAAuB,CACrC,WAAW,EAAE,eAAe,GAAG,IAAI,GAClC,MAAM,CASR;AAED,wBAAgB,0BAA0B,CACxC,WAAW,EAAE,MAAM,EACnB,WAAW,EAAE,eAAe,GAAG,IAAI,GAClC,MAAM,CAWR"}

package/dist/tools/executor/utils/pr-description.js

@@ -0,0 +1,24 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.generateSessionMetadata = generateSessionMetadata;
+exports.addMetadataToPRDescription = addMetadataToPRDescription;
+function generateSessionMetadata(chatSession) {
+    if (!chatSession) {
+        return "";
+    }
+    let userInfo = "";
+    if (chatSession.createdBy) {
+        userInfo = ` by ${chatSession.createdBy}`;
+    }
+    return `PR created from [session #${chatSession.id}](${chatSession.url})${userInfo}`;
+}
+function addMetadataToPRDescription(description, chatSession) {
+    const timestamp = new Date()
+        .toISOString()
+        .replace("T", " ")
+        .replace("Z", " UTC");
+    const sessionInfo = chatSession
+        ? `\n\n${generateSessionMetadata(chatSession)}`
+        : "";
+    return `${description}${sessionInfo}\n\n<sup>Updated at ${timestamp}</sup>`;
+}

package/dist/tools/fetch-image/index.d.ts

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

package/dist/tools/fetch-image/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/tools/fetch-image/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAA8B,MAAM,4BAA4B,CAAC;AAS9E,eAAO,MAAM,cAAc,EAAE,IAiD5B,CAAC"}

package/dist/tools/fetch-image/index.js

@@ -0,0 +1,56 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.fetchImageTool = void 0;
+const zod_1 = require("zod");
+const SUPPORTED_CONTENT_TYPES = ["image/png", "image/jpeg"];
+const fetchImageSchema = zod_1.z.object({
+    url: zod_1.z.string(),
+});
+exports.fetchImageTool = {
+    schema: {
+        name: "fetchImage",
+        description: "Use this tool to fetch image data from any valid URL. It takes an image URL as input and returns the image in base64 format. Automatically call this tool when a user provides an image link or message containing a URL pointing to an image, so that you can view or analyze the image content effectively",
+        parameters: fetchImageSchema,
+    },
+    needsBrowser: false,
+    execute: async ({ input }) => {
+        const { url } = input;
+        try {
+            const response = await fetch(url);
+            if (!response.ok) {
+                console.error(`Failed to fetch image from ${url}: ${response.statusText}`);
+                return {
+                    result: `Failed to fetch image from ${url}: ${response.statusText}`,
+                    isError: true,
+                };
+            }
+            const contentType = response.headers.get("content-type");
+            if (!contentType || !SUPPORTED_CONTENT_TYPES.includes(contentType)) {
+                const errorMessage = `Invalid content type: ${contentType}. URL must return an image.`;
+                console.error(errorMessage);
+                return {
+                    result: errorMessage,
+                    isError: true,
+                };
+            }
+            const buffer = await response.arrayBuffer();
+            const base64 = Buffer.from(buffer).toString("base64");
+            return {
+                result: [
+                    {
+                        type: contentType,
+                        base64Data: base64,
+                    },
+                ],
+                isError: false,
+            };
+        }
+        catch (error) {
+            console.error("Error fetching image", error);
+            return {
+                result: `Error fetching image: ${error}`,
+                isError: true,
+            };
+        }
+    },
+};

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

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/tools/grep/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAEV,IAAI,EAGL,MAAM,4BAA4B,CAAC;AA8HpC,eAAO,MAAM,QAAQ,EAAE,IA4BtB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/tools/grep/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAc,MAAM,4BAA4B,CAAC;AA8HnE,eAAO,MAAM,QAAQ,EAAE,IAkBtB,CAAC"}

package/dist/tools/grep/index.js

@@ -125,7 +125,7 @@ Search is case insensitive and regex patterns are not supported.`,
         parameters: GrepInputSchema,
     },
     needsBrowser: false,
-    execute: async ({ input, repoPath, }) => {
+    execute: async ({ input, repoPath }) => {
         if ((0, ripgrep_1.isRgAvailable)()) {
             return usingRipgrep(input, repoPath);
         }

package/dist/tools/index.d.ts

@@ -0,0 +1,4 @@
+import { ServicePayload, SupportedChatModels } from "@empiricalrun/shared-types";
+export declare function availableTools(model?: SupportedChatModels): import("@empiricalrun/shared-types").ToolDefinition[];
+export declare function sendToolRequestToRemoteQueue(payload: ServicePayload): Promise<void>;
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/tools/index.ts"],"names":[],"mappings":"AACA,OAAO,EAEL,cAAc,EACd,mBAAmB,EACpB,MAAM,4BAA4B,CAAC;AAkBpC,wBAAgB,cAAc,CAAC,KAAK,CAAC,EAAE,mBAAmB,yDAoBzD;AAUD,wBAAsB,4BAA4B,CAAC,OAAO,EAAE,cAAc,iBAoBzE"}

package/dist/tools/index.js

@@ -0,0 +1,60 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.availableTools = availableTools;
+exports.sendToolRequestToRemoteQueue = sendToolRequestToRemoteQueue;
+const chat_1 = require("@empiricalrun/llm/chat");
+const SQSClient_1 = require("../utils/SQSClient");
+const commit_and_create_pr_1 = require("./commit-and-create-pr");
+const run_test_1 = require("./definitions/run-test");
+const str_replace_editor_1 = require("./definitions/str_replace_editor");
+const test_gen_browser_1 = require("./definitions/test-gen-browser");
+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 test_run_fetcher_1 = require("./test-run-fetcher");
+const upgrade_packages_1 = require("./upgrade-packages");
+const queue_1 = require("./utils/queue");
+function availableTools(model) {
+    // TODO: Keep this list in sync with the ToolExecutor
+    const 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,
+        merge_conflicts_1.mergeConflictsTool,
+        list_environments_1.listEnvironmentsTool,
+        download_build_1.downloadBuildTool,
+        upgrade_packages_1.upgradePackagesTool,
+        fetch_image_1.fetchImageTool,
+        delete_file_1.deleteFileTool,
+    ];
+    if (model && (0, chat_1.getProviderForModel)(model) !== "claude") {
+        tools.push(...Object.values(str_replace_editor_1.textEditorTools));
+    }
+    return tools;
+}
+function toolsNeedBrowser(toolCalls) {
+    const allTools = availableTools();
+    return toolCalls.some((toolCall) => allTools.find((t) => t.schema.name === toolCall.name)?.needsBrowser);
+}
+async function sendToolRequestToRemoteQueue(payload) {
+    const needsBrowser = toolsNeedBrowser(payload.toolCalls);
+    const queueUrl = (0, queue_1.getQueueUrl)(payload.toolCalls, needsBrowser);
+    if (!queueUrl) {
+        throw new Error(`queueUrl is required for remote execution.`);
+    }
+    const region = process.env.AWS_REGION;
+    const accessKeyId = process.env.AWS_ACCESS_KEY_ID;
+    const secretAccessKey = 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/list-environments.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"list-environments.d.ts","sourceRoot":"","sources":["../../src/tools/list-environments.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAEV,IAAI,EAEL,MAAM,4BAA4B,CAAC;AAKpC,eAAO,MAAM,oBAAoB,EAAE,IAsDlC,CAAC"}
+{"version":3,"file":"list-environments.d.ts","sourceRoot":"","sources":["../../src/tools/list-environments.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,4BAA4B,CAAC;AAGvD,eAAO,MAAM,oBAAoB,EAAE,IA2ClC,CAAC"}

package/dist/tools/list-environments.js

@@ -2,7 +2,6 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.listEnvironmentsTool = void 0;
 const zod_1 = require("zod");
-const utils_1 = require("./executor/utils");
 exports.listEnvironmentsTool = {
     schema: {
         name: "listEnvironments",
@@ -10,13 +9,11 @@ exports.listEnvironmentsTool = {
         parameters: zod_1.z.object({}),
     },
     needsBrowser: false,
-    execute: async ({ apiKey, }) => {
+    execute: async ({ apiClient }) => {
         let response;
         try {
-            response = await (0, utils_1.makeDashboardRequest)({
-                path: `/api/environments/list`,
+            response = await apiClient.request(`/api/environments/list`, {
                 method: "GET",
-                apiKey,
             });
         }
         catch (error) {

package/dist/tools/merge-conflicts.d.ts

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

package/dist/tools/merge-conflicts.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"merge-conflicts.d.ts","sourceRoot":"","sources":["../../src/tools/merge-conflicts.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,4BAA4B,CAAC;AA6CvD,eAAO,MAAM,kBAAkB,EAAE,IAgEhC,CAAC"}

package/dist/tools/merge-conflicts.js

@@ -0,0 +1,107 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.mergeConflictsTool = void 0;
+const child_process_1 = require("child_process");
+const fs_1 = __importDefault(require("fs"));
+const path_1 = __importDefault(require("path"));
+const zod_1 = require("zod");
+const git_1 = require("./executor/utils/git");
+function runMergeMain(repoPath) {
+    try {
+        (0, child_process_1.execSync)("git merge --no-ff origin/main", { stdio: "pipe", cwd: repoPath });
+        return "CLEAN";
+    }
+    catch (error) {
+        // Check if it's a merge conflict (exit code 1) vs other errors
+        if (error.status === 1) {
+            return "CONFLICTS";
+        }
+        throw error;
+    }
+}
+function getConflictedFilesWithContents(repoPath) {
+    const output = (0, child_process_1.execSync)("git diff --name-only --diff-filter=U", {
+        encoding: "utf8",
+        stdio: "pipe",
+        cwd: repoPath,
+    });
+    const fileNames = output
+        .trim()
+        .split("\n")
+        .filter((file) => file.length > 0);
+    const filesWithContents = {};
+    fileNames.forEach((fileName) => {
+        const fullPath = path_1.default.join(repoPath, fileName);
+        const content = fs_1.default.readFileSync(fullPath, "utf8");
+        filesWithContents[fileName] = content;
+    });
+    return filesWithContents;
+}
+const mergeConflictsToolScheme = zod_1.z.object({});
+exports.mergeConflictsTool = {
+    schema: {
+        name: "checkForMergeConflicts",
+        description: `This tool runs the git merge command to merge main into your
+current working branch with the following command.
+
+\`\`\`
+git merge --no-ff origin/main
+\`\`\`
+
+If this commands fails with merge conflicts, it will still create a dirty commit
+that has merge conflicts, which can be resolved with the file editing tools that you are
+provided with.
+`,
+        parameters: mergeConflictsToolScheme,
+    },
+    needsBrowser: false,
+    execute: async ({ repoPath }) => {
+        try {
+            // Fetch main branch (git clone runs --single-branch), and then run merge
+            (0, child_process_1.execSync)("git remote set-branches --add origin main", { cwd: repoPath });
+            (0, child_process_1.execSync)("git fetch origin main", { cwd: repoPath });
+            // Run merge command and commit (with or without conflict)
+            const mergeResult = runMergeMain(repoPath);
+            const branchName = await (0, git_1.getCurrentBranchName)(repoPath);
+            if (mergeResult === "CLEAN") {
+                (0, child_process_1.execSync)(`git push origin ${branchName}`, { cwd: repoPath });
+                return {
+                    isError: false,
+                    result: `Merge from main was successful, without any conflicts.`,
+                };
+            }
+            else if (mergeResult === "CONFLICTS") {
+                const files = getConflictedFilesWithContents(repoPath);
+                (0, git_1.commitFilesAsBotUser)({
+                    commitMessage: "Merge from main with conflicts",
+                    files: Object.keys(files),
+                    repoPath,
+                });
+                (0, child_process_1.execSync)(`git push origin ${branchName}`, { cwd: repoPath });
+                // TODO: This can be same as file view result
+                const filesResult = Object.entries(files)
+                    .map(([fileName, content]) => `File: ${fileName}\n\n\`\`\`\n${content}\n\`\`\``)
+                    .join("\n\n");
+                return {
+                    isError: false,
+                    result: `Merge from main is committed, with conflicts. Use text edit tools to resolve them.\n\n${filesResult}`,
+                };
+            }
+            else {
+                return {
+                    isError: true,
+                    result: `An unknown error has occured.`,
+                };
+            }
+        }
+        catch (error) {
+            return {
+                isError: true,
+                result: `Failed to run merge conflicts tool: ${error instanceof Error ? error.message : String(error)}`,
+            };
+        }
+    },
+};

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

@@ -1 +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"}
+{"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,IAiFzB,CAAC"}

package/dist/tools/run-test.js

@@ -68,8 +68,8 @@ exports.runTestTool = {
                 projects: [input.project],
                 envOverrides,
             });
-            const attachments = (0, utils_1.extractAttachmentsFromPlaywrightJSONReport)(result.summaryJson);
-            void collectArtifacts?.(attachments);
+            const artifacts = (0, utils_1.extractAttachmentsFromPlaywrightJSONReport)(result.summaryJson);
+            void collectArtifacts?.(artifacts);
             return {
                 isError: false,
                 result: buildResult({

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

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/tools/test-run-fetcher/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,4BAA4B,CAAC;AAcvD,wBAAgB,0BAA0B,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,CAOnE;AAED,eAAO,MAAM,sBAAsB,EAAE,IAgIpC,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/tools/test-run-fetcher/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,4BAA4B,CAAC;AAavD,wBAAgB,0BAA0B,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,CAOnE;AAED,eAAO,MAAM,sBAAsB,EAAE,IA0HpC,CAAC"}

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

@@ -3,7 +3,6 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.fetchTestRunReportTool = void 0;
 exports.extractPathAfterSourceRepo = extractPathAfterSourceRepo;
 const zod_1 = require("zod");
-const utils_1 = require("../executor/utils");
 const TestRunSchema = zod_1.z.object({
     testRunUrl: zod_1.z
         .string()
@@ -25,7 +24,7 @@ exports.fetchTestRunReportTool = {
         parameters: TestRunSchema,
     },
     needsBrowser: false,
-    execute: async ({ input, apiKey, }) => {
+    execute: async ({ input, apiClient }) => {
         const { testRunUrl } = input;
         // Remove query parameters if they exist
         const urlWithoutParams = testRunUrl.split("?")[0] || testRunUrl;
@@ -41,10 +40,7 @@ exports.fetchTestRunReportTool = {
         }
         let data = null;
         try {
-            data = await (0, utils_1.makeDashboardRequest)({
-                path: `/api/test-runs/${runId}?repo_name=${repoName}`,
-                apiKey,
-            });
+            data = await apiClient.request(`/api/test-runs/${runId}?repo_name=${repoName}`, { method: "GET" });
         }
         catch (error) {
             return {

package/dist/tools/upgrade-packages/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/tools/upgrade-packages/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,4BAA4B,CAAC;AAuBvD,eAAO,MAAM,mBAAmB,EAAE,IAwIjC,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/tools/upgrade-packages/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,4BAA4B,CAAC;AAyBvD,eAAO,MAAM,mBAAmB,EAAE,IA2HjC,CAAC"}

package/dist/tools/upgrade-packages/index.js

@@ -4,11 +4,13 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.upgradePackagesTool = void 0;
+const child_process_1 = require("child_process");
 const fs_1 = __importDefault(require("fs"));
 const path_1 = __importDefault(require("path"));
 const zod_1 = require("zod");
 const utils_1 = require("../executor/utils");
 const git_1 = require("../executor/utils/git");
+const pr_description_1 = require("../executor/utils/pr-description");
 const utils_2 = require("./utils");
 const packageSpecSchema = zod_1.z.object({
     name: zod_1.z.string(),
@@ -25,7 +27,7 @@ This tool handles the entire workflow: editing the package.json file, running "n
         parameters: upgradePackagesSchema,
     },
     needsBrowser: false,
-    execute: async ({ input, repoPath, apiKey, chatSessionId, }) => {
+    execute: async ({ input, repoPath, apiClient, chatSession }) => {
         const { packages: packagesToUpdate } = input;
         try {
             const repoName = path_1.default.basename(repoPath);
@@ -52,7 +54,7 @@ This tool handles the entire workflow: editing the package.json file, running "n
                     };
                 }
             }
-            const filesChanged = await (0, git_1.getFilesChanged)(repoPath);
+            const filesChanged = (0, git_1.getFilesChanged)(repoPath);
             const hasChanges = filesChanged.length > 0;
             if (!hasChanges) {
                 return {
@@ -61,17 +63,14 @@ This tool handles the entire workflow: editing the package.json file, running "n
                 };
             }
             const updateMessage = `upgrade ${changes.length} package${changes.length === 1 ? "" : "s"}`;
-            await (0, git_1.commitFilesAndPushBranch)({
+            (0, git_1.commitFilesAsBotUser)({
                 commitMessage: `[upgrade-packages] ${updateMessage}`,
-                branchName,
                 files: ["package.json", "package-lock.json"],
                 repoPath,
             });
+            (0, child_process_1.execSync)(`git push origin ${branchName}`, { cwd: repoPath });
             await new Promise((resolve) => setTimeout(resolve, 5_000));
             const prBody = `Upgraded the following packages:\n\n${changes.map((c) => `- \`${c.name}\` → ${c.version ? c.version : "latest"}`).join("\n")}`;
-            const sessionMetadata = chatSessionId
-                ? `PR created from session #${chatSessionId}`
-                : ``;
             const prTitle = `chore: ${updateMessage}`;
             let shouldMerge = false;
             let prNumber = null;
@@ -80,23 +79,23 @@ This tool handles the entire workflow: editing the package.json file, running "n
                 const { pr } = await (0, utils_1.findOrCreatePullRequest)({
                     owner: "empirical-run",
                     repo: repoName,
-                    apiKey,
+                    apiClient,
                     branchName,
                     title: prTitle,
-                    body: `${prBody}\n\n${sessionMetadata}`,
+                    body: (0, pr_description_1.addMetadataToPRDescription)(prBody, chatSession),
                     labels: ["automated"],
                 });
                 prNumber = pr.number;
                 if (!pr.number) {
                     throw new Error("Failed to create PR");
                 }
-                shouldMerge = await (0, utils_2.shouldMergePR)({ repoName, prNumber, apiKey });
+                shouldMerge = await (0, utils_2.shouldMergePR)({ repoName, prNumber, apiClient });
                 if (shouldMerge) {
                     console.log("All changes are patch updates, proceeding with merge");
                     merged = await (0, git_1.mergePullRequest)({
                         repoName,
                         prNumber,
-                        apiKey,
+                        apiClient,
                     });
                 }
                 else {

package/dist/tools/upgrade-packages/utils.d.ts

@@ -1,3 +1,4 @@
+import { IDashboardAPIClient } from "@empiricalrun/shared-types";
 export declare function getLatestVersion(packageName: string): Promise<any>;
 export declare function parsePackageJsonDiff({ diff, }: {
     diff: string;
@@ -5,10 +6,10 @@ export declare function parsePackageJsonDiff({ diff, }: {
     old?: string;
     new?: string;
 }>;
-export declare function shouldMergePR({ repoName, prNumber, apiKey, }: {
+export declare function shouldMergePR({ repoName, prNumber, apiClient, }: {
     repoName: string;
     prNumber: number;
-    apiKey: string;
+    apiClient: IDashboardAPIClient;
 }): Promise<boolean>;
 export declare function upgradeAndStagePackage({ repoPath, pkgName, version, isDevDep, }: {
     repoPath: string;