@empiricalrun/test-gen 0.71.1 → 0.72.0

package/CHANGELOG.md

@@ -1,5 +1,54 @@
 # @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
+
+- d7f9025: chore: remove old session reporter which is no longer used
+- f44505a: fix: smaller responses from run-test tool
+- 001b324: fix: enforce codegen approach for overlay dismissals
+- Updated dependencies [f44505a]
+  - @empiricalrun/test-run@0.10.8
+  - @empiricalrun/llm@0.19.4
+
 ## 0.71.1
 
 ### Patch Changes

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

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/actions/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAIhD,OAAO,EAAE,WAAW,EAAE,MAAM,SAAS,CAAC;AAEtC,OAAO,EAEL,UAAU,EACV,YAAY,EAEb,MAAM,UAAU,CAAC;AAWlB,qBAAa,iBAAiB;IAQ1B,OAAO,CAAC,IAAI;IACZ,OAAO,CAAC,cAAc;IARxB,OAAO,CAAC,gBAAgB,CAAwC;IAChE,OAAO,CAAC,eAAe,CAInB;gBAEM,IAAI,EAAE,WAAW,EACjB,cAAc,GAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAM;IAgB5C,aAAa,CACjB,IAAI,EAAE,MAAM,YAAK,EACjB,IAAI,EAAE,UAAU,EAChB,KAAK,CAAC,EAAE,WAAW,GAClB,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC;IA0D9B,wBAAwB,IAAI,YAAY,EAAE;IAoB1C,sBAAsB,IAAI,YAAY,EAAE;IAWxC,YAAY,IAAI;QACd,IAAI,EAAE,MAAM,CAAC;QACb,WAAW,EAAE,MAAM,EAAE,CAAC;KACvB;IAUD,gBAAgB,CAAC,KAAK,EAAE,MAAM;IAK9B;;;;;;OAMG;IACH,aAAa,IAAI,OAAO;IAQxB,UAAU;IAWV,iBAAiB;IAIjB,iBAAiB,CAAC,cAAc,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC;CAItD"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/actions/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAIhD,OAAO,EAAE,WAAW,EAAE,MAAM,SAAS,CAAC;AACtC,OAAO,EAEL,UAAU,EACV,YAAY,EAEb,MAAM,UAAU,CAAC;AAWlB,qBAAa,iBAAiB;IAQ1B,OAAO,CAAC,IAAI;IACZ,OAAO,CAAC,cAAc;IARxB,OAAO,CAAC,gBAAgB,CAAwC;IAChE,OAAO,CAAC,eAAe,CAInB;gBAEM,IAAI,EAAE,WAAW,EACjB,cAAc,GAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAM;IAgB5C,aAAa,CACjB,IAAI,EAAE,MAAM,YAAK,EACjB,IAAI,EAAE,UAAU,EAChB,KAAK,CAAC,EAAE,WAAW,GAClB,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC;IAqD9B,wBAAwB,IAAI,YAAY,EAAE;IAoB1C,sBAAsB,IAAI,YAAY,EAAE;IAWxC,YAAY,IAAI;QACd,IAAI,EAAE,MAAM,CAAC;QACb,WAAW,EAAE,MAAM,EAAE,CAAC;KACvB;IAUD,gBAAgB,CAAC,KAAK,EAAE,MAAM;IAK9B;;;;;;OAMG;IACH,aAAa,IAAI,OAAO;IAQxB,UAAU;IAWV,iBAAiB;IAIjB,iBAAiB,CAAC,cAAc,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC;CAItD"}

package/dist/actions/index.js

@@ -3,7 +3,6 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.PlaywrightActions = void 0;
 const logger_1 = require("../bin/logger");
 const errors_1 = require("../errors");
-const reporter_1 = require("../reporter");
 const assert_1 = require("./assert");
 const click_1 = require("./click");
 const done_1 = require("./done");
@@ -45,7 +44,6 @@ class PlaywrightActions {
             throw Error(`No action registered for action: ${name}`);
         }
         const logger = new logger_1.CustomLogger({ useReporter: false });
-        const testgenUpdatesReporter = new reporter_1.TestGenUpdatesReporter();
         logger.logEmptyLine();
         const actionSpan = trace?.span({
             name: "execute-action",
@@ -69,10 +67,6 @@ class PlaywrightActions {
                 },
             });
             this.recordedActions.push({ name, code, importPath });
