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.0rc2.tar.gz (237.5 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.0rc2-py3-none-any.whl (293.9 kB view details)

Uploaded Python 3

File details

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

File metadata

  • Download URL: osmosis_ai-0.3.0rc2.tar.gz
  • Upload date:
  • Size: 237.5 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.0rc2.tar.gz
Algorithm Hash digest
SHA256 13fbf27490014a236baca56a7de5fd58358512d01d18d99a42a47add79023d46
MD5 7ec0e7b52f01f7c6d93a5cd029bf5a34
BLAKE2b-256 d0a091aa87ae2ce46c34c76db613ce5ea6c2191aae26d9189ba9eaf9df2d2633

See more details on using hashes here.

Provenance

The following attestation bundles were made for osmosis_ai-0.3.0rc2.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.0rc2-py3-none-any.whl.

File metadata

  • Download URL: osmosis_ai-0.3.0rc2-py3-none-any.whl
  • Upload date:
  • Size: 293.9 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.0rc2-py3-none-any.whl
Algorithm Hash digest
SHA256 d5ab77311089d9c468335019d137898af53647e21ea678e4ad058526185bf182
MD5 cd06d4e570c0572c06c768857b81386b
BLAKE2b-256 180c4c7c7fd00f2c2c00acb68ef73655dab5ac69a25c0b506a233c5fb4630af9

See more details on using hashes here.

Provenance

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