@empiricalrun/test-gen 0.55.0 → 0.56.1

package/dist/bin/index.js

@@ -233,8 +233,10 @@ async function main() {
         generationId: testGenConfig.options?.metadata.generationId,
         projectRepoName: testGenConfig.options?.metadata.projectRepoName,
     });
-    // Download the build if repo has a download script
-    await (0, test_build_1.downloadBuild)(testGenConfig.build || {});
+    if (testGenConfig.build?.url) {
+        // Download the build if repo has a download script
+        await (0, test_build_1.downloadBuild)(testGenConfig.build.url);
+    }
     if (completedOptions.useChat) {
         await runChatAgent({
             chatSessionId: completedOptions.chatSessionId,

package/dist/bin/utils/context.js

@@ -4,14 +4,14 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.fetchAppKnowledge = exports.generateTxtForRepository = exports.createRepoEditFilter = exports.contextForGeneration = exports.createGitIgnoreFileFilter = void 0;
-const fs_extra_1 = __importDefault(require("fs-extra"));
+const fs_1 = __importDefault(require("fs"));
 const ignore_1 = __importDefault(require("ignore"));
-const fs_1 = require("./fs");
+const fs_2 = require("./fs");
 async function createGitIgnoreFileFilter() {
     const ignoreFn = (0, ignore_1.default)();
-    if (fs_extra_1.default.existsSync(".gitignore")) {
+    if (fs_1.default.existsSync(".gitignore")) {
         // Not checking for nested gitignore
-        let gitignore = (await fs_extra_1.default.readFile(".gitignore")).toString();
+        let gitignore = fs_1.default.readFileSync(".gitignore").toString();
         gitignore = `
 ${gitignore}
 .git/
@@ -27,10 +27,10 @@ exports.createGitIgnoreFileFilter = createGitIgnoreFileFilter;
 async function contextForGeneration(file) {
     const filter = await createGitIgnoreFileFilter();
     return {
-        codePrompt: await (0, fs_1.generatePromptFromDirectory)("./tests", filter),
-        pomPrompt: await (0, fs_1.generatePromptFromDirectory)("./pages", filter),
-        nonSpecFilePrompt: await (0, fs_1.generatePromptFromNonSpecFiles)("./tests", filter),
-        testFileContent: file ? await fs_extra_1.default.readFile(file, "utf-8") : "",
+        codePrompt: await (0, fs_2.generatePromptFromDirectory)("./tests", filter),
+        pomPrompt: await (0, fs_2.generatePromptFromDirectory)("./pages", filter),
+        nonSpecFilePrompt: await (0, fs_2.generatePromptFromNonSpecFiles)("./tests", filter),
+        testFileContent: file ? fs_1.default.readFileSync(file, "utf-8") : "",
     };
 }
 exports.contextForGeneration = contextForGeneration;
@@ -39,14 +39,14 @@ async function createRepoEditFilter() {
     const gitIgnoreFilter = await createGitIgnoreFileFilter();
     return (filePath) => {
         return (gitIgnoreFilter(filePath) &&
-            (fs_extra_1.default.statSync(filePath).isDirectory() || allowedExtensions.test(filePath)));
+            (fs_1.default.statSync(filePath).isDirectory() || allowedExtensions.test(filePath)));
     };
 }
 exports.createRepoEditFilter = createRepoEditFilter;
 async function generateTxtForRepository() {
     const filter = await createRepoEditFilter();
     return {
-        prompt: await (0, fs_1.generatePromptFromDirectory)(".", filter),
+        prompt: await (0, fs_2.generatePromptFromDirectory)(".", filter),
     };
 }
 exports.generateTxtForRepository = generateTxtForRepository;
@@ -54,7 +54,7 @@ async function fetchAppKnowledge() {
     let fileExists = true;
     const appKnowledgePath = "./app_knowledge.md";
     try {
-        await fs_extra_1.default.access(appKnowledgePath);
+        fs_1.default.accessSync(appKnowledgePath);
     }
     catch (e) {
         fileExists = false;
@@ -62,6 +62,6 @@ async function fetchAppKnowledge() {
     if (!fileExists) {
         return "";
     }
-    return await fs_extra_1.default.readFile(appKnowledgePath, "utf-8");
+    return fs_1.default.readFileSync(appKnowledgePath, "utf-8");
 }
 exports.fetchAppKnowledge = fetchAppKnowledge;

package/dist/bin/utils/fs/index.js

@@ -4,19 +4,19 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.generatePromptFromNonSpecFiles = exports.generatePromptFromDirectory = exports.convertFileContentsToString = exports.readFilesInDirectory = void 0;
-const fs_extra_1 = __importDefault(require("fs-extra"));
+const fs_1 = __importDefault(require("fs"));
 const path_1 = __importDefault(require("path"));
 async function readFilesInDirectory(dir = "", filterFunc) {
     let files = [];
-    const items = await fs_extra_1.default.readdir(dir);
+    const items = fs_1.default.readdirSync(dir);
     const filteredItems = items.map((i) => path_1.default.join(dir, i)).filter(filterFunc);
     for (const item of filteredItems) {
-        const stat = await fs_extra_1.default.stat(item);
+        const stat = fs_1.default.statSync(item);
         if (stat.isDirectory()) {
             files = files.concat(await readFilesInDirectory(item, filterFunc));
         }
         else if (stat.isFile()) {
-            const content = await fs_extra_1.default.readFile(item, "utf-8");
+            const content = fs_1.default.readFileSync(item, "utf-8");
             files.push({ filePath: item, content });
         }
     }

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

@@ -6,7 +6,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.isSyntaxValid = exports.getVariableDeclarationsFromCode = exports.buildTestNamePrompt = exports.isTestPresent = exports.appendScopeToCreateTest = exports.addUserContextFixture = exports.importAllExportsStmtFromFilePaths = exports.injectCodeSnippetBySuiteChain = exports.replaceCreateTestWithNewCode = exports.getPageVariableNameFromCreateTest = exports.getFixtureImportPath = exports.removeTestOnly = exports.addNewImport = exports.formatCode = exports.lintErrors = exports.stripAndPrependImports = exports.validateTypescript = exports.appendToTestBlock = exports.findFirstSerialDescribeBlock = exports.hasTopLevelDescribeConfigureWithSerialMode = exports.hasTestBlock = exports.getTypescriptTestBlock = exports.getTestModuleAliasFromSourceFile = void 0;
 const parser_1 = require("@babel/parser");
 const eslint_1 = require("eslint");
-const fs_extra_1 = __importDefault(require("fs-extra"));
+const fs_1 = __importDefault(require("fs"));
 const path_1 = __importDefault(require("path"));
 const prettier_1 = __importDefault(require("prettier"));
 const ts_morph_1 = require("ts-morph");
@@ -73,12 +73,12 @@ function getTypescriptTestBlock({ scenarioName, suites, content, }) {
 }
 exports.getTypescriptTestBlock = getTypescriptTestBlock;
 function hasTestBlock({ testName, testSuites, filePath, }) {
-    if (!fs_extra_1.default.existsSync(filePath)) {
+    if (!fs_1.default.existsSync(filePath)) {
         return false;
     }
     const { testBlock } = getTypescriptTestBlock({
         scenarioName: testName,
-        content: fs_extra_1.default.readFileSync(filePath, "utf-8"),
+        content: fs_1.default.readFileSync(filePath, "utf-8"),
         suites: testSuites,
     });
     return Boolean(testBlock);
@@ -107,7 +107,7 @@ function getParentDescribeNames(node) {
 }
 async function hasTopLevelDescribeConfigureWithSerialMode(filePath) {
     const project = new ts_morph_1.Project();
-    const content = await fs_extra_1.default.readFile(filePath, "utf-8");
+    const content = fs_1.default.readFileSync(filePath, "utf-8");
     const sourceFile = project.createSourceFile("test.ts", content);
     const statements = sourceFile.getStatements();
     for (const statement of statements) {
@@ -188,7 +188,7 @@ exports.appendToTestBlock = appendToTestBlock;
 function validateTypescript(filePath) {
     // Create a compiler host to read files
     const compilerHost = typescript_1.default.createCompilerHost({});
-    compilerHost.readFile = (file) => fs_extra_1.default.readFileSync(file, "utf8");
+    compilerHost.readFile = (file) => fs_1.default.readFileSync(file, "utf8");
     // Create a program with a single source file
     const program = typescript_1.default.createProgram([filePath], {
         esModuleInterop: true,
@@ -245,12 +245,12 @@ async function lintErrors(filePath) {
     });
     const [result] = await eslint.lintFiles(filePath);
     if (result?.output) {
-        await fs_extra_1.default.writeFile(filePath, result.output);
+        fs_1.default.writeFileSync(filePath, result.output);
     }
 }
 exports.lintErrors = lintErrors;
 async function formatCode(filePath, trace) {
-    const fileContent = fs_extra_1.default.readFileSync(filePath, "utf8");
+    const fileContent = fs_1.default.readFileSync(filePath, "utf8");
     if (!fileContent) {
         trace?.span({
             name: "prettier-format-output",
@@ -264,7 +264,7 @@ async function formatCode(filePath, trace) {
         filepath: filePath,
     });
     trace?.span({ name: "prettier-format-output", output: formattedContent });
-    await fs_extra_1.default.writeFile(filePath, formattedContent);
+    fs_1.default.writeFileSync(filePath, formattedContent);
 }
 exports.formatCode = formatCode;
 function addNewImport(contents, modules, pkg) {
@@ -272,11 +272,11 @@ function addNewImport(contents, modules, pkg) {
 }
 exports.addNewImport = addNewImport;
 async function removeTestOnly(filePath) {
-    const contents = await fs_extra_1.default.readFile(filePath, "utf8");
+    const contents = fs_1.default.readFileSync(filePath, "utf8");
     const updatedContent = contents
         .replace("test.only(", "test(")
         .replace("test.describe.only(", "test.describe(");
-    return fs_extra_1.default.writeFile(filePath, updatedContent);
+    fs_1.default.writeFileSync(filePath, updatedContent);
 }
 exports.removeTestOnly = removeTestOnly;
 function getFixtureImportPath(filePath) {
@@ -296,7 +296,7 @@ function getFixtureImportPath(filePath) {
 }
 exports.getFixtureImportPath = getFixtureImportPath;
 async function getPageVariableNameFromCreateTest(filePath) {
-    const contents = await fs_extra_1.default.readFile(filePath, "utf-8");
+    const contents = fs_1.default.readFileSync(filePath, "utf-8");
     const project = new ts_morph_1.Project();
     const sourceFile = project.createSourceFile("test.ts", contents);
     const createTestNode = sourceFile.getFirstDescendant((node) => !!(node.isKind(ts_morph_1.SyntaxKind.CallExpression) &&
@@ -418,7 +418,7 @@ const importAllExportsStmtFromFilePaths = async (repoDir, filePaths, testFilePat
         if (!importPath.startsWith(".")) {
             importPath = "./" + importPath;
         }
-        const file = await fs_extra_1.default.readFile(fullPath, "utf-8");
+        const file = fs_1.default.readFileSync(fullPath, "utf-8");
         const project = new ts_morph_1.Project();
         const sourceFile = project.createSourceFile("index.ts", file);
         const exportedFunctions = sourceFile
@@ -440,7 +440,7 @@ async function addUserContextFixture({ scenarioName, filePath, suites, }) {
     // TODO: remove this check when we support user context across repos
     let fixtureContent = "";
     try {
-        fixtureContent = await fs_extra_1.default.readFile("./tests/fixtures.ts", "utf-8");
+        fixtureContent = fs_1.default.readFileSync("./tests/fixtures.ts", "utf-8");
     }
     catch (e) {
         // do nothing
@@ -448,7 +448,7 @@ async function addUserContextFixture({ scenarioName, filePath, suites, }) {
     if (!fixtureContent || !fixtureContent.includes("userContext")) {
         return;
     }
-    const fileContent = await fs_extra_1.default.readFile(filePath, "utf-8");
+    const fileContent = fs_1.default.readFileSync(filePath, "utf-8");
     const { testNode } = getTypescriptTestBlock({
         scenarioName,
         content: fileContent,
@@ -469,14 +469,14 @@ async function addUserContextFixture({ scenarioName, filePath, suites, }) {
                     .replace(",}", "}") // Remove any trailing commas
                     .replace("}", ", userContext }");
                 destructuringParam.replaceWithText(updatedImport);
-                await fs_extra_1.default.writeFile(filePath, fileContent.replace(currentBlock, testNode.getText()), "utf-8");
+                fs_1.default.writeFileSync(filePath, fileContent.replace(currentBlock, testNode.getText()), "utf-8");
             }
         }
     }
 }
 exports.addUserContextFixture = addUserContextFixture;
 async function appendScopeToCreateTest(filePath, scopeVariables) {
-    const contents = await fs_extra_1.default.readFile(filePath, "utf-8");
+    const contents = fs_1.default.readFileSync(filePath, "utf-8");
     const project = new ts_morph_1.Project();
     const sourceFile = project.createSourceFile("test.ts", contents);
     const createTestNode = sourceFile
@@ -491,17 +491,17 @@ async function appendScopeToCreateTest(filePath, scopeVariables) {
     const args = createTestNode.getArguments();
     const scopeArgStr = `{ ${scopeVariables.join(", ")} }`;
     createTestNode.insertArgument(args.length, scopeArgStr);
-    await fs_extra_1.default.writeFile(filePath, sourceFile.getFullText());
+    fs_1.default.writeFileSync(filePath, sourceFile.getFullText());
 }
 exports.appendScopeToCreateTest = appendScopeToCreateTest;
 function isTestPresent({ specPath, testCase, }) {
     let isTestPresent = true;
-    isTestPresent = fs_extra_1.default.existsSync(specPath);
+    isTestPresent = fs_1.default.existsSync(specPath);
     if (isTestPresent) {
         const { testBlock } = getTypescriptTestBlock({
             suites: testCase.suites,
             scenarioName: testCase.name,
-            content: fs_extra_1.default.readFileSync(specPath, "utf-8"),
+            content: fs_1.default.readFileSync(specPath, "utf-8"),
         });
         isTestPresent = Boolean(testBlock);
     }

package/dist/reporter/index.js

@@ -5,7 +5,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.TestGenUpdatesReporter = exports.setReporterConfig = exports.getReporter = void 0;
 const r2_uploader_1 = require("@empiricalrun/r2-uploader");
-const fs_extra_1 = __importDefault(require("fs-extra"));
+const fs_1 = __importDefault(require("fs"));
 const path_1 = __importDefault(require("path"));
 const logger_1 = require("../bin/logger");
 const uploader_1 = require("../uploader");
@@ -93,10 +93,10 @@ class TestGenUpdatesReporter {
         }
         try {
             // upload current screenshot to r2 and report it to reporter
-            if (!fs_extra_1.default.existsSync(path_1.default.join(this.repoDir, "gen-assets"))) {
-                await fs_extra_1.default.mkdir(path_1.default.join(this.repoDir, "gen-assets"));
+            if (!fs_1.default.existsSync(path_1.default.join(this.repoDir, "gen-assets"))) {
+                fs_1.default.mkdirSync(path_1.default.join(this.repoDir, "gen-assets"));
             }
-            await fs_extra_1.default.writeFile(path_1.default.join(this.repoDir, "gen-assets", `current-view-${Date.now()}.png`), buffer);
+            fs_1.default.writeFileSync(path_1.default.join(this.repoDir, "gen-assets", `current-view-${Date.now()}.png`), buffer);
             const uploadDir = (0, uploader_1.getUploadPathForRun)(reporterConfig?.projectRepoName);
             const files = await (0, r2_uploader_1.uploadDirectory)({
                 sourceDir: path_1.default.join(this.repoDir, "gen-assets"),
@@ -110,7 +110,7 @@ class TestGenUpdatesReporter {
                 type: "current-snapshot",
                 message: JSON.stringify({ type: "current-view", url }),
             }));
-            await fs_extra_1.default.rmdir(path_1.default.join(this.repoDir, "gen-assets"), {
+            fs_1.default.rmdirSync(path_1.default.join(this.repoDir, "gen-assets"), {
                 recursive: true,
             });
         }

package/dist/test-build/index.d.ts

@@ -1,10 +1,3 @@
-import { Build } from "@empiricalrun/shared-types";
-/**
- * method to download the build from the URL provided in the build object
- * this is only used in cases like chrome extension as of yet.
- * @export
- * @param {Build} build
- * @return {*}  {Promise<void>}
- */
-export declare function downloadBuild(build: Build): Promise<void>;
+export declare function hasDownloadScript(): Promise<boolean>;
+export declare function downloadBuild(buildUrl: string): Promise<void>;
 //# sourceMappingURL=index.d.ts.map

package/dist/test-build/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/test-build/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,MAAM,4BAA4B,CAAC;AAUnD;;;;;;GAMG;AACH,wBAAsB,aAAa,CAAC,KAAK,EAAE,KAAK,GAAG,OAAO,CAAC,IAAI,CAAC,CAY/D"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/test-build/index.ts"],"names":[],"mappings":"AAeA,wBAAsB,iBAAiB,IAAI,OAAO,CAAC,OAAO,CAAC,CAG1D;AAED,wBAAsB,aAAa,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAUnE"}

package/dist/test-build/index.js

@@ -3,26 +3,27 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.downloadBuild = void 0;
-const fs_extra_1 = __importDefault(require("fs-extra"));
+exports.downloadBuild = exports.hasDownloadScript = void 0;
+const fs_1 = __importDefault(require("fs"));
 const logger_1 = require("../bin/logger");
 const exec_1 = require("../utils/exec");
-/**
- * method to download the build from the URL provided in the build object
- * this is only used in cases like chrome extension as of yet.
- * @export
- * @param {Build} build
- * @return {*}  {Promise<void>}
- */
-async function downloadBuild(build) {
-    const logger = new logger_1.CustomLogger({ useReporter: false });
+async function getPackageJSON() {
     const packageJSONPath = "package.json";
-    const packageJsonStr = await fs_extra_1.default.readFile(packageJSONPath, "utf-8");
-    const packageJSONData = JSON.parse(packageJsonStr);
-    const buildDownloadScript = packageJSONData.scripts["download"];
-    if (buildDownloadScript && build?.url) {
-        logger.log(`Downloading build from ${build.url}`);
-        await (0, exec_1.cmd)(`npm run download ${build.url}`.split(" "), {
+    const packageJsonStr = fs_1.default.readFileSync(packageJSONPath, "utf-8");
+    return JSON.parse(packageJsonStr);
+}
+async function hasDownloadScript() {
+    const packageJSON = await getPackageJSON();
+    return !!packageJSON.scripts["download"];
+}
+exports.hasDownloadScript = hasDownloadScript;
+async function downloadBuild(buildUrl) {
+    const logger = new logger_1.CustomLogger({ useReporter: false });
+    const packageJSON = await getPackageJSON();
+    const buildDownloadScript = packageJSON.scripts["download"];
+    if (buildDownloadScript && buildUrl) {
+        logger.log(`Downloading build from ${buildUrl}`);
+        await (0, exec_1.cmd)(`npm run download ${buildUrl}`.split(" "), {
             env: { ...Object(process.env) },
         });
     }

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

@@ -1,4 +1,4 @@
-import { Tool, ToolResult } from "@empiricalrun/llm/chat";
+import { PendingToolCall, Tool, ToolResult } from "@empiricalrun/llm/chat";
 import { SupportedChatModels } from "../agent/chat/types";
 export type { SupportedChatModels };
 type ToolExecutors = {
@@ -7,15 +7,12 @@ type ToolExecutors = {
 export declare class ToolCallService {
     tools: Tool[];
     toolExecutors: ToolExecutors;
-    constructor();
-    getTools(selectedModel: SupportedChatModels): Promise<{
+    chatSessionId: number | null;
+    selectedModel: SupportedChatModels;
+    constructor(chatSessionId: number | null, selectedModel: SupportedChatModels);
+    getTools(): Promise<{
         tools: Tool[];
     }>;
-    execute(payload: {
-        tool: {
-            name: string;
-            input: any;
-        };
-    }, isRemote: boolean): Promise<ToolResult>;
+    execute(toolCalls: PendingToolCall[], isRemote: boolean): Promise<ToolResult[]>;
 }
 //# sourceMappingURL=index.d.ts.map

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":"AACA,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AAE1D,OAAO,EAAE,mBAAmB,EAAE,MAAM,qBAAqB,CAAC;AAa1D,YAAY,EAAE,mBAAmB,EAAE,CAAC;AAEpC,KAAK,aAAa,GAAG;IACnB,CAAC,GAAG,EAAE,MAAM,GAAG,CAAC,KAAK,EAAE,GAAG,KAAK,OAAO,CAAC,UAAU,CAAC,CAAC;CACpD,CAAC;AAqBF,qBAAa,eAAe;IAC1B,KAAK,EAAE,IAAI,EAAE,CAAM;IACnB,aAAa,EAAE,aAAa,CAAM;;IAa5B,QAAQ,CAAC,aAAa,EAAE,mBAAmB;;;IAa3C,OAAO,CACX,OAAO,EAAE;QACP,IAAI,EAAE;YACJ,IAAI,EAAE,MAAM,CAAC;YACb,KAAK,EAAE,GAAG,CAAC;SACZ,CAAC;KACH,EACD,QAAQ,EAAE,OAAO,GAChB,OAAO,CAAC,UAAU,CAAC;CA8BvB"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/tool-call-service/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,eAAe,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AAE3E,OAAO,EAAE,mBAAmB,EAAE,MAAM,qBAAqB,CAAC;AAa1D,YAAY,EAAE,mBAAmB,EAAE,CAAC;AAEpC,KAAK,aAAa,GAAG;IACnB,CAAC,GAAG,EAAE,MAAM,GAAG,CAAC,KAAK,EAAE,GAAG,KAAK,OAAO,CAAC,UAAU,CAAC,CAAC;CACpD,CAAC;AAyBF,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;gBAGjC,aAAa,EAAE,MAAM,GAAG,IAAI,EAC5B,aAAa,EAAE,mBAAmB;IAgB9B,QAAQ;;;IAaR,OAAO,CACX,SAAS,EAAE,eAAe,EAAE,EAC5B,QAAQ,EAAE,OAAO,GAChB,OAAO,CAAC,UAAU,EAAE,CAAC;CAqCzB"}

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

@@ -4,13 +4,14 @@ exports.ToolCallService = void 0;
 const client_sqs_1 = require("@aws-sdk/client-sqs");
 const commit_and_create_pr_1 = require("../tools/commit-and-create-pr");
 const diagnosis_fetcher_1 = require("../tools/diagnosis-fetcher");
+const download_build_1 = require("../tools/download-build");
 const environment_crud_1 = require("../tools/environment-crud");
 const grep_1 = require("../tools/grep");
 const str_replace_editor_1 = require("../tools/str_replace_editor");
 const test_gen_browser_1 = require("../tools/test-gen-browser");
 const test_run_1 = require("../tools/test-run");
 const test_run_fetcher_1 = require("../tools/test-run-fetcher");
-async function sendToolRequestToRemoteQueue(toolName, input) {
+async function sendToolRequestToRemoteQueue(payload) {
     const sqs = new client_sqs_1.SQSClient({
         region: process.env.AWS_REGION,
         credentials: {
@@ -18,18 +19,22 @@ async function sendToolRequestToRemoteQueue(toolName, input) {
             secretAccessKey: process.env.AWS_SECRET_ACCESS_KEY,
         },
     });
-    const queueUrl = "https://sqs.us-east-1.amazonaws.com/381492172454/toolRequests.fifo";
+    const queueUrl = process.env.TOOL_REQUEST_QUEUE_URL;
     await sqs.send(new client_sqs_1.SendMessageCommand({
         QueueUrl: queueUrl,
-        MessageBody: JSON.stringify({ toolName, input }),
-        MessageGroupId: toolName, // tool request id
-        MessageDeduplicationId: toolName, // tool request id
+        MessageBody: JSON.stringify(payload),
+        MessageGroupId: payload.requestId,
+        MessageDeduplicationId: payload.requestId, // unique id for the tool request
     }));
 }
 class ToolCallService {
     tools = [];
     toolExecutors = {};
-    constructor() {
+    chatSessionId;
+    selectedModel;
+    constructor(chatSessionId, selectedModel) {
+        this.chatSessionId = chatSessionId;
+        this.selectedModel = selectedModel;
         this.tools = [
             grep_1.grepTool,
             test_run_1.runTestTool,
@@ -38,49 +43,58 @@ class ToolCallService {
             test_gen_browser_1.generateTestWithBrowserAgent,
             commit_and_create_pr_1.commitAndPushChangesTool,
             environment_crud_1.getEnvironmentTool,
+            download_build_1.downloadBuildTool,
         ];
     }
-    async getTools(selectedModel) {
-        if (!selectedModel.startsWith("claude")) {
+    async getTools() {
+        if (!this.selectedModel.startsWith("claude")) {
             this.tools.push(...str_replace_editor_1.textEditorTools);
         }
         this.tools.forEach((tool) => {
             this.toolExecutors[tool.schema.name] = tool.execute;
         });
-        if (selectedModel.startsWith("claude")) {
+        if (this.selectedModel.startsWith("claude")) {
             this.toolExecutors["str_replace_editor"] = str_replace_editor_1.strReplaceEditorExecutor;
         }
         return { tools: this.tools };
     }
-    async execute(payload, isRemote) {
-        const { tool } = payload;
-        const toolExecutor = this.toolExecutors[tool.name];
-        if (!toolExecutor) {
-            return {
-                isError: true,
-                result: `Invalid function/tool call: invalid_tool_call not found`,
-            };
+    async execute(toolCalls, isRemote) {
+        if (isRemote && this.chatSessionId) {
+            console.log("Executing tool remotely", toolCalls);
+            await sendToolRequestToRemoteQueue({
+                toolCalls,
+                requestId: crypto.randomUUID(),
+                chatSessionId: this.chatSessionId,
+                selectedModel: this.selectedModel,
+            });
+            return toolCalls.map(() => ({
+                isError: false,
+                result: `Tool request sent to remote queue to execute.`,
+            }));
         }
-        try {
-            if (isRemote) {
-                console.log("Executing tool remotely", tool.name, tool.input);
-                // push to sqs
-                await sendToolRequestToRemoteQueue(tool.name, tool.input);
-                // TODO: Need to stop the agent loop here
-                return {
-                    isError: false,
-                    result: `Tool request sent to remote queue to execute ${tool.name}.`,
-                };
+        else {
+            const toolResults = [];
+            for (const toolCall of toolCalls) {
+                const toolExecutor = this.toolExecutors[toolCall.name];
+                if (!toolExecutor) {
+                    toolResults.push({
+                        isError: true,
+                        result: `Invalid function/tool call: invalid_tool_call not found`,
+                    });
+                    continue;
+                }
+                try {
+                    const result = await toolExecutor(toolCall.input);
+                    toolResults.push(result);
+                }
+                catch (error) {
+                    toolResults.push({
+                        isError: true,
+                        result: error instanceof Error ? error.message : String(error),
+                    });
+                }
             }
-            else {
-                return await toolExecutor(tool.input);
-            }
-        }
-        catch (error) {
-            return {
-                isError: true,
-                result: error instanceof Error ? error.message : String(error),
-            };
+            return toolResults;
         }
     }
 }

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

@@ -0,0 +1,3 @@
+import type { Tool } from "@empiricalrun/llm/chat";
+export declare const downloadBuildTool: Tool;
+//# sourceMappingURL=download-build.d.ts.map

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

@@ -0,0 +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;AAKnD,eAAO,MAAM,iBAAiB,EAAE,IAkC/B,CAAC"}

package/dist/tools/download-build.js

@@ -0,0 +1,39 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.downloadBuildTool = void 0;
+const zod_1 = require("zod");
+const test_build_1 = require("../test-build");
+exports.downloadBuildTool = {
+    schema: {
+        name: "downloadBuild",
+        description: `Download a build from a build URL. If you do not have
+have a build URL, you can try getting the environment details with the getEnvironment tool.
+Environment details will include the build URL.`,
+        parameters: zod_1.z.object({
+            buildUrl: zod_1.z.string().describe("The URL of the build to download"),
+        }),
+    },
+    execute: async (input) => {
+        if (!(await (0, test_build_1.hasDownloadScript)())) {
+            return {
+                isError: true,
+                result: `This repo does not have a download script in package.json.
+You probably don't need to worry about this, since it means this repo does not have a build to download.`,
+            };
+        }
+        const { buildUrl } = input;
+        try {
+            await (0, test_build_1.downloadBuild)(buildUrl);
+            return {
+                isError: false,
+                result: "Build downloaded successfully",
+            };
+        }
+        catch (error) {
+            return {
+                isError: true,
+                result: `Failed to download build: ${error}`,
+            };
+        }
+    },
+};

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

@@ -1 +1 @@
-{"version":3,"file":"grep.d.ts","sourceRoot":"","sources":["../../src/tools/grep.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAc,MAAM,wBAAwB,CAAC;AAqB/D,eAAO,MAAM,QAAQ,EAAE,IA+CtB,CAAC"}
+{"version":3,"file":"grep.d.ts","sourceRoot":"","sources":["../../src/tools/grep.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAc,MAAM,wBAAwB,CAAC;AAmB/D,eAAO,MAAM,QAAQ,EAAE,IAgDtB,CAAC"}

package/dist/tools/grep.js

@@ -5,7 +5,6 @@ const child_process_1 = require("child_process");
 const util_1 = require("util");
 const zod_1 = require("zod");
 const repo_tree_1 = require("../utils/repo-tree");
-const execAsync = (0, util_1.promisify)(child_process_1.exec);
 const GrepInputSchema = zod_1.z.object({
     pattern: zod_1.z.string().describe("The pattern to search for"),
     directory: zod_1.z
@@ -40,6 +39,7 @@ exports.grepTool = {
                 // Using -n to show line numbers and removed -l to show actual matches
                 cmd = `find ${dir} ${excludeFind} -name "${input.filePattern}" -exec grep -rin "${input.pattern}" {} \\;`;
             }
+            const execAsync = (0, util_1.promisify)(child_process_1.exec);
             const { stdout, stderr } = await execAsync(cmd);
             if (stdout) {
                 return {

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":"AAAA,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;AAED;;;GAGG;AACH,wBAAgB,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,CAwC1D;AAMD;;;GAGG;AACH,wBAAsB,wBAAwB,CAC5C,KAAK,EAAE,eAAe,GACrB,OAAO,CAAC,UAAU,CAAC,CAwIrB;AA+FD,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":"AAAA,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;AAED;;;GAGG;AACH,wBAAgB,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,CAwC1D;AAMD;;;GAGG;AACH,wBAAsB,wBAAwB,CAC5C,KAAK,EAAE,eAAe,GACrB,OAAO,CAAC,UAAU,CAAC,CAmJrB;AA+FD,eAAO,MAAM,eAAe,EAAE,IAAI,EAKjC,CAAC"}