Dinesh310 commited on
Commit
57ea49c
·
verified ·
1 Parent(s): ee784d3

Create core/embeddings.py

Browse files
Files changed (1) hide show
  1. src/core/embeddings.py +12 -0
src/core/embeddings.py ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from langchain_huggingface import HuggingFaceEmbeddings
2
+ from config.config import Config
3
+
4
+ def load_embeddings():
5
+ try:
6
+ return HuggingFaceEmbeddings(
7
+ model_name=Config.EMBEDDING_MODEL,
8
+ model_kwargs={"device": Config.EMBEDDING_DEVICE},
9
+ encode_kwargs={"normalize_embeddings": Config.NORMALIZE_EMBEDDINGS}
10
+ )
11
+ except Exception as e:
12
+ raise RuntimeError(f"Failed to load embeddings: {e}")