🧠 OH MY BRAIN

English | 中文

Multi-Agent Collaborative Development Framework - Let AI Teams Code for You in Parallel

✨ Features

• 🧠 Central Brain Coordination - Brain manages task distribution, context, and Git operations
• 🤖 Parallel Multi-Worker - Multiple AI Agents develop different modules simultaneously
• 🔌 Flexible AI Configuration - Choose different models per task type (cost/performance trade-offs)
• 📋 Standardized Dev Docs - YAML format, can be generated by any AI
• 🔒 Safety Audit - Dangerous commands are pre-reviewed to protect your codebase
• 📊 Real-time Monitoring - Dashboard to view each Worker's progress
• 🖥️ Cross-Platform - Supports Linux, macOS, Windows, and WSL

🏗 Architecture

                    ┌─────────────────────────────────┐
                    │            BRAIN                │
                    │  ┌─────────┐ ┌─────────┐       │
                    │  │ Context │ │  Model  │       │
                    │  │ Manager │ │  Router │       │
                    │  └────┬────┘ └────┬────┘       │
                    │  ┌────┴──────────┴────┐       │
                    │  │   Task Scheduler   │       │
                    │  └─────────┬──────────┘       │
                    └────────────┼─────────────────┘
                                 │ ZeroMQ
             ┌───────────────────┼───────────────────┐
             ▼                   ▼                   ▼
      ┌────────────┐      ┌────────────┐      ┌────────────┐
      │  Worker 1  │      │  Worker 2  │      │  Worker N  │
      │(Mini-Agent)│      │(Mini-Agent)│      │(Mini-Agent)│
      └────────────┘      └────────────┘      └────────────┘

🚀 Quick Start

Installation

pip install oh-my-brain

Or with uv (recommended):

uv add oh-my-brain

1. Configure AI Models

# ~/.oh-my-brain/models.yaml
models:
  - name: "deepseek-coder"
    provider: "openai"
    api_base: "https://api.deepseek.com"
    model: "deepseek-coder"
    api_key_env: "DEEPSEEK_API_KEY"
    cost_per_1k_tokens: 0.001
    capabilities: [code]

  - name: "minimax-m21"
    provider: "anthropic"
    api_base: "https://api.minimax.io"
    model: "MiniMax-M2.1"
    api_key_env: "MINIMAX_API_KEY"
    cost_per_1k_tokens: 0.002
    capabilities: [code, reasoning, planning]

task_model_mapping:
  planning: "minimax-m21"
  coding: "deepseek-coder"
  review: "minimax-m21"
  default: "deepseek-coder"

2. Prepare Development Document

Generate a standardized dev doc using your preferred AI (Claude, DeepSeek, GPT, etc.):

# my_project/dev_doc.yaml
project:
  name: "my-awesome-app"
  version: "0.1.0"
  description: "A sample web application"
  tech_stack:
    language: "Python"
    framework: "FastAPI"
    database: "PostgreSQL"

modules:
  - id: "mod-auth"
    name: "Authentication Module"
    description: "User authentication and authorization"
    priority: 1
    dependencies: []
    acceptance_criteria: "Users can register, login, and logout"
    sub_tasks:
      - id: "task-001"
        name: "Implement JWT Login"
        type: "feature"
        description: "Create login endpoint with JWT token generation"
        estimated_minutes: 30
        files_involved:
          - "src/auth/router.py"
          - "src/auth/service.py"
        requirements: |
          - POST /auth/login endpoint
          - Accept email and password
          - Return JWT token on success

3. Start Brain

oh-my-brain start --config ./brain.yaml

4. Start Workers

oh-my-brain worker --count 4  # Start 4 workers

5. Submit Dev Doc

oh-my-brain submit ./dev_doc.yaml

Then watch your AI team work in the Dashboard!

📖 Documentation

• Getting Started
• Dev Doc Format
• Model Configuration
• Architecture

🖥️ Platform Support

Platform	Status	Notes
Linux	✅ Full Support	Recommended for production
macOS	✅ Full Support
Windows	✅ Supported	Uses TCP transport, slightly lower performance
WSL	✅ Full Support	Recommended for Windows users

🛠 Development

# Clone the repository
git clone https://github.com/zhaofei0923/oh_my_brain.git
cd oh_my_brain

# Install dependencies
uv sync --all-groups

# Run tests
uv run pytest

# Run linters
uv run ruff check .
uv run mypy src/

🤝 Contributing

Contributions are welcome! Please read CONTRIBUTING.md

📄 License

Apache 2.0 - See LICENSE

🙏 Acknowledgments

• Mini-Agent - The foundation for our Worker agents
• MiniMax - AI model provider