@empiricalrun/test-gen 0.71.2 → 0.73.0

package/dist/utils/SQSClient.js

@@ -0,0 +1,116 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.SQSClient = void 0;
+class SQSClient {
+    region;
+    accessKeyId;
+    secretAccessKey;
+    sessionToken;
+    constructor(region, accessKeyId, secretAccessKey, sessionToken) {
+        this.region = region;
+        this.accessKeyId = accessKeyId;
+        this.secretAccessKey = secretAccessKey;
+        this.sessionToken = sessionToken;
+    }
+    async sha256(message) {
+        const msgBuffer = new TextEncoder().encode(message);
+        const hashBuffer = await crypto.subtle.digest("SHA-256", msgBuffer);
+        const hashArray = Array.from(new Uint8Array(hashBuffer));
+        return hashArray.map((b) => b.toString(16).padStart(2, "0")).join("");
+    }
+    async hmacSha256(key, message) {
+        const cryptoKey = await crypto.subtle.importKey("raw", key, { name: "HMAC", hash: "SHA-256" }, false, ["sign"]);
+        return await crypto.subtle.sign("HMAC", cryptoKey, new TextEncoder().encode(message));
+    }
+    async getSignatureKey(key, dateStamp, regionName, serviceName) {
+        const kDate = await this.hmacSha256(new TextEncoder().encode("AWS4" + key), dateStamp);
+        const kRegion = await this.hmacSha256(kDate, regionName);
+        const kService = await this.hmacSha256(kRegion, serviceName);
+        const kSigning = await this.hmacSha256(kService, "aws4_request");
+        return kSigning;
+    }
+    async sign(stringToSign, signingKey) {
+        const signature = await this.hmacSha256(signingKey, stringToSign);
+        const signatureArray = Array.from(new Uint8Array(signature));
+        return signatureArray.map((b) => b.toString(16).padStart(2, "0")).join("");
+    }
+    async sendMessage(queueUrl, payload) {
+        try {
+            const url = new URL(queueUrl);
+            const host = url.hostname;
+            const canonicalUri = url.pathname;
+            const now = new Date();
+            // The following two operations generate AWS-compliant timestamps:
+            // - amzDate: The full ISO8601 timestamp (without milliseconds and with no separators), used for the x-amz-date header.
+            // - dateStamp: The date portion (YYYYMMDD), used in the credential scope for AWS Signature V4.
+            const amzDate = now
+                .toISOString()
+                .replace(/\.\d{3}Z$/, "Z")
+                .replace(/[:-]/g, "");
+            const dateStamp = amzDate.slice(0, 8);
+            const method = "POST";
+            const service = "sqs";
+            const algorithm = "AWS4-HMAC-SHA256";
+            const credentialScope = `${dateStamp}/${this.region}/${service}/aws4_request`;
+            const messageBody = {
+                Action: "SendMessage",
+                Version: "2012-11-05",
+                QueueUrl: queueUrl,
+                MessageBody: JSON.stringify(payload),
+                MessageGroupId: payload.requestId,
+                MessageDeduplicationId: payload.requestId,
+            };
+            const body = Object.entries(messageBody)
+                .map(([key, value]) => `${encodeURIComponent(key)}=${encodeURIComponent(value)}`)
+                .join("&");
+            const payloadHash = await this.sha256(body);
+            const canonicalHeaders = [
+                `host:${host}`,
+                `x-amz-date:${amzDate}`,
+                "",
+            ].join("\n");
+            const signedHeaders = "host;x-amz-date";
+            const canonicalRequest = [
+                method,
+                canonicalUri,
+                "",
+                canonicalHeaders,
+                signedHeaders,
+                payloadHash,
+            ].join("\n");
+            const stringToSign = [
+                algorithm,
+                amzDate,
+                credentialScope,
+                await this.sha256(canonicalRequest),
+            ].join("\n");
+            const signingKey = await this.getSignatureKey(this.secretAccessKey, dateStamp, this.region, service);
+            const signature = await this.sign(stringToSign, signingKey);
+            const authorizationHeader = `${algorithm} Credential=${this.accessKeyId}/${credentialScope}, SignedHeaders=${signedHeaders}, Signature=${signature}`;
+            const headers = {
+                "Content-Type": "application/x-www-form-urlencoded; charset=utf-8",
+                Host: host,
+                "X-Amz-Date": amzDate,
+                Authorization: authorizationHeader,
+            };
+            if (this.sessionToken) {
+                headers["X-Amz-Security-Token"] = this.sessionToken;
+            }
+            const response = await fetch(queueUrl, {
+                method: "POST",
+                headers,
+                body: body,
+            });
+            if (!response.ok) {
+                const errorText = await response.text();
+                console.error(`[SQSClient.sendMessage] SQS request failed: ${response.status} ${response.statusText} - ${errorText}`);
+                throw new Error(`SQS request failed: ${response.status} ${response.statusText} - ${errorText}`);
+            }
+        }
+        catch (err) {
+            console.error("[SQSClient.sendMessage] Error:", err instanceof Error ? err.stack || err.message : err);
+            throw err;
+        }
+    }
+}
+exports.SQSClient = SQSClient;

