ai-codegen-cli-vrk 2.0.8 → 2.1.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-codegen-cli-vrk",
-  "version": "2.0.8",
+  "version": "2.1.0",
   "description": "Minimalist Terminal-based AI code generator",
   "type": "module",
   "bin": {
@@ -14,5 +14,8 @@
     "@google/generative-ai": "^0.21.0",
     "fs-extra": "^11.2.0",
     "readline-sync": "^1.4.10"
+  },
+  "engines": {
+    "node": ">=18.0.0"
   }
 }

package/src/aiClient.js

@@ -7,38 +7,39 @@ export function setApiKey(apiKey) {
   genAI = new GoogleGenerativeAI(apiKey.trim());
 }
 
+/**
+ * Ultimate discovery: Asks Google for a list of working models for your key.
+ */
 async function getWorkingModel() {
   if (activeModel) return activeModel;
 
-  const modelNames = ["gemini-1.5-flash", "gemini-pro", "gemini-1.5-flash-latest"];
-  const apiVersions = ["v1beta", "v1"];
-  
-  let lastError = null;
-
-  for (const ver of apiVersions) {
-    for (const name of modelNames) {
-      try {
-        const m = genAI.getGenerativeModel({ model: name }, { apiVersion: ver });
-        // Minimal verification call
-        await m.generateContent({ 
-          contents: [{ role: "user", parts: [{ text: "hi" }] }], 
-          generationConfig: { maxOutputTokens: 1 } 
-        });
-        activeModel = m;
-        return m;
-      } catch (err) {
-        lastError = err;
-        // If it's a 404, this combination is invalid; try next.
-        if (err.message.includes("404") || err.message.includes("not found")) continue;
-        // If it's an Auth error, stop and throw it.
-        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");
-        }
+  try {
+    // Attempt to list models to see what this specific key is allowed to use
+    // We use v1beta as it has the most complete model list
+    const response = await fetch(`https://generativelanguage.googleapis.com/v1beta/models?key=${genAI.apiKey}`);
+    const data = await response.json();
+    
+    if (data.models) {
+      // Look for 1.5-flash or 2.0-flash in the permitted list
+      const best = data.models.find(m => 
+        m.supportedGenerationMethods.includes("generateContent") && 
+        (m.name.includes("1.5-flash") || m.name.includes("2.0-flash"))
+      );
+      
+      if (best) {
+        // Strip "models/" prefix if present for the SDK
+        const modelName = best.name.split('/').pop();
+        activeModel = genAI.getGenerativeModel({ model: modelName }, { apiVersion: "v1beta" });
+        return activeModel;
       }
     }
+  } catch (e) {
+    // If listing fails, fall back to a safe guess
   }
-  
-  throw new Error(`Connection Failed. Last error: ${lastError?.message || "Unknown"}`);
+
+  // Final fallback to the most common working model
+  activeModel = genAI.getGenerativeModel({ model: "gemini-1.5-flash" }, { apiVersion: "v1beta" });
+  return activeModel;
 }
 
 export async function generateFullProject(task, tests, retryCount = 0) {
@@ -46,7 +47,6 @@ export async function generateFullProject(task, tests, retryCount = 0) {
     const model = await getWorkingModel();
     
     const prompt = `
-You are an expert automated coding exam solver. 
 Generate the ENTIRE project in a SINGLE response. 
 Strictly pass all test cases.
 
@@ -74,10 +74,11 @@ ${tests}
     return text.replace(/```[a-z]*\n([\s\S]*?)\n```/gi, "$1").trim();
 
   } catch (error) {
-    // Handle Overloaded (503) or Rate Limit (429)
-    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));
+    // Handle Overloaded (503) or Rate Limit (429) - Critical for Free Tier
+    const msg = error.message || "";
+    if ((msg.includes("503") || msg.includes("429") || msg.includes("overloaded")) && retryCount < 5) {
+      console.log(`..... (Server busy, waiting 15s to retry ${retryCount + 1}/5)`);
+      await new Promise(r => setTimeout(r, 15000));
       return generateFullProject(task, tests, retryCount + 1);
     }
     throw error;