src/binary_shield/embedding.py

import numpy as np
from sentence_transformers import SentenceTransformer


def extract_embedding(text: str, model: SentenceTransformer) -> np.ndarray:
    return model.encode(text, convert_to_numpy=True)