@empiricalrun/test-gen 0.63.0 → 0.64.1

package/dist/artifacts/utils.d.ts

@@ -0,0 +1,21 @@
+import { ArtifactInput, ArtifactInputPath } from "@empiricalrun/shared-types";
+import { JSONReport as PlaywrightJSONReport } from "@playwright/test/reporter";
+/**
+ * Extracts attachment information from a Playwright JSON report.
+ * @param report The Playwright JSON report to extract attachments from
+ * @returns An array of objects containing path and contentType for each attachment
+ */
+export declare function extractAttachmentsFromPlaywrightJSONReport(report: PlaywrightJSONReport): ArtifactInputPath[];
+/**
+ * Scans the given repository directory for Playwright artifacts (attachments) by:
+ * - Constructing the path to the Playwright report directory using the provided repoDir.
+ * - Building the absolute path to the summary.json file within the report directory.
+ * - Checking if summary.json exists, and if so, reading and parsing its contents.
+ * - Extracting attachment information (such as file paths and content types) from the parsed Playwright JSON report.
+ * - Returning an array of ArtifactInput objects, each containing details about a found attachment.
+ *
+ * @param repoDir Absolute path to the root of the repository. Used to construct paths to Playwright artifacts.
+ * @returns Array of ArtifactInput objects representing found Playwright attachments, including their resolved paths and content types.
+ */
+export declare function findPlaywrightArtifacts(repoDir: string): ArtifactInput[];
+//# sourceMappingURL=utils.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../src/artifacts/utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,iBAAiB,EAAE,MAAM,4BAA4B,CAAC;AAC9E,OAAO,EACL,UAAU,IAAI,oBAAoB,EAGnC,MAAM,2BAA2B,CAAC;AAGnC;;;;GAIG;AACH,wBAAgB,0CAA0C,CACxD,MAAM,EAAE,oBAAoB,GAC3B,iBAAiB,EAAE,CAqErB;AAED;;;;;;;;;;GAUG;AACH,wBAAgB,uBAAuB,CAAC,OAAO,EAAE,MAAM,GAAG,aAAa,EAAE,CAmBxE"}

package/dist/artifacts/utils.js

@@ -0,0 +1,102 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.extractAttachmentsFromPlaywrightJSONReport = extractAttachmentsFromPlaywrightJSONReport;
+exports.findPlaywrightArtifacts = findPlaywrightArtifacts;
+const fs_1 = __importDefault(require("fs"));
+/**
+ * Extracts attachment information from a Playwright JSON report.
+ * @param report The Playwright JSON report to extract attachments from
+ * @returns An array of objects containing path and contentType for each attachment
+ */
+function extractAttachmentsFromPlaywrightJSONReport(report) {
+    const attachments = [];
+    if (!report || !report.suites || report.suites.length === 0) {
+        return attachments;
+    }
+    /**
+     * Helper function to process a single spec and extract its attachments.
+     * @param {JSONReportSpec} spec - The spec object to process.
+     */
+    function processSpec(spec) {
+        const testTitle = spec.title;
+        if (!spec.tests) {
+            return;
+        }
+        for (const test of spec.tests) {
+            if (!test.results) {
+                continue;
+            }
+            for (const result of test.results) {
+                if (!result.attachments) {
+                    continue;
+                }
+                for (const attachment of result.attachments) {
+                    // Only collect attachments that have a 'path' (i.e., not embedded content)
+                    if (attachment.path && attachment.contentType) {
+                        attachments.push({
+                            name: `${testTitle} ${attachment.name}`,
+                            path: attachment.path,
+                            contentType: attachment.contentType,
+                        });
+                    }
+                }
+            }
+        }
+    }
+    /**
+     * Recursive helper function to traverse the suite structure.
+     * @param {JSONReportSuite[]} suites - An array of suites to process.
+     */
+    function traverseSuites(suites) {
+        if (!suites) {
+            return;
+        }
+        for (const suite of suites) {
+            // Process specs directly within the current suite
+            if (suite.specs) {
+                for (const spec of suite.specs) {
+                    processSpec(spec);
+                }
+            }
+            // Recursively call for any nested suites
+            if (suite.suites) {
+                traverseSuites(suite.suites);
+            }
+        }
+    }
+    // Start the recursive traversal from the top-level suites
+    traverseSuites(report.suites);
+    return attachments;
+}
+/**
+ * Scans the given repository directory for Playwright artifacts (attachments) by:
+ * - Constructing the path to the Playwright report directory using the provided repoDir.
+ * - Building the absolute path to the summary.json file within the report directory.
+ * - Checking if summary.json exists, and if so, reading and parsing its contents.
+ * - Extracting attachment information (such as file paths and content types) from the parsed Playwright JSON report.
+ * - Returning an array of ArtifactInput objects, each containing details about a found attachment.
+ *
+ * @param repoDir Absolute path to the root of the repository. Used to construct paths to Playwright artifacts.
+ * @returns Array of ArtifactInput objects representing found Playwright attachments, including their resolved paths and content types.
+ */
+function findPlaywrightArtifacts(repoDir) {
+    const artifacts = [];
+    const reportPath = require("path").join(repoDir, "playwright-report");
+    const summaryJsonPath = require("path").join(reportPath, "summary.json");
+    if (fs_1.default.existsSync(summaryJsonPath)) {
+        try {
+            const summaryJson = JSON.parse(fs_1.default.readFileSync(summaryJsonPath, "utf-8"));
+            const found = extractAttachmentsFromPlaywrightJSONReport(summaryJson);
+            if (Array.isArray(found) && found.length > 0) {
+                artifacts.push(...found);
+            }
+        }
+        catch (e) {
+            console.error(`[findPlaywrightArtifacts] Error parsing summary.json: ${e}`);
+        }
+    }
+    return artifacts;
+}

