@atharvaspatil5/aish 1.0.1 → 1.0.2

package/README.md

@@ -1,6 +1,20 @@
 # AI.sh
 
-AI.sh is a CLI tool that converts natural language into safe shell commands.
+AI.sh is a CLI tool that converts natural language into safe shell commands, powered by Groq LLM.
+
+## Install
+
+```bash
+npm install -g @atharvaspatil5/aish
+```
+
+## Setup
+
+Get a free API key at [console.groq.com](https://console.groq.com), then create a `.env` file in your working directory:
+
+```env
+GROQ_API_KEY=your_groq_api_key
+```
 
 ## Usage
 
@@ -37,22 +51,9 @@ ai "delete everything"
 
 ## Safety
 
-* Blocks critical system-level commands
 * Warns on risky operations (like file deletion)
 * Requires confirmation before execution
-
-## Setup
-
-```bash
-npm install
-npm link
-```
-
-Create a `.env` file:
-
-```env
-GROQ_API_KEY=your_api_key
-```
+* Dangerous commands always prompt, even with `-y`
 
 ## Structure
 
@@ -60,11 +61,10 @@ GROQ_API_KEY=your_api_key
 bin/        CLI entry
 src/        core logic
   llm.js
-  validator.js
   executor.js
   index.js
 ```
 
 ## Notes
 
-This is a local CLI tool. Use carefully when executing destructive commands.
+Use carefully when executing destructive commands. The `.env` file must be present in the directory where you run `ai`.

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@atharvaspatil5/aish",
-  "version": "1.0.1",
+  "version": "1.0.2",
   "description": "AI-powered CLI that converts natural language to shell commands",
   "main": "index.js",
   "scripts": {
@@ -10,7 +10,7 @@
   "author": "athPATIL247",
   "license": "ISC",
   "bin": {
-    "ai": "./bin/ai.js"
+    "ai": "bin/ai.js"
   },
   "type": "module",
   "dependencies": {

package/src/llm.js

@@ -1,38 +1,24 @@
 import Groq from "groq-sdk";
-import { readFileSync } from 'fs';
-import { fileURLToPath } from 'url';
-import { dirname, join } from 'path';
-
-const __filename = fileURLToPath(import.meta.url);
-const __dirname = dirname(__filename);
-
-const envPath = join(__dirname, '..', '.env');
-const envContent = readFileSync(envPath, 'utf8');
-const envLines = envContent.split('\n');
-for (const line of envLines) {
-    const [key, ...valueParts] = line.split('=');
-    if (key && valueParts.length) {
-        process.env[key.trim()] = valueParts.join('=').trim();
-    }
-}
+import 'dotenv/config';
 
 const groq = new Groq({
     apiKey: process.env.GROQ_API_KEY,
 });
 
-export async function getCommand(query){
+export async function getCommand(query) {
     const response = await groq.chat.completions.create({
         model: "llama-3.1-8b-instant",
         messages: [
             {
                 role: "system",
-                content: `You are a Linux shell expert. Convert user instructions into a single shell command and assess its risk level.
+                content: `You are a shell command generator. Your ONLY output must be a single raw JSON object. No markdown, no code blocks, no backticks, no explanations, no extra text — just the JSON.
 
-Output ONLY a JSON object with exactly these fields:
-- command: the shell command (string)
-- risk_level: "safe", "warning", or "danger" (string)
+The JSON must have exactly these two fields:
+- "command": the shell command as a string
+- "risk_level": one of "safe", "warning", or "danger"
 
-Do not output any other text, explanations, code blocks, or notes. Just the raw JSON.`,
+Example output:
+{"command":"ls -lh","risk_level":"safe"}`,
             },
             {
                 role: "user",
@@ -42,22 +28,25 @@ Do not output any other text, explanations, code blocks, or notes. Just the raw
     });
 
     const content = response.choices[0].message.content.trim();
-    // Remove markdown code blocks if present
+
+    // Strip markdown code blocks if model misbehaves
     let jsonString = content;
-    if (jsonString.startsWith('```json')) {
-        jsonString = jsonString.replace(/^```json\s*/, '').replace(/\s*```$/, '');
-    } else if (jsonString.startsWith('```')) {
-        jsonString = jsonString.replace(/^```\s*/, '').replace(/\s*```$/, '');
+    if (jsonString.startsWith('```')) {
+        jsonString = jsonString.replace(/^```[a-z]*\s*/i, '').replace(/\s*```$/, '');
     }
+
+    // Extract first JSON object if there's extra text
+    const match = jsonString.match(/\{[\s\S]*?\}/);
+    if (match) {
+        jsonString = match[0];
+    }
+
     try {
-        const result = JSON.parse(jsonString);
-        return result;
+        return JSON.parse(jsonString);
     } catch (e) {
-        // Fallback: assume safe if parsing fails
         return {
             command: content,
             risk_level: "safe",
-            reason: "Unable to assess risk"
         };
     }
 }