Generic LLM/RAG platform framework
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.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file axi_platform-0.10.8.tar.gz.
File metadata
- Download URL: axi_platform-0.10.8.tar.gz
- Upload date:
- Size: 2.4 MB
- Tags: Source
- Uploaded using Trusted Publishing? Yes
- Uploaded via: twine/6.1.0 CPython/3.13.12
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
114bc23088ec2067616077af0542761225ec719f40d3fe81b0c6d3a23d343b1d
|
|
| MD5 |
95c19975b554ab154e6e2f3e49e79817
|
|
| BLAKE2b-256 |
10571743710cf24a31a6d5e3f468601f9155e7e7534446698349a5128cecb642
|
Provenance
The following attestation bundles were made for axi_platform-0.10.8.tar.gz:
Publisher:
publish.yml on benjaminbooth/axiom
-
Statement:
-
Statement type:
https://in-toto.io/Statement/v1 -
Predicate type:
https://docs.pypi.org/attestations/publish/v1 -
Subject name:
axi_platform-0.10.8.tar.gz -
Subject digest:
114bc23088ec2067616077af0542761225ec719f40d3fe81b0c6d3a23d343b1d - Sigstore transparency entry: 1309729998
- Sigstore integration time:
-
Permalink:
benjaminbooth/axiom@fad6b93a1c3a26a33200df135e5a07a9e1ab12ad -
Branch / Tag:
refs/tags/v0.10.8 - Owner: https://github.com/benjaminbooth
-
Access:
private
-
Token Issuer:
https://token.actions.githubusercontent.com -
Runner Environment:
github-hosted -
Publication workflow:
publish.yml@fad6b93a1c3a26a33200df135e5a07a9e1ab12ad -
Trigger Event:
push
-
Statement type:
File details
Details for the file axi_platform-0.10.8-py3-none-any.whl.
File metadata
- Download URL: axi_platform-0.10.8-py3-none-any.whl
- Upload date:
- Size: 1.6 MB
- Tags: Python 3
- Uploaded using Trusted Publishing? Yes
- Uploaded via: twine/6.1.0 CPython/3.13.12
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
d8a0ecfecbbdc723b3fc3a0dc7a57bc0aefb9dfe92f6fb3a751d28a74d6a69ff
|
|
| MD5 |
23b2d2eaf2c65fdb7628f2c100613b7f
|
|
| BLAKE2b-256 |
c05835d50d32f0525912042c4abfbac2184ee803544313fd9be6bccf4c20298b
|
Provenance
The following attestation bundles were made for axi_platform-0.10.8-py3-none-any.whl:
Publisher:
publish.yml on benjaminbooth/axiom
-
Statement:
-
Statement type:
https://in-toto.io/Statement/v1 -
Predicate type:
https://docs.pypi.org/attestations/publish/v1 -
Subject name:
axi_platform-0.10.8-py3-none-any.whl -
Subject digest:
d8a0ecfecbbdc723b3fc3a0dc7a57bc0aefb9dfe92f6fb3a751d28a74d6a69ff - Sigstore transparency entry: 1309730072
- Sigstore integration time:
-
Permalink:
benjaminbooth/axiom@fad6b93a1c3a26a33200df135e5a07a9e1ab12ad -
Branch / Tag:
refs/tags/v0.10.8 - Owner: https://github.com/benjaminbooth
-
Access:
private
-
Token Issuer:
https://token.actions.githubusercontent.com -
Runner Environment:
github-hosted -
Publication workflow:
publish.yml@fad6b93a1c3a26a33200df135e5a07a9e1ab12ad -
Trigger Event:
push
-
Statement type: