toni-cli 0.1.0__tar.gz

toni_cli-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 MatrixDynamo
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

toni_cli-0.1.0/PKG-INFO

@@ -0,0 +1,16 @@
+Metadata-Version: 2.2
+Name: toni-cli
+Version: 0.1.0
+Summary: Terminal Operation Natural Instruction - AI-powered terminal assistant
+Author-email: Dakai Ou <oudakai@gmail.com>
+License: MIT
+Project-URL: Homepage, https://github.com/dakai/toni
+Project-URL: Bug Tracker, https://github.com/dakai/toni/issues
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.7
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: openai>=1.0.0
+Requires-Dist: google-generativeai>=0.1.0

toni_cli-0.1.0/pyproject.toml

@@ -0,0 +1,28 @@
+[build-system]
+requires = ["setuptools>=61.0", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "toni-cli"
+version = "0.1.0"
+description = "Terminal Operation Natural Instruction - AI-powered terminal assistant"
+readme = "README.md"
+authors = [{name = "Dakai Ou", email = "oudakai@gmail.com"}]
+license = {text = "MIT"}
+classifiers = [
+    "Programming Language :: Python :: 3",
+    "License :: OSI Approved :: MIT License",
+    "Operating System :: OS Independent",
+]
+requires-python = ">=3.7"
+dependencies = [
+    "openai>=1.0.0",
+    "google-generativeai>=0.1.0",
+]
+
+[project.urls]
+"Homepage" = "https://github.com/dakai/toni"
+"Bug Tracker" = "https://github.com/dakai/toni/issues"
+
+[project.scripts]
+toni = "toni.cli:main"

toni_cli-0.1.0/setup.cfg

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

toni_cli-0.1.0/setup.py

@@ -0,0 +1,5 @@
+#!/usr/bin/env python
+from setuptools import setup
+
+if __name__ == "__main__":
+    setup()

toni_cli-0.1.0/src/toni/__init__.py

@@ -0,0 +1,3 @@
+"""TONI - Terminal Operation Natural Instruction."""
+
+__version__ = "0.1.0"

toni_cli-0.1.0/src/toni/cli.py

@@ -0,0 +1,89 @@
+import argparse
+import os
+import json
+
+from toni.core import (
+    get_system_info,
+    get_gemini_response,
+    get_open_ai_response,
+    command_exists,
+    execute_command,
+)
+
+
+def main():
+    try:
+        parser = argparse.ArgumentParser(
+            description="TONI: Terminal Operation Natural Instruction"
+        )
+        parser.add_argument("query", nargs="+", help="Your natural language query")
+        args = parser.parse_args()
+
+        # Remove trailing question mark if present
+        query = " ".join(args.query).rstrip("?")
+
+        system_info = get_system_info()
+        print(f"Detected system: {system_info}")
+
+        google_api_key = os.environ.get("GOOGLEAI_API_KEY")
+        openai_api_key = os.environ.get("OPENAI_API_KEY")
+        response = None
+
+        # Try Gemini first, fall back to OpenAI
+        if google_api_key:
+            response = get_gemini_response(google_api_key, query, system_info)
+
+        if response is None and openai_api_key:
+            response = get_open_ai_response(openai_api_key, query, system_info)
+
+        if response is None:
+            print(
+                "Please set the GOOGLEAI_API_KEY or OPENAI_API_KEY environment variable."
+            )
+            return
+
+        try:
+            data = json.loads(response)
+        except Exception as e:
+            print(f"An error occurred while parsing the response: {e}")
+            print(f"Raw response: {response}")
+            return
+
+        if data.get("exec") == False:
+            print(data.get("exp"))
+            return
+
+        cmd = data.get("cmd")
+
+        # Check if the command exists
+        if cmd and not command_exists(cmd):
+            print(
+                f"Warning: The command '{cmd.split()[0]}' doesn't appear to be installed on your system."
+            )
+            print(f"Suggested command: {cmd}")
+            print(f"Explanation: {data.get('exp')}")
+            print("Please verify that this command will work on your system.")
+        else:
+            print(f"Suggested command: {cmd}")
+            print(f"Explanation: {data.get('exp')}")
+
+        try:
+            confirmation = input("Do you want to execute this command? (y/n): ").lower()
+            if confirmation == "y" or confirmation == "":
+                execute_command(cmd)
+            else:
+                print("Command execution cancelled.")
+        except KeyboardInterrupt:
+            print("\nOperation cancelled by user.")
+            return
+
+    except KeyboardInterrupt:
+        print("\nOperation cancelled by user.")
+        return
+
+
+if __name__ == "__main__":
+    try:
+        main()
+    except KeyboardInterrupt:
+        print("\nOperation cancelled by user.")

toni_cli-0.1.0/src/toni/core.py

@@ -0,0 +1,161 @@
+from openai import OpenAI
+import os
+import subprocess
+import time
+from google import genai
+import platform
+import shutil
+
+system_message = """Your are a powerful terminal assistant generating a JSON containing a command line for my input.
+You will always reply using the following json structure: {{"cmd":"the command", "exp": "some explanation", "exec": true}}.
+Your answer will always only contain the json structure, never add any advice or supplementary detail or information,
+even if I asked the same question before.
+The field cmd will contain a single line command (don't use new lines, use separators like && and ; instead).
+The field exp will contain an short explanation of the command if you managed to generate an executable command, otherwise it will contain the reason of your failure.
+The field exec will contain true if you managed to generate an executable command, false otherwise.
+
+The host system is using {system_info}. Please ensure commands are compatible with this environment.
+
+Examples:
+Me: list all files in my home dir
+You: {{"cmd":"ls ~", "exp": "list all files in your home dir", "exec": true}}
+Me: list all pods of all namespaces
+You: {{"cmd":"kubectl get pods --all-namespaces", "exp": "list pods form all k8s namespaces", "exec": true}}
+Me: how are you ?
+You: {{"cmd":"", "exp": "I'm good thanks but I cannot generate a command for this.", "exec": false}}"""
+
+
+def get_system_info():
+    system = platform.system()
+    if system == "Linux":
+        try:
+            distro = (
+                subprocess.check_output("cat /etc/os-release | grep -w ID", shell=True)
+                .decode()
+                .strip()
+                .split("=")[1]
+                .strip('"')
+            )
+            return f"Linux ({distro})"
+        except:
+            return "Linux"
+    elif system == "Darwin":
+        return "macOS"
+    elif system == "Windows":
+        return "Windows"
+    else:
+        return system
+
+
+def get_gemini_response(api_key, prompt, system_info):
+    try:
+        client = genai.Client(api_key=api_key)
+
+        formatted_system_message = system_message.format(system_info=system_info)
+
+        # Create the generation config with system instructions
+        # generation_config = {
+        #    "temperature": 0.2,
+        #    "top_p": 0.95,
+        #    "top_k": 0,
+        #    "max_output_tokens": 1024,
+        # }
+
+        # The new Gemini API doesn't always handle system messages properly
+        # Let's combine the system message with the user prompt
+        combined_prompt = f"{formatted_system_message}\n\nUser request: {prompt}"
+
+        response = client.models.generate_content(
+            model="gemini-2.0-flash",
+            contents=[{"parts": [{"text": combined_prompt}]}],
+        )
+
+        # Extract just the JSON part from the response
+        response_text = response.text
+        # Find JSON between curly braces if there's extra text
+        import re
+
+        if response_text:
+            json_match = re.search(r"(\{.*?\})", response_text, re.DOTALL)
+            if json_match:
+                return json_match.group(1)
+            return response_text
+
+    except Exception as e:
+        print(f"An error occurred with Gemini: {e}")
+        return None
+
+
+def get_open_ai_response(api_key, prompt, system_info):
+    try:
+        client = OpenAI(api_key=api_key)
+
+        formatted_system_message = system_message.format(system_info=system_info)
+
+        chat_completion = client.chat.completions.create(
+            messages=[
+                {"role": "system", "content": formatted_system_message},
+                {"role": "user", "content": prompt},
+            ],
+            model="gpt-4o-mini",
+            temperature=0.2,
+        )
+
+        response = chat_completion.choices[0].message.content
+
+        # Extract just the JSON part from the response
+        import re
+
+        if response:
+            json_match = re.search(r"(\{.*?\})", response, re.DOTALL)
+            if json_match:
+                return json_match.group(1)
+            return response
+
+    except Exception as e:
+        print(f"An error occurred with OpenAI: {e}")
+        return None
+
+
+def write_to_zsh_history(command):
+    try:
+        current_time = int(time.time())  # Get current Unix timestamp
+        timestamped_command = (
+            f": {current_time}:0;{command}"  # Assuming duration of 0 for now
+        )
+        with open("/home/dakai/.zsh_history", "a") as f:
+            f.write(timestamped_command + "\n")
+    except Exception as e:
+        print(f"An error occurred while writing to .zsh_history: {e}")
+
+
+def reload_zsh_history():
+    try:
+        os.system("source ~/.zshrc")
+        result = subprocess.run(
+            "source ~/.zshrc", shell=True, check=True, text=True, capture_output=True
+        )
+        print(result.stdout)
+    except Exception as e:
+        print(f"An error occurred while reloading .zshrc: {e}")
+
+
+def execute_command(command):
+    try:
+        result = subprocess.run(
+            command, shell=True, check=True, text=True, capture_output=True
+        )
+        print("Command output:")
+        print(result.stdout)
+        write_to_zsh_history(command)
+        # reload_zsh_history()
+    except subprocess.CalledProcessError as e:
+        print(f"An error occurred while executing the command: {e}")
+        print("Error output:")
+        print(e.stderr)
+
+
+def command_exists(command):
+    # Extract the base command (before any options or arguments)
+    base_command = command.split()[0]
+    return shutil.which(base_command) is not None

toni_cli-0.1.0/src/toni_cli.egg-info/PKG-INFO

@@ -0,0 +1,16 @@
+Metadata-Version: 2.2
+Name: toni-cli
+Version: 0.1.0
+Summary: Terminal Operation Natural Instruction - AI-powered terminal assistant
+Author-email: Dakai Ou <oudakai@gmail.com>
+License: MIT
+Project-URL: Homepage, https://github.com/dakai/toni
+Project-URL: Bug Tracker, https://github.com/dakai/toni/issues
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.7
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: openai>=1.0.0
+Requires-Dist: google-generativeai>=0.1.0

toni_cli-0.1.0/src/toni_cli.egg-info/SOURCES.txt

@@ -0,0 +1,13 @@
+LICENSE
+pyproject.toml
+setup.py
+src/toni/__init__.py
+src/toni/cli.py
+src/toni/core.py
+src/toni_cli.egg-info/PKG-INFO
+src/toni_cli.egg-info/SOURCES.txt
+src/toni_cli.egg-info/dependency_links.txt
+src/toni_cli.egg-info/entry_points.txt
+src/toni_cli.egg-info/requires.txt
+src/toni_cli.egg-info/top_level.txt
+tests/test_toni.py

toni_cli-0.1.0/src/toni_cli.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

toni_cli-0.1.0/src/toni_cli.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+toni = toni.cli:main

toni_cli-0.1.0/src/toni_cli.egg-info/requires.txt

@@ -0,0 +1,2 @@
+openai>=1.0.0
+google-generativeai>=0.1.0

toni_cli-0.1.0/src/toni_cli.egg-info/top_level.txt

@@ -0,0 +1 @@
+toni

toni_cli-0.1.0/tests/test_toni.py

@@ -0,0 +1,28 @@
+"""Tests for the TONI package."""
+
+import unittest
+from unittest.mock import patch, MagicMock
+from toni.core import get_system_info, command_exists
+
+
+class TestToni(unittest.TestCase):
+    """Test cases for TONI functionality."""
+
+    def test_get_system_info(self):
+        """Test that system info is returned."""
+        result = get_system_info()
+        self.assertIsInstance(result, str)
+        self.assertTrue(len(result) > 0)
+
+    @patch("toni.core.shutil.which")
+    def test_command_exists(self, mock_which):
+        """Test command_exists function."""
+        mock_which.return_value = "/usr/bin/ls"
+        self.assertTrue(command_exists("ls"))
+
+        mock_which.return_value = None
+        self.assertFalse(command_exists("nonexistentcommand"))
+
+
+if __name__ == "__main__":
+    unittest.main()