import os
from scale_gp import SGPClient

client = SGPClient(
    api_key=os.environ.get("SGP_API_KEY"),  # This is the default and can be omitted
)
page = client.training_datasets.contents.list(
    "training_dataset_id",
)
page = page.items[0]
print(page.input)

[
  {
    "input": "<string>",
    "output": "<string>",
    "schema_type": "GENERATION"
  }
]