Metadata-Version: 2.4
Name: code-rag-mcp
Version: 0.3.3
Summary: MCP server for efficient code search
Author-email: Duc Nguyen <duc@qduc.me>
License: MIT License
        
        Copyright (c) 2025 Duc Nguyen
        
        Permission is hereby granted, free of charge, to any person obtaining a copy
        of this software and associated documentation files (the "Software"), to deal
        in the Software without restriction, including without limitation the rights
        to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
        copies of the Software, and to permit persons to whom the Software is
        furnished to do so, subject to the following conditions:
        
        The above copyright notice and this permission notice shall be included in all
        copies or substantial portions of the Software.
        
        THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
        IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
        FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
        AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
        LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
        OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
        SOFTWARE.
        
Project-URL: Homepage, https://github.com/qduc/code-rag
Project-URL: Issues, https://github.com/qduc/code-rag/issues
Keywords: rag,code-search,embeddings,semantic-search,mcp
Classifier: Development Status :: 3 - Alpha
Classifier: Intended Audience :: Developers
Classifier: License :: OSI Approved :: MIT License
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.8
Classifier: Programming Language :: Python :: 3.9
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Classifier: Topic :: Software Development :: Libraries :: Python Modules
Requires-Python: <3.14,>=3.10
Description-Content-Type: text/markdown
License-File: LICENSE
Requires-Dist: chromadb>=1.3.5
Requires-Dist: qdrant-client>=1.7.0
Requires-Dist: python-dotenv>=0.21.0
Requires-Dist: mcp>=0.9.0
Requires-Dist: tree-sitter>=0.22.0
Requires-Dist: tree-sitter-python>=0.21.0
Requires-Dist: tree-sitter-javascript>=0.21.0
Requires-Dist: tree-sitter-typescript>=0.21.0
Requires-Dist: tree-sitter-go>=0.21.0
Requires-Dist: tree-sitter-rust>=0.21.0
Requires-Dist: tree-sitter-java>=0.21.0
Requires-Dist: tree-sitter-cpp>=0.21.0
Requires-Dist: tree-sitter-c>=0.21.0
Requires-Dist: fastapi>=0.104.0
Requires-Dist: uvicorn>=0.24.0
Requires-Dist: requests>=2.31.0
Provides-Extra: local
Requires-Dist: sentence-transformers>=2.2.0; extra == "local"
Requires-Dist: einops>=0.6.0; extra == "local"
Provides-Extra: cloud
Requires-Dist: openai>=1.50.0; extra == "cloud"
Requires-Dist: litellm>=1.0.0; extra == "cloud"
Provides-Extra: reranker
Requires-Dist: sentence-transformers>=2.2.0; extra == "reranker"
Provides-Extra: all
Requires-Dist: sentence-transformers>=2.2.0; extra == "all"
Requires-Dist: einops>=0.6.0; extra == "all"
Requires-Dist: openai>=1.50.0; extra == "all"
Requires-Dist: litellm>=1.0.0; extra == "all"
Provides-Extra: dev
Requires-Dist: pytest>=7.0.0; extra == "dev"
Requires-Dist: flake8>=6.0.0; extra == "dev"
Requires-Dist: black>=23.0.0; extra == "dev"
Requires-Dist: isort>=5.12.0; extra == "dev"
Requires-Dist: twine>=4.0.0; extra == "dev"
Requires-Dist: build>=1.0.0; extra == "dev"
Requires-Dist: pytest-asyncio>=0.21.0; extra == "dev"
Requires-Dist: pre-commit>=3.6.0; extra == "dev"
Dynamic: license-file

# Code-RAG

