Metadata-Version: 2.4
Name: local-brain
Version: 0.5.0
Summary: Chat with local Ollama models that can explore your codebase
Project-URL: Homepage, https://github.com/IsmaelMartinez/local-brain
Project-URL: Repository, https://github.com/IsmaelMartinez/local-brain
Project-URL: Documentation, https://github.com/IsmaelMartinez/local-brain#readme
Author-email: Ismael Martinez <ismaelmartinez@gmail.com>
License: MIT
License-File: LICENSE
Keywords: ai,cli,llm,ollama,smolagents,tool-calling
Classifier: Development Status :: 4 - Beta
Classifier: Environment :: Console
Classifier: Intended Audience :: Developers
Classifier: License :: OSI Approved :: MIT License
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Classifier: Programming Language :: Python :: 3.13
Classifier: Topic :: Software Development :: Quality Assurance
Requires-Python: <3.14,>=3.10
Requires-Dist: click>=8.0.0
Requires-Dist: litellm>=1.0.0
Requires-Dist: ollama>=0.6.1
Requires-Dist: smolagents>=1.0.0
Provides-Extra: dev
Requires-Dist: mypy>=1.0.0; extra == 'dev'
Requires-Dist: pytest-cov>=4.0.0; extra == 'dev'
Requires-Dist: pytest>=7.0.0; extra == 'dev'
Requires-Dist: ruff>=0.1.0; extra == 'dev'
Provides-Extra: tracing
Requires-Dist: openinference-instrumentation-smolagents>=0.1.20; extra == 'tracing'
Requires-Dist: opentelemetry-sdk>=1.39.0; extra == 'tracing'
Description-Content-Type: text/markdown

# Local Brain — Claude Code Plugin Marketplace

