Pre-release

This release is a pre-release and may not be stable for production use.

Osmosis

Platform PyPI Python Codecov License Docs

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.

Project links

Data verified by PyPI on
Data provided by the project maintainers, verified at the time the release was uploaded to PyPI.

Key dates

PyPI data
Data sourced directly from PyPI's database.
  • Released:
Stable release available (0.2.31)

2 maintainers

PyPI data
Data sourced directly from PyPI's database.
Avatar for gulpai from gravatar.com gulpai Avatar for JoyboyBrian from gravatar.com JoyboyBrian

Credits

Author: Osmosis AI

GitHub Statistics

Data verified by PyPI on 
The GitHub source repository was provided by the project maintainers and verified by PyPI at the time of upload. Stars, forks, and open issues/PRs are derived from that repository and have not been independently verified.

License expression

MIT
View SPDX License List

Requires

Python >=3.12

Provides Extra

server strands openai-agents harbor rubric parquet full

Classifiers

Report project as malware

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.0rc3.tar.gz (237.7 kB view details)

Uploaded Source

Built Distribution

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

osmosis_ai-0.3.0rc3-py3-none-any.whl (295.0 kB view details)

Uploaded Python 3

File details

Details for the file osmosis_ai-0.3.0rc3.tar.gz.

File metadata

  • Download URL: osmosis_ai-0.3.0rc3.tar.gz
  • Upload date:
  • Size: 237.7 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.0rc3.tar.gz
Algorithm Hash digest
SHA256 549dcfd97990a4a73c023093a53b76a4bdd373c538e5f39b19b829bad70ddbac
MD5 baab523ec1dda542f04b36038f68eadb
BLAKE2b-256 95cd91a12d669c447e608fd91be622631a2870c02046068519d360a0159eb0d2

See more details on using hashes here.

Provenance

The following attestation bundles were made for osmosis_ai-0.3.0rc3.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.

File details

Details for the file osmosis_ai-0.3.0rc3-py3-none-any.whl.

File metadata

  • Download URL: osmosis_ai-0.3.0rc3-py3-none-any.whl
  • Upload date:
  • Size: 295.0 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/7.0.0 CPython/3.13.14

File hashes

Hashes for osmosis_ai-0.3.0rc3-py3-none-any.whl
Algorithm Hash digest
SHA256 0e81fdfeb21d9e5bc1571c14cc8b49f052e0338aa55fffa0e1b665e761904f3c
MD5 70dcf9e3ffebe95c8579d7c327b813e7
BLAKE2b-256 b78404e1c48751d5c63e2cb40565de12b9579cf49f97cff70a6d27a51faebfdc

See more details on using hashes here.

Provenance

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

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.

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