ai-codegen-cli-vrk 2.0.1 → 2.0.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-codegen-cli-vrk",
-  "version": "2.0.1",
+  "version": "2.0.3",
   "description": "Minimalist Terminal-based AI code generator",
   "type": "module",
   "bin": {
@@ -11,7 +11,6 @@
     "src"
   ],
   "dependencies": {
-    "@google/generative-ai": "^0.21.0",
     "fs-extra": "^11.2.0",
     "readline-sync": "^1.4.10"
   },

package/src/aiClient.js

@@ -1,26 +1,54 @@
-import { GoogleGenerativeAI } from "@google/generative-ai";
-
-let genAI = null;
-let model = null;
+let API_KEY = null;
+let SELECTED_MODEL_PATH = 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" });
+  API_KEY = apiKey;
 }
 
-export async function generateFullProject(task, tests) {
-  if (!genAI) throw new Error("API Key not set");
+/**
+ * Automatically finds a model that is active for your account.
+ * This prevents the "404 Not Found" error.
+ */
+async function findActiveModel() {
+  if (SELECTED_MODEL_PATH) return SELECTED_MODEL_PATH;
 
-  const prompt = `
-You are an expert automated coding exam solver. 
-Generate the ENTIRE project in a SINGLE continuous text response.
+  // Use the stable v1 API to list models
+  const url = `https://generativelanguage.googleapis.com/v1/models?key=${API_KEY}`;
+  
+  try {
+    const response = await fetch(url);
+    const data = await response.json();
 
-### 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.
+    if (!response.ok) throw new Error(data.error?.message || "Invalid API Key");
+
+    // Find the first model that supports content generation
+    const models = data.models || [];
+    const match = models.find(m => 
+      m.supportedGenerationMethods.includes("generateContent") && 
+      (m.name.includes("flash") || m.name.includes("pro"))
+    );
+
+    if (match) {
+      SELECTED_MODEL_PATH = match.name; // e.g. "models/gemini-1.5-flash"
+      return SELECTED_MODEL_PATH;
+    }
+  } catch (err) {
+    // If listing fails, fall back to a standard guess
+  }
+
+  SELECTED_MODEL_PATH = "models/gemini-1.5-flash";
+  return SELECTED_MODEL_PATH;
+}
+
+export async function generateFullProject(task, tests) {
+  const modelPath = await findActiveModel();
+  
+  // Use v1 (stable) to avoid v1beta 404 issues
+  const url = `https://generativelanguage.googleapis.com/v1/${modelPath}:generateContent?key=${API_KEY}`;
+  
+  const prompt = `
+Generate the ENTIRE coding project in a SINGLE continuous text response.
+Strictly pass all test cases.
 
 ### FORMATTING (STRICT):
 Every file MUST start with this EXACT header style:
@@ -43,14 +71,23 @@ ${task}
 ### TESTS:
 ${tests}
 
-### OUTPUT:
-Return ONLY the code sections. No talk. No markdown backticks.
+### RULES:
+- Minimalist logic. No extra logic.
+- Bare minimum code to pass.
+- No explanations.
 `;
 
-  const result = await model.generateContent(prompt);
-  const response = await result.response;
-  const text = response.text();
-  
-  // Strip markdown backticks in case AI adds them
+  const response = await fetch(url, {
+    method: "POST",
+    headers: { "Content-Type": "application/json" },
+    body: JSON.stringify({
+      contents: [{ parts: [{ text: prompt }] }]
+    })
+  });
+
+  const data = await response.json();
+  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/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());
 
@@ -28,14 +27,13 @@ async function main() {
     console.log(".....");
     const projectContent = await generateFullProject(task, tests);
     
-    if (!projectContent || projectContent.length < 20) {
-      throw new Error("AI returned no project content. Check your prompt.");
+    if (!projectContent || projectContent.length < 50) {
+      throw new Error("AI returned empty content. Check your task/tests.");
     }
 
     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);
   }