A [Claude Code plugin marketplace](https://code.claude.com/docs/en/plugins) that extends Claude with local capabilities. The first skill lets Claude delegate codebase exploration to local Ollama models.

## Install Marketplace

Add this marketplace to Claude Code:

```bash
/plugin marketplace add IsmaelMartinez/local-brain
```

Then install the plugin:

```bash
/plugin install local-brain@local-brain-marketplace
```

## Available Plugins

### [`local-brain`](./local-brain/skills/local-brain/SKILL.md)

Delegate codebase exploration to local Ollama models. Claude offloads read-only tasks to your machine—no cloud round-trips, full privacy.

```
┌─────────────┐     delegates      ┌─────────────┐     calls      ┌─────────┐
│ Claude Code │ ──────────────────►│ Local Brain │ ──────────────►│ Ollama  │
│   (Cloud)   │                    │ (Smolagents)│                │ (Local) │
│             │◄────────────────── │             │◄────────────── │         │
└─────────────┘     returns        └─────────────┘    responds    └─────────┘
                    results                        with code execution
```

**What Claude can delegate:**
- "Review the code changes"
- "Explain how the auth module works"
- "Generate a commit message"
- "Find all TODO comments"

---

## Marketplace Structure

This repo follows the [Claude Code plugin structure](https://code.claude.com/docs/en/plugins):

```
local-brain/                          # MARKETPLACE ROOT
├── .claude-plugin/
│   └── marketplace.json              # Marketplace manifest
└── local-brain/                      # PLUGIN
    ├── plugin.json                   # Plugin manifest
    └── skills/
        └── local-brain/
            └── SKILL.md              # Skill documentation
```

---

## `local-brain` Plugin Details

### Prerequisites

1. **Install the CLI:**

```bash
uv pip install local-brain
```

Or with pipx:
```bash
pipx install local-brain
```

2. **Install Ollama** from [ollama.ai](https://ollama.ai) and pull a model:

```bash
ollama pull qwen3
```

### CLI Usage

```bash
local-brain "What files changed recently?"
local-brain "Review the code in src/"
local-brain "Generate a commit message"
local-brain "Explain how auth works"
```

```bash
local-brain "prompt"                       # Ask anything (auto-selects best model)
local-brain -v "prompt"                    # Verbose (show tool calls)
local-brain -m qwen2.5-coder:7b "prompt"   # Specific model
local-brain --list-models                  # Show available models
local-brain --root /path/to/project "prompt"  # Set project root
```

### Model Discovery

Local Brain automatically detects installed Ollama models and picks the best one:

```bash
local-brain --list-models
```

**Recommended models:**
- `qwen3:latest` — General purpose (default)
- `qwen2.5-coder:7b` — Code-focused
- `llama3.2:3b` — Fast, lightweight
- `mistral:7b` — Balanced

### Tools

Custom read-only tools registered with Smolagents' `@tool` decorator:

| Tool | What it does |
|------|--------------|
| `read_file` | Read file contents (path-jailed) |
| `list_directory` | List files with glob patterns (path-jailed) |
| `file_info` | Get file metadata (path-jailed) |
| `git_diff` | Show git changes (staged or unstaged) |
| `git_status` | Show current branch and changes |
| `git_log` | View recent commit history |
| `git_changed_files` | List modified/staged files |

### Architecture

Local Brain uses [Smolagents](https://github.com/huggingface/smolagents) as the agent framework:

```
local_brain/
├── __init__.py      # Version
├── cli.py           # Click CLI entry point
├── models.py        # Ollama model discovery & selection
├── security.py      # Path jailing utilities
└── smolagent.py     # CodeAgent + custom tools
```

**What comes from Smolagents:**
- `CodeAgent` — Agent that executes tasks via code generation
- `LiteLLMModel` — Connects to Ollama via LiteLLM
- `@tool` decorator — Registers our custom tools with the agent

**What we implement:**
- All 7 tools (read_file, git_diff, etc.) — our code, registered via `@tool`
- Path jailing security — restricts file access to project root
- Model discovery — detects installed Ollama models

### Security

**Two-layer security model:**

1. **Tool layer** — Our pre-defined tools are trusted code:
   - ✅ Read files within project directory (path-jailed)
   - ✅ Execute git commands (read-only via subprocess)
   - ❌ File I/O outside project root blocked
   - ❌ Sensitive files blocked (`.env`, keys)

2. **LLM sandbox** — Code generated by the LLM runs in LocalPythonExecutor:
   - ❌ Cannot import subprocess, socket, os.system, etc.
   - ❌ Cannot access network directly
   - ✅ Can only call our pre-defined tools

The LLM writes Python code that calls our tools—it cannot bypass them to run arbitrary shell commands.

**Why no web access?** Claude Code already has web access—delegate web research to Claude, local codebase work to Local Brain. This separation prevents data exfiltration and prompt injection from fetched content.

### Future Ideas

- **MCP Bridge** — Ollama ↔ Model Context Protocol bridge when MCP adoption increases
- **Docker Sandbox** — Stronger isolation via container when Docker is available
- **CLI Wrappers** — Wrap existing tools (ripgrep, gh, tokei) instead of custom implementations
- **Observability** — Add tracing and logging for debugging agent behavior

### Architecture Decisions

See [docs/adrs/](./docs/adrs/) for Architecture Decision Records:
- [ADR-001](./docs/adrs/001-custom-implementation.md) — Why custom implementation over frameworks
- [ADR-002](./docs/adrs/002-smolagents.md) — Why Smolagents for code execution
- [ADR-003](./docs/adrs/003-no-web-tools.md) — Why no web tools

---

## Adding New Plugins

Want to add a plugin to this marketplace?

1. Create a new directory at the root:

```
your-plugin/
├── plugin.json
└── skills/
    └── your-skill/
        └── SKILL.md
```

2. Register it in `.claude-plugin/marketplace.json`:

```json
{
  "plugins": [
    { "name": "local-brain", "source": "./local-brain", "description": "..." },
    { "name": "your-plugin", "source": "./your-plugin", "description": "..." }
  ]
}
```

See the [Claude Code plugin docs](https://code.claude.com/docs/en/plugins) for full specifications.

---

## Development

```bash
git clone https://github.com/IsmaelMartinez/local-brain.git
cd local-brain
uv sync
uv run local-brain "Hello!"
```

**Note:** Requires Python 3.10-3.13 (grpcio build issue with 3.14).

### Run Tests

```bash
uv run pytest tests/ -v
```

### Test Plugin Locally

```bash
# In Claude Code
/plugin marketplace add ./path/to/local-brain
/plugin install local-brain@local-brain-marketplace
```

## License

MIT
