datdevsteve commited on
Commit
2bdd8d2
·
verified ·
1 Parent(s): 9f4943c

Update agent/rag_retriever.py

Browse files
Files changed (1) hide show
  1. agent/rag_retriever.py +1 -1
agent/rag_retriever.py CHANGED
@@ -10,7 +10,7 @@ class NivraRAGRetriever:
10
  )
11
  self.vectorstore = Chroma(
12
  collection_name="nivra_medical_docs",
13
- embedding_function=your_embeddings
14
  )
15
  self.retriever = self.vectorstore.as_retriever(
16
  search_type="similarity_score_threshold",
 
10
  )
11
  self.vectorstore = Chroma(
12
  collection_name="nivra_medical_docs",
13
+ embedding_function=self.embeddings
14
  )
15
  self.retriever = self.vectorstore.as_retriever(
16
  search_type="similarity_score_threshold",