ai-codegen-cli-vrk 2.0.0 → 2.0.1

package/package.json

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

package/src/aiClient.js

@@ -1,43 +1,24 @@
-let API_KEY = null;
-let SELECTED_MODEL = null;
+import { GoogleGenerativeAI } from "@google/generative-ai";
 
-export function setApiKey(apiKey) {
-  API_KEY = apiKey;
-}
+let genAI = null;
+let model = null;
 
-/**
- * Automatically finds the exact model name assigned to your key.
- */
-async function getValidModel() {
-  if (SELECTED_MODEL) return SELECTED_MODEL;
-  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();
-    const found = (data.models || []).find(m => 
-      m.supportedGenerationMethods.includes("generateContent") && 
-      (m.name.includes("flash") || m.name.includes("pro"))
-    );
-    SELECTED_MODEL = found ? found.name : "models/gemini-1.5-flash";
-    return SELECTED_MODEL;
-  } catch {
-    SELECTED_MODEL = "models/gemini-1.5-flash";
-    return SELECTED_MODEL;
-  }
+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" });
 }
 
 export async function generateFullProject(task, tests) {
-  const modelPath = await getValidModel();
-  const url = `https://generativelanguage.googleapis.com/v1/${modelPath}:generateContent?key=${API_KEY}`;
-  
+  if (!genAI) throw new Error("API Key not set");
+
   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, no comments, no console logs.
+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.
 
@@ -61,18 +42,15 @@ ${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();
-  if (!response.ok) throw new Error();
+### OUTPUT:
+Return ONLY the code sections. No talk. No markdown backticks.
+`;
 
-  const text = data.candidates?.[0]?.content?.parts?.[0]?.text || "";
-  // Strip markdown backticks to return clean text
+  const result = await model.generateContent(prompt);
+  const response = await result.response;
+  const text = response.text();
+  
+  // Strip markdown backticks in case AI adds them
   return text.replace(/```[a-z]*\n([\s\S]*?)\n```/gi, "$1").trim();
 }

package/src/runner.js

@@ -4,16 +4,14 @@ import { setApiKey, generateFullProject } from "./aiClient.js";
 import { writeSingleFile } from "./fileWriter.js";
 
 async function main() {
-  // Input: API Key
-  const apiKey = readlineSync.question("--- ", { hideEchoBack: true });
+  // Visible prompt for API key
+  const apiKey = readlineSync.question("--- ");
   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: Test cases (Path or Raw Text)
   const testsInput = readlineSync.question("-- ");
   let tests = "";
   try {
@@ -27,19 +25,18 @@ async function main() {
   }
 
   try {
-    // Silent Indicator
     console.log(".....");
-    
-    // Process Single Request
     const projectContent = await generateFullProject(task, tests);
     
-    // Save to file
+    if (!projectContent || projectContent.length < 20) {
+      throw new Error("AI returned no project content. Check your prompt.");
+    }
+
     await writeSingleFile(process.cwd(), projectContent);
-    
-    // Silent Exit on Success
     process.exit(0);
-  } catch {
-    // Silent Exit on Failure
+  } catch (error) {
+    // Report errors explicitly instead of exiting silently
+    console.error("\n❌ Error:", error.message);
     process.exit(1);
   }
 }