osmosis-ai

⚠️ Warning: osmosis-ai is still in active development. APIs may change between versions.

Python SDK and CLI for Osmosis AI, a platform for training LLMs with reinforcement learning. Implement an AgentWorkflow and a concrete Grader in Python, then use the CLI to submit evaluation and training runs from an Osmosis workspace directory.

Installation

Requires Python 3.12+.

pip install osmosis-ai            # Core SDK
pip install osmosis-ai[server]    # + FastAPI rollout server
# or with uv:  uv add osmosis-ai

See Installation for the full extras matrix and CONTRIBUTING.md for development setup.

Documentation

Guides, quickstart, and the full CLI reference live at docs.osmosis.ai.

• Quickstart — run the multiply example end to end, from onboarding to evaluation run to training run
• CLI command reference — every osmosis command and flag, plus the --json / --plain output contract for AI agents and CI/CD
• Workspace setup — repository layout, config files, and Git Sync
• Rollouts — AgentWorkflow, Grader, integrations, and execution backends
• Releases — version history and breaking changes between releases

Building on or contributing to the SDK itself? See the code-anchored developer docs in docs/ (start with docs/architecture.md) alongside CONTRIBUTING.md.

Contributing

See CONTRIBUTING.md for development setup, testing, linting, and PR guidelines.

License

MIT License - see LICENSE file for details.