@empiricalrun/test-gen 0.75.0 → 0.76.0

package/CHANGELOG.md

@@ -1,5 +1,20 @@
 # @empiricalrun/test-gen
 
+## 0.76.0
+
+### Minor Changes
+
+- 9c9d6e5: feat: code review basic agent
+- ee8ce00: feat: fetchVideoAnalysis multi modal tool response
+- 86a1549: feat: code review agent v1 with workflow as an ad hoc tool call
+- 5339d2c: feat: VideoAnalysisAgent working in CLI with extractFrames tool
+
+### Patch Changes
+
+- a6aa6d8: fix: bug where video analysis was set to empty string
+  - @empiricalrun/llm@0.22.0
+  - @empiricalrun/test-run@0.11.1
+
 ## 0.75.0
 
 ### Minor Changes

package/dist/agent/base/index.d.ts

@@ -1,35 +1,42 @@
-import { IChatModel } from "@empiricalrun/llm/chat";
-import { FileInfo, PendingToolCall, ReporterFunction, StreamingMessageReporterFunc, SupportedChatModels, ToolDefinition, TraceClient } from "@empiricalrun/shared-types";
+import { type IChatModel } from "@empiricalrun/llm/chat";
+import type { CanonicalMessage, ChatState, FileInfo, PendingToolCall, ReporterFunction, StreamingMessageReporterFunc, SupportedChatModels, ToolDefinition, ToolResult, TraceClient } from "@empiricalrun/shared-types";
 type RunLoopParams<T> = {
-    chatModel: IChatModel<T>;
+    messages: CanonicalMessage[];
     reporter: ReporterFunction;
     streamingMessageReporter?: StreamingMessageReporterFunc;
     trace?: TraceClient;
     signal?: AbortSignal;
     onPendingToolCall: (toolCalls: PendingToolCall[]) => Promise<void>;
     onLLMResponse?: (response: T, selectedModel: SupportedChatModels) => Promise<void>;
+    repoInfoBuilder?: () => Promise<FileInfo>;
 };
