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 inline — blocks for the whole solve
    result = await client.run_demand_model(request)

    # Run TSD (supply) model inline — blocks for the whole solve
    result = await client.run_tsd_model(optimization)

run_demand_model and run_tsd_model hold the HTTP request open for the length of the solve. They raise ModelRequestInvalid when the server rejects the body, with one readable line per field error in .issues.

Queue-based models

Every model can also be run through the job queue, where a worker picks the run up and the client polls for it. The run_*_model_async methods handle the whole queue → poll → fetch lifecycle and block until the result is ready:

from gravi_model_client import GraviModelClient

async with GraviModelClient(api_key="your-api-key") as client:
    demand = await client.run_demand_model_async(demand_request)  # -> ModelResultResponse
    ltl = await client.run_ltl_model_async(ltl_request)  # -> LtlModelResultResponse
    vrp = await client.run_vrp_model_async(vrp_request)  # -> VrpModelResultResponse
    osa = await client.run_osa_model_async(osa_input)  # -> OsaAPIResponse
    tsd = await client.run_tsd_model_async(tsd_input)  # -> OptimizationAPIResponse

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

For fire-and-forget / webhook workflows, use the primitives instead. Each model has the same trio:

The result fetch raises ModelResultNotReady while a run is still queued or running, and ModelRunError once it has ended failed or cancelled — so a caller polling the fetch directly can tell "come back later" from "this will never succeed".

Job administration

get_job_details, get_ltl_job_details and get_tsd_job_details return a JobDetails with the job's status alongside the request and response stored on the server — that is where to read back the input a finished run was given. cancel_demand_jobs, cancel_ltl_jobs, cancel_osa_jobs and cancel_tsd_jobs cancel every job under a history_id; get_queue_stats, get_ltl_queue_stats and get_tsd_queue_stats report queue depth; clear_files and clear_ltl_files delete the server's temporary request and response files.

Retries

Every request retries transient failures — connection errors, timeouts, 429 and 5xx — with jittered exponential backoff:

GraviModelClient(api_key, retries=3, retry_backoff=1.0, retry_backoff_max=20.0)

retries=0 disables it. The queue endpoints are deliberately more conservative: a failure that may have reached the server is not replayed, because that would enqueue the same solve twice. Only a connection failure, which provably never arrived, is retried there.

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