from retab import Retab

client = Retab()

split = client.splits.create(
    document="invoice_batch.pdf",
    model="retab-small",
    subdocuments=[
        {
            "name": "invoice",
            "description": "Invoice documents with billing information",
            "partition_key": "invoice number",
            "allow_multiple_instances": True,
        },
        {"name": "receipt", "description": "Receipt documents for payments"},
        {"name": "contract", "description": "Legal contract documents"},
    ],
    context="Processing Q4 2024 vendor document batch",
    n_consensus=1,
)

for sub in split.output:
    print(f"{sub.name}: pages {sub.pages}")
    for partition in sub.partitions:
        print(f"  {partition.key}: pages {partition.pages}")

{
  "id": "split_01G34H8J2K",
  "organization_id": "org_abc123",
  "file": {
    "id": "file_6dd6eb00688ad8d1",
    "filename": "invoice_batch.pdf",
    "mime_type": "application/pdf"
  },
  "model": "retab-small",
  "subdocuments": [
    {"name": "invoice", "description": "Invoice documents", "partition_key": "invoice number", "allow_multiple_instances": true},
    {"name": "receipt", "description": "Receipts", "partition_key": null, "allow_multiple_instances": false}
  ],
  "n_consensus": 1,
  "output": [
    {
      "name": "invoice",
      "pages": [1, 2, 3],
      "partitions": [
        {"key": "INV-001", "pages": [1, 2]},
        {"key": "INV-002", "pages": [3]}
      ]
    },
    {"name": "receipt", "pages": [4, 5], "partitions": []}
  ],
  "consensus": null,
  "created_at": "2024-03-15T10:30:00Z",
  "updated_at": "2024-03-15T10:30:00Z"
}