Source code for nlp_shap.domain.generation

"""Generation payloads scored by value functions."""

from dataclasses import dataclass



[docs]
@dataclass(frozen=True)
class GenerationRecord:
    """Concrete generation output consumed by built-in value functions."""

    text: str
    """Generated text payload."""

    text_token_rows: tuple[tuple[int, ...], ...] = ()
    """Token rows hashed for TF-IDF similarity (multimodal text stream)."""

    audio_token_rows: tuple[tuple[int, ...], ...] = ()
    """Token rows hashed for TF-IDF similarity (multimodal audio stream)."""

    logprobs: tuple[float, ...] = ()
    """Per-token log probabilities when the backend exposes them."""

    embedding: tuple[float, ...] = ()
    """Static embedding vector for embedding-based value functions."""

    contextual_embedding: tuple[float, ...] = ()
    """Contextual embedding vector for U2-style scoring."""