aurafarmer 0.1.0__tar.gz

aurafarmer-0.1.0/PKG-INFO

@@ -0,0 +1,126 @@
+Metadata-Version: 2.4
+Name: aurafarmer
+Version: 0.1.0
+Summary: Aurex — Terminal AI Coding Agent powered by NVIDIA NIM
+License: MIT
+Keywords: ai,coding,terminal,cli,agent,nvidia,nim
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+Requires-Dist: openai>=1.0.0
+Requires-Dist: rich>=13.0.0
+
+# Aurex
+
+> Terminal AI Coding Agent powered by NVIDIA NIM
+
+```
+ █████╗ ██╗   ██╗██████╗ ███████╗██╗  ██╗
+██╔══██╗██║   ██║██╔══██╗██╔════╝╚██╗██╔╝
+███████║██║   ██║██████╔╝█████╗   ╚███╔╝ 
+██╔══██║██║   ██║██╔══██╗██╔══╝   ██╔██╗ 
+██║  ██║╚██████╔╝██║  ██║███████╗██╔╝ ██╗
+╚═╝  ╚═╝ ╚═════╝ ╚═╝  ╚═╝╚══════╝╚═╝  ╚═╝
+```
+
+Aurex is a Claude Code-style AI agent that runs in your terminal. It can read/write files, run commands, search your codebase, and reason through multi-step coding tasks — all powered by NVIDIA NIM.
+
+---
+
+## Install
+
+### Option 1 — From PyPI (after publishing)
+
+```bash
+pip install aurex
+```
+
+### Option 2 — From GitHub (recommended for now)
+
+```bash
+pip install git+https://github.com/YOUR_USERNAME/aurex.git
+```
+
+### Option 3 — Local development install
+
+```bash
+git clone https://github.com/YOUR_USERNAME/aurex.git
+cd aurex
+pip install -e .
+```
+
+Then just run:
+
+```bash
+aurex
+```
+
+---
+
+## Commands
+
+| Command      | Description                          |
+|--------------|--------------------------------------|
+| `model`      | View / change the active AI model    |
+| `clear`      | Wipe conversation history            |
+| `files`      | List files in current directory      |
+| `help`       | Show command list                    |
+| `exit`       | Quit                                 |
+
+---
+
+## Tools Aurex can use
+
+- **read_file** — Read any file
+- **write_file** — Create or overwrite files
+- **run_command** — Execute shell commands
+- **list_files** — Browse directories
+- **search_files** — Search code with grep
+- **delete_file** — Remove files
+
+---
+
+## Available Models
+
+Switch models at any time by typing `model` inside Aurex:
+
+- moonshotai/kimi-k2.6
+- mistralai/mistral-medium-3.5-128b
+- nvidia/nemotron-3-nano-omni-30b-a3b-reasoning
+- deepseek-ai/deepseek-v4-flash
+- deepseek-ai/deepseek-v4-pro
+- mistralai/mistral-large-3-675b-instruct-2512
+- qwen/qwen3-coder-480b-a35b-instruct
+- meta/llama-3.2-90b-vision-instruct
+- ...and more
+
+Type `C` for a completely custom model name.
+
+---
+
+## Publishing to PyPI (when ready)
+
+```bash
+pip install build twine
+
+# Build
+python -m build
+
+# Upload
+twine upload dist/*
+```
+
+You'll need a PyPI account and API token.
+
+---
+
+## Requirements
+
+- Python 3.9+
+- `openai` >= 1.0.0
+- `rich` >= 13.0.0
+
+---
+
+## License
+
+MIT

aurafarmer-0.1.0/README.md

