@empiricalrun/test-gen 0.64.1 → 0.64.2

package/CHANGELOG.md

@@ -1,5 +1,14 @@
 # @empiricalrun/test-gen
 
+## 0.64.2
+
+### Patch Changes
+
+- f05aac2: feat: add environment overrides for projects in tool calls
+- Updated dependencies [f05aac2]
+  - @empiricalrun/test-run@0.10.1
+  - @empiricalrun/llm@0.17.3
+
 ## 0.64.1
 
 ### Patch Changes

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

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/agent/chat/index.ts"],"names":[],"mappings":"AACA,OAAO,EAGL,mBAAmB,EACpB,MAAM,wBAAwB,CAAC;AAyChC,wBAAsB,kBAAkB,CAAC,EACvC,mBAAmB,EACnB,aAAa,EACb,oBAAoB,EACpB,SAAS,GACV,EAAE;IACD,aAAa,EAAE,mBAAmB,CAAC;IACnC,mBAAmB,EAAE,OAAO,CAAC;IAC7B,oBAAoB,EAAE,MAAM,GAAG,SAAS,CAAC;IACzC,SAAS,CAAC,EAAE,OAAO,CAAC;CACrB,iBAuHA;AAuBD,wBAAsB,wBAAwB,CAAC,EAC7C,aAAa,EACb,aAAa,GACd,EAAE;IACD,aAAa,EAAE,mBAAmB,CAAC;IACnC,aAAa,EAAE,MAAM,CAAC;CACvB,iBA6DA"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/agent/chat/index.ts"],"names":[],"mappings":"AACA,OAAO,EAGL,mBAAmB,EACpB,MAAM,wBAAwB,CAAC;AAyChC,wBAAsB,kBAAkB,CAAC,EACvC,mBAAmB,EACnB,aAAa,EACb,oBAAoB,EACpB,SAAS,GACV,EAAE;IACD,aAAa,EAAE,mBAAmB,CAAC;IACnC,mBAAmB,EAAE,OAAO,CAAC;IAC7B,oBAAoB,EAAE,MAAM,GAAG,SAAS,CAAC;IACzC,SAAS,CAAC,EAAE,OAAO,CAAC;CACrB,iBAwHA;AAuBD,wBAAsB,wBAAwB,CAAC,EAC7C,aAAa,EACb,aAAa,GACd,EAAE;IACD,aAAa,EAAE,mBAAmB,CAAC;IACnC,aAAa,EAAE,MAAM,CAAC;CACvB,iBA6DA"}

package/dist/agent/chat/index.js

