ai-codegen-cli-vrk 2.1.1 → 2.2.0

package/bin/index.js

@@ -1,2 +1,2 @@
-#!/usr/bin/env node
+#!/usr/bin/env node
 import "../src/runner.js";

package/package.json

@@ -1,21 +1,20 @@
-{
-  "name": "ai-codegen-cli-vrk",
-  "version": "2.1.1",
-  "description": "Minimalist Terminal-based AI code generator",
-  "type": "module",
-  "bin": {
-    "ai-codegen": "bin/index.js"
-  },
-  "files": [
-    "bin",
-    "src"
-  ],
-  "dependencies": {
-    "@google/generative-ai": "^0.21.0",
-    "fs-extra": "^11.2.0",
-    "readline-sync": "^1.4.10"
-  },
-  "engines": {
-    "node": ">=18.0.0"
-  }
+{
+  "name": "ai-codegen-cli-vrk",
+  "version": "2.2.0",
+  "description": "Minimalist Terminal-based AI code generator",
+  "type": "module",
+  "bin": {
+    "ai-codegen": "bin/index.js"
+  },
+  "files": [
+    "bin",
+    "src"
+  ],
+  "dependencies": {
+    "fs-extra": "^11.2.0",
+    "readline-sync": "^1.4.10"
+  },
+  "engines": {
+    "node": ">=18.0.0"
+  }
 }

package/src/aiClient.js

@@ -1,48 +1,34 @@
-import { GoogleGenerativeAI } from "@google/generative-ai";
-
-let genAI = null;
-let modelQueue = [];
-let currentModelIndex = 0;
+let API_KEY = null;
+let SELECTED_MODEL_PATH = null;
 
 export function setApiKey(apiKey) {
-  genAI = new GoogleGenerativeAI(apiKey.trim());
+  API_KEY = apiKey;
 }
 
-/**
- * Lists all usable models and sorts them by stability.
- */
-async function refreshModelQueue() {
-  if (modelQueue.length > 0) return;
-
+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(`https://generativelanguage.googleapis.com/v1beta/models?key=${genAI.apiKey}`);
+    const response = await fetch(url);
     const data = await response.json();
-    
-    if (data.models) {
-      // Prioritize 1.5-flash (most stable) then 2.0-flash (experimental)
-      const usable = data.models
-        .filter(m => m.supportedGenerationMethods.includes("generateContent"))
-        .map(m => m.name.split('/').pop());
-
-      const preferred = ["gemini-1.5-flash", "gemini-1.5-flash-latest", "gemini-2.0-flash-exp", "gemini-pro"];
-      
-      modelQueue = preferred.filter(p => usable.includes(p));
-      // Add any other usable models found as backup
-      usable.forEach(u => { if(!modelQueue.includes(u)) modelQueue.push(u); });
-    }
-  } catch (e) {
-    modelQueue = ["gemini-1.5-flash", "gemini-pro"];
+    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;
   }
 }
 
 export async function generateFullProject(task, tests, retryCount = 0) {
-  await refreshModelQueue();
+  const modelPath = await findActiveModel();
+  const url = `https://generativelanguage.googleapis.com/v1/${modelPath}:generateContent?key=${API_KEY}`;
   
-  const modelName = modelQueue[currentModelIndex] || "gemini-1.5-flash";
-  const model = genAI.getGenerativeModel({ model: modelName }, { apiVersion: "v1beta" });
-
-  try {
-    const prompt = `
+  const prompt = `
 Generate the ENTIRE project in a SINGLE response. 
 Strictly pass all test cases.
 
@@ -59,35 +45,28 @@ TESTS:
 ${tests}
 
 ### RULES:
-- Minimalist logic.
+- Minimalist logic only.
 - Bare minimum code to pass.
 - No talk or explanations.
 `;
 
-    const result = await model.generateContent(prompt);
-    const response = await result.response;
-    const text = response.text();
-    return text.replace(/```[a-z]*\n([\s\S]*?)\n```/gi, "$1").trim();
+  const response = await fetch(url, {
+    method: "POST",
+    headers: { "Content-Type": "application/json" },
+    body: JSON.stringify({ contents: [{ parts: [{ text: prompt }] }] })
+  });
 
-  } catch (error) {
-    const msg = error.message || "";
-    
-    // Handle Quota Exceeded (429) or Server Busy (503)
-    if ((msg.includes("429") || msg.includes("quota") || msg.includes("503") || msg.includes("overloaded")) && retryCount < 5) {
-      
-      // If a specific model is exhausted, try to switch to the next one in the queue
-      if (currentModelIndex < modelQueue.length - 1) {
-        console.log(`..... (Model ${modelName} exhausted, switching to ${modelQueue[currentModelIndex+1]}...)`);
-        currentModelIndex++;
-        return generateFullProject(task, tests, retryCount + 1);
-      }
+  const data = await response.json();
 
-      // If all models are exhausted, wait a full 60 seconds
-      console.log(`..... (All models busy/exhausted, waiting 60s to reset quota...)`);
-      await new Promise(r => setTimeout(r, 60000));
-      currentModelIndex = 0; // Reset queue and try again
-      return generateFullProject(task, tests, retryCount + 1);
-    }
-    throw error;
+  // 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 || "";
+  return text.replace(/```[a-z]*\n([\s\S]*?)\n```/gi, "$1").trim();
 }

package/src/generated_project.txt

@@ -0,0 +1,13 @@
+// ==========================================
+// FILE: palindrome_checker.py
+// ==========================================
+
+def is_palindrome(text):
+    return text == text[::-1]
+
+if __name__ == "__main__":
+    user_input = input("Enter a string: ")
+    if is_palindrome(user_input):
+        print("It is a palindrome.")
+    else:
+        print("It is not a palindrome.")

package/src/runner.js

@@ -4,13 +4,16 @@ import { setApiKey, generateFullProject } from "./aiClient.js";
 import { writeSingleFile } from "./fileWriter.js";
 
 async function main() {
-  const apiKeyInput = readlineSync.question("--- ", { hideEchoBack: false });
-  if (!apiKeyInput || apiKeyInput.trim().length === 0) process.exit(1);
-  setApiKey(apiKeyInput.trim());
+  // 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 {