Aizen AI Agent — A professional-grade AI coding assistant for your terminal.
Project description
Aizen AI Agent 🚀
A helpful AI coding assistant you can use right in your terminal. Aizen reads your code, edits files safely, runs commands, and helps you get things done faster — all with a friendly command‑line interface.
✨ Features
Core
- Asynchronous Architecture — Fully asynchronous operations leveraging
asyncioandAsyncOpenAIfor concurrent processing, parallel tool runs, and streaming. - Rich Markdown Rendering — AI responses are rendered with full Markdown formatting (headers, code blocks, lists, bold/italic) via Rich's live display.
- Streaming with Live Preview — Watch responses render in real-time inside a styled panel with an animated thinking spinner.
- Surgical File Editing — The
edit_filetool makes precise search-and-replace edits with color-coded diff previews, instead of rewriting entire files. - SQLite Session Persistence — Session storage is powered by a SQLite database (
~/.aizen_sessions/aizen.db), auto-migrating older JSON sessions. - Project-Specific Rules — Customizes agent behavior per repository by auto-loading
.aizen_rulesor.cursorrulesfrom the current working directory. - Smart Autocomplete —
@-mention files with Tab completion that respects.gitignoreand supports directory traversal.
Tools
Aizen has 9 built-in tools the AI can use:
| Tool | Description |
|---|---|
read_file |
Read file contents before making changes |
write_file |
Create new files or overwrite entirely (with preview) |
replace_file_content |
Surgical search-and-replace on existing files (with line-bounds and diff preview) |
multi_replace_file_content |
Perform multiple, non-adjacent surgical edits sequentially in a single pass |
run_command |
Execute shell commands (supports background execution; safe commands auto-run, dangerous ones require approval) |
check_background_task |
Check the status and read recent output of a command running in the background |
kill_background_task |
Kill a running background task |
list_directory |
List files/folders with sizes, respecting .gitignore |
grep_search |
Search for text or regex patterns across the codebase |
find_files |
Find files by glob pattern (e.g., *.py, Dockerfile) |
Commands
| Command | Description |
|---|---|
/help |
Show all available commands |
/model [name] |
View or switch the active AI model (saves as default) |
/clear |
Clear conversation history |
/drop |
Drop attached files/URLs/commands from history to save tokens |
/save [name] |
Save current conversation to SQLite database |
/load [name] |
Load a previously saved conversation |
/checkpoint [name] |
Save a conversation snapshot to memory |
/restore [name] |
Restore a saved conversation checkpoint |
/usage |
Show token usage, estimated session cost (USD), and statistics |
/commit |
Auto-generate a commit message for staged/unstaged changes and commit them |
/diff |
Show all uncommitted changes (staged, unstaged, untracked) |
/compact |
Summarize older messages using AI (fallback to text-summarization) to save tokens |
/undo |
Undo the last file modification |
/retry |
Retry the last message |
/copy |
Copy last AI response to clipboard |
/export [file] |
Export conversation to a Markdown file |
/config |
View current configuration |
/mcp |
View configured MCP servers and their connection status |
Safety & UX
- Command Safety — Read-only commands (
ls,cat,git status, etc.) auto-execute. Destructive commands (rm,sudo, etc.) always require confirmation. --yoloMode — Auto-approve all operations for power users.- Background Tasks — Run builds, tests, or other long-running tasks asynchronously while continuing to interact with Aizen.
- File Backups — Every file modification creates a backup. Use
/undoto restore. - Multi-line Input — End a line with
\to continue on the next line. - Session Persistence — Conversations auto-save on exit to SQLite. Use
/saveand/loadto manage. - Cost Tracking & Token Usage — Live tracking of input/output tokens, session duration, and estimated session cost in USD.
- Structured Logging — Rotated file logging at
~/.aizen_logs/aizen.logplus verbose console debugging logs via--verbose. - Graceful Error Recovery — Helpful hints for common API errors (invalid key, rate limits, timeouts).
Dependencies
openai— OpenAI-compatible API clientpython-dotenv— Environment variable managementrich— Rich text, Markdown rendering, panels, tables, and live displayprompt_toolkit— Interactive command line with autocomplete
Installation
1. Python (pip / pipx) — Recommended
pipx install aizen-ai-cli
# Or:
pip install aizen-ai-cli
2. NPM (Node.js)
npm install -g aizen-ai-cli
3. Homebrew (macOS)
brew tap irtaza302/aizen
brew install aizen
4. Local Development
git clone https://github.com/irtaza302/aizen-agent.git
cd aizen-agent
pip install -r requirements.txt
python aizen.py
Usage
aizen
On first launch, you'll be prompted for your OpenRouter API key. It's saved securely to ~/.aizen_config.json.
Command Line Arguments
| Flag | Description |
|---|---|
--version |
Show version |
--model <name> |
Override the default model for this session |
--reset-key |
Clear and re-enter your API key |
--set-base-url <url> |
Set custom API base URL (e.g., http://localhost:11434/v1 for Ollama) |
--yolo |
Auto-approve all file writes and command executions |
--verbose |
Enable verbose logging output to the console |
Attaching Context (@)
Type @ followed by a filename, directory, web URL, or command to give Aizen context. Autocomplete filters out .gitignored files:
- Files:
@aizen.pyattaches the file contents. - Directories:
@tests/generates and attaches a visual directory tree respecting.gitignore. - URLs:
@https://docs.python.org/...fetches the webpage, converts it to markdown, and attaches it. - Commands:
@cmd:"pytest"or@cmd:lssecurely runs the command in the background and injects itsstdoutandstderrdirectly into the prompt.
👤 You
❯ Can you refactor @aizen.py to use async?
👤 You
❯ Explain this output: @cmd:"npm run build"
Multi-line Input
End a line with \ to continue typing on the next line:
👤 You
❯ Write a function that \
⋮ takes a list of numbers \
⋮ and returns the sorted unique values
Configuration
Aizen stores its config in ~/.aizen_config.json:
{
"OPENROUTER_API_KEY": "sk-or-...",
"API_BASE_URL": "https://openrouter.ai/api/v1",
"DEFAULT_MODEL": "anthropic/claude-sonnet-4"
}
Model Context Protocol (MCP) Support
Aizen supports integrating with external Model Context Protocol (MCP) servers to extend its capabilities (e.g. connecting to local databases, searching the web, or accessing custom APIs).
To configure MCP servers, add an "mcp_servers" block to your ~/.aizen_config.json:
{
"mcp_servers": {
"sqlite": {
"command": "uvx",
"args": ["mcp-server-sqlite", "--db-path", "~/test.db"]
},
"everything": {
"command": "npx",
"args": ["-y", "@modelcontextprotocol/server-everything"]
}
}
}
When you start Aizen, it will automatically connect to these servers and make their tools available to the AI.
Sessions are saved in a SQLite database at ~/.aizen_sessions/aizen.db, and file backups are placed in ~/.aizen_backups/.
📂 Project-Specific Rules
Aizen supports loading custom, project-specific rules files (such as .aizen_rules or .cursorrules) from the root of your project directory.
When Aizen starts, it checks for these files in the current working directory in the following order:
.aizen_rules.cursorrules
If one is found, Aizen automatically appends its contents to the system prompt. This allows you to enforce codebase-specific styling guidelines, coding standards, or project rules without editing Aizen's global configuration.
🔄 Background Task Management
For long-running processes (e.g., running test suites, starting local dev servers, or building bundles), you can run commands in the background asynchronously:
- Aizen's
run_commandtool supports a booleanbackgroundparameter. If set totrue, the tool immediately returns a uniquetask_id(e.g.,bg_a1b2c3d4). - You can inspect the status and read the recent stdout/stderr output of a background task using the
check_background_tasktool. - You can terminate any active background task using the
kill_background_tasktool.
This allows you to continue discussing other topics or refactoring files with Aizen while your tests or builds run in parallel.
💰 Cost Tracking
Aizen dynamically estimates session costs in USD for known models based on token usage:
- Input and output tokens are tracked in real-time.
- The estimated session cost is displayed in the CLI status bar and summary tables (via the
/usagecommand). - The cost calculations support popular models from Anthropic (Claude 3.5/3.7 Sonnet, Opus, Haiku), Google (Gemini 2.5 Pro/Flash), and OpenAI (GPT-4o, o1, o3-mini).
📌 Session Checkpoints & Restoring
You can save and restore conversation snapshots at any point during your session:
/checkpoint [name]: Save the current conversation messages history as a named snapshot in memory./restore [name]: Revert the conversation history to the specified checkpoint. If run without a name, it lists all currently active checkpoints.
This is extremely useful when experimenting with different implementation approaches or when recovering from an unintended direction.
📝 Structured Logging
All internal activities, tool calls, and API events are written to a rotating file logger:
- Logs are located at
~/.aizen_logs/aizen.log. - Up to three rotated log files are kept (5 MB per file limit).
- You can run Aizen with the
--verboseflag to mirror log output directly to the console stderr stream.
Publishing & Development
Use the included publish.sh script to build and publish across all platforms (PyPI, NPM, and PyInstaller binaries).
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file aizen_ai_cli-2.2.5.tar.gz.
File metadata
- Download URL: aizen_ai_cli-2.2.5.tar.gz
- Upload date:
- Size: 60.8 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.14.3
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
0d50ecc9247315ae3a9d2cf1a5c2da416345571d209e0cc51cb5c4c276b5869e
|
|
| MD5 |
011d61e315ce8e720df22d822b758cc3
|
|
| BLAKE2b-256 |
e863e6bc5cec1267934c7b31b3d565d164e1b86f6730c8a582bfa319fd7817a3
|
File details
Details for the file aizen_ai_cli-2.2.5-py3-none-any.whl.
File metadata
- Download URL: aizen_ai_cli-2.2.5-py3-none-any.whl
- Upload date:
- Size: 51.2 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.14.3
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
2c61808eb79a7d0dac8e8a603c1641d17f3eca6598f4d5bed4154908fd4a7b55
|
|
| MD5 |
c1cab8ea50b2f262053df722c0dc1fd6
|
|
| BLAKE2b-256 |
a6b946204b0861805c8a87d7c74caa304647f4bb1d7d1f76327683ebc6d36d8b
|