-            if (code) {
-                logger.log(`action: ${name} \ncode: ${code} \nreason: ${args.reason}`);
-                void testgenUpdatesReporter.sendMessage("```ts\n" + code + "\n```");
-            }
             return code;
         }
         catch (e) {

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

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/agent/browsing/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAiB,GAAG,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AACpE,OAAO,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAElC,OAAO,EAAE,iBAAiB,EAAE,MAAM,eAAe,CAAC;AAMlD,MAAM,MAAM,kBAAkB,GAAG;IAC/B,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,MAAM,GAAG,SAAS,CAAC;CAC1B,CAAC;AAEF,wBAAsB,6BAA6B,CAAC,EAClD,MAAM,EACN,IAAI,EACJ,OAAO,EACP,GAAG,EACH,KAAK,GACN,EAAE;IACD,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,IAAI,CAAC;IACX,OAAO,EAAE,iBAAiB,CAAC;IAC3B,GAAG,CAAC,EAAE,GAAG,CAAC;IACV,KAAK,CAAC,EAAE,WAAW,CAAC;CACrB,GAAG,OAAO,CAAC,kBAAkB,GAAG,SAAS,CAAC,CA8D1C"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/agent/browsing/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAiB,GAAG,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AACpE,OAAO,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAElC,OAAO,EAAE,iBAAiB,EAAE,MAAM,eAAe,CAAC;AAKlD,MAAM,MAAM,kBAAkB,GAAG;IAC/B,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,MAAM,GAAG,SAAS,CAAC;CAC1B,CAAC;AAEF,wBAAsB,6BAA6B,CAAC,EAClD,MAAM,EACN,IAAI,EACJ,OAAO,EACP,GAAG,EACH,KAAK,GACN,EAAE;IACD,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,IAAI,CAAC;IACX,OAAO,EAAE,iBAAiB,CAAC;IAC3B,GAAG,CAAC,EAAE,GAAG,CAAC;IACV,KAAK,CAAC,EAAE,WAAW,CAAC;CACrB,GAAG,OAAO,CAAC,kBAAkB,GAAG,SAAS,CAAC,CAwD1C"}

package/dist/agent/browsing/index.js

@@ -4,12 +4,10 @@ exports.executeTaskUsingBrowsingAgent = executeTaskUsingBrowsingAgent;
 const llm_1 = require("@empiricalrun/llm");
 const constants_1 = require("../../constants");
 const promptTemplate_0 = "{{#section \"system\"}}\nYou are a browser automation agent who is given a task to generate code for navigation and assertion. This task is your\ngoal and you must achieve it.\n\nYou will be provided with already executed actions and basis that you need to pick the next step to achieve the task.\nRemember that the goal must be achieved.\n\nYou will be provided with the web page snapshot in the form of Document Object Model. Based on the goal and available\ntool calls you need to pick the appropriate tool call.\n\nInstructions:\n- Take actions one at a time. Do not try to take multiple actions\n- You can respond with multiple assertions in one shot\n- Do not repeat the same actions again otherwise your response will be marked INVALID\n- Avoid repeating errors which we got while executing the last action\n- Stick to the task provided to you and mark the task done once the task is complete\n- Do not execute any action which is not mentioned in the task\n- Do not repeat actions which are already executed more than twice otherwise your response will be marked INVALID\n- Always refer to \"Executed actions\" before deciding your next action for completion of the task.\n- End the task done if all actions required for task are executed\n{{/section}}\n\n{{#section \"user\"}}\nTask:\n{{task}}\n\nCurrent page snapshot:\n{{pageSnapshot}}\n{{/section}}";
-const reporter_1 = require("../../reporter");
 const html_1 = require("../../utils/html");
 async function executeTaskUsingBrowsingAgent({ action, page, actions, llm, trace, }) {
     let generatedCodeSteps;
     const tools = actions.getBrowsingActionSchemas();
-    const testgenUpdatesReporter = new reporter_1.TestGenUpdatesReporter();
     const browsingAgentSpan = trace?.span({
         name: `browsing-agent`,
         input: {
@@ -48,18 +46,12 @@ async function executeTaskUsingBrowsingAgent({ action, page, actions, llm, trace
     for (const i in toolCalls) {
         const toolCall = toolCalls[i];
         const args = JSON.parse(toolCall.function.arguments);
-        try {
-            const code = await actions.executeAction(toolCall.function.name, JSON.parse(toolCall.function.arguments), toolCallsSpan);
-            generatedCodeSteps = {
-                // Passing reason as action, in order to pass the correct action that took place to runtime planner
-                action: args.reason,
-                code,
-            };
-        }
-        catch (e) {
-            void testgenUpdatesReporter.sendMessage(e.message);
-            throw e;
-        }
+        const code = await actions.executeAction(toolCall.function.name, JSON.parse(toolCall.function.arguments), toolCallsSpan);
+        generatedCodeSteps = {
+            // Passing reason as action, in order to pass the correct action that took place to runtime planner
+            action: args.reason,
+            code,
+        };
     }
     toolCallsSpan?.end({ output: { toolCalls, generatedCodeSteps } });
     return generatedCodeSteps;

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"}