@empiricalrun/test-gen 0.72.0 → 0.73.1

package/dist/dashboard/index.js

@@ -1,83 +1,9 @@
 "use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.RetryableHTTPError = exports.NonRetryableHTTPError = exports.HTTPError = void 0;
-exports.makeAuthenticatedRequest = makeAuthenticatedRequest;
-exports.callGitHubProxy = callGitHubProxy;
-const async_retry_1 = __importDefault(require("async-retry"));
-const totp_1 = require("./totp");
-class HTTPError extends Error {
-    status;
-    constructor(message, status) {
-        super(message);
-        this.status = status;
-    }
-}
-exports.HTTPError = HTTPError;
-class NonRetryableHTTPError extends HTTPError {
-}
-exports.NonRetryableHTTPError = NonRetryableHTTPError;
-class RetryableHTTPError extends HTTPError {
-}
-exports.RetryableHTTPError = RetryableHTTPError;
-async function makeAuthenticatedRequest(url, options, secretKey, dashboardDomain) {
-    const baseUrl = dashboardDomain || process.env.DASHBOARD_DOMAIN;
-    const fullUrl = new URL(url, baseUrl);
-    return await (0, async_retry_1.default)(async (bail) => {
-        if (options.params) {
-            Object.entries(options.params).forEach(([key, value]) => {
-                fullUrl.searchParams.append(key, value);
-            });
-        }
-        const headers = {
-            ...options.headers,
-            "Content-Type": "application/json",
-        };
-        if (!headers["Authorization"]) {
-            // Use API key auth if provided, else use TOTP
-            const totp = await (0, totp_1.getTOTP)(secretKey);
-            headers["X-Empirical-Auth-TOTP"] = totp;
-        }
-        const response = await fetch(fullUrl.toString(), {
-            headers,
-            method: options.method,
-            body: options.body ? JSON.stringify(options.body) : undefined,
-        });
-        if (!response.ok) {
-            let error;
-            const msg = `HTTP error in ${options.method} request to ${fullUrl.toString()}! Status code: ${response.status}`;
-            if (response.status < 500) {
-                // Don't retry for status codes less than 500 (client errors)
-                error = new NonRetryableHTTPError(msg, response.status);
-                bail(error);
-                return;
-            }
-            else {
-                error = new RetryableHTTPError(msg, response.status);
-            }
-            throw error;
-        }
-        return response.json();
-    }, {
-        retries: 3,
-        factor: 2,
-        minTimeout: 1_000,
-        maxTimeout: 5_000,
-        onRetry: (error, retryCount) => {
-            console.log(`Retrying... ${retryCount}`);
-            console.log(error);
-        },
-    });
-}
-async function callGitHubProxy({ method, url, body, }) {
-    return await makeAuthenticatedRequest(`${process.env.DASHBOARD_DOMAIN}/api/github/proxy`, {
-        method: "POST",
-        body: {
-            method,
-            url,
-            body,
-        },
-    });
-}
+exports.RetryableHTTPError = exports.NonRetryableHTTPError = exports.HTTPError = exports.DashboardAPIClient = void 0;
+var client_1 = require("./client");
+Object.defineProperty(exports, "DashboardAPIClient", { enumerable: true, get: function () { return client_1.DashboardAPIClient; } });
+var types_1 = require("./types");
+Object.defineProperty(exports, "HTTPError", { enumerable: true, get: function () { return types_1.HTTPError; } });
+Object.defineProperty(exports, "NonRetryableHTTPError", { enumerable: true, get: function () { return types_1.NonRetryableHTTPError; } });
+Object.defineProperty(exports, "RetryableHTTPError", { enumerable: true, get: function () { return types_1.RetryableHTTPError; } });

package/dist/dashboard/totp.d.ts

@@ -1,2 +1,2 @@
-export declare function getTOTP(secretKey?: string): Promise<string>;
+export declare function getTOTP(secretKey: string): Promise<string>;
 //# sourceMappingURL=totp.d.ts.map

