Document parsing as a tool your agent can call.

import json
from datalab_sdk import DatalabClient, ExtractOptions

client = DatalabClient()

def extract_invoice(file_path: str) -> dict:
    """Tool exposed to the agent. Returns typed fields with citations."""
    schema = {
        "type": "object",
        "properties": {
            "invoice_number": {"type": "string"},
            "total_amount":   {"type": "number"},
            "vendor_name":    {"type": "string"},
        },
    }
    options = ExtractOptions(page_schema=json.dumps(schema), mode="balanced")
    result = client.extract(file_path, options=options)
    return json.loads(result.extraction_schema_json)

# The SDK submits the job, then polls until the typed result is ready.
# Each field arrives with `*_citations` referencing source blocks.