"""Text encoder."""

from sentence_transformers import SentenceTransformer
from torch import Tensor


class TextEncoder:
    """Text encoder."""

    def __init__(self) -> None:
        """Initialize the text encoder."""
        self.model = SentenceTransformer("sentence-transformers/all-MiniLM-L12-v1")

    def encode(self, text: str) -> Tensor:
        """Encode a string into an embedding."""
        return self.model.encode(text)