axon-cli 0.1.0__py3-none-any.whl

axon/commands.py

@@ -0,0 +1,54 @@
+import ollama
+import questionary
+from rich.console import Console
+
+from axon import config, memory, tools
+
+console = Console()
+
+
+def handle_command(user_input):
+    if user_input == "/help":
+        print("/help - show commands")
+        print("/clear - clear history")
+        print("/save - save current session")
+        print("/load - load a saved session")
+        print("/model - change model")
+        print("/load-project - load a project folder into context")
+        print("/exit - quit")
+        return True
+    elif user_input == "/clear":
+        memory.clear_history()
+        return True
+    elif user_input == "/exit":
+        exit()
+        return True
+    elif user_input == "/save":
+        filename = input("Enter session name: ")
+        memory.save_session(filename)
+        print(f"Session saved as '{filename}'")
+        return True
+    elif user_input == "/load":
+        filename = input("Enter session name to load: ")
+        memory.load_session(filename)
+        print(f"Session '{filename}' loaded!")
+        return True
+    elif user_input == "/model":
+        models = ollama.list()
+        model_names = model_names = [
+            m.model for m in models.models if m.model is not None
+        ]
+        selected = questionary.select("Select a model:", choices=model_names).ask()
+        if selected:
+            config.current_model = selected
+            print(f"Model switched to '{selected}'")
+        return True
+    elif user_input == "/load-project":
+        folder = input("Enter folder path: ")
+        print("Loading project...")
+        result = tools.load_project(folder)
+        memory.add_message("user", f"Here is my project:\n{result}")
+        console.print("[green]Project loaded![/green]")
+        return True
+    else:
+        return False

axon/config.py

@@ -0,0 +1 @@
+current_model = "llama3.2"

axon/llm.py

@@ -0,0 +1,12 @@
+import ollama
+
+from axon import config
+
+
+def stream_response(conversation_history):
+    """Streams response tokens from Ollama one by one."""
+    bot = ollama.chat(
+        model=config.current_model, messages=conversation_history, stream=True
+    )
+    for chunk in bot:
+        yield chunk["message"]["content"]

axon/main.py

@@ -0,0 +1,75 @@
+import ollama
+import questionary
+from prompt_toolkit import PromptSession
+from prompt_toolkit.completion import WordCompleter
+
+from axon import commands, config, memory, tools
+from axon.llm import stream_response
+from axon.renderer import print_welcome, render_markdown, thinking
+
+models = ollama.list()
+model_names = [m.model for m in models.models if m.model is not None]
+selected = questionary.select(
+    "Select a model to start with:", choices=model_names
+).ask()
+config.current_model = selected
+print(f"Starting with '{selected}'\n")
+print_welcome()
+
+command_completer = WordCompleter(
+    ["/help", "/clear", "/save", "/load", "/model", "/load-project", "/exit"],
+    sentence=True,
+)
+
+session = PromptSession(completer=command_completer)
+
+
+def main():
+    while True:
+        user_input = session.prompt("\n[>] ")
+        if user_input.startswith("/"):
+            commands.handle_command(user_input)
+        else:
+            memory.add_message("user", user_input)
+            response = stream_response(memory.conversation_history)
+            with thinking():
+                full_response = ""
+                for token in response:
+                    full_response += token
+            tool, args = tools.parse_tool_call(full_response)
+            if tool == "read_file":
+                args = args.split("/")[-1].strip()
+                result = tools.read_file(args)
+                memory.add_message("assistant", full_response)
+                memory.add_message("user", f"Tool result:\n{result}")
+                with thinking():
+                    full_response = ""
+                    response = stream_response(memory.conversation_history)
+                    for token in response:
+                        full_response += token
+            elif tool == "run_command":
+                result = tools.run_command(args)
+                memory.add_message("assistant", full_response)
+                memory.add_message("user", f"Tool result:\n{result}")
+                with thinking():
+                    full_response = ""
+                    response = stream_response(memory.conversation_history)
+                    for token in response:
+                        full_response += token
+            elif tool == "write_file":
+                filepath, content = args.split("|", 1)
+                filepath = filepath.split("/")[-1].strip()
+                result = tools.write_file(filepath, content)
+                memory.add_message("assistant", full_response)
+                memory.add_message("user", f"Tool result:\n{result}")
+                with thinking():
+                    full_response = ""
+                    response = stream_response(memory.conversation_history)
+                    for token in response:
+                        full_response += token
+            render_markdown(full_response)
+            memory.add_message("assistant", full_response)
+
+
+if __name__ == "__main__":
+    main()

