Skip to main content

AURORA: Adaptive Unified Reasoning and Orchestration Architecture with MCP Integration

Project description

AURORA

Adaptive Unified Reasoning and Orchestration Architecture

Python 3.10+ PyPI version License: MIT

A cognitive architecture framework that brings intelligent memory, reasoning, and orchestration to AI coding tools. Built on cognitive science principles (ACT-R, SOAR), Aurora enables AI assistants to understand your codebase through persistent memory, semantic search, and multi-step reasoning.


What is Aurora?

Aurora is an AI-powered codebase intelligence system that:

  • Indexes your code with semantic understanding (not just text search)
  • Provides context to AI coding assistants (Claude Code, Cursor, etc.)
  • Remembers everything with persistent memory (frequency, recency, relevance)
  • Reasons intelligently using cognitive architecture patterns
  • No API keys needed for search and memory features

Think of it as a "smart memory layer" for AI coding tools that makes them understand your specific codebase.


Key Features

🔍 Semantic Code Search

  • Tree-sitter powered parsing (Python, with extensible language support)
  • Hybrid retrieval: activation scoring + semantic embeddings
  • Sub-500ms search on 10K+ code chunks
  • Intelligent ranking by relevance, frequency, and recency

🧠 Persistent Memory

  • ACT-R activation-based memory (cognitive science principles)
  • Learns from usage patterns (which code gets referenced most)
  • Multi-tier caching for fast retrieval
  • Project-local .aurora/ directory (safe, isolated)

🤖 AI Tool Integration

  • MCP Protocol: Works with Claude Code CLI, Cursor, Cline, Continue
  • Slash Commands: Quick /aurora-search, /aurora-query commands
  • No API keys required for MCP tools (uses host LLM)
  • Native integration with popular AI coding tools

🎯 SOAR Reasoning Pipeline

  • 9-phase cognitive architecture (Assess → Retrieve → Decompose → ...)
  • Automatic escalation (simple vs complex queries)
  • Multi-agent orchestration and task decomposition
  • Groundedness scoring (prevents hallucination)

📋 Planning System

  • Project-local plan management (.aurora/plans/)
  • Sequential numbering (0001, 0002, etc.)
  • Track features, tasks, and implementation progress
  • Integrated with CLI and MCP tools

🔧 Production Ready

  • Cross-platform (Windows, macOS, Linux)
  • Comprehensive health checks (aur doctor)
  • Budget tracking and cost control
  • Rich terminal output with progress bars

Installation

From PyPI (Recommended)

# Install Aurora (includes all features)
pip install aurora-actr

# Verify installation
aur --version

Current Version: 0.4.0

Package Size: ~528KB (tiny!)

What's Included:

  • CLI commands (aur)
  • MCP server (aurora-mcp)
  • Memory system (semantic search)
  • Planning system
  • SOAR reasoning engine
  • All dependencies

Optional Extras

# With ML features (semantic embeddings - requires torch)
pip install aurora-actr[ml]

# Development tools (for contributors)
pip install aurora-actr[dev]

From Source

git clone https://github.com/your-org/aurora.git
cd aurora
pip install -e .

Quick Start

1. Initialize Aurora in Your Project

cd /path/to/your/project
aur init

This runs 3 unified steps:

  1. Planning Setup - Creates .aurora/ directory structure
  2. Memory Indexing - Indexes your codebase for semantic search
  3. Tool Configuration - Sets up MCP servers and slash commands for AI tools

What gets created:

  • .aurora/ directory (project-local, git-ignored)
  • .aurora/memory.db (SQLite database for indexed code)
  • .aurora/plans/ (for feature planning)
  • MCP and slash command configurations

2. Use with AI Coding Tools

No additional setup required! Restart your AI tool and Aurora's capabilities are available.

Claude Code CLI Example:

You: "Search my codebase for authentication logic"
Claude: [Uses aurora_search tool automatically]

You: "Show me the UserService class"
Claude: [Uses aurora_context tool to retrieve full code]

You: "How does the payment flow work?"
Claude: [Uses aurora_query with SOAR reasoning]

Cursor/Cline/Continue:

Same natural language interaction - tools work automatically via MCP.

3. Use CLI Directly (Optional)

# Search indexed code
aur mem search "authentication"

# Get memory stats
aur mem stats

# Create a plan
aur plan create "Add user authentication"

# Health check
aur doctor

See: COMMANDS.md for complete command reference.


What Aurora Offers

For AI Coding Assistants:

  • Accurate code search instead of guessing file locations
  • Semantic understanding of your codebase structure
  • Context-aware responses using relevant code chunks
  • No hallucination - grounded in actual code

For Developers:

  • One-time indexing (aur mem index) - then instant search
  • Project-local setup - .aurora/ directory per project
  • No API keys needed for search/memory (MCP tools use host LLM)
  • Works offline for search and planning features

For Teams:

  • Consistent codebase understanding across team members
  • Shared plans in .aurora/plans/ (can be committed to git)
  • Budget tracking prevents runaway API costs
  • Audit trail of queries and reasoning

Commands Overview

Aurora provides three ways to interact with your codebase:

1. CLI Commands (aur)

aur init              # Initialize Aurora
aur mem search "..."  # Search code
aur mem index         # Re-index codebase
aur mem stats         # Show stats
aur plan create "..." # Create plan
aur doctor            # Health check
aur --version         # Show version

2. MCP Tools (via AI Assistants)

Natural language prompts automatically use:

  • aurora_search - Fast code search
  • aurora_context - Get code context
  • aurora_query - Complex reasoning
  • aurora_mem_stats - Memory stats
  • aurora_doctor - Health check

