axon-cli 0.1.0__tar.gz

axon_cli-0.1.0/PKG-INFO

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

@@ -0,0 +1,98 @@
+# AXON 🤖
+> A local CLI AI assistant powered by Ollama — works like Gemini CLI or Claude Code, completely offline and private.
+
+## Features
+- 🧠 Conversational AI with full memory across turns
+- 🌊 Streaming responses with a thinking spinner
+- 📄 Markdown rendering in the terminal
+- 📁 Read, write files in your current directory
+- ⚡ Run shell commands directly from chat
+- 📂 Load full project folders into context
+- 💾 Save and load chat sessions
+- 🔀 Switch between Ollama models at runtime
+- ⌨️ Slash command autocomplete
+
+## Requirements
+- Python 3.10+
+- [Ollama](https://ollama.com) installed and running
+- At least one Ollama model pulled (e.g. `ollama pull llama3.2`)
+
+## Installation
+
+**1. Clone the repository**
+
+    git clone https://github.com/yourusername/axon-cli.git
+    cd axon-cli
+
+**2. Install dependencies**
+
+    pip install -r requirements.txt
+
+**3. Install AXON as a command**
+
+    pip install -e .
+
+**4. Make sure Ollama is running**
+
+    ollama serve
+
+**5. Run AXON from anywhere**
+
+    axon
+
+## Usage
+
+On startup, AXON will ask you to select a model from your installed Ollama models using arrow keys. Then you can start chatting right away.
+
+    ? Select a model to start with:
+     ❯ llama3.2:latest
+       gemma4:E4B
+
+## Commands
+
+| Command | Description |
+|---|---|
+| `/help` | Show all available commands |
+| `/clear` | Clear conversation history |
+| `/save` | Save current session to disk |
+| `/load` | Load a previously saved session |
+| `/model` | Switch Ollama model at runtime |
+| `/load-project` | Load a full project folder into context |
+| `/exit` | Quit AXON |
+
+## Tools
+
+AXON can interact with your filesystem and shell when asked:
+
+- **Read a file** — `read the file main.py`
+- **Write a file** — `write a file called notes.txt with...`
+- **Run a command** — `run dir` or `run git status`
+- **Load a project** — `/load-project` then provide folder path
+
+## Project Structure
+
+    axon-cli/
+    ├── axon/
+    │   ├── __init__.py
+    │   ├── main.py        # Entry point and main loop
+    │   ├── llm.py         # Ollama streaming client
+    │   ├── memory.py      # Conversation history management
+    │   ├── renderer.py    # Rich terminal UI and markdown rendering
+    │   ├── commands.py    # Slash command handler
+    │   ├── tools.py       # File and shell tools
+    │   └── config.py      # Shared configuration
+    ├── sessions/          # Saved chat sessions (auto-created)
+    ├── pyproject.toml
+    ├── requirements.txt
+    └── README.md
+
+## Built With
+
+- [Ollama](https://ollama.com) — local LLM inference
+- [Rich](https://github.com/Textualize/rich) — terminal UI and markdown
+- [prompt_toolkit](https://github.com/prompt-toolkit/python-prompt-toolkit) — autocomplete input
+- [pyfiglet](https://github.com/pwaller/pyfiglet) — ASCII art banner
+- [questionary](https://github.com/tmbo/questionary) — interactive model selection
+
+## License
+MIT

axon_cli-0.1.0/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_cli-0.1.0/axon/config.py

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

axon_cli-0.1.0/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_cli-0.1.0/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_cli-0.1.0/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_cli-0.1.0/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_cli-0.1.0/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/axon_cli.egg-info/PKG-INFO

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

@@ -0,0 +1,16 @@
+README.md
+pyproject.toml
+axon/__init__.py
+axon/commands.py
+axon/config.py
+axon/llm.py
+axon/main.py
+axon/memory.py
+axon/renderer.py
+axon/tools.py
+axon_cli.egg-info/PKG-INFO
+axon_cli.egg-info/SOURCES.txt
+axon_cli.egg-info/dependency_links.txt
+axon_cli.egg-info/entry_points.txt
+axon_cli.egg-info/requires.txt
+axon_cli.egg-info/top_level.txt

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

@@ -0,0 +1 @@
+

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

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

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

@@ -0,0 +1,5 @@
+ollama
+rich
+prompt_toolkit
+pyfiglet
+questionary

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

@@ -0,0 +1,3 @@
+axon
+dist
+sessions

axon_cli-0.1.0/pyproject.toml

@@ -0,0 +1,21 @@
+[project]
+name = "axon-cli"
+version = "0.1.0"
+description = "A local CLI AI assistant powered by Ollama"
+dependencies = [
+    "ollama",
+    "rich",
+    "prompt_toolkit",
+    "pyfiglet",
+    "questionary"
+]
+
+[project.scripts]
+axon = "axon.main:main"
+
+[build-system]
+requires = ["setuptools>=64"]
+build-backend = "setuptools.build_meta"
+
+[tool.setuptools.packages.find]
+where = ["."]

axon_cli-0.1.0/setup.cfg

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