@empiricalrun/test-gen 0.79.3 → 0.79.4

package/dist/tools/fetch-file/utils.js

@@ -1,136 +1,62 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.downscaleImageIfNeeded = downscaleImageIfNeeded;
+exports.fetchAndDownscaleImage = fetchAndDownscaleImage;
 const MAX_IMAGE_DIMENSION = 1440;
-function detectImageFormat(buffer) {
-    const arr = new Uint8Array(buffer.slice(0, 12));
-    // PNG: 89 50 4E 47
-    if (arr[0] === 0x89 &&
-        arr[1] === 0x50 &&
-        arr[2] === 0x4e &&
-        arr[3] === 0x47) {
-        return "png";
-    }
-    // JPEG: FF D8 FF
-    if (arr[0] === 0xff && arr[1] === 0xd8 && arr[2] === 0xff) {
-        return "jpeg";
-    }
-    // WebP: RIFF ... WEBP
-    if (arr[0] === 0x52 &&
-        arr[1] === 0x49 &&
-        arr[2] === 0x46 &&
-        arr[3] === 0x46 &&
-        arr[8] === 0x57 &&
-        arr[9] === 0x45 &&
-        arr[10] === 0x42 &&
-        arr[11] === 0x50) {
-        return "webp";
-    }
-    // GIF: 47 49 46
-    if (arr[0] === 0x47 && arr[1] === 0x49 && arr[2] === 0x46) {
-        return "gif";
-    }
-    return null;
+// Cloudflare zone used for image transformations via /cdn-cgi/image/
+const CF_TRANSFORM_ZONE = "https://dashboard-uploads.empirical.run";
+/**
+ * Build a Cloudflare Image Transform URL that downscales the source image
+ * so that its longest dimension is at most `MAX_IMAGE_DIMENSION`.
+ *
+ * @see https://developers.cloudflare.com/images/transform-images/transform-via-url/
+ */
+function buildCfImageTransformUrl(sourceUrl) {
+    const options = `width=${MAX_IMAGE_DIMENSION},height=${MAX_IMAGE_DIMENSION},fit=scale-down`;
+    return `${CF_TRANSFORM_ZONE}/cdn-cgi/image/${options}/${sourceUrl}`;
 }
-async function downscaleWithJsquash(buffer, logger) {
-    const format = detectImageFormat(buffer);
-    if (!format) {
-        logger.warn("Unable to detect image format for downscaling", {
-            tool: "fetchFileTool",
-        });
-        return buffer;
-    }
+/**
+ * Fetch a remote image, using Cloudflare's image transform URL to downscale
+ * large images on the fly. Falls back to a direct fetch if the transform
+ * request fails (e.g. unsupported format, CF feature not enabled).
+ *
+ * Returns both the image buffer and the actual content type from the response,
+ * since CF transforms may convert the format (e.g. PNG → JPEG).
+ */
+async function fetchAndDownscaleImage(sourceUrl, logger) {
+    const transformUrl = buildCfImageTransformUrl(sourceUrl);
     try {
-        let imageData;
-        if (format === "png") {
-            const { decode } = await import("@jsquash/png");
-            imageData = await decode(buffer);
-        }
-        else if (format === "jpeg") {
-            const { decode } = await import("@jsquash/jpeg");
-            imageData = await decode(buffer);
-        }
-        else if (format === "webp") {
-            const { decode } = await import("@jsquash/webp");
-            imageData = await decode(buffer);
-        }
-        else if (format === "gif") {
-            logger.warn("GIF resizing not supported in edge environment", {
-                tool: "fetchFileTool",
-            });
-            return buffer;
-        }
-        else {
-            logger.warn("Unsupported format for edge resizing", {
-                tool: "fetchFileTool",
-                format,
-            });
-            return buffer;
-        }
-        const { width, height } = imageData;
-        const maxDimension = Math.max(width, height);
-        if (maxDimension <= MAX_IMAGE_DIMENSION) {
-            logger.info("Image dimensions within limits, no downscaling needed", {
+        const response = await fetch(transformUrl);
+        if (response.ok) {
+            logger.info("Fetched image via Cloudflare transform", {
                 tool: "fetchFileTool",
-                width,
-                height,
+                sourceUrl,
+                transformUrl,
             });
-            return buffer;
-        }
-        const scale = MAX_IMAGE_DIMENSION / maxDimension;
-        const newWidth = Math.round(width * scale);
-        const newHeight = Math.round(height * scale);
-        const resize = (await import("@jsquash/resize")).default;
-        const resizedImageData = await resize(imageData, {
-            width: newWidth,
-            height: newHeight,
-        });
-        let encodedBuffer;
-        if (format === "png") {
-            const { encode } = await import("@jsquash/png");
-            encodedBuffer = await encode(resizedImageData);
-        }
-        else if (format === "jpeg") {
-            const { encode } = await import("@jsquash/jpeg");
-            encodedBuffer = await encode(resizedImageData);
-        }
-        else if (format === "webp") {
-            const { encode } = await import("@jsquash/webp");
-            encodedBuffer = await encode(resizedImageData);
-        }
-        else {
-            return buffer;
+            return {
+                buffer: await response.arrayBuffer(),
+                contentType: response.headers.get("content-type"),
+            };
         }
-        logger.info("Image downscaled with jsquash", {
+        logger.warn("Cloudflare transform returned non-OK, falling back to direct fetch", {
             tool: "fetchFileTool",
-            originalWidth: width,
-            originalHeight: height,
-            newWidth,
-            newHeight,
-            scale,
-            format,
+            status: response.status,
+            sourceUrl,
         });
-        return encodedBuffer;
     }
     catch (error) {
-        logger.error("Error downscaling with jsquash", {
+        logger.warn("Cloudflare transform fetch failed, falling back to direct fetch", {
             tool: "fetchFileTool",
             error,
-            format,
+            sourceUrl,
         });
-        return buffer;
     }
-}
-async function downscaleImageIfNeeded(buffer, logger) {
-    try {
-        return await downscaleWithJsquash(buffer, logger);
-    }
-    catch (error) {
-        logger.error("Error processing image for downscaling", {
-            tool: "fetchFileTool",
-            error,
-        });
-        // Return original buffer if processing fails
-        return buffer;
+    // Fallback: fetch the original image directly (no resize)
+    const directResponse = await fetch(sourceUrl);
+    if (!directResponse.ok) {
+        throw new Error(`Failed to fetch image: ${directResponse.status}`);
     }
+    return {
+        buffer: await directResponse.arrayBuffer(),
+        contentType: directResponse.headers.get("content-type"),
+    };
 }

package/dist/tools/index.d.ts

@@ -4,10 +4,10 @@ export { safeBashTool } from "./definitions/safe-bash";
 export { scrapeHtmlTool } from "./firecrawl-scrape";
 export { validateToolInput } from "./utils/validate-schema";
 type ToolOrToolDefinition = Tool | ToolDefinition;
-export declare const commonTools: ToolOrToolDefinition[];
 export declare const testGenerationTools: ToolOrToolDefinition[];
 export declare const allToolsDefinitions: ToolOrToolDefinition[];
 export declare function hasBuiltInTextEditor(model: SupportedChatModels): boolean;
+export declare function hasBuiltInWebFetch(model: SupportedChatModels): boolean;
 export declare function textEditorToolsForModel(model: SupportedChatModels): ToolOrToolDefinition[];
 export declare function textViewToolsForModel(model: SupportedChatModels): ToolOrToolDefinition[];
 export declare function toolsNeedBrowser(toolCalls: PendingToolCall[]): boolean;

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

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/tools/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,eAAe,EACf,mBAAmB,EACnB,IAAI,EACJ,cAAc,EACf,MAAM,uCAAuC,CAAC;AAC/C,OAAO,EAAE,cAAc,EAAE,MAAM,iDAAiD,CAAC;AAiCjF,OAAO,EAAE,YAAY,EAAE,MAAM,yBAAyB,CAAC;AACvD,OAAO,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AACpD,OAAO,EAAE,iBAAiB,EAAE,MAAM,yBAAyB,CAAC;AAE5D,KAAK,oBAAoB,GAAG,IAAI,GAAG,cAAc,CAAC;AAIlD,eAAO,MAAM,WAAW,EAAE,oBAAoB,EAS7C,CAAC;AAEF,eAAO,MAAM,mBAAmB,EAAE,oBAAoB,EAOrD,CAAC;AAEF,eAAO,MAAM,mBAAmB,EAAE,oBAAoB,EAerD,CAAC;AAEF,wBAAgB,oBAAoB,CAAC,KAAK,EAAE,mBAAmB,WAE9D;AAED,wBAAgB,uBAAuB,CACrC,KAAK,EAAE,mBAAmB,GACzB,oBAAoB,EAAE,CAExB;AAED,wBAAgB,qBAAqB,CACnC,KAAK,EAAE,mBAAmB,GACzB,oBAAoB,EAAE,CAExB;AAED,wBAAgB,gBAAgB,CAAC,SAAS,EAAE,eAAe,EAAE,WAM5D;AAED,wBAAgB,gBAAgB,CAAC,QAAQ,EAAE,eAAe,GAAG,MAAM,CAQlE;AAED,wBAAsB,4BAA4B,CAAC,OAAO,EAAE,cAAc,iBAoBzE"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/tools/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,eAAe,EACf,mBAAmB,EACnB,IAAI,EACJ,cAAc,EACf,MAAM,uCAAuC,CAAC;AAC/C,OAAO,EAAE,cAAc,EAAE,MAAM,iDAAiD,CAAC;AAkCjF,OAAO,EAAE,YAAY,EAAE,MAAM,yBAAyB,CAAC;AACvD,OAAO,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AACpD,OAAO,EAAE,iBAAiB,EAAE,MAAM,yBAAyB,CAAC;AAE5D,KAAK,oBAAoB,GAAG,IAAI,GAAG,cAAc,CAAC;AAelD,eAAO,MAAM,mBAAmB,EAAE,oBAAoB,EAOrD,CAAC;AAEF,eAAO,MAAM,mBAAmB,EAAE,oBAAoB,EAcrD,CAAC;AAEF,wBAAgB,oBAAoB,CAAC,KAAK,EAAE,mBAAmB,WAE9D;AAED,wBAAgB,kBAAkB,CAAC,KAAK,EAAE,mBAAmB,WAE5D;AAED,wBAAgB,uBAAuB,CACrC,KAAK,EAAE,mBAAmB,GACzB,oBAAoB,EAAE,CAExB;AAED,wBAAgB,qBAAqB,CACnC,KAAK,EAAE,mBAAmB,GACzB,oBAAoB,EAAE,CAExB;AAED,wBAAgB,gBAAgB,CAAC,SAAS,EAAE,eAAe,EAAE,WAM5D;AAED,wBAAgB,gBAAgB,CAAC,QAAQ,EAAE,eAAe,GAAG,MAAM,CAQlE;AAED,wBAAsB,4BAA4B,CAAC,OAAO,EAAE,cAAc,iBAoBzE"}

package/dist/tools/index.js

@@ -1,7 +1,8 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.allToolsDefinitions = exports.testGenerationTools = exports.commonTools = exports.validateToolInput = exports.scrapeHtmlTool = exports.safeBashTool = void 0;
+exports.allToolsDefinitions = exports.testGenerationTools = exports.validateToolInput = exports.scrapeHtmlTool = exports.safeBashTool = void 0;
 exports.hasBuiltInTextEditor = hasBuiltInTextEditor;
+exports.hasBuiltInWebFetch = hasBuiltInWebFetch;
 exports.textEditorToolsForModel = textEditorToolsForModel;
 exports.textViewToolsForModel = textViewToolsForModel;
 exports.toolsNeedBrowser = toolsNeedBrowser;
@@ -15,6 +16,7 @@ const download_build_1 = require("./definitions/download-build");
 const grep_1 = require("./definitions/grep");
 const list_tests_and_projects_1 = require("./definitions/list-tests-and-projects");
 const merge_conflicts_1 = require("./definitions/merge-conflicts");
+const publish_markdown_report_1 = require("./definitions/publish-markdown-report");
 const rename_file_1 = require("./definitions/rename-file");
 const run_test_1 = require("./definitions/run-test");
 const safe_bash_1 = require("./definitions/safe-bash");
@@ -39,7 +41,7 @@ var validate_schema_1 = require("./utils/validate-schema");
 Object.defineProperty(exports, "validateToolInput", { enumerable: true, get: function () { return validate_schema_1.validateToolInput; } });
 // TODO: Keep this list in sync with the ToolExecutor
 // Search for "Tool Executor Tools List"
-exports.commonTools = [
+const commonTools = [
     run_test_1.runTestTool,
     grep_1.grepTool,
     list_environments_1.listEnvironmentsTool,
@@ -58,13 +60,11 @@ exports.testGenerationTools = [
     upgrade_packages_1.upgradePackagesTool,
 ];
 exports.allToolsDefinitions = [
-    ...exports.commonTools,
+    ...commonTools,
     ...exports.testGenerationTools,
     analyse_video_1.analyseVideo,
     triage_summary_1.sendTriageSummaryTool,
     issues_v1_1.listIssuesTool,
-    issues_v1_1.createIssueTool,
-    issues_v1_1.updateIssueTool,
     diagnosis_fetcher_1.fetchDiagnosisReportTool,
     test_run_fetcher_1.fetchTestRunDetailsTool,
     fetch_session_diff_1.fetchSessionDiffTool,
@@ -72,10 +72,14 @@ exports.allToolsDefinitions = [
     list_tests_and_projects_1.listTestsForProjectTool,
     safe_bash_1.safeBashTool,
     firecrawl_scrape_1.scrapeHtmlTool,
+    publish_markdown_report_1.publishMarkdownReportDefinition,
 ];
 function hasBuiltInTextEditor(model) {
     return model.includes("claude");
 }
+function hasBuiltInWebFetch(model) {
+    return model.includes("claude");
+}
 function textEditorToolsForModel(model) {
     return !hasBuiltInTextEditor(model) ? Object.values(str_replace_editor_1.textEditorTools) : [];
 }

package/dist/tools/issues-v1/index.d.ts

@@ -1,4 +1,2 @@
-export { createIssueTool } from "./create-issue";
 export { listIssuesTool } from "./list-issues";
-export { updateIssueTool } from "./update-issue";
 //# sourceMappingURL=index.d.ts.map

package/dist/tools/issues-v1/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/tools/issues-v1/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,gBAAgB,CAAC;AACjD,OAAO,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAC/C,OAAO,EAAE,eAAe,EAAE,MAAM,gBAAgB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/tools/issues-v1/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC"}

package/dist/tools/issues-v1/index.js

@@ -1,9 +1,5 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.updateIssueTool = exports.listIssuesTool = exports.createIssueTool = void 0;
-var create_issue_1 = require("./create-issue");
-Object.defineProperty(exports, "createIssueTool", { enumerable: true, get: function () { return create_issue_1.createIssueTool; } });
+exports.listIssuesTool = void 0;
 var list_issues_1 = require("./list-issues");
 Object.defineProperty(exports, "listIssuesTool", { enumerable: true, get: function () { return list_issues_1.listIssuesTool; } });
-var update_issue_1 = require("./update-issue");
-Object.defineProperty(exports, "updateIssueTool", { enumerable: true, get: function () { return update_issue_1.updateIssueTool; } });

package/dist/tools/publish-markdown-report/index.d.ts

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

package/dist/tools/publish-markdown-report/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/tools/publish-markdown-report/index.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAGV,IAAI,EAEL,MAAM,uCAAuC,CAAC;AAyG/C,eAAO,MAAM,yBAAyB,EAAE,IAsJvC,CAAC"}

package/dist/tools/publish-markdown-report/index.js

@@ -0,0 +1,196 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.publishMarkdownReportTool = void 0;
+const fs = __importStar(require("fs"));
+const mime_1 = __importDefault(require("mime"));
+const path_1 = __importDefault(require("path"));
+const publish_markdown_report_1 = require("../definitions/publish-markdown-report");
+const FILE_URL_REGEX = /file:\/\/([^\s)>\]"']+)/g;
+const R2_BUCKET = "test-report";
+const R2_BASE_URL = "https://reports-r2.empirical.run";
+function extractFileUrls(markdown) {
+    const matches = markdown.matchAll(FILE_URL_REGEX);
+    const urls = new Set();
+    for (const match of matches) {
+        urls.add(match[0]);
+    }
+    return Array.from(urls);
+}
+function fileUrlToPath(fileUrl) {
+    // Remove file:// prefix to get the absolute path
+    return fileUrl.replace(/^file:\/\//, "");
+}
+async function getPresignedUrls(apiClient, files, destinationDir, logger) {
+    const response = await apiClient.request("/api/upload", {
+        method: "POST",
+        body: {
+            files,
+            destination_dir: destinationDir,
+            bucket: R2_BUCKET,
+        },
+    });
+    if (response.error) {
+        logger.error("Failed to get presigned URLs", {
+            error: response.error.message,
+        });
+        throw new Error(`Failed to get presigned URLs: ${response.error.message}`);
+    }
+    if (!response.data) {
+        throw new Error("No presigned URL data returned");
+    }
+    return response.data.presigned_urls;
+}
+async function uploadFileToPresignedUrl(filePath, uploadUrl, contentType, logger) {
+    const fileBuffer = fs.readFileSync(filePath);
+    const response = await fetch(uploadUrl, {
+        method: "PUT",
+        headers: {
+            "Content-Type": contentType,
+        },
+        body: fileBuffer,
+    });
+    if (!response.ok) {
+        const errorText = await response.text();
+        logger.error("Failed to upload file to presigned URL", {
+            filePath,
+            status: response.status,
+            error: errorText,
+        });
+        throw new Error(`Failed to upload ${filePath}: ${response.status} ${errorText}`);
+    }
+}
+exports.publishMarkdownReportTool = {
+    ...publish_markdown_report_1.publishMarkdownReportDefinition,
+    execute: async ({ input, apiClient, chatSession, logger, }) => {
+        try {
+            if (!apiClient) {
+                return {
+                    isError: true,
+                    result: "Dashboard API client is not available.",
+                };
+            }
+            if (!chatSession?.id) {
+                return {
+                    isError: true,
+                    result: "Session ID is not available. Cannot publish report without a fast-triage session.",
+                };
+            }
+            let { markdown, slack_message } = input;
+            const fileUrls = extractFileUrls(markdown);
+            if (fileUrls.length > 0) {
+                logger.info(`Found ${fileUrls.length} file:// URL(s) in markdown, uploading...`);
+                // Collect files that exist on disk
+                const filesToUpload = [];
+                const seenFileNames = new Set();
+                for (const fileUrl of fileUrls) {
+                    const filePath = fileUrlToPath(fileUrl);
+                    if (!fs.existsSync(filePath)) {
+                        logger.warn(`File not found, skipping: ${filePath}`);
+                        continue;
+                    }
+                    let fileName = path_1.default.basename(filePath);
+                    // Deduplicate file names to avoid collisions from different paths
+                    if (seenFileNames.has(fileName)) {
+                        const ext = path_1.default.extname(fileName);
+                        const base = path_1.default.basename(fileName, ext);
+                        const suffix = Math.random().toString(36).substring(2, 8);
+                        fileName = `${base}-${suffix}${ext}`;
+                    }
+                    seenFileNames.add(fileName);
+                    const contentType = mime_1.default.getType(filePath) || "application/octet-stream";
+                    filesToUpload.push({ fileUrl, filePath, fileName, contentType });
+                }
+                if (filesToUpload.length > 0) {
+                    const destinationDir = `fast-triage/${chatSession.id}/report`;
+                    // Get presigned URLs from the dashboard
+                    const presignedUrls = await getPresignedUrls(apiClient, filesToUpload.map((f) => ({
+                        name: f.fileName,
+                        type: f.contentType,
+                    })), destinationDir, logger);
+                    // Upload each file and build replacement map
+                    const urlReplacements = new Map();
+                    for (const file of filesToUpload) {
+                        const presigned = presignedUrls.find((p) => p.file_name === file.fileName);
+                        if (!presigned) {
+                            logger.warn(`No presigned URL found for ${file.fileName}, skipping`);
+                            continue;
+                        }
+                        await uploadFileToPresignedUrl(file.filePath, presigned.upload_url, file.contentType, logger);
+                        const publicUrl = `${R2_BASE_URL}/${presigned.key}`;
+                        urlReplacements.set(file.fileUrl, publicUrl);
+                    }
+                    // Replace all file:// URLs with public HTTP URLs
+                    for (const [fileUrl, httpUrl] of urlReplacements) {
+                        markdown = markdown.replaceAll(fileUrl, httpUrl);
+                        if (slack_message) {
+                            slack_message = slack_message.replaceAll(fileUrl, httpUrl);
+                        }
+                    }
+                    logger.info(`Uploaded ${urlReplacements.size} file(s) and replaced URLs`);
+                }
+            }
+            // PATCH the markdown report to the fast-triage endpoint
+            const patchBody = {
+                markdown_report: markdown,
+            };
+            if (slack_message) {
+                patchBody.slack_message = slack_message;
+            }
+            await apiClient.request(`/api/fast-triage/${chatSession.id}`, {
+                method: "PATCH",
+                body: patchBody,
+            });
+            logger.info(`Published markdown report for fast-triage ${chatSession.id}`);
+            return {
+                isError: false,
+                result: `Report published successfully with ${fileUrls.length} file(s) processed.`,
+            };
+        }
+        catch (error) {
+            logger.error("Error publishing markdown report", {
+                tool: "publishMarkdownReport",
+                error,
+            });
+            return {
+                isError: true,
+                result: error instanceof Error ? error.message : String(error),
+            };
+        }
+    },
+};

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

@@ -125,7 +125,7 @@ exports.generateTestWithBrowserAgent = {
         fs_1.default.writeFileSync(absoluteFilePath, fileBackup, "utf-8");
         const { isError, error, result: toolResult, usage } = agentResult;
         if (usage) {
-            (0, telemetry_1.trackLLMResponse)({
+            await (0, telemetry_1.trackLLMResponse)({
                 modelName: model_1.CUA_MODEL,
                 projectRepoName: chatSession?.repoName || "unknown",
                 chatSessionId: chatSession?.id,

package/dist/tools/trace-dot-zip/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/tools/trace-dot-zip/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAc,MAAM,uCAAuC,CAAC;AACzE,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,EAEL,iBAAiB,EAClB,MAAM,8BAA8B,CAAC;AAEtC,eAAO,MAAM,eAAe,EAAE,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,iBAAiB,CAAC,CA4CnE,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/tools/trace-dot-zip/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAc,MAAM,uCAAuC,CAAC;AAMzE,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAEL,iBAAiB,EAClB,MAAM,8BAA8B,CAAC;AAEtC,eAAO,MAAM,eAAe,EAAE,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,iBAAiB,CAAC,CAqDnE,CAAC"}

package/dist/tools/trace-dot-zip/index.js

@@ -1,7 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.traceDotZipTool = void 0;
-const trace_utils_1 = require("../../trace-utils");
+const trace_utils_1 = require("@empiricalrun/trace-utils");
 const trace_dot_zip_1 = require("../definitions/trace-dot-zip");
 exports.traceDotZipTool = {
     ...trace_dot_zip_1.traceDotZip,
@@ -14,16 +14,26 @@ exports.traceDotZipTool = {
                     result: "No trace.zip URL provided",
                 };
             }
-            const [networkResult, consoleResult] = await Promise.all([
-                (0, trace_utils_1.searchNetwork)({ file: traceFileUrl, errors: true }),
-                (0, trace_utils_1.getConsoleLogs)({ file: traceFileUrl, level: "error" }),
-            ]);
-            return {
-                isError: false,
-                result: JSON.stringify({
+            const pageIds = await (0, trace_utils_1.getUniquePageIds)(traceFileUrl);
+            const pagesToQuery = pageIds.length > 0 ? pageIds : [undefined];
+            const results = await Promise.all(pagesToQuery.map(async (pageId) => {
+                const [networkResult, consoleResult] = await Promise.all([
+                    (0, trace_utils_1.searchNetwork)({ file: traceFileUrl, errors: true, page: pageId }),
+                    (0, trace_utils_1.getConsoleLogs)({
+                        file: traceFileUrl,
+                        level: "error",
+                        page: pageId,
+                    }),
+                ]);
+                return {
+                    pageId,
                     failed_network_requests: networkResult.matches,
                     console_error_logs: consoleResult.logs,
-                }, null, 2),
+                };
+            }));
+            return {
+                isError: false,
+                result: JSON.stringify(results, null, 2),
             };
         }
         catch (error) {

package/dist/utils/model.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"model.d.ts","sourceRoot":"","sources":["../../src/utils/model.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,mBAAmB,EAAE,MAAM,uCAAuC,CAAC;AAE5E,eAAO,MAAM,qBAAqB,EAAE,MAAM,CAAC,MAAM,EAAE,mBAAmB,CAcrE,CAAC"}
+{"version":3,"file":"model.d.ts","sourceRoot":"","sources":["../../src/utils/model.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,mBAAmB,EAAE,MAAM,uCAAuC,CAAC;AAE5E,eAAO,MAAM,qBAAqB,EAAE,MAAM,CAAC,MAAM,EAAE,mBAAmB,CAUrE,CAAC"}

package/dist/utils/model.js

@@ -2,13 +2,9 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.CLI_ARGS_TO_MODEL_MAP = void 0;
 exports.CLI_ARGS_TO_MODEL_MAP = {
-    "claude-4": "claude-sonnet-4-20250514",
-    "claude-4-5": "claude-sonnet-4-5-20250929",
-    "claude-opus": "claude-opus-4-5-20251101",
-    "claude-sonnet": "claude-sonnet-4-5-20250929",
+    "claude-opus": "claude-opus-4-6",
+    "claude-sonnet": "claude-sonnet-4-6",
     "claude-haiku": "claude-haiku-4-5-20251001",
-    "gemini-2.5": "gemini-2.5-pro",
-    "gemini-2.5-pro": "gemini-2.5-pro",
     "gemini-3": "gemini-3-pro-preview",
     "gemini-3-pro": "gemini-3-pro-preview",
     "gpt-5": "gpt-5-2025-08-07",

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@empiricalrun/test-gen",
-  "version": "0.79.3",
+  "version": "0.79.4",
   "publishConfig": {
     "registry": "https://registry.npmjs.org/",
     "access": "public"
@@ -50,10 +50,6 @@
       "types": "./dist/file-info/index.d.ts",
       "default": "./dist/file-info/index.js"
     },
-    "./trace-utils": {
-      "types": "./dist/trace-utils/index.d.ts",
-      "default": "./dist/trace-utils/index.js"
-    },
     "./telemetry": {
       "types": "./dist/telemetry/index.d.ts",
       "default": "./dist/telemetry/index.js"
@@ -68,10 +64,6 @@
     "@babel/parser": "^7.26.3",
     "@commander-js/extra-typings": "^14.0.0",
     "@epic-web/totp": "^4.0.1",
-    "@jsquash/jpeg": "^1.6.0",
-    "@jsquash/png": "^3.1.1",
-    "@jsquash/resize": "^2.1.0",
-    "@jsquash/webp": "^1.5.0",
     "@types/sanitize-html": "^2.11.0",
     "async-retry": "^1.3.3",
     "commander": "^14.0.2",
@@ -83,7 +75,6 @@
     "ignore": "^5.3.1",
     "inquirer": "^12.4.2",
     "istextorbinary": "^9.5.0",
-    "jsdom": "^26.0.0",
     "lodash.isequal": "^4.5.0",
     "mime": "3.0.0",
     "minimatch": "^10.0.1",
@@ -103,6 +94,7 @@
     "@empiricalrun/ast-parser": "^0.0.10",
     "@empiricalrun/llm": "^0.25.2",
     "@empiricalrun/r2-uploader": "^0.9.1",
+    "@empiricalrun/trace-utils": "^0.1.0",
     "@empiricalrun/reporter": "^0.27.0",
     "@empiricalrun/test-run": "^0.14.2",
     "videostil": "0.3.4"
@@ -114,7 +106,6 @@
     "@types/detect-port": "^1.3.5",
     "@types/express": "^4.17.21",
     "@types/js-levenshtein": "^1.1.3",
-    "@types/jsdom": "^21.1.7",
     "@types/lodash.isequal": "^4.5.8",
     "@types/mime": "3.0.0",
     "@types/node": "^24.3.0",