File size: 389 Bytes
094f515
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
from langchain.embeddings import HuggingFaceEmbeddings

def get_hf_embeddings(model_name=None):
    """Retrieve a Hugging Face embedding model using the specified model name.."""

    if model_name is None:
        # "sentence-transformers/all-mpnet-base-v2"
        model_name = "BAAI/bge-base-en-v1.5"

    embeddings = HuggingFaceEmbeddings(model_name=model_name)

    return embeddings