Generic LLM/RAG platform framework
Reason this release was yanked:
classroom/memory/directive extensions missing from wheel due to release-process bug; fixed in 0.10.11
Project description
Axiom
A domain-agnostic platform framework for building federated, AI-powered operational systems.
Axiom provides the intelligence infrastructure — LLM routing, RAG, federation, agents, CLI — that domain products build on top of. NeutronOS is the first domain consumer (nuclear facilities).
Install
pip install axi-platform
What Axiom Does
User (laptop, mobile, IDE)
|
v
axi chat ───> LLM Gateway ───> Qwen (self-hosted) / Claude / OpenAI
| |
| v
| Query Router (keyword + SLM classification + VPN check)
| |
| v
+---------> RAG Store (pgvector) ──> local + federated knowledge
|
v
Agents: EVE (signals) | M-O (steward) | PR-T (publisher) | BURN-E (releases) | DFIB (diagnostics)
|
v
Federation: Node identity | Peer discovery | Trust | .axiompack distribution
Core Capabilities
| Capability | Description |
|---|---|
| LLM Gateway | Multi-provider routing with tier classification, VPN check, circuit breakers, audit logging |
| RAG | Three-tier corpus (community / organization / personal), pgvector embeddings, hybrid vector+full-text search |
| Federation | Ed25519 node identity, peer discovery, trust establishment, .axiompack portable knowledge bundles, A2A agent cards |
| Agents | EVE (signal ingestion), M-O (resource steward), PR-T (document publisher), BURN-E (release management), DFIB (diagnostics), Neut (chat assistant) |
| CLI | 28+ nouns, 150+ commands via extension discovery. Noun-verb pattern: axi <noun> <verb> |
| Interactive Chat | Multi-turn tool calling, RAG grounding, streaming, approval gates, session persistence |
| Knowledge Observatory | Velocity, accumulation, and impact metrics across knowledge bases |
| Call to Research | 5-level distributed research coordination protocol |
| SECUR-T | Security guardian — content verification, anomaly detection, trust scoring |
| Canary Nodes | Federated release promotion — edge/stable channels, signed attestations, quorum-based promotion |
| Document Publishing | Markdown to DOCX lifecycle with provider-based generation, storage, review gates |
| Signal Pipeline | Ingest from voice memos, Teams, GitLab; extract structured insights; publish briefings |
Quick Start
pip install axi-platform
# Check platform health
axi status
# Start an interactive chat session
axi chat
# Index documents for RAG
axi rag index ./docs/
# Search the knowledge base
axi rag search "how does the cooling system work?"
# Initialize federation identity
axi federation init
# Run diagnostics
axi doctor
Architecture
Axiom is designed as a platform framework — it provides generic infrastructure that domain products extend with domain-specific knowledge, agents, and tools.
Domain Products (e.g., NeutronOS)
|
| extends via extensions + domain packs
|
Axiom Platform
|
+-- CLI Framework (axi) 28+ nouns via extension discovery
+-- LLM Gateway Multi-provider routing + audit
+-- RAG Engine pgvector, 3-tier corpus
+-- Agent Framework 6 builtin agents
+-- Federation Protocol Node identity, trust, .axiompack
+-- State Management PostgreSQL + JSON fallback
+-- Prompt Registry Versioned, cacheable templates
+-- Extension System 3-tier discovery (builtin, installed, user)
|
PostgreSQL (pgvector) Single database for RAG + sessions + state
Deployment Progression
| Tier | Role | Infrastructure |
|---|---|---|
| Local | Development, personal RAG | Laptop, SQLite fallback |
| Private server | Shared LLM (GPU), team RAG | VPN, K3D, PostgreSQL |
| HPC cluster | Export-controlled workloads | Institutional VPN |
Nodes at any tier join the federation independently. See execution-plan-2026.md for the full roadmap.
CLI Reference
# Core
axi chat # Interactive agent with tool calling
axi status # System health dashboard
axi doctor # AI-powered diagnostics
# Knowledge
axi rag index|search|status # RAG corpus management
axi knowledge status # Knowledge observatory
axi research create|list # Call to Research protocol
# Federation
axi federation init|join|leave|peers
axi nodes add|status|list # Fleet monitoring
axi security status|alerts # SECUR-T guardian
axi chaos run # Resilience testing
# Operations
axi signal brief|draft # Signal pipeline (EVE agent)
axi pub publish|review # Document lifecycle (PR-T agent)
axi mo status|vitals # Resource steward (M-O agent)
axi release status|check # Release management (BURN-E agent)
# Infrastructure
axi connect [name] # Manage external connections
axi db up|migrate|status # PostgreSQL lifecycle
axi agents start|stop # Agent service management
axi serve # HTTP API server
Full command reference: spec-axi-cli.md
Development
# Clone and install
git clone https://github.com/benjaminbooth/axiom.git
cd axiom
pip install -e ".[all]"
# Run tests (2,270+)
pytest
# Lint
ruff check src/ tests/
ruff format src/ tests/
# Build
python -m build
Extension System
Axiom discovers CLI commands, tools, and agents from extensions via axiom-extension.toml manifests:
[extension]
name = "my-extension"
version = "0.1.0"
description = "What it does"
[[cli.commands]]
noun = "myext"
module = "my_extension.cli"
description = "My custom commands"
Discovery tiers:
- Builtin (
src/axiom/extensions/builtins/) - Installed PyPI packages with
axiom-extension.toml - User extensions (
~/.axi/extensions/)
Domain Products
Axiom is domain-agnostic. Domain products add domain-specific knowledge, agents, and tools:
| Product | Package | Domain |
|---|---|---|
| NeutronOS | neutron-os |
Nuclear facilities |
| (your product) | — | Any domain |
Documentation
| Document | Description |
|---|---|
| Execution Plan 2026 | Roadmap: what gets built, in what order |
| CLI Spec | Full command reference |
| Federation Spec | Multi-node protocol |
| RAG Architecture | Knowledge retrieval design |
| Agent Architecture | Agent capabilities |
| Session Store | PostgreSQL session backend |
| Canary Nodes | Release promotion protocol |
License
MIT License — see LICENSE file.
Acknowledgments
Developed at The University of Texas at Austin, Department of Mechanical Engineering — Nuclear & Radiation Engineering Program. Released as open source under the MIT License with approval from UT Austin Discovery to Impact.