axon/memory.py

@@ -0,0 +1,42 @@
+import json
+
+conversation_history = [
+    {
+        "role": "system",
+        "content": """You are AXON, a helpful CLI assistant running locally via Ollama.
+
+        ONLY use tools when the user EXPLICITLY asks you to read, write, or run something.
+        NEVER use tools to answer general questions — just respond normally.
+
+        To read a file respond ONLY with:
+        <tool>read_file</tool><args>filepath</args>
+
+        To run a command respond ONLY with:
+        <tool>run_command</tool><args>command</args>
+
+        To write a file respond ONLY with:
+        <tool>write_file</tool><args>filepath|content</args>
+
+        Use the EXACT filename the user mentioned. Never use placeholder paths.
+        For everything else including explaining code, answering questions, summarizing — respond normally and be concise.""",
+    }
+]
+
+
+def add_message(role, content):
+    conversation_history.append({"role": role, "content": content})
+
+
+def clear_history():
+    conversation_history[:] = [conversation_history[0]]
+
+
+def save_session(filename):
+    with open(f"sessions/{filename}.json", "w") as f:
+        json.dump(conversation_history, f, indent=2)
+
+
+def load_session(filename):
+    with open(f"sessions/{filename}.json", "r") as f:
+        data = json.load(f)
+        conversation_history[:] = data

axon/renderer.py

@@ -0,0 +1,64 @@
+import os
+
+import pyfiglet
+from rich.columns import Columns
+from rich.console import Console
+from rich.live import Live
+from rich.markdown import Markdown
+from rich.panel import Panel
+from rich.spinner import Spinner
+from rich.text import Text
+
+from axon import config
+
+console = Console()
+
+
+def print_token(token):
+    print(token, end="", flush=True)
+
+
+def render_markdown(markdown_text):
+    md = Markdown(markdown_text)
+    console.print(md)
+
+
+def thinking():
+    return Live(Spinner("dots", text=" Thinking..."), console=console)
+
+
+def print_banner():
+    ascii_art = pyfiglet.figlet_format("AXON", font="slant")
+    text = Text(ascii_art)
+    text.stylize("bold cyan")
+    console.print(text)
+
+
+def print_welcome():
+    print_banner()
+
+    left_content = Text()
+    left_content.append("\n  Welcome back!\n\n", style="bold white")
+    left_content.append("  ◈\n\n", style="bold cyan")
+    left_content.append(f"  {config.current_model}\n", style="dim")
+    left_content.append("  Local via Ollama\n", style="dim")
+    left_panel = Panel(left_content, style="cyan", width=35)
+
+    sessions = os.listdir("sessions") if os.path.exists("sessions") else []
+    activity = (
+        f"Last session: {sorted(sessions)[-1].replace('.json', '')}"
+        if sessions
+        else "No recent activity"
+    )
+
+    right_content = Text()
+    right_content.append("Tips for getting started\n\n", style="bold yellow")
+    right_content.append("1. Ask questions, edit files, or run commands.\n")
+    right_content.append("2. Use /load-project to load a full codebase.\n")
+    right_content.append("3. Use /model to switch models.\n\n")
+    right_content.append("Recent activity\n\n", style="bold yellow")
+    right_content.append(activity, style="dim")
+    right_panel = Panel(right_content, style="cyan", width=60)
+
+    console.print(Columns([left_panel, right_panel]))
+    console.print()

axon/tools.py

