notrack-cli-agent 0.6.0__py3-none-any.whl

notrack_cli/main.py

@@ -0,0 +1,811 @@
+"""
+NoTrack CLI — v0.6.0
+AGY-style autonomous AI terminal. Full shell, sub-agents, slash commands.
+"""
+import sys
+import re
+import os
+import time
+import threading
+import subprocess
+import shlex
+import argparse
+from datetime import datetime
+
+from rich.console import Console
+from rich.panel   import Panel
+from rich.text    import Text
+from rich.markdown import Markdown
+from rich.live    import Live
+from rich.table   import Table
+from rich.rule    import Rule
+from rich.syntax  import Syntax
+from rich.columns import Columns
+from prompt_toolkit              import PromptSession
+from prompt_toolkit.history      import FileHistory
+from prompt_toolkit.styles       import Style as PTStyle
+from prompt_toolkit.formatted_text import HTML
+from prompt_toolkit.completion   import WordCompleter
+
+from .client  import NoTrackClient
+from .tools   import LocalTools, SYSTEM_PROMPT
+from .parser  import extract_tool_calls
+from .memory  import MemoryManager
+
+VERSION      = "0.6.0"
+PYPI_PACKAGE = "notrack-cli"
+
+console = Console(highlight=False)
+
+PT_STYLE = PTStyle.from_dict({
+    "prompt":        "#00d787 bold",
+    "bottom-toolbar": "bg:#1a1a1a #555555",
+})
+
+HISTORY_PATH = os.path.expanduser("~/.notrack_cli/input_history")
+
+# ──────────────────────────────────────────────────────────────
+#  SLASH COMMAND REGISTRY
+# ──────────────────────────────────────────────────────────────
+SLASH_COMMANDS = {
+    "/help":     "Show this help",
+    "/clear":    "Clear terminal screen",
+    "/history":  "Show last 10 conversation turns",
+    "/chats":    "List saved NoTrack conversations",
+    "/resume":   "Resume a chat by chat_id",
+    "/reset":    "Clear all history and start fresh",
+    "/new":      "Start a new chat context (keeps history)",
+    "/model":    "Show current model / API info",
+    "/tools":    "List available system tools",
+    "/save":     "Export conversation to a file",
+    "/shell":    "Drop into interactive shell mode",
+    "/git":      "Quick git status + log (3)",
+    "/ls":       "List current directory",
+    "/pwd":      "Print working directory",
+    "/cd":       "Change working directory: /cd <path>",
+    "/version":  "Show NoTrack CLI version",
+    "/exit":     "Exit NoTrack CLI",
+}
+
+# ──────────────────────────────────────────────────────────────
+#  SESSION STATE
+# ──────────────────────────────────────────────────────────────
+class SessionState:
+    always_yes  = False
+    always_no   = False
+    cwd         = os.getcwd()
+    shell_mode  = False
+
+state = SessionState()
+
+DISPATCH_PROMPT_LIMIT = 3900  # Server error reported a 4,000-character maximum.
+
+def fit_dispatch_prompt(prompt: str, max_chars: int = DISPATCH_PROMPT_LIMIT,
+                        preserve_tail: bool = False) -> str:
+    """Fit dispatch user_input to the observed server limit."""
+    if len(prompt) <= max_chars:
+        return prompt
+    marker = "[Earlier context omitted for NoTrack's 4000-character limit.]\n"
+    budget = max_chars - len(marker)
+    if preserve_tail:
+        return marker + prompt[-budget:]
+    return prompt[:budget] + "\n[Prompt shortened to fit NoTrack's limit.]"
+
+
+# ──────────────────────────────────────────────────────────────
+#  AUTO UPDATE CHECK
+# ──────────────────────────────────────────────────────────────
+def check_for_updates():
+    try:
+        import requests
+        r = requests.get(f"https://pypi.org/pypi/{PYPI_PACKAGE}/json", timeout=4)
+        if r.ok:
+            latest = r.json()["info"]["version"]
+            if latest != VERSION:
+                console.print(
+                    f"\n[bold yellow]⬆  Update available:[/bold yellow] "
+                    f"[bold white]v{latest}[/bold white] "
+                    f"[dim](current v{VERSION})[/dim] → "
+                    f"[bold cyan]pip install -U {PYPI_PACKAGE}[/bold cyan]\n"
+                )
+    except Exception:
+        pass
+
+
+# ──────────────────────────────────────────────────────────────
+#  BANNER
+# ──────────────────────────────────────────────────────────────
+def print_banner():
+    os.system("clear")
+    lines = [
+        ("  ███╗   ██╗ ██████╗ ████████╗██████╗  █████╗  ██████╗██╗  ██╗", "bold green"),
+        ("  ████╗  ██║██╔═══██╗╚══██╔══╝██╔══██╗██╔══██╗██╔════╝██║ ██╔╝", "bold green"),
+        ("  ██╔██╗ ██║██║   ██║   ██║   ██████╔╝███████║██║     █████╔╝ ", "bold green"),
+        ("  ██║╚██╗██║██║   ██║   ██║   ██╔══██╗██╔══██║██║     ██╔═██╗ ", "green"),
+        ("  ██║ ╚████║╚██████╔╝   ██║   ██║  ██║██║  ██║╚██████╗██║  ██╗", "green"),
+        ("  ╚═╝  ╚═══╝ ╚═════╝    ╚═╝   ╚═╝  ╚═╝╚═╝  ╚═╝ ╚═════╝╚═╝  ╚═╝", "dim green"),
+    ]
+    logo = Text()
+    for txt, style in lines:
+        logo.append(txt + "\n", style=style)
+    logo.append(
+        f"\n  Autonomous Master Agent  ·  v{VERSION}  ·  Termux & Linux\n",
+        style="bold white",
+    )
+    logo.append(
+        "  Type /help  ·  Ctrl+C to pause  ·  /exit to quit\n",
+        style="dim white",
+    )
+    console.print(Panel(logo, border_style="green", padding=(0, 1)))
+    console.print()
+
+
+# ──────────────────────────────────────────────────────────────
+#  YES / NO / ALWAYS / NEVER  (AGY-style confirmation)
+# ──────────────────────────────────────────────────────────────
+def ask_confirm(question: str, default=True) -> bool:
+    if state.always_yes:
+        console.print(f"[dim green]  ✔ Auto-yes:[/dim green] [dim]{question}[/dim]")
+        return True
+    if state.always_no:
+        console.print(f"[dim red]  ✗ Auto-no:[/dim red]  [dim]{question}[/dim]")
+        return False
+
+    hint = "[Y]es  [N]o  [A]lways  [never] (default: No)"
+    console.print(f"\n[bold white]❓  {question}[/bold white]")
+    console.print(f"[dim]    {hint}[/dim]")
+    try:
+        ans = console.input("[bold green]❯ [/bold green]").strip().lower()
+    except (KeyboardInterrupt, EOFError):
+        return False
+
+    if ans in ("y", "yes"):
+        return True
+    if ans in ("a", "always", "always yes"):
+        state.always_yes = True
+        console.print("[dim green]  ✔ 'Always yes' set for this session.[/dim green]")
+        return True
+    if ans in ("n", "no", "", "never"):
+        state.always_no = True
+        console.print("[dim red]  ✗ 'Never' set for this session.[/dim red]")
+        return False
+    return False
+
+
+# ──────────────────────────────────────────────────────────────
+#  STORAGE SELECTION MENU
+# ──────────────────────────────────────────────────────────────
+def choose_save_location(default_filename: str) -> str | None:
+    console.print("\n[bold white]📁  Where should the file be saved?[/bold white]")
+    options = [
+        ("1", os.path.expanduser("~"), "Termux / Linux home"),
+        ("2", "/sdcard",               "Android internal storage"),
+        ("3", state.cwd,               "Current working directory"),
+        ("4", "(custom)",              "Enter a custom path"),
+    ]
+    t = Table(show_header=False, box=None, padding=(0, 2))
+    for n, path, label in options:
+        t.add_row(f"[bold cyan]{n}[/bold cyan]", path, f"[dim]{label}[/dim]")
+    console.print(t)
+    try:
+        c = console.input("\n[bold green]❯ Select [1-4]:[/bold green] ").strip()
+    except (KeyboardInterrupt, EOFError):
+        return None
+
+    dirs = {"1": os.path.expanduser("~"), "2": "/sdcard", "3": state.cwd}
+    if c in dirs:
+        return os.path.join(dirs[c], default_filename)
+    if c == "4":
+        try:
+            p = console.input("[bold green]❯ Path:[/bold green] ").strip()
+            return p or None
+        except (KeyboardInterrupt, EOFError):
+            return None
+    return None
+
+
+# ──────────────────────────────────────────────────────────────
+#  CODE CREATION ANIMATION
+# ──────────────────────────────────────────────────────────────
+def animate_code_creation(path: str, content: str) -> str:
+    console.print()
+    console.print(Rule(f"[bold cyan]Writing File[/bold cyan]", style="cyan"))
+    console.print(f"[bold green]✦[/bold green] [bold white]{path}[/bold white]\n")
+
+    ext = os.path.splitext(path)[1].lower()
+    lang_map = {
+        ".py": "python",   ".js": "javascript", ".ts": "typescript",
+        ".json": "json",   ".sh": "bash",        ".html": "html",
+        ".css": "css",     ".md": "markdown",    ".yaml": "yaml",
+        ".yml": "yaml",    ".toml": "toml",      ".rs": "rust",
+        ".go": "go",       ".c": "c",            ".cpp": "cpp",
+        ".java": "java",   ".sql": "sql",        ".rb": "ruby",
+        ".php": "php",     ".kt": "kotlin",      ".swift": "swift",
+        ".dockerfile": "dockerfile",
+    }
+    lang = lang_map.get(ext, "text")
+
+    lines   = content.splitlines()
+    display = ""
+    with Live(refresh_per_second=24, console=console) as live:
+        for line in lines:
+            display += line + "\n"
+            syn = Syntax(
+                display, lang,
+                theme="monokai",
+                line_numbers=True,
+                word_wrap=True,
+            )
+            live.update(Panel(
+                syn,
+                title=f"[bold cyan]📄 {os.path.basename(path)}[/bold cyan]",
+                border_style="cyan",
+            ))
+            time.sleep(0.006)
+
+    result = LocalTools.write_file(path, content, cwd=state.cwd)
+    console.print(f"[bold green]✔[/bold green] {result}\n")
+    return result
+
+
+# ──────────────────────────────────────────────────────────────
+#  COMMAND EXECUTION (full shell, no restrictions)
+# ──────────────────────────────────────────────────────────────
+def show_command_exec(cmd: str, ask: bool = True) -> str:
+    if ask and not ask_confirm(f"Run: `{cmd}`?"):
+        return "Skipped by user."
+    console.print()
+    console.print(Rule(f"[bold yellow]Shell[/bold yellow]", style="yellow"))
+    console.print(f"[bold yellow]⚡[/bold yellow] [bold white]{cmd}[/bold white]\n")
+    with console.status("[bold green]Executing…[/bold green]", spinner="dots"):
+        result = LocalTools.run_command(cmd, cwd=state.cwd)
+    console.print(Panel(
+        result,
+        title="[bold yellow]Output[/bold yellow]",
+        border_style="yellow",
+    ))
+    return result
+
+
+# ──────────────────────────────────────────────────────────────
+#  INTERACTIVE SHELL MODE
+# ──────────────────────────────────────────────────────────────
+def interactive_shell():
+    console.print(Panel(
+        "[bold white]Interactive Shell Mode[/bold white]\n"
+        "[dim]All commands run in your terminal. Type [bold]exit[/bold] or Ctrl+D to return.[/dim]",
+        border_style="yellow",
+    ))
+    os.makedirs(os.path.expanduser("~/.notrack_cli"), exist_ok=True)
+    sh_hist = FileHistory(os.path.expanduser("~/.notrack_cli/shell_history"))
+    sh_sess = PromptSession(history=sh_hist, style=PT_STYLE)
+
+    while True:
+        try:
+            cwd_short = state.cwd.replace(os.path.expanduser("~"), "~")
+            cmd = sh_sess.prompt(
+                HTML(f'<ansiyellow><b>shell:{cwd_short}</b></ansiyellow><ansigreen><b> $ </b></ansigreen>')
+            ).strip()
+        except (KeyboardInterrupt, EOFError):
+            console.print("\n[dim]Returning to NoTrack Agent…[/dim]\n")
+            return
+
+        if not cmd:
+            continue
+        if cmd in ("exit", "quit"):
+            console.print("[dim]Returning to NoTrack Agent…[/dim]\n")
+            return
+
+        if cmd.startswith("cd "):
+            target = cmd[3:].strip().strip('"').strip("'")
+            target = os.path.expanduser(target)
+            if not os.path.isabs(target):
+                target = os.path.join(state.cwd, target)
+            if os.path.isdir(target):
+                state.cwd = os.path.realpath(target)
+                console.print(f"[dim green]{state.cwd}[/dim green]")
+            else:
+                console.print(f"[bold red]cd: no such directory: {target}[/bold red]")
+            continue
+
+        result = LocalTools.run_command(cmd, cwd=state.cwd)
+        if result.strip():
+            console.print(result)
+
+
+# ──────────────────────────────────────────────────────────────
+#  SUB-AGENT  (parallel NoTrack session)
+# ──────────────────────────────────────────────────────────────
+class SubAgent:
+    def __init__(self, task: str):
+        self.task   = task
+        self.result = None
+        self._t     = None
+
+    def _run(self):
+        client = NoTrackClient()
+        prompt = fit_dispatch_prompt(
+            f"{SYSTEM_PROMPT}\n## Sub-Agent Task\n{self.task}",
+            max_chars=3900,
+        )
+        full = ""
+        try:
+            for ev in client.send_message_stream(prompt):
+                et = ev.get("type")
+                if et == "message":
+                    full = ev.get("content", "")
+                elif et == "delta":
+                    full += ev.get("chunk", "")
+        except Exception as e:
+            full = f"[Sub-Agent Error]: {e}"
+        self.result = full
+
+    def start(self):
+        self._t = threading.Thread(target=self._run, daemon=True)
+        self._t.start()
+        return self
+
+    def join(self, timeout=180):
+        if self._t:
+            self._t.join(timeout=timeout)
+        return self.result or ""
+
+
+def dispatch_sub_agent(task: str) -> str:
+    console.print()
+    console.print(Rule("[bold magenta]Sub-Agent Spawned[/bold magenta]", style="magenta"))
+    console.print(f"[bold magenta]▶[/bold magenta] [white]{task[:140]}[/white]\n")
+    agent = SubAgent(task).start()
+    with console.status("[bold magenta]Sub-agent working…[/bold magenta]", spinner="aesthetic"):
+        result = agent.join()
+    console.print(Panel(
+        Markdown(result or "(no output)"),
+        title="[bold magenta]Sub-Agent Result[/bold magenta]",
+        border_style="magenta",
+    ))
+    return result
+
+
+# ──────────────────────────────────────────────────────────────
+#  TOOL EXECUTOR
+# ──────────────────────────────────────────────────────────────
+def execute_tool_call(tool_name: str, args: dict, client: NoTrackClient) -> str:
+    if tool_name == "write_file":
+        path    = args.get("path", "output.txt")
+        content = args.get("content", "")
+        if ask_confirm(f"Write file '{path}' ({len(content):,} bytes)?"):
+            return animate_code_creation(path, content)
+        return "Skipped by user."
+
+    elif tool_name == "run_command":
+        cmd = args.get("command", "")
+        return show_command_exec(cmd, ask=True)
+
+    elif tool_name == "view_file":
+        path = args.get("path", "")
+        console.print(f"\n[bold cyan]👁[/bold cyan]  [bold white]{path}[/bold white]")
+        result = LocalTools.view_file(path, cwd=state.cwd)
+        console.print(Panel(
+            result[:3000] + ("…" if len(result) > 3000 else ""),
+            title=f"[bold cyan]{os.path.basename(path)}[/bold cyan]",
+            border_style="cyan",
+        ))
+        return result
+
+    elif tool_name == "list_dir":
+        path = args.get("path", state.cwd)
+        console.print(f"\n[bold magenta]📁[/bold magenta]  [bold white]{path}[/bold white]")
+        result = LocalTools.list_dir(path, cwd=state.cwd)
+        console.print(Panel(result, title=path, border_style="magenta"))
+        return result
+
+    elif tool_name == "sub_agent":
+        return dispatch_sub_agent(args.get("task", ""))
+
+    # ── NOUVEAUX OUTILS ──────────────────────────────────────
+    elif tool_name == "fetch_url":
+        url = args.get("url", "")
+        console.print(f"\n[bold cyan]🌐[/bold cyan]  [bold white]Fetching {url}[/bold white]")
+        result = LocalTools.fetch_url(url)
+        console.print(Panel(
+            result[:1500] + ("…" if len(result) > 1500 else ""),
+            title="[bold cyan]Web Content[/bold cyan]",
+            border_style="cyan"
+        ))
+        return result
+
+    elif tool_name == "search_in_files":
+        query = args.get("query", "")
+        path = args.get("path", ".")
+        console.print(f"\n[bold magenta]🔍[/bold magenta]  [bold white]Searching for '{query}' in {path}[/bold white]")
+        result = LocalTools.search_in_files(query, path=path, cwd=state.cwd)
+        console.print(Panel(result, title="[bold magenta]Search Results[/bold magenta]", border_style="magenta"))
+        return result
+
+    elif tool_name == "get_system_info":
+        console.print(f"\n[bold green]💻[/bold green]  [bold white]Gathering system specs...[/bold white]")
+        result = LocalTools.get_system_info()
+        console.print(Panel(result, title="[bold green]System Info[/bold green]", border_style="green"))
+        return result
+    # ─────────────────────────────────────────────────────────
+
+    return f"[Unknown tool: {tool_name}]"
+
+
+# ──────────────────────────────────────────────────────────────
+#  SLASH COMMAND HANDLER
+# ──────────────────────────────────────────────────────────────
+def handle_slash(cmd_raw: str, history: list, client: NoTrackClient) -> bool:
+    parts = cmd_raw.strip().split(None, 1)
+    cmd   = parts[0].lower()
+    rest  = parts[1] if len(parts) > 1 else ""
+
+    if cmd == "/help":
+        t = Table(title="[bold green]Slash Commands[/bold green]",
+                  border_style="green", show_lines=False)
+        t.add_column("Command",     style="bold cyan", no_wrap=True)
+        t.add_column("Description", style="white")
+        for k, v in SLASH_COMMANDS.items():
+            t.add_row(k, v)
+        console.print(t)
+        return True
+
+    if cmd == "/clear":
+        print_banner()
+        return True
+
+    if cmd == "/chats":
+        try:
+            chats = client.list_chats()
+            if not chats:
+                console.print("[dim]No chats returned for this anonymous session.[/dim]")
+            else:
+                t = Table("Chat ID", "Title", "Messages", "Status")
+                for item in chats[:50]:
+                    t.add_row(str(item.get("chat_id", "")), str(item.get("title", "")),
+                              str(item.get("message_count", "?")), str(item.get("status", "")))
+                console.print(t)
+        except Exception as e:
+            console.print(f"[bold red]Could not load chats:[/bold red] {e}")
+        return True
+
+    if cmd == "/resume":
+        chat_id = rest.strip()
+        if not chat_id:
+            console.print("[yellow]Usage: /resume <chat_id>[/yellow]")
+        else:
+            client.chat_id = chat_id
+            console.print(f"[green]Resuming chat {chat_id}[/green]")
+        return True
+
+    if cmd == "/history":
+        if not history:
+            console.print("[dim]No history yet.[/dim]")
+            return True
+        for i, item in enumerate(history[-10:], 1):
+            console.print(f"[bold cyan]{i}.[/bold cyan] [bold white]You:[/bold white] "
+                          f"{str(item.get('user',''))[:90]}")
+            console.print(f"   [dim green]Agent:[/dim green] "
+                          f"{str(item.get('assistant',''))[:130]}…\n")
+        return True
+
+    if cmd == "/reset":
+        if ask_confirm("Clear ALL conversation history?"):
+            MemoryManager.clear_history()
+            history.clear()
+            client.new_chat()
+            console.print("[bold green]✔ History cleared.[/bold green]")
+        return True
+
+    if cmd == "/new":
+        client.new_chat()
+        console.print("[bold green]✔ New chat context started.[/bold green]")
+        return True
+
+    if cmd == "/model":
+        console.print(Panel(
+            f"[bold white]Package:[/bold white]  [bold cyan]{PYPI_PACKAGE}[/bold cyan]\n"
+            f"[bold white]Version:[/bold white]  [bold green]v{VERSION}[/bold green]\n"
+            f"[bold white]API:[/bold white]      [dim]notrack.ai/api/dispatch[/dim]\n"
+            f"[bold white]Mode:[/bold white]     [dim]Streaming · Autonomous · Multi-model fallback[/dim]\n"
+            f"[bold white]Chat ID:[/bold white]  [dim]{client.chat_id or 'none (new)'}[/dim]",
+            title="[bold green]Model Info[/bold green]",
+            border_style="green",
+        ))
+        return True
+
+    if cmd == "/tools":
+        t = Table(title="[bold cyan]Available Tools[/bold cyan]",
+                  border_style="cyan", show_lines=False)
+        t.add_column("Tool",        style="bold cyan")
+        t.add_column("Description", style="white")
+        rows = [
+            ("write_file",   "Create or overwrite any file (with animated preview)"),
+            ("view_file",    "Read any file on the system"),
+            ("run_command",  "Execute any shell command (git, pip, bash, etc.)"),
+            ("list_dir",     "List directory contents"),
+            ("sub_agent",    "Spawn a parallel AI sub-agent for complex sub-tasks"),
+            ("fetch_url",       "Download text content from a URL or documentation"),
+            ("search_in_files", "Search text across project files recursively (Grep)"),
+            ("get_system_info", "Return system specs, disk space, and OS details"),
+        ]
+        for r in rows:
+            t.add_row(*r)
+        console.print(t)
+        return True
+
+    if cmd == "/save":
+        if not history:
+            console.print("[dim]Nothing to save.[/dim]")
+            return True
+        fname = f"notrack_{datetime.now().strftime('%Y%m%d_%H%M%S')}.md"
+        path  = choose_save_location(fname)
+        if path:
+            md  = [f"# NoTrack CLI Export — {datetime.now().strftime('%Y-%m-%d %H:%M')}\n"]
+            for item in history:
+                md.append(f"**You:** {item.get('user','')}\n\n"
+                           f"**NoTrack:** {item.get('assistant','')}\n\n---")
+            try:
+                os.makedirs(os.path.dirname(os.path.abspath(path)), exist_ok=True)
+                with open(path, "w", encoding="utf-8") as f:
+                    f.write("\n".join(md))
+                console.print(f"[bold green]✔ Saved:[/bold green] {path}")
+            except Exception as e:
+                console.print(f"[bold red]Save failed:[/bold red] {e}")
+        return True
+
+    if cmd == "/shell":
+        interactive_shell()
+        return True
+
+    if cmd == "/git":
+        out = LocalTools.run_command("git status && git log --oneline -3", cwd=state.cwd)
+        console.print(Panel(out, title="[bold green]git[/bold green]", border_style="green"))
+        return True
+
+    if cmd == "/ls":
+        out = LocalTools.list_dir(rest.strip() or state.cwd, cwd=state.cwd)
+        console.print(Panel(out, title=rest.strip() or state.cwd, border_style="magenta"))
+        return True
+
+    if cmd == "/pwd":
+        console.print(f"[bold cyan]{state.cwd}[/bold cyan]")
+        return True
+
+    if cmd == "/cd":
+        if not rest:
+            target = os.path.expanduser("~")
+        else:
+            target = os.path.expanduser(rest.strip())
+            if not os.path.isabs(target):
+                target = os.path.join(state.cwd, target)
+        if os.path.isdir(target):
+            state.cwd = os.path.realpath(target)
+            console.print(f"[bold green]✔[/bold green] [dim]{state.cwd}[/dim]")
+        else:
+            console.print(f"[bold red]cd: not a directory: {target}[/bold red]")
+        return True
+
+    if cmd == "/version":
+        console.print(f"[bold green]NoTrack CLI[/bold green] "
+                      f"v[bold white]{VERSION}[/bold white]")
+        return True
+
+    if cmd in ("/exit", "/quit", "/q"):
+        console.print("[dim]Goodbye![/dim]")
+        sys.exit(0)
+
+    return False
+
+
+# ──────────────────────────────────────────────────────────────
+#  STREAMING RESPONSE
+# ──────────────────────────────────────────────────────────────
+def stream_response(client: NoTrackClient, prompt: str) -> str:
+    full = ""
+    try:
+        with Live(console=console, refresh_per_second=15) as live:
+            for ev in client.send_message_stream(prompt):
+                et = ev.get("type")
+                if et == "message":
+                    full = ev.get("content", "")
+                elif et == "delta":
+                    full += ev.get("chunk", "")
+                else:
+                    continue
+                live.update(Panel(
+                    Markdown(full),
+                    title="[bold green]NoTrack Agent[/bold green]",
+                    border_style="green",
+                    padding=(0, 1),
+                ))
+    except KeyboardInterrupt:
+        console.print(
+            "\n[bold yellow]⚡ Interrupted — partial response saved.[/bold yellow]"
+        )
+    except RuntimeError as e:
+        console.print(f"\n[bold red]Connection Error:[/bold red] {e}")
+    return full
+
+
+# ──────────────────────────────────────────────────────────────
+#  PROCESS TOOL CALLS + FEEDBACK LOOP (up to 8 levels deep)
+# ──────────────────────────────────────────────────────────────
+def run_tool_loop(full_response: str, client: NoTrackClient,
+                  history: list, depth: int = 0):
+    if depth > 8:
+        return
+    tool_calls = extract_tool_calls(full_response)
+    if not tool_calls:
+        return
+
+    for tool_name, args, _raw in tool_calls:
+        tool_output = execute_tool_call(tool_name, args, client)
+
+        feedback = (
+            f"## Tool Result\n"
+            f"**Tool:** `{tool_name}`\n"
+            f"**Output:**\n```\n{tool_output}\n```\n\n"
+            f"Continue completing the task based on this result."
+        )
+        console.print(
+            f"\n[dim]↩  Sending [bold cyan]{tool_name}[/bold cyan] "
+            f"output back to agent… (depth {depth+1})[/dim]\n"
+        )
+        feedback_resp = stream_response(client, feedback)
+        if feedback_resp:
+            history.append({
+                "user":      f"[tool result: {tool_name}]",
+                "assistant": feedback_resp,
+            })
+            MemoryManager.save_history(history)
+            run_tool_loop(feedback_resp, client, history, depth + 1)
+
+
+# ──────────────────────────────────────────────────────────────
+#  ONE-SHOT  `notrack ask "..."`
+# ──────────────────────────────────────────────────────────────
+def one_shot(question: str):
+    """Run a single question and exit. Used by `notrack ask '...'`"""
+    client  = NoTrackClient()
+    history = MemoryManager.load_history()
+    prompt  = f"{SYSTEM_PROMPT}\n\nUser: {question}"
+
+    console.print(
+        Panel(
+            f"[bold white]{question}[/bold white]",
+            title="[bold cyan]notrack ask[/bold cyan]",
+            border_style="cyan",
+        )
+    )
+    console.print()
+
+    response = stream_response(client, prompt)
+    console.print()
+
+    if response:
+        history.append({"user": question, "assistant": response})
+        MemoryManager.save_history(history)
+        run_tool_loop(response, client, history)
+
+
+# ──────────────────────────────────────────────────────────────
+#  INTERACTIVE MAIN LOOP
+# ──────────────────────────────────────────────────────────────
+def interactive_loop():
+    print_banner()
+    #threading.Thread(target=check_for_updates, daemon=True).start()
+
+    os.makedirs(os.path.expanduser("~/.notrack_cli"), exist_ok=True)
+    client  = NoTrackClient()
+    history = MemoryManager.load_history()
+
+    if history:
+        console.print(
+            f"[dim green]✔ Loaded {len(history)} turns from memory.[/dim green]\n"
+        )
+
+    # Checkpoint recovery
+    ckpt = MemoryManager.load_checkpoint()
+    if ckpt:
+        console.print("[bold yellow]⚠  Interrupted session detected.[/bold yellow]")
+        console.print(f"[dim]   Last prompt: {str(ckpt.get('user_input',''))[:80]}…[/dim]")
+        if ask_confirm("Resume checkpoint?"):
+            client.chat_id = ckpt.get("chat_id")
+        MemoryManager.clear_checkpoint()
+
+    # Prompt completer (slash commands)
+    completer = WordCompleter(
+        list(SLASH_COMMANDS.keys()), pattern=re.compile(r"/\w*"), sentence=True
+    )
+
+    session = PromptSession(
+        history=FileHistory(HISTORY_PATH),
+        style=PT_STYLE,
+        completer=completer,
+    )
+
+    while True:
+        cwd_short = state.cwd.replace(os.path.expanduser("~"), "~")
+        try:
+            user_input = session.prompt(
+                HTML(f'<ansigreen><b>❯ </b></ansigreen>'),
+            )
+        except (KeyboardInterrupt, EOFError):
+            console.print("\n[dim yellow]Session paused. /exit to quit.[/dim yellow]")
+            break
+
+        user_input = user_input.strip()
+        if not user_input:
+            continue
+
+        # Slash command
+        if user_input.startswith("/"):
+            if handle_slash(user_input, history, client):
+                continue
+            console.print(
+                f"[bold red]Unknown command:[/bold red] {user_input}  "
+                f"([bold cyan]/help[/bold cyan] for list)"
+            )
+            continue
+
+        # Quick exit
+        if user_input.lower() in ("exit", "quit", "q"):
+            console.print("[dim]Goodbye![/dim]")
+            break
+
+        # Build prompt avec SYSTEM_PROMPT garanti pour que l'agent voie toujours les outils
+        ctx = ""
+        if history:
+            ctx = "## Context (last 6 turns)\n"
+            for item in history[-6:]:
+                ctx += f"User: {item.get('user','')}\nAssistant: {item.get('assistant','')}\n---\n"
+
+        full_prompt = fit_dispatch_prompt(
+            f"{SYSTEM_PROMPT}\n\n{ctx}\nUser: {user_input}",
+            max_chars=3900,
+            preserve_tail=True,
+        )
+
+        # Save checkpoint before streaming
+        MemoryManager.save_checkpoint(user_input, "", client.chat_id)
+
+        console.print()
+        response = stream_response(client, full_prompt)
+        console.print()
+
+        if not response:
+            continue
+
+        history.append({"user": user_input, "assistant": response})
+        MemoryManager.save_history(history)
+        MemoryManager.clear_checkpoint()
+
+        # Tool call loop
+        run_tool_loop(response, client, history)
+
+
+# ──────────────────────────────────────────────────────────────
+#  ENTRY POINT
+# ──────────────────────────────────────────────────────────────
+def main():
+    parser = argparse.ArgumentParser(
+        prog="notrack",
+        description=f"NoTrack CLI v{VERSION} — Autonomous AI terminal",
+    )
+    sub = parser.add_subparsers(dest="subcommand")
+
+    ask_p = sub.add_parser("ask", help="One-shot: notrack ask 'your question'")
+    ask_p.add_argument("question", nargs="+", help="Question or task")
+
+    args = parser.parse_args()
+
+    if args.subcommand == "ask":
+        question = " ".join(args.question)
+        one_shot(question)
+    else:
+        interactive_loop()
+
+
+if __name__ == "__main__":
+    main()