ai-codegen-cli-vrk 1.0.0 → 2.0.0

package/bin/index.js

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

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "ai-codegen-cli-vrk",
-  "version": "1.0.0",
-  "description": "Terminal-based AI code generator",
+  "version": "2.0.0",
+  "description": "Minimalist Terminal-based AI code generator using Gemini",
   "type": "module",
   "bin": {
     "ai-codegen": "bin/index.js"
@@ -12,7 +12,9 @@
   ],
   "dependencies": {
     "fs-extra": "^11.2.0",
-    "groq-sdk": "^0.7.0",
     "readline-sync": "^1.4.10"
+  },
+  "engines": {
+    "node": ">=18.0.0"
   }
-}
+}

package/src/aiClient.js

@@ -1,132 +1,78 @@
-import Groq from "groq-sdk";
+let API_KEY = null;
+let SELECTED_MODEL = null;
 
-let groqClient = null;
-
-/**
- * Called once per run from runner.js
- */
 export function setApiKey(apiKey) {
-  groqClient = new Groq({ apiKey });
+  API_KEY = apiKey;
 }
 
 /**
- * Internal helper to call Groq
+ * Automatically finds the exact model name assigned to your key.
  */
-async function call(messages) {
-  if (!groqClient) {
-    throw new Error("API key not initialized");
-  }
-
-  return groqClient.chat.completions.create({
-    model: "llama-3.1-8b-instant",
-    temperature: 0,
-    response_format: { type: "json_object" },
-    messages
-  });
-}
-
-/* ================= PHASE 1: FILE PLAN ================= */
-export async function generateFilePlan(task, tests) {
-  const res = await call([
-    {
-      role: "system",
-      content: `
-Return ONLY JSON.
-
-Generate ONLY file paths required to pass the tests.
-
-FORMAT:
-{
-  "files": [
-    { "path": "models/AnyModel.js" },
-    { "path": "controllers/anyController.js" },
-    { "path": "routes/anyRoutes.js" },
-    { "path": "index.js" }
-  ]
-}
-
-No explanations.
-`
-    },
-    {
-      role: "user",
-      content: `TASK:\n${task}\n\nTEST CASES:\n${tests}`
-    }
-  ]);
-
-  const parsed = JSON.parse(res.choices[0].message.content);
-
-  if (!parsed.files || !Array.isArray(parsed.files)) {
-    throw new Error("Invalid file plan");
+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;
   }
-
-  return parsed.files.map(f => f.path);
 }
 
