ai-codegen-cli-vrk 1.0.1 → 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.1",
-  "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,110 +1,78 @@
-import Groq from "groq-sdk";
-
-let groqClient = null;
+let API_KEY = null;
+let SELECTED_MODEL = null;
 
 export function setApiKey(apiKey) {
-  groqClient = new Groq({ apiKey });
-}
-
-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,
-    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 solve the task.
-
-FORMAT:
-{
-  "files": [
-    { "path": "file.ext" }
-  ]
-}
-`
-    },
-    {
-      role: "user",
-      content: `TASK:\n${task}\n\nTESTS:\n${tests}`
-    }
-  ]);
-
-  const parsed = JSON.parse(res.choices[0].message.content);
-
-  if (!Array.isArray(parsed.files)) {
-    throw new Error("Invalid file plan from AI");
-  }
-
-  return parsed.files.map(f => f.path);
+  API_KEY = apiKey;
 }
 
-/* ========= PHASE 2: FILE CONTENT ========= */
-export async function generateFileContent(filePath, task, tests) {
+/**
+ * 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 res = await call([
-      {
-        role: "system",
-        content: `
-Generate ONLY the source code for this file:
-
-${filePath}
-
-No explanations.
-No markdown.
-Return RAW CODE ONLY.
-`
-      },
-      {
-        role: "user",
-        content: `TASK:\n${task}\n\nTESTS:\n${tests}`
-      }
-    ]);
-
-    const content = res.choices[0].message.content?.trim();
-
-    if (content && content.length > 0) {
-      return { path: filePath, content };
-    }
+    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 {
-    /* fall through */
+    SELECTED_MODEL = "models/gemini-1.5-flash";
+    return SELECTED_MODEL;
   }
+}
 
-  /* ========= GENERIC FALLBACK ========= */
-
-  if (filePath.endsWith(".py")) {
-    return {
-      path: filePath,
-      content: `def main():
-    pass
-
-if __name__ == "__main__":
-    main()
-`
-    };
-  }
+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.
+
+### 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.
+
+### FORMATTING (STRICT):
+Every file MUST start with this EXACT header style:
+
+/**
+ * ==========================================
+ * FILE: package.json (Reference)
+ * ==========================================
+ * [List Dependencies here]
+ */
+
+// ==========================================
+// FILE: path/to/filename.js
+// ==========================================
+[CODE HERE]
+
+### TASK:
+${task}
+
+### TESTS:
+${tests}
+`;
+
+  const response = await fetch(url, {
+    method: "POST",
+    headers: { "Content-Type": "application/json" },
+    body: JSON.stringify({ contents: [{ parts: [{ text: prompt }] }] })
+  });
 
-  if (filePath.endsWith(".js")) {
-    return {
-      path: filePath,
-      content: `module.exports = {};`
-    };
-  }
+  const data = await response.json();
+  if (!response.ok) throw new Error();
 
-  return {
-    path: filePath,
-    content: ""
-  };
-}
+  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,63 +1,47 @@
 import readlineSync from "readline-sync";
 import fs from "fs";
-
-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) {
-  process.exit(1);
-}
-
-setApiKey(apiKey);
-
-/* -------- TASK (-) -------- */
-const task = readlineSync.question("- ");
-
-if (!task || task.trim().length === 0) {
-  process.exit(1);
-}
-
-/* -------- TESTS (--) OPTIONAL -------- */
-const testsInput = readlineSync.question("-- ");
-let tests = "";
-
-try {
-  if (
-    testsInput &&
-    (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;
-}
-
-/* ---- SILENT FALLBACK ---- */
-if (!tests || tests.trim().length === 0) {
-  tests = "No tests provided. Generate a simple correct implementation.";
-}
 
-(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);
+    // 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,34 +0,0 @@
-export const systemPrompt = `
-You are solving an automated coding exam.
-
-ABSOLUTE PRIORITY:
-The given test cases are the ONLY specification.
-
-RULES:
-- Generate the MINIMUM code needed
-- No extra logic
-- No comments
-- No console logs
-- No refactoring
-- No optimization
-
-ARCHITECTURE:
-- Create ONLY required files
-- Names must match tests EXACTLY
-- Error messages must match EXACTLY
-
-OUTPUT:
-- ONLY valid JSON
-- No markdown
-- No explanations
-
-FORMAT:
-{
-  "files": [
-    {
-      "path": "relative/path/file.js",
-      "content": "exact code"
-    }
-  ]
-}
-`;