@@ -0,0 +1,115 @@
+# Aurex
+
+> Terminal AI Coding Agent powered by NVIDIA NIM
+
+```
+ █████╗ ██╗   ██╗██████╗ ███████╗██╗  ██╗
+██╔══██╗██║   ██║██╔══██╗██╔════╝╚██╗██╔╝
+███████║██║   ██║██████╔╝█████╗   ╚███╔╝ 
+██╔══██║██║   ██║██╔══██╗██╔══╝   ██╔██╗ 
+██║  ██║╚██████╔╝██║  ██║███████╗██╔╝ ██╗
+╚═╝  ╚═╝ ╚═════╝ ╚═╝  ╚═╝╚══════╝╚═╝  ╚═╝
+```
+
+Aurex is a Claude Code-style AI agent that runs in your terminal. It can read/write files, run commands, search your codebase, and reason through multi-step coding tasks — all powered by NVIDIA NIM.
+
+---
+
+## Install
+
+### Option 1 — From PyPI (after publishing)
+
+```bash
+pip install aurex
+```
+
+### Option 2 — From GitHub (recommended for now)
+
+```bash
+pip install git+https://github.com/YOUR_USERNAME/aurex.git
+```
+
+### Option 3 — Local development install
+
+```bash
+git clone https://github.com/YOUR_USERNAME/aurex.git
+cd aurex
+pip install -e .
+```
+
+Then just run:
+
+```bash
+aurex
+```
+
+---
+
+## Commands
+
+| Command      | Description                          |
+|--------------|--------------------------------------|
+| `model`      | View / change the active AI model    |
+| `clear`      | Wipe conversation history            |
+| `files`      | List files in current directory      |
+| `help`       | Show command list                    |
+| `exit`       | Quit                                 |
+
+---
+
+## Tools Aurex can use
+
+- **read_file** — Read any file
+- **write_file** — Create or overwrite files
+- **run_command** — Execute shell commands
+- **list_files** — Browse directories
+- **search_files** — Search code with grep
+- **delete_file** — Remove files
+
+---
+
+## Available Models
+
+Switch models at any time by typing `model` inside Aurex:
+
+- moonshotai/kimi-k2.6
+- mistralai/mistral-medium-3.5-128b
+- nvidia/nemotron-3-nano-omni-30b-a3b-reasoning
+- deepseek-ai/deepseek-v4-flash
+- deepseek-ai/deepseek-v4-pro
+- mistralai/mistral-large-3-675b-instruct-2512
+- qwen/qwen3-coder-480b-a35b-instruct
+- meta/llama-3.2-90b-vision-instruct
+- ...and more
+
+Type `C` for a completely custom model name.
+
+---
+
+## Publishing to PyPI (when ready)
+
+```bash
+pip install build twine
+
+# Build
+python -m build
+
+# Upload
+twine upload dist/*
+```
+
+You'll need a PyPI account and API token.
+
+---
+
+## Requirements
+
+- Python 3.9+
+- `openai` >= 1.0.0
+- `rich` >= 13.0.0
+
+---
+
+## License
+
+MIT

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

@@ -0,0 +1,126 @@
+Metadata-Version: 2.4
+Name: aurafarmer
+Version: 0.1.0
+Summary: Aurex — Terminal AI Coding Agent powered by NVIDIA NIM
+License: MIT
+Keywords: ai,coding,terminal,cli,agent,nvidia,nim
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+Requires-Dist: openai>=1.0.0
+Requires-Dist: rich>=13.0.0
+
+# Aurex
+
+> Terminal AI Coding Agent powered by NVIDIA NIM
+
+```
+ █████╗ ██╗   ██╗██████╗ ███████╗██╗  ██╗
+██╔══██╗██║   ██║██╔══██╗██╔════╝╚██╗██╔╝
+███████║██║   ██║██████╔╝█████╗   ╚███╔╝ 
+██╔══██║██║   ██║██╔══██╗██╔══╝   ██╔██╗ 
+██║  ██║╚██████╔╝██║  ██║███████╗██╔╝ ██╗
+╚═╝  ╚═╝ ╚═════╝ ╚═╝  ╚═╝╚══════╝╚═╝  ╚═╝
+```
+
+Aurex is a Claude Code-style AI agent that runs in your terminal. It can read/write files, run commands, search your codebase, and reason through multi-step coding tasks — all powered by NVIDIA NIM.
+
+---
+
+## Install
+
+### Option 1 — From PyPI (after publishing)
+
+```bash
+pip install aurex
+```
+
+### Option 2 — From GitHub (recommended for now)
+
+```bash
+pip install git+https://github.com/YOUR_USERNAME/aurex.git
+```
+
+### Option 3 — Local development install
+
+```bash
+git clone https://github.com/YOUR_USERNAME/aurex.git
+cd aurex
+pip install -e .
+```
+
+Then just run:
+
+```bash
+aurex
+```
+
+---
+
+## Commands
+
+| Command      | Description                          |
+|--------------|--------------------------------------|
+| `model`      | View / change the active AI model    |
+| `clear`      | Wipe conversation history            |
+| `files`      | List files in current directory      |
+| `help`       | Show command list                    |
+| `exit`       | Quit                                 |
+
+---
+
+## Tools Aurex can use
+
+- **read_file** — Read any file
+- **write_file** — Create or overwrite files
+- **run_command** — Execute shell commands
+- **list_files** — Browse directories
+- **search_files** — Search code with grep
+- **delete_file** — Remove files
+
+---
+
+## Available Models
+
+Switch models at any time by typing `model` inside Aurex:
+
+- moonshotai/kimi-k2.6
+- mistralai/mistral-medium-3.5-128b
+- nvidia/nemotron-3-nano-omni-30b-a3b-reasoning
+- deepseek-ai/deepseek-v4-flash
+- deepseek-ai/deepseek-v4-pro
+- mistralai/mistral-large-3-675b-instruct-2512
+- qwen/qwen3-coder-480b-a35b-instruct
+- meta/llama-3.2-90b-vision-instruct
+- ...and more
+
+Type `C` for a completely custom model name.
+
+---
+
+## Publishing to PyPI (when ready)
+
+```bash
+pip install build twine
+
+# Build
+python -m build
+
+# Upload
+twine upload dist/*
+```
+
+You'll need a PyPI account and API token.
+
+---
+
+## Requirements
+
+- Python 3.9+
+- `openai` >= 1.0.0
+- `rich` >= 13.0.0
+
+---
+
+## License
+
+MIT

