ai-unit-test-generator 3.1.1 → 3.1.3

package/CHANGELOG.md

@@ -5,6 +5,30 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [3.1.3] - 2025-10-15
+
+### 🐛 Critical Bug Fix (Complete)
+- **Fixed**: All remaining `lib/` path references in batch.ts and parallel-generate.ts
+- **Fixed**: prompt-builder.js, client.js, extractor.js, runner.js, analyzer.js path resolution
+- **Fixed**: Generate workflow now fully functional when installed as npm package
+
+### 📝 Files Modified
+- `src/workflows/batch.ts` - Fixed 5 module path references
+- `src/workflows/parallel-generate.ts` - Fixed 3 module path references
+
+## [3.1.2] - 2025-10-15
+
+### 🐛 Critical Bug Fix
+- **Fixed**: Module path resolution in workflows - corrected hardcoded `lib/workflows/batch.mjs` to `dist/workflows/batch.js`
+- **Fixed**: Generate command now works correctly when installed as npm package
+- **Fixed**: Iterative improvement workflow path resolution
+- **Fixed**: Parallel generation workflow path resolution
+
+### 📝 Files Modified
+- `src/workflows/generate.ts` - Fixed batch script path
+- `src/workflows/iterative-improve.ts` - Fixed 2 batch script path references
+- `src/workflows/all.ts` - Fixed batch script path
+
 ## [3.1.1] - 2025-10-15
 
 ### 🐛 Bug Fixes

package/dist/workflows/all.js

