Agent Brain
A RAG-based (Retrieval-Augmented Generation) document indexing and semantic search system for AI agents and applications. Agent Brain enables intelligent querying of documentation and source code using natural language.
Overview
Agent Brain provides AI-first document and code search through a Claude Code plugin with skills, commands, and agents. Use slash commands to search, agents for complex research tasks, and skills for intelligent query optimization.
| Component | Description |
|---|
| Plugin | 24 slash commands, 3 agents, 2 skills for Claude Code |
| Skills | Intelligent search mode selection and query optimization |
| Agents | Research assistant, search assistant, setup assistant |
| Server | FastAPI backend for indexing and retrieval |
| CLI | Command-line tool (also used by plugin internally) |
Quick Start (Claude Code Plugin)
1. Install the Plugin
claude plugins install github:SpillwaveSolutions/agent-brain
2. Set Up Your Project
In Claude Code, run:
/agent-brain-setup
This interactive wizard will:
- Install the Python packages (
agent-brain-rag, agent-brain-cli)
- Configure your API keys
- Initialize the project
- Start the server
- Index your documentation
3. Search with Commands
/agent-brain-search "how does authentication work"
That's it! The plugin handles everything automatically.
Plugin Commands
Search Commands
| Command | Description | Use When |
|---|
/agent-brain-search | Smart hybrid search (recommended) | General questions |
/agent-brain-semantic | Pure semantic/vector search | Conceptual queries |
/agent-brain-keyword | BM25 keyword search | Error messages, function names |
/agent-brain-hybrid | Hybrid with alpha tuning | Fine-tuned searches |
/agent-brain-graph | Knowledge graph search | "What calls X?", dependencies |
/agent-brain-multi | All modes combined (RRF) | Maximum recall |
Server Commands
| Command | Description |
|---|
/agent-brain-start | Start the server (auto-port) |
/agent-brain-stop | Stop the server |
/agent-brain-status | Check health and document count |
/agent-brain-index | Index documents or code |
Setup Commands
| Command | Description |
|---|
/agent-brain-setup | Complete guided setup wizard |
/agent-brain-install | Install pip packages |
/agent-brain-init | Initialize project directory |
/agent-brain-verify | Verify configuration |
/agent-brain-providers | Configure embedding/summarization providers |
Plugin Agents
Agent Brain includes three intelligent agents for complex tasks:
| Agent | Description | Triggered By |
|---|
| Search Assistant | Multi-step search across modes, synthesizes answers | "Find all references to...", "Research how..." |
| Research Assistant | Deep exploration with follow-up queries | "Investigate...", "Analyze the architecture of..." |
| Setup Assistant | Guided installation and troubleshooting | "Help me set up Agent Brain", configuration issues |
Example Agent Interaction
You: "Research how authentication is implemented across the codebase"
Research Assistant:
- Searches documentation for auth concepts
- Queries code for auth-related functions
- Uses graph mode to find dependencies
- Synthesizes comprehensive answer with references
Plugin Skills
Skills provide intelligent context to Claude for optimal searching:
| Skill | Purpose |
|---|
| using-agent-brain | Search mode selection, query optimization, API knowledge |
| configuring-agent-brain | Installation, provider configuration, troubleshooting |
When you ask about documentation or code, Claude automatically uses the skill to:
- Choose the best search mode for your query
- Set appropriate parameters (top_k, threshold, alpha)
- Interpret and synthesize results
Search Modes
| Mode | Best For | Example Query |
|---|
HYBRID | General questions (default) | "How does caching work?" |
VECTOR | Conceptual understanding | "Explain the architecture" |
BM25 | Exact terms, error codes | "NullPointerException", "getUserById" |
GRAPH | Relationships, dependencies | "What classes use AuthService?" |
MULTI | Comprehensive search | "Everything about data validation" |
Pluggable Providers
Agent Brain supports multiple providers for embeddings and summarization:
Embedding Providers
| Provider | Models | Local |
|---|
| OpenAI | text-embedding-3-large, text-embedding-3-small | No |
| Ollama | nomic-embed-text, mxbai-embed-large | Yes |
| Cohere | embed-english-v3.0, embed-multilingual-v3.0 | No |
SpillwaveSolutions