from django.db.models.fields import UUIDField


def get_entity_pk_type(entity_cls) -> str:
    """Returns the entity PK type to use.

    These values map to `models.Value` as potential fields to use to relate
    to the proper entity via the correct PK type.
    """
    if isinstance(entity_cls._meta.pk, UUIDField):  # noqa: SLF001
        return "entity_uuid"
    return "entity_id"