ai-codegen-cli-vrk 2.0.2 → 2.0.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-codegen-cli-vrk",
-  "version": "2.0.2",
+  "version": "2.0.3",
   "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,50 +1,54 @@
-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
+ * Automatically finds a model that is active for your account.
+ * This prevents the "404 Not Found" error.
  */
-async function getWorkingModel() {
-  if (SELECTED_MODEL_NAME) return genAI.getGenerativeModel({ model: SELECTED_MODEL_NAME });
+async function findActiveModel() {
+  if (SELECTED_MODEL_PATH) return SELECTED_MODEL_PATH;
 
-  const candidates = ["gemini-1.5-flash", "gemini-1.5-flash-latest", "gemini-pro"];
+  // Use the stable v1 API to list models
+  const url = `https://generativelanguage.googleapis.com/v1/models?key=${API_KEY}`;
   
-  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
+  try {
+    const response = await fetch(url);
+    const data = await response.json();
+
+    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
   }
-  throw new Error("No compatible Gemini models found for your API key.");
+
+  SELECTED_MODEL_PATH = "models/gemini-1.5-flash";
+  return SELECTED_MODEL_PATH;
 }
 
 export async function generateFullProject(task, tests) {
-  if (!genAI) throw new Error("API Key not initialized");
-
-  const model = await getWorkingModel();
-
+  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 = `
-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.
+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:
@@ -66,12 +70,24 @@ ${task}
 
 ### TESTS:
 ${tests}
+
+### 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();
-  
-  // Clean markdown backticks
+  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

@@ -27,8 +27,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 empty content. Check your task/tests.");
     }
 
     await writeSingleFile(process.cwd(), projectContent);