@empiricalrun/test-gen 0.62.0 → 0.64.0

package/CHANGELOG.md

@@ -1,5 +1,38 @@
 # @empiricalrun/test-gen
 
+## 0.64.0
+
+### Minor Changes
+
+- 1a5ec8d: feat: stateless tool execution and named args for tool execute fn
+
+### Patch Changes
+
+- e5b9f7e: feat: consume repoPath in browser agent
+- 43dc453: fix: record locators before execution for more reliable codegen
+- 24d9415: fix: error message in browser agent tool call + removed unused code
+- Updated dependencies [1a5ec8d]
+  - @empiricalrun/llm@0.17.2
+
+## 0.63.0
+
+### Minor Changes
+
+- f926e40: feat: consume repoPath in test run tool
+- 55c7913: feat: artifact collection is fire-and-forget at tool level
+- 2f1ee31: chore: consume repoPath and apikey in tools requiring Dashboard api call
+
+### Patch Changes
+
+- c491cdd: fix: insert_line description in text editor tools
+- 0fea8f1: test: add a multiline old_str replacement test
+- 115a023: chore: allow grep tool to consume repoPath for the scope of search
+- 6c7740b: feat: consume repoPath for all file edit tools
+- Updated dependencies [f926e40]
+- Updated dependencies [2f1ee31]
+  - @empiricalrun/test-run@0.9.4
+  - @empiricalrun/llm@0.17.1
+
 ## 0.62.0
 
 ### Minor Changes

