src/embedder.py

from sentence_transformers import SentenceTransformer


model = SentenceTransformer("all-MiniLM-L6-v2")


def embed_documents(documents):

    embeddings = model.encode(documents, show_progress_bar=True)

    return embeddings