package/dist/utils/repo-tree.d.ts

@@ -1,4 +1,4 @@
-import { FileInfo } from "../types";
+import { FileInfo } from "@empiricalrun/shared-types";
 export declare const DEFAULT_EXCLUDE: (string | RegExp)[];
 export declare function generateAsciiTree(root: FileInfo, options?: {}): string;
 //# sourceMappingURL=repo-tree.d.ts.map

package/dist/utils/repo-tree.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"repo-tree.d.ts","sourceRoot":"","sources":["../../src/utils/repo-tree.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,UAAU,CAAC;AAEpC,eAAO,MAAM,eAAe,qBAW3B,CAAC;AAEF,wBAAgB,iBAAiB,CAAC,IAAI,EAAE,QAAQ,EAAE,OAAO,KAAK,UAqF7D"}
+{"version":3,"file":"repo-tree.d.ts","sourceRoot":"","sources":["../../src/utils/repo-tree.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,4BAA4B,CAAC;AAEtD,eAAO,MAAM,eAAe,qBAW3B,CAAC;AAEF,wBAAgB,iBAAiB,CAAC,IAAI,EAAE,QAAQ,EAAE,OAAO,KAAK,UAqF7D"}

package/dist/utils/repo-tree.js

@@ -21,7 +21,7 @@ function generateAsciiTree(root, options = {}) {
         maxDepth: 10,
     };
     const opts = { ...defaultOptions, ...options };
-    if (!root.isDirectory) {
+    if (root.type !== "directory") {
         throw new Error(`"${root.path}" is not a valid directory`);
     }
     let result = root.name + "\n";
@@ -31,9 +31,9 @@ function generateAsciiTree(root, options = {}) {
         .filter((child) => child !== undefined)
         .sort((a, b) => {
         // Files first, then directories
-        if (!a.isDirectory && b.isDirectory)
+        if (a.type !== "directory" && b.type === "directory")
             return -1;
-        if (a.isDirectory && !b.isDirectory)
+        if (a.type === "directory" && b.type !== "directory")
             return 1;
         return a.name.localeCompare(b.name);
     });
@@ -62,17 +62,17 @@ function generateAsciiTree(root, options = {}) {
             continue;
         // Add the current item to the result
         const symbol = isLast ? "└── " : "├── ";
-        result += `${prefix}${symbol}${node.name}${node.isDirectory ? "/" : ""}\n`;
-        if (node.isDirectory && node.children.length > 0) {
+        result += `${prefix}${symbol}${node.name}${node.type === "directory" ? "/" : ""}\n`;
+        if (node.type === "directory" && node.children.length > 0) {
             const nextPrefix = prefix + (isLast ? "    " : "│   ");
             // Sort and add children to stack in reverse order
             const children = [...node.children]
                 .filter((child) => child !== undefined)
                 .sort((a, b) => {
                 // Files first, then directories
-                if (!a.isDirectory && b.isDirectory)
+                if (a.type !== "directory" && b.type === "directory")
                     return -1;
-                if (a.isDirectory && !b.isDirectory)
+                if (a.type === "directory" && b.type !== "directory")
                     return 1;
                 return a.name.localeCompare(b.name);
             });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@empiricalrun/test-gen",
-  "version": "0.71.2",
+  "version": "0.73.0",
   "publishConfig": {
     "registry": "https://registry.npmjs.org/",
     "access": "public"
@@ -26,9 +26,17 @@
       "types": "./dist/index.d.ts",
       "default": "./dist/index.js"
     },
-    "./tool-call-service": {
-      "types": "./dist/tool-call-service/index.d.ts",
-      "default": "./dist/tool-call-service/index.js"
+    "./tool-caller": {
+      "types": "./dist/tools/index.d.ts",
+      "default": "./dist/tools/index.js"
+    },
+    "./tool-executor": {
+      "types": "./dist/tools/executor/index.d.ts",
+      "default": "./dist/tools/executor/index.js"
+    },
+    "./dashboard": {
+      "types": "./dist/dashboard/index.d.ts",
+      "default": "./dist/dashboard/index.js"
     }
   },
   "repository": {
@@ -37,8 +45,8 @@
   },
   "author": "Empirical Team <hey@empirical.run>",
   "dependencies": {
-    "@aws-sdk/client-sqs": "^3.787.0",
     "@babel/parser": "^7.26.3",
+    "@epic-web/totp": "^4.0.1",
     "@types/sanitize-html": "^2.11.0",
     "async-retry": "^1.3.3",
     "commander": "^12.1.0",
@@ -64,9 +72,9 @@
     "tsx": "^4.16.2",
     "typescript": "^5.3.3",
     "zod": "^3.23.8",
-    "@empiricalrun/llm": "^0.19.4",
+    "@empiricalrun/llm": "^0.20.1",
     "@empiricalrun/r2-uploader": "^0.3.9",
-    "@empiricalrun/test-run": "^0.10.8"
+    "@empiricalrun/test-run": "^0.10.9"
   },
   "devDependencies": {
     "@playwright/test": "1.53.2",
@@ -82,7 +90,7 @@
     "playwright": "1.53.2",
     "serve-handler": "^6.1.6",
     "ts-patch": "^3.3.0",
-    "@empiricalrun/shared-types": "0.6.4"
+    "@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/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/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/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/tool-call-service/index.ts","./src/tool-call-service/utils.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/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/index.ts","./src/types/handlebars.d.ts","./src/types/index.ts","./src/utils/checkpoint.ts","./src/utils/env.ts","./src/utils/exec.ts","./src/utils/file-tree.ts","./src/utils/file.ts","./src/utils/git.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;YA2B/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,103 +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 = {}) {
-        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, accessToken) {
-        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 ${accessToken}`,
-            ...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/tool-call-service/index.d.ts

@@ -1,32 +0,0 @@
-import { TraceClient } from "@empiricalrun/llm";
-import { PendingToolCall, ToolResult } from "@empiricalrun/llm/chat";
-import { SupportedChatModels, Tool, ToolExecute } from "@empiricalrun/shared-types";
-type ToolExecutors = {
-    [key: string]: ToolExecute;
-};
-export declare class ToolCallService {
-    tools: Tool[];
-    toolExecutors: ToolExecutors;
-    chatSessionId: number | null;
-    selectedModel: SupportedChatModels;
-    branchName: string;
-    repoPath: string;
-    apiKey: string | undefined;
-    trace?: TraceClient;
-    featureFlags: string[];
-    environmentOverrides: Record<string, string>;
-    constructor({ chatSessionId, selectedModel, branchName, repoPath, apiKey, trace, featureFlags, environmentOverrides, }: {
-        chatSessionId: number | null;
-        selectedModel: SupportedChatModels;
-        branchName: string;
-        repoPath: string;
-        apiKey: string | undefined;
-        trace?: TraceClient;
-        featureFlags: string[];
-        environmentOverrides?: Record<string, string>;
-    });
-    sendToQueue(requestId: string, toolCalls: PendingToolCall[]): Promise<void>;
-    execute(toolCalls: PendingToolCall[]): Promise<ToolResult[]>;
-}
-export {};
-//# sourceMappingURL=index.d.ts.map

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

@@ -1 +0,0 @@
-{"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,UAAU,EACX,MAAM,wBAAwB,CAAC;AAChC,OAAO,EAEL,mBAAmB,EACnB,IAAI,EACJ,WAAW,EACZ,MAAM,4BAA4B,CAAC;AAmBpC,KAAK,aAAa,GAAG;IACnB,CAAC,GAAG,EAAE,MAAM,GAAG,WAAW,CAAC;CAC5B,CAAC;AAEF,qBAAa,eAAe;IAC1B,KAAK,EAAE,IAAI,EAAE,CAAM;IACnB,aAAa,EAAE,aAAa,CAAM;IAClC,aAAa,EAAE,MAAM,GAAG,IAAI,CAAC;IAC7B,aAAa,EAAE,mBAAmB,CAAC;IACnC,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC;IAC3B,KAAK,CAAC,EAAE,WAAW,CAAC;IACpB,YAAY,EAAE,MAAM,EAAE,CAAC;IACvB,oBAAoB,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;gBAEjC,EACV,aAAa,EACb,aAAa,EACb,UAAU,EACV,QAAQ,EACR,MAAM,EACN,KAAK,EACL,YAAY,EACZ,oBAAyB,GAC1B,EAAE;QACD,aAAa,EAAE,MAAM,GAAG,IAAI,CAAC;QAC7B,aAAa,EAAE,mBAAmB,CAAC;QACnC,UAAU,EAAE,MAAM,CAAC;QACnB,QAAQ,EAAE,MAAM,CAAC;QACjB,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC;QAC3B,KAAK,CAAC,EAAE,WAAW,CAAC;QACpB,YAAY,EAAE,MAAM,EAAE,CAAC;QACvB,oBAAoB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;KAC/C;IAoCK,WAAW,CACf,SAAS,EAAE,MAAM,EACjB,SAAS,EAAE,eAAe,EAAE,GAC3B,OAAO,CAAC,IAAI,CAAC;IAcV,OAAO,CAAC,SAAS,EAAE,eAAe,EAAE,GAAG,OAAO,CAAC,UAAU,EAAE,CAAC;CAyEnE"}

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

@@ -1,6 +0,0 @@
-import { PendingToolCall, ServicePayload, Tool } 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: Tool[]): string;
-export declare function sendToolRequestToRemoteQueue(queueUrl: string, payload: ServicePayload): Promise<void>;
-//# sourceMappingURL=utils.d.ts.map

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

@@ -1 +0,0 @@
-{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../src/tool-call-service/utils.ts"],"names":[],"mappings":"AACA,OAAO,EACL,eAAe,EACf,cAAc,EACd,IAAI,EACL,MAAM,4BAA4B,CAAC;AAEpC,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,IAAI,EAAE,GACZ,MAAM,CAeR;AAED,wBAAsB,4BAA4B,CAChD,QAAQ,EAAE,MAAM,EAChB,OAAO,EAAE,cAAc,iBAiBxB"}

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

@@ -1,57 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.appendBranchNameToQueueUrl = appendBranchNameToQueueUrl;
-exports.replaceRegionInUrl = replaceRegionInUrl;
-exports.getQueueUrl = getQueueUrl;
-exports.sendToolRequestToRemoteQueue = sendToolRequestToRemoteQueue;
-const client_sqs_1 = require("@aws-sdk/client-sqs");
-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) {
-    const needsBrowser = toolCalls.some((toolCall) => tools.find((t) => t.schema.name === toolCall.name)?.needsBrowser);
-    const baseQueueUrl = needsBrowser
-        ? process.env.TOOL_EXECUTION_SQS_URL_BROWSER
-        : process.env.TOOL_EXECUTION_SQS_URL;
-    if (process.env.VERCEL_ENV !== "preview") {
-        return baseQueueUrl;
-    }
-    else {
-        return buildQueueUrlsForPreview(baseQueueUrl);
-    }
-}
-async function sendToolRequestToRemoteQueue(queueUrl, payload) {
-    const sqs = new client_sqs_1.SQSClient({
-        region: process.env.AWS_REGION,
-        credentials: {
-            accessKeyId: process.env.AWS_ACCESS_KEY_ID,
-            secretAccessKey: process.env.AWS_SECRET_ACCESS_KEY,
-        },
-    });
-    await sqs.send(new client_sqs_1.SendMessageCommand({
-        QueueUrl: queueUrl,
-        MessageGroupId: payload.requestId,
-        MessageDeduplicationId: payload.requestId,
-        MessageBody: JSON.stringify(payload),
-    }));
-}