[![PyPI version](https://img.shields.io/pypi/v/code-rag-mcp.svg)](https://pypi.org/project/code-rag-mcp/)
[![Build Status](https://github.com/qduc/code-rag/actions/workflows/ci.yml/badge.svg)](https://github.com/qduc/code-rag/actions)
[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
[![Python Versions](https://img.shields.io/pypi/pyversions/code-rag-mcp.svg)](https://pypi.org/project/code-rag-mcp/)

**Semantic code search for your entire codebase.** Ask questions in plain English, get relevant code snippets with source locations.

Instead of grepping for function names, ask "authentication logic" and find all related auth code across your project.

## Table of Contents

- [Quick Start](#quick-start)
- [Claude's Opinion About Code-RAG](#claudes-opinion-about-code-rag-take-it-with-a-huge-grain-of-salt)
- [Why Use Code-RAG?](#why-use-code-rag)
- [Use with Claude Code (MCP Integration)](#use-with-claude-code-mcp-integration)
- [Configuration](#configuration)
- [How It Works](#how-it-works)
- [API Usage](#api-usage)
- [Supported Languages](#supported-languages)
- [Requirements](#requirements)
- [Troubleshooting](#troubleshooting)
- [Development](#development)
- [Contributing](#contributing)

## Quick Start

### 1. Install

**Using [uv](https://github.com/astral-sh/uv) (recommended):**
```bash
uvx --from code-rag-mcp code-rag-setup --install
```
*Note: This ensures your configuration persists even after updates.*

**Using pip:**
```bash
pip install code-rag-mcp && code-rag-setup
```

**New to Python?** Use the one-command installer:
```bash
curl -sSL https://raw.githubusercontent.com/qduc/code-rag/main/scripts/install.sh | bash
```

### 2. Add to Claude

```bash
claude mcp add code-rag -- code-rag-mcp
```

That's it! Claude can now search your codebase semantically.

### 3. Try it out (optional)

Test with the CLI before using with Claude:
```bash
code-rag-cli --path /path/to/your/project
```

## Claude's Opinion About Code-RAG (take it with a huge grain of salt!)

<details>
  <summary>I have taken a lot of salt, show me!</summary>

## Code-RAG Review

I ran 9 diverse semantic queries to test the tool across different aspects of the codebase. Here's my assessment:

### Strengths

1. Excellent Semantic Understanding
- Query "shared embedding server lifecycle management" correctly identified the core design documentation and implementation (0.54 relevance)
- "syntax-aware chunking for different languages" found both the implementation guide and actual code
- Searches understood concepts, not just keywords - found relevant code even when exact terms didn't match

2. Good Relevance Scoring
- Scores generally correlate well with actual relevance (0.4-0.6+ range for good matches)
- Top results consistently more relevant than lower-ranked ones
- The tool correctly prioritizes implementation code over test fixtures

3. Metadata Quality
- Results include file paths, line numbers, and function/class context
- Example: src/code_rag/api.py:443-462 incremental_reindex() | CodeRAGAPI (0.61)
- This makes it easy to navigate to the exact location

4. Handles Complex Queries Well
- Multi-concept queries like "incremental reindexing detect modified files" found the right code
- Domain-specific terminology works well ("embedding model configuration", "MCP server tools")

### Weaknesses

1. Generic Queries Return Scattered Results
- "error handling and retry logic" (query 5) returned less focused results
- Found timeout handlers, error printing, but no cohesive retry logic
- This is expected for overly broad queries, but worth noting

2. Some Test Code Noise
- "database initialization and connection" returned test fixtures as top results instead of actual implementation
- The mock database in test_mcp_server.py ranked higher (0.45) than real database code (0.35)
- Could be improved with result filtering or boosting production code

3. Limited Context Window
- Results show ~20-30 lines per chunk, which is reasonable but sometimes cuts off mid-implementation
- The expand_context parameter helps but isn't always used

### Specific Observations

#### What Works Great:
- Architecture/design queries: "shared embedding server lifecycle management" → perfect results
- Implementation queries: "syntax-aware chunking" → found both docs and code
- Specific subsystem queries: "incremental reindexing" → found the exact functions

#### What Could Be Better:
- Very broad queries ("error handling") need more filtering
- Test vs. production code disambiguation
- Maybe add file type boosting (prefer src/ over tests/)

### Recommendations

1. Add query guidance in the tool description - Users should know that specific queries work better than generic ones (you already have this in the MCP tool description, which is good)
2. Consider result filtering options - Allow excluding test files or prioritizing certain directories
3. Expand context by default - The expand_context parameter could be enabled by default for better chunk visibility
4. Add result grouping - Group results by file to show "This file has 3 relevant chunks"

### Overall Assessment

**Rating: 8.5/10**

The semantic search works remarkably well for its intended purpose. It successfully finds relevant code based on conceptual queries, not just keyword matching. The relevance scoring is solid, and the metadata makes results actionable.

The main improvement areas are around filtering test code and handling overly broad queries. For a developer using this tool, the key insight is: be specific in your queries. "authentication token refresh logic" will work better than just "authentication."

This is a genuinely useful tool that would save significant time when exploring unfamiliar codebases.
</details>

## Why Use Code-RAG?

- **Understand unfamiliar codebases** - Ask questions instead of reading everything
- **Find examples** - "error handling with retries" finds all relevant patterns
- **Refactoring aid** - Locate all code related to a feature you're changing
- **Documentation** - Extract context for writing docs or onboarding

## Use with Claude Code (MCP Integration)

Code-RAG works as an MCP server, letting Claude automatically search your codebase during conversations.

> **Note on `uv`:** Many examples below use [uv](https://github.com/astral-sh/uv) (specifically `uvx`) for fast, zero-config execution. If you don't have `uv` installed, you can use standard `pip` or `npx` (if using a wrapper).

### Quick Setup

**Option 1: Using uvx (Recommended)**
```bash
# Install uv first: https://github.com/astral-sh/uv

# Claude Code
claude mcp add code-rag --transport stdio uvx code-rag-mcp
```

**Option 2: Using pip (Standard)**
```bash
# Install in your environment
pip install code-rag-mcp

# Register with Claude Code using the absolute path to the binary
claude mcp add code-rag --transport stdio $(which code-rag-mcp)
```

**Option 3: Local development installation**
```bash
# Clone and install
git clone https://github.com/qduc/code-rag.git
cd code-rag
python -m venv .venv
source .venv/bin/activate
pip install -e .

# Register with Claude Code
claude mcp add code-rag --transport stdio $(which code-rag-mcp)
```

### Configuration

The MCP server reads configuration from environment variables or config files. Configure via your MCP client's settings:

**Claude Desktop (`~/Library/Application Support/Claude/claude_desktop_config.json`)**:
```json
{
  "mcpServers": {
    "code-rag": {
      "command": "uvx",
      "args": ["code-rag-mcp"],
      "env": {
        "CODE_RAG_EMBEDDING_MODEL": "nomic-ai/CodeRankEmbed",
        "CODE_RAG_DATABASE_TYPE": "chroma",
        "CODE_RAG_RERANKER_ENABLED": "true"
      }
    }
  }
}
```

**Claude Code (via claude mcp add)**:
```bash
# Basic setup
claude mcp add code-rag --transport stdio uvx code-rag-mcp

# Then configure via environment variables or config files
```

**Common Configuration Options**:
- `CODE_RAG_EMBEDDING_MODEL` - Embedding model (default: `nomic-ai/CodeRankEmbed`)
  - `nomic-ai/CodeRankEmbed` - Code-optimized, runs locally, requires GPU for best performance
  - `text-embedding-3-small` - OpenAI embeddings, no GPU required (requires `OPENAI_API_KEY`)
- `CODE_RAG_DATABASE_TYPE` - Database backend: `chroma` or `qdrant` (default: `chroma`)
- `CODE_RAG_CHUNK_SIZE` - Chunk size in characters (default: `1024`)
- `CODE_RAG_RERANKER_ENABLED` - Enable result reranking, may yield better results but slower (default: `false`)
- `CODE_RAG_SHARED_SERVER` - Share embedding server across instances, reduce memory footprint (default: `true`)

**Example with OpenAI embeddings**:
```json
{
  "mcpServers": {
    "code-rag": {
      "command": "uvx",
      "args": ["code-rag-mcp"],
      "env": {
        "CODE_RAG_EMBEDDING_MODEL": "text-embedding-3-small",
        "OPENAI_API_KEY": "sk-...",
        "CODE_RAG_RERANKER_ENABLED": "true"
      }
    }
  }
}
```

### Usage

Once configured, Claude can automatically search your codebase:

```
You: "Find the database connection logic"

Claude: [Automatically searches and finds the code]
        "I found the database connection logic in src/code_rag/db/connection.py..."
```

See [docs/mcp.md](docs/mcp.md) for detailed setup and troubleshooting.

## Basic Usage

```bash
# Different codebase
code-rag-cli --path /path/to/repo

# Force reindex
code-rag-cli --reindex

# More results
code-rag-cli --results 10

# Different embedding model (better for code)
code-rag-cli --model text-embedding-3-small # need to set OPENAI_API_KEY env

# Use Qdrant instead of ChromaDB
code-rag-cli --database qdrant
```

## Configuration

### Priority Order

Configuration is loaded in this order (higher priority overrides lower):

1. **Environment variables** (highest priority)
2. Custom config file via `CODE_RAG_CONFIG_FILE` environment variable
3. Project config: `./code-rag.config`
4. User config: `~/.config/code-rag/config` (auto-created with defaults)

**For MCP servers**: Set environment variables in your MCP client config (see MCP Integration section above).

**For CLI usage**: Use environment variables or config files.

### Environment Variables

```bash
# Use code-optimized embeddings (recommended)
export CODE_RAG_EMBEDDING_MODEL="nomic-ai/CodeRankEmbed"

# Or OpenAI embeddings
export OPENAI_API_KEY="sk-..."
export CODE_RAG_EMBEDDING_MODEL="text-embedding-3-small"

# Use Qdrant
export CODE_RAG_DATABASE_TYPE="qdrant"

# Adjust chunk size
export CODE_RAG_CHUNK_SIZE="2048"

# Enable reranking for better results
export CODE_RAG_RERANKER_ENABLED="true"

# Add custom ignore patterns (comma-separated)
export CODE_RAG_ADDITIONAL_IGNORE_PATTERNS="*.tmp,*.bak,logs/"
```

### Supported Cloud Providers

Code-RAG supports various cloud embedding providers via [LiteLLM](https://docs.litellm.ai/docs/providers). Set `CODE_RAG_EMBEDDING_MODEL` to the provider-specific model name and provide the necessary credentials:

| Provider | Model Example | Required Environment Variables |
|----------|---------------|--------------------------------|
| **OpenAI** | `text-embedding-3-small` | `OPENAI_API_KEY` |
| **Azure OpenAI** | `azure/text-embedding-3-small` | `AZURE_API_KEY`, `AZURE_API_BASE`, `AZURE_API_VERSION` |
| **Google Vertex AI** | `vertex_ai/text-embedding-004` | `VERTEX_AI_PROJECT`, `VERTEX_AI_LOCATION`, plus `gcloud auth application-default login` |
| **Cohere** | `cohere/embed-english-v3.0` | `COHERE_API_KEY` |
| **AWS Bedrock** | `bedrock/amazon.titan-embed-text-v1` | `AWS_ACCESS_KEY_ID`, `AWS_SECRET_ACCESS_KEY`, `AWS_REGION_NAME` |

For other providers (HuggingFace, Mistral, etc.), refer to the [LiteLLM documentation](https://docs.litellm.ai/docs/providers) for model names and required environment variables.


### Config File Format

Config files use the same format (key=value):

```bash
# ~/.config/code-rag/config or ./code-rag.config
CODE_RAG_EMBEDDING_MODEL=nomic-ai/CodeRankEmbed
CODE_RAG_DATABASE_TYPE=chroma
CODE_RAG_CHUNK_SIZE=1024
CODE_RAG_RERANKER_ENABLED=false
```

Full configuration options in [docs/IMPLEMENTATION.md](docs/IMPLEMENTATION.md#configuration-system).

## How It Works

1. **Scans** your codebase (respects `.gitignore`)
2. **Chunks** code intelligently (syntax-aware for Python, JS, Go, Rust, Java, C/C++)
3. **Embeds** chunks as vectors using ML models
4. **Stores** in vector database (ChromaDB or Qdrant)
5. **Searches** semantically when you query

Pluggable architecture - swap databases, embedding models, or add new ones.

## API Usage

Use programmatically:

```python
from code_rag.api import CodeRAGAPI

api = CodeRAGAPI(database_type="chroma", embedding_model="all-MiniLM-L6-v2")
api.initialize_collection("myproject")

# Index
chunks = api.index_codebase("/path/to/project")

# Search
results = api.search("authentication logic", n_results=5)
for r in results:
    print(f"{r['file_path']} - {r['similarity']:.2f}")
```

## Documentation

- **[AGENTS.md](AGENTS.md)** - Developer onboarding and architecture overview
- **[docs/IMPLEMENTATION.md](docs/IMPLEMENTATION.md)** - Detailed implementation reference
- **[docs/mcp.md](docs/mcp.md)** - MCP server setup guide

## Supported Languages

Syntax-aware chunking for: Python, JavaScript, TypeScript, Go, Rust, Java, C, C++

Other languages use line-aware chunking (still works, just less context-aware).

## Requirements

- Python 3.10+
- Minimal dependencies (ChromaDB + sentence-transformers by default)
- Optional: OpenAI API key, Qdrant server

## Troubleshooting

**Import errors?** `pip install --force-reinstall --upgrade code-rag-mcp` (or `pip install -e .` if developing locally)

**Database issues?** `code-rag-cli --reindex`

**Memory issues?** `export CODE_RAG_BATCH_SIZE="16"`

## Development

### Setup

```bash
# Install with dev dependencies
pip install -e ".[dev]"
```

### Testing & Linting

```bash
# Run tests
pytest

# Format code
black .
isort .

# Linting
flake8
```

## Contributing

1. Fork the repo
2. Create feature branch
3. Make changes
4. Add tests
5. Submit PR

See [AGENTS.md](AGENTS.md) for architecture and [docs/IMPLEMENTATION.md](docs/IMPLEMENTATION.md) for internals.

## License

MIT License. See [LICENSE](LICENSE) for details.

---

Built with [ChromaDB](https://www.trychroma.com/), [Qdrant](https://qdrant.tech/), [sentence-transformers](https://www.sbert.net/), and [Tree-sitter](https://tree-sitter.github.io/)
