ai-codegen-cli-vrk 2.0.2 → 2.0.4

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-codegen-cli-vrk",
-  "version": "2.0.2",
+  "version": "2.0.4",
   "description": "Minimalist Terminal-based AI code generator",
   "type": "module",
   "bin": {
@@ -11,8 +11,10 @@
     "src"
   ],
   "dependencies": {
-    "@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,77 +1,56 @@
-import { GoogleGenerativeAI } from "@google/generative-ai";
-
-let genAI = null;
-let SELECTED_MODEL_NAME = null;
+let API_KEY = null;
+let SELECTED_MODEL_PATH = null;
 
 export function setApiKey(apiKey) {
-  genAI = new GoogleGenerativeAI(apiKey);
+  API_KEY = apiKey;
 }
 
-/**
- * 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
-    }
+async function findActiveModel() {
+  if (SELECTED_MODEL_PATH) return SELECTED_MODEL_PATH;
+  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");
+    const match = (data.models || []).find(m => 
+      m.supportedGenerationMethods.includes("generateContent") && 
+      (m.name.includes("flash") || m.name.includes("pro"))
+    );
+    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;
   }
-  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 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. 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:
-
-/**
- * ==========================================
- * FILE: package.json (Reference)
- * ==========================================
- * [List Dependencies here]
- */
-
-// ==========================================
-// FILE: path/to/filename.js
-// ==========================================
-[CODE HERE]
-
-### TASK:
-${task}
+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 response = await fetch(url, {
+    method: "POST",
+    headers: { "Content-Type": "application/json" },
+    body: JSON.stringify({ contents: [{ parts: [{ text: prompt }] }] })
+  });
+
+  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
+    return generateFullProject(task, tests, retryCount + 1);
+  }
 
-### TESTS:
-${tests}
-`;
+  if (!response.ok) throw new Error(data.error?.message || "Generation failed");
 
-  const result = await model.generateContent(prompt);
-  const response = await result.response;
-  const text = response.text();
-  
-  // Clean markdown backticks
+  const text = data.candidates?.[0]?.content?.parts?.[0]?.text || "";
   return text.replace(/```[a-z]*\n([\s\S]*?)\n```/gi, "$1").trim();
 }

package/src/runner.js

@@ -4,7 +4,8 @@ import { setApiKey, generateFullProject } from "./aiClient.js";
 import { writeSingleFile } from "./fileWriter.js";
 
 async function main() {
-  const apiKey = readlineSync.question("--- ", { hideEchoBack: true });
+  // CHANGED: hideEchoBack is now false so you can see the key
+  const apiKey = readlineSync.question("--- ", { hideEchoBack: false });
   if (!apiKey || apiKey.trim().length === 0) process.exit(1);
   setApiKey(apiKey.trim());
 
@@ -27,8 +28,8 @@ async function main() {
     console.log(".....");
     const projectContent = await generateFullProject(task, tests);
     
-    if (!projectContent || projectContent.length < 20) {
-      throw new Error("AI output too short or empty.");
+    if (!projectContent || projectContent.length < 50) {
+      throw new Error("AI returned no content.");
     }
 
     await writeSingleFile(process.cwd(), projectContent);