adaption-ai

Adaption AI SDK Skill

Build dataset augmentation pipelines with Adaption's Adaptive Data platform. This skill covers the full lifecycle: ingest → adapt → wait → evaluate → export.

Quick Reference

pip install adaption

from adaption import Adaption
client = Adaption(api_key="pt_live_...")  # or set ADAPTION_API_KEY env var

Async client: from adaption import AsyncAdaption

Core Lifecycle

1. Ingest — Upload local file, import from HuggingFace, or import from Kaggle
2. Adapt — Start an augmentation run with column mapping + optional controls
3. Wait — Poll for completion with exponential backoff
4. Evaluate — Fetch quality metrics (score_before/after, improvement %)
5. Export — Download augmented dataset via presigned URL

Ingest Methods

Local file upload

Supported formats: .csv, .json, .jsonl, .parquet

result = client.datasets.upload_file("training_data.csv")
result = client.datasets.upload_file("data.csv", name="my-dataset")  # custom name
dataset_id = result.dataset_id

Hugging Face import (async on server — poll before running)

resp = client.datasets.create_from_huggingface(
    url="https://huggingface.co/datasets/org/repo",
    files=["train.csv"],
)

Kaggle import (requires Kaggle API creds registered in Adaption settings)

resp = client.datasets.create_from_kaggle(
    url="https://www.kaggle.com/datasets/org/dataset-name",
    files=["data.csv"],
)

Running Adaptation Jobs

datasets.run() — the central method

Required: dataset_id, column_mapping with at least "prompt" key.

run = client.datasets.run(
    dataset_id,
    column_mapping={
        "prompt": "instruction",       # required
        "completion": "response",      # optional
        # "chat": "conversation",      # optional — alternative to prompt+completion
        # "context": ["source", "ref"],# optional — list of context columns
    },
    # --- Optional controls (see references/ for details) ---
    brand_controls={...},
    recipe_specification={...},
    job_specification={...},
    estimate=True,  # dry-run: get cost quote without starting
)