Skip to main content

AI Assistant and CLI for OpenMC simulation workflows

Project description

PromptMC logo

PromptMC

Python License CI

OpenMC is powerful but can be painful to use: you hand-write XML, manage batch runs, and read results out of HDF5. It would be great if we could safely use AI to reduce that friction.

PromptMC does that by providing infrastructure and tooling that allows both AI LLM assistants and humans to interact with OpenMC through typed, schema-driven workflows.

It works like a grammar checker between an AI LLM assistant and OpenMC: your AI proposes a configuration, PromptMC validates XML structure and supported schema constraints before the simulator runs, and catches malformed inputs early.

Because AI hallucination is a valid concern in reactor physics, the system is designed with deterministic blast walls. Every configuration (from a human, the deterministic local planner, or AI) is validated against the same typed Pydantic schemas before it reaches the simulator.

What you can do

Most planning and schema-validation workflows work without OpenMC installed; execution, geometry-debug checks, and 2D plot rendering require OpenMC.

Without OpenMC installed:

  • Describe a simulation in plain English → a validated plan and settings.xml (the default planner uses no generative AI)
  • Validate XML structure and PromptMC's supported OpenMC schemas
  • Drive planning and schema validation from an AI client via the MCP server

With OpenMC installed:

  • Run simulations (subprocess or Python API)
  • Run geometry-debug overlap checks and generate 2D slice plots inside your AI chat client
  • Parse statepoint and tally outputs without touching HDF5

Installation

Prerequisites:

  • Python 3.10 or higher
  • Pip

PromptMC can be installed using pip. Use the following commands to install PromptMC:

pip install promptmc              # core (includes CLI, MCP server, and Gemini planner)
pip install promptmc[telemetry]   # + OpenTelemetry tracing

OpenMC (required for simulation execution, geometry-debug checks, and plot rendering) can be installed via Conda, Spack, Docker, or build from source per docs.openmc.org. Planning and XML/schema validation work without it.

Cross-section data (for running simulations):

pip install openmc-data-downloader
openmc_data_downloader -l TENDL-2019 -i U235 U238 O16 H1 -d cross_sections
export OPENMC_CROSS_SECTIONS=$(pwd)/cross_sections/cross_sections.xml

See installation for more details.

MCP server

PromptMC exposes a Model Context Protocol server so AI assistants can run OpenMC workflows natively — validation, plotting, execution, and result parsing from inside your LLM chat client, such as Claude Desktop, Cursor, and Google Antigravity.

Tools: openmc_validate, openmc_schema_check, openmc_template, openmc_list_templates, openmc_run, openmc_analyze, openmc_plot (2D slice, returned to the chat client), openmc_geometry_debug, openmc_check_installation, openmc_check_cross_sections.

Resources expose the configured cross-sections path, the session's tool-call history, and the bundled examples.

Setup: see the MCP server configuration guide for per-client steps (Claude Desktop/Code, Cursor, Google Antigravity, VS Code).

CLI

By default, plan uses a deterministic local planner, needing no API key, no network, no generative AI. The optional --llm flag calls Google Gemini (set GEMINI_API_KEY), which can interpret more open-ended natural-language requests. Customize the model name with GEMINI_MODEL (defaults to gemini-3.5-flash).

promptmc plan "concrete shielding calculation with 1 million particles"
promptmc plan "pin cell criticality with 50k particles" --write
promptmc template criticality --particles 10000                 # generate settings.xml
promptmc validate input.xml --schema
promptmc run input.xml --threads 4                              # needs OpenMC
promptmc batch batch_spec.yaml --parallel threads --workers 4
promptmc analyze ./output --json results.json
promptmc info                                                   # environment status

Full options in the CLI reference.

Safety

PromptMC is an engineering-assist tool that keeps a human in the loop. It is not a substitute for professional engineering judgment, independent verification and validation, or regulatory review, and is not for safety, licensing, or other regulated decisions. Reproducing a published benchmark is not qualification for safety analysis. Provided as-is (see LICENSE).

The goal is not autonomous reactor design; the goal is safer, faster OpenMC iteration.

Documentation

About

I studied nuclear engineering at MIT over 20 years ago, running MCNP 4 for my senior thesis. Though I left during my senior year, I eventually went back to university to get a degree in Computer Science, and I have spent the last 11 years working as a software engineer and site reliability engineer at a major FAANG cloud provider.

PromptMC bridges those two worlds. It is also, for me, an exploration of using agentic programming to build software for fun but still holding professional production principles and standards.

Contributions and Support

I welcome contributions! Please ensure all checks pass (pytest, ruff check, mypy src/) before opening a PR.

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

promptmc-0.3.0.tar.gz (58.8 kB view details)

Uploaded Source

Built Distribution

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

promptmc-0.3.0-py3-none-any.whl (71.0 kB view details)

Uploaded Python 3

File details

Details for the file promptmc-0.3.0.tar.gz.

File metadata

  • Download URL: promptmc-0.3.0.tar.gz
  • Upload date:
  • Size: 58.8 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.12

File hashes

Hashes for promptmc-0.3.0.tar.gz
Algorithm Hash digest
SHA256 21df10d961da6436e881e07eb84b9d4cb8f6b2a0a8196c6d2b6ad6763bad861e
MD5 497b871da21e87cdc6c0ab04c8844bee
BLAKE2b-256 47c2eebda00e4370747fe7df69ba2b46fdd44319e0edea7aea8eae2024ef148f

See more details on using hashes here.

Provenance

The following attestation bundles were made for promptmc-0.3.0.tar.gz:

Publisher: publish.yml on rjonace/promptmc

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file promptmc-0.3.0-py3-none-any.whl.

File metadata

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

File hashes

Hashes for promptmc-0.3.0-py3-none-any.whl
Algorithm Hash digest
SHA256 9a644ea5e80eef7d6e99b072507a98cc785312bf86efd1d5914e1add36f371ff
MD5 fd226a605855934120aa6432741fc4b4
BLAKE2b-256 ded227998b0d56c5da8a6ff8cbc69a94e87aff7cbd179c0ab0693b5438583621

See more details on using hashes here.

Provenance

The following attestation bundles were made for promptmc-0.3.0-py3-none-any.whl:

Publisher: publish.yml on rjonace/promptmc

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

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