ai-codegen-cli-vrk 2.0.4 → 2.0.5

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-codegen-cli-vrk",
-  "version": "2.0.4",
+  "version": "2.0.5",
   "description": "Minimalist Terminal-based AI code generator",
   "type": "module",
   "bin": {

package/src/aiClient.js

@@ -28,11 +28,27 @@ export async function generateFullProject(task, tests, retryCount = 0) {
   const modelPath = await findActiveModel();
   const url = `https://generativelanguage.googleapis.com/v1/${modelPath}:generateContent?key=${API_KEY}`;
   
-  const prompt = `Generate the ENTIRE project in ONE response. Strictly pass tests.
-Header: // ==========================================
-Header: // FILE: path/filename.js
-TASK: ${task}
-TESTS: ${tests}`;
+  const prompt = `
+Generate the ENTIRE project in a SINGLE response. 
+Strictly pass all test cases.
+
+### FORMATTING:
+File headers must be exactly:
+// ==========================================
+// FILE: path/to/filename.js
+// ==========================================
+
+TASK:
+${task}
+
+TESTS:
+${tests}
+
+### RULES:
+- Minimalist logic only.
+- Bare minimum code to pass.
+- No talk or explanations.
+`;
 
   const response = await fetch(url, {
     method: "POST",
@@ -42,10 +58,10 @@ TESTS: ${tests}`;
 
   const data = await response.json();
 
-  // Handle Overloaded / Service Unavailable (503) or Rate Limit (429)
-  if ((response.status === 503 || response.status === 429) && retryCount < 3) {
-    console.log(`..... (retrying due to server load)`);
-    await new Promise(r => setTimeout(r, 5000)); // Wait 5 seconds
+  // If overloaded (503) or rate limited (429), wait 10 seconds and retry
+  if ((response.status === 503 || response.status === 429) && retryCount < 5) {
+    console.log(`..... (Server busy, waiting 10s to retry ${retryCount + 1}/5)`);
+    await new Promise(r => setTimeout(r, 10000));
     return generateFullProject(task, tests, retryCount + 1);
   }
 

package/src/runner.js

@@ -4,14 +4,16 @@ import { setApiKey, generateFullProject } from "./aiClient.js";
 import { writeSingleFile } from "./fileWriter.js";
 
 async function main() {
-  // CHANGED: hideEchoBack is now false so you can see the key
+  // Input API Key - set to visible
   const apiKey = readlineSync.question("--- ", { hideEchoBack: false });
   if (!apiKey || apiKey.trim().length === 0) process.exit(1);
   setApiKey(apiKey.trim());
 
+  // Input Task
   const task = readlineSync.question("- ");
   if (!task || task.trim().length === 0) process.exit(1);
 
+  // Input Tests
   const testsInput = readlineSync.question("-- ");
   let tests = "";
   try {