shivam-cli-generator 1.0.3 → 1.0.4

package/.env

@@ -0,0 +1 @@
+GROQ_API_KEY=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

package/ai.js

@@ -1,47 +1,83 @@
+import dotenv from 'dotenv';
+import { fileURLToPath } from 'url';
+import { dirname, join } from 'path';
+
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = dirname(__filename);
+
+// Load .env from the package directory, not the current working directory
+dotenv.config({ path: join(__dirname, '.env') });
 import Groq from "groq-sdk";
 import fs from "fs";
 import readline from "readline-sync";
-
-// ✅ Get API Key from system environment only
 const apiKey = process.env.GROQ_API_KEY;
 
 if (!apiKey) {
-  console.error("❌ GROQ_API_KEY is missing.\n👉 Set it using:\n\nsetx GROQ_API_KEY \"your_groq_api_key_here\"\n");
+  console.error("❌ GROQ_API_KEY is missing.\n👉 Please check your .env file or system environment variables.\n");
   process.exit(1);
 }
 
 // ✅ Correct Groq client initialization
 const client = new Groq({ apiKey });
 
-const inputURL = process.argv[2] || readline.question("Enter URL: ");
+// ✅ Use the inputURL if provided
+const inputURL = process.argv[2];
+
+if (inputURL) {
+  console.log("\n🔗 Context URL:", inputURL);
+}
 
-console.log("\n🔗 URL received:", inputURL);
 console.log("💡 Ask anything to AI. Type 'exit' to quit.\n");
 
 async function main() {
   while (true) {
     const question = readline.question("AI > ");
 
-    if (question.toLowerCase() === "exit") break;
+    if (question.toLowerCase() === "exit") {
+      console.log("👋 Goodbye!");
+      break;
+    }
 
     console.log("⏳ Generating...");
 
-    const response = await client.chat.completions.create({
-      model: "llama-3.1-8b-instant",
-      messages: [
-        { role: "system", content: "You generate clean code. Only code or required output." },
-        { role: "user", content: question }
-      ]
-    });
+    try {
+      let systemPrompt = "You generate clean code. Only code or required output.";
+      let userContent = question;
+
+      if (inputURL) {
+        userContent = `Context URL: ${inputURL}\n\nQuestion: ${question}`;
+      }
+
+      const response = await client.chat.completions.create({
+        model: "llama-3.1-8b-instant",
+        messages: [
+          { role: "system", content: systemPrompt },
+          { role: "user", content: userContent }
+        ]
+      });
+
+      const output = response.choices[0]?.message?.content?.trim();
 
-    const output = response.choices[0].message.content.trim();
+      if (output) {
+        if (!fs.existsSync("generated")) fs.mkdirSync("generated");
 
-    if (!fs.existsSync("generated")) fs.mkdirSync("generated");
+        const fileName = `generated/output_${Date.now()}.txt`;
+        fs.writeFileSync(fileName, output);
 
-    const fileName = `generated/output_${Date.now()}.txt`;
-    fs.writeFileSync(fileName, output);
+        console.log(`\n✅ File generated: ${fileName}\n`);
+        console.log("-----------------------------------");
+        console.log(output); // Also print to console for immediate feedback
+        console.log("-----------------------------------\n");
+      } else {
+        console.log("⚠️ No content received from AI.");
+      }
 
-    console.log(`\n✅ File generated: ${fileName}\n`);
+    } catch (error) {
+      console.error("\n❌ Error communicating with AI:", error.message);
+      if (error.code === 'ECONNREFUSED' || error.code === 'ENOTFOUND') {
+        console.error("👉 Check your internet connection.\n");
+      }
+    }
   }
 }
 

package/cli.js

@@ -1,3 +1,3 @@
-#!/usr/bin/env node
+#!/usr/bin/env node
 
 import './ai.js';

package/package.json

@@ -1,12 +1,12 @@
 {
   "name": "shivam-cli-generator",
-  "version": "1.0.3",
+  "version": "1.0.4",
   "description": "",
   "main": "ai.js",
   "type": "module",
-   "bin": {
-  "shivam-ai": "./cli.js"
-},
+  "bin": {
+    "shivam-ai": "./cli.js"
+  },
   "scripts": {
     "start": "node ai.js"
   },
@@ -16,9 +16,6 @@
   "dependencies": {
     "dotenv": "^17.2.3",
     "groq-sdk": "^0.37.0",
-    "openai": "^6.16.0",
     "readline-sync": "^1.4.10"
   }
- 
-
 }