Spaces:
Runtime error
Runtime error
chore: Update chroma_retriever search type to "similarity"
Browse files
app.py
CHANGED
|
@@ -154,7 +154,9 @@ chroma_db = Chroma(
|
|
| 154 |
|
| 155 |
# Create retrievers
|
| 156 |
faiss_retriever = faiss_db.as_retriever(search_type="mmr", search_kwargs={"k": 10})
|
| 157 |
-
chroma_retriever = chroma_db.as_retriever(
|
|
|
|
|
|
|
| 158 |
bm25_retriever = BM25Retriever.from_documents(combined_documents)
|
| 159 |
bm25_retriever.k = 10
|
| 160 |
ensemble_retriever = EnsembleRetriever(
|
|
|
|
| 154 |
|
| 155 |
# Create retrievers
|
| 156 |
faiss_retriever = faiss_db.as_retriever(search_type="mmr", search_kwargs={"k": 10})
|
| 157 |
+
chroma_retriever = chroma_db.as_retriever(
|
| 158 |
+
search_type="similarity", search_kwargs={"k": 10}
|
| 159 |
+
)
|
| 160 |
bm25_retriever = BM25Retriever.from_documents(combined_documents)
|
| 161 |
bm25_retriever.k = 10
|
| 162 |
ensemble_retriever = EnsembleRetriever(
|