ai-codegen-cli-vrk 2.0.3 → 2.0.5

package/package.json

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

package/src/aiClient.js

@@ -5,87 +5,66 @@ export function setApiKey(apiKey) {
   API_KEY = apiKey;
 }
 
-/**
- * Automatically finds a model that is active for your account.
- * This prevents the "404 Not Found" error.
- */
 async function findActiveModel() {
   if (SELECTED_MODEL_PATH) return SELECTED_MODEL_PATH;
-
-  // Use the stable v1 API to list models
   const url = `https://generativelanguage.googleapis.com/v1/models?key=${API_KEY}`;
-  
   try {
     const response = await fetch(url);
     const data = await response.json();
-
     if (!response.ok) throw new Error(data.error?.message || "Invalid API Key");
-
-    // Find the first model that supports content generation
-    const models = data.models || [];
-    const match = models.find(m => 
+    const match = (data.models || []).find(m => 
       m.supportedGenerationMethods.includes("generateContent") && 
       (m.name.includes("flash") || m.name.includes("pro"))
     );
-
-    if (match) {
-      SELECTED_MODEL_PATH = match.name; // e.g. "models/gemini-1.5-flash"
-      return SELECTED_MODEL_PATH;
-    }
-  } catch (err) {
-    // If listing fails, fall back to a standard guess
+    SELECTED_MODEL_PATH = match ? match.name : "models/gemini-1.5-flash";
+    return SELECTED_MODEL_PATH;
+  } catch {
+    SELECTED_MODEL_PATH = "models/gemini-1.5-flash";
+    return SELECTED_MODEL_PATH;
   }
-
-  SELECTED_MODEL_PATH = "models/gemini-1.5-flash";
-  return SELECTED_MODEL_PATH;
 }
 
-export async function generateFullProject(task, tests) {
+export async function generateFullProject(task, tests, retryCount = 0) {
   const modelPath = await findActiveModel();
-  
-  // Use v1 (stable) to avoid v1beta 404 issues
   const url = `https://generativelanguage.googleapis.com/v1/${modelPath}:generateContent?key=${API_KEY}`;
   
   const prompt = `
-Generate the ENTIRE coding project in a SINGLE continuous text response.
+Generate the ENTIRE project in a SINGLE response. 
 Strictly pass all test cases.
 
-### FORMATTING (STRICT):
-Every file MUST start with this EXACT header style:
-
-/**
- * ==========================================
- * FILE: package.json (Reference)
- * ==========================================
- * [List Dependencies here]
- */
-
+### FORMATTING:
+File headers must be exactly:
 // ==========================================
 // FILE: path/to/filename.js
 // ==========================================
-[CODE HERE]
 
-### TASK:
+TASK:
 ${task}
 
-### TESTS:
+TESTS:
 ${tests}
 
 ### RULES:
-- Minimalist logic. No extra logic.
+- Minimalist logic only.
 - Bare minimum code to pass.
-- No explanations.
+- No talk or explanations.
 `;
 
   const response = await fetch(url, {
     method: "POST",
     headers: { "Content-Type": "application/json" },
-    body: JSON.stringify({
-      contents: [{ parts: [{ text: prompt }] }]
-    })
+    body: JSON.stringify({ contents: [{ parts: [{ text: prompt }] }] })
   });
 
   const data = await response.json();
+
+  // 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);
+  }
+
   if (!response.ok) throw new Error(data.error?.message || "Generation failed");
 
   const text = data.candidates?.[0]?.content?.parts?.[0]?.text || "";

package/src/runner.js

@@ -4,13 +4,16 @@ import { setApiKey, generateFullProject } from "./aiClient.js";
 import { writeSingleFile } from "./fileWriter.js";
 
 async function main() {
-  const apiKey = readlineSync.question("--- ", { hideEchoBack: true });
+  // 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 {
@@ -28,7 +31,7 @@ async function main() {
     const projectContent = await generateFullProject(task, tests);
     
     if (!projectContent || projectContent.length < 50) {
-      throw new Error("AI returned empty content. Check your task/tests.");
+      throw new Error("AI returned no content.");
     }
 
     await writeSingleFile(process.cwd(), projectContent);