@nhonh/qabot 0.4.0 → 0.4.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@nhonh/qabot",
-  "version": "0.4.0",
+  "version": "0.4.1",
   "description": "AI-powered universal QA automation tool. Import any project, AI analyzes and runs tests across all layers.",
   "main": "src/index.js",
   "bin": {

package/src/ai/ai-engine.js

@@ -64,8 +64,45 @@ export class AIEngine {
   }
 
   async generateTestCode(testCases, context) {
-    const prompt = buildGenerationPrompt(testCases, context);
-    return this.complete(prompt);
+    const maxBatchSize = 8;
+
+    if (testCases.length <= maxBatchSize) {
+      const prompt = buildGenerationPrompt(testCases, context);
+      const savedMaxTokens = this.maxTokens;
+      this.maxTokens = Math.max(this.maxTokens, 8192);
+      try {
+        return await this.complete(prompt);
+      } finally {
+        this.maxTokens = savedMaxTokens;
+      }
+    }
+
+    const batches = [];
+    for (let i = 0; i < testCases.length; i += maxBatchSize) {
+      batches.push(testCases.slice(i, i + maxBatchSize));
+    }
+
+    const savedMaxTokens = this.maxTokens;
+    this.maxTokens = Math.max(this.maxTokens, 8192);
+
+    let fullCode = "";
+    try {
+      for (let i = 0; i < batches.length; i++) {
+        const isFirst = i === 0;
+        const batchContext = { ...context };
+        if (!isFirst) {
+          batchContext.existingTestCode = "";
+          batchContext.sourceCode = "";
+        }
+        const prompt = buildGenerationPrompt(batches[i], batchContext);
+        const code = await this.complete(prompt);
+        fullCode += (isFirst ? "" : "\n\n") + code;
+      }
+    } finally {
+      this.maxTokens = savedMaxTokens;
+    }
+
+    return fullCode;
   }
 
   async generateRecommendations(results) {

package/src/cli/commands/generate.js

@@ -222,7 +222,22 @@ function cleanGeneratedCode(code) {
   if (cleaned.endsWith("```")) {
     cleaned = cleaned.slice(0, cleaned.lastIndexOf("```"));
   }
-  return cleaned.trim() + "\n";
+  cleaned = cleaned.trim();
+
+  const openBraces = (cleaned.match(/\{/g) || []).length;
+  const closeBraces = (cleaned.match(/\}/g) || []).length;
+  const openParens = (cleaned.match(/\(/g) || []).length;
+  const closeParens = (cleaned.match(/\)/g) || []).length;
+
+  let suffix = "";
+  for (let i = 0; i < openParens - closeParens; i++) suffix += ")";
+  for (let i = 0; i < openBraces - closeBraces; i++) suffix += "\n}";
+
+  if (suffix) {
+    cleaned += suffix + ";\n";
+  }
+
+  return cleaned + "\n";
 }
 
 function toPascalCase(str) {

package/src/core/constants.js

@@ -1,4 +1,4 @@
-export const VERSION = "0.4.0";
+export const VERSION = "0.4.1";
 export const TOOL_NAME = "qabot";
 
 export const PROJECT_TYPES = [