src/api/classify/results.py

from __future__ import annotations
from dataclasses import dataclass


@dataclass(frozen=True, slots=True)
class StageTimings:
    total_ms: int
    domain_ms: int
    labels_ms: int


@dataclass(frozen=True, slots=True)
class ClassificationResult:
    domain_hits: list[tuple[str, float]]
    chosen_domains: list[str]
    label_hits: list[tuple[str, float]]
    timings: StageTimings