@empiricalrun/test-gen 0.69.8 → 0.70.0

package/CHANGELOG.md

@@ -1,5 +1,17 @@
 # @empiricalrun/test-gen
 
+## 0.70.0
+
+### Minor Changes
+
+- 2daef2d: feat: upgrade-pkgs tool schema and PR details updated
+
+### Patch Changes
+
+- 1348b1a: chore: refactor slack client into internal vs external usage
+- 4274dc5: fix: json report output should exist for video upload
+- 39b105b: fix: auto-merge checks in upgrade package tool
+
 ## 0.69.8
 
 ### Patch Changes

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

@@ -5,8 +5,4 @@ export declare function runChatAgentForCLI({ useDiskForChatState, selectedModel,
     useDiskForChatState: boolean;
     initialPromptContent: string | undefined;
 }): Promise<void>;
-export declare function runChatAgentForDashboard({ chatSessionId, selectedModel, }: {
-    selectedModel: SupportedChatModels;
-    chatSessionId: number;
-}): Promise<void>;
 //# sourceMappingURL=index.d.ts.map

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

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

package/dist/agent/chat/index.js

@@ -2,7 +2,6 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.fetchEnvironmentVariables = fetchEnvironmentVariables;
 exports.runChatAgentForCLI = runChatAgentForCLI;
-exports.runChatAgentForDashboard = runChatAgentForDashboard;
 const llm_1 = require("@empiricalrun/llm");
 const chat_1 = require("@empiricalrun/llm/chat");
 const picocolors_1 = require("picocolors");
