Skip to content

pixano_inference.models.capabilities

Capability helpers for HTTP-exposed inference model families.

infer_http_capability(model_cls)

Infer the HTTP capability implemented by a model class.

Source code in pixano_inference/models/capabilities.py 
26
27
28
29
30
31
32
33
34
35
36
37
38
def infer_http_capability(model_cls: type[InferenceModel]) -> str:
    """Infer the HTTP capability implemented by a model class."""
    for base_cls in HTTP_CAPABILITY_BASES:
        if issubclass(model_cls, base_cls):
            capability = getattr(base_cls, "capability_name", None)
            if capability is not None:
                return capability

    supported = ", ".join(base.__name__ for base in HTTP_CAPABILITY_BASES)
    raise ValueError(
        f"Model class '{model_cls.__name__}' is not supported by the HTTP inference API. "
        f"Supported base classes: {supported}."
    )