@empiricalrun/test-gen 0.74.2 → 0.75.0

package/CHANGELOG.md

@@ -1,5 +1,27 @@
 # @empiricalrun/test-gen
 
+## 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,36 @@
+import { IChatModel } from "@empiricalrun/llm/chat";
+import { FileInfo, PendingToolCall, ReporterFunction, StreamingMessageReporterFunc, SupportedChatModels, ToolDefinition, TraceClient } from "@empiricalrun/shared-types";
+type RunLoopParams<T> = {
+    chatModel: IChatModel<T>;
+    reporter: ReporterFunction;
+    streamingMessageReporter?: StreamingMessageReporterFunc;
+    trace?: TraceClient;
+    signal?: AbortSignal;
+    onPendingToolCall: (toolCalls: PendingToolCall[]) => Promise<void>;
+    onLLMResponse?: (response: T, selectedModel: SupportedChatModels) => Promise<void>;
+};
+export declare const getChatAgentMode: (source: string | undefined) => "triage" | "test_generation";
+type AgentParams = {
+    featureFlags: string[];
+    selectedModel: SupportedChatModels;
+    repoInfo: FileInfo;
+};
+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 {
+    featureFlags: string[];
+    tools: ToolDefinition[];
+    selectedModel: SupportedChatModels;
+    repoInfo: FileInfo;
+    constructor({ featureFlags, selectedModel, repoInfo }: AgentParams);
+    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,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"}

package/dist/agent/base/index.js

@@ -0,0 +1,74 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TriageAgent = exports.ChatAgent = exports.getChatAgentMode = void 0;
+exports.getAgentForMode = getAgentForMode;
+const tools_1 = require("../../tools");
+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 {
+    featureFlags = [];
+    tools = [];
+    selectedModel;
+    repoInfo;
+    constructor({ featureFlags, selectedModel, repoInfo }) {
+        this.featureFlags = featureFlags;
+        this.tools = (0, tools_1.availableTools)(selectedModel, false, []);
+        this.selectedModel = selectedModel;
+        this.repoInfo = repoInfo;
+    }
+    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,
+        });
+    }
+}
+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;
+    }
+    async runLoop(params) {
+        const mode = (0, exports.getChatAgentMode)("triage");
+        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,
+        });
+    }
+}
+exports.TriageAgent = TriageAgent;

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

@@ -1,18 +1,17 @@
 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";
