from retab import Retab

client = Retab()

response = client.partitions.create(
    document="invoice_batch.pdf",
    key="invoice_number",
    instructions="Return one chunk per invoice number and keep all pages for the same invoice together.",
    model="retab-small",
    n_consensus=3,
    allow_overlap=True,
    bust_cache=False,
)

for chunk in response.output:
    print(chunk.key, chunk.pages)

{
  "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] }
      ],
      [
        { "key": "INV-001", "pages": [1, 2] },
        { "key": "INV-002", "pages": [3, 4] }
      ],
      [
        { "key": "INV-001", "pages": [1, 2] },
        { "key": "INV-002", "pages": [3, 4] }
      ]
    ]
  },
  "usage": {
    "credits": 3.0
  }
}