@empiricalrun/test-gen 0.71.2 → 0.72.0

package/CHANGELOG.md

@@ -1,5 +1,43 @@
 # @empiricalrun/test-gen
 
+## 0.72.0
+
+### Minor Changes
+
+- 6ac65ed: feat: add custom sqs client with bug fix and error boundary
+- 23708d1: feat: add agent-worker app to run chat agent over durable objects
+
+### Patch Changes
+
+- 6b1d98c: feat: trigger impacted tests computation on tool-responses
+- b6a04f5: feat: use webocket to get realtime updates on session
+- 4cbc287: feat: add (de)compress layer in the worker message management
+- a96a03c: chore: follow-ups from agent-worker first land
+- 32905df: chore: move dashboard methods from tool-execute-service to test-gen
+- 8e3c7a4: feat: working agent-worker for text messages
+- 5b0d43a: chore: logging tool call flow between agent-worker and dashboard
+- fc6f97c: fix: execSync handling for ubuntu and windows ci
+- 440e851: feat: extended thinking in claude
+- a23b38f: chore: add worker env in createChatModel & ToolCaller methods
+- 12c69cc: fix: error handling for non-retryable errors
+- 5ed01c4: fix: stop should reset askUserForInput state
+- e1d01c8: feat: add file info builder for github in test-gen
+- 450b79a: feat: show resolved projects for environments in cli
+- 7009d67: feat: show tool result preview in chat ui
+- d570c55: feat: add reporter function in agent worker
+- 622aa35: chore: split tool call service into caller and executor
+- 162e461: chore: file info clean up
+- 349003e: feat: working tool calls on agent-worker
+- d5c7696: test: add agent loop test harness
+- 7afa5c1: feat: better error handling & state updates, add langfuse trace id to worker session state
+- Updated dependencies [df226a5]
+- Updated dependencies [440e851]
+- Updated dependencies [a23b38f]
+- Updated dependencies [450b79a]
+- Updated dependencies [d5c7696]
+  - @empiricalrun/llm@0.20.0
+  - @empiricalrun/test-run@0.10.9
+
 ## 0.71.2
 
 ### Patch Changes

package/dist/agent/chat/agent-loop.d.ts

@@ -1,17 +1,19 @@
 import { TraceClient } from "@empiricalrun/llm";
 import { IChatModel } from "@empiricalrun/llm/chat";