aurafarmer-0.1.0/aurafarmer.egg-info/SOURCES.txt

@@ -0,0 +1,14 @@
+README.md
+pyproject.toml
+aurafarmer.egg-info/PKG-INFO
+aurafarmer.egg-info/SOURCES.txt
+aurafarmer.egg-info/dependency_links.txt
+aurafarmer.egg-info/entry_points.txt
+aurafarmer.egg-info/requires.txt
+aurafarmer.egg-info/top_level.txt
+aurex/__init__.py
+aurex/agent.py
+aurex/config.py
+aurex/main.py
+aurex/tools.py
+aurex/ui.py

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

@@ -0,0 +1 @@
+

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

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

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

@@ -0,0 +1,2 @@
+openai>=1.0.0
+rich>=13.0.0

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

@@ -0,0 +1 @@
+aurex

aurafarmer-0.1.0/aurex/__init__.py

@@ -0,0 +1,2 @@
+__version__ = "0.1.0"
+__author__ = "aurex"

aurafarmer-0.1.0/aurex/agent.py

@@ -0,0 +1,148 @@
+import json
+from openai import OpenAI
+
+from . import config
+from . import tools as tool_module
+
+SYSTEM_PROMPT = """\
+You are Aurex, an elite AI coding agent running directly in the developer's terminal.
+
+You have access to tools that let you read and write files, run shell commands, search code, \
+and more. When asked to do something, do it — don't just describe how. Be concise, decisive, \
+and action-oriented.
+
+Rules:
+- Always use tools to act on the filesystem rather than only describing changes.
+- When writing code, write the entire file, not just snippets.
+- When running commands, show what you're doing and why.
+- Keep responses tight — no filler, no disclaimers.
+"""
+
+
+class AurexAgent:
+    def __init__(self):
+        self.history: list[dict] = []
+        self.client = OpenAI(api_key=config.API_KEY, base_url=config.BASE_URL)
+
+    # ── Public ─────────────────────────────────────────────────────────────────
+
+    def clear_history(self) -> None:
+        self.history = []
+
+    def chat(self, user_message: str):
+        """
+        Generator — yields (event_type, data).
+
+        event_type  data
+        ----------  ----------------------------------------------------
+        "text"      str chunk
+        "tool_use"  {"name": str, "args": dict}
+        "tool_res"  {"name": str, "result": str}
+        "error"     str
+        """
+        self.history.append({"role": "user", "content": user_message})
+        messages = [{"role": "system", "content": SYSTEM_PROMPT}] + self.history
+
+        MAX_ITERS = 10
+        for _ in range(MAX_ITERS):
+            text, tool_calls, error = yield from self._call_api(messages)
+
+            if error:
+                yield ("error", error)
+                return
+
+            if tool_calls:
+                # Build the assistant turn with tool_calls attached
+                assistant_msg: dict = {"role": "assistant", "content": text or None,
+                                       "tool_calls": self._format_tool_calls(tool_calls)}
+                messages.append(assistant_msg)
+
+                # Execute each tool and feed results back
+                tool_result_msgs = []
+                for tc in tool_calls:
+                    try:
+                        args = json.loads(tc["arguments"] or "{}")
+                    except json.JSONDecodeError:
+                        args = {}
+
+                    yield ("tool_use", {"name": tc["name"], "args": args})
+                    result = tool_module.execute_tool(tc["name"], args)
+                    yield ("tool_res", {"name": tc["name"], "result": result})
+
+                    tool_result_msgs.append({
+                        "role": "tool",
+                        "tool_call_id": tc["id"],
+                        "content": result,
+                    })
+
+                messages.extend(tool_result_msgs)
+                # Loop again so the model can see tool results
+
+            else:
+                # No tool calls — model is done
+                if text:
+                    self.history.append({"role": "assistant", "content": text})
+                return
+
+    # ── Internal ───────────────────────────────────────────────────────────────
+
+    def _call_api(self, messages: list):
+        """
+        Streams one completion turn.
+        Yields (event_type, data) for text chunks.
+        Returns (full_text, tool_calls, error_or_None).
+        """
+        full_text = ""
+        tool_calls_acc: dict[int, dict] = {}
+        error = None
+
+        try:
+            stream = self.client.chat.completions.create(
+                model=config.get_model(),
+                messages=messages,
+                tools=tool_module.TOOL_DEFINITIONS,
+                stream=True,
+                max_tokens=4096,
+            )
+
+            for chunk in stream:
+                if not chunk.choices:
+                    continue
+                choice = chunk.choices[0]
+                delta  = choice.delta
+
+                # ── text ──────────────────────────────────────────────────────
+                if delta.content:
+                    full_text += delta.content
+                    yield ("text", delta.content)
+
+                # ── tool call fragments ────────────────────────────────────────
+                if delta.tool_calls:
+                    for tc in delta.tool_calls:
+                        i = tc.index
+                        if i not in tool_calls_acc:
+                            tool_calls_acc[i] = {"id": "", "name": "", "arguments": ""}
+                        if tc.id:
+                            tool_calls_acc[i]["id"] = tc.id
+                        if tc.function:
+                            if tc.function.name:
+                                tool_calls_acc[i]["name"] += tc.function.name
+                            if tc.function.arguments:
+                                tool_calls_acc[i]["arguments"] += tc.function.arguments
+
+        except Exception as exc:
+            error = str(exc)
+
+        tool_calls = list(tool_calls_acc.values()) if tool_calls_acc else []
+        return full_text, tool_calls, error
+
+    @staticmethod
+    def _format_tool_calls(tool_calls: list[dict]) -> list[dict]:
+        return [
+            {
+                "id":       tc["id"],
+                "type":     "function",
+                "function": {"name": tc["name"], "arguments": tc["arguments"]},
+            }
+            for tc in tool_calls
+        ]