package/dist/agent/browsing/run.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"run.d.ts","sourceRoot":"","sources":["../../../src/agent/browsing/run.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,MAAM,4BAA4B,CAAC;AAc3D,KAAK,iBAAiB,GAAG;IACvB,YAAY,EAAE,MAAM,CAAC;IACrB,cAAc,EAAE,MAAM,EAAE,CAAC;IACzB,YAAY,EAAE,MAAM,CAAC;IACrB,gBAAgB,EAAE,MAAM,CAAC;IACzB,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,EAAE,MAAM,CAAC;IACrB,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,yBAAyB,EAAE,OAAO,CAAC;CACpC,CAAC;AAEF,wBAAsB,8BAA8B,CAAC,EACnD,gBAAgB,EAChB,OAAO,EACP,YAAY,GACb,EAAE;IACD,gBAAgB,EAAE,MAAM,EAAE,CAAC;IAC3B,OAAO,EAAE,MAAM,CAAC;IAChB,YAAY,EAAE,MAAM,CAAC;CACtB,GAAG,OAAO,CAAC,MAAM,CAAC,CAQlB;AAED,wBAAsB,6BAA6B,CAAC,EAClD,YAAY,EACZ,cAAc,EACd,YAAY,EACZ,gBAAgB,EAChB,WAAW,EACX,YAAY,EACZ,OAAO,EACP,OAAO,EACP,yBAAyB,GAC1B,EAAE,iBAAiB,GAAG,OAAO,CAAC;IAC7B,OAAO,EAAE,OAAO,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,SAAS,CAAC,EAAE,aAAa,EAAE,CAAC;CAC7B,CAAC,CAmFD"}
+{"version":3,"file":"run.d.ts","sourceRoot":"","sources":["../../../src/agent/browsing/run.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,MAAM,4BAA4B,CAAC;AAe3D,KAAK,iBAAiB,GAAG;IACvB,YAAY,EAAE,MAAM,CAAC;IACrB,cAAc,EAAE,MAAM,EAAE,CAAC;IACzB,YAAY,EAAE,MAAM,CAAC;IACrB,gBAAgB,EAAE,MAAM,CAAC;IACzB,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,EAAE,MAAM,CAAC;IACrB,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,yBAAyB,EAAE,OAAO,CAAC;CACpC,CAAC;AAEF,wBAAsB,8BAA8B,CAAC,EACnD,gBAAgB,EAChB,OAAO,EACP,YAAY,GACb,EAAE;IACD,gBAAgB,EAAE,MAAM,EAAE,CAAC;IAC3B,OAAO,EAAE,MAAM,CAAC;IAChB,YAAY,EAAE,MAAM,CAAC;CACtB,GAAG,OAAO,CAAC,MAAM,CAAC,CAQlB;AAED,wBAAsB,6BAA6B,CAAC,EAClD,YAAY,EACZ,cAAc,EACd,YAAY,EACZ,gBAAgB,EAChB,WAAW,EACX,YAAY,EACZ,OAAO,EACP,OAAO,EACP,yBAAyB,GAC1B,EAAE,iBAAiB,GAAG,OAAO,CAAC;IAC7B,OAAO,EAAE,OAAO,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,SAAS,CAAC,EAAE,aAAa,EAAE,CAAC;CAC7B,CAAC,CA2FD"}

package/dist/agent/browsing/run.js

@@ -8,6 +8,7 @@ exports.generateTestsUsingMasterAgent = generateTestsUsingMasterAgent;
 const test_run_1 = require("@empiricalrun/test-run");
 const detect_port_1 = __importDefault(require("detect-port"));
 const fs_1 = __importDefault(require("fs"));
+const path_1 = __importDefault(require("path"));
 const web_1 = require("../../bin/utils/platform/web");
 const server_1 = require("../../file/server");
 const exec_1 = require("../../utils/exec");
@@ -18,8 +19,10 @@ async function convertProjectsFilterToProject({ pwProjectsFilter, repoDir, testF
     return project;
 }
 async function generateTestsUsingMasterAgent({ testCaseName, testCaseSuites, testFilePath, filePathToUpdate, projectName, testGenToken, repoDir, traceId, editFileWithGeneratedCode, }) {
-    if (!fs_1.default.existsSync(testFilePath)) {
-        throw new Error(`File for master agent to run not found: ${testFilePath}`);
+    const absTestFilePath = path_1.default.join(repoDir, testFilePath);
+    if (!fs_1.default.existsSync(absTestFilePath)) {
+        const errorMsg = `File for master agent to run not found: ${testFilePath}`;
+        throw new Error(errorMsg);
     }
     const pm = new exec_1.ProcessManager();
     const availablePort = await (0, detect_port_1.default)(3030);
@@ -32,7 +35,8 @@ async function generateTestsUsingMasterAgent({ testCaseName, testCaseSuites, tes
     });
     await fileServer.startFileService();
     fileServer.setFilePath(filePathToUpdate);
-    const pageVar = await (0, web_1.getPageVariableNameFromCreateTest)(filePathToUpdate);
+    const absFilePathToUpdate = path_1.default.join(repoDir, filePathToUpdate);
+    const pageVar = await (0, web_1.getPageVariableNameFromCreateTest)(absFilePathToUpdate);
     const testsDir = "tests";
     const cmd = await (0, test_run_1.runSpecificTestsCmd)({
         tests: [
@@ -66,11 +70,12 @@ async function generateTestsUsingMasterAgent({ testCaseName, testCaseSuites, tes
                 // Enable this for the page.pause approach in codegen
                 PW_CODEGEN_NO_INSPECTOR: "1",
             },
+            cwd: repoDir,
         });
     }
     catch (e) {
         error = `Failed to generate test using master agent ${e}`;
-        console.error(error);
+        console.error(`[generateTestsUsingMasterAgent] ${error}`);
     }
     if (error) {
         // Clean up the file if there is any error
@@ -81,7 +86,7 @@ async function generateTestsUsingMasterAgent({ testCaseName, testCaseSuites, tes
             await (0, web_1.lintErrors)(filePathToUpdate);
         }
         catch (e) {
-            console.error("Failed to remove extra scripts from files post test gen error", e);
+            console.error(`[generateTestsUsingMasterAgent] Failed to remove extra scripts from files post test gen error:`, e);
         }
     }
     await fileServer.stop();

package/dist/agent/browsing/utils.d.ts

@@ -4,7 +4,7 @@ import { Page } from "playwright";
 import { PlaywrightTestConfig } from "playwright/test";
 export declare function isRegExp(obj: any): obj is RegExp;
 export declare function prepareBrowsingAgentTask(steps: string[]): string;
-export declare function replaceTodoWithCreateTest(testFilePath: string): Promise<void>;
+export declare function replaceTodoWithCreateTest(testFilePath: string, repoDir: string): Promise<void>;
 export declare function markTestAsOnly({ testCaseName, testCaseSuites, specPath, }: {
     testCaseName: string;
     testCaseSuites: string[];

package/dist/agent/browsing/utils.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../src/agent/browsing/utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,4BAA4B,CAAC;AAI3D,OAAO,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAClC,OAAO,EAAE,oBAAoB,EAAE,MAAM,iBAAiB,CAAC;AAsBvD,wBAAgB,QAAQ,CAAC,GAAG,EAAE,GAAG,GAAG,GAAG,IAAI,MAAM,CAKhD;AAED,wBAAgB,wBAAwB,CAAC,KAAK,EAAE,MAAM,EAAE,UAIvD;AAiFD,wBAAsB,yBAAyB,CAAC,YAAY,EAAE,MAAM,iBAsBnE;AAED,wBAAsB,cAAc,CAAC,EACnC,YAAY,EACZ,cAAc,EACd,QAAQ,GACT,EAAE;IACD,YAAY,EAAE,MAAM,CAAC;IACrB,cAAc,EAAE,MAAM,EAAE,CAAC;IACzB,QAAQ,EAAE,MAAM,CAAC;CAClB,iBAoBA;AAED,wBAAsB,yBAAyB,CAAC,EAC9C,QAAQ,EACR,QAAQ,EACR,KAAK,GACN,EAAE;IACD,QAAQ,EAAE,QAAQ,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,CAAC,EAAE,WAAW,CAAC;CACrB,GAAG,OAAO,CAAC,MAAM,CAAC,CAyDlB;AAyBD,wBAAsB,wBAAwB,CAAC,IAAI,EAAE,IAAI,iBA8HxD;AAED;;;GAGG;AACH,wBAAsB,oBAAoB,CACxC,OAAO,EAAE,MAAM,GACd,OAAO,CAAC,oBAAoB,CAAC,CA2B/B;AAWD,wBAAsB,oBAAoB,CACxC,gBAAgB,EAAE,oBAAoB,GACrC,OAAO,CAAC,MAAM,EAAE,CAAC,CAQnB;AAED;;;;;GAKG;AACH,wBAAsB,iBAAiB,CACrC,YAAY,EAAE,MAAM,EACpB,gBAAgB,EAAE,oBAAoB,EACtC,gBAAgB,GAAE,MAAM,EAAU,GACjC,OAAO,CAAC,MAAM,CAAC,CA+CjB"}
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../src/agent/browsing/utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,4BAA4B,CAAC;AAI3D,OAAO,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAClC,OAAO,EAAE,oBAAoB,EAAE,MAAM,iBAAiB,CAAC;AAsBvD,wBAAgB,QAAQ,CAAC,GAAG,EAAE,GAAG,GAAG,GAAG,IAAI,MAAM,CAKhD;AAED,wBAAgB,wBAAwB,CAAC,KAAK,EAAE,MAAM,EAAE,UAIvD;AAiFD,wBAAsB,yBAAyB,CAC7C,YAAY,EAAE,MAAM,EACpB,OAAO,EAAE,MAAM,iBAwBhB;AAED,wBAAsB,cAAc,CAAC,EACnC,YAAY,EACZ,cAAc,EACd,QAAQ,GACT,EAAE;IACD,YAAY,EAAE,MAAM,CAAC;IACrB,cAAc,EAAE,MAAM,EAAE,CAAC;IACzB,QAAQ,EAAE,MAAM,CAAC;CAClB,iBAoBA;AAED,wBAAsB,yBAAyB,CAAC,EAC9C,QAAQ,EACR,QAAQ,EACR,KAAK,GACN,EAAE;IACD,QAAQ,EAAE,QAAQ,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,CAAC,EAAE,WAAW,CAAC;CACrB,GAAG,OAAO,CAAC,MAAM,CAAC,CAyDlB;AAyBD,wBAAsB,wBAAwB,CAAC,IAAI,EAAE,IAAI,iBA8HxD;AAED;;;GAGG;AACH,wBAAsB,oBAAoB,CACxC,OAAO,EAAE,MAAM,GACd,OAAO,CAAC,oBAAoB,CAAC,CA2B/B;AAWD,wBAAsB,oBAAoB,CACxC,gBAAgB,EAAE,oBAAoB,GACrC,OAAO,CAAC,MAAM,EAAE,CAAC,CAQnB;AAED;;;;;GAKG;AACH,wBAAsB,iBAAiB,CACrC,YAAY,EAAE,MAAM,EACpB,gBAAgB,EAAE,oBAAoB,EACtC,gBAAgB,GAAE,MAAM,EAAU,GACjC,OAAO,CAAC,MAAM,CAAC,CA+CjB"}

package/dist/agent/browsing/utils.js

@@ -93,10 +93,11 @@ async function prepareFileForUpdateScenario({ testCase, specPath, trace, }) {
     });
     return createTestFilePath;
 }
-async function replaceTodoWithCreateTest(testFilePath) {
+async function replaceTodoWithCreateTest(testFilePath, repoDir) {
     // This method is an alternative to prepareFileForUpdateScenario
     // TODO: Does not support scoped variables and updates in POM files
-    const fileContent = fs_1.default.readFileSync(testFilePath, "utf-8");
+    const absoluteTestFilePath = path_1.default.join(repoDir, testFilePath);
+    const fileContent = fs_1.default.readFileSync(absoluteTestFilePath, "utf-8");
     const todoRegex = /\/\/ TODO\(agent(?:\s+on\s+(\w+))?\):\s*(.*)/;
     const todoMatch = fileContent.match(todoRegex);
     if (!todoMatch) {
@@ -104,8 +105,8 @@ async function replaceTodoWithCreateTest(testFilePath) {
     }
     const [, pageVarName] = todoMatch;
     const pageVariable = pageVarName || "page"; // Default to "page" if not specified
-    fs_1.default.writeFileSync(testFilePath, fileContent.replace(todoRegex, (_, __, todoText) => `await createTest("${todoText.replace(/"/g, '\\"')}", ${pageVariable});`));
-    await addImportForCreateTest(testFilePath);
+    fs_1.default.writeFileSync(absoluteTestFilePath, fileContent.replace(todoRegex, (_, __, todoText) => `await createTest("${todoText.replace(/"/g, '\\"')}", ${pageVariable});`));
+    await addImportForCreateTest(absoluteTestFilePath);
 }
 async function markTestAsOnly({ testCaseName, testCaseSuites, specPath, }) {
     const testFileContent = fs_1.default.readFileSync(specPath, "utf-8");

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,iBAmHA;AAuBD,wBAAsB,wBAAwB,CAAC,EAC7C,aAAa,EACb,aAAa,GACd,EAAE;IACD,aAAa,EAAE,mBAAmB,CAAC;IACnC,aAAa,EAAE,MAAM,CAAC;CACvB,iBAyDA"}
+{"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,iBAsHA;AAuBD,wBAAsB,wBAAwB,CAAC,EAC7C,aAAa,EACb,aAAa,GACd,EAAE;IACD,aAAa,EAAE,mBAAmB,CAAC;IACnC,aAAa,EAAE,MAAM,CAAC;CACvB,iBA4DA"}

package/dist/agent/chat/index.js

@@ -29,7 +29,7 @@ async function runChatAgentForCLI({ useDiskForChatState, selectedModel, initialP
     // TODO: Store branch name in chat state so that we don't recreate it every time
     const randomId = crypto.randomUUID().substring(0, 8);
     const branchName = `branch-${randomId}`;
-    await (0, git_1.checkoutBranch)(branchName);
+    await (0, git_1.checkoutBranch)(branchName, process.cwd());
     let messagesLoadedFromDisk = chatState?.messages || [];
     let chatModel = (0, chat_1.createChatModel)(messagesLoadedFromDisk, selectedModel);
     if (initialPromptContent && chatModel.messages.length === 0) {
@@ -100,6 +100,9 @@ async function runChatAgentForCLI({ useDiskForChatState, selectedModel, initialP
         }
         else {
             // TODO: Should we pass a loader function? That would allow us to show a spinner
+            if (!process.env.EMPIRICALRUN_API_KEY) {
+                throw new Error("EMPIRICALRUN_API_KEY is not set");
+            }
             const toolCallService = new tool_call_service_1.ToolCallService({
                 chatSessionId: null,
                 selectedModel,
@@ -159,6 +162,9 @@ async function runChatAgentForDashboard({ chatSessionId, selectedModel, }) {
             chatSessionId,
         },
     });
+    if (!process.env.EMPIRICALRUN_API_KEY) {
+        throw new Error("EMPIRICALRUN_API_KEY is not set");
+    }
     const toolCallService = new tool_call_service_1.ToolCallService({
         chatSessionId,
         selectedModel,
@@ -167,7 +173,7 @@ async function runChatAgentForDashboard({ chatSessionId, selectedModel, }) {
         apiKey: process.env.EMPIRICALRUN_API_KEY,
         trace,
     });
-    await (0, git_1.checkoutBranch)(branchName);
+    await (0, git_1.checkoutBranch)(branchName, process.cwd());
     let chatModel = (0, chat_1.createChatModel)(chatState.messages, selectedModel);
     let reporterFunc = async (chatState, latest) => {
         const response = await fetch(`${DASHBOARD_DOMAIN}/api/chat-sessions/${chatSessionId}`, {

package/dist/agent/cua/computer.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"computer.d.ts","sourceRoot":"","sources":["../../../src/agent/cua/computer.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,wBAAwB,EAAE,MAAM,0CAA0C,CAAC;AACpF,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAEvC,OAAO,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAE3D,KAAK,oBAAoB,GACrB,wBAAwB,CAAC,KAAK,GAC9B,wBAAwB,CAAC,WAAW,GACpC,wBAAwB,CAAC,IAAI,GAC7B,wBAAwB,CAAC,QAAQ,GACjC,wBAAwB,CAAC,IAAI,GAC7B,wBAAwB,CAAC,UAAU,GACnC,wBAAwB,CAAC,MAAM,GAC/B,wBAAwB,CAAC,IAAI,GAC7B,wBAAwB,CAAC,IAAI,CAAC;AAElC,wBAAsB,aAAa,CAAC,IAAI,EAAE,IAAI,mBAG7C;AAgCD,wBAAsB,kBAAkB,CACtC,IAAI,EAAE,IAAI,EACV,MAAM,EAAE,oBAAoB,GAAG;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,GAAG,EAAE,MAAM,CAAA;CAAE,EAC5D,OAAO,EAAE,qBAAqB,GAC7B,OAAO,CAAC;IACT,aAAa,EAAE,MAAM,CAAC;IACtB,UAAU,EAAE,MAAM,CAAC;CACpB,CAAC,CAkJD"}
+{"version":3,"file":"computer.d.ts","sourceRoot":"","sources":["../../../src/agent/cua/computer.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,wBAAwB,EAAE,MAAM,0CAA0C,CAAC;AACpF,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAEvC,OAAO,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAE3D,KAAK,oBAAoB,GACrB,wBAAwB,CAAC,KAAK,GAC9B,wBAAwB,CAAC,WAAW,GACpC,wBAAwB,CAAC,IAAI,GAC7B,wBAAwB,CAAC,QAAQ,GACjC,wBAAwB,CAAC,IAAI,GAC7B,wBAAwB,CAAC,UAAU,GACnC,wBAAwB,CAAC,MAAM,GAC/B,wBAAwB,CAAC,IAAI,GAC7B,wBAAwB,CAAC,IAAI,CAAC;AAElC,wBAAsB,aAAa,CAAC,IAAI,EAAE,IAAI,mBAG7C;AAgCD,wBAAsB,kBAAkB,CACtC,IAAI,EAAE,IAAI,EACV,MAAM,EAAE,oBAAoB,GAAG;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,GAAG,EAAE,MAAM,CAAA;CAAE,EAC5D,OAAO,EAAE,qBAAqB,GAC7B,OAAO,CAAC;IACT,aAAa,EAAE,MAAM,CAAC;IACtB,UAAU,EAAE,MAAM,CAAC;CACpB,CAAC,CAmKD"}

package/dist/agent/cua/computer.js

@@ -40,14 +40,21 @@ async function executeModelAction(page, action, codegen) {
     let actionCode = "";
     let actionSummary = "";
     try {
+        // For every action, this method:
+        // 1. Record the action in codegen
+        //    -> This allows codegen to find locators for the element
+        //       before the action is executed - execution can be a destructive
+        //       operation that modifies the DOM
+        // 2. Execute the action
+        // 3. Add to action summary
+        // 4. Add to action code
         switch (actionType) {
-            // For every action, this method 1. executes the action,
-            // 2. adds to action summary, and 3. adds to action code
             case "goto": {
                 const { url } = action;
+                await codegen.recordAction({ type: "goto", url });
                 await page.goto(url);
                 actionSummary = `Navigated page to ${url}`;
-                actionCode = await codegen.codeForAction({ type: "goto", url });
+                actionCode = await codegen.getCodeForLastAction();
                 break;
             }
             case "click": {
@@ -61,20 +68,23 @@ async function executeModelAction(page, action, codegen) {
                     pwButton = "middle";
                 }
                 if (pwButton) {
+                    await codegen.recordAction({ type: "click", x, y });
                     await page.mouse.click(x, y, { button: pwButton });
-                    actionCode = await codegen.codeForAction({ type: "click", x, y });
                     actionSummary = `Click at (${x}, ${y}) with button '${button}'`;
+                    actionCode = await codegen.getCodeForLastAction();
                 }
                 if (button === "back" || button === "forward") {
                     if (button === "back") {
+                        await codegen.recordAction({ type: "back" });
                         await page.goBack();
-                        actionCode = await codegen.codeForAction({ type: "back" });
                         actionSummary = `Navigated page back`;
+                        actionCode = await codegen.getCodeForLastAction();
                     }
                     else if (button === "forward") {
+                        await codegen.recordAction({ type: "forward" });
                         await page.goForward();
-                        actionCode = await codegen.codeForAction({ type: "forward" });
                         actionSummary = `Navigated page forward`;
+                        actionCode = await codegen.getCodeForLastAction();
                     }
                 }
                 break;
@@ -82,17 +92,28 @@ async function executeModelAction(page, action, codegen) {
             case "double_click": {
                 const { x, y } = action;
                 console.log(`Action: doubleclick at (${x}, ${y})`);
+                await codegen.recordAction({ type: "doubleclick", x, y });
                 await page.mouse.dblclick(x, y, { button: "left" });
                 actionSummary = `Double click at (${x}, ${y})`;
-                actionCode = await codegen.codeForAction({ type: "doubleclick", x, y });
+                actionCode = await codegen.getCodeForLastAction();
+                break;
+            }
+            case "type": {
+                const { text } = action;
+                console.log(`Action: type text '${text}'`);
+                await codegen.recordAction({ type: "type", text });
+                await page.keyboard.type(text);
+                actionSummary = `Type text '${text}'`;
+                actionCode = await codegen.getCodeForLastAction();
                 break;
             }
             case "move": {
                 const { x, y } = action;
                 console.log(`Action: mouse move to (${x}, ${y})`);
+                await codegen.recordAction({ type: "move", x, y });
                 await page.mouse.move(x, y);
                 actionSummary = `Mouse move to (${x}, ${y})`;
-                actionCode = await codegen.codeForAction({ type: "move", x, y });
+                actionCode = await codegen.getCodeForLastAction();
                 break;
             }
             case "drag": {
@@ -102,32 +123,35 @@ async function executeModelAction(page, action, codegen) {
                 if (!path || path.length === 0) {
                     break;
                 }
+                await codegen.recordAction({ type: "drag", path });
                 await page.mouse.move(path[0].x, path[0].y);
                 await page.mouse.down();
                 for (let i = 1; i < path.length; i++) {
                     await page.mouse.move(path[i].x, path[i].y);
                 }
                 await page.mouse.up();
-                actionCode = await codegen.codeForAction({ type: "drag", path });
+                actionCode = await codegen.getCodeForLastAction();
                 break;
             }
             case "scroll": {
                 const { x, y, scroll_x, scroll_y } = action;
                 console.log(`Action: scroll at (${x}, ${y}) with offsets (scroll_x=${scroll_x}, scroll_y=${scroll_y})`);
-                await page.mouse.move(x, y);
-                await page.evaluate(`window.scrollBy(${scroll_x}, ${scroll_y})`);
-                actionSummary = `Scroll at (${x}, ${y}) with offsets (scroll_x=${scroll_x}, scroll_y=${scroll_y})`;
-                actionCode = await codegen.codeForAction({
+                await codegen.recordAction({
                     type: "scroll",
                     x,
                     y,
                     scroll_x,
                     scroll_y,
                 });
+                await page.mouse.move(x, y);
+                await page.evaluate(`window.scrollBy(${scroll_x}, ${scroll_y})`);
+                actionSummary = `Scroll at (${x}, ${y}) with offsets (scroll_x=${scroll_x}, scroll_y=${scroll_y})`;
+                actionCode = await codegen.getCodeForLastAction();
                 break;
             }
             case "keypress": {
                 const { keys } = action;
+                await codegen.recordAction({ type: "keypress", keys });
                 const mappedKeys = keys.map((k) => {
                     return CUA_KEY_TO_PLAYWRIGHT_KEY[k.toLowerCase()] || k;
                 });
@@ -135,22 +159,15 @@ async function executeModelAction(page, action, codegen) {
                 console.log(`Action: keypress for keys ${keys} -> '${mappedKey}'`);
                 await page.keyboard.press(mappedKey);
                 actionSummary = `Keypress for keys ${keys} (mapped to '${mappedKey}' for Playwright)`;
-                actionCode = await codegen.codeForAction({ type: "keypress", keys });
-                break;
-            }
-            case "type": {
-                const { text } = action;
-                console.log(`Action: type text '${text}'`);
-                await page.keyboard.type(text);
-                actionSummary = `Type text '${text}'`;
-                actionCode = await codegen.codeForAction({ type: "type", text });
+                actionCode = await codegen.getCodeForLastAction();
                 break;
             }
             case "wait": {
                 console.log(`Action: wait`);
+                await codegen.recordAction({ type: "wait" });
                 await page.waitForTimeout(2000);
                 actionSummary = `Wait for 2 seconds`;
-                actionCode = await codegen.codeForAction({ type: "wait" });
+                actionCode = await codegen.getCodeForLastAction();
                 break;
             }
             case "screenshot": {

package/dist/agent/cua/pw-codegen/element-from-point.d.ts

@@ -2,7 +2,9 @@ import type { Page } from "playwright";
 import { Action, BasePlaywrightCodegen } from "./types";
 export declare class ElementFromPointCodegen implements BasePlaywrightCodegen {
     private page;
+    private codeForLastAction;
     initialize(page: Page): Promise<void>;
-    codeForAction(action: Action): Promise<string>;
+    recordAction(action: Action): Promise<void>;
+    getCodeForLastAction(): Promise<string>;
 }
 //# sourceMappingURL=element-from-point.d.ts.map

package/dist/agent/cua/pw-codegen/element-from-point.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"element-from-point.d.ts","sourceRoot":"","sources":["../../../../src/agent/cua/pw-codegen/element-from-point.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAGvC,OAAO,EAAE,MAAM,EAAE,qBAAqB,EAAE,MAAM,SAAS,CAAC;AA4DxD,qBAAa,uBAAwB,YAAW,qBAAqB;IACnE,OAAO,CAAC,IAAI,CAAmB;IACzB,UAAU,CAAC,IAAI,EAAE,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;IAKrC,aAAa,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;CA0DrD"}
+{"version":3,"file":"element-from-point.d.ts","sourceRoot":"","sources":["../../../../src/agent/cua/pw-codegen/element-from-point.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAGvC,OAAO,EAAE,MAAM,EAAE,qBAAqB,EAAE,MAAM,SAAS,CAAC;AA4DxD,qBAAa,uBAAwB,YAAW,qBAAqB;IACnE,OAAO,CAAC,IAAI,CAAmB;IAC/B,OAAO,CAAC,iBAAiB,CAAqB;IAExC,UAAU,CAAC,IAAI,EAAE,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;IAKrC,YAAY,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IA4D3C,oBAAoB,IAAI,OAAO,CAAC,MAAM,CAAC;CAG9C"}

package/dist/agent/cua/pw-codegen/element-from-point.js

@@ -51,35 +51,38 @@ async function getLocatorForFill(page) {
 }
 class ElementFromPointCodegen {
     page;
+    codeForLastAction;
     async initialize(page) {
         this.page = page;
         await (0, utils_1.injectPwLocatorGenerator)(page);
     }
-    async codeForAction(action) {
+    async recordAction(action) {
         try {
             if (action.type === "goto") {
-                return `await page.goto("${action.url}");\n`;
+                this.codeForLastAction = `await page.goto("${action.url}");\n`;
             }
             if (action.type === "click") {
                 const { x, y } = action;
                 if (!this.page) {
                     throw new Error("Page is not initialized");
                 }
+                // TODO: Need execption handling
                 const locator = await getLocatorForClick(this.page, { x, y });
-                return `await page.${locator}.click();\n`;
+                this.codeForLastAction = `await page.${locator}.click();\n`;
             }
             if (action.type === "type") {
                 if (!this.page) {
                     throw new Error("Page is not initialized");
                 }
+                // TODO: Need execption handling
                 const locator = await getLocatorForFill(this.page);
-                return `await page.${locator}.fill("${action.text}");\n`;
+                this.codeForLastAction = `await page.${locator}.fill("${action.text}");\n`;
             }
             if (action.type === "back") {
-                return `await page.goBack();\n`;
+                this.codeForLastAction = `await page.goBack();\n`;
             }
             if (action.type === "forward") {
-                return `await page.goForward();\n`;
+                this.codeForLastAction = `await page.goForward();\n`;
             }
             if (action.type === "doubleclick") {
                 const { x, y } = action;
@@ -87,32 +90,34 @@ class ElementFromPointCodegen {
                     throw new Error("Page is not initialized");
                 }
                 const locator = await getLocatorForClick(this.page, { x, y });
-                return `await page.${locator}.dblclick();\n`;
+                this.codeForLastAction = `await page.${locator}.dblclick();\n`;
             }
             if (action.type === "move") {
                 const { x, y } = action;
-                return `await page.mouse.move(${x}, ${y});\n`;
+                this.codeForLastAction = `await page.mouse.move(${x}, ${y});\n`;
             }
             if (action.type === "drag") {
                 // const { path } = action;
                 // TODO: This needs to be fixed
-                return `No code generated: drag action not supported by code generation`;
+                this.codeForLastAction = `No code generated: drag action not supported by code generation`;
             }
             if (action.type === "scroll") {
-                return `No code to generate: will rely on Playwright's ability to auto-wait`;
+                this.codeForLastAction = `No code to generate: will rely on Playwright's ability to auto-wait`;
             }
             if (action.type === "keypress") {
                 const { keys } = action;
-                return `await page.keyboard.press("${keys.join("+")}");\n`;
+                this.codeForLastAction = `await page.keyboard.press("${keys.join("+")}");\n`;
             }
             if (action.type === "wait") {
-                return `No code to generate: will rely on Playwright's ability to auto-wait`;
+                this.codeForLastAction = `No code to generate: will rely on Playwright's ability to auto-wait`;
             }
-            throw new Error(`Unsupported action: ${JSON.stringify(action)}`);
         }
         catch (e) {
-            return `Error generating code for action: ${e.message || e.toString()}`;
+            this.codeForLastAction = `Error generating code for action: ${e.message || e.toString()}`;
         }
     }
+    async getCodeForLastAction() {
+        return this.codeForLastAction || "No code generated for the last action";
+    }
 }
 exports.ElementFromPointCodegen = ElementFromPointCodegen;

package/dist/agent/cua/pw-codegen/types.d.ts

@@ -41,6 +41,7 @@ export type Action = {
 };
 export interface BasePlaywrightCodegen {
     initialize(page: Page): Promise<void>;
-    codeForAction(action: Action): Promise<string>;
+    recordAction(action: Action): Promise<void>;
+    getCodeForLastAction(): Promise<string>;
 }
 //# sourceMappingURL=types.d.ts.map

package/dist/agent/cua/pw-codegen/types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../../src/agent/cua/pw-codegen/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAEvC,MAAM,MAAM,MAAM,GACd;IACE,IAAI,EAAE,OAAO,CAAC;IACd,CAAC,EAAE,MAAM,CAAC;IACV,CAAC,EAAE,MAAM,CAAC;CACX,GACD;IACE,IAAI,EAAE,aAAa,CAAC;IACpB,CAAC,EAAE,MAAM,CAAC;IACV,CAAC,EAAE,MAAM,CAAC;CACX,GACD;IACE,IAAI,EAAE,MAAM,CAAC;CACd,GACD;IACE,IAAI,EAAE,SAAS,CAAC;CACjB,GACD;IACE,IAAI,EAAE,MAAM,CAAC;IACb,CAAC,EAAE,MAAM,CAAC;IACV,CAAC,EAAE,MAAM,CAAC;CACX,GACD;IACE,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE;QAAE,CAAC,EAAE,MAAM,CAAC;QAAC,CAAC,EAAE,MAAM,CAAA;KAAE,EAAE,CAAC;CAClC,GACD;IACE,IAAI,EAAE,QAAQ,CAAC;IACf,CAAC,EAAE,MAAM,CAAC;IACV,CAAC,EAAE,MAAM,CAAC;IACV,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;CAClB,GACD;IACE,IAAI,EAAE,UAAU,CAAC;IACjB,IAAI,EAAE,MAAM,EAAE,CAAC;CAChB,GACD;IACE,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;CACd,GACD;IACE,IAAI,EAAE,MAAM,CAAC;IACb,GAAG,EAAE,MAAM,CAAC;CACb,GACD;IACE,IAAI,EAAE,MAAM,CAAC;CACd,CAAC;AAEN,MAAM,WAAW,qBAAqB;IACpC,UAAU,CAAC,IAAI,EAAE,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACtC,aAAa,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;CAChD"}
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../../src/agent/cua/pw-codegen/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAEvC,MAAM,MAAM,MAAM,GACd;IAEE,IAAI,EAAE,OAAO,CAAC;IACd,CAAC,EAAE,MAAM,CAAC;IACV,CAAC,EAAE,MAAM,CAAC;CACX,GACD;IACE,IAAI,EAAE,aAAa,CAAC;IACpB,CAAC,EAAE,MAAM,CAAC;IACV,CAAC,EAAE,MAAM,CAAC;CACX,GACD;IACE,IAAI,EAAE,MAAM,CAAC;CACd,GACD;IACE,IAAI,EAAE,SAAS,CAAC;CACjB,GACD;IACE,IAAI,EAAE,MAAM,CAAC;IACb,CAAC,EAAE,MAAM,CAAC;IACV,CAAC,EAAE,MAAM,CAAC;CACX,GACD;IACE,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE;QAAE,CAAC,EAAE,MAAM,CAAC;QAAC,CAAC,EAAE,MAAM,CAAA;KAAE,EAAE,CAAC;CAClC,GACD;IACE,IAAI,EAAE,QAAQ,CAAC;IACf,CAAC,EAAE,MAAM,CAAC;IACV,CAAC,EAAE,MAAM,CAAC;IACV,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;CAClB,GACD;IACE,IAAI,EAAE,UAAU,CAAC;IACjB,IAAI,EAAE,MAAM,EAAE,CAAC;CAChB,GACD;IACE,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;CACd,GACD;IACE,IAAI,EAAE,MAAM,CAAC;IACb,GAAG,EAAE,MAAM,CAAC;CACb,GACD;IACE,IAAI,EAAE,MAAM,CAAC;CACd,CAAC;AAEN,MAAM,WAAW,qBAAqB;IACpC,UAAU,CAAC,IAAI,EAAE,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACtC,YAAY,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC5C,oBAAoB,IAAI,OAAO,CAAC,MAAM,CAAC,CAAC;CACzC"}

package/dist/artifacts/index.d.ts

@@ -40,4 +40,13 @@ export declare function isArtifactCollectionEnabled(): string | undefined;
  * // Throws: "Invalid path: relative/path/test.png..."
  */
 export declare function collectArtifacts(inputs: ArtifactInput[], repoDir: string, toolCallId: string): Promise<Artifact[]>;
+export declare class UploadArtifactsQueue {
+    private toolCallId;
+    private baseRepoPath;
+    private artifactResults;
+    private uploadPromise;
+    constructor(baseRepoPath: string, toolCallId: string);
+    addTask(artifacts: ArtifactInput[]): Promise<void>;
+    waitForCompletion(): Promise<Artifact[]>;
+}
 //# sourceMappingURL=index.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/artifacts/index.ts"],"names":[],"mappings":"AAIA,OAAO,EACL,QAAQ,EACR,aAAa,EAGd,MAAM,4BAA4B,CAAC;AAkBpC,wBAAgB,2BAA2B,uBAM1C;AA6HD;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAsCG;AACH,wBAAsB,gBAAgB,CACpC,MAAM,EAAE,aAAa,EAAE,EACvB,OAAO,EAAE,MAAM,EACf,UAAU,EAAE,MAAM,GACjB,OAAO,CAAC,QAAQ,EAAE,CAAC,CA6DrB"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/artifacts/index.ts"],"names":[],"mappings":"AAIA,OAAO,EACL,QAAQ,EACR,aAAa,EAGd,MAAM,4BAA4B,CAAC;AAkBpC,wBAAgB,2BAA2B,uBAM1C;AA6HD;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAsCG;AACH,wBAAsB,gBAAgB,CACpC,MAAM,EAAE,aAAa,EAAE,EACvB,OAAO,EAAE,MAAM,EACf,UAAU,EAAE,MAAM,GACjB,OAAO,CAAC,QAAQ,EAAE,CAAC,CA6DrB;AAED,qBAAa,oBAAoB;IAC/B,OAAO,CAAC,UAAU,CAAS;IAC3B,OAAO,CAAC,YAAY,CAAS;IAC7B,OAAO,CAAC,eAAe,CAAkB;IACzC,OAAO,CAAC,aAAa,CAA8B;gBAEvC,YAAY,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM;IAKvC,OAAO,CAAC,SAAS,EAAE,aAAa,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAelD,iBAAiB,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;CAMtD"}

package/dist/artifacts/index.js

@@ -3,6 +3,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
+exports.UploadArtifactsQueue = void 0;
 exports.isArtifactCollectionEnabled = isArtifactCollectionEnabled;
 exports.collectArtifacts = collectArtifacts;
 const r2_uploader_1 = require("@empiricalrun/r2-uploader");
@@ -207,3 +208,30 @@ async function collectArtifacts(inputs, repoDir, toolCallId) {
         return [];
     }
 }
+class UploadArtifactsQueue {
+    toolCallId;
+    baseRepoPath;
+    artifactResults = [];
+    uploadPromise = null;
+    constructor(baseRepoPath, toolCallId) {
+        this.baseRepoPath = baseRepoPath;
+        this.toolCallId = toolCallId;
+    }
+    async addTask(artifacts) {
+        this.uploadPromise = collectArtifacts(artifacts, this.baseRepoPath, this.toolCallId)
+            .then((results) => {
+            this.artifactResults.push(...results);
+        })
+            .catch((error) => {
+            console.error("Error processing artifact upload task:", error);
+            throw error;
+        });
+    }
+    async waitForCompletion() {
+        if (this.uploadPromise) {
+            await this.uploadPromise;
+        }
+        return this.artifactResults;
+    }
+}
+exports.UploadArtifactsQueue = UploadArtifactsQueue;

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/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;AAiBhC,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;CAmDnE"}
+{"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"}

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

@@ -79,6 +79,13 @@ class ToolCallService {
                 name: `tool: ${toolCall.name}`,
                 input: toolCall.input,
             });
+            const uploadArtifactsQueue = new artifacts_1.UploadArtifactsQueue(this.repoPath, toolCall.id);
+            const collectArtifactsFn = (artifactsInput) => {
+                uploadArtifactsQueue.addTask(artifactsInput).catch((error) => {
+                    console.error("Error collecting artifacts:", error);
+                });
+                return;
+            };
             const toolExecutor = this.toolExecutors[toolCall.name];
             if (!toolExecutor) {
                 const errorResult = {
@@ -90,9 +97,19 @@ class ToolCallService {
                 span?.end({ output: errorResult });
                 continue;
             }
+            if (!this.apiKey) {
+                throw new Error("API key is required for tool execution");
+            }
             try {
-                const result = await toolExecutor(toolCall.input, this.repoPath, this.apiKey, this.trace, artifacts_1.collectArtifacts);
-                toolResults.push(result);
+                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 });
             }
             catch (error) {
@@ -105,7 +122,11 @@ class ToolCallService {
                 span?.end({ output: errorResult });
             }
         }
-        await (0, checkpoint_1.createCheckpoint)(toolCalls, this.branchName);
+        await (0, checkpoint_1.createCheckpoint)({
+            toolCalls,
+            branchName: this.branchName,
+            repoPath: this.repoPath,
+        });
         executeSpan?.end({ output: { toolResults } });
         return toolResults;
     }

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,IA2EnC,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,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,IA0FnC,CAAC"}