3. Slash Commands (Quick Access)

/aurora-search authentication
/aurora-context UserService
/aurora-query How does X work?
/aurora-doctor

Complete Reference: COMMANDS.md


How It Works

Step 1: Indexing (One-Time)

aur mem index
  • Parses your codebase with tree-sitter
  • Extracts functions, classes, methods, docstrings
  • Creates semantic embeddings (if ML features installed)
  • Stores in local .aurora/memory.db SQLite database

Step 2: Activation Scoring (ACT-R)

When you search or query:

  • Frequency: How often this code is referenced
  • Recency: When it was last accessed
  • Semantic Similarity: How relevant to your query
  • Context Boost: Related to current conversation

Step 3: Retrieval & Reasoning

  • Simple queries: Direct memory search (< 2s)
  • Complex queries: SOAR 9-phase pipeline (< 10s)
  • Groundedness check: Prevents hallucination on weak matches
  • Multi-step reasoning: Decomposes complex questions

Step 4: Integration

  • MCP tools provide context to AI assistant
  • AI assistant uses host LLM to process and respond
  • No extra API calls for search (only for AI reasoning)

Architecture

Aurora is built as a single bundled Python package with modular components:

Core Components:

  • aurora_core - Storage, configuration, data structures
  • aurora_cli - CLI commands and initialization
  • aurora_context_code - Code parsing and analysis
  • aurora_soar - SOAR reasoning pipeline
  • aurora_reasoning - LLM integration (Anthropic, OpenAI, Ollama)
  • aurora_planning - Plan management

Key Patterns:

  • 9-Phase SOAR Pipeline: Assess → Retrieve → Decompose → Verify → Route → Collect → Synthesize → Record → Respond
  • Hybrid Retrieval: 60% activation + 40% semantic similarity
  • Multi-Tier Caching: LRU cache + persistent cache + activation scores
  • Cognitive Memory: ACT-R principles for intelligent retrieval

Performance:

  • Simple query: < 2s (measured: 0.002s)
  • Complex query: < 10s
  • Memory usage: < 100MB for 10K chunks (measured: 39MB)
  • Retrieval: < 500ms for 10K chunks

Documentation

Getting Started

Architecture & Design

Development


FAQ

Q: Do I need an API key? A: No API key needed for search, memory, planning, or MCP tools. Only aur query (direct LLM) and aur headless (autonomous mode) require ANTHROPIC_API_KEY.

Q: What AI tools work with Aurora? A: Claude Code CLI, Cursor, Cline, Continue - any MCP-compatible tool.

Q: Does Aurora send my code to the cloud? A: No. Code stays local in .aurora/memory.db. Only when you use aur query with API key does it send context to LLM.

Q: How much does it cost? A: Aurora itself is free. If you use aur query with Anthropic API, you pay for API usage (~$0.01-0.10 per query depending on size).

Q: Can I use it offline? A: Yes! Search (aur mem search), memory, and planning work offline. Only aur query and autonomous mode need internet.

Q: What languages are supported? A: Python is fully supported. Extensible to other languages via tree-sitter parsers.

Q: How do I update Aurora? A: pip install --upgrade aurora-actr

Q: Where is my data stored? A: Project-local .aurora/ directory (can be git-ignored or committed).


Development

Setup

git clone https://github.com/your-org/aurora.git
cd aurora
pip install -e .[dev]

Testing

# Run all tests
pytest tests/

# Unit tests only
pytest tests/unit/

# Integration tests
pytest tests/integration/

# With coverage
pytest tests/ --cov=aurora_cli --cov-report=html

Code Quality

# Format code
ruff check --fix .

# Type checking
mypy packages/cli/src/

# Security scan
bandit -r packages/

Contributing

Contributions welcome! Please:

  1. Fork the repository
  2. Create a feature branch
  3. Make your changes with tests
  4. Run quality checks: ruff check --fix . && mypy packages/*/src/
  5. Submit a pull request

License

MIT License - see LICENSE for details.


Support


Aurora - Intelligent memory and reasoning for AI coding assistants • Version 0.4.0 • PyPI

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

aurora_actr-0.4.2.tar.gz (410.9 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

aurora_actr-0.4.2-py3-none-any.whl (527.3 kB view details)

Uploaded Python 3

File details

Details for the file aurora_actr-0.4.2.tar.gz.

File metadata

  • Download URL: aurora_actr-0.4.2.tar.gz
  • Upload date:
  • Size: 410.9 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.10.12

File hashes

Hashes for aurora_actr-0.4.2.tar.gz
Algorithm Hash digest
SHA256 b21ae444a6e493b60999de05ec851325e721a7dede363e6f8a79df356167835b
MD5 8b97620be2b17fea26a01602a4a0f05d
BLAKE2b-256 364c4020ee9c69bf73de3c297b2731673c5cf2b1ea1f975df574273d42aa329c

See more details on using hashes here.

File details

Details for the file aurora_actr-0.4.2-py3-none-any.whl.

File metadata

  • Download URL: aurora_actr-0.4.2-py3-none-any.whl
  • Upload date:
  • Size: 527.3 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.10.12

File hashes

Hashes for aurora_actr-0.4.2-py3-none-any.whl
Algorithm Hash digest
SHA256 900991adab4925035359254189f44c21a979a92ec52305e370852899648799f5
MD5 50232c7a08c12cb0a985c351f56d67ab
BLAKE2b-256 1caa71be67afbdba7003bd8573ade8a24ce78f9e5a0c955f34f079526fc4c83e

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page