Dinesh310 commited on
Commit
e694569
·
verified ·
1 Parent(s): 4c645b4

Update src/core/embeddings.py

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