ai-codegen-cli-vrk 2.0.7 → 2.0.9

package/package.json

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

package/src/aiClient.js

@@ -4,31 +4,43 @@ let genAI = null;
 let activeModel = null;
 
 export function setApiKey(apiKey) {
-  // Auto-trim to remove any accidental spaces from copy-pasting
-  const cleanKey = apiKey.trim();
-  genAI = new GoogleGenerativeAI(cleanKey);
+  genAI = new GoogleGenerativeAI(apiKey.trim());
 }
 
+/**
+ * Robust model discovery. Tries newer Flash models first.
+ */
 async function getWorkingModel() {
   if (activeModel) return activeModel;
 
-  const candidates = ["gemini-1.5-flash", "gemini-1.5-flash-latest", "gemini-pro"];
+  // Added 2.0 Flash to the list
+  const modelNames = ["gemini-1.5-flash", "gemini-2.0-flash-exp", "gemini-1.5-flash-latest", "gemini-pro"];
+  const apiVersions = ["v1beta", "v1"];
   
-  for (const name of candidates) {
-    try {
-      // Force the stable 'v1' API version to avoid Beta key issues
-      const m = genAI.getGenerativeModel({ model: name }, { apiVersion: "v1" });
-      
-      // Verification call
-      await m.generateContent({ contents: [{ role: "user", parts: [{ text: "hi" }] }], generationConfig: { maxOutputTokens: 1 } });
-      activeModel = m;
-      return m;
-    } catch (err) {
-      if (err.message.includes("404") || err.message.includes("not found")) continue;
-      throw err;
+  let lastError = null;
+
+  for (const ver of apiVersions) {
+    for (const name of modelNames) {
+      try {
+        const m = genAI.getGenerativeModel({ model: name }, { apiVersion: ver });
+        // Verification call
+        await m.generateContent({ 
+          contents: [{ role: "user", parts: [{ text: "hi" }] }], 
+          generationConfig: { maxOutputTokens: 1 } 
+        });
+        activeModel = m;
+        return m;
+      } catch (err) {
+        lastError = err;
+        if (err.message.includes("404") || err.message.includes("not found")) continue;
+        if (err.message.includes("API key not valid") || err.message.includes("401")) {
+           throw new Error("Invalid API Key. Please check it at aistudio.google.com");
+        }
+      }
     }
   }
-  throw new Error("No compatible Gemini models found.");
+  
+  throw new Error(`Connection Failed. Google's server rejected the model request. Error: ${lastError?.message}`);
 }
 
 export async function generateFullProject(task, tests, retryCount = 0) {
@@ -63,14 +75,11 @@ ${tests}
     return text.replace(/```[a-z]*\n([\s\S]*?)\n```/gi, "$1").trim();
 
   } catch (error) {
-    // If key is truly invalid, give a clear message
-    if (error.message.includes("API key not valid")) {
-      throw new Error("The API key you entered is invalid. Please double-check it at aistudio.google.com");
-    }
-
+    // Handle Overloaded (503) or Rate Limit (429) - Critical for Free Tier
     if ((error.message.includes("503") || error.message.includes("429")) && retryCount < 5) {
-      console.log(`..... (Server busy, retrying ${retryCount + 1}/5)`);
-      await new Promise(r => setTimeout(r, 10000));
+      console.log(`..... (Quota hit or Server busy, waiting to retry ${retryCount + 1}/5)`);
+      // Wait 15 seconds to safely clear the Free Tier 15-RPM limit
+      await new Promise(r => setTimeout(r, 15000));
       return generateFullProject(task, tests, retryCount + 1);
     }
     throw error;

package/src/runner.js

@@ -6,8 +6,6 @@ import { writeSingleFile } from "./fileWriter.js";
 async function main() {
   const apiKeyInput = readlineSync.question("--- ", { hideEchoBack: false });
   if (!apiKeyInput || apiKeyInput.trim().length === 0) process.exit(1);
-  
-  // Trimming again here just to be safe
   setApiKey(apiKeyInput.trim());
 
   const task = readlineSync.question("- ");