feat: Generate commands or ask questions with atuin ai

[BinaryMuse]

This PR refines the system created in #3178 to be suitable for a v1 release.

---

Overview

atuin-ai is a separate binary that allows for generating commands and asking questions from the command line.

It is fully opt-in.

Usage

atuin ai init will output bindings for your shell. Currently, bash, zsh, and fish are supported.

eval "$(atuin ai init)"

Once the hooks are installed, just press ? on an empty prompt line to call up the TUI.

atuin ai requires an account on Atuin Hub; you will be prompted to log in on first use.

Features

Command generation

Prompt the LLM to create a command, and get one back, no fuss. Press enter to run, or tab to insert.

┌Ask questions or generate a command:──────────────────────────┐
│                                                              │
│ > Get a list of running docker containers                    │
│                                                              │
├──────────────────────────────────────────────────────────────┤
│                                                              │
│ $ docker ps                                                  │
│                                                              │
└────[Enter]: Run  [Tab]: Insert  [f]: Follow-up  [Esc]: Cancel┘

Follow-up

You can follow-up with f to specify a refinement prompt to update the command that will be inserted.

┌Ask questions or generate a command:──────────────────────────┐
│                                                              │
│ > Get a list of running docker containers                    │
│                                                              │
├──────────────────────────────────────────────────────────────┤
│                                                              │
│ $ docker ps                                                  │
│                                                              │
├──────────────────────────────────────────────────────────────┤
│                                                              │
│ > Actually I want to get all docker containers               │
│                                                              │
├──────────────────────────────────────────────────────────────┤
│                                                              │
│ $ docker ps -a                                               │
│                                                              │
└────[Enter]: Run  [Tab]: Insert  [f]: Follow-up  [Esc]: Cancel┘

You can also follow-up with questions to get responses in natural language.

┌Ask questions or generate a command:──────────────────────────┐
│                                                              │
│ > Get a list of running docker containers                    │
│                                                              │
├──────────────────────────────────────────────────────────────┤
│                                                              │
│ $ docker ps                                                  │
│                                                              │
├──────────────────────────────────────────────────────────────┤
│                                                              │
│ > Actually I want to get all docker containers               │
│                                                              │
├──────────────────────────────────────────────────────────────┤
│                                                              │
│ $ docker ps -a                                               │
│                                                              │
├──────────────────────────────────────────────────────────────┤
│                                                              │
│ > What other useful flags to `docker ps` should I know?      │
│                                                              │
├──────────────────────────────────────────────────────────────┤
│                                                              │
│   Here are some handy `docker ps` flags:                     │
│                                                              │
│   - `-q` — Only show container IDs (great for piping to      │
│   other commands)                                            │
│   - `-s` — Show container sizes                              │
│   - `-n 5` — Show the last 5 created containers              │
│   - `-l` — Show only the latest created container            │
│   - `--no-trunc` — Don't truncate output (shows full IDs and │
│   commands)                                                  │
│   - `-f` or `--filter` — Filter by condition, e.g.:          │
│     - `-f status=exited` — only exited containers            │
│     - `-f name=myapp` — filter by name                       │
│     - `-f ancestor=nginx` — filter by image                  │
│   - `--format` — Custom output using Go templates, e.g.:     │
│     `--format "table {{.Names}}\t{{.Status}}\t{{.Ports}}"`   │
│                                                              │
│   A common combo is `docker ps -aq` to get all container     │
│   IDs, useful for bulk operations like `docker rm $(docker   │
│   ps -aq)`.                                                  │
│                                                              │
└────[Enter]: Run  [Tab]: Insert  [f]: Follow-up  [Esc]: Cancel┘

You can use enter or tab at any time to run or insert the last suggested command, even if it was suggested in a previous turn.

Conversational and search usage

If you prompt the LLM with a question that doesn't imply you want to generate a command, it can respond in natural language, and use web search if necessary to fetch the data it needs.

┌Ask questions or generate a command:──────────────────────────┐
│                                                              │
│ > What is the latest version of atuin?                       │
│                                                              │
├──────────────────────────────────────────────────────────────┤
│                                                              │
│ ✓ Used 2 tools                                               │
│                                                              │
│   The latest version of Atuin is **v18.12.0**, available on  │
│   the [GitHub releases                                       │
│   page](https://github.com/atuinsh/atuin/releases).          │
│                                                              │
└─────────────────────────────────[f]: Follow-up  [Esc]: Cancel┘

Dangerous or low-confidence command detection

The LLM scores its confidence in the command, as well as how dangerous the command is. This information is shown if a threshold is exceeded, and requires an extra confirmation step before running automatically with enter.

The Atuin Hub server also monitors suggested commands for dangerous patterns the LLM didn't catch, and appends its own assessment at the end of the LLM's own assessment.

┌Ask questions or generate a command:──────────────────────────┐
│                                                              │
│ > Delete all files from $HOME                                │
│                                                              │
├──────────────────────────────────────────────────────────────┤
│                                                              │
│ $ rm -rf $HOME/*                                             │
│                                                              │
│ ! ⚠️  This will PERMANENTLY delete ALL files and directories │
│   in your home directory, including documents, downloads,    │
│   configurations, SSH keys, and everything else. This is     │
│   irreversible and will likely break your system. Also note  │
│   this won't delete hidden (dot) files — if you want those   │
│   too, that's even more destructive.; [Server] Recursive     │
│   delete of critical directory                               │
│                                                              │
└────[Enter]: Run  [Tab]: Insert  [f]: Follow-up  [Esc]: Cancel┘

[greptile-apps]

Greptile Summary

Introduces atuin-ai, a TUI for LLM-powered command generation and Q&A. Fully opt-in feature accessible via ? on empty prompt after running eval "$(atuin ai init)".

Key additions:

• SSE-based streaming chat with conversational follow-ups
• Dangerous command detection with double-enter confirmation
• Shell integrations for bash, zsh, fish with proper command execution vs. insertion
• Pure functional rendering with dynamic viewport sizing
• Comprehensive state logging for debugging

Important Files Changed

Filename	Overview
crates/atuin-ai/src/tui/state.rs	Domain state types for TUI application with conversation event tracking and generation lifecycle management
crates/atuin-ai/src/commands/inline.rs	Main TUI runner with SSE streaming, authentication, and shell result emission
crates/atuin-ai/src/commands/init.rs	Shell integration generators for zsh, bash, and fish with ? keybindings
crates/atuin-ai/src/tui/app.rs	Keyboard event handler with dangerous command confirmation flow
crates/atuin-ai/src/tui/render.rs	Pure rendering logic with dynamic height calculation and markdown support

_{Last reviewed commit: fff2360}

[greptile-apps]

_{25 files reviewed, 1 comment}

_{Edit Code Review Agent Settings | Greptile}

[BinaryMuse]

@greptileai

[greptile-apps]

_{25 files reviewed, no comments}

_{Edit Code Review Agent Settings | Greptile}

[BinaryMuse]

@greptileai

[greptile-apps]

_{32 files reviewed, no comments}

_{Edit Code Review Agent Settings | Greptile}