Skip to main content

CLI and MCP server for Pretorin Compliance API

Project description

Pretorin

Compliance tools for developers. Integrate with AI agents or your CI pipeline.

PyPI version MCP Registry MCP Compatible License: Apache-2.0 Tests Python 3.10+


Beta — Pretorin is currently in closed beta. Framework/control browsing works for authenticated users. Platform write features (evidence, narratives, monitoring) require a beta code. Sign up for early access.

Pretorin CLI gives developers and AI agents direct access to compliance data, implementation context, and evidence workflows.

The CLI and MCP tooling in this repository are open source. Access to Pretorin-hosted platform services, APIs, and account-scoped data is authenticated and governed separately by the applicable platform terms.

mcp-name: io.github.pretorin-ai/pretorin

Two Usage Modes

  1. Pretorin-hosted model mode: run pretorin agent run and route model calls through Pretorin /v1 endpoints.
  2. Bring-your-own-agent mode: run pretorin mcp-serve and connect the MCP server to your existing AI tool (Claude Code, Codex CLI, Cursor, etc.).

Quick Start

uv tool install pretorin
pretorin login
pretorin skill install

Run the walkthrough:

bash tools/demo-walkthrough.sh

Hosted Model Workflow (Recommended)

Use this flow when you want pretorin agent run to go through Pretorin-hosted model endpoints.

  1. Authenticate with your Pretorin API key:
pretorin login
  1. Optional: point model traffic to a custom/self-hosted Pretorin endpoint:
pretorin config set model_api_base_url https://platform.pretorin.com/api/v1/public/model
  1. Verify runtime setup:
pretorin agent doctor
pretorin agent install
  1. Run an agent task:
pretorin agent run "Assess AC-2 implementation gaps for my system"

Key behavior:

  • Preferred setup is pretorin login with no shell-level OPENAI_API_KEY override.
  • Model key precedence is: OPENAI_API_KEY -> config.api_key -> config.openai_api_key.
  • If OPENAI_API_KEY is set in your shell, it overrides stored login credentials.

Add to Your AI Tool

Use this flow when you already have an AI agent/tool and want Pretorin as an MCP capability provider.

Rome-bot

Install the Skill

The Pretorin skill teaches your AI agent how to use MCP tools effectively for compliance workflows. Install it for Claude Code and/or Codex CLI:

pretorin skill install                # both agents
pretorin skill install --agent claude # claude only
pretorin skill install --agent codex  # codex only
pretorin skill status                 # check what's installed

1. Claude Code

claude mcp add --transport stdio pretorin -- pretorin mcp-serve

Team setup via .mcp.json:

{
  "mcpServers": {
    "pretorin": {
      "type": "stdio",
      "command": "pretorin",
      "args": ["mcp-serve"]
    }
  }
}

2. Codex CLI

Add to ~/.codex/config.toml:

[mcp_servers.pretorin]
command = "pretorin"
args = ["mcp-serve"]

If you installed Pretorin with uv tool install or pipx, prefer pinning the absolute path from command -v pretorin to avoid PATH drift between shells and GUI apps.

For Claude Desktop, Cursor, and Windsurf setup, see docs/MCP.md.

Core Commands

Platform-backed review and update workflows are single-scope: set one active system + framework first with pretorin context set, then run evidence, note, monitoring, narrative, or MCP-assisted compliance commands inside that scope. Multi-framework work must be split into separate runs. Evidence, narratives, and notes all support a local-first workflow: create locally, list, then push to the platform.

Command Purpose
pretorin login Authenticate with the Pretorin API
pretorin logout Clear stored credentials
pretorin whoami Show current authenticated user
pretorin frameworks list List available frameworks
pretorin frameworks control <framework> <control> Get control details and guidance
pretorin context set Set active system/framework context
pretorin context show Inspect and validate the active context
pretorin context clear Clear the active context
pretorin control status Start/reopen control authoring with in_progress
pretorin control context Get full control context for the active scope
pretorin evidence create Create local evidence file
pretorin evidence list List local evidence files
pretorin evidence push Push local evidence to Pretorin
pretorin evidence search Search platform evidence
pretorin evidence link Link evidence to a control
pretorin evidence link-cci Link evidence to a CCI implementation row
pretorin evidence link-stig Link evidence to a STIG rule workflow
pretorin evidence upsert <ctrl> <fw> Find-or-create evidence and link it
pretorin evidence mark-current <id> Re-affirm evidence freshness (bump expires_at, transition expired→valid)
pretorin evidence upload Upload a file as evidence to the platform
pretorin evidence delete Delete an evidence record
pretorin narrative create Create local narrative file
pretorin narrative list List local narrative files
pretorin narrative push Push local narratives to Pretorin
pretorin narrative get <ctrl> <fw> Get current control narrative
pretorin narrative push-file <ctrl> <fw> <sys> <file> Push a single narrative file
pretorin notes create Create local note file
pretorin notes list --local List local note files
pretorin notes push Push local notes to Pretorin
pretorin notes list <ctrl> <fw> List platform control notes
pretorin notes add <ctrl> <fw> --content ... Add control note directly
pretorin notes resolve <ctrl> <fw> <note_id> --resolution-note ... Resolve (close) a control note
pretorin monitoring push Push a monitoring event
pretorin agent run "<task>" Run Codex-powered compliance task
pretorin review run --control-id <id> --path <dir> Review local code for control coverage
pretorin skill install Install Pretorin skill for AI agents
pretorin skill status Check skill install status per agent
pretorin mcp-serve Start MCP server

