Skip to content

yetone/smart-suggestion

Repository files navigation

Smart Suggestion for Zsh

Note

This project is a fork of zsh-copilot by Myzel394.

Get AI-powered command suggestions directly in your zsh shell. No complex setup, no external tools - just press CTRL + O and get intelligent command suggestions powered by OpenAI, Anthropic Claude, Google Gemini, or DeepSeek.

Note

This project is still in its early stages, and some features may be immature and unstable. I appreciate your understanding.

smart-suggestion.mp4
ffmpeg.mp4

Features

  • 🚀 Context-aware intelligent prediction: Predicts the next command you are likely to input based on context (history, aliases, terminal buffer)
  • 🤖 Multiple AI Providers: Support for OpenAI GPT, Anthropic Claude, Google Gemini, and DeepSeek
  • 🔧 Highly Configurable: Customize keybindings, AI provider, context sharing, and more

Questions

  • Why don't I use zsh-copilot and instead fork a separate version?

    Because the context of zsh-copilot only includes history commands and does not include the terminal buffer (i.e., the stdout/stderr of history commands), it cannot achieve the context-aware intelligent prediction I want, this is the feature I want the most, and it's also the main reason why I forked. Additionally, since zsh-copilot is written in shell, it's very difficult to concatenate JSON and implement stdio interception. Therefore, I re-implemented almost all logic using Go, which made it too different from the original project to merge back.

Installation

Prerequisites

Make sure you have the following installed:

Method 1: Quick Install (Recommended)

The easiest way to install smart-suggestion is using our installation script:

curl -fsSL https://raw.githubusercontent.com/yetone/smart-suggestion/main/install.sh | bash

This script will:

  • Detect your platform (Linux, macOS, Windows)
  • Download the appropriate pre-built binary
  • Install the plugin to ~/.config/smart-suggestion
  • Configure your ~/.zshrc automatically with proxy mode enabled by default
  • Check for zsh-autosuggestions dependency

Uninstall:

curl -fsSL https://raw.githubusercontent.com/yetone/smart-suggestion/main/install.sh | bash -s -- --uninstall

Method 2: Oh My Zsh

  1. Clone the repository into your Oh My Zsh custom plugins directory:
git clone https://github.com/yetone/smart-suggestion ${ZSH_CUSTOM:-~/.oh-my-zsh/custom}/plugins/smart-suggestion
  1. Add smart-suggestion to your plugins array in ~/.zshrc:
omz plugin enable smart-suggestion
  1. Build the Go binary:
cd ${ZSH_CUSTOM:-~/.oh-my-zsh/custom}/plugins/smart-suggestion
./build.sh
  1. Reload your shell:
source ~/.zshrc

Method 3: Zinit

  1. Add the following to your ~/.zshrc:
zinit as"program" atclone'./build.sh' \
    atpull'%atclone' pick"smart-suggestion" src"smart-suggestion.plugin.zsh" for \
        yetone/smart-suggestion
  1. Update Zinit:
zi update

Method 4: Manual Installation from Source

  1. Clone the repository:
git clone https://github.com/yetone/smart-suggestion ~/.config/smart-suggestion
  1. Build the Go binary (requires Go 1.21+):
cd ~/.config/smart-suggestion
./build.sh
  1. Add to your ~/.zshrc:
source ~/.config/smart-suggestion/smart-suggestion.plugin.zsh
  1. Reload your shell:
source ~/.zshrc

Method 5: Manual Installation from Release

  1. Download the latest release for your platform from GitHub Releases

  2. Extract the archive:

mkdir -p ~/.config/smart-suggestion
tar -xzf smart-suggestion-*.tar.gz -C ~/.config/smart-suggestion --strip-components=1
  1. Add to your ~/.zshrc:
source ~/.config/smart-suggestion/smart-suggestion.plugin.zsh
  1. Reload your shell:
source ~/.zshrc

Configuration

AI Provider Setup

You need an API key for at least one of the supported AI providers:

OpenAI (default)

export OPENAI_API_KEY="your-openai-api-key"

Azure OpenAI

export AZURE_OPENAI_API_KEY="your-azure-openai-api-key" # i.e. c0123456789012345678901234567890
export AZURE_OPENAI_RESOURCE_NAME="your-azure-openai-resource-name" # i.e. awesome-corp when your endpoint is https://awesome-corp.openai.azure.com
export AZURE_OPENAI_DEPLOYMENT_NAME="your-deployment-name" # i.e. gpt-4o
export AZURE_OPENAI_API_VERSION="2024-10-21"  # Optional, defaults to 2024-10-21
export AZURE_OPENAI_BASE_URL="https://your-azure-openai-base-url" # Optional, default to https://$AZURE_OPENAI_RESOURCE_NAME.openai.azure.com