@@ -54,7 +54,7 @@ Batch starting... priority=${priority}, size=${batchSize}, skip=${skip}
 Remaining TODO: ${remain}
 ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━`);
         try {
-            await sh('node', [join(pkgRoot, 'lib/workflows/batch.mjs'), priority, String(batchSize), String(skip)]);
+            await sh('node', [join(pkgRoot, 'dist/workflows/batch.js'), priority, String(batchSize), String(skip)]);
         }
         catch (err) {
             const message = err instanceof Error ? err.message : String(err);

package/dist/workflows/batch.js

@@ -102,7 +102,7 @@ async function main(argv = process.argv) {
     console.log(`📊 Initial coverage: ${beforeCov.toFixed(2)}%`);
     // 1) 生成 Prompt（只针对 TODO 函数，加入上一轮失败提示）
     const promptArgs = [
-        join(pkgRoot, 'lib/ai/prompt-builder.mjs'),
+        join(pkgRoot, 'dist/ai/prompt-builder.js'),
         '--report', reportPath
     ];
     // 只有当 priority 存在时才添加 -p 参数
@@ -125,17 +125,17 @@ async function main(argv = process.argv) {
     writeFileSync('prompt.txt', promptText);
     // 2) 调用 AI
     console.log('\n🤖 Calling AI...');
-    await spawnCommand('node', [join(pkgRoot, 'lib/ai/client.mjs'), 'prompt.txt'], { captureStdout: true });
+    await spawnCommand('node', [join(pkgRoot, 'dist/ai/client.js'), 'prompt.txt'], { captureStdout: true });
     // 3) 提取测试
     console.log('\n📦 Extracting tests...');
-    await spawnCommand('node', [join(pkgRoot, 'lib/ai/extractor.mjs'), 'reports/ai_response.txt', '--overwrite']);
+    await spawnCommand('node', [join(pkgRoot, 'dist/ai/extractor.js'), 'reports/ai_response.txt', '--overwrite']);
     // 4) 运行 Jest（按优先级自适应重跑）
     console.log('\n🧪 Running tests...');
     const reruns = priority === 'P0' ? 1 : 0;
     let testsPassed = false;
     for (let i = 0; i < Math.max(1, reruns + 1); i++) {
         try {
-            await spawnCommand('node', [join(pkgRoot, 'lib/testing/runner.mjs')]);
+            await spawnCommand('node', [join(pkgRoot, 'dist/testing/runner.js')]);
             testsPassed = true;
             break;
         }
@@ -172,7 +172,7 @@ async function main(argv = process.argv) {
     const { spawn: spawnLocal } = await import('node:child_process');
     const { writeFileSync: writeFileSyncLocal } = await import('node:fs');
     await new Promise((resolve) => {
-        const child = spawnLocal('node', [join(pkgRoot, 'lib/testing/analyzer.mjs')], { stdio: ['inherit', 'pipe', 'inherit'] });
+        const child = spawnLocal('node', [join(pkgRoot, 'dist/testing/analyzer.js')], { stdio: ['inherit', 'pipe', 'inherit'] });
         const chunks = [];
         if (child.stdout) {
             child.stdout.on('data', (d) => chunks.push(d));

package/dist/workflows/generate.js

@@ -21,7 +21,7 @@ const PKG_ROOT = join(__dirname, '../..');
  * @returns 批次结果
  */
 async function generateBatch(priority, count, skip, report) {
-    const batchScript = join(PKG_ROOT, 'lib/workflows/batch.mjs');
+    const batchScript = join(PKG_ROOT, 'dist/workflows/batch.js');
     return new Promise((resolve, reject) => {
         const child = spawn('node', [batchScript, priority, String(count), String(skip), report], {
             stdio: 'inherit',

package/dist/workflows/iterative-improve.js

@@ -198,7 +198,7 @@ export async function iterativeImprove(options = {}) {
                     console.log(`\n   🎲 Sample ${sampleIdx + 1}/${samplesPerIteration}...`);
                     // 生成候选测试
                     await spawnCommand('node', [
-                        join(PKG_ROOT, 'lib/workflows/batch.mjs'),
+                        join(PKG_ROOT, 'dist/workflows/batch.js'),
                         'null', // priority (as string for CLI arg)
                         '10', // limit
                         '0', // skip
@@ -235,7 +235,7 @@ export async function iterativeImprove(options = {}) {
             else {
                 // 单样本模式（原有逻辑）
                 await spawnCommand('node', [
-                    join(PKG_ROOT, 'lib/workflows/batch.mjs'),
+                    join(PKG_ROOT, 'dist/workflows/batch.js'),
                     'null', // priority
                     '10', // limit
                     '0', // skip

package/dist/workflows/parallel-generate.js

@@ -178,7 +178,7 @@ async function generateBatch(batch, batchIndex, options = {}) {
         // 1. 生成 Prompt（只包含这个批次的函数）
         const reportArg = reportPath || 'reports/ut_scores.md';
         const promptArgs = [
-            join(PKG_ROOT, 'lib/ai/prompt-builder.mjs'),
+            join(PKG_ROOT, 'dist/ai/prompt-builder.js'),
             '--report', reportArg,
             '--only-todo'
         ];
@@ -207,13 +207,13 @@ async function generateBatch(batch, batchIndex, options = {}) {
         // 2. 调用 AI
         const aiResponsePath = join(batchDir, 'ai_response.txt');
         await sh('node', [
-            join(PKG_ROOT, 'lib/ai/client.mjs'),
+            join(PKG_ROOT, 'dist/ai/client.js'),
             '--prompt', promptPath || '',
             '--out', aiResponsePath || ''
         ], { captureStdout: false, cwd: process.cwd(), env: {} });
         // 3. 提取测试
         await sh('node', [
-            join(PKG_ROOT, 'lib/ai/extractor.mjs'),
+            join(PKG_ROOT, 'dist/ai/extractor.js'),
             aiResponsePath || '',
             '--overwrite'
         ], { captureStdout: false, cwd: process.cwd(), env: {} });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-unit-test-generator",
-  "version": "3.1.1",
+  "version": "3.1.3",
   "description": "AI-powered unit test generator with smart priority scoring, parallel generation, behavior classification, test validation, and deduplication",
   "keywords": [
     "unit-test",