Skip to content

Quick Start

From zero to your first generated document — from either the command line or Python. Both paths produce the same artifacts.

1. Install

pip install seed-data

The default renderer (xhtml2pdf) is pure Python — there are no system libraries to install. For the richer-CSS WeasyPrint renderer, see Renderers.

2. Configure AWS Bedrock

seed-data generates content with Amazon Bedrock models, so you need AWS credentials with Bedrock access.

export AWS_PROFILE=your-profile-name
export AWS_REGION=us-east-1

3. Get the schemas

A schema defines a document type — its fields, data rules, and visual style. Copy the built-in library into a local, editable folder:

seed-data clone-schema-library ./schemas

Browse the library on GitHub: 👉 awslabs/…/schemas

Built-in types include invoice, fcc-invoice, bank-statement, w2, pay-stub, credit-report, loan-application, police-report, and more.

4. Generate your first document

Generate a single document — pass a bundled schema name or a path:

seed-data --schema-dir invoice --output ./output

This runs the full single-document pipeline (data generation → validation → PDF render → visual critique). It writes three paired artifacts:

output/
├── data/<id>.json                 # ground-truth JSON label
├── generation_scripts/<id>.html   # the HTML the renderer used
└── pdfs/<id>.pdf                  # the final document

More CLI examples:

# Steer the content with a brief; pick faster/cheaper models
seed-data --schema-dir invoice \
  --scenario "Invoices for midwest food distributors, 20-40 line items" \
  --doc-model gpt-oss --critic-model haiku --threshold 3

# A diverse batch of N documents from one brief
seed-data --schema-dir fcc-invoice --count 10 \
  --scenario "Local TV stations across different US regions"

# A coordinated multi-document packet (merged PDF + shared context)
seed-data packet lending-package \
  --scenario "First-time homebuyer in Portland, OR"

# Add scanning/aging artifacts, or a different renderer
seed-data --schema-dir invoice --augment --renderer weasyprint

Run seed-data --help (and seed-data packet --help) to see every option.

Python API Usage

Everything the CLI does is available programmatically through the Generator class — configure it once, then call a verb. Each returns a typed result.

from seed_data import Generator, ModelConfig

gen = Generator(
    models=ModelConfig(doc="gpt-oss", critic="haiku"),
    threshold=3,
    output_dir="./output",
)

Single document → GeneratedDoc:

doc = gen.generate("invoice", scenario="Midwest food-distributor invoice")

print(doc.success, doc.verdict, doc.score)
print(doc.pdf_path)          # the rendered PDF
print(doc.data)              # ground-truth JSON label (lazy-loaded dict)
print(doc.data_json_path)    # ...and its path on disk

Diverse batch → BatchResult:

batch = gen.generate_batch(
    "fcc-invoice",
    count=10,
    scenario="Local TV stations across different US regions",
    on_document=lambda i, n, d: print(f"[{i+1}/{n}] {d.verdict}"),
)
print(batch.count_succeeded, "of", batch.count_requested)
for d in batch.succeeded:
    print(d.pdf_path, d.data_json_path)

Coordinated packet → PacketResult:

pkt = gen.generate_packet("lending-package", scenario="First-time homebuyer in Portland, OR")
print(pkt.merged_pdf)
for s in pkt.sections:
    print(s.document_class, s.page_indices)

Schema from real documents → a Schema you can generate from. Point SEED at example documents and a vision model reverse-engineers the schema for you:

schema = gen.infer_schema("./samples/*.pdf", name="invoice")
doc = gen.generate(schema, scenario="Midwest food distributor")

See Schema from Documents for inferring from images, S3, or one concatenated multi-document PDF.

Define a document type in code — no files on disk, from a pydantic model:

from pydantic import BaseModel
from seed_data import Schema

class Invoice(BaseModel):
    invoice_number: str
    total: float

schema = Schema(name="invoice", model=Invoice,
                generation_guidance="Totals must equal the sum of line items.")
doc = gen.generate(schema, scenario="IT consulting services")

See the Python API reference for full signatures.

Edit or create a schema

Because you cloned the library locally, editing a document type is just editing files:

schemas/invoice/
├── schema.json              # fields and types
└── generation_guidance.md   # data-realism rules and visual style

Point --schema-dir (or gen.generate(...)) at any folder with that structure — your own custom types work exactly the same way.

Next steps