Campaign Workflows

Campaigns let you run bulk compliance operations across multiple controls, policies, or scope questions in a single coordinated run. Campaigns support an external-agent-first pattern with checkpoint persistence and lease-based concurrency.

Command Purpose
pretorin campaign controls --mode initial --family AC Draft narratives/evidence for a control family
pretorin campaign controls --mode notes-fix --family AC Fix controls flagged by platform notes
pretorin campaign controls --mode review-fix --family AC Fix controls flagged by family review
pretorin campaign policy --mode answer --all-incomplete Answer all incomplete policy questions
pretorin campaign scope --mode answer Answer scope questions for a system/framework
pretorin campaign status --checkpoint <path> Check campaign progress

Vendor Management & Inheritance

Manage vendor entities (CSPs, SaaS, managed services) and track control inheritance through vendor responsibility edges.

Command Purpose
pretorin vendor list List all vendors
pretorin vendor create <name> --type csp Create a vendor entity
pretorin vendor get <id> Get vendor details
pretorin vendor upload-doc <id> <file> Upload vendor evidence document
pretorin vendor list-docs <id> List vendor documents

Risk Management

Populate a system's risk register: list and create risks, attach artifact links (controls, evidence, findings, vendors), record mitigation strategies, and refresh AI summaries. Risks are system-scoped; the library is org-level.

Command Purpose
pretorin risk list <system_id> List risks for a system (filter by category, level, status)
pretorin risk show <system_id> <risk_id> Show full risk including artifact links
pretorin risk create <system_id> --title ... Create a risk; pass --framework for control auto-link
pretorin risk seed <system_id> Seed risks from library templates
pretorin risk update <system_id> <risk_id> Update fields including --treatment (mitigation surface)
pretorin risk link add <system_id> <risk_id> Attach a control / evidence / finding / vendor to a risk
pretorin risk link rm <system_id> <risk_id> <link_id> Remove a risk artifact link
pretorin risk refresh-summary <system_id> <risk_id> Re-score risk and trigger best-effort AI summary
pretorin risk library list Browse the org-level template library

Policy & Scope Questionnaires

Stateful questionnaire workflows for organization policies and system scope. Answer questions interactively or in bulk via campaigns.

Command Purpose
pretorin policy list List organization policies
pretorin policy show <policy_id> Show policy detail and questions
pretorin policy populate Auto-populate policy answers from context
pretorin scope show Show scope questionnaire for active context
pretorin scope populate Auto-populate scope answers from context

STIG & CCI Browsing

Browse STIG benchmarks, rules, and CCIs with full traceability from NIST 800-53 controls down to individual STIG check rules.

Command Purpose
pretorin stig list List STIG benchmarks
pretorin stig show <id> Show STIG benchmark detail
pretorin stig rules <id> List rules for a benchmark
pretorin stig applicable Show applicable STIGs for active system
pretorin stig infer AI-infer applicable STIGs from system profile
pretorin cci list List CCIs with optional control filter
pretorin cci show <id> Show CCI detail with linked SRGs and rules
pretorin cci chain <control_id> Full traceability: Control -> CCIs -> STIG rules
pretorin cci impl <cci_uuid> Show per-system CCI implementation (status, narrative, evidence)

Custom Frameworks

Author your own compliance framework or fork an existing one. The toolchain ships a vendored unified-framework JSON Schema, OSCAL ↔ unified converters with lossless round-trip, and an end-to-end build / validate / upload workflow against the platform's framework revision lifecycle.