@@ -159,73 +158,3 @@ async function runChatAgentForCLI({ useDiskForChatState, selectedModel, initialP
     const usageSummary = chatModel.getUsageSummary();
     console.log(`\n${(0, picocolors_1.gray)("Usage summary -> " + usageSummary)}`);
 }
-async function getChatSessionFromDashboard(chatSessionId) {
-    const response = await fetch(`${DASHBOARD_DOMAIN}/api/chat-sessions/${chatSessionId}`, {
-        headers: {
-            "Content-Type": "application/json",
-            Authorization: `Bearer ${process.env.EMPIRICALRUN_API_KEY}`,
-        },
-    });
-    if (!response.ok) {
-        throw new Error(`Failed to get chat session: ${response.statusText}`);
-    }
-    const data = await response.json();
-    return data.data.chat_session;
-}
-async function runChatAgentForDashboard({ chatSessionId, selectedModel, }) {
-    const chatSession = await getChatSessionFromDashboard(chatSessionId);
-    let chatState = chatSession.chat_state;
-    // If not already canonical, migrate to canonical format
-    if (!chatState.version || chatState.version !== state_1.LATEST_CHAT_STATE_VERSION) {
-        chatState = (0, state_1.migrateChatState)(chatState);
-    }
-    const branchName = chatSession.branch_name;
-    const trace = llm_1.langfuseInstance?.trace({
-        id: chatSession.langfuse_trace_id,
-        name: "chat_agent",
-        input: chatState,
-        tags: [selectedModel, "chat_agent"],
-        metadata: {
-            chatSessionId,
-        },
-    });
-    if (!process.env.EMPIRICALRUN_API_KEY) {
-        throw new Error("EMPIRICALRUN_API_KEY is not set");
-    }
-    const toolCallService = new tool_call_service_1.ToolCallService({
-        chatSessionId,
-        selectedModel,
-        branchName,
-        repoPath: process.cwd(),
-        apiKey: process.env.EMPIRICALRUN_API_KEY,
-        trace,
-        featureFlags: [],
-    });
-    await (0, git_1.checkoutBranch)(branchName, process.cwd());
-    let chatModel = (0, chat_1.createChatModel)(chatState.messages, selectedModel);
-    let reporterFunc = async (chatState, latest) => {
-        const response = await fetch(`${DASHBOARD_DOMAIN}/api/chat-sessions/${chatSessionId}`, {
-            method: "PATCH",
-            body: JSON.stringify({
-                chat_state: chatState,
-                last_assistant_message: latest?.textMessage,
-            }),
-            headers: {
-                "Content-Type": "application/json",
-                Authorization: `Bearer ${process.env.EMPIRICALRUN_API_KEY}`,
-            },
-        });
-        const data = await response.json();
-        console.log(`Patch request sent for chat session: ${JSON.stringify(data)}`);
-    };
-    const fileInfo = await (0, file_tree_1.getFileInfoFromFS)(process.cwd());
-    await (0, agent_loop_1.chatAgentLoop)({
-        chatModel,
-        selectedModel,
-        reporter: reporterFunc,
-        trace,
-        toolCallService,
-        fileInfo,
-        isToolExecutionRemote: false,
-    });
-}

package/dist/bin/index.js

@@ -31,20 +31,12 @@ const scenarios_1 = require("./utils/scenarios");
 dotenv_1.default.config({
     path: [".env.local", ".env"],
 });
-async function runChatAgent({ modelInput, chatSessionId, useDiskForChatState, initialPromptPath, }) {
+async function runChatAgent({ modelInput, useDiskForChatState, initialPromptPath, }) {
     if (modelInput && !utils_2.ARGS_TO_MODEL_MAP[modelInput]) {
         throw new Error(`Invalid chat model: ${modelInput}`);
     }
     const defaultModel = "claude-3-7-sonnet-20250219";
     const specifiedModel = modelInput && utils_2.ARGS_TO_MODEL_MAP[modelInput];
-    if (chatSessionId) {
-        // If --chat-session-id is provided, we run the chat agent for the dashboard
-        // and not CLI (where user can input their own prompt)
-        return await (0, chat_1.runChatAgentForDashboard)({
-            chatSessionId: Number(chatSessionId),
-            selectedModel: specifiedModel || defaultModel,
-        });
-    }
     let initialPromptContent = undefined;
     if (initialPromptPath) {
         try {
@@ -318,10 +310,8 @@ async function main() {
         .option("--chat-model <model>", "LLM to use (claude-3-7, claude-4 or gemini-2.5)")
         .option("--use-disk-for-chat-state", "Save and load chat state from disk")
         .option("--initial-prompt <path>", "Path to an initial prompt file (e.g. prompt.md)")
-        .option("--chat-session-id <chat-session-id>", "Identifier for chat session (fetched from dash.empirical.run)")
         .action(async (options) => {
         await runChatAgent({
-            chatSessionId: options.chatSessionId,
             modelInput: options.chatModel,
             useDiskForChatState: options.useDiskForChatState,
             initialPromptPath: options.initialPrompt,

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

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/recorder/index.ts"],"names":[],"mappings":"AA4BA,wBAAsB,WAAW,CAAC,EAAE,IAAI,EAAE,EAAE;IAAE,IAAI,EAAE,MAAM,CAAA;CAAE,iBA8I3D"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/recorder/index.ts"],"names":[],"mappings":"AAgCA,wBAAsB,WAAW,CAAC,EAAE,IAAI,EAAE,EAAE;IAAE,IAAI,EAAE,MAAM,CAAA;CAAE,iBA+I3D"}

package/dist/recorder/index.js

@@ -114,6 +114,7 @@ async function runRecorder({ name }) {
             IPC_FILE_SERVICE_PORT: availablePort.toString(),
         },
     });
+    await (0, upload_1.waitForSummaryJson)(repoDir);
     const videoPaths = (0, upload_1.extractVideoAttachments)(repoDir);
     let attachments = [];
     if (videoPaths.length === 0) {

package/dist/recorder/upload.d.ts

@@ -1,3 +1,4 @@
+export declare function waitForSummaryJson(repoDir: string): Promise<void>;
 export declare function extractVideoAttachments(repoDir: string): string[];
 export declare function uploadVideosWithSpinner(videoPaths: string[], testName: string): Promise<void | string[]>;
 //# sourceMappingURL=upload.d.ts.map

package/dist/recorder/upload.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"upload.d.ts","sourceRoot":"","sources":["../../src/recorder/upload.ts"],"names":[],"mappings":"AAaA,wBAAgB,uBAAuB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,EAAE,CAoBjE;AA6FD,wBAAsB,uBAAuB,CAC3C,UAAU,EAAE,MAAM,EAAE,EACpB,QAAQ,EAAE,MAAM,GACf,OAAO,CAAC,IAAI,GAAG,MAAM,EAAE,CAAC,CAiB1B"}
+{"version":3,"file":"upload.d.ts","sourceRoot":"","sources":["../../src/recorder/upload.ts"],"names":[],"mappings":"AAcA,wBAAsB,kBAAkB,CAAC,OAAO,EAAE,MAAM,iBAmBvD;AAED,wBAAgB,uBAAuB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,EAAE,CAoBjE;AA6FD,wBAAsB,uBAAuB,CAC3C,UAAU,EAAE,MAAM,EAAE,EACpB,QAAQ,EAAE,MAAM,GACf,OAAO,CAAC,IAAI,GAAG,MAAM,EAAE,CAAC,CAiB1B"}

package/dist/recorder/upload.js

@@ -3,6 +3,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
+exports.waitForSummaryJson = waitForSummaryJson;
 exports.extractVideoAttachments = extractVideoAttachments;
 exports.uploadVideosWithSpinner = uploadVideosWithSpinner;
 const fs_1 = __importDefault(require("fs"));
@@ -15,9 +16,31 @@ const BUCKET_DOMAINS = {
     "empirical-assets-staging": "assets-staging.empirical.run",
     "empirical-assets-production": "assets.empirical.run",
 };
+const SUMMARY_JSON = (repoDir) => path_1.default.join(repoDir, "summary.json");
+async function waitForSummaryJson(repoDir) {
+    const summaryPath = SUMMARY_JSON(repoDir);
+    const maxTimeout = 5_000;
+    const timerPromise = new Promise((resolve, reject) => {
+        setTimeout(() => {
+            reject(new Error("Timeout waiting for summary.json"));
+        }, maxTimeout);
+    });
+    const fsExistsPromise = new Promise((resolve) => {
+        const checkFile = () => {
+            if (fs_1.default.existsSync(summaryPath)) {
+                resolve();
+            }
+            else {
+                setTimeout(checkFile, 100);
+            }
+        };
+        checkFile();
+    });
+    return Promise.race([timerPromise, fsExistsPromise]);
+}
 function extractVideoAttachments(repoDir) {
     try {
-        const summaryPath = path_1.default.join(repoDir, "summary.json");
+        const summaryPath = SUMMARY_JSON(repoDir);
         if (!fs_1.default.existsSync(summaryPath)) {
             console.log("summary.json not found");
             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;AA+BvD,eAAO,MAAM,mBAAmB,EAAE,IA+GjC,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;AAuBvD,eAAO,MAAM,mBAAmB,EAAE,IAiIjC,CAAC"}

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

@@ -10,58 +10,65 @@ const zod_1 = require("zod");
 const git_1 = require("../../utils/git");
 const utils_1 = require("../utils");
 const utils_2 = require("./utils");
-const pkgs = [
-    "@empiricalrun/playwright-utils",
-    "@empiricalrun/eslint-config",
-    "@empiricalrun/typescript-config",
-];
-const packageSpecSchema = zod_1.z.union([
-    zod_1.z.string(),
-    zod_1.z.object({
-        name: zod_1.z.string(),
-        version: zod_1.z.string().optional(),
-    }),
-]);
+const packageSpecSchema = zod_1.z.object({
+    name: zod_1.z.string(),
+    version: zod_1.z.string(),
+});
 const upgradePackagesSchema = zod_1.z.object({
-    packages: zod_1.z.array(packageSpecSchema),
+    packages: zod_1.z.array(packageSpecSchema).min(1),
 });
 exports.upgradePackagesTool = {
     schema: {
         name: "upgradePackages",
-        description: `Automatically upgrades specific packages. You can optionally specify a version for each package. If a version is not provided, the package will be upgraded to the latest version. The tool handles the entire workflow: updating package files, committing changes, creating/updating PRs, and managing merges. For patch updates (e.g., 1.0.1 → 1.0.2), it automatically merges the PR and cleans up. For minor/major updates, it leaves the PR open for review. Returns a success message with the PR URL and merge status.`,
+        description: `This tool can be used to upgrade NPM packages in this repository. To run this tool, specify a list of packages with their names and version strings. At least one package must be specified.
+This tool handles the entire workflow: editing the package.json file, running "npm install", generating lockfile, committing the changes and creating a pull request. For patch updates (e.g., 1.0.1 → 1.0.2), the tool will automatically merge the PR for minor updates. For major updates, the user needs to review the PR and merge it manually.`,
         parameters: upgradePackagesSchema,
     },
     needsBrowser: false,
     execute: async ({ input, repoPath, apiKey, }) => {
-        const { packages } = input;
-        let packagesToUpdate = packages.length ? packages : pkgs;
-        const packageSpecs = packagesToUpdate.map((p) => typeof p === "string" ? { name: p, version: undefined } : p);
+        const { packages: packagesToUpdate } = input;
         try {
             const repoName = path_1.default.basename(repoPath);
             const branchName = await (0, git_1.getCurrentBranchName)(repoPath);
             const packageJsonPath = path_1.default.join(repoPath, "package.json");
             const packageJson = JSON.parse(fs_1.default.readFileSync(packageJsonPath, "utf-8"));
-            for (let { name: pkgName, version } of packageSpecs) {
+            const changes = [];
+            for (let { name: pkgName, version } of packagesToUpdate) {
                 const isDevDep = !!packageJson.devDependencies?.[pkgName];
                 try {
-                    await (0, utils_2.upgradeAndStagePackage)({
+                    const updatedPackage = await (0, utils_2.upgradeAndStagePackage)({
                         repoPath,
                         pkgName,
                         version,
                         isDevDep,
                     });
+                    changes.push(updatedPackage);
                 }
                 catch (err) {
                     console.error(`Failed to upgrade ${pkgName}:`, err);
+                    return {
+                        result: `Failed to upgrade ${pkgName}: ${err.message}`,
+                        isError: true,
+                    };
                 }
             }
+            const filesChanged = await (0, git_1.getFilesChanged)(repoPath);
+            const hasChanges = filesChanged.length > 0;
+            if (!hasChanges) {
+                return {
+                    result: "Success: No package changes detected, nothing to commit. Skipping PR creation.",
+                    isError: false,
+                };
+            }
             await (0, git_1.commitFilesAndPushBranch)({
-                commitMessage: "[create-pull-request] automated change [skip ci]",
+                commitMessage: `[upgrade-packages] upgrade ${changes.length} packages [skip ci]`,
                 branchName,
                 files: ["package.json", "package-lock.json"],
                 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 prTitle = `chore: upgrade ${changes.length} package(s)`;
             let shouldMerge = false;
             let prNumber = null;
             let merged = false;
@@ -71,8 +78,8 @@ exports.upgradePackagesTool = {
                     repo: repoName,
                     apiKey,
                     branchName,
-                    title: "Upgrade packages",
-                    body: "Upgrade packages to specified versions",
+                    title: prTitle,
+                    body: prBody,
                     labels: ["automated"],
                 });
                 prNumber = pr.number;
@@ -109,7 +116,7 @@ exports.upgradePackagesTool = {
                     ? "created but merge failed"
                     : "created but not merged (non-patch updates)";
             return {
-                result: `Success: PR ${status}. Link: ${prLink}`,
+                result: `Success: Successfully upgraded ${changes.length} package(s). PR ${status}. Link: ${prLink}`,
                 isError: false,
             };
         }

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

@@ -1,4 +1,10 @@
 export declare function getLatestVersion(packageName: string): Promise<any>;
+export declare function parsePackageJsonDiff({ diff, }: {
+    diff: string;
+}): Record<string, {
+    old?: string;
+    new?: string;
+}>;
 export declare function shouldMergePR({ repoName, prNumber, apiKey, }: {
     repoName: string;
     prNumber: number;
@@ -9,5 +15,8 @@ export declare function upgradeAndStagePackage({ repoPath, pkgName, version, isD
     pkgName: string;
     version?: string;
     isDevDep: boolean;
-}): Promise<void>;
+}): Promise<{
+    name: string;
+    version: any;
+}>;
 //# sourceMappingURL=utils.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../src/tools/upgrade-packages/utils.ts"],"names":[],"mappings":"AAYA,wBAAsB,gBAAgB,CAAC,WAAW,EAAE,MAAM,gBAczD;AAmED,wBAAsB,aAAa,CAAC,EAClC,QAAQ,EACR,QAAQ,EACR,MAAM,GACP,EAAE;IACD,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;CAChB,oBA6BA;AAED,wBAAsB,sBAAsB,CAAC,EAC3C,QAAQ,EACR,OAAO,EACP,OAAO,EACP,QAAQ,GACT,EAAE;IACD,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,OAAO,CAAC;CACnB,iBAUA"}
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../src/tools/upgrade-packages/utils.ts"],"names":[],"mappings":"AAoBA,wBAAsB,gBAAgB,CAAC,WAAW,EAAE,MAAM,gBAkBzD;AA8BD,wBAAgB,oBAAoB,CAAC,EACnC,IAAI,GACL,EAAE;IACD,IAAI,EAAE,MAAM,CAAC;CACd,GAAG,MAAM,CAAC,MAAM,EAAE;IAAE,GAAG,CAAC,EAAE,MAAM,CAAC;IAAC,GAAG,CAAC,EAAE,MAAM,CAAA;CAAE,CAAC,CAuBjD;AAED,wBAAsB,aAAa,CAAC,EAClC,QAAQ,EACR,QAAQ,EACR,MAAM,GACP,EAAE;IACD,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;CAChB,oBA8BA;AAED,wBAAsB,sBAAsB,CAAC,EAC3C,QAAQ,EACR,OAAO,EACP,OAAO,EACP,QAAQ,GACT,EAAE;IACD,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,OAAO,CAAC;CACnB;;;GAcA"}

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

@@ -1,11 +1,17 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.getLatestVersion = getLatestVersion;
+exports.parsePackageJsonDiff = parsePackageJsonDiff;
 exports.shouldMergePR = shouldMergePR;
 exports.upgradeAndStagePackage = upgradeAndStagePackage;
 const child_process_1 = require("child_process");
 const utils_1 = require("../utils");
 function isPatchUpdate(currentVersion, newVersion) {
+    if (!currentVersion || !newVersion) {
+        // If current is undefined, we might be installing a new package
+        // Both must be truthy for this to be a patch version update
+        return false;
+    }
     const current = currentVersion.replace(/^[\^~]/, "").split(".");
     const next = newVersion.replace(/^[\^~]/, "").split(".");
     return (current[0] === next[0] && current[1] === next[1] && current[2] !== next[2]);
@@ -15,7 +21,10 @@ async function getLatestVersion(packageName) {
     try {
         const response = await fetch(url);
         if (!response.ok) {
-            throw new Error(`Failed to fetch data for package: ${packageName}`);
+            const errorMsg = response.status === 404
+                ? `Package '${packageName}' not found on npm registry`
+                : `Failed to fetch data for package: ${packageName}`;
+            throw new Error(errorMsg);
         }
         const packageInfo = await response.json();
         console.log(`Latest version of ${packageName}: ${packageInfo.version}`);
@@ -26,68 +35,63 @@ async function getLatestVersion(packageName) {
         throw error;
     }
 }
-async function getPackageJsonChanges({ repoName, prNumber, apiKey, }) {
+async function getGitDiffForPackageJson({ repoName, prNumber, apiKey, }) {
     const url = `${utils_1.GITHUB_API_BASE}/${repoName}/pulls/${prNumber}/files`;
-    try {
-        const files = await (0, utils_1.callGitHubProxy)({
-            method: "GET",
-            url: url,
-            apiKey,
-        });
-        if (!files) {
-            throw new Error("Failed to fetch PR files via proxy");
-        }
-        const packageJsonChanges = files.find((file) => file.filename === "package.json");
-        if (!packageJsonChanges) {
-            console.log("No package.json changes found in PR");
-            return null;
-        }
-        const patchLines = packageJsonChanges.patch.split("\n");
-        const changes = {};
-        // Process the patch lines to find version changes
-        for (let i = 0; i < patchLines.length; i++) {
-            const line = patchLines[i];
-            // Look for lines that change version numbers
-            const match = line.match(/^[-+]\s*"([^"]+)":\s*"([^"]+)"/);
-            if (match) {
-                const [, pkg, version] = match;
-                const changeType = line.startsWith("+") ? "new" : "old";
-                if (!changes[pkg]) {
-                    changes[pkg] = {};
-                }
-                changes[pkg][changeType] = version;
+    const files = await (0, utils_1.callGitHubProxy)({
+        method: "GET",
+        url: url,
+        apiKey,
+    });
+    if (!files) {
+        throw new Error("Failed to fetch PR files via proxy");
+    }
+    const packageJsonChanges = files.find((file) => file.filename === "package.json");
+    if (!packageJsonChanges) {
+        console.log("No package.json changes found in PR");
+        return null;
+    }
+    return packageJsonChanges.patch;
+}
+function parsePackageJsonDiff({ diff, }) {
+    const patchLines = diff.split("\n");
+    const changes = {};
+    for (let i = 0; i < patchLines.length; i++) {
+        const line = patchLines[i];
+        // Look for lines that change version numbers
+        const match = line.match(/^[-+]\s*"([^"]+)":\s*"([^"]+)"/);
+        if (match) {
+            const [, pkg, version] = match;
+            if (!pkg) {
+                console.warn("No package name found in line:", line);
+                continue;
             }
-        }
-        // Filter out any packages that don't have both old and new versions
-        const validChanges = {};
-        for (const [pkg, versions] of Object.entries(changes)) {
-            if (versions.old && versions.new) {
-                validChanges[pkg] = {
-                    oldVersion: versions.old,
-                    newVersion: versions.new,
-                };
+            const changeType = line.startsWith("+") ? "new" : "old";
+            if (!changes[pkg]) {
+                changes[pkg] = {};
             }
+            changes[pkg][changeType] = version;
         }
-        return validChanges;
-    }
-    catch (error) {
-        console.error(`Error fetching PR changes: ${error.message}`);
-        throw error;
     }
+    return changes;
 }
 async function shouldMergePR({ repoName, prNumber, apiKey, }) {
-    const changes = await getPackageJsonChanges({
+    const gitDiff = await getGitDiffForPackageJson({
         repoName,
         prNumber,
         apiKey,
     });
-    if (!changes) {
+    if (!gitDiff) {
+        console.log("No package.json changes found, skipping merge");
+        return false;
+    }
+    const changes = parsePackageJsonDiff({ diff: gitDiff });
+    if (Object.keys(changes).length === 0) {
         console.log("No package.json changes found, skipping merge");
         return false;
     }
     let allPatchUpdates = true;
     for (const [pkg, versions] of Object.entries(changes)) {
-        const { oldVersion, newVersion } = versions;
+        const { old: oldVersion, new: newVersion } = versions;
         if (!isPatchUpdate(oldVersion, newVersion)) {
             console.log(`${pkg}: ${oldVersion} -> ${newVersion} is not a patch update`);
             allPatchUpdates = false;
@@ -103,4 +107,8 @@ async function upgradeAndStagePackage({ repoPath, pkgName, version, isDevDep, })
     (0, child_process_1.execSync)(`npm i ${pkgName}@${pkgVersion} ${isDevDep ? "--save-dev" : "--save"}`, { cwd: repoPath });
     console.log(`Updated package: ${pkgName} version to ${pkgVersion} in ${repoPath}`);
     (0, child_process_1.execSync)("git add package.json package-lock.json", { cwd: repoPath });
+    return {
+        name: pkgName,
+        version: pkgVersion,
+    };
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@empiricalrun/test-gen",
-  "version": "0.69.8",
+  "version": "0.70.0",
   "publishConfig": {
     "registry": "https://registry.npmjs.org/",
     "access": "public"