package/dist/bin/index.js

@@ -30,7 +30,7 @@ const flushEvents = async () => {
 };
 function setupProcessListeners(cleanup) {
     const events = ["beforeExit", "exit", "SIGINT", "SIGTERM"];
-    events.forEach((event) => process.on(event, cleanup));
+    events.forEach((event) => process.once(event, cleanup));
     return () => {
         events.forEach((event) => process.removeListener(event, cleanup));
     };
@@ -191,6 +191,7 @@ async function runAgentsWorkflow(testGenConfig, testGenToken) {
             testGenToken,
             repoDir: process.cwd(),
             editFileWithGeneratedCode: true,
+            envOverrides: {},
         });
         if (isError) {
             throw new Error(error);

package/dist/bin/utils/platform/web/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../../src/bin/utils/platform/web/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,4BAA4B,CAAC;AAM3D,OAAO,EAGL,IAAI,EAEJ,UAAU,EAEX,MAAM,UAAU,CAAC;AAMlB,eAAO,MAAM,gCAAgC,GAC3C,YAAY,UAAU,KACrB,MAgBF,CAAC;AAEF;;;;;;GAMG;AACH,wBAAgB,sBAAsB,CAAC,EACrC,YAAY,EACZ,MAAM,EACN,OAAO,GACR,EAAE;IACD,YAAY,EAAE,MAAM,CAAC;IACrB,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;CACjB,GAAG;IACF,SAAS,EAAE,MAAM,GAAG,SAAS,CAAC;IAC9B,QAAQ,EAAE,IAAI,GAAG,SAAS,CAAC;IAC3B,SAAS,EAAE,MAAM,CAAC;CACnB,CA2CA;AAED,wBAAgB,YAAY,CAAC,EAC3B,QAAQ,EACR,UAAU,EACV,QAAQ,GACT,EAAE;IACD,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,MAAM,EAAE,CAAC;IACrB,QAAQ,EAAE,MAAM,CAAC;CAClB,WAUA;AAwBD,wBAAsB,0CAA0C,CAC9D,QAAQ,EAAE,MAAM,oBA+BjB;AAED;;;;;;;;;;;;;;;;;;;;;;GAsBG;AACH,wBAAgB,4BAA4B,CAC1C,IAAI,EAAE,IAAI,GAAG,SAAS,GACrB,IAAI,GAAG,SAAS,CA4BlB;AAED,wBAAgB,iBAAiB,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,MAAM,CAG5E;AAED,wBAAsB,qBAAqB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC;IACpE,OAAO,EAAE,OAAO,CAAC;IACjB,MAAM,EAAE,MAAM,EAAE,CAAC;CAClB,CAAC,CAwBD;AAED,wBAAgB,kBAAkB,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,EAAE,CA8C7D;AAED,wBAAsB,sBAAsB,CAC1C,OAAO,EAAE,MAAM,EACf,QAAQ,EAAE,MAAM,mCAWjB;AAED,wBAAsB,UAAU,CAAC,QAAQ,EAAE,MAAM,iBAShD;AAED,wBAAsB,UAAU,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,WAAW,iBAgBrE;AAED,wBAAgB,YAAY,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE,GAAG,EAAE,MAAM,UAE5E;AAED,wBAAgB,oBAAoB,CAAC,QAAQ,EAAE,MAAM,UAcpD;AAED,wBAAsB,iCAAiC,CAAC,QAAQ,EAAE,MAAM,+BAoBvE;AA+CD,wBAAgB,4BAA4B,CAC1C,QAAQ,EAAE,MAAM,EAChB,QAAQ,EAAE,MAAM,EAChB,aAAa,EAAE,MAAM,UAoCtB;AAED,eAAO,MAAM,6BAA6B,GAAI,2CAI3C;IACD,eAAe,EAAE,MAAM,CAAC;IACxB,WAAW,EAAE,MAAM,CAAC;IACpB,MAAM,EAAE,MAAM,EAAE,CAAC;CAClB,WA0DA,CAAC;AAEF,eAAO,MAAM,iCAAiC,GAC5C,SAAS,MAAM,EACf,WAAW,MAAM,EAAE,EACnB,cAAc,MAAM,sBAyBrB,CAAC;AAEF,wBAAsB,qBAAqB,CAAC,EAC1C,YAAY,EACZ,QAAQ,EACR,MAAM,GACP,EAAE;IACD,YAAY,EAAE,MAAM,CAAC;IACrB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,EAAE,CAAC;CAClB,iBAgDA;AAED,wBAAsB,uBAAuB,CAC3C,QAAQ,EAAE,MAAM,EAChB,cAAc,EAAE,MAAM,EAAE,iBAsBzB;AAED,wBAAgB,aAAa,CAAC,EAC5B,QAAQ,EACR,QAAQ,GACT,EAAE;IACD,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,QAAQ,CAAC;CACpB,WAYA;AAED,wBAAgB,mBAAmB,CAAC,EAClC,QAAQ,EACR,MAAM,GACP,EAAE;IACD,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,EAAE,CAAC;CAClB,UAOA;AAED,wBAAgB,+BAA+B,CAAC,UAAU,EAAE,MAAM,GAAG,MAAM,EAAE,CA4B5E;AAED,wBAAgB,aAAa,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAQnD"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../../src/bin/utils/platform/web/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,4BAA4B,CAAC;AAM3D,OAAO,EAGL,IAAI,EAEJ,UAAU,EAEX,MAAM,UAAU,CAAC;AAMlB,eAAO,MAAM,gCAAgC,GAC3C,YAAY,UAAU,KACrB,MAgBF,CAAC;AAEF;;;;;;GAMG;AACH,wBAAgB,sBAAsB,CAAC,EACrC,YAAY,EACZ,MAAM,EACN,OAAO,GACR,EAAE;IACD,YAAY,EAAE,MAAM,CAAC;IACrB,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;CACjB,GAAG;IACF,SAAS,EAAE,MAAM,GAAG,SAAS,CAAC;IAC9B,QAAQ,EAAE,IAAI,GAAG,SAAS,CAAC;IAC3B,SAAS,EAAE,MAAM,CAAC;CACnB,CA2CA;AAED,wBAAgB,YAAY,CAAC,EAC3B,QAAQ,EACR,UAAU,EACV,QAAQ,GACT,EAAE;IACD,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,MAAM,EAAE,CAAC;IACrB,QAAQ,EAAE,MAAM,CAAC;CAClB,WAUA;AAwBD,wBAAsB,0CAA0C,CAC9D,QAAQ,EAAE,MAAM,oBA+BjB;AAED;;;;;;;;;;;;;;;;;;;;;;GAsBG;AACH,wBAAgB,4BAA4B,CAC1C,IAAI,EAAE,IAAI,GAAG,SAAS,GACrB,IAAI,GAAG,SAAS,CA4BlB;AAED,wBAAgB,iBAAiB,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,MAAM,CAG5E;AAED,wBAAsB,qBAAqB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC;IACpE,OAAO,EAAE,OAAO,CAAC;IACjB,MAAM,EAAE,MAAM,EAAE,CAAC;CAClB,CAAC,CAwBD;AAED,wBAAgB,kBAAkB,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,EAAE,CA8C7D;AAED,wBAAsB,sBAAsB,CAC1C,OAAO,EAAE,MAAM,EACf,QAAQ,EAAE,MAAM,mCAWjB;AAED,wBAAsB,UAAU,CAAC,QAAQ,EAAE,MAAM,iBAUhD;AAED,wBAAsB,UAAU,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,WAAW,iBAiBrE;AAED,wBAAgB,YAAY,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE,GAAG,EAAE,MAAM,UAE5E;AAED,wBAAgB,oBAAoB,CAAC,QAAQ,EAAE,MAAM,UAcpD;AAED,wBAAsB,iCAAiC,CAAC,QAAQ,EAAE,MAAM,+BAoBvE;AA+CD,wBAAgB,4BAA4B,CAC1C,QAAQ,EAAE,MAAM,EAChB,QAAQ,EAAE,MAAM,EAChB,aAAa,EAAE,MAAM,UAoCtB;AAED,eAAO,MAAM,6BAA6B,GAAI,2CAI3C;IACD,eAAe,EAAE,MAAM,CAAC;IACxB,WAAW,EAAE,MAAM,CAAC;IACpB,MAAM,EAAE,MAAM,EAAE,CAAC;CAClB,WA0DA,CAAC;AAEF,eAAO,MAAM,iCAAiC,GAC5C,SAAS,MAAM,EACf,WAAW,MAAM,EAAE,EACnB,cAAc,MAAM,sBAyBrB,CAAC;AAEF,wBAAsB,qBAAqB,CAAC,EAC1C,YAAY,EACZ,QAAQ,EACR,MAAM,GACP,EAAE;IACD,YAAY,EAAE,MAAM,CAAC;IACrB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,EAAE,CAAC;CAClB,iBAgDA;AAED,wBAAsB,uBAAuB,CAC3C,QAAQ,EAAE,MAAM,EAChB,cAAc,EAAE,MAAM,EAAE,iBAsBzB;AAED,wBAAgB,aAAa,CAAC,EAC5B,QAAQ,EACR,QAAQ,GACT,EAAE;IACD,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,QAAQ,CAAC;CACpB,WAYA;AAED,wBAAgB,mBAAmB,CAAC,EAClC,QAAQ,EACR,MAAM,GACP,EAAE;IACD,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,EAAE,CAAC;CAClB,UAOA;AAED,wBAAgB,+BAA+B,CAAC,UAAU,EAAE,MAAM,GAAG,MAAM,EAAE,CA4B5E;AAED,wBAAgB,aAAa,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAQnD"}