-export declare const getChatAgentMode: (source: string | undefined) => "triage" | "test_generation";
-type AgentParams = {
-    featureFlags: string[];
+export type AgentParams = {
     selectedModel: SupportedChatModels;
-    repoInfo: FileInfo;
+    featureFlags?: string[];
+    workerEnv?: Record<string, string>;
 };
-export declare function getAgentForMode(isTriage: boolean, params: AgentParams): TriageAgent | ChatAgent;
-export declare class ChatAgent {
-    featureFlags: string[];
-    tools: ToolDefinition[];
-    selectedModel: SupportedChatModels;
-    repoInfo: FileInfo;
-    constructor({ featureFlags, selectedModel, repoInfo }: AgentParams);
-    runLoop(params: RunLoopParams<any>): Promise<void>;
-}
-export declare class TriageAgent {
+export declare abstract class BaseAgent {
     featureFlags: string[];
     tools: ToolDefinition[];
     selectedModel: SupportedChatModels;
-    repoInfo: FileInfo;
-    constructor({ featureFlags, selectedModel, repoInfo }: AgentParams);
+    protected chatModel?: IChatModel<any>;
+    protected workerEnv?: Record<string, string>;
+    protected currentChatState?: ChatState;
+    constructor({ featureFlags, selectedModel, workerEnv }: AgentParams);
+    protected abstract getTools(): ToolDefinition[];
+    protected abstract buildSystemPrompt(repoContext?: string): Promise<string>;
+    get messages(): CanonicalMessage[] | undefined;
+    get askUserForInput(): boolean | undefined;
+    private rejectPendingToolCallsAndUpdateState;
+    private processValidToolCallsAndUpdateState;
+    initializeWithState(chatState: ChatState): void;
+    stop(): ChatState;
+    processToolCallsFromState({ toolCalls, toolResults, chatState, }: {
+        toolCalls: PendingToolCall[];
+        toolResults: ToolResult[];
+        chatState: ChatState;
+    }): ChatState | undefined;
+    processToolResults(toolCalls: PendingToolCall[], toolResults: ToolResult[]): ChatState | undefined;
     runLoop(params: RunLoopParams<any>): Promise<void>;
 }
 export {};

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

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/agent/base/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AACpD,OAAO,EACL,QAAQ,EACR,eAAe,EACf,gBAAgB,EAChB,4BAA4B,EAC5B,mBAAmB,EACnB,cAAc,EACd,WAAW,EACZ,MAAM,4BAA4B,CAAC;AAMpC,KAAK,aAAa,CAAC,CAAC,IAAI;IACtB,SAAS,EAAE,UAAU,CAAC,CAAC,CAAC,CAAC;IACzB,QAAQ,EAAE,gBAAgB,CAAC;IAC3B,wBAAwB,CAAC,EAAE,4BAA4B,CAAC;IACxD,KAAK,CAAC,EAAE,WAAW,CAAC;IACpB,MAAM,CAAC,EAAE,WAAW,CAAC;IACrB,iBAAiB,EAAE,CAAC,SAAS,EAAE,eAAe,EAAE,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACnE,aAAa,CAAC,EAAE,CACd,QAAQ,EAAE,CAAC,EACX,aAAa,EAAE,mBAAmB,KAC/B,OAAO,CAAC,IAAI,CAAC,CAAC;CACpB,CAAC;AAEF,eAAO,MAAM,gBAAgB,GAAI,QAAQ,MAAM,GAAG,SAAS,iCAW1D,CAAC;AAEF,KAAK,WAAW,GAAG;IACjB,YAAY,EAAE,MAAM,EAAE,CAAC;IACvB,aAAa,EAAE,mBAAmB,CAAC;IACnC,QAAQ,EAAE,QAAQ,CAAC;CACpB,CAAC;AAEF,wBAAgB,eAAe,CAAC,QAAQ,EAAE,OAAO,EAAE,MAAM,EAAE,WAAW,2BAMrE;AAED,qBAAa,SAAS;IACpB,YAAY,EAAE,MAAM,EAAE,CAAM;IAC5B,KAAK,EAAE,cAAc,EAAE,CAAM;IAC7B,aAAa,EAAE,mBAAmB,CAAC;IACnC,QAAQ,EAAE,QAAQ,CAAC;gBAEP,EAAE,YAAY,EAAE,aAAa,EAAE,QAAQ,EAAE,EAAE,WAAW;IAO5D,OAAO,CAAC,MAAM,EAAE,aAAa,CAAC,GAAG,CAAC;CAUzC;AAED,qBAAa,WAAW;IACtB,YAAY,EAAE,MAAM,EAAE,CAAM;IAC5B,KAAK,EAAE,cAAc,EAAE,CAAM;IAC7B,aAAa,EAAE,mBAAmB,CAAC;IACnC,QAAQ,EAAE,QAAQ,CAAC;gBAEP,EAAE,YAAY,EAAE,aAAa,EAAE,QAAQ,EAAE,EAAE,WAAW;IAO5D,OAAO,CAAC,MAAM,EAAE,aAAa,CAAC,GAAG,CAAC;CAUzC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/agent/base/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAmB,KAAK,UAAU,EAAE,MAAM,wBAAwB,CAAC;AAC1E,OAAO,KAAK,EACV,gBAAgB,EAChB,SAAS,EACT,QAAQ,EACR,eAAe,EACf,gBAAgB,EAChB,4BAA4B,EAC5B,mBAAmB,EACnB,cAAc,EACd,UAAU,EACV,WAAW,EACZ,MAAM,4BAA4B,CAAC;AAOpC,KAAK,aAAa,CAAC,CAAC,IAAI;IACtB,QAAQ,EAAE,gBAAgB,EAAE,CAAC;IAC7B,QAAQ,EAAE,gBAAgB,CAAC;IAC3B,wBAAwB,CAAC,EAAE,4BAA4B,CAAC;IACxD,KAAK,CAAC,EAAE,WAAW,CAAC;IACpB,MAAM,CAAC,EAAE,WAAW,CAAC;IACrB,iBAAiB,EAAE,CAAC,SAAS,EAAE,eAAe,EAAE,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACnE,aAAa,CAAC,EAAE,CACd,QAAQ,EAAE,CAAC,EACX,aAAa,EAAE,mBAAmB,KAC/B,OAAO,CAAC,IAAI,CAAC,CAAC;IACnB,eAAe,CAAC,EAAE,MAAM,OAAO,CAAC,QAAQ,CAAC,CAAC;CAC3C,CAAC;AAEF,MAAM,MAAM,WAAW,GAAG;IACxB,aAAa,EAAE,mBAAmB,CAAC;IACnC,YAAY,CAAC,EAAE,MAAM,EAAE,CAAC;IACxB,SAAS,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;CACpC,CAAC;AAEF,8BAAsB,SAAS;IAC7B,YAAY,EAAE,MAAM,EAAE,CAAM;IAC5B,KAAK,EAAE,cAAc,EAAE,CAAM;IAC7B,aAAa,EAAE,mBAAmB,CAAC;IACnC,SAAS,CAAC,SAAS,CAAC,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC;IACtC,SAAS,CAAC,SAAS,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC7C,SAAS,CAAC,gBAAgB,CAAC,EAAE,SAAS,CAAC;gBAE3B,EAAE,YAAY,EAAE,aAAa,EAAE,SAAS,EAAE,EAAE,WAAW;IAOnE,SAAS,CAAC,QAAQ,CAAC,QAAQ,IAAI,cAAc,EAAE;IAC/C,SAAS,CAAC,QAAQ,CAAC,iBAAiB,CAAC,WAAW,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAE3E,IAAI,QAAQ,IAAI,gBAAgB,EAAE,GAAG,SAAS,CAE7C;IAED,IAAI,eAAe,IAAI,OAAO,GAAG,SAAS,CAEzC;IAED,OAAO,CAAC,oCAAoC;IA4B5C,OAAO,CAAC,mCAAmC;IA6B3C,mBAAmB,CAAC,SAAS,EAAE,SAAS,GAAG,IAAI;IAS/C,IAAI,IAAI,SAAS;IAWjB,yBAAyB,CAAC,EACxB,SAAS,EACT,WAAW,EACX,SAAS,GACV,EAAE;QACD,SAAS,EAAE,eAAe,EAAE,CAAC;QAC7B,WAAW,EAAE,UAAU,EAAE,CAAC;QAC1B,SAAS,EAAE,SAAS,CAAC;KACtB,GAAG,SAAS,GAAG,SAAS;IAUzB,kBAAkB,CAChB,SAAS,EAAE,eAAe,EAAE,EAC5B,WAAW,EAAE,UAAU,EAAE,GACxB,SAAS,GAAG,SAAS;IAalB,OAAO,CAAC,MAAM,EAAE,aAAa,CAAC,GAAG,CAAC;CAqBzC"}

package/dist/agent/base/index.js

@@ -1,74 +1,106 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.TriageAgent = exports.ChatAgent = exports.getChatAgentMode = void 0;
-exports.getAgentForMode = getAgentForMode;
-const tools_1 = require("../../tools");
+exports.BaseAgent = void 0;
+const chat_1 = require("@empiricalrun/llm/chat");
 const agent_loop_1 = require("../chat/agent-loop");
-const prompt_1 = require("../chat/prompt");
-const getChatAgentMode = (source) => {
-    switch (source) {
-        case "triage":
-            return "triage";
-        case "dashboard":
-            return "test_generation";
-        case "manual":
-            return "test_generation";
-        default:
-            return "test_generation";
-    }
-};
-exports.getChatAgentMode = getChatAgentMode;
-function getAgentForMode(isTriage, params) {
-    if (isTriage) {
-        return new TriageAgent(params);
-    }
-    else {
-        return new ChatAgent(params);
-    }
-}
-class ChatAgent {
+const repo_1 = require("../chat/prompt/repo");
+const REJECTION_RESULT = "This tool call was rejected by the user.";
+class BaseAgent {
     featureFlags = [];
     tools = [];
     selectedModel;
-    repoInfo;
-    constructor({ featureFlags, selectedModel, repoInfo }) {
-        this.featureFlags = featureFlags;
-        this.tools = (0, tools_1.availableTools)(selectedModel, false, []);
+    chatModel;
+    workerEnv;
+    currentChatState;
+    constructor({ featureFlags, selectedModel, workerEnv }) {
+        this.featureFlags = featureFlags || [];
         this.selectedModel = selectedModel;
-        this.repoInfo = repoInfo;
+        this.workerEnv = workerEnv;
+        this.tools = this.getTools();
     }
-    async runLoop(params) {
-        const mode = (0, exports.getChatAgentMode)("test_generation");
-        return (0, agent_loop_1.chatAgentLoop)({
-            ...params,
-            tools: this.tools,
-            selectedModel: this.selectedModel,
-            systemPrompt: await (0, prompt_1.getSystemPrompt)(this.repoInfo, mode),
-            featureFlags: this.featureFlags,
+    get messages() {
+        return this.chatModel?.messages;
+    }
+    get askUserForInput() {
+        return this.chatModel?.askUserForInput;
+    }
+    rejectPendingToolCallsAndUpdateState(chatModel, chatState) {
+        const pendingToolCalls = chatModel.getPendingToolCalls();
+        let updatedChatState = {
+            ...chatState,
+            messages: chatModel.messages,
+            askUserForInput: true,
+        };
+        if (pendingToolCalls.length > 0) {
+            const toolResults = pendingToolCalls.map(() => ({
+                isError: true,
+                result: REJECTION_RESULT,
+                artifacts: null,
+            }));
+            chatModel.pushToolResultsMessage(pendingToolCalls, toolResults);
+            updatedChatState = {
+                ...updatedChatState,
+                messages: chatModel.messages,
+            };
+        }
+        return updatedChatState;
+    }
+    processValidToolCallsAndUpdateState(chatModel, chatState, toolCalls, toolResults) {
+        const pendingToolCalls = chatModel.getPendingToolCalls() || [];
+        // Filter tool calls to only include pending ones
+        const filteredToolResults = [];
+        const filteredToolCalls = toolCalls.filter((call, index) => {
+            if (pendingToolCalls.some((p) => p.id === call.id)) {
+                filteredToolResults.push(toolResults[index]);
+                return true;
+            }
+            return false;
         });
+        if (filteredToolCalls.length > 0) {
+            chatModel.pushToolResultsMessage(filteredToolCalls, filteredToolResults);
+            return {
+                ...chatState,
+                messages: chatModel.messages,
+                askUserForInput: chatModel.askUserForInput,
+            };
+        }
+        return undefined;
     }
-}
-exports.ChatAgent = ChatAgent;
-class TriageAgent {
-    featureFlags = [];
-    tools = [];
-    selectedModel;
-    repoInfo;
-    constructor({ featureFlags, selectedModel, repoInfo }) {
-        this.featureFlags = featureFlags;
-        this.tools = (0, tools_1.availableTools)(selectedModel, true, []);
-        this.selectedModel = selectedModel;
-        this.repoInfo = repoInfo;
+    initializeWithState(chatState) {
+        this.currentChatState = chatState;
+        this.chatModel = (0, chat_1.createChatModel)(chatState.messages, this.selectedModel, this.workerEnv);
+    }
+    stop() {
+        if (!this.chatModel || !this.currentChatState) {
+            throw new Error("Agent not initialized - call initializeWithState first");
+        }
+        return this.rejectPendingToolCallsAndUpdateState(this.chatModel, this.currentChatState);
+    }
+    processToolCallsFromState({ toolCalls, toolResults, chatState, }) {
+        const tempChatModel = (0, chat_1.createChatModel)(chatState.messages, chatState.model);
+        return this.processValidToolCallsAndUpdateState(tempChatModel, chatState, toolCalls, toolResults);
+    }
+    processToolResults(toolCalls, toolResults) {
+        if (!this.chatModel || !this.currentChatState) {
+            throw new Error("Agent not initialized - call initializeWithState first");
+        }
+        return this.processValidToolCallsAndUpdateState(this.chatModel, this.currentChatState, toolCalls, toolResults);
     }
     async runLoop(params) {
-        const mode = (0, exports.getChatAgentMode)("triage");
-        return (0, agent_loop_1.chatAgentLoop)({
+        this.chatModel = (0, chat_1.createChatModel)(params.messages, this.selectedModel, this.workerEnv);
+        let repoContext;
+        if (params.repoInfoBuilder) {
+            const repoInfo = await params.repoInfoBuilder();
+            repoContext = await (0, repo_1.getRepoInfoPrompt)(repoInfo);
+        }
+        const systemPrompt = await this.buildSystemPrompt(repoContext);
+        await (0, agent_loop_1.chatAgentLoop)({
             ...params,
+            chatModel: this.chatModel,
             tools: this.tools,
-            selectedModel: this.selectedModel,
-            systemPrompt: await (0, prompt_1.getSystemPrompt)(this.repoInfo, mode),
+            systemPrompt,
             featureFlags: this.featureFlags,
         });
     }
 }
-exports.TriageAgent = TriageAgent;
+exports.BaseAgent = BaseAgent;

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

@@ -1,9 +1,8 @@
 import { TraceClient } from "@empiricalrun/llm";
 import { IChatModel } from "@empiricalrun/llm/chat";
 import { CanonicalMessage, PendingToolCall, ReporterFunction, StreamingMessageReporterFunc, SupportedChatModels, ToolDefinition } from "@empiricalrun/shared-types";
-export declare function chatAgentLoop<T = CanonicalMessage>({ chatModel, selectedModel, reporter, streamingMessageReporter, trace, signal, featureFlags, tools, onPendingToolCall, onLLMResponse, systemPrompt, }: {
+type AgentLoopParams<T> = {
     chatModel: IChatModel<T>;
-    selectedModel: SupportedChatModels;
     reporter: ReporterFunction;
     streamingMessageReporter?: StreamingMessageReporterFunc;
     trace?: TraceClient;
@@ -13,5 +12,7 @@ export declare function chatAgentLoop<T = CanonicalMessage>({ chatModel, selecte
     onPendingToolCall: (toolCalls: PendingToolCall[]) => Promise<void>;
     onLLMResponse?: (response: T, selectedModel: SupportedChatModels) => Promise<void>;
     systemPrompt: string;
-}): Promise<void>;
+};
+export declare function chatAgentLoop<T = CanonicalMessage>({ chatModel, reporter, streamingMessageReporter, trace, signal, featureFlags, tools, onPendingToolCall, onLLMResponse, systemPrompt, }: AgentLoopParams<T>): Promise<void>;
+export {};
 //# 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,EAAE,MAAM,wBAAwB,CAAC;AACpD,OAAO,EACL,gBAAgB,EAChB,eAAe,EACf,gBAAgB,EAChB,4BAA4B,EAC5B,mBAAmB,EACnB,cAAc,EACf,MAAM,4BAA4B,CAAC;AAKpC,wBAAsB,aAAa,CAAC,CAAC,GAAG,gBAAgB,EAAE,EACxD,SAAS,EACT,aAAa,EACb,QAAQ,EACR,wBAAwB,EACxB,KAAK,EACL,MAAM,EACN,YAAY,EACZ,KAAK,EACL,iBAAiB,EACjB,aAAa,EACb,YAAY,GACb,EAAE;IACD,SAAS,EAAE,UAAU,CAAC,CAAC,CAAC,CAAC;IACzB,aAAa,EAAE,mBAAmB,CAAC;IACnC,QAAQ,EAAE,gBAAgB,CAAC;IAC3B,wBAAwB,CAAC,EAAE,4BAA4B,CAAC;IACxD,KAAK,CAAC,EAAE,WAAW,CAAC;IACpB,MAAM,CAAC,EAAE,WAAW,CAAC;IACrB,YAAY,EAAE,MAAM,EAAE,CAAC;IACvB,KAAK,EAAE,cAAc,EAAE,CAAC;IACxB,iBAAiB,EAAE,CAAC,SAAS,EAAE,eAAe,EAAE,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACnE,aAAa,CAAC,EAAE,CACd,QAAQ,EAAE,CAAC,EACX,aAAa,EAAE,mBAAmB,KAC/B,OAAO,CAAC,IAAI,CAAC,CAAC;IACnB,YAAY,EAAE,MAAM,CAAC;CACtB,GAAG,OAAO,CAAC,IAAI,CAAC,CA2DhB"}
+{"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,EAAE,MAAM,wBAAwB,CAAC;AACpD,OAAO,EACL,gBAAgB,EAChB,eAAe,EACf,gBAAgB,EAChB,4BAA4B,EAC5B,mBAAmB,EACnB,cAAc,EACf,MAAM,4BAA4B,CAAC;AAKpC,KAAK,eAAe,CAAC,CAAC,IAAI;IACxB,SAAS,EAAE,UAAU,CAAC,CAAC,CAAC,CAAC;IACzB,QAAQ,EAAE,gBAAgB,CAAC;IAC3B,wBAAwB,CAAC,EAAE,4BAA4B,CAAC;IACxD,KAAK,CAAC,EAAE,WAAW,CAAC;IACpB,MAAM,CAAC,EAAE,WAAW,CAAC;IACrB,YAAY,EAAE,MAAM,EAAE,CAAC;IACvB,KAAK,EAAE,cAAc,EAAE,CAAC;IACxB,iBAAiB,EAAE,CAAC,SAAS,EAAE,eAAe,EAAE,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACnE,aAAa,CAAC,EAAE,CACd,QAAQ,EAAE,CAAC,EACX,aAAa,EAAE,mBAAmB,KAC/B,OAAO,CAAC,IAAI,CAAC,CAAC;IACnB,YAAY,EAAE,MAAM,CAAC;CACtB,CAAC;AAEF,wBAAsB,aAAa,CAAC,CAAC,GAAG,gBAAgB,EAAE,EACxD,SAAS,EACT,QAAQ,EACR,wBAAwB,EACxB,KAAK,EACL,MAAM,EACN,YAAY,EACZ,KAAK,EACL,iBAAiB,EACjB,aAAa,EACb,YAAY,GACb,EAAE,eAAe,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC,CAkDpC"}

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

@@ -3,7 +3,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.chatAgentLoop = chatAgentLoop;
 const state_1 = require("./state");
 const utils_1 = require("./utils");
-async function chatAgentLoop({ chatModel, selectedModel, reporter, streamingMessageReporter, trace, signal, featureFlags, tools, onPendingToolCall, onLLMResponse, systemPrompt, }) {
+async function chatAgentLoop({ chatModel, reporter, streamingMessageReporter, trace, signal, featureFlags, tools, onPendingToolCall, onLLMResponse, systemPrompt, }) {
     trace?.update({ input: { systemPrompt } });
     while (!chatModel.askUserForInput) {
         if (signal?.aborted) {
@@ -16,11 +16,10 @@ async function chatAgentLoop({ chatModel, selectedModel, reporter, streamingMess
                 await onPendingToolCall(toolCalls);
                 break;
             }
-            (0, utils_1.log)(`${(0, utils_1.getModelName)(selectedModel)} is working...`);
+            (0, utils_1.log)(`${(0, utils_1.getModelName)(chatModel.selectedModel)} is working...`);
             const response = await chatModel.getLLMResponse({
                 systemPrompt,
                 tools: tools.map((t) => t.schema),
-                selectedModel,
                 trace,
                 streamingMessageReporter,
                 hasThinkingEnabled: true,
@@ -31,24 +30,19 @@ async function chatAgentLoop({ chatModel, selectedModel, reporter, streamingMess
             }
             chatModel.pushMessage(response);
             if (onLLMResponse) {
-                await onLLMResponse(response, selectedModel);
+                await onLLMResponse(response, chatModel.selectedModel);
             }
             const latest = chatModel.getHumanReadableLatestMessage();
             if (signal?.aborted) {
                 (0, utils_1.log)("[chatAgentLoop] Aborted by signal");
                 break;
             }
-            await reporter((0, state_1.chatStateFromModel)({
-                chatModel,
-                selectedModel,
-                error: null,
-            }), latest);
+            await reporter((0, state_1.chatStateFromModel)({ chatModel, error: null }), latest);
         }
         catch (error) {
             await (0, utils_1.handleAgentError)({
                 error,
                 chatModel,
-                selectedModel,
                 reporter,
                 trace,
             });

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

@@ -1,10 +1,12 @@
 export { getFileInfoFromGitHub, viewFileUsingGitHub, } from "../../file-info/adapters/github";
 export { allToolsDefinitions } from "../../tools";
-export { ChatAgent, getAgentForMode, TriageAgent } from "../base";
+export { BaseAgent } from "../base";
+export { MODE_TO_AGENT_MAP } from "../index";
+export { TriageAgent } from "../triage";
+export { ChatAgent } from "./index";
 export { getDefaultChatModelId, resolveChatModelBasedOnInput } from "./models";
 export { CHAT_STATE_VERSIONS_MIGRATIONS_MAP, chatStateFromModel, createChatState, createChatStateForMessages, fetchToolCallFromId, getLatestDownloadBuildUrl, LATEST_CHAT_STATE_VERSION, migrateChatState, } from "./state";
 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,OAAO,EACL,qBAAqB,EACrB,mBAAmB,GACpB,MAAM,iCAAiC,CAAC;AACzC,OAAO,EAAE,mBAAmB,EAAE,MAAM,aAAa,CAAC;AAClD,OAAO,EAAE,SAAS,EAAE,eAAe,EAAE,WAAW,EAAE,MAAM,SAAS,CAAC;AAClE,OAAO,EAAE,qBAAqB,EAAE,4BAA4B,EAAE,MAAM,UAAU,CAAC;AAC/E,OAAO,EACL,kCAAkC,EAClC,kBAAkB,EAClB,eAAe,EACf,0BAA0B,EAC1B,mBAAmB,EACnB,yBAAyB,EACzB,yBAAyB,EACzB,gBAAgB,GACjB,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"}
+{"version":3,"file":"exports.d.ts","sourceRoot":"","sources":["../../../src/agent/chat/exports.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,qBAAqB,EACrB,mBAAmB,GACpB,MAAM,iCAAiC,CAAC;AACzC,OAAO,EAAE,mBAAmB,EAAE,MAAM,aAAa,CAAC;AAClD,OAAO,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AACpC,OAAO,EAAE,iBAAiB,EAAE,MAAM,UAAU,CAAC;AAC7C,OAAO,EAAE,WAAW,EAAE,MAAM,WAAW,CAAC;AACxC,OAAO,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AACpC,OAAO,EAAE,qBAAqB,EAAE,4BAA4B,EAAE,MAAM,UAAU,CAAC;AAC/E,OAAO,EACL,kCAAkC,EAClC,kBAAkB,EAClB,eAAe,EACf,0BAA0B,EAC1B,mBAAmB,EACnB,yBAAyB,EACzB,yBAAyB,EACzB,gBAAgB,GACjB,MAAM,SAAS,CAAC;AACjB,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"}

package/dist/agent/chat/exports.js

@@ -1,15 +1,19 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.SUPPORTED_CHAT_MODELS = exports.createChatModel = exports.rejectPendingToolCalls = exports.processToolCallsAndUpdateChatState = exports.extractAttachments = exports.migrateChatState = exports.LATEST_CHAT_STATE_VERSION = exports.getLatestDownloadBuildUrl = exports.fetchToolCallFromId = exports.createChatStateForMessages = exports.createChatState = exports.chatStateFromModel = exports.CHAT_STATE_VERSIONS_MIGRATIONS_MAP = exports.resolveChatModelBasedOnInput = exports.getDefaultChatModelId = exports.TriageAgent = exports.getAgentForMode = exports.ChatAgent = exports.allToolsDefinitions = exports.viewFileUsingGitHub = exports.getFileInfoFromGitHub = void 0;
+exports.SUPPORTED_CHAT_MODELS = exports.createChatModel = exports.extractAttachments = exports.migrateChatState = exports.LATEST_CHAT_STATE_VERSION = exports.getLatestDownloadBuildUrl = exports.fetchToolCallFromId = exports.createChatStateForMessages = exports.createChatState = exports.chatStateFromModel = exports.CHAT_STATE_VERSIONS_MIGRATIONS_MAP = exports.resolveChatModelBasedOnInput = exports.getDefaultChatModelId = exports.ChatAgent = exports.TriageAgent = exports.MODE_TO_AGENT_MAP = exports.BaseAgent = exports.allToolsDefinitions = exports.viewFileUsingGitHub = exports.getFileInfoFromGitHub = void 0;
 var github_1 = require("../../file-info/adapters/github");
 Object.defineProperty(exports, "getFileInfoFromGitHub", { enumerable: true, get: function () { return github_1.getFileInfoFromGitHub; } });
 Object.defineProperty(exports, "viewFileUsingGitHub", { enumerable: true, get: function () { return github_1.viewFileUsingGitHub; } });
 var tools_1 = require("../../tools");
 Object.defineProperty(exports, "allToolsDefinitions", { enumerable: true, get: function () { return tools_1.allToolsDefinitions; } });
 var base_1 = require("../base");
-Object.defineProperty(exports, "ChatAgent", { enumerable: true, get: function () { return base_1.ChatAgent; } });
-Object.defineProperty(exports, "getAgentForMode", { enumerable: true, get: function () { return base_1.getAgentForMode; } });
-Object.defineProperty(exports, "TriageAgent", { enumerable: true, get: function () { return base_1.TriageAgent; } });
+Object.defineProperty(exports, "BaseAgent", { enumerable: true, get: function () { return base_1.BaseAgent; } });
+var index_1 = require("../index");
+Object.defineProperty(exports, "MODE_TO_AGENT_MAP", { enumerable: true, get: function () { return index_1.MODE_TO_AGENT_MAP; } });
+var triage_1 = require("../triage");
+Object.defineProperty(exports, "TriageAgent", { enumerable: true, get: function () { return triage_1.TriageAgent; } });
+var index_2 = require("./index");
+Object.defineProperty(exports, "ChatAgent", { enumerable: true, get: function () { return index_2.ChatAgent; } });
 var models_1 = require("./models");
 Object.defineProperty(exports, "getDefaultChatModelId", { enumerable: true, get: function () { return models_1.getDefaultChatModelId; } });
 Object.defineProperty(exports, "resolveChatModelBasedOnInput", { enumerable: true, get: function () { return models_1.resolveChatModelBasedOnInput; } });
@@ -24,9 +28,6 @@ Object.defineProperty(exports, "LATEST_CHAT_STATE_VERSION", { enumerable: true,
 Object.defineProperty(exports, "migrateChatState", { enumerable: true, get: function () { return state_1.migrateChatState; } });
 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

@@ -1,11 +1,7 @@
-import { SupportedChatModels } from "@empiricalrun/shared-types";
-export declare function fetchEnvironmentVariables(): Promise<Record<string, string>>;
-export declare function runChatAgentForCLI({ useDiskForChatState, selectedModel, initialPromptContent, useTriage, resetChat, useFSCache, }: {
-    selectedModel: SupportedChatModels;
-    useDiskForChatState: boolean;
-    initialPromptContent: string | undefined;
-    useTriage: boolean;
-    useFSCache: boolean;
-    resetChat: boolean;
-}): Promise<void>;
+import type { ToolDefinition } from "@empiricalrun/shared-types";
+import { BaseAgent } from "../base";
+export declare class ChatAgent extends BaseAgent {
+    protected getTools(): ToolDefinition[];
+    protected buildSystemPrompt(repoContext?: string): Promise<string>;
+}
 //# sourceMappingURL=index.d.ts.map

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,EAML,mBAAmB,EACpB,MAAM,4BAA4B,CAAC;AAwCpC,wBAAsB,yBAAyB,IAAI,OAAO,CACxD,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CACvB,CAkCA;AAED,wBAAsB,kBAAkB,CAAC,EACvC,mBAAmB,EACnB,aAAa,EACb,oBAAoB,EACpB,SAAS,EACT,SAAS,EACT,UAAU,GACX,EAAE;IACD,aAAa,EAAE,mBAAmB,CAAC;IACnC,mBAAmB,EAAE,OAAO,CAAC;IAC7B,oBAAoB,EAAE,MAAM,GAAG,SAAS,CAAC;IACzC,SAAS,EAAE,OAAO,CAAC;IACnB,UAAU,EAAE,OAAO,CAAC;IACpB,SAAS,EAAE,OAAO,CAAC;CACpB,iBAyKA"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/agent/chat/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;AAQjE,OAAO,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AAGpC,qBAAa,SAAU,SAAQ,SAAS;IACtC,SAAS,CAAC,QAAQ,IAAI,cAAc,EAAE;cAmBtB,iBAAiB,CAAC,WAAW,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;CAkHzE"}