aivim 0.1.0

AI-enhanced Vim-like text editor with code assistance and generation

AIVim - AI-Enhanced Text Editor

AIVim is an AI-enhanced version of Vim built in Python, offering intelligent code assistance and generation capabilities while maintaining the core modal editing experience. Combining the power of different AI models with familiar Vim interactions, AIVim helps developers understand, improve, and generate code more efficiently.

Features

Modal Editing

• Normal Mode: Navigate and edit text using Vim-style commands
• Insert Mode: Type and modify text directly
• Visual Mode: Select text for operations
• Command Mode: Enter commands with the : prefix

AI-Powered Assistance

• Code Explanation: Understand complex code with detailed explanations
• Code Improvement: Get AI-powered refactoring and optimization suggestions with diff-style presentation
• Code Generation: Generate new code based on natural language descriptions
• Custom AI Queries: Ask questions about your code and receive contextual answers
• Code Analysis: Get complexity analysis and bug detection for your code
• Multi-Provider Support: Choose between OpenAI, Anthropic Claude, or local LLM models

Editor Features

• Vim-Style Commands: Familiar commands like dd (delete line), yy (yank line), p (paste)
• Multi-Tab Interface: View and compare original and AI-improved code in tabs
• Version History: Automatic tracking of file changes with metadata
• File Backups: Automatic backup creation when applying improvements (with timestamps)
• Loading Indicators: Visual feedback when AI operations are in progress
• Line Numbers: Display line numbers for easier navigation and reference
• Status Line: Displays current mode, filename, cursor position, and tab information
• Syntax Highlighting: Basic syntax highlighting for improved code readability
• Animated Loading: Visual indicators during AI operations to show progress

Web Interface

• Browser Access: Access AIVim from a web browser via Flask
• Code Editor: Edit and modify code directly in the browser
• AI Operations: Access all AI capabilities through the web interface
• Results Display: View AI-generated output in a dedicated results area

AI Commands

AIVim provides the following AI-specific commands:

Command	Description
:explain <start> <end>	Get an explanation of lines start through end
:improve <start> <end>	Get improvement suggestions for lines start through end
:generate <line> <description>	Generate code at line based on the description
:analyze <start> <end>	Analyze code complexity and detect bugs in lines start through end
:ai <query>	Ask a custom question about the current file
:set model=<provider>	Set the AI model provider (openai, claude, local)

Tab Navigation Commands

Command	Description
:nexttab or :n	Switch to the next tab
:prevtab or :N	Switch to the previous tab
:tabnew	Create a new empty tab
:tabnew <filename>	Create a new tab and open the specified file
:tabclose	Close the current tab

Vim Commands Supported

Command	Description
dd	Delete current line
yy	Yank (copy) current line
p	Paste after cursor
P	Paste before cursor
o	Open new line below cursor and enter insert mode
O	Open new line above cursor and enter insert mode
x	Delete character under cursor
u	Undo last change
Ctrl+r	Redo
i	Enter insert mode
v	Enter visual mode
gg	Go to first line
G	Go to last line
$	Go to end of line
/pattern	Search forward for pattern
?pattern	Search backward for pattern
n	Find next search match
N	Find previous search match
:%s/old/new/g	Replace all occurrences of 'old' with 'new'
:<line_number>	Jump to line number (integer expected)
:$	Jump to last line of the file

Usage

Basic Editing

• Press i to enter insert mode, ESC to return to normal mode
• Use :w to save, :q to quit, :wq to save and quit
• Navigation works with arrow keys (primary) or h, j, k, l keys (alternative)
• Jump to specific lines with :1, :14 or :$ (last line)
• Search for text with /pattern (forward) or ?pattern (backward)
• Use n to find next match, N to find previous match
• Replace text with :%s/old/new/g syntax

AI Features

