from enum import StrEnum


class Keys(StrEnum):
    DIAG = "diag"
    DOC = "doc"


class Metrics(StrEnum):
    PR = "precision"
    RE = "recall"
    F1 = "f1"


class ScoreNames(StrEnum):
    OPT = "optimistic"
    AVG = "averaged"
    REAL = "realistic"


class Models(StrEnum):
    GIGA_MAX = "giga_max"
    GIGA_PLUS = "giga_plus"
    GIGA_PRO = "giga_pro"
    qwen = "qwen"
    deepseekr1 = "deepseekr1"
    deepseekv3 = "deepseekv3"
    llama = "llama_405b"
    mistral = "mistral"
    gpt_4o = "gpt4o"
    deepseekr1distqwen32b = "deepseekr1distqwen32b"


class Paths:
    CHATS = "../data/input/chats_1-360.json"
    PREDICTS = "../data/input/predicts_1-360"
    TARGETS = "../data/input/targets_1-360.json"
    PAIRS = "../data/input/pair-match.json"
    ONE_VS_ONE = "../data/input/one-vs-one"
    METRICS = "../data/output/metrics_1-360.json"
    PAIR_METRICS = "../data/output/pair_metrics_1-360.json"
    PREPROCESSOR = "../data/input/preprocessor.json"