aurafarmer-0.1.0/aurex/config.py

@@ -0,0 +1,61 @@
+import json
+from pathlib import Path
+
+# ── API ────────────────────────────────────────────────────────────────────────
+API_KEY  = "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"
+BASE_URL = "https://integrate.api.nvidia.com/v1"
+
+# ── Available models ───────────────────────────────────────────────────────────
+MODELS = [
+    "moonshotai/kimi-k2.6",
+    "mistralai/mistral-medium-3.5-128b",
+    "nvidia/nemotron-3-nano-omni-30b-a3b-reasoning",
+    "deepseek-ai/deepseek-v4-flash",
+    "deepseek-ai/deepseek-v4-pro",
+    "z-ai/glm-5.1",
+    "z-ai/glm-4.7",
+    "minimaxai/minimax-m2.7",
+    "google/gemma-4-31b-it",
+    "mistralai/mistral-small-4-119b-2603",
+    "nvidia/nemotron-3-super-120b-a12b",
+    "qwen/qwen3.5-122b-a10b",
+    "qwen/qwen3.5-397b-a17b",
+    "stepfun-ai/step-3.5-flash",
+    "mistralai/mistral-large-3-675b-instruct-2512",
+    "stockmark/stockmark-2-100b-instruct",
+    "qwen/qwen3-coder-480b-a35b-instruct",
+    "openai/gpt-oss-120b",
+    "meta/llama-3.2-90b-vision-instruct",
+]
+
+DEFAULT_MODEL = MODELS[0]
+
+# ── Persistence ────────────────────────────────────────────────────────────────
+CONFIG_DIR  = Path.home() / ".aurex"
+CONFIG_FILE = CONFIG_DIR / "config.json"
+
+
+def load_config() -> dict:
+    if CONFIG_FILE.exists():
+        try:
+            with open(CONFIG_FILE) as f:
+                return json.load(f)
+        except Exception:
+            pass
+    return {"model": DEFAULT_MODEL}
+
+
+def save_config(cfg: dict) -> None:
+    CONFIG_DIR.mkdir(parents=True, exist_ok=True)
+    with open(CONFIG_FILE, "w") as f:
+        json.dump(cfg, f, indent=2)
+
+
+def get_model() -> str:
+    return load_config().get("model", DEFAULT_MODEL)
+
+
+def set_model(model: str) -> None:
+    cfg = load_config()
+    cfg["model"] = model
+    save_config(cfg)