1. Navigate to the code you want to work with
2. In normal mode, type :explain 10 20 to explain lines 10-20
3. Use :improve 10 20 to get improvement suggestions:
   • A new tab opens showing a colorized diff view of proposed changes
   • Review the modifications in the diff view
   • To apply the changes, switch back to the original tab and confirm
   • A backup of the original file is automatically created with a timestamp
4. Generate code with :generate 5 "Create a function that calculates factorial"
5. Analyze code with :analyze 10 20 to identify complexity issues and potential bugs
6. Ask questions with :ai How does this algorithm work?
7. Switch between AI providers with :set model=openai, :set model=claude, or :set model=local

Installation

To install AIVim:

# Install from source
git clone https://github.com/yourusername/aivim.git
cd aivim
pip install -e .

# Run AIVim
python -m aivim.run_editor file.py

System-wide Installation

To make AIVim available as a system-wide command (aivim), follow these steps:

# For temporary use in current session
alias aivim='python -m aivim.run_editor'

# For permanent installation, add to your shell configuration
echo 'alias aivim="python -m aivim.run_editor"' >> ~/.bashrc
# Or for Zsh
echo 'alias aivim="python -m aivim.run_editor"' >> ~/.zshrc

# Apply changes without restarting the terminal
source ~/.bashrc  # or source ~/.zshrc

For a more robust system-wide installation, create a symbolic link:

# Create a symbolic link in a directory that's in your PATH
sudo ln -s "$(which python) $(pwd)/run_editor.py" /usr/local/bin/aivim
sudo chmod +x /usr/local/bin/aivim

# Now you can run AIVim from anywhere
aivim myfile.py

Configuration File

Instead of setting environment variables, you can create a configuration file for storing API keys:

# Create a config directory
mkdir -p ~/.config/aivim

# Create and edit the configuration file
cat > ~/.config/aivim/config.ini << EOF
[api_keys]
openai = your_openai_api_key_here
anthropic = your_anthropic_api_key_here
llama_model_path = /path/to/your/local/model.gguf

[settings]
default_model = openai
EOF

# Set proper permissions to protect your API keys
chmod 600 ~/.config/aivim/config.ini

AIVim will automatically check for this configuration file and use these settings in addition to any environment variables that are set.

Setting Up Local LLM Support

AIVim supports using local LLM models via llama-cpp-python:

# Install the local LLM package
pip install llama-cpp-python

# Download a small model using the provided script
python download_local_model.py --model tinyllama

# Or list available models
python download_local_model.py --list

# Once downloaded, you can use it by setting the model:
# Inside AIVim: :set model=local

The download_local_model.py script supports several small models suitable for local execution:

Model	Size	Description
tinyllama	~1GB	Small but capable model for code assistance
phi2	~1.7GB	Microsoft compact yet powerful model
stablelm	~1.5GB	Stability AI efficient language model

Web Interface

AIVim also includes a web interface for easy access:

# Start the web server
python main.py

# Access AIVim in your browser at:
# http://0.0.0.0:5000

The web interface provides access to all the AI-powered features through a more traditional web IDE experience. It communicates with the same AI services backend as the terminal-based editor.

Embedding AIVim

AIVim can be embedded into other Python applications:

from aivim import embed_editor

# Open a file in the embedded editor
embed_editor("path/to/file.py")

# Or create a new file
embed_editor()

Requirements

• Python 3.8 or higher
• OpenAI API key (set as OPENAI_API_KEY environment variable)
• Optional: Anthropic API key (set as ANTHROPIC_API_KEY environment variable)
• Optional: Local LLM model (set model path as LLAMA_MODEL_PATH environment variable)
• curses library (included with Python on most systems)

License

This project is licensed under the MIT License - see the LICENSE file for details.

Publishing to PyPI

AIVim can be published to PyPI using the GitHub Actions workflow. For detailed instructions, see docs/publishing.md.

# Install from PyPI (once published)
pip install aivim

# Run the editor
aivim myfile.py