from retab import Retab

client = Retab()

partition = client.partitions.get("part_01G34H8J2K")
for chunk in partition.output:
    print(chunk.key, chunk.pages)

{
  "id": "part_01G34H8J2K",
  "file": {
    "id": "file_6dd6eb00688ad8d1",
    "filename": "invoice_batch.pdf",
    "mime_type": "application/pdf"
  },
  "model": "retab-small",
  "key": "invoice_number",
  "instructions": "Return one chunk per invoice number and keep all pages for the same invoice together.",
  "n_consensus": 3,
  "allow_overlap": true,
  "output": [
    {"key": "INV-001", "pages": [1, 2]},
    {"key": "INV-002", "pages": [3, 4]}
  ],
  "consensus": {
    "likelihoods": [
      {"key": 0.99, "pages": [0.99, 0.98]},
      {"key": 0.96, "pages": [0.95, 0.95]}
    ],
    "choices": [
      [
        {"key": "INV-001", "pages": [1, 2]},
        {"key": "INV-002", "pages": [3, 4]}
      ]
    ]
  },
  "usage": {
    "credits": 3.0
  },
  "created_at": "2024-03-15T10:30:00Z"
}