| # embeddings.py | |
| from langchain_community.embeddings import HuggingFaceBgeEmbeddings | |
| from config import EMBEDDING_MODEL | |
| from utils import get_device | |
| def get_embeddings(): | |
| return HuggingFaceBgeEmbeddings( | |
| model_name=EMBEDDING_MODEL, | |
| model_kwargs={"device": get_device()}, | |
| encode_kwargs={"normalize_embeddings": True}, | |
| ) | |