Osmosis

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                     # CLI + framework-neutral rollout core
pip install "osmosis-ai[server]"            # + generic FastAPI rollout server
pip install "osmosis-ai[strands]"           # + Strands integration
pip install "osmosis-ai[openai-agents]"     # + OpenAI Agents integration
pip install "osmosis-ai[harbor]"            # + Harbor backend (uses an externally provided SkyPilot runtime)
pip install "osmosis-ai[rubric]"            # + LLM-as-judge rubric evaluation
pip install "osmosis-ai[parquet]"           # + Parquet dataset support
pip install "osmosis-ai[full]"              # every optional feature
# or with uv:  uv add osmosis-ai

There is one distribution, osmosis-ai. The harbor extra installs plain Harbor only: Daytona is retired, and Harbor's skypilot extra must not be installed because the rollout runtime provides SkyPilot. See Installation for product setup 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.

Operating System
- OS Independent
Programming Language

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

osmosis_ai-0.3.1.tar.gz (336.6 kB view details)

Uploaded Aug 25, 2026 Source

Built Distribution

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

The dropdown lists show the available interpreters, ABIs, and platforms. Enable javascript to be able to filter the list of wheel files.

osmosis_ai-0.3.1-py3-none-any.whl (408.3 kB view details)

Uploaded Aug 25, 2026 Python 3

File details

Details for the file osmosis_ai-0.3.1.tar.gz.

File metadata

Download URL: osmosis_ai-0.3.1.tar.gz
Upload date: Aug 25, 2026
Size: 336.6 kB
Tags: Source
Uploaded using Trusted Publishing? Yes
Uploaded via: twine/7.0.0 CPython/3.13.14

File hashes

Hashes for osmosis_ai-0.3.1.tar.gz
Algorithm	Hash digest
SHA256	`73072beb4b08cf88feca4283e85af09b39060bd43f1b52072dcb852c5d7a7032`
MD5	`489550a751f86deee1e75770fc4ee85f`
BLAKE2b-256	`1d9e9bb2fd323f877e30904e5767a8ba843e185b0f262b68f3c84df824518ef4`

See more details on using hashes here.

Provenance

The following attestation bundles were made for osmosis_ai-0.3.1.tar.gz:

Publisher: publish.yml on Osmosis-AI/osmosis-sdk-python

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

Statement:
- Statement type: https://in-toto.io/Statement/v1
- Predicate type: https://docs.pypi.org/attestations/publish/v1
- Subject name: osmosis_ai-0.3.1.tar.gz
- Subject digest: 73072beb4b08cf88feca4283e85af09b39060bd43f1b52072dcb852c5d7a7032
- Sigstore transparency entry: 2582519530
- Sigstore integration time: Aug 25, 2026
Source repository:
- Permalink: Osmosis-AI/osmosis-sdk-python@2d60976ae1920f0d0b17ac454011ee7264ba5aa4
- Branch / Tag: refs/tags/v0.3.1
- Owner: https://github.com/Osmosis-AI
- Access: public
Publication detail:
- Token Issuer: https://token.actions.githubusercontent.com
- Runner Environment: github-hosted
- Publication workflow: publish.yml@2d60976ae1920f0d0b17ac454011ee7264ba5aa4
- Trigger Event: release