@@ -111,6 +111,7 @@ async function runChatAgentForCLI({ useDiskForChatState, selectedModel, initialP
                 apiKey: process.env.EMPIRICALRUN_API_KEY,
                 trace,
                 featureFlags: [],
+                environmentOverrides: {},
             });
             const fileInfo = await (0, file_tree_1.getFileInfoFromFS)(process.cwd());
             await (0, agent_loop_1.chatAgentLoop)({

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

@@ -15,7 +15,8 @@ export declare class ToolCallService {
     apiKey: string | undefined;
     trace?: TraceClient;
     featureFlags: string[];
-    constructor({ chatSessionId, selectedModel, branchName, repoPath, apiKey, trace, featureFlags, }: {
+    environmentOverrides: Record<string, string>;
+    constructor({ chatSessionId, selectedModel, branchName, repoPath, apiKey, trace, featureFlags, environmentOverrides, }: {
         chatSessionId: number | null;
         selectedModel: SupportedChatModels;
         branchName: string;
@@ -23,6 +24,7 @@ export declare class ToolCallService {
         apiKey: string | undefined;
         trace?: TraceClient;
         featureFlags: string[];
+        environmentOverrides?: Record<string, 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,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"}
+{"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;IACvB,oBAAoB,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;gBACjC,EACV,aAAa,EACb,aAAa,EACb,UAAU,EACV,QAAQ,EACR,MAAM,EACN,KAAK,EACL,YAAY,EACZ,oBAAyB,GAC1B,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;QACvB,oBAAoB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;KAC/C;IAmCK,WAAW,CAAC,SAAS,EAAE,eAAe,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAkBxD,OAAO,CAAC,SAAS,EAAE,eAAe,EAAE,GAAG,OAAO,CAAC,UAAU,EAAE,CAAC;CAuEnE"}

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

@@ -24,7 +24,8 @@ class ToolCallService {
     apiKey;
     trace;
     featureFlags;
-    constructor({ chatSessionId, selectedModel, branchName, repoPath, apiKey, trace, featureFlags, }) {
+    environmentOverrides;
+    constructor({ chatSessionId, selectedModel, branchName, repoPath, apiKey, trace, featureFlags, environmentOverrides = {}, }) {
         this.chatSessionId = chatSessionId;
         this.selectedModel = selectedModel;
         this.branchName = branchName;
@@ -32,6 +33,7 @@ class ToolCallService {
         this.repoPath = repoPath;
         this.apiKey = apiKey;
         this.featureFlags = featureFlags;
+        this.environmentOverrides = environmentOverrides;
         this.tools = [
             grep_1.grepTool,
             test_run_1.runTestTool,
@@ -62,13 +64,16 @@ class ToolCallService {
         if (!requestId) {
             throw new Error("Could not find an id for the tool call.");
         }
-        await (0, utils_1.sendToolRequestToRemoteQueue)({
+        const queueUrl = (0, utils_1.getQueueUrl)(this.featureFlags);
+        if (!queueUrl) {
+            throw new Error(`queueUrl is required for remote execution.`);
+        }
+        await (0, utils_1.sendToolRequestToRemoteQueue)(queueUrl, {
             toolCalls,
             requestId,
             chatSessionId: this.chatSessionId,
             selectedModel: this.selectedModel,
             branchName: this.branchName,
-            featureFlags: this.featureFlags || [],
         });
     }
     async execute(toolCalls) {
@@ -110,6 +115,7 @@ class ToolCallService {
                     apiKey: this.apiKey,
                     trace: this.trace,
                     collectArtifacts: collectArtifactsFn,
+                    environmentOverrides: this.environmentOverrides,
                 });
                 const artifacts = await uploadArtifactsQueue.waitForCompletion();
                 toolResults.push({ ...result, artifacts });

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

@@ -1,11 +1,11 @@
 import { SupportedChatModels } from "@empiricalrun/llm/chat";
 import { PendingToolCall } from "@empiricalrun/shared-types";
-export declare function sendToolRequestToRemoteQueue(payload: {
+export declare function getQueueUrl(featureFlags: string[]): string | undefined;
+export declare function sendToolRequestToRemoteQueue(queueUrl: string, payload: {
     toolCalls: PendingToolCall[];
     requestId: string;
     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;IACnB,YAAY,EAAE,MAAM,EAAE,CAAC;CACxB,iBA4BA"}
+{"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,wBAAgB,WAAW,CAAC,YAAY,EAAE,MAAM,EAAE,sBAMjD;AAED,wBAAsB,4BAA4B,CAChD,QAAQ,EAAE,MAAM,EAChB,OAAO,EAAE;IACP,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,iBAiBF"}

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

@@ -1,8 +1,16 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
+exports.getQueueUrl = getQueueUrl;
 exports.sendToolRequestToRemoteQueue = sendToolRequestToRemoteQueue;
 const client_sqs_1 = require("@aws-sdk/client-sqs");
-async function sendToolRequestToRemoteQueue(payload) {
+function getQueueUrl(featureFlags) {
+    const useToolExecutionOnEcs = featureFlags.includes("useToolExecutionOnEcs");
+    if (useToolExecutionOnEcs) {
+        return process.env.TOOL_EXECUTION_SQS_URL_ECS;
+    }
+    return process.env.TOOL_EXECUTION_SQS_URL;
+}
+async function sendToolRequestToRemoteQueue(queueUrl, payload) {
     const sqs = new client_sqs_1.SQSClient({
         region: process.env.AWS_REGION,
         credentials: {
@@ -10,14 +18,6 @@ async function sendToolRequestToRemoteQueue(payload) {
             secretAccessKey: process.env.AWS_SECRET_ACCESS_KEY,
         },
     });
-    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(`${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,
         MessageBody: JSON.stringify(payload),

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,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"}
+{"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;AAOpC,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,IA6FnC,CAAC"}

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,4BAA4B,CAAC;AAevD,eAAO,MAAM,wBAAwB,EAAE,IAyFtC,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;AAepE,eAAO,MAAM,wBAAwB,EAAE,IAqFtC,CAAC"}

package/dist/tools/diagnosis-fetcher.js

@@ -19,7 +19,9 @@ exports.fetchDiagnosisReportTool = {
         description: "Fetch details about a test case diagnosis using its URL or slug",
         parameters: DiagnosisSchema,
     },
-    execute: async ({ input, repoPath, apiKey, }) => {
+    execute: async (params) => {
+        const { input } = params;
+        const { repoPath, apiKey } = params;
         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.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,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"}
+{"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;AACpE,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"}

package/dist/tools/download-build.js

@@ -14,7 +14,9 @@ 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 (params) => {
+        const { input } = params;
+        const { repoPath, apiKey } = params;
         if (!(await (0, test_build_1.hasDownloadScript)(repoPath))) {
             return {
                 isError: true,

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,4BAA4B,CAAC;AA6DvD,eAAO,MAAM,kBAAkB,EAAE,IAiFhC,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,EAEV,IAAI,EAEL,MAAM,4BAA4B,CAAC;AA6DpC,eAAO,MAAM,kBAAkB,EAAE,IAoFhC,CAAC;AAGF,eAAO,MAAM,gBAAgB,EAAE,IAAI,EAAyB,CAAC"}

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,4BAA4B,CAAC;AA8HnE,eAAO,MAAM,QAAQ,EAAE,IAuBtB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/tools/grep/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAEV,IAAI,EAGL,MAAM,4BAA4B,CAAC;AA8HpC,eAAO,MAAM,QAAQ,EAAE,IA2BtB,CAAC"}

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,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"}
+{"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;AAOhF,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;AA8HD,eAAO,MAAM,eAAe,EAAE,IAAI,EAKjC,CAAC"}

package/dist/tools/str_replace_editor.js

@@ -340,7 +340,9 @@ File contents are returned with line numbers, starting from 1.
             path: zod_1.z.string().describe("The path to the file or directory to view."),
         }),
     },
-    execute: async ({ input, repoPath, }) => {
+    execute: async (params) => {
+        const { input } = params;
+        const { repoPath } = params;
         return strReplaceEditorExecutor({
             input: {
                 command: "view",
@@ -362,7 +364,9 @@ directories "tests", "tests/example", and "tests/example/foo.spec.ts".`,
             file_text: zod_1.z.string().describe("The contents of the new file."),
         }),
     },
-    execute: async ({ input, repoPath, }) => {
+    execute: async (params) => {
+        const { input } = params;
+        const { repoPath } = params;
         return strReplaceEditorExecutor({
             input: {
                 command: "create",
@@ -384,7 +388,9 @@ in the file. If old_str is not unique, the tool will return an error.`,
             new_str: zod_1.z.string().describe("The string to replace old_str with."),
         }),
     },
-    execute: async ({ input, repoPath, }) => {
+    execute: async (params) => {
+        const { input } = params;
+        const { repoPath } = params;
         return strReplaceEditorExecutor({
             input: {
                 command: "str_replace",
@@ -412,7 +418,9 @@ To insert a string at the end of the file, you should use insert_line = (total l
             new_str: zod_1.z.string().describe("The string to insert."),
         }),
     },
-    execute: async ({ input, repoPath, }) => {
+    execute: async (params) => {
+        const { input } = params;
+        const { repoPath } = params;
         return strReplaceEditorExecutor({
             input: {
                 command: "insert",

package/dist/tools/test-gen-browser.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"test-gen-browser.d.ts","sourceRoot":"","sources":["../../src/tools/test-gen-browser.ts"],"names":[],"mappings":"AACA,OAAO,EAAoB,IAAI,EAAE,MAAM,4BAA4B,CAAC;AAmGpE,eAAO,MAAM,4BAA4B,EAAE,IAwI1C,CAAC"}
+{"version":3,"file":"test-gen-browser.d.ts","sourceRoot":"","sources":["../../src/tools/test-gen-browser.ts"],"names":[],"mappings":"AACA,OAAO,EAAoB,IAAI,EAAE,MAAM,4BAA4B,CAAC;AAmGpE,eAAO,MAAM,4BAA4B,EAAE,IA0I1C,CAAC"}

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

@@ -95,7 +95,7 @@ exports.generateTestWithBrowserAgent = {
         description: BROWSER_AGENT_DESCRIPTION,
         parameters: BrowserAgentSchema,
     },
-    execute: async ({ input, repoPath, trace, collectArtifacts, }) => {
+    execute: async ({ input, repoPath, trace, collectArtifacts, environmentOverrides = {}, }) => {
         const { filePath, project } = input;
         const absoluteFilePath = path_1.default.join(repoPath, filePath);
         if (!fs_1.default.existsSync(absoluteFilePath)) {
@@ -152,7 +152,7 @@ exports.generateTestWithBrowserAgent = {
         catch (err) {
             console.warn("[generateTestWithBrowserAgent] Error preparing playwright for codegen", err);
         }
-        let envOverrides = {};
+        let envOverrides = { ...environmentOverrides };
         if (input.buildUrl) {
             envOverrides = {
                 ...envOverrides,

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

@@ -1 +1 @@
-{"version":3,"file":"test-run.d.ts","sourceRoot":"","sources":["../../src/tools/test-run.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAc,MAAM,4BAA4B,CAAC;AAsDnE,eAAO,MAAM,WAAW,EAAE,IAiFzB,CAAC"}
+{"version":3,"file":"test-run.d.ts","sourceRoot":"","sources":["../../src/tools/test-run.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAc,MAAM,4BAA4B,CAAC;AAsDnE,eAAO,MAAM,WAAW,EAAE,IAoFzB,CAAC"}

package/dist/tools/test-run.js

@@ -45,14 +45,15 @@ exports.runTestTool = {
         description: "Run a test",
         parameters: RunTestSchema,
     },
-    execute: async ({ input, repoPath, collectArtifacts, }) => {
+    execute: async ({ input, repoPath, collectArtifacts, environmentOverrides = {}, }) => {
         let reportUrl = undefined;
-        let envOverrides = {};
+        let envOverrides = { ...environmentOverrides };
         if ((0, artifacts_1.isArtifactCollectionEnabled)()) {
             const projectName = "test-gen-chat-agent";
             const testRunId = Date.now().toString();
             reportUrl = buildReportUrl(projectName, testRunId);
             envOverrides = {
+                ...envOverrides,
                 PROJECT_NAME: projectName,
                 TEST_RUN_GITHUB_ACTION_ID: testRunId,
             };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@empiricalrun/test-gen",
-  "version": "0.64.1",
+  "version": "0.64.2",
   "publishConfig": {
     "registry": "https://registry.npmjs.org/",
     "access": "public"
@@ -64,7 +64,7 @@
     "zod": "^3.23.8",
     "@empiricalrun/llm": "^0.17.3",
     "@empiricalrun/r2-uploader": "^0.3.9",
-    "@empiricalrun/test-run": "^0.10.0"
+    "@empiricalrun/test-run": "^0.10.1"
   },
   "devDependencies": {
     "@playwright/test": "1.47.1",
@@ -79,7 +79,7 @@
     "playwright": "1.47.1",
     "serve-handler": "^6.1.6",
     "ts-patch": "^3.3.0",
-    "@empiricalrun/shared-types": "0.4.1"
+    "@empiricalrun/shared-types": "0.4.2"
   },
   "scripts": {
     "dev": "tspc --build --watch",