"""Schema routing for the quantized multilingual extraction runtime."""

from gliner2 import Schema

from extraction_runtime import CoreMLAdaptiveBoundaryExtractor


def record_schema(mode: str):
    schema = Schema()
    schema.structure("employment", mode=mode).field("person", dtype="str").field("company", dtype="str")
    return schema


def test_latent_record_uses_fp32():
    assert CoreMLAdaptiveBoundaryExtractor.selected_precision(record_schema("latent")) == "fp32"


def test_other_selected_schemas_use_fp16():
    assert CoreMLAdaptiveBoundaryExtractor.selected_precision(Schema().entities(["person"])) == "fp16"
    assert CoreMLAdaptiveBoundaryExtractor.selected_precision(record_schema("natural")) == "fp16"
    assert CoreMLAdaptiveBoundaryExtractor.selected_precision(record_schema("anchorless")) == "fp16"