{
  "$schema": "https://json-schema.org/draft/2020-12/schema",
  "$id": "https://raw.githubusercontent.com/api-evangelist/prime-intellect/main/json-schema/prime-intellect-training-run-schema.json",
  "title": "Prime Intellect Training Run",
  "description": "A dedicated reinforcement-learning post-training run executed against a hosted base model.",
  "type": "object",
  "required": ["id", "base_model", "status"],
  "properties": {
    "id": {"type": "string"},
    "base_model": {
      "type": "string",
      "description": "One of the supported base models: Qwen3.5 (0.8B/2B/4B/9B/35B-A3B/122B-A10B/397B-A17B), Llama 1B/3B Instruct, NVIDIA Nemotron 30B/120B, OpenAI gpt-oss 20B/120B."
    },
    "environment": {"type": "string", "description": "Environments Hub slug for the RL environment driving the run."},
    "config": {"type": "object", "description": "Run configuration generated by `prime train init`."},
    "status": {"type": "string", "enum": ["queued", "running", "succeeded", "failed", "cancelled"]},
    "metrics": {"type": "object", "description": "Live training metrics."},
    "checkpoints": {"type": "array", "items": {"type": "string"}},
    "usage": {
      "type": "object",
      "properties": {
        "input_tokens": {"type": "integer"},
        "output_tokens": {"type": "integer"},
        "training_tokens": {"type": "integer"},
        "cost_usd": {"type": "number"}
      }
    },
    "created_at": {"type": "string", "format": "date-time"},
    "team_id": {"type": "string"}
  }
}