@empiricalrun/test-gen 0.75.0 → 0.77.0

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

@@ -18,8 +18,8 @@ async function getAllMarkdownFiles(directory) {
         .filter((file) => file.type === "file" && file.path.endsWith(".md"))
         .map(async (file) => {
         return {
-            // TODO: Give full path
             name: path_1.default.basename(file.path),
+            path: file.path,
             content: file.type === "file" ? await file.getContent() : "",
         };
     }));
@@ -35,13 +35,14 @@ async function knowledgeContext(repoInfo) {
     const mdFiles = await getAllMarkdownFiles(knowledgeDir);
     const knowledge = mdFiles.map((file) => {
         return `
-    <knowledge_file>
-    <file_name>${file.name}</file_name>
-    <file_content>
-    ${file.content}
-    </file_content>
-    </knowledge_file>
-    `;
+<knowledge_file>
+<file_name>${file.name}</file_name>
+<file_path>${file.path}</file_path>
+<file_content>
+${file.content}
+</file_content>
+</knowledge_file>
+`;
     });
     return knowledge.join("\n");
 }
@@ -59,25 +60,13 @@ While specifying paths to files, use relative paths from the current working dir
 - Correct path: "tests/lesson.spec.ts"
 - Incorrect path: "/repo/tests/lesson.spec.ts" or "${repoInfo.name}/tests/lesson.spec.ts"
 `;
-    // try {
-    //   const playwrightConfig = await readPlaywrightConfig(process.cwd());
-    //   const validProjectNames = await getValidProjectNames(playwrightConfig);
-    //   if (validProjectNames.length > 0) {
-    //     REPO_CONTEXT_PROMPT += `
-    //   ## Playwright configuration
-    //   This repo is configured with these Playwright projects (in the playwright.config.ts file):
-    //   ${validProjectNames.map((name) => `- ${name}`).join("\n")}
-    //   `;
-    //   }
-    // } catch (error) {
-    //   console.warn("Failed to read playwright config", error);
-    // }
     const knowledge = await knowledgeContext(repoInfo);
     if (knowledge.length > 0) {
         REPO_CONTEXT_PROMPT += `
 ## Repo-specific knowledge
 
-You can use the following knowledge with your tasks.
+This repo knowledge files that will help you understand the repo and application better.
+
 <knowledge>
 ${knowledge}
 </knowledge>

package/dist/agent/chat/prompt/test-case-def.d.ts

@@ -0,0 +1,2 @@
+export declare const testCasesDefinitionPrompt = "\n# Specifying test cases for tools/functions\n\nMany tools and functions that you use require specifying a test case with the following properties: \n- Test case name: Name of the test (specified to the \"test(...)\" method)\n- Suites: Refers to describe blocks where this test is located, specified as an array of strings\n- File path: Path to the file - must be relative to the repo directory\n\n<example_1>\n// Contents of tests/example.spec.ts\nimport { test, expect } from \"./fixtures\";\n\ntest(\"has title\", async ({ page }) => {\n  await page.goto(\"https://playwright.dev/\");\n  await expect(page).toHaveTitle(/Playwright/);\n});\n\n// Specifying this test case\n- Test case name: \"has title\"\n- Suites: [] (since this test has no describe blocks)\n- File path: \"tests/example.spec.ts\"\n</example_1>\n\n<example_2>\n// Contents of tests/sidebar/nav.spec.ts\nimport { test, expect } from \"./fixtures\";\n\ntest.describe(\"Sidebar Navigation\", () => {\n  test(\"collapse sidebar and expand it by clicking settings gear icon\", async ({ page }) => {\n    await page.goto(\"/\");\n    await page.getByRole('button', { name: 'Collapse sidebar' }).click();\n    await page.getByRole('navigation').getByRole('button').click();\n  });\n});\n\n// Specifying this test case\n- Test case name: \"collapse sidebar and expand it by clicking settings gear icon\"\n- Suites: [\"Sidebar Navigation\"]\n- File path: \"tests/sidebar/nav.spec.ts\"\n</example_2>\n";
+//# sourceMappingURL=test-case-def.d.ts.map

package/dist/agent/chat/prompt/test-case-def.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"test-case-def.d.ts","sourceRoot":"","sources":["../../../../src/agent/chat/prompt/test-case-def.ts"],"names":[],"mappings":"AAAA,eAAO,MAAM,yBAAyB,47CAwCrC,CAAC"}

package/dist/agent/chat/prompt/test-case-def.js

@@ -0,0 +1,44 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.testCasesDefinitionPrompt = void 0;
+exports.testCasesDefinitionPrompt = `
+# Specifying test cases for tools/functions
+
+Many tools and functions that you use require specifying a test case with the following properties: 
+- Test case name: Name of the test (specified to the "test(...)" method)
+- Suites: Refers to describe blocks where this test is located, specified as an array of strings
+- File path: Path to the file - must be relative to the repo directory
+
+<example_1>
+// Contents of tests/example.spec.ts
+import { test, expect } from "./fixtures";
+
+test("has title", async ({ page }) => {
+  await page.goto("https://playwright.dev/");
+  await expect(page).toHaveTitle(/Playwright/);
+});
+
+// Specifying this test case
+- Test case name: "has title"
+- Suites: [] (since this test has no describe blocks)
+- File path: "tests/example.spec.ts"
+</example_1>
+
+<example_2>
+// Contents of tests/sidebar/nav.spec.ts
+import { test, expect } from "./fixtures";
+
+test.describe("Sidebar Navigation", () => {
+  test("collapse sidebar and expand it by clicking settings gear icon", async ({ page }) => {
+    await page.goto("/");
+    await page.getByRole('button', { name: 'Collapse sidebar' }).click();
+    await page.getByRole('navigation').getByRole('button').click();
+  });
+});
+
+// Specifying this test case
+- Test case name: "collapse sidebar and expand it by clicking settings gear icon"
+- Suites: ["Sidebar Navigation"]
+- File path: "tests/sidebar/nav.spec.ts"
+</example_2>
+`;

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

@@ -1,9 +1,8 @@
-import { IChatModel } from "@empiricalrun/llm/chat";
+import { IChatCanonicalModel, IChatModel } from "@empiricalrun/llm/chat";
 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 createChatState({ userPrompt, attachments, existingState, selectedModel, error, }: {
     userPrompt: string | undefined;
     attachments: Attachment[];
@@ -17,19 +16,20 @@ export declare function createChatStateForMessages({ messages, selectedModel, as
     askUserForInput: boolean;
     error: ChatStateError | null;
 }): ChatState;
-export declare function chatStateFromModel<T>({ chatModel, selectedModel, error, }: {
-    chatModel: IChatModel<T>;
-    selectedModel: SupportedChatModels;
+export declare function chatStateFromModel<T>({ chatModel, error, }: {
+    chatModel: IChatModel<T> | IChatCanonicalModel;
     error: ChatStateError | null;
 }): ChatState;
-export declare function getUsageSummary(chatModel: IChatModel<CanonicalMessage>): string;
-export declare function loadChatState(): ChatState | undefined;
+export declare function getUsageSummary(chatState: ChatState): string;
+export declare function loadChatState({ resetChat, }: {
+    resetChat: boolean;
+}): ChatState | undefined;
 /**
  * Migrates a chat state object from an old version to the latest version.
  * Add migration logic for each version as needed.
  */
 export declare function migrateChatState(oldState: any): ChatState;
-export declare function saveToDisk(messages: Array<CanonicalMessage>, selectedModel: SupportedChatModels, askUserForInput: boolean, error: ChatStateError | null): void;
+export declare function saveToDisk(chatState: ChatState): void;
 export declare function getLatestDownloadBuildUrl(messages: CanonicalMessage[]): string | null;
 export declare function fetchToolCallFromId(toolCallId: String, messages: CanonicalMessage[]): {
     toolCallQueuedAt: Date | undefined;

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,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"}
+{"version":3,"file":"state.d.ts","sourceRoot":"","sources":["../../../src/agent/chat/state.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,mBAAmB,EACnB,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;AAMpC,eAAO,MAAM,kCAAkC,EAAE,MAAM,CACrD,MAAM,EACN,CAAC,KAAK,EAAE,GAAG,KAAK,GAAG,CAGpB,CAAC;AAEF,eAAO,MAAM,yBAAyB,QAAQ,CAAC;AAE/C,eAAO,MAAM,eAAe,QAI3B,CAAC;AAEF,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,KAAK,GACN,EAAE;IACD,SAAS,EAAE,UAAU,CAAC,CAAC,CAAC,GAAG,mBAAmB,CAAC;IAC/C,KAAK,EAAE,cAAc,GAAG,IAAI,CAAC;CAC9B,aAOA;AAED,wBAAgB,eAAe,CAAC,SAAS,EAAE,SAAS,GAAG,MAAM,CA0B5D;AAED,wBAAgB,aAAa,CAAC,EAC5B,SAAS,GACV,EAAE;IACD,SAAS,EAAE,OAAO,CAAC;CACpB,GAAG,SAAS,GAAG,SAAS,CAsBxB;AAED;;;GAGG;AACH,wBAAgB,gBAAgB,CAAC,QAAQ,EAAE,GAAG,GAAG,SAAS,CAqBzD;AAED,wBAAgB,UAAU,CAAC,SAAS,EAAE,SAAS,QAQ9C;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

@@ -4,7 +4,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.createChatState = createChatState;
 exports.createChatStateForMessages = createChatStateForMessages;
 exports.chatStateFromModel = chatStateFromModel;
@@ -17,43 +16,11 @@ exports.fetchToolCallFromId = fetchToolCallFromId;
 const chat_1 = require("@empiricalrun/llm/chat");
 const fs_1 = __importDefault(require("fs"));
 const path_1 = __importDefault(require("path"));
-// Migration wrapper for v20250327.1 -> v0.1 chat state versions
-// v20250327.1 was model-specific, but v0.1 is canonical
-function migrateToV01(oldState) {
-    if (oldState.model &&
-        typeof oldState.model === "string" &&
-        Array.isArray(oldState.messages)) {
-        const provider = (0, chat_1.getProviderForModel)(oldState.model);
-        if (provider === "google") {
-            return {
-                ...oldState,
-                version: "0.1",
-                messages: oldState.messages.map(chat_1.geminiToCanonical),
-            };
-        }
-        else {
-            throw new Error(`Unsupported state for migration with model: ${oldState.model} and version: ${oldState.version}`);
-        }
-    }
-    // If not Gemini or not matching, return the old state
-    return oldState;
-}
 exports.CHAT_STATE_VERSIONS_MIGRATIONS_MAP = {
-    "20250327.1": migrateToV01,
     "0.1": (state) => state,
 };
 exports.LATEST_CHAT_STATE_VERSION = "0.1";
 exports.CHAT_STATE_PATH = path_1.default.join(process.cwd(), ".empiricalrun", "last-chat.json");
-function clearChatState() {
-    try {
-        if (fs_1.default.existsSync(exports.CHAT_STATE_PATH)) {
-            fs_1.default.unlinkSync(exports.CHAT_STATE_PATH);
-        }
-    }
-    catch {
-        // Swallow errors to avoid blocking the CLI; deletion is best-effort
-    }
-}
 function createChatState({ userPrompt, attachments, existingState, selectedModel, error, }) {
     const messages = existingState?.messages || [];
     const chatModel = (0, chat_1.createChatModel)(messages, selectedModel);
@@ -77,16 +44,16 @@ function createChatStateForMessages({ messages, selectedModel, askUserForInput,
         error: error,
     };
 }
-function chatStateFromModel({ chatModel, selectedModel, error, }) {
+function chatStateFromModel({ chatModel, error, }) {
     return createChatStateForMessages({
         messages: chatModel.messages,
-        selectedModel,
+        selectedModel: chatModel.selectedModel,
         askUserForInput: chatModel.askUserForInput,
         error,
     });
 }
-function getUsageSummary(chatModel) {
-    const messages = chatModel.messages;
+function getUsageSummary(chatState) {
+    const messages = chatState.messages;
     const messageUsages = messages.map((msg) => msg.usage).filter(Boolean);
     const inputTokens = messageUsages.reduce((sum, usage) => sum + (usage?.tokens?.input || 0), 0);
     const outputTokens = messageUsages.reduce((sum, usage) => sum + (usage?.tokens?.output || 0), 0);
@@ -103,7 +70,17 @@ function getUsageSummary(chatModel) {
 Input: ${inputTokens.toLocaleString()} tokens / Output: ${outputTokens.toLocaleString()} tokens`;
     return msgsSummary;
 }
-function loadChatState() {
+function loadChatState({ resetChat, }) {
+    if (resetChat) {
+        try {
+            if (fs_1.default.existsSync(exports.CHAT_STATE_PATH)) {
+                fs_1.default.unlinkSync(exports.CHAT_STATE_PATH);
+            }
+        }
+        catch {
+            // Swallow errors to avoid blocking the CLI; deletion is best-effort
+        }
+    }
     if (!fs_1.default.existsSync(exports.CHAT_STATE_PATH)) {
         return undefined;
     }
@@ -141,21 +118,14 @@ function migrateChatState(oldState) {
         ...migrated,
     };
 }
-function saveToDisk(messages, selectedModel, askUserForInput, error) {
+function saveToDisk(chatState) {
     const statePath = exports.CHAT_STATE_PATH;
     // Ensure directory exists before trying to read/write
     const dirname = path_1.default.dirname(statePath);
     if (!fs_1.default.existsSync(dirname)) {
         fs_1.default.mkdirSync(dirname, { recursive: true });
     }
-    // Use the helper to build the new state
-    const newState = createChatStateForMessages({
-        messages,
-        selectedModel,
-        askUserForInput,
-        error,
-    });
-    fs_1.default.writeFileSync(statePath, JSON.stringify(newState, null, 2));
+    fs_1.default.writeFileSync(statePath, JSON.stringify(chatState, null, 2));
 }
 function findLatestToolCall(messages, toolName) {
     return messages

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

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

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"}
+{"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,EAGL,mBAAmB,EACnB,UAAU,EACX,MAAM,wBAAwB,CAAC;AAChC,OAAO,EACL,UAAU,EAEV,gBAAgB,EACjB,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,QAAQ,EACR,KAAK,GACN,EAAE;IACD,KAAK,EAAE,OAAO,CAAC;IACf,SAAS,EAAE,UAAU,CAAC,GAAG,CAAC,GAAG,mBAAmB,CAAC;IACjD,QAAQ,EAAE,gBAAgB,CAAC;IAC3B,KAAK,CAAC,EAAE,WAAW,CAAC;CACrB,iBAaA;AASD,wBAAgB,kBAAkB,CAAC,UAAU,EAAE,MAAM,GAAG;IACtD,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,UAAU,EAAE,CAAC;CAC3B,CA4BA"}

package/dist/agent/chat/utils.js

@@ -29,31 +29,37 @@ function getModelName(model) {
         return model;
     return model;
 }
-async function handleAgentError({ error, chatModel, selectedModel, reporter, trace, }) {
+async function handleAgentError({ error, chatModel, reporter, trace, }) {
     const errorObject = {
         message: error.message,
         stack: error.stack || "Stack trace not available",
         timestamp: new Date().toISOString(),
+        type: error instanceof chat_1.CanonicalError ? error.type : null,
     };
-    await reporter((0, state_1.chatStateFromModel)({
-        chatModel,
-        selectedModel,
-        error: errorObject,
-    }), chatModel.getHumanReadableLatestMessage());
+    await reporter((0, state_1.chatStateFromModel)({ chatModel, error: errorObject }), chatModel.getHumanReadableLatestMessage());
     trace?.update({ output: { error: errorObject } });
     logError(error, trace);
 }
+const VALID_ATTACHMENTS = [".webm", ".png"];
+const MIME_MAP = {
+    ".webm": "video/webm",
+    ".png": "image/png",
+};
 function extractAttachments(userPrompt) {
     const attachments = [];
     const urlPattern = /\[\[(https?:\/\/[^\]]+)\]\]/g;
     let match;
     while ((match = urlPattern.exec(userPrompt)) !== null) {
         const name = match[1].split("/").pop();
-        if (!name || !name.endsWith(".webm")) {
-            throw new Error(`Invalid attachment name: ${name}. Only .webm files are supported.`);
+        const ext = name ? name.substring(name.lastIndexOf(".")) : null;
+        if (!ext) {
+            throw new Error(`Attachment URL does not have a valid file extension: ${match[1]}`);
+        }
+        if (!name || !VALID_ATTACHMENTS.some((validExt) => ext === validExt)) {
+            throw new Error(`Invalid attachment name: ${name}. Only ${VALID_ATTACHMENTS.join(", ")} files are supported.`);
         }
         attachments.push({
-            contentType: "video/webm",
+            contentType: MIME_MAP[ext],
             name,
             url: match[1],
         });

package/dist/agent/cli.d.ts

@@ -0,0 +1,11 @@
+import { AgentModeEnum, SupportedChatModels } from "@empiricalrun/shared-types";
+export declare function fetchEnvironmentVariables(): Promise<Record<string, string>>;
+export declare function runChatAgentForCLI({ useDiskForChatState, selectedModel, initialPromptContent, agentMode, resetChat, useFSCache, }: {
+    selectedModel: SupportedChatModels;
+    useDiskForChatState: boolean;
+    initialPromptContent: string | undefined;
+    agentMode: AgentModeEnum;
+    useFSCache: boolean;
+    resetChat: boolean;
+}): Promise<void>;
+//# sourceMappingURL=cli.d.ts.map

package/dist/agent/cli.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"cli.d.ts","sourceRoot":"","sources":["../../src/agent/cli.ts"],"names":[],"mappings":"AACA,OAAO,EACL,aAAa,EAKb,mBAAmB,EACpB,MAAM,4BAA4B,CAAC;AA+BpC,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,aAAa,CAAC;IACzB,UAAU,EAAE,OAAO,CAAC;IACpB,SAAS,EAAE,OAAO,CAAC;CACpB,iBA2KA"}

package/dist/agent/cli.js

@@ -0,0 +1,213 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.fetchEnvironmentVariables = fetchEnvironmentVariables;
+exports.runChatAgentForCLI = runChatAgentForCLI;
+const llm_1 = require("@empiricalrun/llm");
+const picocolors_1 = require("picocolors");
+const client_1 = require("../dashboard/client");
+const reader_1 = require("../file-info/adapters/file-system/reader");
+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 state_1 = require("./chat/state");
+const utils_1 = require("./chat/utils");
+const index_1 = require("./index");
+function stopCriteria(userPrompt) {
+    return userPrompt?.toLowerCase() === "stop";
+}
+function concludeAgent(agent, useDiskForChatState) {
+    const chatState = agent.chatState;
+    console.log(`\n${(0, picocolors_1.gray)("Usage summary -> " + (0, state_1.getUsageSummary)(chatState))}`);
+    if (useDiskForChatState) {
+        (0, state_1.saveToDisk)(chatState);
+    }
+}
+async function fetchEnvironmentVariables() {
+    const projectApiKey = process.env.EMPIRICALRUN_API_KEY;
+    let apiClient;
+    let queryParams = {};
+    if (projectApiKey) {
+        apiClient = new client_1.DashboardAPIClient({
+            authType: "project-api-key",
+            projectApiKey: process.env.EMPIRICALRUN_API_KEY,
+        });
+    }
+    else {
+        apiClient = new client_1.DashboardAPIClient({
+            authType: "user-access-token",
+        });
+        const repoName = await (0, validation_1.validatePackageJson)(process.cwd());
+        queryParams = { project_repo_name: repoName };
+    }
+    const data = await apiClient.request("/api/environment-variables", { method: "GET", params: queryParams });
+    if (!data.data || !data.data.environment_variables) {
+        console.error("Failed to fetch environment variables:", data);
+        throw new Error("Failed to fetch environment variables");
+    }
+    const envVars = data.data.environment_variables.reduce((acc, envVar) => {
+        acc[envVar.name] = envVar.value;
+        return acc;
+    }, {});
+    return envVars;
+}
+async function runChatAgentForCLI({ useDiskForChatState, selectedModel, initialPromptContent, agentMode, resetChat, useFSCache, }) {
+    const enableStreaming = !useFSCache;
+    // TODO: Implement cache support in BaseAgent
+    // const cache = useFSCache ? new FilesystemLLMCache() : undefined;
+    // TODO: Store branch name in chat state so that we don't recreate it every time
+    const randomId = crypto.randomUUID().substring(0, 8);
+    const branchName = `branch-${randomId}`;
+    await (0, git_1.checkoutBranch)(branchName, process.cwd());
+    let chatState;
+    if (useDiskForChatState) {
+        chatState = (0, state_1.loadChatState)({ resetChat });
+    }
+    if (chatState && chatState.error) {
+        // Reset error state as we are attempting a retry
+        chatState.error = null;
+    }
+    if (initialPromptContent) {
+        if (!chatState) {
+            const { text, attachments } = (0, utils_1.extractAttachments)(initialPromptContent);
+            chatState = (0, state_1.createChatState)({
+                userPrompt: text,
+                attachments: attachments,
+                existingState: undefined,
+                selectedModel,
+                error: null,
+            });
+        }
+        else {
+            console.warn(`Ignoring initial prompt because we have existing chat state.`);
+        }
+    }
+    if (!chatState) {
+        chatState = (0, state_1.createChatState)({
+            userPrompt: undefined,
+            attachments: [],
+            selectedModel,
+            existingState: undefined,
+            error: null,
+        });
+    }
+    const handleSigInt = () => {
+        concludeAgent(agent, useDiskForChatState);
+        process.exit(0);
+    };
+    process.once("SIGINT", handleSigInt);
+    process.once("SIGTERM", handleSigInt);
+    let userPrompt;
+    let reporterFunc = async (chatState, latest) => {
+        if (useDiskForChatState) {
+            (0, state_1.saveToDisk)(chatState);
+        }
+        if (latest) {
+            if (!enableStreaming) {
+                console.log(`${(0, picocolors_1.blue)(latest.role)}: ${latest.textMessage}`);
+            }
+            else {
+                process.stdout.write(`\n`);
+            }
+        }
+    };
+    const trace = (0, llm_1.createLangfuseTrace)({
+        name: "chat_agent",
+        input: initialPromptContent || "",
+        tags: [selectedModel, "chat_agent"],
+    });
+    if (trace) {
+        const traceUrl = trace.getTraceUrl();
+        console.log(`Starting ${selectedModel}: ${traceUrl}`);
+    }
+    let authType = "user-access-token";
+    if (process.env.EMPIRICALRUN_API_KEY) {
+        authType = "project-api-key";
+    }
+    const apiClient = new client_1.DashboardAPIClient({
+        authType,
+    });
+    const toolExecutor = new executor_1.ToolExecutor({
+        chatSession: { branchName },
+        repoPath: process.cwd(),
+        apiClient,
+        trace,
+        featureFlags: [],
+        environmentOverrides: await fetchEnvironmentVariables(),
+    });
+    const fileInfoBuilder = () => (0, reader_1.getFileInfoFromFS)(process.cwd());
+    const agentParams = {
+        selectedModel,
+        featureFlags: [],
+        chatState,
+        toolExecutor,
+    };
+    const agent = index_1.MODE_TO_AGENT_MAP[agentMode]({ ...agentParams });
+    if (agent.askUserForInput) {
+        // Show last message to the user for context when we loaded from disk
+        const latest = agent.getHumanReadableLatestMessage();
+        if (latest) {
+            console.log(`${(0, picocolors_1.blue)(latest.role)}: ${latest.textMessage}`);
+        }
+    }
+    while (!stopCriteria(userPrompt)) {
+        if (agent.askUserForInput) {
+            try {
+                userPrompt = await human_in_the_loop_1.humanLoop.getFeedback({
+                    message: "User:",
+                });
+            }
+            catch (e) {
+                // https://github.com/SBoudrias/Inquirer.js/issues/1502#issuecomment-2275991680
+                if (e instanceof Error && e.name === "ExitPromptError") {
+                    concludeAgent(agent, useDiskForChatState);
+                    process.exit(0);
+                }
+                concludeAgent(agent, useDiskForChatState);
+                throw e;
+            }
+            if (!stopCriteria(userPrompt)) {
+                const { text, attachments } = (0, utils_1.extractAttachments)(userPrompt);
+                agent.pushUserMessage(text, attachments);
+            }
+        }
+        else {
+            await agent.runLoop({
+                reporter: reporterFunc,
+                streamingMessageReporter: (() => {
+                    if (!enableStreaming) {
+                        return;
+                    }
+                    let hasStarted = false;
+                    let startedRole = undefined;
+                    return async (delta, snapshot, thinking) => {
+                        if (delta) {
+                            const role = thinking ? "Thinking" : "Assistant";
+                            if (!hasStarted) {
+                                process.stdout.write(`${(0, picocolors_1.blue)(role)}: `);
+                                hasStarted = true;
+                                startedRole = role;
+                            }
+                            else if (hasStarted && role !== startedRole) {
+                                // Changing from thinking -> text block
+                                process.stdout.write("\n");
+                                process.stdout.write(`${(0, picocolors_1.blue)(role)}: `);
+                                startedRole = role;
+                            }
+                            process.stdout.write(delta);
+                        }
+                    };
+                })(),
+                trace,
+                repoInfoBuilder: fileInfoBuilder,
+                onPendingToolCall: async (toolCalls) => {
+                    const { toolResults, checkpoint } = await toolExecutor.execute(toolCalls);
+                    agent.processToolResults(toolCalls, toolResults, checkpoint);
+                },
+            });
+        }
+    }
+    trace?.update({ output: { messages: agent.messages } });
+    await llm_1.langfuseInstance?.flushAsync();
+    console.log(`\n${(0, picocolors_1.gray)("Usage summary -> " + (0, state_1.getUsageSummary)(agent.chatState))}`);
+}

package/dist/agent/code-review/executor/index.d.ts

@@ -0,0 +1,5 @@
+import { BaseToolExecutor, BaseToolExecutorProps } from "../../../tools/executor/base";
+export declare class CodeReviewToolExecutor extends BaseToolExecutor {
+    constructor(params: Omit<BaseToolExecutorProps, "tools">);
+}
+//# sourceMappingURL=index.d.ts.map

package/dist/agent/code-review/executor/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/agent/code-review/executor/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,gBAAgB,EAChB,qBAAqB,EACtB,MAAM,8BAA8B,CAAC;AAItC,qBAAa,sBAAuB,SAAQ,gBAAgB;gBAC9C,MAAM,EAAE,IAAI,CAAC,qBAAqB,EAAE,OAAO,CAAC;CAIzD"}

package/dist/agent/code-review/executor/index.js

@@ -0,0 +1,13 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CodeReviewToolExecutor = void 0;
+const base_1 = require("../../../tools/executor/base");
+const fetch_session_diff_1 = require("../../../tools/fetch-session-diff");
+const file_operations_1 = require("../../../tools/file-operations");
+class CodeReviewToolExecutor extends base_1.BaseToolExecutor {
+    constructor(params) {
+        const tools = [fetch_session_diff_1.fetchSessionDiffTool, ...file_operations_1.viewOnlyTools];
+        super({ ...params, tools });
+    }
+}
+exports.CodeReviewToolExecutor = CodeReviewToolExecutor;

package/dist/agent/code-review/index.d.ts

@@ -0,0 +1,12 @@
+import type { ToolsForLLM } from "@empiricalrun/shared-types";
+import { BaseAgent } from "../base";
+import { type CodeReviewResultV0, type CodeReviewResultV1, type CodeReviewResultV2, CodeReviewSeverity, CodeReviewVerdict } from "./types";
+export type { CodeReviewResultV1, CodeReviewResultV0, CodeReviewResultV2 };
+export { CodeReviewVerdict, CodeReviewSeverity };
+export type CodeReviewVersionedResult = CodeReviewResultV1 | CodeReviewResultV0 | CodeReviewResultV2;
+export declare class CodeReviewAgent extends BaseAgent {
+    protected getTools(): ToolsForLLM;
+    getResult(): CodeReviewVersionedResult | undefined;
+    protected buildSystemPrompt(repoContext: string): Promise<string>;
+}
+//# sourceMappingURL=index.d.ts.map

package/dist/agent/code-review/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/agent/code-review/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAmB,WAAW,EAAE,MAAM,4BAA4B,CAAC;AAG/E,OAAO,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AAEpC,OAAO,EACL,KAAK,kBAAkB,EACvB,KAAK,kBAAkB,EACvB,KAAK,kBAAkB,EACvB,kBAAkB,EAClB,iBAAiB,EAClB,MAAM,SAAS,CAAC;AAEjB,YAAY,EAAE,kBAAkB,EAAE,kBAAkB,EAAE,kBAAkB,EAAE,CAAC;AAC3E,OAAO,EAAE,iBAAiB,EAAE,kBAAkB,EAAE,CAAC;AACjD,MAAM,MAAM,yBAAyB,GACjC,kBAAkB,GAClB,kBAAkB,GAClB,kBAAkB,CAAC;AAEvB,qBAAa,eAAgB,SAAQ,SAAS;IAC5C,SAAS,CAAC,QAAQ,IAAI,WAAW;IAWjC,SAAS,IAAI,yBAAyB,GAAG,SAAS;cAmBlC,iBAAiB,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;CAuHxE"}