from retab import Retab

client = Retab()

# Update predictions after human review
updated = client.extractions.update(
    extraction_id="extr_01G34H8J2K",
    predictions={
        "invoice_number": "INV-2024-0789-CORRECTED",
        "total_amount": 1576.75
    },
    human_review_status="reviewed"
)

# Update only the review status
updated = client.extractions.update(
    extraction_id="extr_01G34H8J2K",
    human_review_status="reviewed"
)

# Update metadata
updated = client.extractions.update(
    extraction_id="extr_01G34H8J2K",
    metadata={"reviewed_by": "[email protected]"}
)

{
  "id": "extr_01G34H8J2K",
  "created_at": "2024-03-15T10:30:00Z",
  "updated_at": "2024-03-15T14:45:00Z",
  "organization_id": "org_01ABC123",
  "origin": {
    "type": "project",
    "id": "proj_F0FE8DFqyouQdZXDTWRg0"
  },
  "file": {
    "id": "file_6dd6eb00688ad8d1",
    "filename": "invoice.pdf"
  },
  "predictions": {
    "invoice_number": "INV-2024-0789-CORRECTED",
    "total_amount": 1576.75
  },
  "human_review_status": "reviewed",
  "json_schema": {
    "type": "object",
    "properties": {
      "invoice_number": { "type": "string" },
      "total_amount": { "type": "number" }
    }
  },
  "metadata": {
    "reviewed_by": "[email protected]"
  }
}