package/dist/dashboard/totp.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"totp.d.ts","sourceRoot":"","sources":["../../src/dashboard/totp.ts"],"names":[],"mappings":"AAAA,wBAAsB,OAAO,CAAC,SAAS,CAAC,EAAE,MAAM,mBAe/C"}
+{"version":3,"file":"totp.d.ts","sourceRoot":"","sources":["../../src/dashboard/totp.ts"],"names":[],"mappings":"AAAA,wBAAsB,OAAO,CAAC,SAAS,EAAE,MAAM,mBAU9C"}

package/dist/dashboard/totp.js

@@ -2,14 +2,10 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.getTOTP = getTOTP;
 async function getTOTP(secretKey) {
-    const secret = secretKey || process.env.EMPIRICAL_TOTP_SK;
-    if (!secret) {
-        throw new Error("EMPIRICAL_TOTP_SK is not set");
-    }
-    // NOTE: Had to import dynamically to avoid ECMA Script Modules error
+    // Dynamic import to avoid ECMA Script Modules error
     const { generateTOTP } = await import("@epic-web/totp");
     const totp = await generateTOTP({
-        secret,
+        secret: secretKey,
         digits: 6,
         period: 30,
         algorithm: "SHA-256",

package/dist/dashboard/types.d.ts

@@ -0,0 +1,9 @@
+export declare class HTTPError extends Error {
+    status: number;
+    constructor(message: string, status: number);
+}
+export declare class NonRetryableHTTPError extends HTTPError {
+}
+export declare class RetryableHTTPError extends HTTPError {
+}
+//# sourceMappingURL=types.d.ts.map

package/dist/dashboard/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/dashboard/types.ts"],"names":[],"mappings":"AAAA,qBAAa,SAAU,SAAQ,KAAK;IAClC,MAAM,EAAE,MAAM,CAAC;gBACH,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM;CAI5C;AAED,qBAAa,qBAAsB,SAAQ,SAAS;CAAG;AAEvD,qBAAa,kBAAmB,SAAQ,SAAS;CAAG"}

package/dist/dashboard/types.js

@@ -0,0 +1,17 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.RetryableHTTPError = exports.NonRetryableHTTPError = exports.HTTPError = void 0;
+class HTTPError extends Error {
+    status;
+    constructor(message, status) {
+        super(message);
+        this.status = status;
+    }
+}
+exports.HTTPError = HTTPError;
+class NonRetryableHTTPError extends HTTPError {
+}
+exports.NonRetryableHTTPError = NonRetryableHTTPError;
+class RetryableHTTPError extends HTTPError {
+}
+exports.RetryableHTTPError = RetryableHTTPError;

package/dist/file-info/github.d.ts

@@ -1,3 +1,3 @@
-import { FileInfo } from "@empiricalrun/shared-types";
-export declare function getFileInfoFromGitHub(repoName: string, secretKey: string, dashboardDomain?: string): Promise<FileInfo>;
+import { FileInfo, IDashboardAPIClient } from "@empiricalrun/shared-types";
+export declare function getFileInfoFromGitHub(repoName: string, apiClient: IDashboardAPIClient): Promise<FileInfo>;
 //# sourceMappingURL=github.d.ts.map

package/dist/file-info/github.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"github.d.ts","sourceRoot":"","sources":["../../src/file-info/github.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,4BAA4B,CAAC;AAuFtD,wBAAsB,qBAAqB,CACzC,QAAQ,EAAE,MAAM,EAChB,SAAS,EAAE,MAAM,EACjB,eAAe,CAAC,EAAE,MAAM,GACvB,OAAO,CAAC,QAAQ,CAAC,CAyFnB"}
+{"version":3,"file":"github.d.ts","sourceRoot":"","sources":["../../src/file-info/github.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,mBAAmB,EAAE,MAAM,4BAA4B,CAAC;AAyE3E,wBAAsB,qBAAqB,CACzC,QAAQ,EAAE,MAAM,EAChB,SAAS,EAAE,mBAAmB,GAC7B,OAAO,CAAC,QAAQ,CAAC,CAwFnB"}

package/dist/file-info/github.js

@@ -5,31 +5,30 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.getFileInfoFromGitHub = getFileInfoFromGitHub;
 const path_1 = __importDefault(require("path"));
-const dashboard_1 = require("../dashboard");
-async function getFileContent(repoName, path, secretKey, dashboardDomain) {
+async function getFileContent(repoName, path, apiClient) {
     // TODO: specify ref (branch name) for the chat session
     const url = `/api/github/files?repo=${repoName}&path=${path}`;
-    const response = await (0, dashboard_1.makeAuthenticatedRequest)(url, {
+    const response = await apiClient.request(url, {
         method: "GET",
-    }, secretKey, dashboardDomain);
+    });
     if (!response.data) {
         throw new Error(`Unable to fetch file for FileInfo`);
     }
     return response.data.fileContents.content;
 }
-async function getContentsForDirectory({ repo, path, }, secretKey, dashboardDomain) {
+async function getContentsForDirectory({ repo, path, }, apiClient) {
     const url = `/api/github/tree?repo=${repo}&path=${path}`;
-    const response = await (0, dashboard_1.makeAuthenticatedRequest)(url, {
+    const response = await apiClient.request(url, {
         method: "GET",
-    }, secretKey, dashboardDomain);
+    });
     if (!response.data) {
         throw new Error(`Unable to fetch file for FileInfo`);
     }
     return response.data.tree;
 }
 // TODO: use this helper method in the dashboard
-async function getFileInfoFromGitHub(repoName, secretKey, dashboardDomain) {
-    const files = await getContentsForDirectory({ repo: repoName, path: "" }, secretKey, dashboardDomain);
+async function getFileInfoFromGitHub(repoName, apiClient) {
+    const files = await getContentsForDirectory({ repo: repoName, path: "" }, apiClient);
     const root = [];
     const nodeMap = {};
     if (!files) {
@@ -54,7 +53,7 @@ async function getFileInfoFromGitHub(repoName, secretKey, dashboardDomain) {
                 type: "file",
                 path: currentPath,
                 name: fileName,
-                getContent: async () => getFileContent(repoName, currentPath, secretKey, dashboardDomain),
+                getContent: async () => getFileContent(repoName, currentPath, apiClient),
             }
             : {
                 type: "directory",

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,YAAY,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAahD,OAAO,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AACpC,OAAO,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAsB7C,wBAAsB,UAAU,CAC9B,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,IAAI,GAAG,YAAY,EAC5B,KAAK,CAAC,EAAE,SAAS,iBAmElB;AAED,wBAAsB,UAAU,CAAC,OAAO,EAAE,IAAI,iBAY7C"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,YAAY,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAYhD,OAAO,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AACpC,OAAO,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAsB7C,wBAAsB,UAAU,CAC9B,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,IAAI,GAAG,YAAY,EAC5B,KAAK,CAAC,EAAE,SAAS,iBA0DlB;AAED,wBAAsB,UAAU,CAAC,OAAO,EAAE,IAAI,iBAY7C"}

package/dist/index.js

@@ -14,7 +14,6 @@ const run_1 = require("./agent/master/run");
 const scenarios_1 = require("./bin/utils/scenarios");
 const client_1 = __importDefault(require("./file/client"));
 const logger_1 = require("./logger");
-const session_1 = require("./session");
 var test_build_1 = require("./test-build");
 Object.defineProperty(exports, "downloadBuild", { enumerable: true, get: function () { return test_build_1.downloadBuild; } });
 const flushEvents = async () => {
@@ -45,14 +44,6 @@ async function createTest(task, pageRef, scope) {
                 },
             });
         }
-        if (testGenConfig.options && testGenConfig.options.metadata) {
-            (0, session_1.setSessionDetails)({
-                sessionId: testGenConfig.options?.metadata.testSessionId,
-                generationId: testGenConfig.options?.metadata.generationId,
-                testCaseId: testGenConfig.testCase.id,
-                projectRepoName: testGenConfig.options?.metadata.projectRepoName,
-            });
-        }
         // pageRef can be a FrameLocator, in which case we need to get the Page where the iframe is located
         let page = "owner" in pageRef ? pageRef.owner().page() : pageRef;
         const fileServiceClient = new client_1.default();

package/dist/recorder/env-variables.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"env-variables.d.ts","sourceRoot":"","sources":["../../src/recorder/env-variables.ts"],"names":[],"mappings":"AAEA,wBAAsB,yBAAyB,CAC7C,QAAQ,EAAE,MAAM,GACf,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAgCjC"}
+{"version":3,"file":"env-variables.d.ts","sourceRoot":"","sources":["../../src/recorder/env-variables.ts"],"names":[],"mappings":"AAEA,wBAAsB,yBAAyB,CAC7C,QAAQ,EAAE,MAAM,GACf,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAyBjC"}

package/dist/recorder/env-variables.js

@@ -1,17 +1,13 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.fetchEnvironmentVariables = fetchEnvironmentVariables;
-const api_client_1 = require("../auth/api-client");
+const client_1 = require("../dashboard/client");
 async function fetchEnvironmentVariables(repoName) {
     try {
-        const response = await api_client_1.apiClient.request(`/api/environment-variables?project_repo_name=${encodeURIComponent(repoName)}`, {
+        const data = await client_1.apiClient.request(`/api/environment-variables`, {
             method: "GET",
+            params: { project_repo_name: repoName },
         });
-        if (!response.ok) {
-            const errorMessage = await response.text();
-            throw new Error(`Failed to fetch environment variables: ${errorMessage}`);
-        }
-        const data = await response.json();
         if (!data.data) {
             console.error("Failed to fetch environment variables:", data);
             throw new Error("Failed to fetch environment variables");

package/dist/recorder/index.js

@@ -10,7 +10,7 @@ const inquirer_1 = __importDefault(require("inquirer"));
 const path_1 = __importDefault(require("path"));
 const utils_1 = require("../agent/browsing/utils");
 const pw_pause_1 = require("../agent/cua/pw-codegen/pw-pause");
-const api_client_1 = require("../auth/api-client");
+const client_1 = require("../dashboard/client");
 const server_1 = require("../file/server");
 const logger_1 = require("../logger");
 const env_variables_1 = require("./env-variables");
@@ -20,7 +20,7 @@ const upload_1 = require("./upload");
 const validation_1 = require("./validation");
 async function runRecorder({ name }) {
     try {
-        await api_client_1.apiClient.ensureAuthenticated();
+        await client_1.apiClient.ensureUserIsAuthenticated();
     }
     catch (error) {
         console.error("Authentication required: please run @empiricalrun/test-gen login first");
@@ -115,13 +115,13 @@ async function runRecorder({ name }) {
         },
     });
     await (0, upload_1.waitForSummaryJson)(repoDir);
-    const videoPaths = (0, upload_1.extractVideoAttachments)(repoDir);
+    const videoArtifacts = (0, upload_1.extractVideoAttachments)(repoDir);
     let attachments = [];
-    if (videoPaths.length === 0) {
+    if (videoArtifacts.length === 0) {
         logger_1.logger.warn("No video attachments found for temp test");
     }
     else {
-        const videoUrls = await (0, upload_1.uploadVideosWithSpinner)(videoPaths, name);
+        const videoUrls = await (0, upload_1.uploadVideosWithSpinner)(videoArtifacts, name);
         if (videoUrls) {
             attachments = [...attachments, ...videoUrls];
         }

package/dist/recorder/request.js

@@ -5,7 +5,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.sendToDashboardAsRequest = sendToDashboardAsRequest;
 const inquirer_1 = __importDefault(require("inquirer"));
-const api_client_1 = require("../auth/api-client");
+const client_1 = require("../dashboard/client");
 function truncateForDisplay(text, maxLineLength) {
     const lines = text.split("\n");
     return lines
@@ -174,9 +174,9 @@ async function createRequest({ repoName, title, description, }) {
         // TODO: Fix the source and sourceIdentifier
         const source = "cli";
         const sourceIdentifier = "random-string";
-        const response = await api_client_1.apiClient.request("/api/requests", {
+        const data = await client_1.apiClient.request("/api/requests", {
             method: "POST",
-            body: JSON.stringify({
+            body: {
                 source,
                 source_identifier: sourceIdentifier,
                 title,
@@ -184,13 +184,8 @@ async function createRequest({ repoName, title, description, }) {
                 project_repo_name: repoName,
                 // TODO: Enable draft requests when we have UX on dashboard sorted
                 // status: "draft",
-            }),
+            },
         });
-        if (!response.ok) {
-            const errorMessage = await response.text();
-            throw new Error(`Failed to create request: ${errorMessage}`);
-        }
-        const data = await response.json();
         const dashboardDomain = process.env.DASHBOARD_DOMAIN || "https://dash.empirical.run";
         const requestUrl = `${dashboardDomain}/${repoName}/requests/${data.data.request.id}`;
         // console.log("Request created successfully:", data);

package/dist/recorder/upload.d.ts

@@ -1,4 +1,5 @@
+import { ArtifactInputPath, ArtifactInputUrl } from "@empiricalrun/shared-types";
 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[]>;
+export declare function extractVideoAttachments(repoDir: string): Array<ArtifactInputPath | ArtifactInputUrl>;
+export declare function uploadVideosWithSpinner(videoArtifacts: Array<ArtifactInputPath | ArtifactInputUrl>, 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":"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"}
+{"version":3,"file":"upload.d.ts","sourceRoot":"","sources":["../../src/recorder/upload.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,iBAAiB,EACjB,gBAAgB,EACjB,MAAM,4BAA4B,CAAC;AAepC,wBAAsB,kBAAkB,CAAC,OAAO,EAAE,MAAM,iBAmBvD;AAED,wBAAgB,uBAAuB,CACrC,OAAO,EAAE,MAAM,GACd,KAAK,CAAC,iBAAiB,GAAG,gBAAgB,CAAC,CAoB7C;AAmFD,wBAAsB,uBAAuB,CAC3C,cAAc,EAAE,KAAK,CAAC,iBAAiB,GAAG,gBAAgB,CAAC,EAC3D,QAAQ,EAAE,MAAM,GACf,OAAO,CAAC,IAAI,GAAG,MAAM,EAAE,CAAC,CA0B1B"}

package/dist/recorder/upload.js

@@ -9,7 +9,7 @@ exports.uploadVideosWithSpinner = uploadVideosWithSpinner;
 const fs_1 = __importDefault(require("fs"));
 const path_1 = __importDefault(require("path"));
 const utils_1 = require("../artifacts/utils");
-const api_client_1 = require("../auth/api-client");
+const client_1 = require("../dashboard/client");
 const slug_1 = require("../utils/slug");
 const ASSETS_PRODUCTION_BUCKET = "empirical-assets-production";
 const BUCKET_DOMAINS = {
@@ -46,11 +46,9 @@ function extractVideoAttachments(repoDir) {
             return [];
         }
         const summaryContent = JSON.parse(fs_1.default.readFileSync(summaryPath, "utf-8"));
-        const attachments = (0, utils_1.extractAttachmentsFromPlaywrightJSONReport)(summaryContent, "temp test");
-        const videoPaths = attachments
-            .filter((attachment) => attachment.contentType === "video/webm")
-            .map((attachment) => attachment.path);
-        return videoPaths;
+        const artifacts = (0, utils_1.extractAttachmentsFromPlaywrightJSONReport)(summaryContent, "temp test");
+        const videoArtifacts = artifacts.filter((artifact) => artifact.contentType === "video/webm");
+        return videoArtifacts;
     }
     catch (error) {
         console.warn("Error processing summary.json:", error);
@@ -62,10 +60,11 @@ function buildVideoUrl(localPath, directory, bucket) {
     const fileName = path_1.default.basename(localPath);
     return `https://${domain}/${path_1.default.join(directory, fileName)}`;
 }
-async function uploadVideos(videoPaths, testSlug) {
-    if (videoPaths.length === 0) {
+async function uploadVideos(videoArtifacts, testSlug) {
+    if (videoArtifacts.length === 0) {
         return null;
     }
+    const videoPaths = videoArtifacts.map((a) => a.path);
     const uploadDestinationDir = `recorder-uploads/${testSlug}`;
     const bucket = ASSETS_PRODUCTION_BUCKET;
     try {
@@ -79,22 +78,14 @@ async function uploadVideos(videoPaths, testSlug) {
         if (files.length === 0) {
             return null;
         }
-        const presignedResponse = await api_client_1.apiClient.request("/api/upload", {
+        const presignedResult = await client_1.apiClient.request("/api/upload", {
             method: "POST",
-            headers: {
-                "Content-Type": "application/json",
-            },
-            body: JSON.stringify({
+            body: {
                 files,
                 destination_dir: uploadDestinationDir,
                 bucket,
-            }),
+            },
         });
-        if (!presignedResponse.ok) {
-            console.warn(`Failed to get pre-signed URLs: ${presignedResponse.status} ${presignedResponse.statusText}`);
-            return null;
-        }
-        const presignedResult = await presignedResponse.json();
         if (presignedResult.error) {
             console.warn("Pre-signed URL error:", presignedResult.error.message);
             return null;
@@ -126,17 +117,24 @@ async function uploadVideos(videoPaths, testSlug) {
         return null;
     }
 }
-async function uploadVideosWithSpinner(videoPaths, testName) {
+async function uploadVideosWithSpinner(videoArtifacts, testName) {
     // Use dynamic imports to avoid issues with ESM modules
     // https://gist.github.com/sindresorhus/a39789f98801d908bbc7ff3ecc99d99c
     const ora = (await import("ora")).default;
     const uploadSpinner = ora("Uploading video recordings...").start();
     try {
         const slugifiedTestName = `${(0, slug_1.slugify)(testName)}-${Math.random().toString(36).substring(2, 8)}`;
-        const uploaded = await uploadVideos(videoPaths, slugifiedTestName);
+        const pathArtifacts = videoArtifacts.filter((v) => "path" in v);
+        const urlArtifacts = videoArtifacts.filter((v) => "url" in v);
+        const urlsForUrlArtifacts = urlArtifacts.map((v) => v.url);
+        if (pathArtifacts.length === 0) {
+            uploadSpinner.succeed("Video recordings uploaded successfully");
+            return urlsForUrlArtifacts;
+        }
+        const uploaded = await uploadVideos(pathArtifacts, slugifiedTestName);
         if (uploaded) {
             uploadSpinner.succeed("Video recordings uploaded successfully");
-            return uploaded;
+            return [...uploaded, ...urlsForUrlArtifacts];
         }
         else {
             uploadSpinner.warn("Video upload skipped or failed");

package/dist/tools/commit-and-create-pr.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"commit-and-create-pr.d.ts","sourceRoot":"","sources":["../../src/tools/commit-and-create-pr.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAEV,IAAI,EAEL,MAAM,4BAA4B,CAAC;AA0CpC,eAAO,MAAM,qBAAqB,EAAE,IAoFnC,CAAC"}
+{"version":3,"file":"commit-and-create-pr.d.ts","sourceRoot":"","sources":["../../src/tools/commit-and-create-pr.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,4BAA4B,CAAC;AA+BvD,eAAO,MAAM,qBAAqB,EAAE,IAwFnC,CAAC"}

package/dist/tools/commit-and-create-pr.js

@@ -5,6 +5,7 @@ const child_process_1 = require("child_process");
 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 createPullRequestSchema = zod_1.z.object({
     pullRequestTitle: zod_1.z
         .string()
@@ -16,16 +17,6 @@ For example, if you used the test run tool, you should include the results (and
 videos and other artifacts that help the reviewer gain more context and confidence in the changes. If tests pass, reviewer will see the video and merge the PR.
 If tests fail, reviewer will see the video and the test artifacts, and will be able to help you debug the issue.`),
 });
-function descriptionWithMetadata(chatSessionId, description) {
-    const timestamp = new Date()
-        .toISOString()
-        .replace("T", " ")
-        .replace("Z", " UTC");
-    const sessionInfo = chatSessionId
-        ? `\n\nPR created from session #${chatSessionId}`
-        : ``;
-    return `${description}${sessionInfo}\n\n<sup>Updated at ${timestamp}</sup>`;
-}
 exports.createPullRequestTool = {
     schema: {
         name: "createPullRequest",
@@ -40,51 +31,67 @@ Don't ask the user for this information, just come up with it yourself.
         parameters: createPullRequestSchema,
     },
     needsBrowser: false,
-    execute: async ({ input, repoPath, apiKey, chatSessionId, }) => {
+    execute: async ({ input, repoPath, apiClient, chatSession }) => {
         try {
             const { pullRequestTitle, pullRequestDescription } = input;
-            const branchName = await (0, git_1.getCurrentBranchName)(repoPath);
-            const repoUrl = (0, child_process_1.execSync)("git config --get remote.origin.url", {
+            const remoteUrl = (0, child_process_1.execSync)("git config --get remote.origin.url", {
                 cwd: repoPath,
             })
                 .toString()
                 .trim();
-            const { owner, repo } = (0, utils_1.parseGitHubUrl)(repoUrl);
+            const { owner, repo } = (0, utils_1.parseGitHubUrl)(remoteUrl);
+            const branchName = await (0, git_1.getCurrentBranchName)(repoPath);
             (0, child_process_1.execSync)(`git push origin ${branchName} --set-upstream`, {
                 cwd: repoPath,
             });
-            const existingPR = await (0, utils_1.findExistingPR)({
+            let pullRequest = await (0, utils_1.getExistingPR)({
                 owner,
                 repo,
                 branchName,
-                apiKey,
+                apiClient,
             });
-            if (existingPR) {
-                await (0, utils_1.updatePullRequest)({
+            if (pullRequest) {
+                pullRequest = await (0, utils_1.updatePullRequest)({
+                    owner,
+                    repo,
+                    prNumber: pullRequest.number,
+                    body: (0, pr_description_1.addMetadataToPRDescription)(pullRequestDescription, chatSession),
+                    apiClient,
+                });
+                const mergeableState = await (0, utils_1.getMergeableState)({
                     owner,
                     repo,
-                    prNumber: existingPR.number,
-                    body: descriptionWithMetadata(chatSessionId, pullRequestDescription),
-                    apiKey,
+                    pullRequest,
+                    apiClient,
                 });
+                const stateDescription = (0, utils_1.getMergeableStateDescription)(mergeableState);
                 return {
                     isError: false,
-                    result: `Committed and pushed changes to existing PR: ${existingPR.html_url}`,
+                    result: `Updated existing PR: ${pullRequest.html_url}\n\nMergeable state: ${mergeableState} - ${stateDescription}`,
+                };
+            }
+            else {
+                pullRequest = await (0, utils_1.createPullRequest)({
+                    owner,
+                    repo,
+                    title: pullRequestTitle,
+                    head: branchName,
+                    base: "main",
+                    body: (0, pr_description_1.addMetadataToPRDescription)(pullRequestDescription, chatSession),
+                    apiClient,
+                });
+                const mergeableState = await (0, utils_1.getMergeableState)({
+                    owner,
+                    repo,
+                    pullRequest,
+                    apiClient,
+                });
+                const stateDescription = (0, utils_1.getMergeableStateDescription)(mergeableState);
+                return {
+                    isError: false,
+                    result: `Created a new PR: ${pullRequest.html_url}\n\nMergeable state: ${mergeableState} - ${stateDescription}`,
                 };
             }
-            const pr = await (0, utils_1.createPullRequest)({
-                owner,
-                repo,
-                title: pullRequestTitle,
-                head: branchName,
-                base: "main",
-                body: descriptionWithMetadata(chatSessionId, pullRequestDescription),
-                apiKey,
-            });
-            return {
-                isError: false,
-                result: `Committed and pushed changes to new PR: ${pr.html_url}`,
-            };
         }
         catch (error) {
             return {

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"}