-import { PendingToolCall, SupportedChatModels } from "@empiricalrun/shared-types";
-import { ToolCallService } from "../../tool-call-service";
-import { FileInfo } from "../../types";
-import { ReporterFunction } from "./types";
-export declare function chatAgentLoop({ chatModel, selectedModel, reporter, trace, toolCallService, fileInfo, isToolExecutionRemote, onToolCallQueued, }: {
+import { FileInfo, PendingToolCall, ReporterFunction, StreamingMessageReporterFunc, SupportedChatModels, ToolDefinition } from "@empiricalrun/shared-types";
+import { ToolExecutor } from "../../tools/executor";
+export declare function chatAgentLoop({ chatModel, selectedModel, reporter, streamingMessageReporter, trace, isToolExecutionRemote, toolExecutor, repoInfo, tools, onPendingToolCall, signal, featureFlags, }: {
     chatModel: IChatModel<any>;
     selectedModel: SupportedChatModels;
-    toolCallService: ToolCallService;
+    isToolExecutionRemote: boolean;
+    toolExecutor: ToolExecutor | undefined;
     reporter: ReporterFunction;
+    streamingMessageReporter?: StreamingMessageReporterFunc;
     trace?: TraceClient;
-    fileInfo: FileInfo;
-    isToolExecutionRemote: boolean;
-    onToolCallQueued?: (requestId: string, toolCalls: PendingToolCall[]) => void;
+    repoInfo: FileInfo;
+    tools: ToolDefinition[];
+    onPendingToolCall: (toolCalls: PendingToolCall[]) => Promise<void>;
+    signal?: AbortSignal;
+    featureFlags: string[];
 }): Promise<void>;
 //# sourceMappingURL=agent-loop.d.ts.map

package/dist/agent/chat/agent-loop.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"agent-loop.d.ts","sourceRoot":"","sources":["../../../src/agent/chat/agent-loop.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAE,UAAU,EAAmB,MAAM,wBAAwB,CAAC;AACrE,OAAO,EACL,eAAe,EACf,mBAAmB,EACpB,MAAM,4BAA4B,CAAC;AAEpC,OAAO,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAC1D,OAAO,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AAGvC,OAAO,EAAE,gBAAgB,EAAE,MAAM,SAAS,CAAC;AAG3C,wBAAsB,aAAa,CAAC,EAClC,SAAS,EACT,aAAa,EACb,QAAQ,EACR,KAAK,EACL,eAAe,EACf,QAAQ,EACR,qBAAqB,EACrB,gBAAgB,GACjB,EAAE;IACD,SAAS,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC;IAC3B,aAAa,EAAE,mBAAmB,CAAC;IACnC,eAAe,EAAE,eAAe,CAAC;IACjC,QAAQ,EAAE,gBAAgB,CAAC;IAC3B,KAAK,CAAC,EAAE,WAAW,CAAC;IACpB,QAAQ,EAAE,QAAQ,CAAC;IACnB,qBAAqB,EAAE,OAAO,CAAC;IAC/B,gBAAgB,CAAC,EAAE,CAAC,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,eAAe,EAAE,KAAK,IAAI,CAAC;CAC9E,iBAwDA"}
+{"version":3,"file":"agent-loop.d.ts","sourceRoot":"","sources":["../../../src/agent/chat/agent-loop.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAE,UAAU,EAAmB,MAAM,wBAAwB,CAAC;AACrE,OAAO,EACL,QAAQ,EACR,eAAe,EACf,gBAAgB,EAChB,4BAA4B,EAC5B,mBAAmB,EACnB,cAAc,EACf,MAAM,4BAA4B,CAAC;AAEpC,OAAO,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAC;AAKpD,wBAAsB,aAAa,CAAC,EAClC,SAAS,EACT,aAAa,EACb,QAAQ,EACR,wBAAwB,EACxB,KAAK,EACL,qBAAqB,EACrB,YAAY,EACZ,QAAQ,EACR,KAAK,EACL,iBAAiB,EACjB,MAAM,EACN,YAAY,GACb,EAAE;IACD,SAAS,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC;IAC3B,aAAa,EAAE,mBAAmB,CAAC;IACnC,qBAAqB,EAAE,OAAO,CAAC;IAC/B,YAAY,EAAE,YAAY,GAAG,SAAS,CAAC;IACvC,QAAQ,EAAE,gBAAgB,CAAC;IAC3B,wBAAwB,CAAC,EAAE,4BAA4B,CAAC;IACxD,KAAK,CAAC,EAAE,WAAW,CAAC;IACpB,QAAQ,EAAE,QAAQ,CAAC;IACnB,KAAK,EAAE,cAAc,EAAE,CAAC;IACxB,iBAAiB,EAAE,CAAC,SAAS,EAAE,eAAe,EAAE,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACnE,MAAM,CAAC,EAAE,WAAW,CAAC;IACrB,YAAY,EAAE,MAAM,EAAE,CAAC;CACxB,GAAG,OAAO,CAAC,IAAI,CAAC,CAiEhB"}

package/dist/agent/chat/agent-loop.js

@@ -5,38 +5,49 @@ const chat_1 = require("@empiricalrun/llm/chat");
 const prompt_1 = require("./prompt");
 const state_1 = require("./state");
 const utils_1 = require("./utils");
-async function chatAgentLoop({ chatModel, selectedModel, reporter, trace, toolCallService, fileInfo, isToolExecutionRemote, onToolCallQueued, }) {
-    const systemPrompt = await (0, prompt_1.buildSystemPrompt)(fileInfo);
+async function chatAgentLoop({ chatModel, selectedModel, reporter, streamingMessageReporter, trace, isToolExecutionRemote, toolExecutor, repoInfo, tools, onPendingToolCall, signal, featureFlags, }) {
+    const systemPrompt = await (0, prompt_1.buildSystemPrompt)(repoInfo);
     trace?.update({ input: { systemPrompt } });
     while (!chatModel.askUserForInput) {
+        // Abort if signal is set
+        if (signal?.aborted) {
+            (0, utils_1.log)("[chatAgentLoop] Aborted by signal");
+            break;
+        }
         try {
             const toolCalls = chatModel.getPendingToolCalls();
             if (toolCalls.length > 0) {
+                await onPendingToolCall(toolCalls);
                 if (isToolExecutionRemote) {
-                    const requestId = toolCalls[0].id;
-                    await toolCallService.sendToQueue(requestId, toolCalls);
-                    onToolCallQueued?.(requestId, toolCalls);
                     (0, utils_1.log)(`Tool call remote execution in progress`);
                     break;
                 }
                 else {
-                    const toolResults = await toolCallService.execute(toolCalls);
+                    if (!toolExecutor) {
+                        throw new Error("ToolExecutor is required for local execution");
+                    }
+                    const toolResults = await toolExecutor.execute(toolCalls);
                     chatModel.pushToolResultsMessage(toolCalls, toolResults);
                 }
             }
             (0, utils_1.log)(`${(0, utils_1.getModelName)(selectedModel)} is working...`);
-            let response = null;
-            response = await chatModel.getLLMResponse({
+            const response = await chatModel.getLLMResponse({
                 systemPrompt,
-                tools: toolCallService.tools.map((tool) => (0, chat_1.zodToOpenAITool)(tool.schema)),
+                tools: tools.map((tool) => (0, chat_1.zodToOpenAITool)(tool.schema)),
                 selectedModel,
                 trace,
+                streamingMessageReporter,
+                hasThinkingEnabled: featureFlags.includes("extendedThinkingClaude"),
             });
             if (!response) {
                 throw new Error("Error getting response from LLM");
             }
             chatModel.pushMessage(response);
             const latest = chatModel.getHumanReadableLatestMessage();
+            if (signal?.aborted) {
+                (0, utils_1.log)("[chatAgentLoop] Aborted by signal");
+                break;
+            }
             await reporter((0, state_1.chatStateFromModel)({
                 chatModel,
                 selectedModel,
@@ -45,7 +56,6 @@ async function chatAgentLoop({ chatModel, selectedModel, reporter, trace, toolCa
         }
         catch (error) {
             await (0, utils_1.handleAgentError)({
-                context: "chatAgentLoop",
                 error,
                 chatModel,
                 selectedModel,

package/dist/agent/chat/exports.d.ts

@@ -1,9 +1,9 @@
-export type { FileInfo } from "../../types";
+export { getFileInfoFromGitHub } from "../../file-info/github";
 export { chatAgentLoop } from "./agent-loop";
 export { defaultModel } from "./models";
 export { CHAT_STATE_VERSIONS_MIGRATIONS_MAP, chatStateFromModel, createChatState, createChatStateForMessages, fetchToolCallAvailability, getLatestDownloadBuildUrl, LATEST_CHAT_STATE_VERSION, migrateChatState, processAttachments, } from "./state";
-export type { ReporterFunction } from "./types";
 export { extractAttachments } from "./utils";
+export { processToolCallsAndUpdateChatState, rejectPendingToolCalls, } from "./utils/tool-calls";
 export type { IChatModel } from "@empiricalrun/llm/chat";
 export { createChatModel, SUPPORTED_CHAT_MODELS } from "@empiricalrun/llm/chat";
 //# sourceMappingURL=exports.d.ts.map

package/dist/agent/chat/exports.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"exports.d.ts","sourceRoot":"","sources":["../../../src/agent/chat/exports.ts"],"names":[],"mappings":"AAAA,YAAY,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AAC5C,OAAO,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAC7C,OAAO,EAAE,YAAY,EAAE,MAAM,UAAU,CAAC;AACxC,OAAO,EACL,kCAAkC,EAClC,kBAAkB,EAClB,eAAe,EACf,0BAA0B,EAC1B,yBAAyB,EACzB,yBAAyB,EACzB,yBAAyB,EACzB,gBAAgB,EAChB,kBAAkB,GACnB,MAAM,SAAS,CAAC;AACjB,YAAY,EAAE,gBAAgB,EAAE,MAAM,SAAS,CAAC;AAChD,OAAO,EAAE,kBAAkB,EAAE,MAAM,SAAS,CAAC;AAC7C,YAAY,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,EAAE,eAAe,EAAE,qBAAqB,EAAE,MAAM,wBAAwB,CAAC"}
+{"version":3,"file":"exports.d.ts","sourceRoot":"","sources":["../../../src/agent/chat/exports.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,qBAAqB,EAAE,MAAM,wBAAwB,CAAC;AAC/D,OAAO,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAC7C,OAAO,EAAE,YAAY,EAAE,MAAM,UAAU,CAAC;AACxC,OAAO,EACL,kCAAkC,EAClC,kBAAkB,EAClB,eAAe,EACf,0BAA0B,EAC1B,yBAAyB,EACzB,yBAAyB,EACzB,yBAAyB,EACzB,gBAAgB,EAChB,kBAAkB,GACnB,MAAM,SAAS,CAAC;AACjB,OAAO,EAAE,kBAAkB,EAAE,MAAM,SAAS,CAAC;AAC7C,OAAO,EACL,kCAAkC,EAClC,sBAAsB,GACvB,MAAM,oBAAoB,CAAC;AAC5B,YAAY,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,EAAE,eAAe,EAAE,qBAAqB,EAAE,MAAM,wBAAwB,CAAC"}

package/dist/agent/chat/exports.js

@@ -1,6 +1,8 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.SUPPORTED_CHAT_MODELS = exports.createChatModel = exports.extractAttachments = exports.processAttachments = exports.migrateChatState = exports.LATEST_CHAT_STATE_VERSION = exports.getLatestDownloadBuildUrl = exports.fetchToolCallAvailability = exports.createChatStateForMessages = exports.createChatState = exports.chatStateFromModel = exports.CHAT_STATE_VERSIONS_MIGRATIONS_MAP = exports.defaultModel = exports.chatAgentLoop = void 0;
+exports.SUPPORTED_CHAT_MODELS = exports.createChatModel = exports.rejectPendingToolCalls = exports.processToolCallsAndUpdateChatState = exports.extractAttachments = exports.processAttachments = exports.migrateChatState = exports.LATEST_CHAT_STATE_VERSION = exports.getLatestDownloadBuildUrl = exports.fetchToolCallAvailability = exports.createChatStateForMessages = exports.createChatState = exports.chatStateFromModel = exports.CHAT_STATE_VERSIONS_MIGRATIONS_MAP = exports.defaultModel = exports.chatAgentLoop = exports.getFileInfoFromGitHub = void 0;
+var github_1 = require("../../file-info/github");
+Object.defineProperty(exports, "getFileInfoFromGitHub", { enumerable: true, get: function () { return github_1.getFileInfoFromGitHub; } });
 var agent_loop_1 = require("./agent-loop");
 Object.defineProperty(exports, "chatAgentLoop", { enumerable: true, get: function () { return agent_loop_1.chatAgentLoop; } });
 var models_1 = require("./models");
@@ -17,6 +19,9 @@ Object.defineProperty(exports, "migrateChatState", { enumerable: true, get: func
 Object.defineProperty(exports, "processAttachments", { enumerable: true, get: function () { return state_1.processAttachments; } });
 var utils_1 = require("./utils");
 Object.defineProperty(exports, "extractAttachments", { enumerable: true, get: function () { return utils_1.extractAttachments; } });
+var tool_calls_1 = require("./utils/tool-calls");
+Object.defineProperty(exports, "processToolCallsAndUpdateChatState", { enumerable: true, get: function () { return tool_calls_1.processToolCallsAndUpdateChatState; } });
+Object.defineProperty(exports, "rejectPendingToolCalls", { enumerable: true, get: function () { return tool_calls_1.rejectPendingToolCalls; } });
 var chat_1 = require("@empiricalrun/llm/chat");
 Object.defineProperty(exports, "createChatModel", { enumerable: true, get: function () { return chat_1.createChatModel; } });
 Object.defineProperty(exports, "SUPPORTED_CHAT_MODELS", { enumerable: true, get: function () { return chat_1.SUPPORTED_CHAT_MODELS; } });

package/dist/agent/chat/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/agent/chat/index.ts"],"names":[],"mappings":"AAEA,OAAO,EAGL,mBAAmB,EACpB,MAAM,4BAA4B,CAAC;AAoCpC,wBAAsB,yBAAyB,IAAI,OAAO,CACxD,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CACvB,CAwBA;AAED,wBAAsB,kBAAkB,CAAC,EACvC,mBAAmB,EACnB,aAAa,EACb,oBAAoB,GACrB,EAAE;IACD,aAAa,EAAE,mBAAmB,CAAC;IACnC,mBAAmB,EAAE,OAAO,CAAC;IAC7B,oBAAoB,EAAE,MAAM,GAAG,SAAS,CAAC;CAC1C,iBAkIA"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/agent/chat/index.ts"],"names":[],"mappings":"AAEA,OAAO,EAIL,mBAAmB,EACpB,MAAM,4BAA4B,CAAC;AAoCpC,wBAAsB,yBAAyB,IAAI,OAAO,CACxD,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CACvB,CAwBA;AAED,wBAAsB,kBAAkB,CAAC,EACvC,mBAAmB,EACnB,aAAa,EACb,oBAAoB,GACrB,EAAE;IACD,aAAa,EAAE,mBAAmB,CAAC;IACnC,mBAAmB,EAAE,OAAO,CAAC;IAC7B,oBAAoB,EAAE,MAAM,GAAG,SAAS,CAAC;CAC1C,iBAmKA"}

package/dist/agent/chat/index.js

@@ -5,10 +5,11 @@ exports.runChatAgentForCLI = runChatAgentForCLI;
 const llm_1 = require("@empiricalrun/llm");
 const chat_1 = require("@empiricalrun/llm/chat");
 const picocolors_1 = require("picocolors");
+const file_system_1 = require("../../file-info/file-system");
 const human_in_the_loop_1 = require("../../human-in-the-loop");
-const tool_call_service_1 = require("../../tool-call-service");
-const file_tree_1 = require("../../utils/file-tree");
-const git_1 = require("../../utils/git");
+const definitions_1 = require("../../tools/definitions");
+const executor_1 = require("../../tools/executor");
+const git_1 = require("../../tools/executor/utils/git");
 const agent_loop_1 = require("./agent-loop");
 const state_1 = require("./state");
 const utils_1 = require("./utils");
@@ -43,6 +44,7 @@ async function fetchEnvironmentVariables() {
 }
 async function runChatAgentForCLI({ useDiskForChatState, selectedModel, initialPromptContent, }) {
     let chatState;
+    let enableStreaming = true;
     if (useDiskForChatState) {
         chatState = (0, state_1.loadChatState)();
     }
@@ -83,7 +85,12 @@ async function runChatAgentForCLI({ useDiskForChatState, selectedModel, initialP
             (0, state_1.saveToDisk)(chatState.messages, selectedModel, chatState.askUserForInput, chatState.error);
         }
         if (latest) {
-            console.log(`${(0, picocolors_1.blue)(latest.role)}: ${latest.textMessage}`);
+            if (!enableStreaming) {
+                console.log(`${(0, picocolors_1.blue)(latest.role)}: ${latest.textMessage}`);
+            }
+            else {
+                process.stdout.write(`\n`);
+            }
         }
     };
     const trace = (0, llm_1.createLangfuseTrace)({
@@ -127,7 +134,7 @@ async function runChatAgentForCLI({ useDiskForChatState, selectedModel, initialP
             if (!process.env.EMPIRICALRUN_API_KEY) {
                 throw new Error("EMPIRICALRUN_API_KEY is not set");
             }
-            const toolCallService = new tool_call_service_1.ToolCallService({
+            const toolExecutor = new executor_1.ToolExecutor({
                 chatSessionId: null,
                 selectedModel,
                 branchName,
@@ -137,23 +144,51 @@ async function runChatAgentForCLI({ useDiskForChatState, selectedModel, initialP
                 featureFlags: [],
                 environmentOverrides: await fetchEnvironmentVariables(),
             });
-            const fileInfo = await (0, file_tree_1.getFileInfoFromFS)(process.cwd());
+            const toolCaller = new definitions_1.ToolCaller({
+                chatSessionId: null,
+                selectedModel,
+                branchName,
+            });
+            const fileInfo = await (0, file_system_1.getFileInfoFromFS)(process.cwd());
             await (0, agent_loop_1.chatAgentLoop)({
                 chatModel,
                 selectedModel,
                 reporter: reporterFunc,
+                streamingMessageReporter: (() => {
+                    if (!enableStreaming) {
+                        return;
+                    }
+                    let hasStarted = false;
+                    let startedRole = undefined;
+                    return async (delta, snapshot, thinking) => {
+                        if (delta) {
+                            const role = thinking ? "Thinking" : "Assistant";
+                            if (!hasStarted) {
+                                process.stdout.write(`${(0, picocolors_1.blue)(role)}: `);
+                                hasStarted = true;
+                                startedRole = role;
+                            }
+                            else if (hasStarted && role !== startedRole) {
+                                // Changing from thinking -> text block
+                                process.stdout.write("\n");
+                                process.stdout.write(`${(0, picocolors_1.blue)(role)}: `);
+                                startedRole = role;
+                            }
+                            process.stdout.write(delta);
+                        }
+                    };
+                })(),
                 trace,
-                toolCallService,
-                fileInfo,
+                repoInfo: fileInfo,
                 isToolExecutionRemote: false,
+                toolExecutor,
+                tools: toolCaller.tools,
+                onPendingToolCall: async () => { },
+                featureFlags: ["extendedThinkingClaude"],
             });
         }
     }
-    trace?.update({
-        output: {
-            messages: chatModel.messages,
-        },
-    });
+    trace?.update({ output: { messages: chatModel.messages } });
     await llm_1.langfuseInstance?.flushAsync();
     const usageSummary = chatModel.getUsageSummary();
     console.log(`\n${(0, picocolors_1.gray)("Usage summary -> " + usageSummary)}`);

package/dist/agent/chat/prompt/index.d.ts

@@ -1,3 +1,3 @@
-import { FileInfo } from "../../../types";
-export declare function buildSystemPrompt(fileInfo: FileInfo): Promise<string>;
+import { FileInfo } from "@empiricalrun/shared-types";
+export declare function buildSystemPrompt(repoInfo: FileInfo): Promise<string>;
 //# sourceMappingURL=index.d.ts.map

package/dist/agent/chat/prompt/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/agent/chat/prompt/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAI1C,wBAAsB,iBAAiB,CAAC,QAAQ,EAAE,QAAQ,mBA4GzD"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/agent/chat/prompt/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,4BAA4B,CAAC;AAKtD,wBAAsB,iBAAiB,CAAC,QAAQ,EAAE,QAAQ,mBA4GzD"}

package/dist/agent/chat/prompt/index.js

@@ -3,7 +3,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.buildSystemPrompt = buildSystemPrompt;
 const pw_utils_docs_1 = require("./pw-utils-docs");
 const repo_1 = require("./repo");
-async function buildSystemPrompt(fileInfo) {
+async function buildSystemPrompt(repoInfo) {
     const preamble = `
 You are a helpful assistant that can answer questions and help with tasks related to writing and maintaining Playwright tests.
 
@@ -101,7 +101,7 @@ if (await saveButton.isVisible()) {
 \`\`\`
 
 `;
-    const repoContext = await (0, repo_1.getRepoInfoPrompt)(fileInfo);
+    const repoContext = await (0, repo_1.getRepoInfoPrompt)(repoInfo);
     return `${preamble}
 
 # Recipes

package/dist/agent/chat/prompt/repo.d.ts

@@ -1,3 +1,3 @@
-import { FileInfo } from "../../../types";
-export declare function getRepoInfoPrompt(directory: FileInfo): Promise<string>;
+import { FileInfo } from "@empiricalrun/shared-types";
+export declare function getRepoInfoPrompt(repoInfo: FileInfo): Promise<string>;
 //# sourceMappingURL=repo.d.ts.map

package/dist/agent/chat/prompt/repo.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"repo.d.ts","sourceRoot":"","sources":["../../../../src/agent/chat/prompt/repo.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAwC1C,wBAAsB,iBAAiB,CAAC,SAAS,EAAE,QAAQ,mBAyC1D"}
+{"version":3,"file":"repo.d.ts","sourceRoot":"","sources":["../../../../src/agent/chat/prompt/repo.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,4BAA4B,CAAC;AAqDtD,wBAAsB,iBAAiB,CAAC,QAAQ,EAAE,QAAQ,mBA0CzD"}

package/dist/agent/chat/prompt/repo.js

@@ -7,24 +7,31 @@ exports.getRepoInfoPrompt = getRepoInfoPrompt;
 const path_1 = __importDefault(require("path"));
 const repo_tree_1 = require("../../../utils/repo-tree");
 async function getAllMarkdownFiles(directory) {
-    if (!directory.isDirectory) {
+    if (directory.type !== "directory") {
         return [];
     }
-    const files = directory.children.find((child) => child.isDirectory && child.path === ".empiricalrun")?.children;
-    if (!files) {
+    const files = directory.children;
+    if (!files || files.length === 0) {
         return [];
     }
     return Promise.all(files
-        .filter((file) => file.isFile && file.path.endsWith(".md"))
+        .filter((file) => file.type === "file" && file.path.endsWith(".md"))
         .map(async (file) => {
         return {
             name: path_1.default.basename(file.path),
-            content: await file.getContent(),
+            content: file.type === "file" ? await file.getContent() : "",
         };
     }));
 }
-async function knowledgeContext(directory) {
-    const mdFiles = await getAllMarkdownFiles(directory);
+async function knowledgeContext(repoInfo) {
+    if (repoInfo.type !== "directory") {
+        throw new Error(`"${repoInfo.path}" is not a valid directory`);
+    }
+    const knowledgeDir = repoInfo.children.find((child) => child.type === "directory" && child.path === ".empiricalrun");
+    if (!knowledgeDir) {
+        return "";
+    }
+    const mdFiles = await getAllMarkdownFiles(knowledgeDir);
     const knowledge = mdFiles.map((file) => {
         return `
     <knowledge_file>
@@ -37,17 +44,19 @@ async function knowledgeContext(directory) {
     });
     return knowledge.join("\n");
 }
-async function getRepoInfoPrompt(directory) {
+async function getRepoInfoPrompt(repoInfo) {
+    if (!repoInfo)
+        return "";
     let REPO_CONTEXT_PROMPT = `
 You are running as a CLI tool inside the directory of the repo that has Playwright tests.
 
 Here is the repo directory structure:
 
-${(0, repo_tree_1.generateAsciiTree)(directory)}
+${(0, repo_tree_1.generateAsciiTree)(repoInfo)}
 
 While specifying paths to files, use relative paths from the current working directory. For example:
 - Correct path: "tests/lesson.spec.ts"
-- Incorrect path: "/repo/tests/lesson.spec.ts" or "${directory.name}/tests/lesson.spec.ts"
+- Incorrect path: "/repo/tests/lesson.spec.ts" or "${repoInfo.name}/tests/lesson.spec.ts"
 `;
     // try {
     //   const playwrightConfig = await readPlaywrightConfig(process.cwd());
@@ -62,7 +71,7 @@ While specifying paths to files, use relative paths from the current working dir
     // } catch (error) {
     //   console.warn("Failed to read playwright config", error);
     // }
-    const knowledge = await knowledgeContext(directory);
+    const knowledge = await knowledgeContext(repoInfo);
     if (knowledge.length > 0) {
         REPO_CONTEXT_PROMPT += `
 ## Repo-specific knowledge

package/dist/agent/chat/state.d.ts

@@ -12,7 +12,7 @@ export declare function createChatState({ userPrompt, attachments, existingState
     error: ChatStateError | null;
 }): ChatState;
 export declare function createChatStateForMessages({ messages, selectedModel, askUserForInput, error, }: {
-    messages: any;
+    messages: CanonicalMessage[];
     selectedModel: SupportedChatModels;
     askUserForInput: boolean;
     error: ChatStateError | null;
@@ -28,7 +28,7 @@ export declare function loadChatState(): ChatState | undefined;
  * Add migration logic for each version as needed.
  */
 export declare function migrateChatState(oldState: any): ChatState;
-export declare function saveToDisk<T>(messages: Array<T>, selectedModel: SupportedChatModels, askUserForInput: boolean, error: ChatStateError | null): void;
+export declare function saveToDisk(messages: Array<CanonicalMessage>, selectedModel: SupportedChatModels, askUserForInput: boolean, error: ChatStateError | null): void;
 export declare function getLatestDownloadBuildUrl(messages: CanonicalMessage[]): string | null;
 export declare function fetchToolCallAvailability(toolCallId: String, messages: CanonicalMessage[]): {
     hasToolRequest: boolean;

package/dist/agent/chat/state.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"state.d.ts","sourceRoot":"","sources":["../../../src/agent/chat/state.ts"],"names":[],"mappings":"AAAA,OAAO,EAIL,UAAU,EACX,MAAM,wBAAwB,CAAC;AAChC,OAAO,EACL,UAAU,EACV,gBAAgB,EAChB,SAAS,EACT,cAAc,EACd,mBAAmB,EACnB,mBAAmB,EAEpB,MAAM,4BAA4B,CAAC;AA+BpC,eAAO,MAAM,kCAAkC,EAAE,MAAM,CACrD,MAAM,EACN,CAAC,KAAK,EAAE,GAAG,KAAK,GAAG,CAIpB,CAAC;AAEF,eAAO,MAAM,yBAAyB,QAAQ,CAAC;AAE/C,eAAO,MAAM,eAAe,QAI3B,CAAC;AAEF,wBAAsB,kBAAkB,CACtC,WAAW,EAAE,UAAU,EAAE,EACzB,aAAa,EAAE,mBAAmB,GACjC,OAAO,CAAC,mBAAmB,EAAE,CAAC,CAMhC;AAED,wBAAgB,eAAe,CAAC,EAC9B,UAAU,EACV,WAAW,EACX,aAAa,EACb,aAAa,EACb,KAAK,GACN,EAAE;IACD,UAAU,EAAE,MAAM,GAAG,SAAS,CAAC;IAC/B,WAAW,EAAE,mBAAmB,EAAE,CAAC;IACnC,aAAa,EAAE,SAAS,GAAG,SAAS,CAAC;IACrC,aAAa,EAAE,mBAAmB,CAAC;IACnC,KAAK,EAAE,cAAc,GAAG,IAAI,CAAC;CAC9B,aAYA;AAED,wBAAgB,0BAA0B,CAAC,EACzC,QAAQ,EACR,aAAa,EACb,eAAe,EACf,KAAK,GACN,EAAE;IACD,QAAQ,EAAE,GAAG,CAAC;IACd,aAAa,EAAE,mBAAmB,CAAC;IACnC,eAAe,EAAE,OAAO,CAAC;IACzB,KAAK,EAAE,cAAc,GAAG,IAAI,CAAC;CAC9B,GAAG,SAAS,CASZ;AAED,wBAAgB,kBAAkB,CAAC,CAAC,EAAE,EACpC,SAAS,EACT,aAAa,EACb,KAAK,GACN,EAAE;IACD,SAAS,EAAE,UAAU,CAAC,CAAC,CAAC,CAAC;IACzB,aAAa,EAAE,mBAAmB,CAAC;IACnC,KAAK,EAAE,cAAc,GAAG,IAAI,CAAC;CAC9B,aAOA;AAED,wBAAgB,aAAa,IAAI,SAAS,GAAG,SAAS,CAarD;AAED;;;GAGG;AACH,wBAAgB,gBAAgB,CAAC,QAAQ,EAAE,GAAG,GAAG,SAAS,CAqBzD;AAED,wBAAgB,UAAU,CAAC,CAAC,EAC1B,QAAQ,EAAE,KAAK,CAAC,CAAC,CAAC,EAClB,aAAa,EAAE,mBAAmB,EAClC,eAAe,EAAE,OAAO,EACxB,KAAK,EAAE,cAAc,GAAG,IAAI,QAgB7B;AA2BD,wBAAgB,yBAAyB,CACvC,QAAQ,EAAE,gBAAgB,EAAE,GAC3B,MAAM,GAAG,IAAI,CAef;AAED,wBAAgB,yBAAyB,CACvC,UAAU,EAAE,MAAM,EAClB,QAAQ,EAAE,gBAAgB,EAAE;;;;;EA2B7B"}
+{"version":3,"file":"state.d.ts","sourceRoot":"","sources":["../../../src/agent/chat/state.ts"],"names":[],"mappings":"AAAA,OAAO,EAIL,UAAU,EACX,MAAM,wBAAwB,CAAC;AAChC,OAAO,EACL,UAAU,EACV,gBAAgB,EAChB,SAAS,EACT,cAAc,EACd,mBAAmB,EACnB,mBAAmB,EAEpB,MAAM,4BAA4B,CAAC;AA+BpC,eAAO,MAAM,kCAAkC,EAAE,MAAM,CACrD,MAAM,EACN,CAAC,KAAK,EAAE,GAAG,KAAK,GAAG,CAIpB,CAAC;AAEF,eAAO,MAAM,yBAAyB,QAAQ,CAAC;AAE/C,eAAO,MAAM,eAAe,QAI3B,CAAC;AAEF,wBAAsB,kBAAkB,CACtC,WAAW,EAAE,UAAU,EAAE,EACzB,aAAa,EAAE,mBAAmB,GACjC,OAAO,CAAC,mBAAmB,EAAE,CAAC,CAMhC;AAED,wBAAgB,eAAe,CAAC,EAC9B,UAAU,EACV,WAAW,EACX,aAAa,EACb,aAAa,EACb,KAAK,GACN,EAAE;IACD,UAAU,EAAE,MAAM,GAAG,SAAS,CAAC;IAC/B,WAAW,EAAE,mBAAmB,EAAE,CAAC;IACnC,aAAa,EAAE,SAAS,GAAG,SAAS,CAAC;IACrC,aAAa,EAAE,mBAAmB,CAAC;IACnC,KAAK,EAAE,cAAc,GAAG,IAAI,CAAC;CAC9B,aAYA;AAED,wBAAgB,0BAA0B,CAAC,EACzC,QAAQ,EACR,aAAa,EACb,eAAe,EACf,KAAK,GACN,EAAE;IACD,QAAQ,EAAE,gBAAgB,EAAE,CAAC;IAC7B,aAAa,EAAE,mBAAmB,CAAC;IACnC,eAAe,EAAE,OAAO,CAAC;IACzB,KAAK,EAAE,cAAc,GAAG,IAAI,CAAC;CAC9B,GAAG,SAAS,CASZ;AAED,wBAAgB,kBAAkB,CAAC,CAAC,EAAE,EACpC,SAAS,EACT,aAAa,EACb,KAAK,GACN,EAAE;IACD,SAAS,EAAE,UAAU,CAAC,CAAC,CAAC,CAAC;IACzB,aAAa,EAAE,mBAAmB,CAAC;IACnC,KAAK,EAAE,cAAc,GAAG,IAAI,CAAC;CAC9B,aAOA;AAED,wBAAgB,aAAa,IAAI,SAAS,GAAG,SAAS,CAarD;AAED;;;GAGG;AACH,wBAAgB,gBAAgB,CAAC,QAAQ,EAAE,GAAG,GAAG,SAAS,CAqBzD;AAED,wBAAgB,UAAU,CACxB,QAAQ,EAAE,KAAK,CAAC,gBAAgB,CAAC,EACjC,aAAa,EAAE,mBAAmB,EAClC,eAAe,EAAE,OAAO,EACxB,KAAK,EAAE,cAAc,GAAG,IAAI,QAgB7B;AA2BD,wBAAgB,yBAAyB,CACvC,QAAQ,EAAE,gBAAgB,EAAE,GAC3B,MAAM,GAAG,IAAI,CAef;AAED,wBAAgB,yBAAyB,CACvC,UAAU,EAAE,MAAM,EAClB,QAAQ,EAAE,gBAAgB,EAAE;;;;;EA2B7B"}

package/dist/agent/chat/types.d.ts

@@ -1,8 +1,2 @@
-import { ChatState } from "@empiricalrun/shared-types";
-type LatestMessage = {
-    role: string;
-    textMessage: string;
-};
-export type ReporterFunction = (state: ChatState, latestHumanReadableMessage: LatestMessage | undefined) => Promise<void>;
 export {};
 //# sourceMappingURL=types.d.ts.map

package/dist/agent/chat/types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../src/agent/chat/types.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,4BAA4B,CAAC;AAEvD,KAAK,aAAa,GAAG;IACnB,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;CACrB,CAAC;AAEF,MAAM,MAAM,gBAAgB,GAAG,CAC7B,KAAK,EAAE,SAAS,EAChB,0BAA0B,EAAE,aAAa,GAAG,SAAS,KAClD,OAAO,CAAC,IAAI,CAAC,CAAC"}
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../src/agent/chat/types.ts"],"names":[],"mappings":""}

package/dist/agent/chat/utils/tool-calls.d.ts

@@ -0,0 +1,21 @@
+import { IChatModel } from "@empiricalrun/llm/chat";
+import type { ChatState, PendingToolCall, SupportedChatModels, ToolResult } from "@empiricalrun/shared-types";
+export declare function rejectPendingToolCalls({ chatState, chatModel, }: {
+    chatState: ChatState;
+    chatModel: IChatModel<any>;
+}): ChatState;
+export declare function filterPendingToolCalls({ toolCalls, toolResults, pendingToolCalls, }: {
+    toolCalls: PendingToolCall[];
+    toolResults: ToolResult[];
+    pendingToolCalls: PendingToolCall[];
+}): {
+    filteredToolCalls: PendingToolCall[];
+    filteredToolResults: ToolResult[];
+};
+export declare function processToolCallsAndUpdateChatState({ toolCalls, toolResults, chatModel, selectedModel, }: {
+    toolCalls: PendingToolCall[];
+    toolResults: ToolResult[];
+    chatModel: IChatModel<any>;
+    selectedModel: SupportedChatModels;
+}): ChatState | undefined;
+//# sourceMappingURL=tool-calls.d.ts.map

package/dist/agent/chat/utils/tool-calls.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-calls.d.ts","sourceRoot":"","sources":["../../../../src/agent/chat/utils/tool-calls.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AACpD,OAAO,KAAK,EACV,SAAS,EACT,eAAe,EACf,mBAAmB,EACnB,UAAU,EACX,MAAM,4BAA4B,CAAC;AAOpC,wBAAgB,sBAAsB,CAAC,EACrC,SAAS,EACT,SAAS,GACV,EAAE;IACD,SAAS,EAAE,SAAS,CAAC;IACrB,SAAS,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC;CAC5B,GAAG,SAAS,CA8BZ;AAED,wBAAgB,sBAAsB,CAAC,EACrC,SAAS,EACT,WAAW,EACX,gBAAgB,GACjB,EAAE;IACD,SAAS,EAAE,eAAe,EAAE,CAAC;IAC7B,WAAW,EAAE,UAAU,EAAE,CAAC;IAC1B,gBAAgB,EAAE,eAAe,EAAE,CAAC;CACrC;;;EAgBA;AAED,wBAAgB,kCAAkC,CAAC,EACjD,SAAS,EACT,WAAW,EACX,SAAS,EACT,aAAa,GACd,EAAE;IACD,SAAS,EAAE,eAAe,EAAE,CAAC;IAC7B,WAAW,EAAE,UAAU,EAAE,CAAC;IAC1B,SAAS,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC;IAC3B,aAAa,EAAE,mBAAmB,CAAC;CACpC,GAAG,SAAS,GAAG,SAAS,CAkBxB"}

package/dist/agent/chat/utils/tool-calls.js

@@ -0,0 +1,64 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.rejectPendingToolCalls = rejectPendingToolCalls;
+exports.filterPendingToolCalls = filterPendingToolCalls;
+exports.processToolCallsAndUpdateChatState = processToolCallsAndUpdateChatState;
+const state_1 = require("../state");
+const TOOL_CALL_REJECTED_BY_USER_ERROR = "This tool call was rejected by the user.";
+function rejectPendingToolCalls({ chatState, chatModel, }) {
+    const pendingToolCalls = chatModel.getPendingToolCalls();
+    let updatedChatState = {
+        ...chatState,
+        // When tool calls are rejected, we reset askUserForInput to true
+        // so that the user can provide input for the next tool call. The agent
+        // loop is not run again until the user provides input.
+        askUserForInput: true,
+    };
+    if (pendingToolCalls.length > 0) {
+        const toolResults = pendingToolCalls.map((toolCall) => {
+            console.log(`Rejecting pending tool calls with toolCallId: ${toolCall.id} `);
+            return {
+                isError: true,
+                result: TOOL_CALL_REJECTED_BY_USER_ERROR,
+            };
+        });
+        chatModel.pushToolResultsMessage(pendingToolCalls, toolResults);
+        return {
+            ...updatedChatState,
+            messages: chatModel.messages,
+        };
+    }
+    return updatedChatState;
+}
+function filterPendingToolCalls({ toolCalls, toolResults, pendingToolCalls, }) {
+    const filteredToolResults = [];
+    const filteredToolCalls = toolCalls.filter((call, index) => {
+        if (pendingToolCalls.some((p) => p.id === call.id)) {
+            filteredToolResults.push(toolResults[index]);
+            return true;
+        }
+        return false;
+    });
+    return {
+        filteredToolCalls,
+        filteredToolResults,
+    };
+}
+function processToolCallsAndUpdateChatState({ toolCalls, toolResults, chatModel, selectedModel, }) {
+    const pendingToolCalls = chatModel.getPendingToolCalls() || [];
+    const { filteredToolCalls, filteredToolResults } = filterPendingToolCalls({
+        toolCalls,
+        toolResults,
+        pendingToolCalls,
+    });
+    if (filteredToolCalls.length > 0) {
+        chatModel.pushToolResultsMessage(filteredToolCalls, filteredToolResults);
+        const newChatState = (0, state_1.createChatStateForMessages)({
+            messages: chatModel.messages || [],
+            selectedModel,
+            askUserForInput: chatModel.askUserForInput,
+            error: null,
+        });
+        return newChatState;
+    }
+}

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

@@ -1,11 +1,9 @@
 import { TraceClient } from "@empiricalrun/llm";
 import { IChatModel } from "@empiricalrun/llm/chat";
-import { Attachment, SupportedChatModels } from "@empiricalrun/shared-types";
-import { ReporterFunction } from "./types";
+import { Attachment, ReporterFunction, SupportedChatModels } from "@empiricalrun/shared-types";
 export declare const log: (...args: any[]) => void;
 export declare function getModelName(model: string): string;
-export declare function handleAgentError({ context, error, chatModel, selectedModel, reporter, trace, }: {
-    context: string;
+export declare function handleAgentError({ error, chatModel, selectedModel, reporter, trace, }: {
     error: unknown;
     chatModel: IChatModel<any>;
     selectedModel: SupportedChatModels;

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

@@ -1 +1 @@
-{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../src/agent/chat/utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAuB,UAAU,EAAE,MAAM,wBAAwB,CAAC;AACzE,OAAO,EACL,UAAU,EAEV,mBAAmB,EACpB,MAAM,4BAA4B,CAAC;AAIpC,OAAO,EAAE,gBAAgB,EAAE,MAAM,SAAS,CAAC;AAE3C,eAAO,MAAM,GAAG,GAAI,GAAG,MAAM,GAAG,EAAE,SAEjC,CAAC;AAcF,wBAAgB,YAAY,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,CAMlD;AAED,wBAAsB,gBAAgB,CAAC,EACrC,OAAO,EACP,KAAK,EACL,SAAS,EACT,aAAa,EACb,QAAQ,EACR,KAAK,GACN,EAAE;IACD,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,OAAO,CAAC;IACf,SAAS,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC;IAC3B,aAAa,EAAE,mBAAmB,CAAC;IACnC,QAAQ,EAAE,gBAAgB,CAAC;IAC3B,KAAK,CAAC,EAAE,WAAW,CAAC;CACrB,iBAsBA;AAED,wBAAgB,kBAAkB,CAAC,UAAU,EAAE,MAAM,GAAG;IACtD,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,UAAU,EAAE,CAAC;CAC3B,CAsBA"}
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../src/agent/chat/utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAuB,UAAU,EAAE,MAAM,wBAAwB,CAAC;AACzE,OAAO,EACL,UAAU,EAEV,gBAAgB,EAChB,mBAAmB,EACpB,MAAM,4BAA4B,CAAC;AAKpC,eAAO,MAAM,GAAG,GAAI,GAAG,MAAM,GAAG,EAAE,SAEjC,CAAC;AAcF,wBAAgB,YAAY,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,CAMlD;AAED,wBAAsB,gBAAgB,CAAC,EACrC,KAAK,EACL,SAAS,EACT,aAAa,EACb,QAAQ,EACR,KAAK,GACN,EAAE;IACD,KAAK,EAAE,OAAO,CAAC;IACf,SAAS,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC;IAC3B,aAAa,EAAE,mBAAmB,CAAC;IACnC,QAAQ,EAAE,gBAAgB,CAAC;IAC3B,KAAK,CAAC,EAAE,WAAW,CAAC;CACrB,iBAgBA;AAED,wBAAgB,kBAAkB,CAAC,UAAU,EAAE,MAAM,GAAG;IACtD,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,UAAU,EAAE,CAAC;CAC3B,CAsBA"}

package/dist/agent/chat/utils.js

@@ -11,8 +11,8 @@ const log = (...args) => {
     console.log((0, picocolors_1.gray)(args.join(" ")));
 };
 exports.log = log;
-function logError(context, error, trace) {
-    console.error((0, picocolors_1.gray)(`[Error in ${context}]:`), error instanceof Error ? error.stack || error.message : error);
+function logError(error, trace) {
+    console.error((0, picocolors_1.gray)(`[Error]:`), error instanceof Error ? error.stack || error.message : error);
     trace?.update({
         output: {
             error: error instanceof Error ? error.message : String(error),
@@ -29,7 +29,7 @@ function getModelName(model) {
         return "o4";
     return "AI";
 }
-async function handleAgentError({ context, error, chatModel, selectedModel, reporter, trace, }) {
+async function handleAgentError({ error, chatModel, selectedModel, reporter, trace, }) {
     const errorObject = {
         message: error.message,
         stack: error.stack || "Stack trace not available",
@@ -40,14 +40,8 @@ async function handleAgentError({ context, error, chatModel, selectedModel, repo
         selectedModel,
         error: errorObject,
     }), chatModel.getHumanReadableLatestMessage());
-    trace?.update({
-        output: {
-            error: errorObject,
-        },
-    });
-    if (context) {
-        logError(context, error, trace);
-    }
+    trace?.update({ output: { error: errorObject } });
+    logError(error, trace);
 }
 function extractAttachments(userPrompt) {
     const attachments = [];

package/dist/agent/master/browser-tests/index.spec.js

@@ -72,7 +72,7 @@ fixtures_1.test.skip("master agent can click icons accurately", async ({ page, s
     // expect(icons.length).toBe(4); // 1 for each unique icon
     fs_1.default.unlinkSync(iconsRegistryFile);
 });
-(0, fixtures_1.test)("cua agent can click icons accurately", async ({ page, server }) => {
+fixtures_1.test.skip("cua agent can click icons accurately", async ({ page, server }) => {
     await page.goto(`${server.baseURL}/icons-navbar.html`);
     await (0, fixtures_1.expect)(page.getByText("select an icon")).toBeVisible();
     const response = await (0, run_1.createTestUsingComputerUseAgent)({

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

@@ -1 +1 @@
-{"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"}
+{"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"}