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",
            "allow_multiple_instances": True,
        },
        {"name": "receipt", "description": "Receipt documents for payments"},
        {"name": "contract", "description": "Legal contract documents"},
    ],
    instructions="Processing Q4 2024 vendor document batch",
    n_consensus=3,
    bust_cache=False,
)

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

{
  "id": "split_01G34H8J2K",
  "file": {
    "id": "file_6dd6eb00688ad8d1",
    "filename": "invoice_batch.pdf",
    "mime_type": "application/pdf"
  },
  "model": "retab-small",
  "subdocuments": [
    {
      "name": "invoice",
      "description": "Invoice documents with billing information",
      "allow_multiple_instances": true
    },
    {
      "name": "receipt",
      "description": "Receipt documents for payments",
      "allow_multiple_instances": false
    }
  ],
  "n_consensus": 3,
  "instructions": "Processing Q4 2024 vendor document batch",
  "output": [
    {"name": "invoice", "pages": [1, 2, 3]},
    {"name": "invoice", "pages": [4, 5]},
    {"name": "receipt", "pages": [6]}
  ],
  "consensus": {
    "likelihoods": [
      {"name": 0.98, "pages": [0.99, 0.97, 0.96]},
      {"name": 0.95, "pages": [0.95, 0.93]},
      {"name": 0.99, "pages": [0.99]}
    ],
    "choices": [
      [
        {"name": "invoice", "pages": [1, 2, 3]},
        {"name": "invoice", "pages": [4, 5]},
        {"name": "receipt", "pages": [6]}
      ],
      [
        {"name": "invoice", "pages": [1, 2]},
        {"name": "invoice", "pages": [3, 4, 5]},
        {"name": "receipt", "pages": [6]}
      ],
      [
        {"name": "invoice", "pages": [1, 2, 3]},
        {"name": "invoice", "pages": [4, 5]},
        {"name": "receipt", "pages": [6]}
      ]
    ]
  },
  "usage": {
    "credits": 3.0
  },
  "created_at": "2024-03-15T10:30:00Z",
  "updated_at": "2024-03-15T10:30:00Z"
}