@empiricalrun/test-gen 0.71.2 → 0.72.0

package/dist/auth/api-client.js

@@ -9,6 +9,10 @@ class APIClient {
         this.appUrl = process.env.DASHBOARD_DOMAIN || "https://dash.empirical.run";
     }
     async request(endpoint, options = {}) {
+        const apiKey = process.env.EMPIRICALRUN_API_KEY;
+        if (apiKey) {
+            return this.makeRequest(endpoint, options, apiKey);
+        }
         await this.ensureAuthenticated();
         const tokens = await (0, token_store_1.getStoredTokens)();
         if (!tokens) {
@@ -28,13 +32,13 @@ class APIClient {
         }
         return response;
     }
-    async makeRequest(endpoint, options, accessToken) {
+    async makeRequest(endpoint, options, authToken) {
         const url = endpoint.startsWith("http")
             ? endpoint
             : `${this.appUrl}${endpoint}`;
         // Don't set Content-Type for FormData (let browser set it automatically)
         const headers = {
-            Authorization: `Bearer ${accessToken}`,
+            Authorization: `Bearer ${authToken}`,
             ...options.headers,
         };
         // Only set Content-Type to application/json if body is not FormData

package/dist/bin/environments.d.ts

@@ -1,2 +1,2 @@
-export declare function listEnvironments(): Promise<void>;
+export declare function listEnvironments(repoDir: string): Promise<void>;
 //# sourceMappingURL=environments.d.ts.map

package/dist/bin/environments.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"environments.d.ts","sourceRoot":"","sources":["../../src/bin/environments.ts"],"names":[],"mappings":"AASA,wBAAsB,gBAAgB,kBAiErC"}
+{"version":3,"file":"environments.d.ts","sourceRoot":"","sources":["../../src/bin/environments.ts"],"names":[],"mappings":"AAkEA,wBAAsB,gBAAgB,CAAC,OAAO,EAAE,MAAM,iBAqDrD"}

package/dist/bin/environments.js

@@ -1,12 +1,66 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.listEnvironments = listEnvironments;
+const test_run_1 = require("@empiricalrun/test-run");
 const api_client_1 = require("../auth/api-client");
 const validation_1 = require("../recorder/validation");
-// TODO: Match against playwright project names - show that to the user
-async function listEnvironments() {
+async function getMatchingPlaywrightProjects(patterns, repoDir) {
+    if (patterns.length === 0) {
+        return [];
+    }
+    try {
+        // Use test-run's existing logic that handles comma-separated values and all patterns
+        return await (0, test_run_1.generateProjectFilters)({
+            platform: test_run_1.Platform.WEB,
+            filteringSets: patterns,
+            repoDir,
+        });
+    }
+    catch (error) {
+        console.warn("Warning: Could not match Playwright projects:", error);
+        return [];
+    }
+}
+function printLatestBuild(latestBuild) {
+    if (!latestBuild) {
+        console.log(`  └─ No builds available`);
+        return;
+    }
+    console.log("");
+    const commit = latestBuild.commit
+        ? latestBuild.commit.substring(0, 7)
+        : "N/A";
+    const branch = latestBuild.branch || "N/A";
+    // Calculate relative time
+    const createdAt = new Date(latestBuild.created_at);
+    const now = new Date();
+    const diffMs = now.getTime() - createdAt.getTime();
+    const diffDays = Math.floor(diffMs / (1000 * 60 * 60 * 24));
+    const diffHours = Math.floor(diffMs / (1000 * 60 * 60));
+    const diffMinutes = Math.floor(diffMs / (1000 * 60));
+    let timeAgo;
+    if (diffDays > 0) {
+        timeAgo = `${diffDays} day${diffDays > 1 ? "s" : ""} ago`;
+    }
+    else if (diffHours > 0) {
+        timeAgo = `${diffHours} hour${diffHours > 1 ? "s" : ""} ago`;
+    }
+    else if (diffMinutes > 0) {
+        timeAgo = `${diffMinutes} minute${diffMinutes > 1 ? "s" : ""} ago`;
+    }
+    else {
+        timeAgo = "Just now";
+    }
+    console.log(`  ┌─ Latest build`);
+    console.log(`  │ Commit:    ${commit}`);
+    console.log(`  │ Branch:    ${branch}`);
+    console.log(`  │ Created:   ${timeAgo}`);
+    console.log(`  │ URL:       ${latestBuild.build_url}`);
+    console.log(`  └─`);
+}
+async function listEnvironments(repoDir) {
     try {
-        let repoName = await (0, validation_1.validatePackageJson)(process.cwd());
+        let repoName = await (0, validation_1.validatePackageJson)(repoDir);
         const response = await api_client_1.apiClient.request(`/api/environments/list?project_repo_name=${encodeURIComponent(repoName)}`);
         if (!response.ok) {
             console.error(`❌ Failed to fetch environments for repo ${repoName}:`, response.statusText);
@@ -14,47 +68,27 @@ async function listEnvironments() {
         }
         const result = await response.json();
         if (result.data?.environments && result.data.environments.length > 0) {
-            result.data.environments.forEach((env, index) => {
-                if (index > 0) {
-                    console.log(""); // Add spacing between environments
+            for (const env of result.data.environments) {
+                const envIndex = result.data.environments.indexOf(env);
+                if (envIndex > 0) {
+                    console.log("");
                 }
                 console.log(`Environment: ${env.name}`);
-                if (env.latest_build) {
-                    const commit = env.latest_build.commit
-                        ? env.latest_build.commit.substring(0, 7)
-                        : "N/A";
-                    const branch = env.latest_build.branch || "N/A";
-                    // Calculate relative time
-                    const createdAt = new Date(env.latest_build.created_at);
-                    const now = new Date();
-                    const diffMs = now.getTime() - createdAt.getTime();
-                    const diffDays = Math.floor(diffMs / (1000 * 60 * 60 * 24));
-                    const diffHours = Math.floor(diffMs / (1000 * 60 * 60));
-                    const diffMinutes = Math.floor(diffMs / (1000 * 60));
-                    let timeAgo;
-                    if (diffDays > 0) {
-                        timeAgo = `${diffDays} day${diffDays > 1 ? "s" : ""} ago`;
-                    }
-                    else if (diffHours > 0) {
-                        timeAgo = `${diffHours} hour${diffHours > 1 ? "s" : ""} ago`;
-                    }
-                    else if (diffMinutes > 0) {
-                        timeAgo = `${diffMinutes} minute${diffMinutes > 1 ? "s" : ""} ago`;
+                console.log(`  Slug: ${env.slug}`);
+                if (env.playwright_projects && env.playwright_projects.length > 0) {
+                    const matchingProjects = await getMatchingPlaywrightProjects(env.playwright_projects, repoDir);
+                    if (matchingProjects.length > 0) {
+                        console.log(`  Playwright Projects: ${matchingProjects.join(", ")}`);
                     }
                     else {
-                        timeAgo = "Just now";
+                        console.log(`  Playwright Projects: No matches found for patterns: ${env.playwright_projects.join(", ")}`);
                     }
-                    console.log(`  ┌─ Latest build`);
-                    console.log(`  │ Commit:    ${commit}`);
-                    console.log(`  │ Branch:    ${branch}`);
-                    console.log(`  │ Created:   ${timeAgo}`);
-                    console.log(`  │ URL:       ${env.latest_build.build_url}`);
-                    console.log(`  └─`);
                 }
                 else {
-                    console.log(`  └─ No builds available`);
+                    console.log(`  Playwright Projects: None`);
                 }
-            });
+                printLatestBuild(env.latest_build);
+            }
         }
         else {
             console.log(`No environments found for repository: ${repoName}`);

package/dist/bin/index.js

@@ -269,7 +269,7 @@ async function main() {
         .command("environments")
         .description("List environments and their latest builds")
         .action(async () => {
-        await (0, environments_1.listEnvironments)();
+        await (0, environments_1.listEnvironments)(process.cwd());
         process.exit(0);
     });
     program

package/dist/bin/setup.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"setup.d.ts","sourceRoot":"","sources":["../../src/bin/setup.ts"],"names":[],"mappings":"AAmIA,wBAAsB,QAAQ,CAAC,EAAE,QAAQ,EAAE,EAAE;IAAE,QAAQ,EAAE,MAAM,CAAA;CAAE,iBA4BhE"}
+{"version":3,"file":"setup.d.ts","sourceRoot":"","sources":["../../src/bin/setup.ts"],"names":[],"mappings":"AA2HA,wBAAsB,QAAQ,CAAC,EAAE,QAAQ,EAAE,EAAE;IAAE,QAAQ,EAAE,MAAM,CAAA;CAAE,iBA4BhE"}

package/dist/bin/setup.js

@@ -48,17 +48,17 @@ async function installDependencies(targetDir) {
 async function installPlaywrightBrowser(targetDir) {
     console.log(`🎭 Installing Playwright browser...`);
     try {
-        const { stderr: playwrightStderr } = await execAsync(`npx playwright install chromium --with-deps`, {
-            cwd: targetDir,
-        });
-        if (playwrightStderr && !playwrightStderr.includes("Installing")) {
-            console.error(`Playwright install error: ${playwrightStderr}`);
-            process.exit(1);
+        const { stderr } = await execAsync(`npx playwright install chromium --with-deps`, { cwd: targetDir });
+        if (stderr && stderr.trim()) {
+            console.warn(`⚠️ Playwright installation warnings: ${stderr.trim()}`);
         }
         console.log(`✅ Playwright browser installed successfully`);
     }
-    catch (playwrightError) {
-        console.error(`❌ Failed to install Playwright browser: ${playwrightError.message}`);
+    catch (error) {
+        // execAsync already throws on non-zero exit codes
+        const errorMessage = error.message || "Unknown error";
+        const exitCode = error.code || "unknown";
+        console.error(`❌ Failed to install Playwright browser (exit code: ${exitCode}): ${errorMessage}`);
         process.exit(1);
     }
 }
@@ -73,8 +73,8 @@ async function cleanUpRepoCredSetup(repoPath) {
 }
 async function setupRepoCredentials(repoPath) {
     console.log(`🔑 Setting up repository credentials...`);
-    const helperCommand = `!npx @empiricalrun/test-gen github-token --raw`;
-    await execAsync(`git config credential."https://github.com".helper '${helperCommand}'`, { cwd: repoPath });
+    const helperCommand = `"!npx @empiricalrun/test-gen github-token --raw"`;
+    await execAsync(`git config credential."https://github.com".helper ${helperCommand}`, { cwd: repoPath });
     console.log(`✅ Repository credentials configured`);
 }
 async function validateAndSetupRepo(repoPath) {

package/dist/dashboard/index.d.ts

@@ -0,0 +1,21 @@
+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 {
+}
+export interface RequestOptions {
+    params?: Record<string, string>;
+    body?: Record<string, unknown>;
+    method: "GET" | "POST";
+    headers?: Record<string, string>;
+}
+export declare function makeAuthenticatedRequest<T>(url: string, options: RequestOptions, secretKey?: string, dashboardDomain?: string): Promise<T>;
+export declare function callGitHubProxy({ method, url, body, }: {
+    method: "GET" | "POST";
+    url: string;
+    body?: any;
+}): Promise<unknown>;
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/dashboard/index.ts"],"names":[],"mappings":"AAIA,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;AAEpD,MAAM,WAAW,cAAc;IAC7B,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAChC,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAC/B,MAAM,EAAE,KAAK,GAAG,MAAM,CAAC;IACvB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;CAClC;AAED,wBAAsB,wBAAwB,CAAC,CAAC,EAC9C,GAAG,EAAE,MAAM,EACX,OAAO,EAAE,cAAc,EACvB,SAAS,CAAC,EAAE,MAAM,EAClB,eAAe,CAAC,EAAE,MAAM,GACvB,OAAO,CAAC,CAAC,CAAC,CAqDZ;AAED,wBAAsB,eAAe,CAAC,EACpC,MAAM,EACN,GAAG,EACH,IAAI,GACL,EAAE;IACD,MAAM,EAAE,KAAK,GAAG,MAAM,CAAC;IACvB,GAAG,EAAE,MAAM,CAAC;IACZ,IAAI,CAAC,EAAE,GAAG,CAAC;CACZ,oBAYA"}

package/dist/dashboard/index.js

@@ -0,0 +1,83 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.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,
+        },
+    });
+}

package/dist/dashboard/totp.d.ts

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

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

@@ -0,0 +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"}

package/dist/dashboard/totp.js

@@ -0,0 +1,18 @@
+"use strict";
+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
+    const { generateTOTP } = await import("@epic-web/totp");
+    const totp = await generateTOTP({
+        secret,
+        digits: 6,
+        period: 30,
+        algorithm: "SHA-256",
+    });
+    return totp.otp;
+}

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

@@ -0,0 +1,3 @@
+import { FileInfo } from "@empiricalrun/shared-types";
+export declare function getFileInfoFromFS(directory: string): Promise<FileInfo>;
+//# sourceMappingURL=file-system.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"file-system.d.ts","sourceRoot":"","sources":["../../src/file-info/file-system.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,4BAA4B,CAAC;AAItD,wBAAsB,iBAAiB,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,QAAQ,CAAC,CAgC5E"}

package/dist/{utils/file-tree.js → file-info/file-system.js}

@@ -10,14 +10,10 @@ async function getFileInfoFromFS(directory) {
     // TODO: This traverses all node_modules - can be excluded with DEFAULT_EXCLUDE
     let currentDirName = path_1.default.basename(directory);
     let currentDir = {
-        isDirectory: true,
-        isFile: false,
+        type: "directory",
         path: currentDirName,
         name: currentDirName,
         children: [],
-        getContent: async () => {
-            return "";
-        },
     };
     const files = await promises_1.default.readdir(directory);
     for (const file of files) {
@@ -29,11 +25,9 @@ async function getFileInfoFromFS(directory) {
         }
         else {
             currentDir.children.push({
-                isDirectory: false,
-                isFile: true,
+                type: "file",
                 path: filePath,
                 name: file,
-                children: [],
                 getContent: async () => {
                     return await promises_1.default.readFile(filePath, "utf8");
                 },

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

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

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

@@ -0,0 +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"}

package/dist/file-info/github.js

@@ -0,0 +1,108 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": 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) {
+    // 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, {
+        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) {
+    const url = `/api/github/tree?repo=${repo}&path=${path}`;
+    const response = await (0, dashboard_1.makeAuthenticatedRequest)(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);
+    const root = [];
+    const nodeMap = {};
+    if (!files) {
+        return {
+            type: "directory",
+            path: repoName,
+            name: repoName,
+            children: [],
+        };
+    }
+    files.tree.forEach((file) => {
+        if (!file.path) {
+            return;
+        }
+        const pathParts = file.path.split("/");
+        const fileName = path_1.default.basename(file.path);
+        const isFile = file.type === "blob";
+        // Create the current file/directory node
+        const currentPath = file.path;
+        const currentNode = isFile
+            ? {
+                type: "file",
+                path: currentPath,
+                name: fileName,
+                getContent: async () => getFileContent(repoName, currentPath, secretKey, dashboardDomain),
+            }
+            : {
+                type: "directory",
+                path: currentPath,
+                name: fileName,
+                children: [],
+            };
+        nodeMap[currentPath] = currentNode;
+        if (pathParts.length === 1) {
+            // This is a root-level file or directory
+            root.push(currentNode);
+        }
+        else {
+            // This is a nested file or directory
+            const parentPath = path_1.default.dirname(currentPath);
+            // Ensure parent exists
+            if (!nodeMap[parentPath]) {
+                // Create parent directory if it doesn't exist yet
+                const parentNode = {
+                    type: "directory",
+                    path: parentPath,
+                    name: parentPath.split("/").pop() || "",
+                    children: [],
+                };
+                nodeMap[parentPath] = parentNode;
+                // Add parent to its parent or root
+                const parentPathParts = parentPath.split("/");
+                if (parentPathParts.length === 1) {
+                    root.push(parentNode);
+                }
+                else {
+                    const grandparentPath = path_1.default.dirname(parentPath);
+                    if (nodeMap[grandparentPath] &&
+                        nodeMap[grandparentPath].type === "directory") {
+                        nodeMap[grandparentPath].children.push(parentNode);
+                    }
+                }
+            }
+            // Add current node to its parent
+            if (nodeMap[parentPath].type === "directory") {
+                nodeMap[parentPath].children.push(currentNode);
+            }
+        }
+    });
+    return {
+        type: "directory",
+        path: repoName,
+        name: repoName,
+        children: root,
+    };
+}

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

@@ -3,8 +3,8 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.createPullRequestTool = void 0;
 const child_process_1 = require("child_process");
 const zod_1 = require("zod");
-const git_1 = require("../utils/git");
-const utils_1 = require("./utils");
+const utils_1 = require("./executor/utils");
+const git_1 = require("./executor/utils/git");
 const createPullRequestSchema = zod_1.z.object({
     pullRequestTitle: zod_1.z
         .string()

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

@@ -0,0 +1,22 @@
+import { PendingToolCall } from "@empiricalrun/llm/chat";
+import { SupportedChatModels, ToolDefinition } from "@empiricalrun/shared-types";
+export declare class ToolCaller {
+    tools: ToolDefinition[];
+    selectedModel: SupportedChatModels;
+    branchName: string;
+    chatSessionId: number | null;
+    workerEnv?: Record<string, string>;
+    constructor({ chatSessionId, selectedModel, branchName, workerEnv, }: {
+        chatSessionId: number | null;
+        selectedModel: SupportedChatModels;
+        branchName: string;
+        workerEnv?: Record<string, string>;
+    });
+    schemas(): {
+        name: string;
+        description: string;
+        parameters: import("zod").ZodType;
+    }[];
+    sendToQueue(requestId: string, toolCalls: PendingToolCall[]): Promise<void>;
+}
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/tools/definitions/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAuB,eAAe,EAAE,MAAM,wBAAwB,CAAC;AAC9E,OAAO,EACL,mBAAmB,EACnB,cAAc,EACf,MAAM,4BAA4B,CAAC;AAcpC,qBAAa,UAAU;IACrB,KAAK,EAAE,cAAc,EAAE,CAAM;IAC7B,aAAa,EAAE,mBAAmB,CAAC;IACnC,UAAU,EAAE,MAAM,CAAC;IACnB,aAAa,EAAE,MAAM,GAAG,IAAI,CAAC;IAC7B,SAAS,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;gBACvB,EACV,aAAa,EACb,aAAa,EACb,UAAU,EACV,SAAS,GACV,EAAE;QACD,aAAa,EAAE,MAAM,GAAG,IAAI,CAAC;QAC7B,aAAa,EAAE,mBAAmB,CAAC;QACnC,UAAU,EAAE,MAAM,CAAC;QACnB,SAAS,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;KACpC;IAsBD,OAAO;;;;;IAID,WAAW,CACf,SAAS,EAAE,MAAM,EACjB,SAAS,EAAE,eAAe,EAAE,GAC3B,OAAO,CAAC,IAAI,CAAC;CAcjB"}

package/dist/tools/definitions/index.js

@@ -0,0 +1,61 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ToolCaller = void 0;
+const chat_1 = require("@empiricalrun/llm/chat");
+const commit_and_create_pr_1 = require("../commit-and-create-pr");
+const diagnosis_fetcher_1 = require("../diagnosis-fetcher");
+const download_build_1 = require("../download-build");
+const grep_1 = require("../grep");
+const list_environments_1 = require("../list-environments");
+const test_run_fetcher_1 = require("../test-run-fetcher");
+const upgrade_packages_1 = require("../upgrade-packages");
+const run_test_1 = require("./run-test");
+const str_replace_editor_1 = require("./str_replace_editor");
+const test_gen_browser_1 = require("./test-gen-browser");
+const queue_1 = require("./utils/queue");
+class ToolCaller {
+    tools = [];
+    selectedModel;
+    branchName;
+    chatSessionId;
+    workerEnv;
+    constructor({ chatSessionId, selectedModel, branchName, workerEnv, }) {
+        this.chatSessionId = chatSessionId;
+        this.selectedModel = selectedModel;
+        this.branchName = branchName;
+        this.workerEnv = workerEnv;
+        // TODO: Keep this list in sync with the ToolExecutor
+        this.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,
+            list_environments_1.listEnvironmentsTool,
+            download_build_1.downloadBuildTool,
+            upgrade_packages_1.upgradePackagesTool,
+        ];
+        if ((0, chat_1.getProviderForModel)(this.selectedModel) !== "claude") {
+            this.tools.push(...Object.values(str_replace_editor_1.textEditorTools));
+        }
+    }
+    schemas() {
+        return this.tools.map((tool) => tool.schema);
+    }
+    async sendToQueue(requestId, toolCalls) {
+        const queueUrl = (0, queue_1.getQueueUrl)(toolCalls, this.tools, this.workerEnv);
+        if (!queueUrl) {
+            throw new Error(`queueUrl is required for remote execution.`);
+        }
+        await (0, queue_1.sendToolRequestToRemoteQueue)(queueUrl, {
+            toolCalls,
+            requestId,
+            chatSessionId: this.chatSessionId,
+            selectedModel: this.selectedModel,
+            branchName: this.branchName,
+            workerEnv: this.workerEnv,
+        });
+    }
+}
+exports.ToolCaller = ToolCaller;

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

@@ -0,0 +1,23 @@
+import { ToolDefinition } from "@empiricalrun/shared-types";
+import { z } from "zod";
+export declare const RunTestSchema: z.ZodObject<{
+    testName: z.ZodString;
+    suites: z.ZodArray<z.ZodString, "many">;
+    filePath: z.ZodString;
+    project: z.ZodString;
+    buildUrl: z.ZodOptional<z.ZodString>;
+}, "strip", z.ZodTypeAny, {
+    suites: string[];
+    testName: string;
+    filePath: string;
+    project: string;
+    buildUrl?: string | undefined;
+}, {
+    suites: string[];
+    testName: string;
+    filePath: string;
+    project: string;
+    buildUrl?: string | undefined;
+}>;
+export declare const runTestTool: ToolDefinition;
+//# sourceMappingURL=run-test.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"run-test.d.ts","sourceRoot":"","sources":["../../../src/tools/definitions/run-test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;AAC5D,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,eAAO,MAAM,aAAa;;;;;;;;;;;;;;;;;;EAmBxB,CAAC;AAEH,eAAO,MAAM,WAAW,EAAE,cAOzB,CAAC"}