@@ -0,0 +1,80 @@
+import os
+import subprocess
+
+
+def load_project(folder_path):
+    allowed = [
+        ".py",
+        ".js",
+        ".html",
+        ".css",
+        ".json",
+        ".md",
+        ".txt",
+        ".ts",
+        ".jsx",
+        ".tsx",
+    ]
+    skip_folders = ["node_modules", ".git", "__pycache__", ".venv", "venv"]
+    result = ""
+    file_count = 0
+    for root, dirs, files in os.walk(folder_path):
+        # skip junk folders
+        dirs[:] = [d for d in dirs if d not in skip_folders]
+        for file in files:
+            _, ext = os.path.splitext(file)
+            if ext not in allowed:
+                continue
+            full_path = os.path.join(root, file)
+            content = read_file(full_path)
+            result += f"\n=== {full_path} ===\n{content}\n"
+            file_count += 1
+    if file_count == 0:
+        return "No supported files found in that folder."
+    return f"Loaded {file_count} files:\n{result}"
+
+
+def parse_tool_call(text):
+    if "<tool>" not in text:
+        return (None, None)
+
+    start = text.find("<tool>") + len("<tool>")
+    end = text.find("</tool>")
+    tool_name = text[start:end]
+
+    start = text.find("<args>") + len("<args>")
+    end = text.find("</args>")
+    args = text[start:end]
+
+    return (tool_name, args)
+
+
+def read_file(file_path):
+    try:
+        with open(file_path, "r", encoding="utf-8", errors="ignore") as file:
+            return file.read()
+    except FileNotFoundError:
+        return f"Error: file '{file_path}' not found."
+
+
+def write_file(file_path, content):
+    try:
+        with open(file_path, "w") as file:
+            file.write(content)
+        return f"Success: file '{file_path}' written."
+    except Exception as e:
+        return f"Error: {e}"
+
+
+def run_command(command):
+    try:
+        result = subprocess.run(command, shell=True, capture_output=True, text=True)
+        # if there is output return it, otherwise return errors
+        if result.stdout:
+            return result.stdout
+        elif result.stderr:
+            return result.stderr
+        else:
+            return "Command ran successfully with no output."
+    except Exception as e:
+        return f"Error: {e}"

axon_cli-0.1.0.dist-info/METADATA

@@ -0,0 +1,9 @@
+Metadata-Version: 2.4
+Name: axon-cli
+Version: 0.1.0
+Summary: A local CLI AI assistant powered by Ollama
+Requires-Dist: ollama
+Requires-Dist: rich
+Requires-Dist: prompt_toolkit
+Requires-Dist: pyfiglet
+Requires-Dist: questionary

axon_cli-0.1.0.dist-info/RECORD

@@ -0,0 +1,13 @@
+axon/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+axon/commands.py,sha256=TuUdtnnAMS1Fqj6owHlseG8UgATM15MB_zNsmt7dALI,1847
+axon/config.py,sha256=XdKpURgJMFDMp3Xh8X-MsNw2rifslr4AeZvb5LIN3-w,28
+axon/llm.py,sha256=BX9GwfKcKMpSZcmQbxS53_0EoOTInmMPnh_3E8i9Glg,326
+axon/main.py,sha256=24yQa00EwhqUccHac6crhBraNssPT4W_VTWqYngCc0U,3018
+axon/memory.py,sha256=su2KD9kfIT31W6Mv4PmR8Y4OR7-GhTUE32Za3jvw8qM,1356
+axon/renderer.py,sha256=fl1icyy466Jhg6Zlf0NoPOJ6Cn1oN2tFoA7IHyqLdGo,1948
+axon/tools.py,sha256=r80hfWiP7wODLFJnDxTZjEBxOdmTZtjoUyhte7eajdg,2253
+axon_cli-0.1.0.dist-info/METADATA,sha256=xTVrd5a7mmPfhxLCQJuL14l4kAnSD5tjgbLigEpH4vk,236
+axon_cli-0.1.0.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+axon_cli-0.1.0.dist-info/entry_points.txt,sha256=qwp1vV4rHTSDQzACmA9OmKzR5HiwwFp46LZxh1hjeFw,40
+axon_cli-0.1.0.dist-info/top_level.txt,sha256=vRkPFUYD0GspqpZfeuY7c2fzJmpHIpgjOY_nnOHNwVc,5
+axon_cli-0.1.0.dist-info/RECORD,,

axon_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
+

axon_cli-0.1.0.dist-info/entry_points.txt

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

axon_cli-0.1.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+axon