package/dist/bin/utils/platform/web/index.js

@@ -275,6 +275,7 @@ async function stripAndPrependImports(content, testName) {
     return [prependContent, strippedContent];
 }
 async function lintErrors(filePath) {
+    // TODO: use in new tools - return result/error?
     const eslint = new eslint_1.ESLint({
         fix: true,
         useEslintrc: true,
@@ -285,6 +286,7 @@ async function lintErrors(filePath) {
     }
 }
 async function formatCode(filePath, trace) {
+    // TODO: should we delete this?
     const fileContent = fs_1.default.readFileSync(filePath, "utf8");
     if (!fileContent) {
         trace?.span({

package/dist/file/server.js

@@ -31,7 +31,7 @@ class FileServiceServer {
         return this.artifactsInputs;
     }
     setFilePath(filePath) {
-        this.filePath = filePath;
+        this.filePath = path_1.default.resolve(this.repoDir, filePath);
     }
     async startFileService() {
         const app = (0, express_1.default)();

package/dist/test-build/index.js

@@ -26,7 +26,7 @@ async function downloadBuild({ buildUrl, repoPath, apiKey, }) {
         logger.log(`Downloading build from ${buildUrl}`);
         await (0, exec_1.cmd)(`npm`, ["run", "download", buildUrl], {
             cwd: repoPath,
-            env: { ...Object(process.env), EMPIRICALRUN_API_KEY: apiKey },
+            envOverrides: { ...Object(process.env), EMPIRICALRUN_API_KEY: apiKey },
         });
     }
 }

package/dist/tool-call-service/index.d.ts

@@ -1,5 +1,6 @@
 import { TraceClient } from "@empiricalrun/llm";
-import { PendingToolCall, SupportedChatModels, Tool, ToolExecute, ToolResult } from "@empiricalrun/llm/chat";
+import { PendingToolCall, SupportedChatModels, ToolResult } from "@empiricalrun/llm/chat";
+import { Tool, ToolExecute } from "@empiricalrun/shared-types";
 export type { SupportedChatModels };
 type ToolExecutors = {
     [key: string]: ToolExecute;
@@ -13,13 +14,15 @@ export declare class ToolCallService {
     repoPath: string;
     apiKey: string | undefined;
     trace?: TraceClient;
-    constructor({ chatSessionId, selectedModel, branchName, repoPath, apiKey, trace, }: {
+    featureFlags: string[];
+    constructor({ chatSessionId, selectedModel, branchName, repoPath, apiKey, trace, featureFlags, }: {
         chatSessionId: number | null;
         selectedModel: SupportedChatModels;
         branchName: string;
         repoPath: string;
         apiKey: string | undefined;
         trace?: TraceClient;
+        featureFlags: string[];
     });
     sendToQueue(toolCalls: PendingToolCall[]): Promise<void>;
     execute(toolCalls: PendingToolCall[]): Promise<ToolResult[]>;

package/dist/tool-call-service/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/tool-call-service/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAEL,eAAe,EACf,mBAAmB,EACnB,IAAI,EACJ,WAAW,EACX,UAAU,EACX,MAAM,wBAAwB,CAAC;AAkBhC,YAAY,EAAE,mBAAmB,EAAE,CAAC;AAEpC,KAAK,aAAa,GAAG;IACnB,CAAC,GAAG,EAAE,MAAM,GAAG,WAAW,CAAC;CAC5B,CAAC;AAEF,qBAAa,eAAe;IAC1B,KAAK,EAAE,IAAI,EAAE,CAAM;IACnB,aAAa,EAAE,aAAa,CAAM;IAClC,aAAa,EAAE,MAAM,GAAG,IAAI,CAAC;IAC7B,aAAa,EAAE,mBAAmB,CAAC;IACnC,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC;IAC3B,KAAK,CAAC,EAAE,WAAW,CAAC;gBACR,EACV,aAAa,EACb,aAAa,EACb,UAAU,EACV,QAAQ,EACR,MAAM,EACN,KAAK,GACN,EAAE;QACD,aAAa,EAAE,MAAM,GAAG,IAAI,CAAC;QAC7B,aAAa,EAAE,mBAAmB,CAAC;QACnC,UAAU,EAAE,MAAM,CAAC;QACnB,QAAQ,EAAE,MAAM,CAAC;QACjB,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC;QAC3B,KAAK,CAAC,EAAE,WAAW,CAAC;KACrB;IAiCK,WAAW,CAAC,SAAS,EAAE,eAAe,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAcxD,OAAO,CAAC,SAAS,EAAE,eAAe,EAAE,GAAG,OAAO,CAAC,UAAU,EAAE,CAAC;CAsEnE"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/tool-call-service/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAEL,eAAe,EACf,mBAAmB,EACnB,UAAU,EACX,MAAM,wBAAwB,CAAC;AAChC,OAAO,EAAiB,IAAI,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AAiB9E,YAAY,EAAE,mBAAmB,EAAE,CAAC;AAEpC,KAAK,aAAa,GAAG;IACnB,CAAC,GAAG,EAAE,MAAM,GAAG,WAAW,CAAC;CAC5B,CAAC;AAEF,qBAAa,eAAe;IAC1B,KAAK,EAAE,IAAI,EAAE,CAAM;IACnB,aAAa,EAAE,aAAa,CAAM;IAClC,aAAa,EAAE,MAAM,GAAG,IAAI,CAAC;IAC7B,aAAa,EAAE,mBAAmB,CAAC;IACnC,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC;IAC3B,KAAK,CAAC,EAAE,WAAW,CAAC;IACpB,YAAY,EAAE,MAAM,EAAE,CAAC;gBACX,EACV,aAAa,EACb,aAAa,EACb,UAAU,EACV,QAAQ,EACR,MAAM,EACN,KAAK,EACL,YAAY,GACb,EAAE;QACD,aAAa,EAAE,MAAM,GAAG,IAAI,CAAC;QAC7B,aAAa,EAAE,mBAAmB,CAAC;QACnC,UAAU,EAAE,MAAM,CAAC;QACnB,QAAQ,EAAE,MAAM,CAAC;QACjB,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC;QAC3B,KAAK,CAAC,EAAE,WAAW,CAAC;QACpB,YAAY,EAAE,MAAM,EAAE,CAAC;KACxB;IAkCK,WAAW,CAAC,SAAS,EAAE,eAAe,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAexD,OAAO,CAAC,SAAS,EAAE,eAAe,EAAE,GAAG,OAAO,CAAC,UAAU,EAAE,CAAC;CAsEnE"}

package/dist/tool-call-service/index.js

@@ -23,13 +23,15 @@ class ToolCallService {
     repoPath;
     apiKey;
     trace;
-    constructor({ chatSessionId, selectedModel, branchName, repoPath, apiKey, trace, }) {
+    featureFlags;
+    constructor({ chatSessionId, selectedModel, branchName, repoPath, apiKey, trace, featureFlags, }) {
         this.chatSessionId = chatSessionId;
         this.selectedModel = selectedModel;
         this.branchName = branchName;
         this.trace = trace;
         this.repoPath = repoPath;
         this.apiKey = apiKey;
+        this.featureFlags = featureFlags;
         this.tools = [
             grep_1.grepTool,
             test_run_1.runTestTool,
@@ -66,6 +68,7 @@ class ToolCallService {
             chatSessionId: this.chatSessionId,
             selectedModel: this.selectedModel,
             branchName: this.branchName,
+            featureFlags: this.featureFlags || [],
         });
     }
     async execute(toolCalls) {
@@ -101,7 +104,13 @@ class ToolCallService {
                 throw new Error("API key is required for tool execution");
             }
             try {
-                const result = await toolExecutor(toolCall.input, this.repoPath, this.apiKey, this.trace, collectArtifactsFn);
+                const result = await toolExecutor({
+                    input: toolCall.input,
+                    repoPath: this.repoPath,
+                    apiKey: this.apiKey,
+                    trace: this.trace,
+                    collectArtifacts: collectArtifactsFn,
+                });
                 const artifacts = await uploadArtifactsQueue.waitForCompletion();
                 toolResults.push({ ...result, artifacts });
                 span?.end({ output: result });

package/dist/tool-call-service/utils.d.ts

@@ -6,5 +6,6 @@ export declare function sendToolRequestToRemoteQueue(payload: {
     chatSessionId: number;
     selectedModel: SupportedChatModels;
     branchName: string;
+    featureFlags: string[];
 }): Promise<void>;
 //# sourceMappingURL=utils.d.ts.map

package/dist/tool-call-service/utils.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../src/tool-call-service/utils.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,mBAAmB,EAAE,MAAM,wBAAwB,CAAC;AAC7D,OAAO,EAAE,eAAe,EAAE,MAAM,4BAA4B,CAAC;AAE7D,wBAAsB,4BAA4B,CAAC,OAAO,EAAE;IAC1D,SAAS,EAAE,eAAe,EAAE,CAAC;IAC7B,SAAS,EAAE,MAAM,CAAC;IAClB,aAAa,EAAE,MAAM,CAAC;IACtB,aAAa,EAAE,mBAAmB,CAAC;IACnC,UAAU,EAAE,MAAM,CAAC;CACpB,iBAoBA"}
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../src/tool-call-service/utils.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,mBAAmB,EAAE,MAAM,wBAAwB,CAAC;AAC7D,OAAO,EAAE,eAAe,EAAE,MAAM,4BAA4B,CAAC;AAE7D,wBAAsB,4BAA4B,CAAC,OAAO,EAAE;IAC1D,SAAS,EAAE,eAAe,EAAE,CAAC;IAC7B,SAAS,EAAE,MAAM,CAAC;IAClB,aAAa,EAAE,MAAM,CAAC;IACtB,aAAa,EAAE,mBAAmB,CAAC;IACnC,UAAU,EAAE,MAAM,CAAC;IACnB,YAAY,EAAE,MAAM,EAAE,CAAC;CACxB,iBA4BA"}

package/dist/tool-call-service/utils.js

@@ -10,9 +10,13 @@ async function sendToolRequestToRemoteQueue(payload) {
             secretAccessKey: process.env.AWS_SECRET_ACCESS_KEY,
         },
     });
-    const queueUrl = process.env.TOOL_EXECUTION_SQS_URL;
+    const useToolExecutionOnEcs = payload.featureFlags.includes("useToolExecutionOnEcs");
+    let queueUrl = process.env.TOOL_EXECUTION_SQS_URL;
+    if (useToolExecutionOnEcs) {
+        queueUrl = process.env.TOOL_EXECUTION_SQS_URL_ECS;
+    }
     if (!queueUrl) {
-        throw new Error("TOOL_EXECUTION_SQS_URL is required for remote execution.");
+        throw new Error(`${useToolExecutionOnEcs ? "TOOL_EXECUTION_SQS_URL_ECS" : "TOOL_EXECUTION_SQS_URL"} is required for remote execution.`);
     }
     await sqs.send(new client_sqs_1.SendMessageCommand({
         QueueUrl: queueUrl,

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

@@ -1,4 +1,4 @@
-import type { Tool } from "@empiricalrun/llm/chat";
+import type { Tool } from "@empiricalrun/shared-types";
 interface GitHubRepoInfo {
     owner: string;
     repo: string;

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,EAAE,IAAI,EAAE,MAAM,wBAAwB,CAAC;AAOnD,UAAU,cAAc;IACtB,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;CACd;AAED,wBAAgB,cAAc,CAAC,GAAG,EAAE,MAAM,GAAG,cAAc,CAiB1D;AA2CD,eAAO,MAAM,qBAAqB,EAAE,IAsFnC,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;AAOvD,UAAU,cAAc;IACtB,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;CACd;AAED,wBAAgB,cAAc,CAAC,GAAG,EAAE,MAAM,GAAG,cAAc,CAiB1D;AA2CD,eAAO,MAAM,qBAAqB,EAAE,IA0FnC,CAAC"}

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

@@ -55,7 +55,7 @@ Don't ask the user for this information, just come up with it yourself.
 `,
         parameters: createPullRequestSchema,
     },
-    execute: async (input, repoPath, apiKey) => {
+    execute: async ({ input, repoPath, apiKey, }) => {
         try {
             const { pullRequestTitle, pullRequestDescription } = input;
             const branchName = await (0, git_1.getCurrentBranchName)(repoPath);

package/dist/tools/diagnosis-fetcher.d.ts

@@ -1,3 +1,3 @@
-import type { Tool } from "@empiricalrun/llm/chat";
+import type { Tool } from "@empiricalrun/shared-types";
 export declare const fetchDiagnosisReportTool: Tool;
 //# sourceMappingURL=diagnosis-fetcher.d.ts.map

package/dist/tools/diagnosis-fetcher.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"diagnosis-fetcher.d.ts","sourceRoot":"","sources":["../../src/tools/diagnosis-fetcher.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,wBAAwB,CAAC;AAenD,eAAO,MAAM,wBAAwB,EAAE,IAqFtC,CAAC"}
+{"version":3,"file":"diagnosis-fetcher.d.ts","sourceRoot":"","sources":["../../src/tools/diagnosis-fetcher.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,4BAA4B,CAAC;AAevD,eAAO,MAAM,wBAAwB,EAAE,IAyFtC,CAAC"}

package/dist/tools/diagnosis-fetcher.js

@@ -19,7 +19,7 @@ exports.fetchDiagnosisReportTool = {
         description: "Fetch details about a test case diagnosis using its URL or slug",
         parameters: DiagnosisSchema,
     },
-    execute: async (input, repoPath, apiKey) => {
+    execute: async ({ input, repoPath, apiKey, }) => {
         const { diagnosisUrl } = input;
         // Extract the slug from the URL - it's the part after the last '--'
         const slug = diagnosisUrl.split("--").pop();

package/dist/tools/download-build.d.ts

@@ -1,4 +1,4 @@
-import type { Tool } from "@empiricalrun/llm/chat";
+import type { Tool } from "@empiricalrun/shared-types";
 import { z } from "zod";
 export declare const downloadBuildToolSchema: z.ZodObject<{
     buildUrl: z.ZodString;

package/dist/tools/download-build.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"download-build.d.ts","sourceRoot":"","sources":["../../src/tools/download-build.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,wBAAwB,CAAC;AACnD,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAIxB,eAAO,MAAM,uBAAuB;;;;;;EAElC,CAAC;AAEH,MAAM,MAAM,sBAAsB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,uBAAuB,CAAC,CAAC;AAE7E,eAAO,MAAM,iBAAiB,EAAE,IAkC/B,CAAC"}
+{"version":3,"file":"download-build.d.ts","sourceRoot":"","sources":["../../src/tools/download-build.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,4BAA4B,CAAC;AACvD,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAIxB,eAAO,MAAM,uBAAuB;;;;;;EAElC,CAAC;AAEH,MAAM,MAAM,sBAAsB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,uBAAuB,CAAC,CAAC;AAE7E,eAAO,MAAM,iBAAiB,EAAE,IAsC/B,CAAC"}

package/dist/tools/download-build.js

@@ -14,7 +14,7 @@ have a build URL, you can try getting the environment details with the getEnviro
 Environment details will include the build URL.`,
         parameters: exports.downloadBuildToolSchema,
     },
-    execute: async (input, repoPath, apiKey) => {
+    execute: async ({ input, repoPath, apiKey, }) => {
         if (!(await (0, test_build_1.hasDownloadScript)(repoPath))) {
             return {
                 isError: true,

package/dist/tools/environment-crud.d.ts

@@ -1,4 +1,4 @@
-import type { Tool } from "@empiricalrun/llm/chat";
+import type { Tool } from "@empiricalrun/shared-types";
 export declare const getEnvironmentTool: Tool;
 export declare const environmentTools: Tool[];
 //# sourceMappingURL=environment-crud.d.ts.map

package/dist/tools/environment-crud.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"environment-crud.d.ts","sourceRoot":"","sources":["../../src/tools/environment-crud.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,wBAAwB,CAAC;AA6DnD,eAAO,MAAM,kBAAkB,EAAE,IA6EhC,CAAC;AAGF,eAAO,MAAM,gBAAgB,EAAE,IAAI,EAAyB,CAAC"}
+{"version":3,"file":"environment-crud.d.ts","sourceRoot":"","sources":["../../src/tools/environment-crud.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,4BAA4B,CAAC;AA6DvD,eAAO,MAAM,kBAAkB,EAAE,IAiFhC,CAAC;AAGF,eAAO,MAAM,gBAAgB,EAAE,IAAI,EAAyB,CAAC"}

package/dist/tools/environment-crud.js

@@ -31,7 +31,7 @@ exports.getEnvironmentTool = {
         description: "Fetch details of an existing environment",
         parameters: GetEnvironmentSchema,
     },
-    execute: async (input, repoPath, apiKey) => {
+    execute: async ({ input, repoPath, apiKey, }) => {
         // Get project repo name
         let projectRepoName;
         try {

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

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

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

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

package/dist/tools/grep/index.js

@@ -75,8 +75,9 @@ async function usingRipgrep(input, repoPath) {
     try {
         const dir = path_1.default.join(repoPath, input.directory || "");
         const escapedPattern = input.pattern
-            .replace(/\s+/g, "\\ ")
-            .replace(/([[\]()])/g, "\\$1");
+            .replace(/[.*+?^${}()|[\]\\]/g, "\\$&") // Escape special characters that have meaning in regex/grep
+            .replace(/[&;|<>]/g, "\\$&") // Escape shell special characters
+            .replace(/\s+/g, "\\ "); // Handle spaces
         const results = (0, ripgrep_1.ripgrep)(dir, {
             string: escapedPattern,
             globs: input.filePattern ? [input.filePattern] : undefined,
@@ -123,7 +124,7 @@ If ripgrep is not available, it will fall back to using system grep.
 Search is case insensitive and regex patterns are not supported.`,
         parameters: GrepInputSchema,
     },
-    execute: async (input, repoPath) => {
+    execute: async ({ input, repoPath, }) => {
         if ((0, ripgrep_1.isRgAvailable)()) {
             return usingRipgrep(input, repoPath);
         }

package/dist/tools/str_replace_editor.d.ts

@@ -1,4 +1,4 @@
-import { Tool, ToolResult } from "@empiricalrun/llm/chat";
+import type { Tool, ToolResult } from "@empiricalrun/shared-types";
 interface StrReplaceInput {
     command: string;
     path: string;
@@ -8,16 +8,14 @@ interface StrReplaceInput {
     file_text?: string;
     insert_line?: number;
 }
-/**
- * Cleans up any backup files that were created during the editing process
- * @returns The number of backup files that were cleaned up
- */
-export declare function cleanupBackupFiles(repoDir: string): number;
 /**
  * Our implementation of Claude's built-in text editor tool
  * https://docs.anthropic.com/en/docs/build-with-claude/tool-use/text-editor-tool
  */
-export declare function strReplaceEditorExecutor(input: StrReplaceInput, repoPath: string): Promise<ToolResult>;
+export declare function strReplaceEditorExecutor({ input, repoPath, }: {
+    input: StrReplaceInput;
+    repoPath: string;
+}): Promise<ToolResult>;
 export declare const textEditorTools: Tool[];
 export {};
 //# sourceMappingURL=str_replace_editor.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"str_replace_editor.d.ts","sourceRoot":"","sources":["../../src/tools/str_replace_editor.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AA2B1D,UAAU,eAAe;IACvB,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,CAAC,EAAE,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC9B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AAqED;;;GAGG;AACH,wBAAgB,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,CAwC1D;AAMD;;;GAGG;AACH,wBAAsB,wBAAwB,CAC5C,KAAK,EAAE,eAAe,EACtB,QAAQ,EAAE,MAAM,GACf,OAAO,CAAC,UAAU,CAAC,CAkNrB;AAuHD,eAAO,MAAM,eAAe,EAAE,IAAI,EAKjC,CAAC"}
+{"version":3,"file":"str_replace_editor.d.ts","sourceRoot":"","sources":["../../src/tools/str_replace_editor.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,4BAA4B,CAAC;AAOnE,UAAU,eAAe;IACvB,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,CAAC,EAAE,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC9B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AA6UD;;;GAGG;AACH,wBAAsB,wBAAwB,CAAC,EAC7C,KAAK,EACL,QAAQ,GACT,EAAE;IACD,KAAK,EAAE,eAAe,CAAC;IACvB,QAAQ,EAAE,MAAM,CAAC;CAClB,GAAG,OAAO,CAAC,UAAU,CAAC,CA2CtB;AAsID,eAAO,MAAM,eAAe,EAAE,IAAI,EAKjC,CAAC"}