@empiricalrun/test-gen 0.74.2 → 0.76.0

package/CHANGELOG.md

@@ -1,5 +1,42 @@
 # @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
+
+- dde3517: feat: trace zip req & res body update from network trace
+- 8668785: feat: enabled params for video analysis tool
+
+### Patch Changes
+
+- 8c1935f: chore: feature flag in fetchVideo tool, UI changes to accomodate analysis on click with params
+- ab1be01: chore: introduce agent class
+- 9f1cb66: chore: enable none as flag value, lint changes
+- 1a7f487: fix: truncating the request body before appending to req res content in trace zip tool
+- e3f7028: feat: add ff to allow video analysis in test-gen session (admin only)
+- Updated dependencies [2a06e9b]
+- Updated dependencies [9f1cb66]
+- Updated dependencies [9fb0ba1]
+- Updated dependencies [8668785]
+  - @empiricalrun/llm@0.22.0
+  - @empiricalrun/test-run@0.11.1
+  - @empiricalrun/r2-uploader@0.3.11
+
 ## 0.74.2
 
 ### Patch Changes

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

@@ -0,0 +1,43 @@
+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> = {
+    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 type AgentParams = {
+    selectedModel: SupportedChatModels;
+    featureFlags?: string[];
+    workerEnv?: Record<string, string>;
+};
+export declare abstract class BaseAgent {
+    featureFlags: string[];
+    tools: ToolDefinition[];
+    selectedModel: SupportedChatModels;
+    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 {};
+//# sourceMappingURL=index.d.ts.map

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

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

@@ -0,0 +1,106 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.BaseAgent = void 0;
+const chat_1 = require("@empiricalrun/llm/chat");
+const agent_loop_1 = require("../chat/agent-loop");
+const repo_1 = require("../chat/prompt/repo");
+const REJECTION_RESULT = "This tool call was rejected by the user.";
+class BaseAgent {
+    featureFlags = [];
+    tools = [];
+    selectedModel;
+    chatModel;
+    workerEnv;
+    currentChatState;
+    constructor({ featureFlags, selectedModel, workerEnv }) {
+        this.featureFlags = featureFlags || [];
+        this.selectedModel = selectedModel;
+        this.workerEnv = workerEnv;
+        this.tools = this.getTools();
+    }
+    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;
+    }
+    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) {
+        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,
+            systemPrompt,
+            featureFlags: this.featureFlags,
+        });
+    }
+}
+exports.BaseAgent = BaseAgent;

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

@@ -1,18 +1,18 @@
 import { TraceClient } from "@empiricalrun/llm";
 import { IChatModel } from "@empiricalrun/llm/chat";
-import { CanonicalMessage, FileInfo, PendingToolCall, ReporterFunction, StreamingMessageReporterFunc, SupportedChatModels, ToolDefinition } from "@empiricalrun/shared-types";
-export declare function chatAgentLoop<T = CanonicalMessage>({ chatModel, selectedModel, reporter, streamingMessageReporter, trace, repoInfo, signal, featureFlags, tools, onPendingToolCall, onLLMResponse, mode, }: {
+import { CanonicalMessage, PendingToolCall, ReporterFunction, StreamingMessageReporterFunc, SupportedChatModels, ToolDefinition } from "@empiricalrun/shared-types";
+type AgentLoopParams<T> = {
     chatModel: IChatModel<T>;
-    selectedModel: SupportedChatModels;
     reporter: ReporterFunction;
     streamingMessageReporter?: StreamingMessageReporterFunc;
     trace?: TraceClient;
-    repoInfo: FileInfo;
     signal?: AbortSignal;
     featureFlags: string[];
-    tools?: ToolDefinition[];
+    tools: ToolDefinition[];
     onPendingToolCall: (toolCalls: PendingToolCall[]) => Promise<void>;
     onLLMResponse?: (response: T, selectedModel: SupportedChatModels) => Promise<void>;
-    mode: string;
-}): Promise<void>;
+    systemPrompt: string;
+};
+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,QAAQ,EACR,eAAe,EACf,gBAAgB,EAChB,4BAA4B,EAC5B,mBAAmB,EACnB,cAAc,EACf,MAAM,4BAA4B,CAAC;AAOpC,wBAAsB,aAAa,CAAC,CAAC,GAAG,gBAAgB,EAAE,EACxD,SAAS,EACT,aAAa,EACb,QAAQ,EACR,wBAAwB,EACxB,KAAK,EACL,QAAQ,EACR,MAAM,EACN,YAAY,EACZ,KAAK,EACL,iBAAiB,EACjB,aAAa,EACb,IAAI,GACL,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,QAAQ,EAAE,QAAQ,CAAC;IACnB,MAAM,CAAC,EAAE,WAAW,CAAC;IACrB,YAAY,EAAE,MAAM,EAAE,CAAC;IACvB,KAAK,CAAC,EAAE,cAAc,EAAE,CAAC;IACzB,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,IAAI,EAAE,MAAM,CAAC;CACd,GAAG,OAAO,CAAC,IAAI,CAAC,CAmEhB"}
+{"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

@@ -1,17 +1,10 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.chatAgentLoop = chatAgentLoop;
-const tools_1 = require("../../tools");
-const prompt_1 = require("./prompt");
 const state_1 = require("./state");
 const utils_1 = require("./utils");
-async function chatAgentLoop({ chatModel, selectedModel, reporter, streamingMessageReporter, trace, repoInfo, signal, featureFlags, tools, onPendingToolCall, onLLMResponse, mode, }) {
-    const systemPrompt = await (0, prompt_1.getSystemPrompt)(repoInfo, mode);
+async function chatAgentLoop({ chatModel, reporter, streamingMessageReporter, trace, signal, featureFlags, tools, onPendingToolCall, onLLMResponse, systemPrompt, }) {
     trace?.update({ input: { systemPrompt } });
-    if (!mode && !tools) {
-        throw new Error("Either mode or tools must be provided");
-    }
-    const availableToolsData = tools || (0, tools_1.availableTools)(selectedModel, mode === "triage");
     while (!chatModel.askUserForInput) {
         if (signal?.aborted) {
             (0, utils_1.log)("[chatAgentLoop] Aborted by signal");
@@ -23,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: availableToolsData.map((t) => t.schema),
-                selectedModel,
+                tools: tools.map((t) => t.schema),
                 trace,
                 streamingMessageReporter,
                 hasThinkingEnabled: true,
@@ -38,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 { chatAgentLoop } from "./agent-loop";
+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, processAttachments, } from "./state";
-export { extractAttachments, getChatAgentMode } from "./utils";
-export { processToolCallsAndUpdateChatState, rejectPendingToolCalls, } from "./utils/tool-calls";
+export { CHAT_STATE_VERSIONS_MIGRATIONS_MAP, chatStateFromModel, createChatState, createChatStateForMessages, fetchToolCallFromId, getLatestDownloadBuildUrl, LATEST_CHAT_STATE_VERSION, migrateChatState, } from "./state";
+export { extractAttachments } from "./utils";
 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,aAAa,EAAE,MAAM,cAAc,CAAC;AAC7C,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,EAChB,kBAAkB,GACnB,MAAM,SAAS,CAAC;AACjB,OAAO,EAAE,kBAAkB,EAAE,gBAAgB,EAAE,MAAM,SAAS,CAAC;AAC/D,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,13 +1,19 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.SUPPORTED_CHAT_MODELS = exports.createChatModel = exports.rejectPendingToolCalls = exports.processToolCallsAndUpdateChatState = exports.getChatAgentMode = exports.extractAttachments = exports.processAttachments = 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.chatAgentLoop = 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 agent_loop_1 = require("./agent-loop");
-Object.defineProperty(exports, "chatAgentLoop", { enumerable: true, get: function () { return agent_loop_1.chatAgentLoop; } });
+var base_1 = require("../base");
+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; } });
@@ -20,13 +26,8 @@ Object.defineProperty(exports, "fetchToolCallFromId", { enumerable: true, get: f
 Object.defineProperty(exports, "getLatestDownloadBuildUrl", { enumerable: true, get: function () { return state_1.getLatestDownloadBuildUrl; } });
 Object.defineProperty(exports, "LATEST_CHAT_STATE_VERSION", { enumerable: true, get: function () { return state_1.LATEST_CHAT_STATE_VERSION; } });
 Object.defineProperty(exports, "migrateChatState", { enumerable: true, get: function () { return state_1.migrateChatState; } });
-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; } });
-Object.defineProperty(exports, "getChatAgentMode", { enumerable: true, get: function () { return utils_1.getChatAgentMode; } });
-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,iBA2KA"}
+{"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"}