icmd-cli 0.1.0__py3-none-any.whl

icmd/executor.py

@@ -0,0 +1,15 @@
+import subprocess
+
+def execute(command):
+    try:
+        result = subprocess.run(
+            command.split(),
+            capture_output=True,
+            text=True,
+            timeout=5
+        )
+        
+        return result.stdout if result.returncode == 0 else result.stderr
+    
+    except Exception as e:
+        return str(e)

icmd/llm.py

@@ -0,0 +1,53 @@
+import requests
+import subprocess
+
+MODEL = "qwen2.5-coder"
+OLLAMA_URL = 'http://localhost:11434'
+
+def ensure_ollama():
+    try:
+        requests.get(OLLAMA_URL)
+        res = requests.get(f"{OLLAMA_URL}/api/tags").json()
+        models = [m["name"] for m in res.get("models",[])]
+    
+        if MODEL not in models:
+            print(f"Pulling model: {MODEL}...")
+            subprocess.run(["ollama", "pull", MODEL])
+    except:
+        print("Ollama is not installed or running.")
+        print("Install from: https://ollama.com")
+        exit()
+    
+
+def llm(input):
+    ensure_ollama()
+    PROMPT = f"""
+        You are a advanced linux terminal assistant.
+        For every user query you will return corresponding linux bash command as response.
+        You only return commands and should not explain using comments.
+
+        Examples:
+        query: "list me all the files"
+        reponse: "ls"
+
+        query: "print the working directory"
+        response: "pwd"
+        
+        (MUST!!!) Ensure the command does not involve any critical system operations
+        - Deleting a file/folder
+        - Accessing root user
+        - Accessing important system files, etc.
+        If so return "Cannot process this request" as response
+        
+        Now return bash command for this query with out any quotes:
+        '{input}'
+    """
+
+    data = {
+        "model": "qwen2.5-coder",
+        "prompt": PROMPT,
+        "stream": False
+    }
+    
+    response = requests.post(f"{OLLAMA_URL}/api/tags",json=data)
+    return response.json()["response"]

icmd/main.py

@@ -0,0 +1,23 @@
+import sys
+from icmd.llm import llm
+from icmd.executor import execute
+from icmd.safety import is_dangerous
+from icmd.utils import clean
+
+def run():
+    if len(sys.argv) < 2:
+        print("Usage: icmd \"your query\"")
+        return
+    query = " ".join(sys.argv[1:])
+    command = llm(query)
+    cleaned = clean(command)
+    if cleaned.lower() == "cannot process this request":
+        print("cannot process this request")
+        return
+    if is_dangerous(cleaned):
+        print("Unsafe command to execute")
+        return
+    result = execute(cleaned)
+    print(result)
+        
+

icmd/safety.py

@@ -0,0 +1,19 @@
+CRITICAL_COMMANDS = [
+    "rm", "rmdir", "mv", "dd", "mkfs",
+    "shutdown", "reboot", "init", "poweroff",
+    "kill", "killall", "chmod", "chown",
+    "sudo", "su", "mount", "umount",
+    "wget", "curl", "scp", "ssh"
+]
+
+DANGEROUS_SYMBOLS = [";", "|", "&", ">", ">>"]
+
+
+def is_dangerous(command):
+    parts = command.strip().split()
+    if parts[0] in CRITICAL_COMMANDS:
+        return True
+    for symbol in DANGEROUS_SYMBOLS:
+        if symbol in command:
+            return True
+    return False

icmd/utils.py

@@ -0,0 +1,2 @@
+def clean(command):
+    return command.replace("```bash","").replace("```","").strip()

icmd_cli-0.1.0.dist-info/METADATA

@@ -0,0 +1,17 @@
+Metadata-Version: 2.4
+Name: icmd-cli
+Version: 0.1.0
+Summary: AI-powered command line tool using local LLM
+Author: Deepak Karthick
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+Requires-Dist: requests
+
+# iCMD
+
+An AI-powered command line tool using local LLM (qwen2.5-coder).
+
+## Installation
+
+```bash
+pip install icmd

icmd_cli-0.1.0.dist-info/RECORD

@@ -0,0 +1,11 @@
+icmd/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+icmd/executor.py,sha256=9zLIffCqRECiDIux_tDredCbLaUP5i0HS3hUX4mkS30,336
+icmd/llm.py,sha256=zGuq_MesSNkv7BR9v6fpSCITH5Ll9m0IGw0iBhdwX7M,1559
+icmd/main.py,sha256=xPWlEiOAm8wYxRstrH7EgE41b-RM2SszALMq85TNhSk,585
+icmd/safety.py,sha256=QTfaBU25PDCQJjYaGNJZLSwSn4V6PXezLgJUdngG5IM,498
+icmd/utils.py,sha256=f5by9hEEMtBueV9SN8bLJZdQn1SCSkBhGBKm8zKv7Ls,86
+icmd_cli-0.1.0.dist-info/METADATA,sha256=zvGbYxd4slN7FBIPXRQdUWVgMS2e3OHAJ32bw9-7-Eo,334
+icmd_cli-0.1.0.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+icmd_cli-0.1.0.dist-info/entry_points.txt,sha256=TJVEpIQe2iVfW9xJxRhkH2qTNqItzqB6_XkDzSU9_GY,39
+icmd_cli-0.1.0.dist-info/top_level.txt,sha256=olzM6Z_QEwg7RrFVOnkgO0I2Ps2m7vhSqYQZYeiawcw,5
+icmd_cli-0.1.0.dist-info/RECORD,,

icmd_cli-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (82.0.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

icmd_cli-0.1.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+icmd = icmd.main:run

icmd_cli-0.1.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+icmd