gravi-model-client

Python client for Gravitate's demand and supply optimization models.

Installation

pip install gravi-model-client

Usage

from gravi_model_client import GraviModelClient, ModelRunRequest, OptimizationInput

async with GraviModelClient(api_key="your-api-key", base_url="https://...") as client:
    # Health check
    await client.health_check()

    # Run demand model (synchronous)
    result = await client.run_demand_model(request)

    # Run TSD (supply) model (synchronous)
    result = await client.run_tsd_model(optimization)

Async (queue-based) models: VRP and OSA

VRP (tank wagon turn) and OSA (order-based supply) runs are queued and processed by a worker. run_vrp_model / run_osa_model handle the whole queue → poll → fetch lifecycle and block until the result is ready:

from gravi_model_client import GraviModelClient, VrpModelRunRequest, OsaInput

async with GraviModelClient(api_key="your-api-key") as client:
    vrp_result = await client.run_vrp_model(vrp_request)   # -> VrpModelResultResponse
    osa_result = await client.run_osa_model(osa_input)     # -> OsaAPIResponse

They raise ModelRunError if the run fails or is cancelled, and ModelRunTimeout if it doesn't finish within timeout (default 300s; tune poll_interval too).

For fire-and-forget / webhook workflows, use the primitives instead: queue_vrp_model returns a job id, get_job_status / get_osa_job_status return a JobStatus, and get_vrp_model_run_response / get_osa_model_run_response fetch the finished result.

Models

This package is the single source of truth for all pydantic models used by both the client and the server. Import models directly:

from gravi_model_client.demand import ModelRunRequest, StoreRequest, TankRequest, VolumeRange
from gravi_model_client.tsd import OptimizationInput, Directive, Site, Tank

Download files

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

Source Distribution

gravi_model_client-0.6.0.tar.gz (36.2 kB view details)

Uploaded Aug 19, 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.

gravi_model_client-0.6.0-py3-none-any.whl (27.6 kB view details)

Uploaded Aug 19, 2026 Python 3

File details

Details for the file gravi_model_client-0.6.0.tar.gz.

File metadata

Download URL: gravi_model_client-0.6.0.tar.gz
Upload date: Aug 19, 2026
Size: 36.2 kB
Tags: Source
Uploaded using Trusted Publishing? Yes
Uploaded via: twine/7.0.0 CPython/3.13.14

File hashes

Hashes for gravi_model_client-0.6.0.tar.gz
Algorithm	Hash digest
SHA256	`48dfbc7244b0a48b0be2403fad326dc371e5e958499bd849f0b7c1b2bc7d96be`
MD5	`bc96b830554834e22bee03dcc1f0ed44`
BLAKE2b-256	`889a567d9fc0f7bb2e7ac9250f9685a5fe2e35d9ab4dd118b05dce42e174616c`

See more details on using hashes here.

Provenance

The following attestation bundles were made for gravi_model_client-0.6.0.tar.gz:

Publisher: publish-gravi-model-client.yml on gravitate-energy/demand_only_model

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: gravi_model_client-0.6.0.tar.gz
- Subject digest: 48dfbc7244b0a48b0be2403fad326dc371e5e958499bd849f0b7c1b2bc7d96be
- Sigstore transparency entry: 2518082939
- Sigstore integration time: Aug 19, 2026
Source repository:
- Permalink: gravitate-energy/demand_only_model@2550d1af6034531175427f03ca1a631a77c81a3c
- Branch / Tag: refs/tags/gravi-model-client-v0.6.0
- Owner: https://github.com/gravitate-energy
- Access: private
Publication detail:
- Token Issuer: https://token.actions.githubusercontent.com
- Runner Environment: github-hosted
- Publication workflow: publish-gravi-model-client.yml@2550d1af6034531175427f03ca1a631a77c81a3c
- Trigger Event: push