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)