embeddings.py

from langchain.embeddings import HuggingFaceEmbeddings

# Load Hugging Face model for embedding generation
embedding_model = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")

def get_embedding(text):
    """Generate embeddings for a given text."""
    return embedding_model.embed_query(text)