axi-platform 0.10.4

Generic LLM/RAG platform framework

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:

1. Builtin (src/axiom/extensions/builtins/)
2. Installed PyPI packages with axiom-extension.toml
3. 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.