+export declare function chatAgentLoop<T = CanonicalMessage>({ chatModel, selectedModel, reporter, streamingMessageReporter, trace, signal, featureFlags, tools, onPendingToolCall, onLLMResponse, systemPrompt, }: {
     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;
+    systemPrompt: 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,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,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"}

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, selectedModel, 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");
@@ -26,7 +19,7 @@ async function chatAgentLoop({ chatModel, selectedModel, reporter, streamingMess
             (0, utils_1.log)(`${(0, utils_1.getModelName)(selectedModel)} is working...`);
             const response = await chatModel.getLLMResponse({
                 systemPrompt,
-                tools: availableToolsData.map((t) => t.schema),
+                tools: tools.map((t) => t.schema),
                 selectedModel,
                 trace,
                 streamingMessageReporter,

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

@@ -1,9 +1,9 @@
 export { getFileInfoFromGitHub, viewFileUsingGitHub, } from "../../file-info/adapters/github";
 export { allToolsDefinitions } from "../../tools";
-export { chatAgentLoop } from "./agent-loop";
+export { ChatAgent, getAgentForMode, TriageAgent } from "../base";
 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 { 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";

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

package/dist/agent/chat/exports.js

@@ -1,13 +1,15 @@
 "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.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;
 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, "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; } });
 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,10 +22,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; } });

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

package/dist/agent/chat/index.js

@@ -11,7 +11,7 @@ const human_in_the_loop_1 = require("../../human-in-the-loop");
 const validation_1 = require("../../recorder/validation");
 const executor_1 = require("../../tools/executor");
 const git_1 = require("../../tools/executor/utils/git");
-const agent_loop_1 = require("./agent-loop");
+const base_1 = require("../base");
 const filesystem_cache_1 = require("./filesystem-cache");
 const state_1 = require("./state");
 const utils_1 = require("./utils");
@@ -122,6 +122,11 @@ async function runChatAgentForCLI({ useDiskForChatState, selectedModel, initialP
     const apiClient = new client_1.DashboardAPIClient({
         authType,
     });
+    const fileInfo = await (0, reader_1.getFileInfoFromFS)(process.cwd());
+    const agentParams = { featureFlags: [], repoInfo: fileInfo, selectedModel };
+    const agent = useTriage
+        ? new base_1.TriageAgent(agentParams)
+        : new base_1.ChatAgent(agentParams);
     while (!stopCriteria(userPrompt)) {
         if (chatModel.askUserForInput) {
             try {
@@ -144,10 +149,7 @@ async function runChatAgentForCLI({ useDiskForChatState, selectedModel, initialP
             }
             if (!stopCriteria(userPrompt)) {
                 const { text, attachments } = (0, utils_1.extractAttachments)(userPrompt);
-                const processed = attachments.length > 0
-                    ? await chatModel.processAttachments(attachments)
-                    : [];
-                chatModel.pushUserMessage(text, processed);
+                chatModel.pushUserMessage(text, attachments);
             }
         }
         else {
@@ -160,10 +162,8 @@ async function runChatAgentForCLI({ useDiskForChatState, selectedModel, initialP
                 featureFlags: [],
                 environmentOverrides: await fetchEnvironmentVariables(),
             });
-            const fileInfo = await (0, reader_1.getFileInfoFromFS)(process.cwd());
-            await (0, agent_loop_1.chatAgentLoop)({
+            await agent.runLoop({
                 chatModel,
-                selectedModel,
                 reporter: reporterFunc,
                 streamingMessageReporter: (() => {
                     if (!enableStreaming) {
@@ -190,13 +190,10 @@ async function runChatAgentForCLI({ useDiskForChatState, selectedModel, initialP
                     };
                 })(),
                 trace,
-                repoInfo: fileInfo,
                 onPendingToolCall: async (toolCalls) => {
                     const toolResults = await toolExecutor.execute(toolCalls);
                     chatModel.pushToolResultsMessage(toolCalls, toolResults);
                 },
-                featureFlags: [],
-                mode: (0, utils_1.getChatAgentMode)(useTriage ? "triage" : "test_generation"),
             });
         }
     }

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

@@ -1,13 +1,12 @@
 import { IChatModel } from "@empiricalrun/llm/chat";
-import { Attachment, CanonicalMessage, ChatState, ChatStateError, PendingToolCall, ProcessedAttachment, SupportedChatModels, ToolResult } from "@empiricalrun/shared-types";
+import { Attachment, CanonicalMessage, ChatState, ChatStateError, PendingToolCall, SupportedChatModels, ToolResult } from "@empiricalrun/shared-types";
 export declare const CHAT_STATE_VERSIONS_MIGRATIONS_MAP: Record<string, (state: any) => any>;
 export declare const LATEST_CHAT_STATE_VERSION = "0.1";
 export declare const CHAT_STATE_PATH: string;
 export declare function clearChatState(): void;
-export declare function processAttachments(attachments: Attachment[], selectedModel: SupportedChatModels): Promise<ProcessedAttachment[]>;
 export declare function createChatState({ userPrompt, attachments, existingState, selectedModel, error, }: {
     userPrompt: string | undefined;
-    attachments: ProcessedAttachment[];
+    attachments: Attachment[];
     existingState: ChatState | undefined;
     selectedModel: SupportedChatModels;
     error: ChatStateError | null;

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,eAAe,EACf,mBAAmB,EACnB,mBAAmB,EAEnB,UAAU,EACX,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,wBAAgB,cAAc,SAQ7B;AAED,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,eAAe,CAC7B,SAAS,EAAE,UAAU,CAAC,gBAAgB,CAAC,GACtC,MAAM,CA0BR;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,mBAAmB,CACjC,UAAU,EAAE,MAAM,EAClB,QAAQ,EAAE,gBAAgB,EAAE,GAC3B;IACD,gBAAgB,EAAE,IAAI,GAAG,SAAS,CAAC;IACnC,QAAQ,EAAE,eAAe,GAAG,SAAS,CAAC;IACtC,UAAU,EAAE,UAAU,GAAG,SAAS,CAAC;CACpC,CA0BA"}
+{"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,eAAe,EACf,mBAAmB,EAEnB,UAAU,EACX,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,wBAAgB,cAAc,SAQ7B;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,UAAU,EAAE,CAAC;IAC1B,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,eAAe,CAC7B,SAAS,EAAE,UAAU,CAAC,gBAAgB,CAAC,GACtC,MAAM,CA0BR;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,mBAAmB,CACjC,UAAU,EAAE,MAAM,EAClB,QAAQ,EAAE,gBAAgB,EAAE,GAC3B;IACD,gBAAgB,EAAE,IAAI,GAAG,SAAS,CAAC;IACnC,QAAQ,EAAE,eAAe,GAAG,SAAS,CAAC;IACtC,UAAU,EAAE,UAAU,GAAG,SAAS,CAAC;CACpC,CA0BA"}

package/dist/agent/chat/state.js

@@ -5,7 +5,6 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.CHAT_STATE_PATH = exports.LATEST_CHAT_STATE_VERSION = exports.CHAT_STATE_VERSIONS_MIGRATIONS_MAP = void 0;
 exports.clearChatState = clearChatState;
-exports.processAttachments = processAttachments;
 exports.createChatState = createChatState;
 exports.createChatStateForMessages = createChatStateForMessages;
 exports.chatStateFromModel = chatStateFromModel;
@@ -55,13 +54,6 @@ function clearChatState() {
         // Swallow errors to avoid blocking the CLI; deletion is best-effort
     }
 }
-async function processAttachments(attachments, selectedModel) {
-    const chatModel = (0, chat_1.createChatModel)([], selectedModel);
-    if (attachments.length === 0) {
-        return [];
-    }
-    return chatModel.processAttachments(attachments);
-}
 function createChatState({ userPrompt, attachments, existingState, selectedModel, error, }) {
     const messages = existingState?.messages || [];
     const chatModel = (0, chat_1.createChatModel)(messages, selectedModel);

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

@@ -14,5 +14,4 @@ export declare function extractAttachments(userPrompt: string): {
     text: string;
     attachments: Attachment[];
 };
-export declare const getChatAgentMode: (source: string | undefined) => "triage" | "test_generation";
 //# sourceMappingURL=utils.d.ts.map

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

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

package/dist/agent/chat/utils.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.getChatAgentMode = exports.log = void 0;
+exports.log = void 0;
 exports.getModelName = getModelName;
 exports.handleAgentError = handleAgentError;
 exports.extractAttachments = extractAttachments;
@@ -64,16 +64,3 @@ function extractAttachments(userPrompt) {
         attachments,
     };
 }
-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;

package/dist/file-info/adapters/github/index.d.ts

@@ -1,11 +1,11 @@
 import { IDashboardAPIClient, ToolResult } from "@empiricalrun/shared-types";
 import { type StrReplaceInputParams } from "../../../tools/file-operations/shared/helpers";
 export { getFileInfoFromGitHub } from "./reader";
-export declare function viewFileUsingGitHub({ input, filePath, repoName, apiClient, branchName, }: {
+export declare function viewFileUsingGitHub({ input, repoName, apiClient, branchName, baseBranch, }: {
     input: StrReplaceInputParams;
-    filePath: string;
     repoName: string;
     apiClient: IDashboardAPIClient;
     branchName: string;
+    baseBranch: string;
 }): Promise<ToolResult>;
 //# sourceMappingURL=index.d.ts.map

package/dist/file-info/adapters/github/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/file-info/adapters/github/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,mBAAmB,EAAE,UAAU,EAAE,MAAM,4BAA4B,CAAC;AAE7E,OAAO,EAEL,KAAK,qBAAqB,EAC3B,MAAM,+CAA+C,CAAC;AAEvD,OAAO,EAAE,qBAAqB,EAAE,MAAM,UAAU,CAAC;AAIjD,wBAAsB,mBAAmB,CAAC,EACxC,KAAK,EACL,QAAQ,EACR,QAAQ,EACR,SAAS,EACT,UAAU,GACX,EAAE;IACD,KAAK,EAAE,qBAAqB,CAAC;IAC7B,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,EAAE,mBAAmB,CAAC;IAC/B,UAAU,EAAE,MAAM,CAAC;CACpB,GAAG,OAAO,CAAC,UAAU,CAAC,CAsBtB"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/file-info/adapters/github/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,mBAAmB,EAAE,UAAU,EAAE,MAAM,4BAA4B,CAAC;AAE7E,OAAO,EAEL,KAAK,qBAAqB,EAC3B,MAAM,+CAA+C,CAAC;AAEvD,OAAO,EAAE,qBAAqB,EAAE,MAAM,UAAU,CAAC;AAIjD,wBAAsB,mBAAmB,CAAC,EACxC,KAAK,EACL,QAAQ,EACR,SAAS,EACT,UAAU,EACV,UAAU,GACX,EAAE;IACD,KAAK,EAAE,qBAAqB,CAAC;IAC7B,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,EAAE,mBAAmB,CAAC;IAC/B,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;CACpB,GAAG,OAAO,CAAC,UAAU,CAAC,CA2BtB"}

package/dist/file-info/adapters/github/index.js

@@ -7,9 +7,10 @@ const reader_1 = require("./reader");
 var reader_2 = require("./reader");
 Object.defineProperty(exports, "getFileInfoFromGitHub", { enumerable: true, get: function () { return reader_2.getFileInfoFromGitHub; } });
 const REPO_OWNER = "empirical-run";
-async function viewFileUsingGitHub({ input, filePath, repoName, apiClient, branchName, }) {
+async function viewFileUsingGitHub({ input, repoName, apiClient, branchName, baseBranch, }) {
+    const filePath = input.path;
     const githubReader = new reader_1.GitHubFileReader(repoName, apiClient, REPO_OWNER);
-    const fileData = await githubReader.readFile(filePath, branchName);
+    const fileData = await githubReader.readFile(filePath, branchName, baseBranch);
     if (!fileData) {
         return {
             result: `Error: File or directory ${filePath} is not found.`,

package/dist/file-info/adapters/github/reader.d.ts

@@ -7,10 +7,10 @@ export declare class GitHubFileReader {
     private fetchFileFromGitHub;
     private fetchTreeFromGitHub;
     private resolveBranchName;
-    readFile(filePath: string, branchName: string): Promise<FileReadResult | null>;
+    readFile(filePath: string, branchName: string, baseBranch: string): Promise<FileReadResult | null>;
     private getFileContent;
     private getContentsForDirectory;
-    getFileInfo(branchName: string): Promise<FileInfo>;
+    getFileInfo(branchName: string, baseBranch: string): Promise<FileInfo>;
 }
-export declare function getFileInfoFromGitHub(repoName: string, apiClient: IDashboardAPIClient, branchName: string): Promise<FileInfo>;
+export declare function getFileInfoFromGitHub(repoName: string, apiClient: IDashboardAPIClient, branchName: string, baseBranch: string): Promise<FileInfo>;
 //# sourceMappingURL=reader.d.ts.map

package/dist/file-info/adapters/github/reader.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"reader.d.ts","sourceRoot":"","sources":["../../../../src/file-info/adapters/github/reader.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,QAAQ,EACR,cAAc,EACd,mBAAmB,EACpB,MAAM,4BAA4B,CAAC;AA4CpC,qBAAa,gBAAgB;IAEzB,OAAO,CAAC,QAAQ;IAChB,OAAO,CAAC,SAAS;IACjB,OAAO,CAAC,SAAS;gBAFT,QAAQ,EAAE,MAAM,EAChB,SAAS,EAAE,mBAAmB,EAC9B,SAAS,EAAE,MAAM;YAGb,mBAAmB;YAUnB,mBAAmB;YASnB,iBAAiB;IAkBzB,QAAQ,CACZ,QAAQ,EAAE,MAAM,EAChB,UAAU,EAAE,MAAM,GACjB,OAAO,CAAC,cAAc,GAAG,IAAI,CAAC;YAmCnB,cAAc;YAiBd,uBAAuB;IAuB/B,WAAW,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,QAAQ,CAAC;CA+EzD;AAED,wBAAsB,qBAAqB,CACzC,QAAQ,EAAE,MAAM,EAChB,SAAS,EAAE,mBAAmB,EAC9B,UAAU,EAAE,MAAM,GACjB,OAAO,CAAC,QAAQ,CAAC,CAGnB"}
+{"version":3,"file":"reader.d.ts","sourceRoot":"","sources":["../../../../src/file-info/adapters/github/reader.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,QAAQ,EACR,cAAc,EACd,mBAAmB,EACpB,MAAM,4BAA4B,CAAC;AA4CpC,qBAAa,gBAAgB;IAEzB,OAAO,CAAC,QAAQ;IAChB,OAAO,CAAC,SAAS;IACjB,OAAO,CAAC,SAAS;gBAFT,QAAQ,EAAE,MAAM,EAChB,SAAS,EAAE,mBAAmB,EAC9B,SAAS,EAAE,MAAM;YAGb,mBAAmB;YAUnB,mBAAmB;YASnB,iBAAiB;IAqBzB,QAAQ,CACZ,QAAQ,EAAE,MAAM,EAChB,UAAU,EAAE,MAAM,EAClB,UAAU,EAAE,MAAM,GACjB,OAAO,CAAC,cAAc,GAAG,IAAI,CAAC;YAmCnB,cAAc;YAkBd,uBAAuB;IAwB/B,WAAW,CAAC,UAAU,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,QAAQ,CAAC;CAoF7E;AAED,wBAAsB,qBAAqB,CACzC,QAAQ,EAAE,MAAM,EAChB,SAAS,EAAE,mBAAmB,EAC9B,UAAU,EAAE,MAAM,EAClB,UAAU,EAAE,MAAM,GACjB,OAAO,CAAC,QAAQ,CAAC,CAGnB"}

package/dist/file-info/adapters/github/reader.js

@@ -29,9 +29,9 @@ class GitHubFileReader {
             url: `/repos/${this.repoOwner}/${this.repoName}/git/trees/${branch}?recursive=1`,
         }));
     }
-    async resolveBranchName(branchName) {
-        if (!branchName || branchName === "main") {
-            return "main";
+    async resolveBranchName(branchName, baseBranch) {
+        if (!branchName || branchName === baseBranch) {
+            return baseBranch;
         }
         try {
             // Try to fetch the branch to see if it exists
@@ -42,12 +42,12 @@ class GitHubFileReader {
             return branchName;
         }
         catch {
-            // Branch doesn't exist, fallback to main
-            return "main";
+            // Branch doesn't exist, fallback to base branch
+            return baseBranch;
         }
     }
-    async readFile(filePath, branchName) {
-        const branch = await this.resolveBranchName(branchName);
+    async readFile(filePath, branchName, baseBranch) {
+        const branch = await this.resolveBranchName(branchName, baseBranch);
         const normalizedPath = filePath === "" ? "." : filePath;
         try {
             const response = await this.fetchFileFromGitHub(normalizedPath, branch);
@@ -76,8 +76,8 @@ class GitHubFileReader {
         }
         return null;
     }
-    async getFileContent(path, branchName) {
-        const branch = await this.resolveBranchName(branchName);
+    async getFileContent(path, branchName, baseBranch) {
+        const branch = await this.resolveBranchName(branchName, baseBranch);
         const response = await this.fetchFileFromGitHub(path, branch);
         const isDirectory = Array.isArray(response);
         if (isDirectory) {
@@ -88,8 +88,8 @@ class GitHubFileReader {
         }
         throw new Error(`Unable to fetch file content for ${path}`);
     }
-    async getContentsForDirectory(path, branchName) {
-        const branch = await this.resolveBranchName(branchName);
+    async getContentsForDirectory(path, branchName, baseBranch) {
+        const branch = await this.resolveBranchName(branchName, baseBranch);
         const response = await this.fetchTreeFromGitHub(branch);
         if (response.tree) {
             // Convert to expected RepoTree format
@@ -106,8 +106,9 @@ class GitHubFileReader {
         }
         throw new Error(`Unable to fetch directory contents for ${path}`);
     }
-    async getFileInfo(branchName) {
-        const files = await this.getContentsForDirectory("", branchName);
+    async getFileInfo(branchName, baseBranch) {
+        const rootPath = "";
+        const files = await this.getContentsForDirectory(rootPath, branchName, baseBranch);
         const root = [];
         const nodeMap = {};
         if (!files) {
@@ -131,7 +132,7 @@ class GitHubFileReader {
                     type: "file",
                     path: currentPath,
                     name: fileName,
-                    getContent: async () => this.getFileContent(currentPath, branchName),
+                    getContent: async () => this.getFileContent(currentPath, branchName, baseBranch),
                 }
                 : {
                     type: "directory",
@@ -179,7 +180,7 @@ class GitHubFileReader {
     }
 }
 exports.GitHubFileReader = GitHubFileReader;
-async function getFileInfoFromGitHub(repoName, apiClient, branchName) {
+async function getFileInfoFromGitHub(repoName, apiClient, branchName, baseBranch) {
     const fileReader = new GitHubFileReader(repoName, apiClient, REPO_OWNER);
-    return fileReader.getFileInfo(branchName);
+    return fileReader.getFileInfo(branchName, baseBranch);
 }

package/dist/tools/commit-and-create-pr/index.js

@@ -56,7 +56,7 @@ exports.createPullRequestTool = {
                     repo,
                     title: pullRequestTitle,
                     head: branchName,
-                    base: "main",
+                    base: chatSession?.baseBranchName || "main",
                     body: (0, pr_description_1.addMetadataToPRDescription)(pullRequestDescription, chatSession),
                     apiClient,
                 });

package/dist/tools/definitions/fetch-video-analysis.d.ts

@@ -2,10 +2,38 @@ import { ToolDefinition } from "@empiricalrun/shared-types";
 import z from "zod";
 export declare const videoAnalysisSchema: z.ZodObject<{
     videoUrl: z.ZodString;
+    params: z.ZodOptional<z.ZodObject<{
+        fps: z.ZodOptional<z.ZodDefault<z.ZodNumber>>;
+        threshold: z.ZodOptional<z.ZodDefault<z.ZodNumber>>;
+        model: z.ZodOptional<z.ZodDefault<z.ZodString>>;
+        featureFlag: z.ZodOptional<z.ZodDefault<z.ZodEnum<["none", "send-all-frames", "separate-threshold", "ai-based-frames"]>>>;
+    }, "strip", z.ZodTypeAny, {
+        model?: string | undefined;
+        fps?: number | undefined;
+        threshold?: number | undefined;
+        featureFlag?: "none" | "send-all-frames" | "separate-threshold" | "ai-based-frames" | undefined;
+    }, {
+        model?: string | undefined;
+        fps?: number | undefined;
+        threshold?: number | undefined;
+        featureFlag?: "none" | "send-all-frames" | "separate-threshold" | "ai-based-frames" | undefined;
+    }>>;
 }, "strip", z.ZodTypeAny, {
     videoUrl: string;
+    params?: {
+        model?: string | undefined;
+        fps?: number | undefined;
+        threshold?: number | undefined;
+        featureFlag?: "none" | "send-all-frames" | "separate-threshold" | "ai-based-frames" | undefined;
+    } | undefined;
 }, {
     videoUrl: string;
+    params?: {
+        model?: string | undefined;
+        fps?: number | undefined;
+        threshold?: number | undefined;
+        featureFlag?: "none" | "send-all-frames" | "separate-threshold" | "ai-based-frames" | undefined;
+    } | undefined;
 }>;
 export declare const fetchVideoAnalysis: ToolDefinition<z.infer<typeof videoAnalysisSchema>>;
 //# sourceMappingURL=fetch-video-analysis.d.ts.map

package/dist/tools/definitions/fetch-video-analysis.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"fetch-video-analysis.d.ts","sourceRoot":"","sources":["../../../src/tools/definitions/fetch-video-analysis.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;AAC5D,OAAO,CAAC,MAAM,KAAK,CAAC;AAEpB,eAAO,MAAM,mBAAmB;;;;;;EAK9B,CAAC;AAEH,eAAO,MAAM,kBAAkB,EAAE,cAAc,CAC7C,CAAC,CAAC,KAAK,CAAC,OAAO,mBAAmB,CAAC,CAapC,CAAC"}
+{"version":3,"file":"fetch-video-analysis.d.ts","sourceRoot":"","sources":["../../../src/tools/definitions/fetch-video-analysis.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;AAC5D,OAAO,CAAC,MAAM,KAAK,CAAC;AAEpB,eAAO,MAAM,mBAAmB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EA2C9B,CAAC;AAEH,eAAO,MAAM,kBAAkB,EAAE,cAAc,CAC7C,CAAC,CAAC,KAAK,CAAC,OAAO,mBAAmB,CAAC,CAcpC,CAAC"}