ai-codegen-cli-vrk 2.1.0 → 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.0",
-  "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,52 +1,34 @@
-import { GoogleGenerativeAI } from "@google/generative-ai";
-
-let genAI = null;
-let activeModel = null;
+let API_KEY = null;
+let SELECTED_MODEL_PATH = null;
 
 export function setApiKey(apiKey) {
-  genAI = new GoogleGenerativeAI(apiKey.trim());
+  API_KEY = apiKey;
 }
 
-/**
- * Ultimate discovery: Asks Google for a list of working models for your key.
- */
-async function getWorkingModel() {
-  if (activeModel) return activeModel;
-
+async function findActiveModel() {
+  if (SELECTED_MODEL_PATH) return SELECTED_MODEL_PATH;
+  const url = `https://generativelanguage.googleapis.com/v1/models?key=${API_KEY}`;
   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 response = await fetch(url);
     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
+    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;
   }
-
-  // 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) {
-  try {
-    const model = await getWorkingModel();
-    
-    const prompt = `
+  const modelPath = await findActiveModel();
+  const url = `https://generativelanguage.googleapis.com/v1/${modelPath}:generateContent?key=${API_KEY}`;
+  
+  const prompt = `
 Generate the ENTIRE project in a SINGLE response. 
 Strictly pass all test cases.
 
@@ -63,24 +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) {
-    // 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;
+  const data = await response.json();
+
+  // 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 {