icmd-cli 0.1.0__tar.gz

icmd_cli-0.1.0/PKG-INFO

@@ -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/README.md

@@ -0,0 +1,8 @@
+# iCMD
+
+An AI-powered command line tool using local LLM (qwen2.5-coder).
+
+## Installation
+
+```bash
+pip install icmd

icmd_cli-0.1.0/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_cli-0.1.0/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_cli-0.1.0/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_cli-0.1.0/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_cli-0.1.0/icmd/utils.py

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

icmd_cli-0.1.0/icmd_cli.egg-info/PKG-INFO

@@ -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/icmd_cli.egg-info/SOURCES.txt

@@ -0,0 +1,14 @@
+README.md
+pyproject.toml
+icmd/__init__.py
+icmd/executor.py
+icmd/llm.py
+icmd/main.py
+icmd/safety.py
+icmd/utils.py
+icmd_cli.egg-info/PKG-INFO
+icmd_cli.egg-info/SOURCES.txt
+icmd_cli.egg-info/dependency_links.txt
+icmd_cli.egg-info/entry_points.txt
+icmd_cli.egg-info/requires.txt
+icmd_cli.egg-info/top_level.txt

icmd_cli-0.1.0/icmd_cli.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

icmd_cli-0.1.0/icmd_cli.egg-info/entry_points.txt

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

icmd_cli-0.1.0/icmd_cli.egg-info/requires.txt

@@ -0,0 +1 @@
+requests

icmd_cli-0.1.0/icmd_cli.egg-info/top_level.txt

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

icmd_cli-0.1.0/pyproject.toml

@@ -0,0 +1,18 @@
+[build-system]
+requires = ["setuptools>=61.0"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "icmd-cli"
+version = "0.1.0"
+description = "AI-powered command line tool using local LLM"
+readme = "README.md"
+authors = [
+    {name = "Deepak Karthick"}
+]
+requires-python = ">=3.8"
+dependencies = [
+    "requests"
+]
+[project.scripts]
+icmd = "icmd.main:run"

icmd_cli-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+