models/distance.py

import enum


distance_strategy_limit = {
    "l2": 1.05,
    "cosine": 0.6,
    "inner": 1.0,
}


class DistanceStrategy(str, enum.Enum):
    """Enumerator of the Distance strategies."""

    EUCLIDEAN = "l2"
    COSINE = "cosine"
    MAX_INNER_PRODUCT = "inner"