ai-codegen-cli-vrk 1.0.0 → 1.0.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-codegen-cli-vrk",
-  "version": "1.0.0",
+  "version": "1.0.1",
   "description": "Terminal-based AI code generator",
   "type": "module",
   "bin": {

package/src/aiClient.js

@@ -2,16 +2,10 @@ import Groq from "groq-sdk";
 
 let groqClient = null;
 
-/**
- * Called once per run from runner.js
- */
 export function setApiKey(apiKey) {
   groqClient = new Groq({ apiKey });
 }
 
-/**
- * Internal helper to call Groq
- */
 async function call(messages) {
   if (!groqClient) {
     throw new Error("API key not initialized");
@@ -20,12 +14,11 @@ async function call(messages) {
   return groqClient.chat.completions.create({
     model: "llama-3.1-8b-instant",
     temperature: 0,
-    response_format: { type: "json_object" },
     messages
   });
 }
 
-/* ================= PHASE 1: FILE PLAN ================= */
+/* ========= PHASE 1: FILE PLAN ========= */
 export async function generateFilePlan(task, tests) {
   const res = await call([
     {
@@ -33,100 +26,85 @@ export async function generateFilePlan(task, tests) {
       content: `
 Return ONLY JSON.
 
-Generate ONLY file paths required to pass the tests.
+Generate ONLY file paths required to solve the task.
 
 FORMAT:
 {
   "files": [
-    { "path": "models/AnyModel.js" },
-    { "path": "controllers/anyController.js" },
-    { "path": "routes/anyRoutes.js" },
-    { "path": "index.js" }
+    { "path": "file.ext" }
   ]
 }
-
-No explanations.
 `
     },
     {
       role: "user",
-      content: `TASK:\n${task}\n\nTEST CASES:\n${tests}`
+      content: `TASK:\n${task}\n\nTESTS:\n${tests}`
     }
   ]);
 
   const parsed = JSON.parse(res.choices[0].message.content);
 
-  if (!parsed.files || !Array.isArray(parsed.files)) {
-    throw new Error("Invalid file plan");
+  if (!Array.isArray(parsed.files)) {
+    throw new Error("Invalid file plan from AI");
   }
 
   return parsed.files.map(f => f.path);
 }
 
-/* ================= PHASE 2: FILE CONTENT ================= */
+/* ========= PHASE 2: FILE CONTENT ========= */
 export async function generateFileContent(filePath, task, tests) {
-  let file = null;
-
   try {
     const res = await call([
       {
         role: "system",
         content: `
-You are generating ONLY this file:
+Generate ONLY the source code for this file:
 
 ${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"
-}
+No explanations.
+No markdown.
+Return RAW CODE ONLY.
 `
       },
       {
         role: "user",
-        content: `TASK:\n${task}\n\nTEST CASES:\n${tests}`
+        content: `TASK:\n${task}\n\nTESTS:\n${tests}`
       }
     ]);
 
-    file = JSON.parse(res.choices[0].message.content);
+    const content = res.choices[0].message.content?.trim();
+
+    if (content && content.length > 0) {
+      return { path: filePath, content };
+    }
   } catch {
-    file = null;
+    /* fall through */
   }
 
-  /* ===== 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", "");
-
-      return {
-        path: filePath,
-        content:
-`const mongoose = require('mongoose');
-
-const schema = new mongoose.Schema({}, { strict: false });
-
-module.exports = mongoose.model('${modelName}', schema);
+  /* ========= GENERIC FALLBACK ========= */
+
+  if (filePath.endsWith(".py")) {
+    return {
+      path: filePath,
+      content: `def main():
+    pass
+
+if __name__ == "__main__":
+    main()
 `
-      };
-    }
+    };
+  }
 
-    throw new Error(`Invalid or empty content for ${filePath}`);
+  if (filePath.endsWith(".js")) {
+    return {
+      path: filePath,
+      content: `module.exports = {};`
+    };
   }
 
-  return file;
+  return {
+    path: filePath,
+    content: ""
+  };
 }

package/src/promptTemplate.js

@@ -3,31 +3,26 @@ 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
+RULES:
+- Generate the MINIMUM code needed
+- No extra logic
+- No comments
+- No console logs
+- No refactoring
+- No optimization
 
-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
+ARCHITECTURE:
+- Create ONLY required files
+- Names must match tests EXACTLY
+- Error messages must match EXACTLY
 
-OUTPUT RULES:
-- Output ONLY valid JSON
+OUTPUT:
+- ONLY valid JSON
 - No markdown
 - No explanations
-- No extra text
 
-JSON FORMAT:
+FORMAT:
 {
   "files": [
     {
@@ -36,8 +31,4 @@ JSON FORMAT:
     }
   ]
 }
-
-IMPORTANT:
-Passing tests is MORE IMPORTANT than clean code.
-If a simpler solution passes tests, choose it.
 `;

package/src/runner.js

@@ -1,31 +1,34 @@
 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 (---) ----------- */
+/* -------- 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 (-) ----------- */
+/* -------- TASK (-) -------- */
 const task = readlineSync.question("- ");
 
-/* ----------- TESTS (-- text OR file path) ----------- */
-let testsInput = readlineSync.question("-- ");
-let tests;
+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)
   ) {
@@ -37,6 +40,11 @@ try {
   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);
@@ -48,9 +56,8 @@ try {
     }
 
     await writeFiles(process.cwd(), files);
-    fs.writeFileSync(".ai_codegen_success", "");
-  } catch (e) {
-    fs.writeFileSync(".ai_codegen_error", e.message || "error");
+    process.exit(0);
+  } catch {
     process.exit(1);
   }
 })();