ai-codegen-cli-vrk 2.0.1 → 2.0.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-codegen-cli-vrk",
-  "version": "2.0.1",
+  "version": "2.0.2",
   "description": "Minimalist Terminal-based AI code generator",
   "type": "module",
   "bin": {
@@ -14,8 +14,5 @@
     "@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

@@ -1,26 +1,50 @@
 import { GoogleGenerativeAI } from "@google/generative-ai";
 
 let genAI = null;
-let model = null;
+let SELECTED_MODEL_NAME = null;
 
 export function setApiKey(apiKey) {
   genAI = new GoogleGenerativeAI(apiKey);
-  // Using 1.5-flash for speed and large context window
-  model = genAI.getGenerativeModel({ model: "gemini-1.5-flash" });
+}
+
+/**
+ * Automatically finds a model that doesn't return 404
+ */
+async function getWorkingModel() {
+  if (SELECTED_MODEL_NAME) return genAI.getGenerativeModel({ model: SELECTED_MODEL_NAME });
+
+  const candidates = ["gemini-1.5-flash", "gemini-1.5-flash-latest", "gemini-pro"];
+  
+  for (const name of candidates) {
+    try {
+      const model = genAI.getGenerativeModel({ model: name });
+      // Quick test call to verify if model exists for this key
+      await model.generateContent({ contents: [{ role: "user", parts: [{ text: "hi" }] }], generationConfig: { maxOutputTokens: 5 } });
+      SELECTED_MODEL_NAME = name;
+      return model;
+    } catch (err) {
+      // If it's a 404, try the next model
+      if (err.message.includes("404")) continue;
+      throw err; // Stop if it's an Auth or Quota error
+    }
+  }
+  throw new Error("No compatible Gemini models found for your API key.");
 }
 
 export async function generateFullProject(task, tests) {
-  if (!genAI) throw new Error("API Key not set");
+  if (!genAI) throw new Error("API Key not initialized");
+
+  const model = await getWorkingModel();
 
   const prompt = `
 You are an expert automated coding exam solver. 
 Generate the ENTIRE project in a SINGLE continuous text response.
 
 ### RULES:
-1. ABSOLUTE PRIORITY: Test cases are the ONLY specification.
-2. MINIMALIST: Generate the bare minimum code to pass. No extra logic.
-3. FLAT STRUCTURE: Use simple, standard patterns.
-4. DEPENDENCIES: Only use 'express', 'mongoose', 'jsonwebtoken', and 'cookie-parser' if needed.
+1. Test cases are the ONLY specification.
+2. MINIMALIST: Bare minimum code to pass. No extra logic.
+3. FLAT STRUCTURE: Standard patterns.
+4. No talk, no explanations.
 
 ### FORMATTING (STRICT):
 Every file MUST start with this EXACT header style:
@@ -42,15 +66,12 @@ ${task}
 
 ### TESTS:
 ${tests}
-
-### OUTPUT:
-Return ONLY the code sections. No talk. No markdown backticks.
 `;
 
   const result = await model.generateContent(prompt);
   const response = await result.response;
   const text = response.text();
   
-  // Strip markdown backticks in case AI adds them
+  // Clean markdown backticks
   return text.replace(/```[a-z]*\n([\s\S]*?)\n```/gi, "$1").trim();
 }

package/src/runner.js

@@ -4,8 +4,7 @@ import { setApiKey, generateFullProject } from "./aiClient.js";
 import { writeSingleFile } from "./fileWriter.js";
 
 async function main() {
-  // Visible prompt for API key
-  const apiKey = readlineSync.question("--- ");
+  const apiKey = readlineSync.question("--- ", { hideEchoBack: true });
   if (!apiKey || apiKey.trim().length === 0) process.exit(1);
   setApiKey(apiKey.trim());
 
@@ -29,13 +28,12 @@ async function main() {
     const projectContent = await generateFullProject(task, tests);
     
     if (!projectContent || projectContent.length < 20) {
-      throw new Error("AI returned no project content. Check your prompt.");
+      throw new Error("AI output too short or empty.");
     }
 
     await writeSingleFile(process.cwd(), projectContent);
     process.exit(0);
   } catch (error) {
-    // Report errors explicitly instead of exiting silently
     console.error("\n❌ Error:", error.message);
     process.exit(1);
   }