src/legisqa_local/core/embeddings.py

"""Embedding functionality for LegisQA"""

from langchain_huggingface import HuggingFaceEmbeddings
from legisqa_local.config.settings import EMBEDDING_MODEL, EMBEDDING_DEVICE


def load_embeddings():
    """Load and return the embedding function"""
    model_kwargs = {"device": EMBEDDING_DEVICE}
    emb_fn = HuggingFaceEmbeddings(
        model_name=EMBEDDING_MODEL,
        model_kwargs=model_kwargs,
    )
    return emb_fn