@empiricalrun/test-gen 0.57.1 → 0.58.0

package/dist/utils/repo-tree.d.ts

@@ -1,3 +1,4 @@
+import { FileInfo } from "../types";
 export declare const DEFAULT_EXCLUDE: (string | RegExp)[];
-export declare function generateAsciiTree(dirPath: string, options?: {}): string;
+export declare function generateAsciiTree(root: FileInfo, options?: {}): string;
 //# sourceMappingURL=repo-tree.d.ts.map

package/dist/utils/repo-tree.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"repo-tree.d.ts","sourceRoot":"","sources":["../../src/utils/repo-tree.ts"],"names":[],"mappings":"AAGA,eAAO,MAAM,eAAe,qBAW3B,CAAC;AAEF,wBAAgB,iBAAiB,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,KAAK,UAsE9D"}
+{"version":3,"file":"repo-tree.d.ts","sourceRoot":"","sources":["../../src/utils/repo-tree.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,UAAU,CAAC;AAEpC,eAAO,MAAM,eAAe,qBAW3B,CAAC;AAEF,wBAAgB,iBAAiB,CAAC,IAAI,EAAE,QAAQ,EAAE,OAAO,KAAK,UAqF7D"}

package/dist/utils/repo-tree.js

@@ -1,11 +1,7 @@
 "use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.generateAsciiTree = exports.DEFAULT_EXCLUDE = void 0;
-const fs_1 = __importDefault(require("fs"));
-const path_1 = __importDefault(require("path"));
+exports.DEFAULT_EXCLUDE = void 0;
+exports.generateAsciiTree = generateAsciiTree;
 exports.DEFAULT_EXCLUDE = [
     "node_modules",
     "dist",
@@ -18,62 +14,75 @@ exports.DEFAULT_EXCLUDE = [
     "auth",
     "package-lock.json",
 ];
-function generateAsciiTree(dirPath, options = {}) {
+function generateAsciiTree(root, options = {}) {
     const defaultOptions = {
         showHidden: false,
         exclude: exports.DEFAULT_EXCLUDE,
         maxDepth: 10,
     };
     const opts = { ...defaultOptions, ...options };
-    // Make sure the path exists and is a directory
-    if (!fs_1.default.existsSync(dirPath) || !fs_1.default.statSync(dirPath).isDirectory()) {
-        throw new Error(`"${dirPath}" is not a valid directory path`);
+    if (!root.isDirectory) {
+        throw new Error(`"${root.path}" is not a valid directory`);
+    }
+    let result = root.name + "\n";
+    const stack = [];
+    // Add root's children to stack in reverse order
+    const rootChildren = [...root.children]
+        .filter((child) => child !== undefined)
+        .sort((a, b) => {
+        // Files first, then directories
+        if (!a.isDirectory && b.isDirectory)
+            return -1;
+        if (a.isDirectory && !b.isDirectory)
+            return 1;
+        return a.name.localeCompare(b.name);
+    });
+    for (let i = rootChildren.length - 1; i >= 0; i--) {
+        const child = rootChildren[i];
+        if (child) {
+            stack.push([child, "", 1, i === 0]);
+        }
     }
-    // Start with the root directory name
-    const rootName = path_1.default.basename(dirPath);
-    let result = rootName + "\n";
-    function processDirectory(currentPath, prefix = "", depth = 1) {
+    while (stack.length > 0) {
+        const current = stack.pop();
+        if (!current)
+            continue;
+        const [node, prefix, depth, isLast] = current;
         if (depth > opts.maxDepth)
-            return;
-        const items = fs_1.default.readdirSync(currentPath);
-        // Sort items: directories first, then files
-        const sortedItems = items.sort((a, b) => {
-            const aIsDir = fs_1.default.statSync(path_1.default.join(currentPath, a)).isDirectory();
-            const bIsDir = fs_1.default.statSync(path_1.default.join(currentPath, b)).isDirectory();
-            if (aIsDir && !bIsDir)
-                return -1;
-            if (!aIsDir && bIsDir)
-                return 1;
-            return a.localeCompare(b);
-        });
-        // Process each item
-        sortedItems.forEach((item, index) => {
-            // Skip hidden files if not showing hidden
-            if (!opts.showHidden && item.startsWith("."))
-                return;
-            // Skip excluded patterns
-            if (opts.exclude.some((pattern) => typeof pattern === "string"
-                ? item === pattern
-                : pattern instanceof RegExp
-                    ? pattern.test(item)
-                    : false))
-                return;
-            const itemPath = path_1.default.join(currentPath, item);
-            const isDirectory = fs_1.default.statSync(itemPath).isDirectory();
-            const isLast = index === sortedItems.length - 1;
-            // Current item symbols
-            const symbol = isLast ? "└── " : "├── ";
-            const nextPrefix = isLast ? "    " : "│   ";
-            // Add the current item to the result
-            result += `${prefix}${symbol}${item}${isDirectory ? "/" : ""}\n`;
-            // Process subdirectories
-            if (isDirectory) {
-                processDirectory(itemPath, prefix + nextPrefix, depth + 1);
+            continue;
+        // Skip hidden files if not showing hidden
+        if (!opts.showHidden && node.name.startsWith("."))
+            continue;
+        // Skip excluded patterns
+        if (opts.exclude.some((pattern) => typeof pattern === "string"
+            ? node.path === pattern
+            : pattern instanceof RegExp
+                ? pattern.test(node.path)
+                : false))
+            continue;
+        // Add the current item to the result
+        const symbol = isLast ? "└── " : "├── ";
+        result += `${prefix}${symbol}${node.name}${node.isDirectory ? "/" : ""}\n`;
+        if (node.isDirectory && node.children.length > 0) {
+            const nextPrefix = prefix + (isLast ? "    " : "│   ");
+            // Sort and add children to stack in reverse order
+            const children = [...node.children]
+                .filter((child) => child !== undefined)
+                .sort((a, b) => {
+                // Files first, then directories
+                if (!a.isDirectory && b.isDirectory)
+                    return -1;
+                if (a.isDirectory && !b.isDirectory)
+                    return 1;
+                return a.name.localeCompare(b.name);
+            });
+            for (let i = children.length - 1; i >= 0; i--) {
+                const child = children[i];
+                if (child) {
+                    stack.push([child, nextPrefix, depth + 1, i === 0]);
+                }
             }
-        });
+        }
     }
-    // Start the recursive processing
-    processDirectory(dirPath);
     return result;
 }
-exports.generateAsciiTree = generateAsciiTree;

package/dist/utils/slug.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"slug.d.ts","sourceRoot":"","sources":["../../src/utils/slug.ts"],"names":[],"mappings":"AAQA,eAAO,MAAM,uBAAuB,SAAgB,MAAM,oBAOzD,CAAC"}
+{"version":3,"file":"slug.d.ts","sourceRoot":"","sources":["../../src/utils/slug.ts"],"names":[],"mappings":"AAQA,eAAO,MAAM,uBAAuB,GAAU,MAAM,MAAM,oBAOzD,CAAC"}

package/dist/utils/slug.js

@@ -1,7 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.asyncGenerateRandomSlug = void 0;
-// @ts-expect-error
+// @ts-ignore
 let customAlphabet;
 const loadNanoid = async () => {
     const { customAlphabet: importedCustomAlphabet } = await import("nanoid");

package/dist/utils/string.js

@@ -1,9 +1,8 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.convertTextToJsComments = void 0;
+exports.convertTextToJsComments = convertTextToJsComments;
 function convertTextToJsComments(text) {
     const lines = text.split("\n");
     const commentedLines = lines.map((line) => `// ${line}`);
     return commentedLines.join("\n");
 }
-exports.convertTextToJsComments = convertTextToJsComments;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@empiricalrun/test-gen",
-  "version": "0.57.1",
+  "version": "0.58.0",
   "publishConfig": {
     "registry": "https://registry.npmjs.org/",
     "access": "public"
@@ -60,9 +60,9 @@
     "tsx": "^4.16.2",
     "typescript": "^5.3.3",
     "zod": "^3.23.8",
-    "@empiricalrun/llm": "^0.15.3",
-    "@empiricalrun/r2-uploader": "^0.3.8",
-    "@empiricalrun/test-run": "^0.8.4"
+    "@empiricalrun/llm": "^0.15.4",
+    "@empiricalrun/r2-uploader": "^0.3.9",
+    "@empiricalrun/test-run": "^0.8.5"
   },
   "devDependencies": {
     "@playwright/test": "1.47.1",
@@ -76,7 +76,7 @@
     "playwright": "1.47.1",
     "serve-handler": "^6.1.6",
     "ts-patch": "^3.3.0",
-    "@empiricalrun/shared-types": "0.0.7"
+    "@empiricalrun/shared-types": "0.0.8"
   },
   "scripts": {
     "dev": "tspc --build --watch",

package/tsconfig.tsbuildinfo

@@ -0,0 +1 @@
+{"root":["./src/index.ts","./src/actions/assert.ts","./src/actions/click.ts","./src/actions/done.ts","./src/actions/fill.ts","./src/actions/goto.ts","./src/actions/hover.ts","./src/actions/index.ts","./src/actions/next-task.ts","./src/actions/press.ts","./src/actions/skill.ts","./src/actions/text-content.ts","./src/actions/constants/index.ts","./src/actions/utils/index.ts","./src/agent/browsing/index.ts","./src/agent/browsing/run.ts","./src/agent/browsing/utils.ts","./src/agent/chat/agent-loop.ts","./src/agent/chat/exports.ts","./src/agent/chat/index.ts","./src/agent/chat/prompt.ts","./src/agent/chat/repo.ts","./src/agent/chat/state.ts","./src/agent/chat/types.ts","./src/agent/codegen/create-test-block.ts","./src/agent/codegen/fix-ts-errors.ts","./src/agent/codegen/generate-code-apply-changes.ts","./src/agent/codegen/lexical-scoped-vars.ts","./src/agent/codegen/repo-edit.ts","./src/agent/codegen/run.ts","./src/agent/codegen/skills-retriever.ts","./src/agent/codegen/test-update-feedback.ts","./src/agent/codegen/types.ts","./src/agent/codegen/update-flow.ts","./src/agent/codegen/use-skill.ts","./src/agent/codegen/utils.ts","./src/agent/cua/computer.ts","./src/agent/cua/index.ts","./src/agent/cua/model.ts","./src/agent/diagnosis-agent/index.ts","./src/agent/diagnosis-agent/strict-mode-violation.ts","./src/agent/enrich-prompt/index.ts","./src/agent/enrich-prompt/utils.ts","./src/agent/infer-agent/index.ts","./src/agent/master/action-tool-calls.ts","./src/agent/master/element-annotation.ts","./src/agent/master/execute-browser-action.ts","./src/agent/master/execute-skill-action.ts","./src/agent/master/next-action.ts","./src/agent/master/planner.ts","./src/agent/master/run.ts","./src/agent/master/scroller.ts","./src/agent/master/with-hints.ts","./src/agent/master/browser-tests/cua.spec.ts","./src/agent/master/browser-tests/fixtures.ts","./src/agent/master/browser-tests/index.spec.ts","./src/agent/master/browser-tests/skills.spec.ts","./src/agent/master/icon-descriptor/index.ts","./src/agent/master/icon-descriptor/normalize-svg.ts","./src/agent/planner/run-time-planner.ts","./src/agent/planner/run.ts","./src/bin/index.ts","./src/bin/logger/index.ts","./src/bin/utils/context.ts","./src/bin/utils/index.ts","./src/bin/utils/fs/index.ts","./src/bin/utils/platform/web/index.ts","./src/bin/utils/platform/web/test-files/ts-path-import-validate.ts","./src/bin/utils/scenarios/index.ts","./src/browser-injected-scripts/annotate-elements.spec.ts","./src/constants/index.ts","./src/errors/index.ts","./src/evals/add-scenario-agent.evals.ts","./src/evals/append-create-test-agent.evals.ts","./src/evals/fetch-pom-skills-agent.evals.ts","./src/evals/infer-master-or-code-agent.evals.ts","./src/evals/master-agent.evals.ts","./src/evals/type.ts","./src/evals/update-scenario-agent.evals.ts","./src/file/client.ts","./src/file/server.ts","./src/human-in-the-loop/cli.ts","./src/human-in-the-loop/index.ts","./src/human-in-the-loop/ipc.ts","./src/page/index.ts","./src/prompts/lib/ts-transformer.ts","./src/reporter/index.ts","./src/reporter/lib.ts","./src/session/index.ts","./src/test-build/index.ts","./src/tool-call-service/index.ts","./src/tools/codegen-agent.ts","./src/tools/commit-and-create-pr.ts","./src/tools/diagnosis-fetcher.ts","./src/tools/download-build.ts","./src/tools/environment-crud.ts","./src/tools/str_replace_editor.ts","./src/tools/test-gen-browser.ts","./src/tools/test-run.ts","./src/tools/grep/index.ts","./src/tools/grep/ripgrep/index.ts","./src/tools/grep/ripgrep/types.ts","./src/tools/test-run-fetcher/index.ts","./src/tools/test-run-fetcher/types.ts","./src/tools/utils/index.ts","./src/types/handlebars.d.ts","./src/types/index.ts","./src/uploader/index.ts","./src/uploader/utils.ts","./src/utils/checkpoint.ts","./src/utils/env.ts","./src/utils/exec.ts","./src/utils/file-tree.ts","./src/utils/file.ts","./src/utils/git.ts","./src/utils/html.ts","./src/utils/index.ts","./src/utils/pw-test.ts","./src/utils/repo-tree.ts","./src/utils/slug.ts","./src/utils/string.ts"],"version":"5.8.3"}

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

@@ -1,4 +0,0 @@
-import { IChatModel } from "@empiricalrun/llm/chat";
-import { SupportedChatModels } from "./types";
-export declare function createChatModel(messages: any[], selectedModel: SupportedChatModels): IChatModel<any>;
-//# sourceMappingURL=model.d.ts.map

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

@@ -1 +0,0 @@
-{"version":3,"file":"model.d.ts","sourceRoot":"","sources":["../../../src/agent/chat/model.ts"],"names":[],"mappings":"AAAA,OAAO,EAGL,UAAU,EAEX,MAAM,wBAAwB,CAAC;AAEhC,OAAO,EAAE,mBAAmB,EAAE,MAAM,SAAS,CAAC;AAE9C,wBAAgB,eAAe,CAC7B,QAAQ,EAAE,GAAG,EAAE,EACf,aAAa,EAAE,mBAAmB,GACjC,UAAU,CAAC,GAAG,CAAC,CAWjB"}

package/dist/agent/chat/model.js

@@ -1,17 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.createChatModel = void 0;
-const chat_1 = require("@empiricalrun/llm/chat");
-function createChatModel(messages, selectedModel) {
-    if (selectedModel.startsWith("claude")) {
-        return new chat_1.ClaudeChatModel(messages);
-    }
-    if (selectedModel.startsWith("gemini")) {
-        return new chat_1.GeminiChatModel(messages);
-    }
-    if (selectedModel.startsWith("o4")) {
-        return new chat_1.OpenAIChatModel(messages);
-    }
-    throw new Error(`Unsupported model: ${selectedModel}`);
-}
-exports.createChatModel = createChatModel;