Anthropic Claude

export ANTHROPIC_API_KEY="your-anthropic-api-key"

Google Gemini

export GEMINI_API_KEY="your-gemini-api-key"

DeepSeek

export DEEPSEEK_API_KEY="your-deepseek-api-key"

Environment Variables

Configure the plugin behavior with these environment variables:

Variable Description Default Options
SMART_SUGGESTION_AI_PROVIDER AI provider to use Auto-detected openai, azure_openai, anthropic, gemini, deepseek
SMART_SUGGESTION_KEY Keybinding to trigger suggestions ^o Any zsh keybinding
SMART_SUGGESTION_SEND_CONTEXT Send shell context to AI true true, false
SMART_SUGGESTION_PROXY_MODE Enable proxy mode for better context true true, false
SMART_SUGGESTION_DEBUG Enable debug logging false true, false
SMART_SUGGESTION_SYSTEM_PROMPT Custom system prompt Built-in Any string
SMART_SUGGESTION_AUTO_UPDATE Enable automatic update checking true true, false
SMART_SUGGESTION_UPDATE_INTERVAL Days between update checks 7 Any positive integer
SMART_SUGGESTION_BINARY Path to the smart_suggestion binary Auto-detected Any valid filepath to a valid smart_suggestion binary

If SMART_SUGGESTION_BINARY is not specified, we look for one in the following locations:

  1. smart_suggestion beside the current smart-suggestion.plugin.zsh
  2. ~/.config/smart-suggestion/smart_suggestion

Advanced Configuration

Custom API URLs

export OPENAI_BASE_URL="your-custom-openai-endpoint.com"
export AZURE_OPENAI_BASE_URL="your-custom-azure-openai-endpoint.com"
export ANTHROPIC_BASE_URL="your-custom-anthropic-endpoint.com"
export GEMINI_BASE_URL="your-custom-gemini-endpoint.com"
export DEEPSEEK_BASE_URL="your-custom-deepseek-endpoint.com"

Custom Models

export GEMINI_MODEL="gemini-1.5-pro"  # Default: gemini-1.5-flash

History Lines for Context

export SMART_SUGGESTION_HISTORY_LINES="20"  # Default: 10

View Current Configuration

To see all available configurations and their current values:

smart-suggestion

Usage

  1. Start typing a command or describe what you want to do
  2. Press CTRL + O (or your configured key)
  3. Wait for the AI suggestion (loading animation will show)
    • Note: On first use, proxy mode will automatically start in the background to capture terminal context
  4. The suggestion will appear as:
    • An autosuggestion you can accept with (for completions)
    • A completely new command that replaces your input (for new commands)

How It Works

  1. Input Capture: The plugin captures your current command line input
  2. Proxy Mode (Default): Automatically starts a background shell recording session to capture terminal output for better context
  3. Context Collection: Gathers rich shell context including user info, directory, command history, aliases, and terminal buffer content via proxy mode
  4. AI Processing: Sends the input and context to your configured AI provider
  5. Smart Response: AI returns either a completion (+) or new command (=)
  6. Shell Integration: The suggestion is displayed using zsh-autosuggestions or replaces your input

Proxy Mode (New Default)

Smart Suggestion now automatically enables proxy mode by default, which provides significantly better context awareness by recording your terminal session. This mode:

  • Automatically starts when you first use smart suggestions
  • Records terminal output using the script command for maximum compatibility
  • Provides rich context to the AI including command outputs and error messages
  • Works seamlessly across different terminal environments

You can disable proxy mode if needed:

export SMART_SUGGESTION_PROXY_MODE=false

For advanced proxy configuration, see PROXY_USAGE.md.

Troubleshooting

Debug Mode

Enable debug logging to troubleshoot issues:

export SMART_SUGGESTION_DEBUG=true

Debug logs are written to /tmp/smart-suggestion.log.

Common Issues

  1. "Binary not found" error: Run ./build.sh in the plugin directory
  2. No suggestions: Check your API key and internet connection
  3. Wrong suggestions: Try adjusting the context settings or system prompt
  4. Key binding conflicts: Change SMART_SUGGESTION_KEY to a different key

Build Issues

If the build fails:

# Check Go installation
go version

# Clean and rebuild
rm -f smart-suggestion
./build.sh

Contributing

Contributions are welcome! Please feel free to submit issues and pull requests.

License

This project is open source. Please check the repository for license details.

About

Get AI-powered command suggestions **directly** in your zsh shell.

Resources

Stars

Watchers

Forks

Packages

No packages published