aurafarmer-0.1.0/aurex/main.py

@@ -0,0 +1,132 @@
+import os
+import sys
+
+from rich.console import Console
+
+from . import config, ui
+from .agent import AurexAgent
+from . import tools as tool_module
+
+console = Console()
+
+
+# ── Model command ──────────────────────────────────────────────────────────────
+
+def handle_model_command(agent: AurexAgent) -> None:
+    models  = config.MODELS
+    current = config.get_model()
+
+    ui.print_current_model(current)
+    ui.print_model_list(models, current)
+
+    try:
+        choice = console.input("  [bold red]❯[/bold red] ").strip()
+    except (KeyboardInterrupt, EOFError):
+        return
+
+    if not choice:
+        ui.print_info(f"Keeping: {current}")
+        return
+
+    if choice.upper() == "C":
+        try:
+            custom = console.input("  [dim red]Custom model name:[/dim red] ").strip()
+        except (KeyboardInterrupt, EOFError):
+            return
+        if custom:
+            config.set_model(custom)
+            ui.print_info(f"Model set to [bold red]{custom}[/bold red]")
+    else:
+        try:
+            idx = int(choice) - 1
+            if 0 <= idx < len(models):
+                selected = models[idx]
+                config.set_model(selected)
+                ui.print_info(f"Model set to [bold red]{selected}[/bold red]")
+            else:
+                ui.print_error(f"Enter a number between 1 and {len(models)}.")
+        except ValueError:
+            ui.print_error("Invalid input — enter a number or C.")
+
+
+# ── Agent runner ───────────────────────────────────────────────────────────────
+
+def run_agent(agent: AurexAgent, user_input: str) -> None:
+    console.print()
+    header_printed = False
+
+    for event_type, data in agent.chat(user_input):
+
+        if event_type == "text":
+            if not header_printed:
+                ui.print_assistant_header()
+                header_printed = True
+            console.print(data, end="", markup=False, highlight=False)
+
+        elif event_type == "tool_use":
+            console.print()
+            ui.print_tool_use(data["name"], data["args"])
+
+        elif event_type == "tool_res":
+            ui.print_tool_result(data["name"], data["result"])
+
+        elif event_type == "error":
+            ui.print_error(data)
+
+    console.print("\n")  # breathing room after response
+
+
+# ── Main REPL ──────────────────────────────────────────────────────────────────
+
+def main() -> None:
+    os.system("clear" if os.name == "posix" else "cls")
+    ui.print_banner()
+
+    model = config.get_model()
+    console.print(f"  [dim]Model  [bold red]{model}[/bold red][/dim]")
+    console.print(f"  [dim]CWD    [bold red]{os.getcwd()}[/bold red][/dim]\n")
+    ui.print_rule()
+    console.print()
+
+    agent = AurexAgent()
+
+    while True:
+        try:
+            user_input = console.input("[bold red]❯ [/bold red]").strip()
+        except (KeyboardInterrupt, EOFError):
+            console.print("\n\n[bold red]Goodbye.[/bold red]\n")
+            sys.exit(0)
+
+        if not user_input:
+            continue
+
+        cmd = user_input.lower()
+
+        if cmd in ("exit", "quit", "bye", "q"):
+            console.print("\n[bold red]Goodbye.[/bold red]\n")
+            sys.exit(0)
+
+        elif cmd == "help":
+            ui.print_help()
+
+        elif cmd == "model":
+            handle_model_command(agent)
+
+        elif cmd == "clear":
+            agent.clear_history()
+            os.system("clear" if os.name == "posix" else "cls")
+            ui.print_banner()
+            ui.print_info("Conversation cleared.")
+
+        elif cmd in ("files", "ls"):
+            result = tool_module.list_files(".")
+            console.print(f"\n[dim red]  {os.getcwd()}[/dim red]")
+            console.print(result)
+            console.print()
+
+        else:
+            run_agent(agent, user_input)
+
+
+if __name__ == "__main__":
+    main()

