MohitRajput45 commited on
Commit
117b594
·
verified ·
1 Parent(s): 9ab0852

Update src/rag_engine/build_vector_db.py

Browse files
Files changed (1) hide show
  1. src/rag_engine/build_vector_db.py +3 -1
src/rag_engine/build_vector_db.py CHANGED
@@ -28,7 +28,9 @@ class MindGuardVectorDB:
28
 
29
  # Initialize the Embedding Engine (all-MiniLM-L6-v2)
30
  # This is a small, lightning-fast Hugging Face model that turns sentences into math vectors
31
- self.embedding_fn = embedding_functions.DefaultEmbeddingFunction()
 
 
32
 
33
  # Create or load the 'clinical_guidelines' collection
34
  self.collection = self.chroma_client.get_or_create_collection(
 
28
 
29
  # Initialize the Embedding Engine (all-MiniLM-L6-v2)
30
  # This is a small, lightning-fast Hugging Face model that turns sentences into math vectors
31
+ self.embedding_fn = embedding_functions.SentenceTransformerEmbeddingFunction(
32
+ model_name="BAAI/bge-base-en-v1.5"
33
+ )
34
 
35
  # Create or load the 'clinical_guidelines' collection
36
  self.collection = self.chroma_client.get_or_create_collection(