@empiricalrun/test-gen 0.72.0 → 0.73.1

package/dist/tools/upgrade-packages/utils.d.ts

@@ -1,3 +1,4 @@
+import { IDashboardAPIClient } from "@empiricalrun/shared-types";
 export declare function getLatestVersion(packageName: string): Promise<any>;
 export declare function parsePackageJsonDiff({ diff, }: {
     diff: string;
@@ -5,10 +6,10 @@ export declare function parsePackageJsonDiff({ diff, }: {
     old?: string;
     new?: string;
 }>;
-export declare function shouldMergePR({ repoName, prNumber, apiKey, }: {
+export declare function shouldMergePR({ repoName, prNumber, apiClient, }: {
     repoName: string;
     prNumber: number;
-    apiKey: string;
+    apiClient: IDashboardAPIClient;
 }): Promise<boolean>;
 export declare function upgradeAndStagePackage({ repoPath, pkgName, version, isDevDep, }: {
     repoPath: string;

package/dist/tools/upgrade-packages/utils.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../src/tools/upgrade-packages/utils.ts"],"names":[],"mappings":"AAoBA,wBAAsB,gBAAgB,CAAC,WAAW,EAAE,MAAM,gBAkBzD;AA8BD,wBAAgB,oBAAoB,CAAC,EACnC,IAAI,GACL,EAAE;IACD,IAAI,EAAE,MAAM,CAAC;CACd,GAAG,MAAM,CAAC,MAAM,EAAE;IAAE,GAAG,CAAC,EAAE,MAAM,CAAC;IAAC,GAAG,CAAC,EAAE,MAAM,CAAA;CAAE,CAAC,CAuBjD;AAED,wBAAsB,aAAa,CAAC,EAClC,QAAQ,EACR,QAAQ,EACR,MAAM,GACP,EAAE;IACD,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;CAChB,oBA8BA;AAED,wBAAsB,sBAAsB,CAAC,EAC3C,QAAQ,EACR,OAAO,EACP,OAAO,EACP,QAAQ,GACT,EAAE;IACD,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,OAAO,CAAC;CACnB;;;GAcA"}
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../src/tools/upgrade-packages/utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,mBAAmB,EAAE,MAAM,4BAA4B,CAAC;AAmBjE,wBAAsB,gBAAgB,CAAC,WAAW,EAAE,MAAM,gBAkBzD;AA4BD,wBAAgB,oBAAoB,CAAC,EACnC,IAAI,GACL,EAAE;IACD,IAAI,EAAE,MAAM,CAAC;CACd,GAAG,MAAM,CAAC,MAAM,EAAE;IAAE,GAAG,CAAC,EAAE,MAAM,CAAC;IAAC,GAAG,CAAC,EAAE,MAAM,CAAA;CAAE,CAAC,CAuBjD;AAED,wBAAsB,aAAa,CAAC,EAClC,QAAQ,EACR,QAAQ,EACR,SAAS,GACV,EAAE;IACD,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,EAAE,mBAAmB,CAAC;CAChC,oBA8BA;AAED,wBAAsB,sBAAsB,CAAC,EAC3C,QAAQ,EACR,OAAO,EACP,OAAO,EACP,QAAQ,GACT,EAAE;IACD,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,OAAO,CAAC;CACnB;;;GAcA"}

package/dist/tools/upgrade-packages/utils.js

@@ -5,7 +5,6 @@ exports.parsePackageJsonDiff = parsePackageJsonDiff;
 exports.shouldMergePR = shouldMergePR;
 exports.upgradeAndStagePackage = upgradeAndStagePackage;
 const child_process_1 = require("child_process");
-const utils_1 = require("../executor/utils");
 function isPatchUpdate(currentVersion, newVersion) {
     if (!currentVersion || !newVersion) {
         // If current is undefined, we might be installing a new package
@@ -35,12 +34,10 @@ async function getLatestVersion(packageName) {
         throw error;
     }
 }
-async function getGitDiffForPackageJson({ repoName, prNumber, apiKey, }) {
-    const url = `${utils_1.GITHUB_API_BASE}/${repoName}/pulls/${prNumber}/files`;
-    const files = await (0, utils_1.callGitHubProxy)({
+async function getGitDiffForPackageJson({ repoName, prNumber, apiClient, }) {
+    const files = await apiClient.callGitHubProxy({
         method: "GET",
-        url: url,
-        apiKey,
+        url: `/repos/empirical-run/${repoName}/pulls/${prNumber}/files`,
     });
     if (!files) {
         throw new Error("Failed to fetch PR files via proxy");
@@ -74,11 +71,11 @@ function parsePackageJsonDiff({ diff, }) {
     }
     return changes;
 }
-async function shouldMergePR({ repoName, prNumber, apiKey, }) {
+async function shouldMergePR({ repoName, prNumber, apiClient, }) {
     const gitDiff = await getGitDiffForPackageJson({
         repoName,
         prNumber,
-        apiKey,
+        apiClient,
     });
     if (!gitDiff) {
         console.log("No package.json changes found, skipping merge");

package/dist/tools/utils/queue.d.ts

@@ -0,0 +1,5 @@
+import { PendingToolCall } from "@empiricalrun/shared-types";
+export declare function appendSuffixToQueueUrl(queueUrl: string, suffix: string): string;
+export declare function replaceRegionInUrl(url: string, newRegion: string): string;
+export declare function getQueueUrl(toolCalls: PendingToolCall[], needsBrowser: boolean): string;
+//# sourceMappingURL=queue.d.ts.map

package/dist/tools/utils/queue.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"queue.d.ts","sourceRoot":"","sources":["../../../src/tools/utils/queue.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,4BAA4B,CAAC;AAE7D,wBAAgB,sBAAsB,CACpC,QAAQ,EAAE,MAAM,EAChB,MAAM,EAAE,MAAM,GACb,MAAM,CAKR;AAED,wBAAgB,kBAAkB,CAAC,GAAG,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,MAAM,CAOzE;AAkBD,wBAAgB,WAAW,CACzB,SAAS,EAAE,eAAe,EAAE,EAC5B,YAAY,EAAE,OAAO,GACpB,MAAM,CAeR"}

package/dist/tools/utils/queue.js

@@ -0,0 +1,41 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.appendSuffixToQueueUrl = appendSuffixToQueueUrl;
+exports.replaceRegionInUrl = replaceRegionInUrl;
+exports.getQueueUrl = getQueueUrl;
+function appendSuffixToQueueUrl(queueUrl, suffix) {
+    const parts = queueUrl.split("/");
+    const queueName = parts.pop();
+    const queueNameWithoutFifo = queueName?.replace(/\.fifo$/, "");
+    return `${parts.join("/")}/${queueNameWithoutFifo}-${suffix}.fifo`;
+}
+function replaceRegionInUrl(url, newRegion) {
+    const parsedUrl = new URL(url);
+    parsedUrl.hostname = parsedUrl.hostname.replace(/\.([a-z0-9-]+)\.amazonaws\.com$/, `.${newRegion}.amazonaws.com`);
+    return parsedUrl.toString();
+}
+function buildQueueUrlsForPreview(baseQueueUrl) {
+    const buildHash = process.env.TOOL_EXECUTION_SERVICE_BUILD_HASH;
+    if (!buildHash) {
+        throw new Error("TOOL_EXECUTION_SERVICE_BUILD_HASH is not set");
+    }
+    const updatedRegionForQueue = replaceRegionInUrl(baseQueueUrl, process.env.AWS_REGION);
+    const queueUrl = appendSuffixToQueueUrl(updatedRegionForQueue, buildHash);
+    console.log("queueUrl for preview", queueUrl);
+    return queueUrl;
+}
+function getQueueUrl(toolCalls, needsBrowser) {
+    const browserSqsQueueUrl = process.env.TOOL_EXECUTION_SQS_URL_BROWSER;
+    const baseSqsQueueUrl = process.env.TOOL_EXECUTION_SQS_URL;
+    const queueUrl = needsBrowser ? browserSqsQueueUrl : baseSqsQueueUrl;
+    if (!queueUrl) {
+        throw new Error("No queue URL found");
+    }
+    const vercelEnv = process.env.VERCEL_ENV || "";
+    if (vercelEnv !== "preview") {
+        return queueUrl;
+    }
+    else {
+        return buildQueueUrlsForPreview(queueUrl);
+    }
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@empiricalrun/test-gen",
-  "version": "0.72.0",
+  "version": "0.73.1",
   "publishConfig": {
     "registry": "https://registry.npmjs.org/",
     "access": "public"
@@ -27,8 +27,8 @@
       "default": "./dist/index.js"
     },
     "./tool-caller": {
-      "types": "./dist/tools/definitions/index.d.ts",
-      "default": "./dist/tools/definitions/index.js"
+      "types": "./dist/tools/index.d.ts",
+      "default": "./dist/tools/index.js"
     },
     "./tool-executor": {
       "types": "./dist/tools/executor/index.d.ts",
@@ -72,7 +72,7 @@
     "tsx": "^4.16.2",
     "typescript": "^5.3.3",
     "zod": "^3.23.8",
-    "@empiricalrun/llm": "^0.20.0",
+    "@empiricalrun/llm": "^0.20.2",
     "@empiricalrun/r2-uploader": "^0.3.9",
     "@empiricalrun/test-run": "^0.10.9"
   },
@@ -90,7 +90,7 @@
     "playwright": "1.53.2",
     "serve-handler": "^6.1.6",
     "ts-patch": "^3.3.0",
-    "@empiricalrun/shared-types": "0.7.0"
+    "@empiricalrun/shared-types": "0.8.0"
   },
   "scripts": {
     "dev": "tspc --build --watch",

package/tsconfig.tsbuildinfo

@@ -1 +1 @@
-{"root":["./src/index.ts","./src/logger.ts","./src/actions/assert.ts","./src/actions/click.ts","./src/actions/done.ts","./src/actions/fill.ts","./src/actions/goto.ts","./src/actions/hover.ts","./src/actions/index.ts","./src/actions/next-task.ts","./src/actions/press.ts","./src/actions/skill.ts","./src/actions/text-content.ts","./src/actions/constants/index.ts","./src/actions/utils/index.ts","./src/agent/browsing/index.ts","./src/agent/browsing/run.ts","./src/agent/browsing/utils.ts","./src/agent/chat/agent-loop.ts","./src/agent/chat/exports.ts","./src/agent/chat/index.ts","./src/agent/chat/models.ts","./src/agent/chat/state.ts","./src/agent/chat/types.ts","./src/agent/chat/utils.ts","./src/agent/chat/prompt/index.ts","./src/agent/chat/prompt/pw-utils-docs.ts","./src/agent/chat/prompt/repo.ts","./src/agent/chat/utils/tool-calls.ts","./src/agent/codegen/create-test-block.ts","./src/agent/codegen/fix-ts-errors.ts","./src/agent/codegen/generate-code-apply-changes.ts","./src/agent/codegen/lexical-scoped-vars.ts","./src/agent/codegen/repo-edit.ts","./src/agent/codegen/run.ts","./src/agent/codegen/skills-retriever.ts","./src/agent/codegen/test-update-feedback.ts","./src/agent/codegen/types.ts","./src/agent/codegen/update-flow.ts","./src/agent/codegen/use-skill.ts","./src/agent/codegen/utils.ts","./src/agent/cua/computer.ts","./src/agent/cua/index.ts","./src/agent/cua/model.ts","./src/agent/cua/pw-codegen/element-from-point.ts","./src/agent/cua/pw-codegen/types.ts","./src/agent/cua/pw-codegen/pw-pause/for-recorder.ts","./src/agent/cua/pw-codegen/pw-pause/index.ts","./src/agent/cua/pw-codegen/pw-pause/ipc.ts","./src/agent/cua/pw-codegen/pw-pause/patch.ts","./src/agent/cua/pw-codegen/pw-pause/types.ts","./src/agent/cua/pw-codegen/pw-pause/utils.ts","./src/agent/diagnosis-agent/index.ts","./src/agent/diagnosis-agent/strict-mode-violation.ts","./src/agent/enrich-prompt/index.ts","./src/agent/enrich-prompt/utils.ts","./src/agent/infer-agent/index.ts","./src/agent/master/action-tool-calls.ts","./src/agent/master/element-annotation.ts","./src/agent/master/execute-browser-action.ts","./src/agent/master/execute-skill-action.ts","./src/agent/master/next-action.ts","./src/agent/master/planner.ts","./src/agent/master/run.ts","./src/agent/master/scroller.ts","./src/agent/master/with-hints.ts","./src/agent/master/browser-tests/cua.spec.ts","./src/agent/master/browser-tests/fixtures.ts","./src/agent/master/browser-tests/index.spec.ts","./src/agent/master/browser-tests/skills.spec.ts","./src/agent/master/icon-descriptor/index.ts","./src/agent/master/icon-descriptor/normalize-svg.ts","./src/agent/planner/run-time-planner.ts","./src/agent/planner/run.ts","./src/artifacts/index.ts","./src/artifacts/utils.ts","./src/auth/api-client.ts","./src/auth/cli-auth.ts","./src/auth/index.ts","./src/auth/token-store.ts","./src/bin/environments.ts","./src/bin/index.ts","./src/bin/setup.ts","./src/bin/logger/index.ts","./src/bin/utils/context.ts","./src/bin/utils/index.ts","./src/bin/utils/fs/index.ts","./src/bin/utils/platform/web/index.ts","./src/bin/utils/platform/web/test-files/ts-path-import-validate.ts","./src/bin/utils/scenarios/index.ts","./src/browser-injected-scripts/annotate-elements.spec.ts","./src/constants/index.ts","./src/dashboard/index.ts","./src/dashboard/totp.ts","./src/errors/index.ts","./src/evals/add-scenario-agent.evals.ts","./src/evals/append-create-test-agent.evals.ts","./src/evals/fetch-pom-skills-agent.evals.ts","./src/evals/infer-master-or-code-agent.evals.ts","./src/evals/master-agent.evals.ts","./src/evals/type.ts","./src/evals/update-scenario-agent.evals.ts","./src/file/client.ts","./src/file/server.ts","./src/file-info/file-system.ts","./src/file-info/github.ts","./src/human-in-the-loop/cli.ts","./src/human-in-the-loop/index.ts","./src/human-in-the-loop/ipc.ts","./src/page/index.ts","./src/prompts/lib/ts-transformer.ts","./src/recorder/env-variables.ts","./src/recorder/index.ts","./src/recorder/request.ts","./src/recorder/temp-files.ts","./src/recorder/upload.ts","./src/recorder/validation.ts","./src/session/index.ts","./src/test-build/index.ts","./src/tools/commit-and-create-pr.ts","./src/tools/diagnosis-fetcher.ts","./src/tools/download-build.ts","./src/tools/list-environments.ts","./src/tools/run-test.ts","./src/tools/str_replace_editor.ts","./src/tools/test-gen-browser.ts","./src/tools/definitions/index.ts","./src/tools/definitions/run-test.ts","./src/tools/definitions/str_replace_editor.ts","./src/tools/definitions/test-gen-browser.ts","./src/tools/definitions/utils/queue.ts","./src/tools/executor/index.ts","./src/tools/executor/utils/checkpoint.ts","./src/tools/executor/utils/git.ts","./src/tools/executor/utils/index.ts","./src/tools/grep/index.ts","./src/tools/grep/ripgrep/index.ts","./src/tools/grep/ripgrep/types.ts","./src/tools/test-run-fetcher/index.ts","./src/tools/test-run-fetcher/types.ts","./src/tools/upgrade-packages/index.ts","./src/tools/upgrade-packages/utils.ts","./src/types/handlebars.d.ts","./src/types/index.ts","./src/utils/SQSClient.ts","./src/utils/env.ts","./src/utils/exec.ts","./src/utils/file.ts","./src/utils/html.ts","./src/utils/index.ts","./src/utils/json.ts","./src/utils/repo-tree.ts","./src/utils/slug.ts","./src/utils/string.ts","./src/utils/stripAnsi.ts"],"version":"5.8.3"}
+{"root":["./src/index.ts","./src/logger.ts","./src/actions/assert.ts","./src/actions/click.ts","./src/actions/done.ts","./src/actions/fill.ts","./src/actions/goto.ts","./src/actions/hover.ts","./src/actions/index.ts","./src/actions/next-task.ts","./src/actions/press.ts","./src/actions/skill.ts","./src/actions/text-content.ts","./src/actions/constants/index.ts","./src/actions/utils/index.ts","./src/agent/browsing/index.ts","./src/agent/browsing/run.ts","./src/agent/browsing/utils.ts","./src/agent/chat/agent-loop.ts","./src/agent/chat/exports.ts","./src/agent/chat/index.ts","./src/agent/chat/models.ts","./src/agent/chat/state.ts","./src/agent/chat/types.ts","./src/agent/chat/utils.ts","./src/agent/chat/prompt/index.ts","./src/agent/chat/prompt/pw-utils-docs.ts","./src/agent/chat/prompt/repo.ts","./src/agent/chat/utils/tool-calls.ts","./src/agent/code-review/prompt.ts","./src/agent/codegen/create-test-block.ts","./src/agent/codegen/fix-ts-errors.ts","./src/agent/codegen/generate-code-apply-changes.ts","./src/agent/codegen/lexical-scoped-vars.ts","./src/agent/codegen/repo-edit.ts","./src/agent/codegen/run.ts","./src/agent/codegen/skills-retriever.ts","./src/agent/codegen/test-update-feedback.ts","./src/agent/codegen/types.ts","./src/agent/codegen/update-flow.ts","./src/agent/codegen/use-skill.ts","./src/agent/codegen/utils.ts","./src/agent/cua/computer.ts","./src/agent/cua/index.ts","./src/agent/cua/model.ts","./src/agent/cua/pw-codegen/element-from-point.ts","./src/agent/cua/pw-codegen/types.ts","./src/agent/cua/pw-codegen/pw-pause/for-recorder.ts","./src/agent/cua/pw-codegen/pw-pause/index.ts","./src/agent/cua/pw-codegen/pw-pause/ipc.ts","./src/agent/cua/pw-codegen/pw-pause/patch.ts","./src/agent/cua/pw-codegen/pw-pause/types.ts","./src/agent/cua/pw-codegen/pw-pause/utils.ts","./src/agent/diagnosis-agent/index.ts","./src/agent/diagnosis-agent/strict-mode-violation.ts","./src/agent/enrich-prompt/index.ts","./src/agent/enrich-prompt/utils.ts","./src/agent/infer-agent/index.ts","./src/agent/master/action-tool-calls.ts","./src/agent/master/element-annotation.ts","./src/agent/master/execute-browser-action.ts","./src/agent/master/execute-skill-action.ts","./src/agent/master/next-action.ts","./src/agent/master/planner.ts","./src/agent/master/run.ts","./src/agent/master/scroller.ts","./src/agent/master/with-hints.ts","./src/agent/master/browser-tests/cua.spec.ts","./src/agent/master/browser-tests/fixtures.ts","./src/agent/master/browser-tests/index.spec.ts","./src/agent/master/browser-tests/skills.spec.ts","./src/agent/master/icon-descriptor/index.ts","./src/agent/master/icon-descriptor/normalize-svg.ts","./src/agent/planner/run-time-planner.ts","./src/agent/planner/run.ts","./src/artifacts/index.ts","./src/artifacts/utils.ts","./src/auth/cli-auth.ts","./src/auth/index.ts","./src/auth/token-store.ts","./src/bin/environments.ts","./src/bin/index.ts","./src/bin/setup.ts","./src/bin/logger/index.ts","./src/bin/utils/context.ts","./src/bin/utils/index.ts","./src/bin/utils/fs/index.ts","./src/bin/utils/platform/web/index.ts","./src/bin/utils/platform/web/test-files/ts-path-import-validate.ts","./src/bin/utils/scenarios/index.ts","./src/browser-injected-scripts/annotate-elements.spec.ts","./src/constants/index.ts","./src/dashboard/client.ts","./src/dashboard/index.ts","./src/dashboard/totp.ts","./src/dashboard/types.ts","./src/errors/index.ts","./src/evals/add-scenario-agent.evals.ts","./src/evals/append-create-test-agent.evals.ts","./src/evals/fetch-pom-skills-agent.evals.ts","./src/evals/infer-master-or-code-agent.evals.ts","./src/evals/master-agent.evals.ts","./src/evals/type.ts","./src/evals/update-scenario-agent.evals.ts","./src/file/client.ts","./src/file/server.ts","./src/file-info/file-system.ts","./src/file-info/github.ts","./src/human-in-the-loop/cli.ts","./src/human-in-the-loop/index.ts","./src/human-in-the-loop/ipc.ts","./src/page/index.ts","./src/prompts/lib/ts-transformer.ts","./src/recorder/env-variables.ts","./src/recorder/index.ts","./src/recorder/request.ts","./src/recorder/temp-files.ts","./src/recorder/upload.ts","./src/recorder/validation.ts","./src/test-build/index.ts","./src/tools/commit-and-create-pr.ts","./src/tools/delete-file.ts","./src/tools/diagnosis-fetcher.ts","./src/tools/download-build.ts","./src/tools/index.ts","./src/tools/list-environments.ts","./src/tools/merge-conflicts.ts","./src/tools/run-test.ts","./src/tools/str_replace_editor.ts","./src/tools/test-gen-browser.ts","./src/tools/definitions/run-test.ts","./src/tools/definitions/str_replace_editor.ts","./src/tools/definitions/test-gen-browser.ts","./src/tools/executor/index.ts","./src/tools/executor/utils/checkpoint.ts","./src/tools/executor/utils/git.ts","./src/tools/executor/utils/index.ts","./src/tools/executor/utils/pr-description.ts","./src/tools/fetch-image/index.ts","./src/tools/grep/index.ts","./src/tools/grep/ripgrep/index.ts","./src/tools/grep/ripgrep/types.ts","./src/tools/test-run-fetcher/index.ts","./src/tools/test-run-fetcher/types.ts","./src/tools/upgrade-packages/index.ts","./src/tools/upgrade-packages/utils.ts","./src/tools/utils/queue.ts","./src/types/handlebars.d.ts","./src/types/index.ts","./src/utils/SQSClient.ts","./src/utils/env.ts","./src/utils/exec.ts","./src/utils/file.ts","./src/utils/html.ts","./src/utils/index.ts","./src/utils/json.ts","./src/utils/repo-tree.ts","./src/utils/slug.ts","./src/utils/string.ts","./src/utils/stripAnsi.ts"],"version":"5.8.3"}

package/dist/auth/api-client.d.ts

@@ -1,13 +0,0 @@
-declare class APIClient {
-    private appUrl;
-    constructor();
-    request<T>(endpoint: string, options?: RequestInit): Promise<Response & {
-        json(): Promise<T>;
-    }>;
-    private makeRequest;
-    ensureAuthenticated(): Promise<void>;
-    private refreshToken;
-}
-export declare const apiClient: APIClient;
-export {};
-//# sourceMappingURL=api-client.d.ts.map

package/dist/auth/api-client.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"api-client.d.ts","sourceRoot":"","sources":["../../src/auth/api-client.ts"],"names":[],"mappings":"AAQA,cAAM,SAAS;IACb,OAAO,CAAC,MAAM,CAAS;;IAMjB,OAAO,CAAC,CAAC,EACb,QAAQ,EAAE,MAAM,EAChB,OAAO,GAAE,WAAgB,GACxB,OAAO,CAAC,QAAQ,GAAG;QAAE,IAAI,IAAI,OAAO,CAAC,CAAC,CAAC,CAAA;KAAE,CAAC;YA+B/B,WAAW;IAuBnB,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC;YAiB5B,YAAY;CA4C3B;AAED,eAAO,MAAM,SAAS,WAAkB,CAAC"}

package/dist/auth/api-client.js

@@ -1,107 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.apiClient = void 0;
-const logger_1 = require("../logger");
-const token_store_1 = require("./token-store");
-class APIClient {
-    appUrl;
-    constructor() {
-        this.appUrl = process.env.DASHBOARD_DOMAIN || "https://dash.empirical.run";
-    }
-    async request(endpoint, options = {}) {
-        const apiKey = process.env.EMPIRICALRUN_API_KEY;
-        if (apiKey) {
-            return this.makeRequest(endpoint, options, apiKey);
-        }
-        await this.ensureAuthenticated();
-        const tokens = await (0, token_store_1.getStoredTokens)();
-        if (!tokens) {
-            throw new Error("Not authenticated. Please run the login command first.");
-        }
-        const response = await this.makeRequest(endpoint, options, tokens.access_token);
-        if (response.status === 401) {
-            logger_1.logger.debug("Access token expired, attempting to refresh...");
-            const refreshed = await this.refreshToken();
-            if (refreshed) {
-                const newTokens = await (0, token_store_1.getStoredTokens)();
-                if (newTokens) {
-                    return this.makeRequest(endpoint, options, newTokens.access_token);
-                }
-            }
-            throw new Error("Authentication failed. Please run the login command again.");
-        }
-        return response;
-    }
-    async makeRequest(endpoint, options, authToken) {
-        const url = endpoint.startsWith("http")
-            ? endpoint
-            : `${this.appUrl}${endpoint}`;
-        // Don't set Content-Type for FormData (let browser set it automatically)
-        const headers = {
-            Authorization: `Bearer ${authToken}`,
-            ...options.headers,
-        };
-        // Only set Content-Type to application/json if body is not FormData
-        if (!(options.body instanceof FormData)) {
-            headers["Content-Type"] = "application/json";
-        }
-        return fetch(url, {
-            ...options,
-            headers,
-        });
-    }
-    async ensureAuthenticated() {
-        if (!(await (0, token_store_1.isAuthenticated)())) {
-            const tokens = await (0, token_store_1.getStoredTokens)();
-            if (!tokens) {
-                throw new Error("Not authenticated. Please run the login command first.");
-            }
-            // Token exists but is expired, try to refresh
-            const refreshed = await this.refreshToken();
-            if (!refreshed) {
-                throw new Error("Session expired. Please run the login command again.");
-            }
-        }
-    }
-    async refreshToken() {
-        const tokens = await (0, token_store_1.getStoredTokens)();
-        if (!tokens || !tokens.refresh_token) {
-            return false;
-        }
-        try {
-            // Use dashboard refresh endpoint instead of Supabase directly
-            const response = await fetch(`${this.appUrl}/api/cli/refresh`, {
-                method: "POST",
-                headers: {
-                    "Content-Type": "application/json",
-                },
-                body: JSON.stringify({
-                    refresh_token: tokens.refresh_token,
-                }),
-            });
-            if (!response.ok) {
-                const errorText = await response.text();
-                console.error("Token refresh failed:", errorText);
-                await (0, token_store_1.clearTokens)();
-                return false;
-            }
-            const data = await response.json();
-            // Store the new tokens (refresh token rotation means we get a new refresh token)
-            await (0, token_store_1.storeTokens)({
-                access_token: data.access_token,
-                refresh_token: data.refresh_token,
-                expires_at: data.expires_at,
-                user_id: data.user?.id || tokens.user_id,
-                user_email: data.user?.email || tokens.user_email,
-            });
-            logger_1.logger.debug("Access token refreshed successfully via dashboard");
-            return true;
-        }
-        catch (error) {
-            console.error("Token refresh error:", error);
-            await (0, token_store_1.clearTokens)();
-            return false;
-        }
-    }
-}
-exports.apiClient = new APIClient();

package/dist/session/index.d.ts

@@ -1,20 +0,0 @@
-declare function getSessionDetails(): {
-    generationId: number | undefined;
-    sessionId: number | undefined;
-    testUrl: string | undefined;
-    version: string;
-};
-export declare function setSessionDetails({ sessionId, generationId, testCaseId, projectRepoName, }: {
-    sessionId: number;
-    generationId: number;
-    testCaseId: number;
-    projectRepoName: string;
-}): void;
-export declare function shouldStopSession(): Promise<boolean>;
-export declare function getSessionState(): Promise<"started" | "completed" | "request_complete">;
-export declare function endSession(): Promise<void>;
-export { getSessionDetails };
-export declare function updateSessionStatus(sessionId: number, payload: {
-    status: string;
-}): Promise<void>;
-//# sourceMappingURL=index.d.ts.map

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

@@ -1 +0,0 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/session/index.ts"],"names":[],"mappings":"AAmBA,iBAAS,iBAAiB;;;;;EAOzB;AAED,wBAAgB,iBAAiB,CAAC,EAChC,SAAS,EACT,YAAY,EACZ,UAAU,EACV,eAAe,GAChB,EAAE;IACD,SAAS,EAAE,MAAM,CAAC;IAClB,YAAY,EAAE,MAAM,CAAC;IACrB,UAAU,EAAE,MAAM,CAAC;IACnB,eAAe,EAAE,MAAM,CAAC;CACzB,QAIA;AAED,wBAAsB,iBAAiB,qBAItC;AAED,wBAAsB,eAAe,0DA0BpC;AAED,wBAAsB,UAAU,kBAyB/B;AAED,OAAO,EAAE,iBAAiB,EAAE,CAAC;AAE7B,wBAAsB,mBAAmB,CACvC,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE;IACP,MAAM,EAAE,MAAM,CAAC;CAChB,iBAmBF"}

package/dist/session/index.js

@@ -1,104 +0,0 @@
-"use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.setSessionDetails = setSessionDetails;
-exports.shouldStopSession = shouldStopSession;
-exports.getSessionState = getSessionState;
-exports.endSession = endSession;
-exports.getSessionDetails = getSessionDetails;
-exports.updateSessionStatus = updateSessionStatus;
-const package_json_1 = __importDefault(require("../../package.json"));
-const env_1 = require("../utils/env");
-const sessionDetails = {
-    sessionId: undefined,
-    version: package_json_1.default.version,
-    generationId: undefined,
-    testUrl: undefined,
-};
-const DASHBOARD_DOMAIN = process.env.DASHBOARD_DOMAIN ||
-    ((0, env_1.isRunningOnCloud)() ? "https://dash.empirical.run" : "");
-function getSessionDetails() {
-    return {
-        generationId: sessionDetails.generationId,
-        sessionId: sessionDetails.sessionId,
-        testUrl: sessionDetails.testUrl,
-        version: package_json_1.default.version,
-    };
-}
-function setSessionDetails({ sessionId, generationId, testCaseId, projectRepoName, }) {
-    sessionDetails.sessionId = sessionId;
-    sessionDetails.generationId = generationId;
-    sessionDetails.testUrl = `${process.env.DASHBOARD_DOMAIN ?? ""}/${projectRepoName}/test-cases/${testCaseId}`;
-}
-async function shouldStopSession() {
-    const sessionState = await getSessionState();
-    const shouldStopSession = sessionState !== "started";
-    return shouldStopSession;
-}
-async function getSessionState() {
-    // if there is no dashboard domain to connect to or session details missing, then continue the generation session.
-    if (!DASHBOARD_DOMAIN ||
-        !sessionDetails.sessionId ||
-        !sessionDetails.generationId) {
-        return "started";
-    }
-    const apiPath = `${DASHBOARD_DOMAIN}/api/sessions/${sessionDetails.sessionId}/generations/${sessionDetails.generationId}/state`;
-    const response = await fetch(apiPath, {
-        method: "GET",
-        headers: {
-            "Content-Type": "application/json",
-            Authorization: `Bearer ${process.env.EMPIRICALRUN_API_KEY}`,
-        },
-    });
-    const generationStateData = (await response.json());
-    // TODO: fix this heirarchy
-    return generationStateData.data.state.state;
-}
-async function endSession() {
-    if (!DASHBOARD_DOMAIN) {
-        return;
-    }
-    if (!sessionDetails.sessionId || !sessionDetails.generationId) {
-        console.warn("Session details missing, skipping session end");
-        return;
-    }
-    const apiPath = `${DASHBOARD_DOMAIN}/api/sessions/${sessionDetails.sessionId}/generations/${sessionDetails.generationId}/state`;
-    try {
-        await fetch(apiPath, {
-            method: "POST",
-            headers: {
-                "Content-Type": "application/json",
-                Authorization: `Bearer ${process.env.EMPIRICALRUN_API_KEY}`,
-            },
-            body: JSON.stringify({
-                state: {
-                    state: "completed",
-                },
-            }),
-        });
-    }
-    catch (error) {
-        console.error("Failed to end session:", error);
-    }
-}
-async function updateSessionStatus(sessionId, payload) {
-    if (!DASHBOARD_DOMAIN) {
-        return;
-    }
-    try {
-        const body = JSON.stringify(payload);
-        await fetch(`${DASHBOARD_DOMAIN}/api/sessions/${sessionId}`, {
-            method: "PATCH",
-            headers: {
-                "Content-Type": "application/json",
-                Authorization: `Bearer ${process.env.EMPIRICALRUN_API_KEY}`,
-            },
-            body,
-        });
-    }
-    catch (error) {
-        console.error("Failed to update session status:", error);
-    }
-}

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

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

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

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

package/dist/tools/definitions/index.js

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

package/dist/tools/definitions/utils/queue.d.ts

@@ -1,8 +0,0 @@
-import { PendingToolCall, ServicePayload, ToolDefinition } from "@empiricalrun/shared-types";
-export declare function appendBranchNameToQueueUrl(queueUrl: string, branchName: string): string;
-export declare function replaceRegionInUrl(url: string, newRegion: string): string;
-export declare function getQueueUrl(toolCalls: PendingToolCall[], tools: ToolDefinition[], workerEnv?: Record<string, string>): string;
-export declare function sendToolRequestToRemoteQueue(queueUrl: string, payload: ServicePayload & {
-    workerEnv?: Record<string, string>;
-}): Promise<void>;
-//# sourceMappingURL=queue.d.ts.map

package/dist/tools/definitions/utils/queue.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"queue.d.ts","sourceRoot":"","sources":["../../../../src/tools/definitions/utils/queue.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,eAAe,EACf,cAAc,EACd,cAAc,EACf,MAAM,4BAA4B,CAAC;AAIpC,wBAAgB,0BAA0B,CACxC,QAAQ,EAAE,MAAM,EAChB,UAAU,EAAE,MAAM,GACjB,MAAM,CAOR;AAED,wBAAgB,kBAAkB,CAAC,GAAG,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,MAAM,CAOzE;AAgBD,wBAAgB,WAAW,CACzB,SAAS,EAAE,eAAe,EAAE,EAC5B,KAAK,EAAE,cAAc,EAAE,EACvB,SAAS,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GACjC,MAAM,CAyBR;AAED,wBAAsB,4BAA4B,CAChD,QAAQ,EAAE,MAAM,EAChB,OAAO,EAAE,cAAc,GAAG;IAAE,SAAS,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;CAAE,iBAiBjE"}

package/dist/tools/definitions/utils/queue.js

@@ -1,58 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.appendBranchNameToQueueUrl = appendBranchNameToQueueUrl;
-exports.replaceRegionInUrl = replaceRegionInUrl;
-exports.getQueueUrl = getQueueUrl;
-exports.sendToolRequestToRemoteQueue = sendToolRequestToRemoteQueue;
-const SQSClient_1 = require("../../../utils/SQSClient");
-function appendBranchNameToQueueUrl(queueUrl, branchName) {
-    const cleanBranchName = branchName.replace(/[/.]/g, "-");
-    const suffix = cleanBranchName.slice(0, 50);
-    const parts = queueUrl.split("/");
-    const queueName = parts.pop();
-    const queueNameWithoutFifo = queueName?.replace(/\.fifo$/, "");
-    return `${parts.join("/")}/${queueNameWithoutFifo}-${suffix}.fifo`;
-}
-function replaceRegionInUrl(url, newRegion) {
-    const parsedUrl = new URL(url);
-    parsedUrl.hostname = parsedUrl.hostname.replace(/\.([a-z0-9-]+)\.amazonaws\.com$/, `.${newRegion}.amazonaws.com`);
-    return parsedUrl.toString();
-}
-function buildQueueUrlsForPreview(baseQueueUrl) {
-    const branch = process.env.VERCEL_GIT_COMMIT_REF;
-    if (!branch) {
-        throw new Error("VERCEL_GIT_COMMIT_REF is not set");
-    }
-    const updatedRegionForQueue = replaceRegionInUrl(baseQueueUrl, process.env.AWS_REGION);
-    const queueUrl = appendBranchNameToQueueUrl(updatedRegionForQueue, branch);
-    console.log("queueUrl for preview", queueUrl);
-    return queueUrl;
-}
-function getQueueUrl(toolCalls, tools, workerEnv) {
-    const needsBrowser = toolCalls.some((toolCall) => tools.find((t) => t.schema.name === toolCall.name)?.needsBrowser);
-    const browserSwsQueueUrl = workerEnv?.TOOL_EXECUTION_SQS_URL_BROWSER ||
-        process.env.TOOL_EXECUTION_SQS_URL_BROWSER;
-    const baseSqsQueueUrl = workerEnv?.TOOL_EXECUTION_SQS_URL || process.env.TOOL_EXECUTION_SQS_URL;
-    const baseQueueUrl = needsBrowser ? browserSwsQueueUrl : baseSqsQueueUrl;
-    if (!baseQueueUrl) {
-        throw new Error("No queue URL found");
-    }
-    const vercelEnv = workerEnv?.VERCEL_ENV || process.env.VERCEL_ENV || "";
-    if (vercelEnv !== "preview") {
-        return baseQueueUrl;
-    }
-    else {
-        return buildQueueUrlsForPreview(baseQueueUrl);
-    }
-}
-async function sendToolRequestToRemoteQueue(queueUrl, payload) {
-    const { workerEnv } = payload;
-    const region = workerEnv?.AWS_REGION || process.env.AWS_REGION;
-    const accessKeyId = workerEnv?.AWS_ACCESS_KEY_ID || process.env.AWS_ACCESS_KEY_ID;
-    const secretAccessKey = workerEnv?.AWS_SECRET_ACCESS_KEY || process.env.AWS_SECRET_ACCESS_KEY;
-    if (!region || !accessKeyId || !secretAccessKey) {
-        throw new Error("AWS_REGION, AWS_ACCESS_KEY_ID, and AWS_SECRET_ACCESS_KEY must be set");
-    }
-    const sqs = new SQSClient_1.SQSClient(region, accessKeyId, secretAccessKey);
-    await sqs.sendMessage(queueUrl, payload);
-}