aurafarmer-0.1.0/aurex/tools.py

@@ -0,0 +1,189 @@
+import subprocess
+from pathlib import Path
+
+# ── Tool schemas (OpenAI function-calling format) ──────────────────────────────
+TOOL_DEFINITIONS = [
+    {
+        "type": "function",
+        "function": {
+            "name": "read_file",
+            "description": "Read the full contents of a file on disk.",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "path": {"type": "string", "description": "Path to the file"}
+                },
+                "required": ["path"],
+            },
+        },
+    },
+    {
+        "type": "function",
+        "function": {
+            "name": "write_file",
+            "description": "Create or overwrite a file with the given content.",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "path":    {"type": "string", "description": "Path to the file"},
+                    "content": {"type": "string", "description": "Content to write"},
+                },
+                "required": ["path", "content"],
+            },
+        },
+    },
+    {
+        "type": "function",
+        "function": {
+            "name": "run_command",
+            "description": "Execute a shell command and return stdout + stderr.",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "command": {"type": "string", "description": "Shell command to run"}
+                },
+                "required": ["command"],
+            },
+        },
+    },
+    {
+        "type": "function",
+        "function": {
+            "name": "list_files",
+            "description": "List files and directories at a given path.",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "path": {"type": "string", "description": "Directory path (default: .)"}
+                },
+                "required": [],
+            },
+        },
+    },
+    {
+        "type": "function",
+        "function": {
+            "name": "search_files",
+            "description": "Search for a text pattern inside files recursively.",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "pattern":      {"type": "string", "description": "Text to search for"},
+                    "path":         {"type": "string", "description": "Root directory (default: .)"},
+                    "file_pattern": {"type": "string", "description": "Glob filter e.g. *.py"},
+                },
+                "required": ["pattern"],
+            },
+        },
+    },
+    {
+        "type": "function",
+        "function": {
+            "name": "delete_file",
+            "description": "Delete a file from disk.",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "path": {"type": "string", "description": "Path to the file to delete"}
+                },
+                "required": ["path"],
+            },
+        },
+    },
+]
+
+
+# ── Implementations ────────────────────────────────────────────────────────────
+
+def read_file(path: str) -> str:
+    try:
+        return Path(path).read_text(encoding="utf-8")
+    except FileNotFoundError:
+        return f"Error: '{path}' not found."
+    except Exception as e:
+        return f"Error reading file: {e}"
+
+
+def write_file(path: str, content: str) -> str:
+    try:
+        p = Path(path)
+        p.parent.mkdir(parents=True, exist_ok=True)
+        p.write_text(content, encoding="utf-8")
+        return f"Wrote {len(content)} chars to '{path}'."
+    except Exception as e:
+        return f"Error writing file: {e}"
+
+
+def run_command(command: str) -> str:
+    try:
+        result = subprocess.run(
+            command, shell=True, capture_output=True, text=True, timeout=60
+        )
+        out = result.stdout.strip()
+        err = result.stderr.strip()
+        parts = []
+        if out:
+            parts.append(out)
+        if err:
+            parts.append(f"[stderr]\n{err}")
+        if not parts:
+            parts.append(f"(exit code {result.returncode})")
+        return "\n".join(parts)
+    except subprocess.TimeoutExpired:
+        return "Error: command timed out after 60 s."
+    except Exception as e:
+        return f"Error: {e}"
+
+
+def list_files(path: str = ".") -> str:
+    try:
+        p = Path(path)
+        lines = []
+        for item in sorted(p.iterdir()):
+            if item.is_dir():
+                lines.append(f"  📁  {item.name}/")
+            else:
+                size = item.stat().st_size
+                lines.append(f"  📄  {item.name}  ({size:,} B)")
+        return "\n".join(lines) if lines else "(empty)"
+    except Exception as e:
+        return f"Error: {e}"
+
+
+def search_files(pattern: str, path: str = ".", file_pattern: str = "*") -> str:
+    cmd = f"grep -rn --include='{file_pattern}' '{pattern}' {path}"
+    return run_command(cmd)
+
+
+def delete_file(path: str) -> str:
+    try:
+        Path(path).unlink()
+        return f"Deleted '{path}'."
+    except FileNotFoundError:
+        return f"Error: '{path}' not found."
+    except Exception as e:
+        return f"Error: {e}"
+
+
+# ── Dispatcher ─────────────────────────────────────────────────────────────────
+
+def execute_tool(name: str, args: dict) -> str:
+    dispatch = {
+        "read_file":    lambda a: read_file(a["path"]),
+        "write_file":   lambda a: write_file(a["path"], a["content"]),
+        "run_command":  lambda a: run_command(a["command"]),
+        "list_files":   lambda a: list_files(a.get("path", ".")),
+        "search_files": lambda a: search_files(
+                            a["pattern"],
+                            a.get("path", "."),
+                            a.get("file_pattern", "*"),
+                        ),
+        "delete_file":  lambda a: delete_file(a["path"]),
+    }
+    fn = dispatch.get(name)
+    if fn is None:
+        return f"Unknown tool: {name}"
+    try:
+        return fn(args)
+    except Exception as e:
+        return f"Tool error: {e}"