Command Purpose
pretorin frameworks init-custom <id> Scaffold a minimal valid unified.json template
pretorin frameworks validate-custom <path> Local JSON Schema pre-flight before upload
pretorin frameworks build-custom <input> -f <id> Auto-detect input shape and normalize to unified.json
pretorin frameworks upload-custom <path> [--publish] POST a draft revision; --publish immediately promotes it
pretorin frameworks fork-framework <upstream_id> <new_id> Create a linked-fork draft anchored on upstream
pretorin frameworks rebase-fork <id> Create a fresh rebase draft against the latest upstream
pretorin frameworks revisions <id> List draft and published revisions
pretorin frameworks export-oscal <unified.json> Regenerate an OSCAL catalog from a unified artifact

Recipes

Recipes are agent-executable playbooks bundled with pretorin (markdown + scripts) that the calling AI agent reads, plans against, and runs. Built-in recipes include OpenSCAP, InSpec, and AWS/Azure cloud baselines, manual attestation, and code-evidence capture. Authoring docs live at docs/src/recipes/ and the contract spec at docs/rfcs/0001-recipes.md.

Command Purpose
pretorin recipe list List all loaded recipes (built-in + user + project)
pretorin recipe show <id> Show recipe detail and shadowing sources
pretorin recipe new <id> Scaffold a new community recipe in the user folder
pretorin recipe validate <id> Validate a recipe against the RFC 0001 contract
pretorin recipe run <id> Run a recipe end-to-end (typically invoked by an agent)

Quick context checks:

pretorin context show --quiet
pretorin context show --quiet --check

pretorin login clears the stored active context when you switch API keys or platform endpoints, which helps prevent old localhost or deleted-system scope from leaking into a new environment.

Artifact Authoring Rules

  • Narrative and evidence markdown must be human-readable for auditors: no markdown headings, use lists/tables/code blocks/links.
  • Markdown image embeds are temporarily disabled until platform-side file upload support is available.

Configuration

Credentials are stored at ~/.pretorin/config.json.

Variable Description
PRETORIN_API_KEY API key for platform access (overrides stored config)
PRETORIN_PLATFORM_API_BASE_URL Platform REST API base URL (/api/v1/public)
PRETORIN_API_BASE_URL Backward-compatible alias for PRETORIN_PLATFORM_API_BASE_URL
PRETORIN_MODEL_API_BASE_URL Model API base URL used by agent/harness flows (default: https://platform.pretorin.com/api/v1/public/model)
OPENAI_API_KEY Optional model key override for agent runtime

Documentation

Full documentation is built with mdbook. To view it locally:

# Install mdbook (if you don't have it)
cargo install mdbook

# Serve the docs and open in your browser
cd docs && mdbook serve --open

This starts a local server at http://localhost:3000 with live-reload.

To build static HTML without serving:

./tools/build-docs.sh
# Output is in docs/book/ and includes llms.txt / llms-full.txt

Quick links

Development

git clone https://github.com/pretorin-ai/pretorin-cli.git
cd pretorin-cli
uv pip install -e ".[dev]"
pytest
ruff check src/pretorin
ruff format --check src/pretorin

License

Licensed under the Apache License, Version 2.0. See LICENSE.

The Apache-2.0 license applies to the source code in this repository. It does not grant rights to Pretorin trademarks, logos, or branding, and it does not change the separate terms that govern access to Pretorin-hosted platform services and data. See TRADEMARKS.md.

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

pretorin-0.21.0.tar.gz (2.3 MB view details)

Uploaded Source

Built Distribution

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

pretorin-0.21.0-py3-none-any.whl (414.2 kB view details)

Uploaded Python 3

File details

Details for the file pretorin-0.21.0.tar.gz.

File metadata

  • Download URL: pretorin-0.21.0.tar.gz
  • Upload date:
  • Size: 2.3 MB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.12

File hashes

Hashes for pretorin-0.21.0.tar.gz
Algorithm Hash digest
SHA256 d5c9611218702c889305180663e302c841d4adfe86669c5ef2f2bcdf924d5a42
MD5 ccfab714d7bf10f6b750c7863ce4546b
BLAKE2b-256 45656a9409c8249b06fe077119046e1c34ac8b6b31137890696b5f42fc91b3fb

See more details on using hashes here.

File details

Details for the file pretorin-0.21.0-py3-none-any.whl.

File metadata

  • Download URL: pretorin-0.21.0-py3-none-any.whl
  • Upload date:
  • Size: 414.2 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.12

File hashes

Hashes for pretorin-0.21.0-py3-none-any.whl
Algorithm Hash digest
SHA256 eb0c8fc4fe0c70c86b871dc84419e9a2d7d076cf6f588e081c4927b0b263e654
MD5 f4d9ef332bec3039caab3100cfac36fa
BLAKE2b-256 a0a8ef59a5664953f5f95c9413df71c46a6a1d1af1c5bb03a5646737de17cd8d

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