Classify

from ndi_sdk import NdiClient
from ndi_sdk.models.document_ops import ClassifyClass

with NdiClient() as client:
    document = client.documents.create_upload("invoice.pdf")
    job = client.jobs.wait(
        client.documents.classify(
            document,
            classes=[
                ClassifyClass(id="invoice", label="Invoice", description="A supplier invoice"),
            ],
            wait_seconds=30,
        ).job_id
    )
    for unit in job.result.units:
        print(unit.unknown, unit.labels)

def classify(
    source: UploadSource | UrlSource | WorkspaceFileSource | UploadResponse,
    *,
    classes: list[ClassifyClass],
    granularity: Literal["document", "page"] = "document",
    page_ranges: list[PageRange] | None = None,
    unknown_policy: Literal["allow", "force_best"] = "allow",
    output: ClassifyOutputOptions | None = None,
    wait_seconds: int = 0,
    idempotency_key: str | None = None,
) -> Job