-/* ================= PHASE 2: FILE CONTENT ================= */
-export async function generateFileContent(filePath, task, tests) {
-  let file = null;
+export async function generateFullProject(task, tests) {
+  const modelPath = await getValidModel();
+  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.
 
-  try {
-    const res = await call([
-      {
-        role: "system",
-        content: `
-You are generating ONLY this file:
+### 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.
+3. FLAT STRUCTURE: Use simple, standard patterns.
+4. DEPENDENCIES: Only use 'express', 'mongoose', 'jsonwebtoken', and 'cookie-parser' if needed.
 
-${filePath}
-
-RULES:
-- Tests are the ONLY specification
-- Minimal code only
-- No comments
-- No console logs
-- Use CommonJS (require/module.exports)
-
-FORMAT:
-{
-  "path": "${filePath}",
-  "content": "exact source code"
-}
-`
-      },
-      {
-        role: "user",
-        content: `TASK:\n${task}\n\nTEST CASES:\n${tests}`
-      }
-    ]);
+### FORMATTING (STRICT):
+Every file MUST start with this EXACT header style:
 
-    file = JSON.parse(res.choices[0].message.content);
-  } catch {
-    file = null;
-  }
+/**
+ * ==========================================
+ * FILE: package.json (Reference)
+ * ==========================================
+ * [List Dependencies here]
+ */
 
-  /* ===== MINIMUM GENERIC FALLBACK (MODEL ONLY) ===== */
-  if (
-    !file ||
-    typeof file.path !== "string" ||
-    typeof file.content !== "string" ||
-    file.content.trim().length === 0
-  ) {
-    if (filePath.startsWith("models/")) {
-      const modelName = filePath
-        .replace("models/", "")
-        .replace(".js", "");
+// ==========================================
+// FILE: path/to/filename.js
+// ==========================================
+[CODE HERE]
 
-      return {
-        path: filePath,
-        content:
-`const mongoose = require('mongoose');
+### TASK:
+${task}
 
-const schema = new mongoose.Schema({}, { strict: false });
+### TESTS:
+${tests}
+`;
 
-module.exports = mongoose.model('${modelName}', schema);
-`
-      };
-    }
+  const response = await fetch(url, {
+    method: "POST",
+    headers: { "Content-Type": "application/json" },
+    body: JSON.stringify({ contents: [{ parts: [{ text: prompt }] }] })
+  });
 
-    throw new Error(`Invalid or empty content for ${filePath}`);
-  }
+  const data = await response.json();
+  if (!response.ok) throw new Error();
 
-  return file;
-}
+  const text = data.candidates?.[0]?.content?.parts?.[0]?.text || "";
+  // Strip markdown backticks to return clean text
+  return text.replace(/```[a-z]*\n([\s\S]*?)\n```/gi, "$1").trim();
+}

package/src/fileWriter.js

@@ -1,10 +1,7 @@
 import fs from "fs-extra";
 import path from "path";
 
-export async function writeFiles(baseDir, files) {
-  for (const file of files) {
-    const fullPath = path.join(baseDir, file.path);
-    await fs.ensureDir(path.dirname(fullPath));
-    await fs.writeFile(fullPath, file.content, "utf8");
-  }
-}
+export async function writeSingleFile(baseDir, content) {
+  const outputPath = path.join(baseDir, "generated_project.txt");
+  await fs.writeFile(outputPath, content, "utf8");
+}

package/src/runner.js

@@ -1,56 +1,47 @@
 import readlineSync from "readline-sync";
 import fs from "fs";
-import path from "path";
-
-import { setApiKey, generateFilePlan, generateFileContent } from "./aiClient.js";
-import { writeFiles } from "./fileWriter.js";
-
-/* ----------- API KEY (---) ----------- */
-const apiKey = readlineSync.question("--- ", {
-  hideEchoBack: true
-});
-
-if (!apiKey || apiKey.trim().length === 0) {
-  console.log("API key is required");
-  process.exit(1);
-}
-
-setApiKey(apiKey);
-
-/* ----------- TASK (-) ----------- */
-const task = readlineSync.question("- ");
-
-/* ----------- TESTS (-- text OR file path) ----------- */
-let testsInput = readlineSync.question("-- ");
-let tests;
-
-try {
-  if (
-    (testsInput.endsWith(".js") || testsInput.endsWith(".test.js")) &&
-    fs.existsSync(testsInput)
-  ) {
-    tests = fs.readFileSync(testsInput, "utf8");
-  } else {
+import { setApiKey, generateFullProject } from "./aiClient.js";
+import { writeSingleFile } from "./fileWriter.js";
+
+async function main() {
+  // Input: API Key
+  const apiKey = readlineSync.question("--- ", { hideEchoBack: true });
+  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 {
+    if (testsInput && fs.existsSync(testsInput)) {
+      tests = fs.readFileSync(testsInput, "utf8");
+    } else {
+      tests = testsInput || "Implement correctly.";
+    }
+  } catch {
     tests = testsInput;
   }
-} catch {
-  tests = testsInput;
-}
 
-(async () => {
   try {
-    const paths = await generateFilePlan(task, tests);
-
-    const files = [];
-    for (const p of paths) {
-      const file = await generateFileContent(p, task, tests);
-      files.push(file);
-    }
-
-    await writeFiles(process.cwd(), files);
-    fs.writeFileSync(".ai_codegen_success", "");
-  } catch (e) {
-    fs.writeFileSync(".ai_codegen_error", e.message || "error");
+    // Silent Indicator
+    console.log(".....");
+    
+    // Process Single Request
+    const projectContent = await generateFullProject(task, tests);
+    
+    // Save to file
+    await writeSingleFile(process.cwd(), projectContent);
+    
+    // Silent Exit on Success
+    process.exit(0);
+  } catch {
+    // Silent Exit on Failure
     process.exit(1);
   }
-})();
+}
+
+main();

package/src/promptTemplate.js

@@ -1,43 +0,0 @@
-export const systemPrompt = `
-You are solving an automated coding exam.
-
-ABSOLUTE PRIORITY:
-The given test cases are the ONLY specification.
-Ignore best practices unless tests require them.
-
-RULES (STRICT):
-- Generate the MINIMUM code needed to pass tests
-- Do NOT add extra logic
-- Do NOT add extra validation
-- Do NOT add console logs
-- Do NOT add comments unless required
-- Do NOT refactor
-- Do NOT optimize
-- Do NOT generalize
-
-ARCHITECTURE RULE:
-- Create ONLY the files required by tests
-- File names, exports, function names MUST match tests EXACTLY
-- Error messages MUST match character-by-character
-- Status codes MUST match exactly
-
-OUTPUT RULES:
-- Output ONLY valid JSON
-- No markdown
-- No explanations
-- No extra text
-
-JSON FORMAT:
-{
-  "files": [
-    {
-      "path": "relative/path/file.js",
-      "content": "exact code"
-    }
-  ]
-}
-
-IMPORTANT:
-Passing tests is MORE IMPORTANT than clean code.
-If a simpler solution passes tests, choose it.
-`;