aurafarmer-0.1.0/aurex/ui.py

@@ -0,0 +1,150 @@
+import json
+from rich.console import Console
+from rich.panel import Panel
+from rich.table import Table
+from rich.text import Text
+from rich.rule import Rule
+from rich.markdown import Markdown
+from rich import box
+
+console = Console()
+
+# ── Banner ─────────────────────────────────────────────────────────────────────
+BANNER = """\
+ █████╗ ██╗   ██╗██████╗ ███████╗██╗  ██╗
+██╔══██╗██║   ██║██╔══██╗██╔════╝╚██╗██╔╝
+███████║██║   ██║██████╔╝█████╗   ╚███╔╝ 
+██╔══██║██║   ██║██╔══██╗██╔══╝   ██╔██╗ 
+██║  ██║╚██████╔╝██║  ██║███████╗██╔╝ ██╗
+╚═╝  ╚═╝ ╚═════╝ ╚═╝  ╚═╝╚══════╝╚═╝  ╚═╝"""
+
+TOOL_ICONS = {
+    "read_file":    "📖",
+    "write_file":   "✏️ ",
+    "run_command":  "⚡",
+    "list_files":   "📁",
+    "search_files": "🔍",
+    "delete_file":  "🗑️ ",
+}
+
+
+# ── Core ───────────────────────────────────────────────────────────────────────
+
+def print_banner() -> None:
+    console.print(BANNER, style="bold red")
+    console.print(
+        "\n  [dim]AI Coding Agent · Powered by NVIDIA NIM[/dim]\n"
+        "  [dim]Type [bold red]help[/] for commands · "
+        "[bold red]model[/] to switch · "
+        "[bold red]exit[/] to quit[/dim]\n"
+    )
+
+
+def print_rule() -> None:
+    console.print(Rule(style="dim red"))
+
+
+def print_error(msg: str) -> None:
+    console.print(f"\n[bold red]✗[/bold red] [red]{msg}[/red]\n")
+
+
+def print_info(msg: str) -> None:
+    console.print(f"\n[dim red]ℹ[/dim red]  {msg}\n")
+
+
+# ── Tool events ────────────────────────────────────────────────────────────────
+
+def print_tool_use(name: str, args: dict) -> None:
+    icon   = TOOL_ICONS.get(name, "🔧")
+    title  = f"[bold red]{icon} {name}[/bold red]"
+    body   = Text()
+    for k, v in args.items():
+        body.append(f"  {k}: ", style="dim red")
+        val = v if len(str(v)) <= 120 else str(v)[:120] + "…"
+        body.append(f"{val}\n", style="white")
+    console.print(
+        Panel(body, title=title, border_style="red", padding=(0, 1), box=box.ROUNDED)
+    )
+
+
+def print_tool_result(name: str, result: str) -> None:
+    MAX = 600
+    display = result if len(result) <= MAX else result[:MAX] + f"\n[dim]… {len(result)-MAX} more chars[/dim]"
+    console.print(
+        Panel(
+            display,
+            title="[dim red]result[/dim red]",
+            border_style="dim red",
+            padding=(0, 1),
+            box=box.ROUNDED,
+        )
+    )
+
+
+# ── Assistant output ───────────────────────────────────────────────────────────
+
+def print_assistant_header() -> None:
+    console.print("\n[bold red]◆ Aurex[/bold red]\n")
+
+
+# ── Help ───────────────────────────────────────────────────────────────────────
+
+def print_help() -> None:
+    t = Table(
+        title="[bold red]Aurex Commands[/bold red]",
+        border_style="red",
+        header_style="bold red",
+        box=box.ROUNDED,
+        show_lines=True,
+    )
+    t.add_column("Command",     style="red",   no_wrap=True)
+    t.add_column("Description", style="white")
+
+    rows = [
+        ("model",        "Show current model / change model"),
+        ("clear",        "Wipe conversation history"),
+        ("files",        "List files in the current directory"),
+        ("help",         "Show this help panel"),
+        ("exit / quit",  "Exit Aurex"),
+    ]
+    for cmd, desc in rows:
+        t.add_row(cmd, desc)
+
+    console.print("\n", t, "\n")
+
+
+# ── Model UI ───────────────────────────────────────────────────────────────────
+
+def print_current_model(model: str) -> None:
+    console.print(
+        Panel(
+            f"[white]{model}[/white]",
+            title="[bold red]Active Model[/bold red]",
+            border_style="red",
+            padding=(0, 2),
+            box=box.ROUNDED,
+        )
+    )
+
+
+def print_model_list(models: list[str], current: str) -> None:
+    t = Table(
+        title="[bold red]Available Models[/bold red]",
+        border_style="red",
+        header_style="bold red",
+        box=box.ROUNDED,
+        show_lines=True,
+    )
+    t.add_column("#",     style="dim red", width=4, justify="right")
+    t.add_column("Model", style="white")
+    t.add_column("",      style="green",   width=12)
+
+    for i, m in enumerate(models, 1):
+        t.add_row(str(i), m, "◀ active" if m == current else "")
+
+    console.print("\n", t)
+    console.print(
+        "\n  [dim]Enter [bold red]number[/] to select · "
+        "[bold red]C[/] for custom model · "
+        "[bold red]Enter[/] to keep current[/dim]\n"
+    )

aurafarmer-0.1.0/pyproject.toml

@@ -0,0 +1,24 @@
+[build-system]
+requires = ["setuptools>=61.0", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "aurafarmer"
+version = "0.1.0"
+description = "Aurex — Terminal AI Coding Agent powered by NVIDIA NIM"
+readme = "README.md"
+requires-python = ">=3.9"
+license = { text = "MIT" }
+keywords = ["ai", "coding", "terminal", "cli", "agent", "nvidia", "nim"]
+
+dependencies = [
+    "openai>=1.0.0",
+    "rich>=13.0.0",
+]
+
+[project.scripts]
+aurex = "aurex.main:main"
+
+[tool.setuptools.packages.find]
+where = ["."]
+include = ["aurex*"]

aurafarmer-0.1.0/setup.cfg

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