@empiricalrun/test-gen 0.77.0 → 0.78.1

package/dist/tools/diagnosis-fetcher.js

@@ -1,45 +1,59 @@
 "use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.fetchDiagnosisReportTool = void 0;
-const path_1 = __importDefault(require("path"));
 const zod_1 = require("zod");
-const slug_1 = require("../utils/slug");
+const playwright_report_parser_1 = require("../utils/playwright-report-parser");
 const DiagnosisSchema = zod_1.z.object({
-    diagnosisUrl: zod_1.z
+    diagnosisId: zod_1.z
         .string()
-        .describe("The full URL of the diagnosis (e.g. https://dash.empirical.run/shopflo-tests/diagnosis/split-cod-place-the-order--byynrPjCml57)"),
+        .describe("12 character random string that identifies a failing test case in a test run"),
 });
 exports.fetchDiagnosisReportTool = {
     schema: {
         name: "fetchDiagnosisDetails",
-        description: "Fetch details about a test case diagnosis using its URL or slug. Returns diagnosis data including test metadata, error details, and file path. Use fileViewTool separately to read the actual test source code if needed.",
+        description: `Fetch details about a failed test case using its URL or slug.
+This tool returns error stack, failing line, across multiple retries (generally, 3) for the
+test case in a particular test run (identified by an id, like #3450).
+
+## How to use this tool?
+- The input to this tool is a diagnosis identifier, which is random 12 character string (e.g byynrPjCml57).
+  Internally, this string is generated by nanoid.
+
+- To get this diagnosis id, you can interpret incoming URLs that contain this id.
+
+### URL format 1
+
+https://dash.empirical.run/<project-name>/test-runs/34548?detail=ULRHHNwiGZ2W
+
+In the above, the test run id is #34548 and the diagnosis id is "ULRHHNwiGZ2W"
+
+### URL format 2
+
+https://dash.empirical.run/<project-name>/diagnosis/slug-of-test-case-name--ULRHHNwiGZ2W
+
+In the above, the diagnosis id is after the double hyphen, which is "ULRHHNwiGZ2W"
+
+## Related tools
+- You can use the file view tool to read the actual test source code, or helper methods that
+  are imported in the test to get a complete sense of what the test case was testing.
+- You can also fetch the entire test run report, which contains a list of failed test cases.`,
         parameters: DiagnosisSchema,
     },
     needsBrowser: false,
     isInlineTool: true,
-    execute: async ({ input, repoPath, apiClient }) => {
-        const { diagnosisUrl } = input;
-        let slug;
-        try {
-            slug = (0, slug_1.extractSlugFromDiagnosisUrl)(diagnosisUrl);
-        }
-        catch (error) {
-            return {
-                isError: true,
-                result: `Invalid diagnosis URL - could not extract slug: ${error instanceof Error ? error.message : String(error)}`,
-            };
-        }
+    execute: async ({ input, apiClient }) => {
+        const { diagnosisId } = input;
         let data = null;
         if (!apiClient) {
             throw new Error("Dashboard API client is not available.");
         }
         try {
-            data = await apiClient.request(`/api/diagnosis/${slug}/detailed`, {
+            data = await apiClient.request(`/api/diagnosis/${diagnosisId}/detailed`, {
                 method: "GET",
             });
+            if (!data || !data.data) {
+                throw new Error("No API response from /api/diagnosis/<slug>/detailed");
+            }
         }
         catch (error) {
             return {
@@ -48,37 +62,33 @@ exports.fetchDiagnosisReportTool = {
             };
         }
         const { test_case, diagnosis } = data.data;
-        const project = diagnosis?.test_project || "unknown";
-        const repoName = path_1.default.basename(repoPath);
-        const cleanErrorStack = diagnosis?.failed_run_metadata?.stack?.replace(`"/runner/_work/${repoName}/${repoName}/source-repo/"`, "");
-        // Format the response as markdown
+        const project = diagnosis.test_project || "unknown";
+        const retries = diagnosis.retry_details?.retries;
+        const retryInfo = retries?.reduce((acc, current) => {
+            return `${acc}\n\n## Retry ${current.retryIndex}\n\n${JSON.stringify(current, null, 2)}`;
+        }, "");
+        const lastSuccessRunForProject = await getDataForLastSuccessfulRuns(test_case);
+        let lastSuccessRunInfo = "No successful runs found.";
+        if (Object.keys(lastSuccessRunForProject).length > 0) {
+            lastSuccessRunInfo = Object.entries(lastSuccessRunForProject).reduce((acc, current) => {
+                const [project, value] = current;
+                return `${acc}\n\n## Playwright project: ${project}\n\n${JSON.stringify(value, null, 2)}`;
+            }, "");
+        }
         const markdownResponse = `
-# Test Case Diagnosis
-
-## Test Case Information
-- **Test Case Name**: ${test_case.name}
-- **Test Suite**: ${test_case.suites}
-- **File Path**: tests/${test_case.file_path}
-- **Project**: ${project}
-
-## What Happened in the Test Run
-
-### Failure Details
-- **Failing Line**: ${diagnosis?.failing_line || "No failing line available"}
+# Test Case Information
+- **Name**: ${test_case.name}
+- **Test suite** (Describe blocks): ${test_case.suites}
+- **File path**: tests/${test_case.file_path}
+- **Playwright project**: ${project}
 
-#### Error Stack
-\`\`\`
-${cleanErrorStack || "No error stack available"}
-\`\`\`
+# What happened in this test run
 
-#### Error Summary
-${diagnosis?.error_stack_summary?.content || "No error summary available"}
+${retryInfo}
 
-#### Visual Analysis
-${diagnosis?.visual_diff_summary?.summary || "No visual analysis available"}
+# Last successful runs
 
-#### Merged Summary
-${diagnosis?.merged_summary?.content || "No merged summary available"}
+${lastSuccessRunInfo}
 `;
         return {
             result: markdownResponse,
@@ -86,3 +96,38 @@ ${diagnosis?.merged_summary?.content || "No merged summary available"}
         };
     },
 };
+async function getDataForLastSuccessfulRuns(testCase) {
+    const lastSuccessfulRun = testCase.metadata?.last_successful_run;
+    const runData = {};
+    const projectPromises = Object.entries(lastSuccessfulRun ?? {}).map(async ([project, testCaseMetaData]) => {
+        if (!testCaseMetaData.summary_url)
+            return { project, success: false, data: null };
+        try {
+            const testCaseData = await (0, playwright_report_parser_1.fetchAndFindTestCase)(testCaseMetaData.summary_url, {
+                name: testCase.name,
+                suites: testCase.suites,
+                file_path: testCase.file_path,
+            });
+            return {
+                project,
+                success: true,
+                data: {
+                    ...testCaseMetaData,
+                    test_case: testCaseData,
+                },
+            };
+        }
+        catch {
+            return { project, success: false, data: null };
+        }
+    });
+    const results = await Promise.allSettled(projectPromises);
+    for (const result of results) {
+        if (result.status === "fulfilled" &&
+            result.value.success &&
+            result.value.data) {
+            runData[result.value.project] = result.value.data;
+        }
+    }
+    return runData;
+}

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

@@ -1,3 +1,3 @@
-import type { Tool } from "@empiricalrun/shared-types";
+import type { Tool } from "@empiricalrun/shared-types/chat-agent";
 export declare const downloadBuildTool: Tool;
 //# sourceMappingURL=index.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/tools/download-build/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAe,MAAM,4BAA4B,CAAC;AAQpE,eAAO,MAAM,iBAAiB,EAAE,IA4B/B,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/tools/download-build/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAe,MAAM,uCAAuC,CAAC;AAQ/E,eAAO,MAAM,iBAAiB,EAAE,IA4B/B,CAAC"}

package/dist/tools/executor/base.d.ts

@@ -1,6 +1,7 @@
-import { TraceClient } from "@empiricalrun/llm";
+import type { TraceClient } from "@empiricalrun/llm";
 import { PendingToolCall, ToolResult } from "@empiricalrun/llm/chat";
-import type { ChatSessionInfo, IDashboardAPIClient, Logger, MessageCheckpoint, Tool, ToolExecute } from "@empiricalrun/shared-types";
+import type { IDashboardAPIClient } from "@empiricalrun/shared-types/api/base";
+import type { ChatSessionInfo, Logger, MessageCheckpoint, Tool, ToolExecute } from "@empiricalrun/shared-types/chat-agent";
 export type BaseToolExecutorProps = {
     tools: Array<Tool>;
     chatSession: ChatSessionInfo | {
@@ -14,7 +15,7 @@ export type BaseToolExecutorProps = {
     logger?: Logger;
 };
 export declare class BaseToolExecutor {
-    toolExecutors: Record<string, ToolExecute>;
+    toolExecutors: Record<string, Tool | ToolExecute>;
     featureFlags: string[];
     chatSession: ChatSessionInfo | null;
     branchName: string | null;

package/dist/tools/executor/base.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"base.d.ts","sourceRoot":"","sources":["../../../src/tools/executor/base.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAE,eAAe,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AACrE,OAAO,KAAK,EACV,eAAe,EAEf,mBAAmB,EACnB,MAAM,EACN,iBAAiB,EACjB,IAAI,EACJ,WAAW,EACZ,MAAM,4BAA4B,CAAC;AAapC,MAAM,MAAM,qBAAqB,GAAG;IAClC,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;IACnB,WAAW,EAAE,eAAe,GAAG;QAAE,UAAU,EAAE,MAAM,CAAA;KAAE,GAAG,IAAI,CAAC;IAC7D,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,CAAC,EAAE,mBAAmB,CAAC;IAChC,KAAK,CAAC,EAAE,WAAW,CAAC;IACpB,YAAY,EAAE,MAAM,EAAE,CAAC;IACvB,oBAAoB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC9C,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB,CAAC;AAEF,qBAAa,gBAAgB;IAC3B,aAAa,EAAE,MAAM,CAAC,MAAM,EAAE,WAAW,CAAC,CAAM;IAChD,YAAY,EAAE,MAAM,EAAE,CAAC;IACvB,WAAW,EAAE,eAAe,GAAG,IAAI,CAAC;IACpC,UAAU,EAAE,MAAM,GAAG,IAAI,CAAC;IAC1B,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,EAAE,mBAAmB,GAAG,SAAS,CAAC;IAC3C,KAAK,CAAC,EAAE,WAAW,CAAC;IACpB,oBAAoB,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC7C,MAAM,EAAE,MAAM,CAAC;gBAEH,EACV,KAAK,EACL,WAAW,EACX,QAAQ,EACR,SAAS,EACT,KAAK,EACL,YAAY,EACZ,oBAAoB,EACpB,MAAM,GACP,EAAE,qBAAqB;IAuBlB,OAAO,CAAC,SAAS,EAAE,eAAe,EAAE,GAAG,OAAO,CAAC;QACnD,WAAW,EAAE,UAAU,EAAE,CAAC;QAC1B,UAAU,EAAE,iBAAiB,GAAG,IAAI,CAAC;KACtC,CAAC;CAiFH"}
+{"version":3,"file":"base.d.ts","sourceRoot":"","sources":["../../../src/tools/executor/base.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AACrD,OAAO,EAAE,eAAe,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AACrE,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,qCAAqC,CAAC;AAC/E,OAAO,KAAK,EACV,eAAe,EAEf,MAAM,EACN,iBAAiB,EACjB,IAAI,EACJ,WAAW,EACZ,MAAM,uCAAuC,CAAC;AAc/C,MAAM,MAAM,qBAAqB,GAAG;IAClC,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;IACnB,WAAW,EAAE,eAAe,GAAG;QAAE,UAAU,EAAE,MAAM,CAAA;KAAE,GAAG,IAAI,CAAC;IAC7D,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,CAAC,EAAE,mBAAmB,CAAC;IAChC,KAAK,CAAC,EAAE,WAAW,CAAC;IACpB,YAAY,EAAE,MAAM,EAAE,CAAC;IACvB,oBAAoB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC9C,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB,CAAC;AAEF,qBAAa,gBAAgB;IAC3B,aAAa,EAAE,MAAM,CAAC,MAAM,EAAE,IAAI,GAAG,WAAW,CAAC,CAAM;IACvD,YAAY,EAAE,MAAM,EAAE,CAAC;IACvB,WAAW,EAAE,eAAe,GAAG,IAAI,CAAC;IACpC,UAAU,EAAE,MAAM,GAAG,IAAI,CAAC;IAC1B,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,EAAE,mBAAmB,GAAG,SAAS,CAAC;IAC3C,KAAK,CAAC,EAAE,WAAW,CAAC;IACpB,oBAAoB,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC7C,MAAM,EAAE,MAAM,CAAC;gBAEH,EACV,KAAK,EACL,WAAW,EACX,QAAQ,EACR,SAAS,EACT,KAAK,EACL,YAAY,EACZ,oBAAoB,EACpB,MAAM,GACP,EAAE,qBAAqB;IAuBlB,OAAO,CAAC,SAAS,EAAE,eAAe,EAAE,GAAG,OAAO,CAAC;QACnD,WAAW,EAAE,UAAU,EAAE,CAAC;QAC1B,UAAU,EAAE,iBAAiB,GAAG,IAAI,CAAC;KACtC,CAAC;CAsGH"}

package/dist/tools/executor/base.js

@@ -3,6 +3,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.BaseToolExecutor = void 0;
 const artifacts_1 = require("../../artifacts");
 const file_operations_1 = require("../file-operations");
+const validate_schema_1 = require("../utils/validate-schema");
 const checkpoint_1 = require("./utils/checkpoint");
 const LOGGER = {
     info: (msg, meta) => console.log(`[info] ${msg}`, meta || ""),
@@ -31,7 +32,7 @@ class BaseToolExecutor {
         // Build toolExecutors
         let toolExecutors = {};
         tools.forEach((tool) => {
-            toolExecutors[tool.schema.name] = tool.execute;
+            toolExecutors[tool.schema.name] = tool;
         });
         this.toolExecutors = {
             ...toolExecutors,
@@ -71,8 +72,21 @@ class BaseToolExecutor {
                 span?.end({ output: errorResult });
                 continue;
             }
+            const toolSchema = typeof toolExecutor === "function" ? null : toolExecutor.schema;
+            const toolInputSchema = toolSchema?.parameters;
+            const executeFn = typeof toolExecutor === "function"
+                ? toolExecutor
+                : toolExecutor.execute;
+            if (toolInputSchema) {
+                const validationErrorToolresult = (0, validate_schema_1.validateToolInput)(toolCall, toolInputSchema);
+                if (validationErrorToolresult) {
+                    toolResults.push(validationErrorToolresult);
+                    span?.end({ output: validationErrorToolresult });
+                    continue;
+                }
+            }
             try {
-                const result = await toolExecutor({
+                const result = await executeFn({
                     input: toolCall.input,
                     repoPath: this.repoPath,
                     apiClient: this.apiClient,

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

@@ -1,4 +1,7 @@
 import { BaseToolExecutor, BaseToolExecutorProps } from "./base";
+export type { BaseToolExecutorProps } from "./base";
+export { BaseToolExecutor } from "./base";
+export { gitPushChanges } from "./utils/git";
 export declare class ToolExecutor extends BaseToolExecutor {
     constructor(params: Omit<BaseToolExecutorProps, "tools">);
 }

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

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/tools/executor/index.ts"],"names":[],"mappings":"AA0BA,OAAO,EAAE,gBAAgB,EAAE,qBAAqB,EAAE,MAAM,QAAQ,CAAC;AAEjE,qBAAa,YAAa,SAAQ,gBAAgB;gBACpC,MAAM,EAAE,IAAI,CAAC,qBAAqB,EAAE,OAAO,CAAC;CAgCzD"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/tools/executor/index.ts"],"names":[],"mappings":"AAwBA,OAAO,EAAE,gBAAgB,EAAE,qBAAqB,EAAE,MAAM,QAAQ,CAAC;AAEjE,YAAY,EAAE,qBAAqB,EAAE,MAAM,QAAQ,CAAC;AACpD,OAAO,EAAE,gBAAgB,EAAE,MAAM,QAAQ,CAAC;AAC1C,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAE7C,qBAAa,YAAa,SAAQ,gBAAgB;gBACpC,MAAM,EAAE,IAAI,CAAC,qBAAqB,EAAE,OAAO,CAAC;CA8BzD"}

package/dist/tools/executor/index.js

@@ -1,17 +1,16 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.ToolExecutor = void 0;
+exports.ToolExecutor = exports.gitPushChanges = exports.BaseToolExecutor = void 0;
 const analyse_video_1 = require("../analyse-video");
 const create_pull_request_1 = require("../create-pull-request");
 const delete_file_1 = require("../delete-file");
 const diagnosis_fetcher_1 = require("../diagnosis-fetcher");
 const download_build_1 = require("../download-build");
 const fetch_file_1 = require("../fetch-file");
-const fetch_last_successful_test_run_1 = require("../fetch-last-successful-test-run");
 const fetch_session_diff_1 = require("../fetch-session-diff");
 const file_operations_1 = require("../file-operations");
 const grep_1 = require("../grep");
-const issues_1 = require("../issues");
+const issues_v1_1 = require("../issues-v1");
 const list_environments_1 = require("../list-environments");
 const list_tests_and_projects_1 = require("../list-tests-and-projects");
 const merge_conflicts_1 = require("../merge-conflicts");
@@ -23,22 +22,25 @@ const test_run_fetcher_1 = require("../test-run-fetcher");
 const trace_dot_zip_1 = require("../trace-dot-zip");
 const triage_summary_1 = require("../triage-summary");
 const upgrade_packages_1 = require("../upgrade-packages");
-const view_failed_test_run_report_1 = require("../view-failed-test-run-report");
 const base_1 = require("./base");
+var base_2 = require("./base");
+Object.defineProperty(exports, "BaseToolExecutor", { enumerable: true, get: function () { return base_2.BaseToolExecutor; } });
+var git_1 = require("./utils/git");
+Object.defineProperty(exports, "gitPushChanges", { enumerable: true, get: function () { return git_1.gitPushChanges; } });
 class ToolExecutor extends base_1.BaseToolExecutor {
     constructor(params) {
         // Found for "Tool Executor Tools List" and add any new tools to the list
         const tools = [
             grep_1.grepTool,
             run_test_1.runTestTool,
-            test_run_fetcher_1.fetchTestRunReportTool,
+            test_run_fetcher_1.fetchTestRunDetailsTool,
             diagnosis_fetcher_1.fetchDiagnosisReportTool,
             test_gen_browser_1.generateTestWithBrowserAgent,
             create_pull_request_1.createPullRequestTool,
             merge_conflicts_1.mergeConflictsTool,
             list_environments_1.listEnvironmentsTool,
-            issues_1.listIssuesTool,
-            issues_1.createIssueTool,
+            issues_v1_1.listIssuesTool,
+            issues_v1_1.createIssueTool,
             download_build_1.downloadBuildTool,
             upgrade_packages_1.upgradePackagesTool,
             fetch_file_1.fetchFileTool,
@@ -46,9 +48,7 @@ class ToolExecutor extends base_1.BaseToolExecutor {
             delete_file_1.deleteFileTool,
             rename_file_1.renameFileTool,
             trace_dot_zip_1.traceDotZipTool,
-            view_failed_test_run_report_1.viewFailedTestRunReportTool,
-            issues_1.updateIssueTool,
-            fetch_last_successful_test_run_1.fetchLastSuccessfulTestRunTool,
+            issues_v1_1.updateIssueTool,
             triage_summary_1.sendTriageSummaryTool,
             list_tests_and_projects_1.listProjectsTool,
             fetch_session_diff_1.fetchSessionDiffTool,

package/dist/tools/executor/utils/checkpoint.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"checkpoint.d.ts","sourceRoot":"","sources":["../../../../src/tools/executor/utils/checkpoint.ts"],"names":[],"mappings":"AAIA,wBAAgB,sBAAsB,CAAC,EACrC,UAAU,EACV,QAAQ,GACT,EAAE;IACD,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;CAClB,GAAG,MAAM,GAAG,IAAI,CA2BhB"}
+{"version":3,"file":"checkpoint.d.ts","sourceRoot":"","sources":["../../../../src/tools/executor/utils/checkpoint.ts"],"names":[],"mappings":"AAEA,wBAAgB,sBAAsB,CAAC,EACrC,UAAU,EACV,QAAQ,GACT,EAAE;IACD,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;CAClB,GAAG,MAAM,GAAG,IAAI,CAwBhB"}

package/dist/tools/executor/utils/checkpoint.js

@@ -1,7 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.createCommitCheckpoint = createCommitCheckpoint;
-const child_process_1 = require("child_process");
 const git_1 = require("./git");
 function createCommitCheckpoint({ branchName, repoPath, }) {
     const filesChanged = (0, git_1.getStagedFiles)(repoPath);
@@ -21,9 +20,6 @@ function createCommitCheckpoint({ branchName, repoPath, }) {
         const commitMessage = commitMessages.join(", ");
         // Files are already staged by individual tools, just commit
         const commitSha = (0, git_1.commitAsBotUser)(commitMessage, repoPath);
-        if (commitSha) {
-            (0, child_process_1.execSync)(`git push origin ${branchName}`, { cwd: repoPath });
-        }
         return commitSha;
     }
     return null;

package/dist/tools/executor/utils/git.d.ts

@@ -1,4 +1,6 @@
-import { IDashboardAPIClient } from "@empiricalrun/shared-types";
+import { IDashboardAPIClient } from "@empiricalrun/shared-types/api/base";
+export declare const GIT_USER_NAME = "empiricalrun[bot]";
+export declare const GIT_USER_EMAIL = "180257021+empiricalrun[bot]@users.noreply.github.com";
 export declare function getGitDiff(filepath: string, cwd: string): string;
 export declare function getGitDiffStaged(filepath: string, cwd: string): string;
 export declare function getGitDiffForStagedOrUnstaged(filepath: string, cwd: string): string;
@@ -24,4 +26,8 @@ export declare function mergePullRequest({ repoName, prNumber, apiClient, }: {
     prNumber: number;
     apiClient: IDashboardAPIClient;
 }): Promise<boolean>;
+export declare function gitPushChanges({ branch, workingDirectory, }: {
+    branch: string;
+    workingDirectory: string;
+}): void;
 //# sourceMappingURL=git.d.ts.map

package/dist/tools/executor/utils/git.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"git.d.ts","sourceRoot":"","sources":["../../../../src/tools/executor/utils/git.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,mBAAmB,EAAE,MAAM,4BAA4B,CAAC;AAMjE,wBAAgB,UAAU,CAAC,QAAQ,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,GAAG,MAAM,CAMhE;AAED,wBAAgB,gBAAgB,CAAC,QAAQ,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,GAAG,MAAM,CAMtE;AAED,wBAAgB,6BAA6B,CAC3C,QAAQ,EAAE,MAAM,EAChB,GAAG,EAAE,MAAM,GACV,MAAM,CAaR;AAED,wBAAgB,oBAAoB,CAAC,QAAQ,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,GAAG,MAAM,CAe1E;AAED,wBAAsB,cAAc,CAAC,UAAU,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,iBAMnE;AAED,wBAAgB,eAAe,CAC7B,aAAa,EAAE,MAAM,EACrB,GAAG,EAAE,MAAM,GACV,MAAM,GAAG,IAAI,CAkBf;AAED,wBAAsB,oBAAoB,CAAC,QAAQ,EAAE,MAAM,mBAO1D;AAED,wBAAsB,UAAU,CAAC,UAAU,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,iBAE/D;AAED,wBAAgB,4BAA4B,CAAC,EAC3C,aAAa,EACb,KAAK,EACL,QAAQ,GACT,EAAE;IACD,aAAa,EAAE,MAAM,CAAC;IACtB,KAAK,EAAE,MAAM,EAAE,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC;CAClB,GAAG,MAAM,GAAG,IAAI,CAGhB;AAED,MAAM,MAAM,UAAU,GAAG;IACvB,IAAI,EAAE,OAAO,GAAG,UAAU,GAAG,SAAS,GAAG,SAAS,CAAC;IACnD,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB,CAAC;AAEF,wBAAgB,cAAc,CAAC,GAAG,EAAE,MAAM,GAAG,UAAU,EAAE,CA6CxD;AAED,wBAAgB,eAAe,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,EAAE,CAUrD;AAED,wBAAsB,gBAAgB,CAAC,EACrC,QAAQ,EACR,QAAQ,EACR,SAAS,GACV,EAAE;IACD,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,EAAE,mBAAmB,CAAC;CAChC,oBA8BA"}
+{"version":3,"file":"git.d.ts","sourceRoot":"","sources":["../../../../src/tools/executor/utils/git.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,mBAAmB,EAAE,MAAM,qCAAqC,CAAC;AAG1E,eAAO,MAAM,aAAa,sBAAsB,CAAC;AACjD,eAAO,MAAM,cAAc,yDAC6B,CAAC;AAEzD,wBAAgB,UAAU,CAAC,QAAQ,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,GAAG,MAAM,CAMhE;AAED,wBAAgB,gBAAgB,CAAC,QAAQ,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,GAAG,MAAM,CAMtE;AAED,wBAAgB,6BAA6B,CAC3C,QAAQ,EAAE,MAAM,EAChB,GAAG,EAAE,MAAM,GACV,MAAM,CAaR;AAED,wBAAgB,oBAAoB,CAAC,QAAQ,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,GAAG,MAAM,CAe1E;AAED,wBAAsB,cAAc,CAAC,UAAU,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,iBAMnE;AAED,wBAAgB,eAAe,CAC7B,aAAa,EAAE,MAAM,EACrB,GAAG,EAAE,MAAM,GACV,MAAM,GAAG,IAAI,CAkBf;AAED,wBAAsB,oBAAoB,CAAC,QAAQ,EAAE,MAAM,mBAO1D;AAED,wBAAsB,UAAU,CAAC,UAAU,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,iBAE/D;AAED,wBAAgB,4BAA4B,CAAC,EAC3C,aAAa,EACb,KAAK,EACL,QAAQ,GACT,EAAE;IACD,aAAa,EAAE,MAAM,CAAC;IACtB,KAAK,EAAE,MAAM,EAAE,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC;CAClB,GAAG,MAAM,GAAG,IAAI,CAGhB;AAED,MAAM,MAAM,UAAU,GAAG;IACvB,IAAI,EAAE,OAAO,GAAG,UAAU,GAAG,SAAS,GAAG,SAAS,CAAC;IACnD,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB,CAAC;AAEF,wBAAgB,cAAc,CAAC,GAAG,EAAE,MAAM,GAAG,UAAU,EAAE,CA6CxD;AAED,wBAAgB,eAAe,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,EAAE,CAUrD;AAED,wBAAsB,gBAAgB,CAAC,EACrC,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,wBAAgB,cAAc,CAAC,EAC7B,MAAM,EACN,gBAAgB,GACjB,EAAE;IACD,MAAM,EAAE,MAAM,CAAC;IACf,gBAAgB,EAAE,MAAM,CAAC;CAC1B,QAIA"}

package/dist/tools/executor/utils/git.js

@@ -1,5 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
+exports.GIT_USER_EMAIL = exports.GIT_USER_NAME = void 0;
 exports.getGitDiff = getGitDiff;
 exports.getGitDiffStaged = getGitDiffStaged;
 exports.getGitDiffForStagedOrUnstaged = getGitDiffForStagedOrUnstaged;
@@ -12,9 +13,10 @@ exports.stageAndCommitFilesAsBotUser = stageAndCommitFilesAsBotUser;
 exports.getStagedFiles = getStagedFiles;
 exports.getFilesChanged = getFilesChanged;
 exports.mergePullRequest = mergePullRequest;
+exports.gitPushChanges = gitPushChanges;
 const child_process_1 = require("child_process");
-const GIT_USER_NAME = "empiricalrun[bot]";
-const GIT_USER_EMAIL = "180257021+empiricalrun[bot]@users.noreply.github.com";
+exports.GIT_USER_NAME = "empiricalrun[bot]";
+exports.GIT_USER_EMAIL = "180257021+empiricalrun[bot]@users.noreply.github.com";
 function getGitDiff(filepath, cwd) {
     const diff = (0, child_process_1.execSync)(`git diff ${filepath}`, {
         encoding: "utf-8",
@@ -77,7 +79,7 @@ function commitAsBotUser(commitMessage, cwd) {
         return null;
     }
     const commitMessageWithSkipCi = `${commitMessage} [skip ci]`;
-    (0, child_process_1.execSync)(`git -c user.name="${GIT_USER_NAME}" -c user.email="${GIT_USER_EMAIL}" commit -m "${commitMessageWithSkipCi}"`, { cwd });
+    (0, child_process_1.execSync)(`git -c user.name="${exports.GIT_USER_NAME}" -c user.email="${exports.GIT_USER_EMAIL}" commit -m "${commitMessageWithSkipCi}"`, { cwd });
     const commitSha = (0, child_process_1.execSync)("git rev-parse HEAD", {
         encoding: "utf-8",
         cwd,
@@ -179,3 +181,8 @@ async function mergePullRequest({ repoName, prNumber, apiClient, }) {
         return false;
     }
 }
+function gitPushChanges({ branch, workingDirectory, }) {
+    (0, child_process_1.execSync)(`git push origin ${branch}`, {
+        cwd: workingDirectory,
+    });
+}

package/dist/tools/executor/utils/index.d.ts

@@ -1,4 +1,4 @@
-import { IDashboardAPIClient } from "@empiricalrun/shared-types";
+import { IDashboardAPIClient } from "@empiricalrun/shared-types/api/base";
 export interface GitHubRepoInfo {
     owner: string;
     repo: string;
@@ -50,7 +50,7 @@ export declare function createPullRequest({ owner, repo, title, head, base, body
     repo: string;
     title: string;
     head: string;
-    base?: string;
+    base: string;
     body: string;
     labels?: string[];
     apiClient: IDashboardAPIClient;
@@ -62,12 +62,13 @@ export declare function updatePullRequest({ owner, repo, prNumber, body, apiClie
     body?: string;
     apiClient: IDashboardAPIClient;
 }): Promise<PullRequestInfo>;
-export declare function findOrCreatePullRequest({ owner, repo, branchName, title, body, labels, apiClient, }: {
+export declare function findOrCreatePullRequest({ owner, repo, branchName, title, body, base, labels, apiClient, }: {
     owner: string;
     repo: string;
     branchName: string;
     title: string;
     body: string;
+    base: string;
     labels?: string[];
     apiClient: IDashboardAPIClient;
 }): Promise<{

package/dist/tools/executor/utils/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/tools/executor/utils/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,mBAAmB,EAAE,MAAM,4BAA4B,CAAC;AAEjE,MAAM,WAAW,cAAc;IAC7B,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;CACd;AAED,wBAAgB,cAAc,CAAC,GAAG,EAAE,MAAM,GAAG,cAAc,CAiB1D;AAED,MAAM,WAAW,eAAe;IAC9B,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE;QACJ,GAAG,EAAE,MAAM,CAAC;QACZ,GAAG,EAAE,MAAM,CAAC;KACb,CAAC;IACF,IAAI,EAAE;QACJ,GAAG,EAAE,MAAM,CAAC;QACZ,GAAG,EAAE,MAAM,CAAC;KACb,CAAC;IACF,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;IAClB,gBAAgB,EAAE,MAAM,CAAC;IACzB,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;IAClB,aAAa,EAAE,MAAM,CAAC;IACtB,IAAI,EAAE;QACJ,KAAK,EAAE,MAAM,CAAC;KACf,CAAC;IACF,eAAe,EAAE,OAAO,GAAG,OAAO,GAAG,UAAU,GAAG,SAAS,CAAC;CAC7D;AAED,wBAAsB,aAAa,CAAC,EAClC,KAAK,EACL,IAAI,EACJ,UAAU,EACV,SAAS,GACV,EAAE;IACD,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE,mBAAmB,CAAC;CAChC,GAAG,OAAO,CAAC,eAAe,GAAG,IAAI,CAAC,CAUlC;AAED,wBAAsB,iBAAiB,CAAC,EACtC,KAAK,EACL,IAAI,EACJ,WAAW,EACX,SAAS,GACV,EAAE;IACD,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,eAAe,CAAC;IAC7B,SAAS,EAAE,mBAAmB,CAAC;CAChC,GAAG,OAAO,CAAC,OAAO,GAAG,OAAO,GAAG,UAAU,GAAG,SAAS,CAAC,CAyBtD;AAED,wBAAgB,4BAA4B,CAC1C,KAAK,EAAE,OAAO,GAAG,OAAO,GAAG,UAAU,GAAG,SAAS,GAAG,MAAM,GACzD,MAAM,CAaR;AAED,wBAAsB,iBAAiB,CAAC,EACtC,KAAK,EACL,IAAI,EACJ,KAAK,EACL,IAAI,EACJ,IAAI,EACJ,IAAI,EACJ,MAAM,EACN,SAAS,GACV,EAAE;IACD,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC;IAClB,SAAS,EAAE,mBAAmB,CAAC;CAChC,GAAG,OAAO,CAAC,eAAe,CAAC,CAoB3B;AAED,wBAAsB,iBAAiB,CAAC,EACtC,KAAK,EACL,IAAI,EACJ,QAAQ,EACR,IAAI,EACJ,SAAS,GACV,EAAE;IACD,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,mBAAmB,CAAC;CAChC,GAAG,OAAO,CAAC,eAAe,CAAC,CAS3B;AAED,wBAAsB,uBAAuB,CAAC,EAC5C,KAAK,EACL,IAAI,EACJ,UAAU,EACV,KAAK,EACL,IAAI,EACJ,MAAM,EACN,SAAS,GACV,EAAE;IACD,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,EAAE,MAAM,CAAC;IACnB,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC;IAClB,SAAS,EAAE,mBAAmB,CAAC;CAChC,GAAG,OAAO,CAAC;IAAE,EAAE,EAAE,eAAe,CAAC;IAAC,KAAK,EAAE,OAAO,CAAA;CAAE,CAAC,CAoBnD"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/tools/executor/utils/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,mBAAmB,EAAE,MAAM,qCAAqC,CAAC;AAE1E,MAAM,WAAW,cAAc;IAC7B,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;CACd;AAED,wBAAgB,cAAc,CAAC,GAAG,EAAE,MAAM,GAAG,cAAc,CAiB1D;AAED,MAAM,WAAW,eAAe;IAC9B,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE;QACJ,GAAG,EAAE,MAAM,CAAC;QACZ,GAAG,EAAE,MAAM,CAAC;KACb,CAAC;IACF,IAAI,EAAE;QACJ,GAAG,EAAE,MAAM,CAAC;QACZ,GAAG,EAAE,MAAM,CAAC;KACb,CAAC;IACF,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;IAClB,gBAAgB,EAAE,MAAM,CAAC;IACzB,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;IAClB,aAAa,EAAE,MAAM,CAAC;IACtB,IAAI,EAAE;QACJ,KAAK,EAAE,MAAM,CAAC;KACf,CAAC;IACF,eAAe,EAAE,OAAO,GAAG,OAAO,GAAG,UAAU,GAAG,SAAS,CAAC;CAC7D;AAED,wBAAsB,aAAa,CAAC,EAClC,KAAK,EACL,IAAI,EACJ,UAAU,EACV,SAAS,GACV,EAAE;IACD,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE,mBAAmB,CAAC;CAChC,GAAG,OAAO,CAAC,eAAe,GAAG,IAAI,CAAC,CAUlC;AAED,wBAAsB,iBAAiB,CAAC,EACtC,KAAK,EACL,IAAI,EACJ,WAAW,EACX,SAAS,GACV,EAAE;IACD,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,eAAe,CAAC;IAC7B,SAAS,EAAE,mBAAmB,CAAC;CAChC,GAAG,OAAO,CAAC,OAAO,GAAG,OAAO,GAAG,UAAU,GAAG,SAAS,CAAC,CAyBtD;AAED,wBAAgB,4BAA4B,CAC1C,KAAK,EAAE,OAAO,GAAG,OAAO,GAAG,UAAU,GAAG,SAAS,GAAG,MAAM,GACzD,MAAM,CAaR;AAED,wBAAsB,iBAAiB,CAAC,EACtC,KAAK,EACL,IAAI,EACJ,KAAK,EACL,IAAI,EACJ,IAAI,EACJ,IAAI,EACJ,MAAM,EACN,SAAS,GACV,EAAE;IACD,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC;IAClB,SAAS,EAAE,mBAAmB,CAAC;CAChC,GAAG,OAAO,CAAC,eAAe,CAAC,CAoB3B;AAED,wBAAsB,iBAAiB,CAAC,EACtC,KAAK,EACL,IAAI,EACJ,QAAQ,EACR,IAAI,EACJ,SAAS,GACV,EAAE;IACD,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,mBAAmB,CAAC;CAChC,GAAG,OAAO,CAAC,eAAe,CAAC,CAS3B;AAED,wBAAsB,uBAAuB,CAAC,EAC5C,KAAK,EACL,IAAI,EACJ,UAAU,EACV,KAAK,EACL,IAAI,EACJ,IAAI,EACJ,MAAM,EACN,SAAS,GACV,EAAE;IACD,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,EAAE,MAAM,CAAC;IACnB,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC;IAClB,SAAS,EAAE,mBAAmB,CAAC;CAChC,GAAG,OAAO,CAAC;IAAE,EAAE,EAAE,eAAe,CAAC;IAAC,KAAK,EAAE,OAAO,CAAA;CAAE,CAAC,CAqBnD"}

package/dist/tools/executor/utils/index.js

@@ -98,7 +98,7 @@ async function updatePullRequest({ owner, repo, prNumber, body, apiClient, }) {
         body: updateBody,
     });
 }
-async function findOrCreatePullRequest({ owner, repo, branchName, title, body, labels, apiClient, }) {
+async function findOrCreatePullRequest({ owner, repo, branchName, title, body, base, labels, apiClient, }) {
     const existingPR = await getExistingPR({
         owner,
         repo,
@@ -113,6 +113,7 @@ async function findOrCreatePullRequest({ owner, repo, branchName, title, body, l
         repo,
         title,
         head: branchName,
+        base,
         body,
         labels,
         apiClient,

package/dist/tools/executor/utils/pr-description.d.ts

@@ -1,4 +1,4 @@
-import type { ChatSessionInfo } from "@empiricalrun/shared-types";
+import type { ChatSessionInfo } from "@empiricalrun/shared-types/chat-agent";
 export declare function generateSessionMetadata(chatSession: ChatSessionInfo | null): string;
 export declare function addMetadataToPRDescription(description: string, chatSession: ChatSessionInfo | null | undefined): string;
 //# sourceMappingURL=pr-description.d.ts.map

package/dist/tools/executor/utils/pr-description.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"pr-description.d.ts","sourceRoot":"","sources":["../../../../src/tools/executor/utils/pr-description.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,4BAA4B,CAAC;AAElE,wBAAgB,uBAAuB,CACrC,WAAW,EAAE,eAAe,GAAG,IAAI,GAClC,MAAM,CASR;AAED,wBAAgB,0BAA0B,CACxC,WAAW,EAAE,MAAM,EACnB,WAAW,EAAE,eAAe,GAAG,IAAI,GAAG,SAAS,GAC9C,MAAM,CAWR"}
+{"version":3,"file":"pr-description.d.ts","sourceRoot":"","sources":["../../../../src/tools/executor/utils/pr-description.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,uCAAuC,CAAC;AAE7E,wBAAgB,uBAAuB,CACrC,WAAW,EAAE,eAAe,GAAG,IAAI,GAClC,MAAM,CASR;AAED,wBAAgB,0BAA0B,CACxC,WAAW,EAAE,MAAM,EACnB,WAAW,EAAE,eAAe,GAAG,IAAI,GAAG,SAAS,GAC9C,MAAM,CAWR"}

package/dist/tools/fetch-file/index.d.ts

@@ -1,4 +1,4 @@
-import { Tool } from "@empiricalrun/shared-types";
+import { Tool } from "@empiricalrun/shared-types/chat-agent";
 import { z } from "zod";
 declare const fetchFileSchema: z.ZodObject<{
     url: z.ZodString;

package/dist/tools/fetch-file/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/tools/fetch-file/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,IAAI,EAGL,MAAM,4BAA4B,CAAC;AACpC,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAUxB,QAAA,MAAM,eAAe;;;;;;EAEnB,CAAC;AAEH,eAAO,MAAM,aAAa,EAAE,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,eAAe,CAAC,CAyF/D,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/tools/fetch-file/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,IAAI,EAGL,MAAM,uCAAuC,CAAC;AAC/C,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAYxB,QAAA,MAAM,eAAe;;;;;;EAEnB,CAAC;AAEH,eAAO,MAAM,aAAa,EAAE,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,eAAe,CAAC,CAoI/D,CAAC"}

package/dist/tools/fetch-file/index.js

@@ -2,6 +2,8 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.fetchFileTool = void 0;
 const zod_1 = require("zod");
+const url_validation_1 = require("../../utils/url-validation");
+const utils_1 = require("./utils");
 const SUPPORTED_IMAGE_TYPES = [
     "image/png",
     "image/jpeg",
@@ -16,7 +18,7 @@ exports.fetchFileTool = {
     schema: {
         name: "fetchFile",
         description: `Use this tool to fetch file data from any valid URL that responds with an image (PNG, JPEG, GIF, WebP) or markdown file.
-For images, it returns the image in base64 format for you to view or analyze. For markdown files, it returns the text content for you to read and process.
+For images, it returns the image in base64 format for you to view or analyse. For markdown files, it returns the text content for you to read and process.
 
 ## Caveats
 1. This will not work to fetch markdown files from a repo due to access control issues. Use file view tools for that.
@@ -30,20 +32,32 @@ time of test failures. Both of these are available in the attachments section of
     },
     needsBrowser: false,
     isInlineTool: true,
-    execute: async ({ input }) => {
+    execute: async ({ input, logger, }) => {
         const { url } = input;
+        logger.info("Starting file fetch", {
+            tool: "fetchFileTool",
+            url,
+        });
         try {
-            const response = await fetch(url);
-            if (!response.ok) {
-                console.error(`Failed to fetch file from ${url}: ${response.statusText}`);
+            const headResponse = await (0, url_validation_1.validateUrlAccess)(url);
+            if (!headResponse.ok) {
+                logger.error("Failed to fetch file", {
+                    tool: "fetchFileTool",
+                    url,
+                    statusText: headResponse.statusText,
+                });
                 return {
-                    result: `Failed to fetch file from ${url}: ${response.statusText}`,
+                    result: `Failed to fetch file from ${url}: ${headResponse.statusText}`,
                     isError: true,
                 };
             }
-            const contentType = response.headers.get("content-type");
+            const contentType = headResponse.headers.get("content-type");
             if (!contentType) {
                 const errorMessage = `No content type header found. URL must return an image or markdown file.`;
+                logger.warn("No content type header found", {
+                    tool: "fetchFileTool",
+                    url,
+                });
                 return {
                     result: errorMessage,
                     isError: true,
@@ -51,17 +65,33 @@ time of test failures. Both of these are available in the attachments section of
             }
             const isImage = SUPPORTED_IMAGE_TYPES.some((type) => contentType.startsWith(type));
             const isText = SUPPORTED_TEXT_TYPES.some((type) => contentType.startsWith(type));
+            const isVideo = contentType.startsWith("video");
             if (!isImage && !isText) {
-                const errorMessage = `Invalid content type: ${contentType}. URL must return an image (PNG, JPEG, GIF, WebP) or markdown file.`;
-                console.error(errorMessage);
+                const errorMessage = isVideo
+                    ? `This tool does not support content type: ${contentType}. Try analyseVideo tool instead for video files.`
+                    : `Invalid content type: ${contentType}. URL must return an image (PNG, JPEG, GIF, WebP) or markdown file.`;
+                logger.warn("Unsupported content type", {
+                    tool: "fetchFileTool",
+                    url,
+                    contentType,
+                    isVideo,
+                });
                 return {
                     result: errorMessage,
                     isError: true,
                 };
             }
+            const contentResponse = await fetch(url);
             if (isImage) {
-                const buffer = await response.arrayBuffer();
-                const base64 = Buffer.from(buffer).toString("base64");
+                const buffer = await contentResponse.arrayBuffer();
+                const processedBuffer = await (0, utils_1.downscaleImageIfNeeded)(buffer, logger);
+                const base64 = Buffer.from(processedBuffer).toString("base64");
+                logger.info("Successfully fetched image file", {
+                    tool: "fetchFileTool",
+                    url,
+                    contentType,
+                    size: processedBuffer.byteLength,
+                });
                 return {
                     result: [
                         {
@@ -74,7 +104,13 @@ time of test failures. Both of these are available in the attachments section of
             }
             else {
                 // Handle text/markdown files
-                const text = await response.text();
+                const text = await contentResponse.text();
+                logger.info("Successfully fetched text file", {
+                    tool: "fetchFileTool",
+                    url,
+                    contentType,
+                    size: text.length,
+                });
                 return {
                     result: [
                         {
@@ -87,7 +123,11 @@ time of test failures. Both of these are available in the attachments section of
             }
         }
         catch (error) {
-            console.error("Error fetching file", error);
+            logger.error("Error fetching file", {
+                tool: "fetchFileTool",
+                url,
+                error,
+            });
             return {
                 result: `Error fetching file: ${error}`,
                 isError: true,

package/dist/tools/fetch-file/utils.d.ts

@@ -0,0 +1,3 @@
+import { Logger } from "@empiricalrun/shared-types/chat-agent";
+export declare function downscaleImageIfNeeded(buffer: ArrayBuffer, logger: Logger): Promise<ArrayBuffer>;
+//# sourceMappingURL=utils.d.ts.map