Spaces:
Sleeping
Sleeping
Update agent/rag_retriever.py
Browse files- 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=
|
| 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",
|