@empiricalrun/test-gen 0.28.3 → 0.28.4

package/CHANGELOG.md

@@ -1,5 +1,11 @@
 # @empiricalrun/test-gen
 
+## 0.28.4
+
+### Patch Changes
+
+- 2d27074: fix: add test for nested test update
+
 ## 0.28.3
 
 ### Patch Changes

package/dist/agent/codegen/run.js

@@ -77,7 +77,7 @@ async function generateTest(testCase, file, options) {
     logger.success("Test generated successfully!");
     const readWriteFileSpan = trace.span({ name: "write-to-file" });
     let contents = fs_extra_1.default.readFileSync(file, "utf-8");
-    const [prependContent, strippedContent] = await (0, web_1.stripAndPrependImports)(response, testCase?.name, testCase?.suites);
+    const [prependContent, strippedContent] = await (0, web_1.stripAndPrependImports)(response, testCase?.name, []);
     let updatedContent = prependContent + contents + `\n\n${strippedContent}`;
     await fs_extra_1.default.writeFile(file, updatedContent, "utf-8");
     readWriteFileSpan.end({ output: { updatedContent } });

package/dist/agent/codegen/update-flow.js

@@ -23,14 +23,14 @@ async function applyFileChanges({ validateTypes = true, trace, testCase, fileCha
         const { testBlock: testBlockUpdate } = (0, web_1.getTypescriptTestBlock)({
             scenarioName: testCase?.name || "",
             content: fileChange.newCode || "",
-            suites: testCase?.suites,
+            suites: [], // suites should be empty here since we ask LLM to send immediate parent AST node for the code update. so there won't be any nesting here, just the test block
         });
         if (testBlockUpdate) {
             // assuming the test case getting updated
             // maintaining the previous accuracy of the test case update
             const readWriteFileSpan = trace.span({ name: "write-to-file" });
             let contents = await fs_extra_1.default.readFile(fileChange.filePath, "utf-8");
-            const [prependContent, strippedContent] = await (0, web_1.stripAndPrependImports)(fileChange.newCode, testCase?.name, testCase?.suites);
+            const [prependContent, strippedContent] = await (0, web_1.stripAndPrependImports)(fileChange.newCode, testCase?.name, []);
             let updatedContent = prependContent + contents + `\n\n${strippedContent}`;
             const { testBlock } = (0, web_1.getTypescriptTestBlock)({
                 scenarioName: testCase?.name,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@empiricalrun/test-gen",
-  "version": "0.28.3",
+  "version": "0.28.4",
   "publishConfig": {
     "registry": "https://registry.npmjs.org/",
     "access": "public"
@@ -44,8 +44,8 @@
     "tsx": "^4.16.2",
     "typescript": "^5.3.3",
     "@empiricalrun/r2-uploader": "^0.3.0",
-    "@empiricalrun/reporter": "^0.18.2",
-    "@empiricalrun/llm": "^0.9.2"
+    "@empiricalrun/llm": "^0.9.2",
+    "@empiricalrun/reporter": "^0.18.2"
   },
   "devDependencies": {
     "@types/detect-port": "^1.3.5",

package/vitest.config.ts

@@ -0,0 +1,10 @@
+import { config as dotenvConfig } from "dotenv";
+import { defineConfig } from "vitest/config";
+
+dotenvConfig({ path: [".env.local", ".env"] });
+
+export default defineConfig({
+  test: {
+    setupFiles: ["dotenv/config"],
+  },
+});