KadiAPY_Coding_Assistant / embeddings.py
bupa1018's picture
Create embeddings.py
094f515
raw
history blame
389 Bytes
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